diff options
Diffstat (limited to 'subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/DefaultMessageIndexer.java')
-rw-r--r-- | subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/DefaultMessageIndexer.java | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/DefaultMessageIndexer.java b/subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/DefaultMessageIndexer.java new file mode 100644 index 00000000..da9bc47e --- /dev/null +++ b/subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/DefaultMessageIndexer.java | |||
@@ -0,0 +1,74 @@ | |||
1 | /******************************************************************************* | ||
2 | * Copyright (c) 2010-2018, Tamas Szabo, Istvan Rath and Daniel Varro | ||
3 | * This program and the accompanying materials are made available under the | ||
4 | * terms of the Eclipse Public License v. 2.0 which is available at | ||
5 | * http://www.eclipse.org/legal/epl-v20.html. | ||
6 | * | ||
7 | * SPDX-License-Identifier: EPL-2.0 | ||
8 | *******************************************************************************/ | ||
9 | package tools.refinery.viatra.runtime.rete.network.indexer; | ||
10 | |||
11 | import java.util.Collections; | ||
12 | import java.util.Map; | ||
13 | |||
14 | import tools.refinery.viatra.runtime.matchers.tuple.Tuple; | ||
15 | import tools.refinery.viatra.runtime.matchers.util.CollectionsFactory; | ||
16 | |||
17 | /** | ||
18 | * @author Tamas Szabo | ||
19 | * @since 2.0 | ||
20 | */ | ||
21 | public class DefaultMessageIndexer implements MessageIndexer { | ||
22 | |||
23 | protected final Map<Tuple, Integer> indexer; | ||
24 | |||
25 | public DefaultMessageIndexer() { | ||
26 | this.indexer = CollectionsFactory.createMap(); | ||
27 | } | ||
28 | |||
29 | public Map<Tuple, Integer> getTuples() { | ||
30 | return Collections.unmodifiableMap(this.indexer); | ||
31 | } | ||
32 | |||
33 | @Override | ||
34 | public int getCount(final Tuple update) { | ||
35 | final Integer count = getTuples().get(update); | ||
36 | if (count == null) { | ||
37 | return 0; | ||
38 | } else { | ||
39 | return count; | ||
40 | } | ||
41 | } | ||
42 | |||
43 | @Override | ||
44 | public void insert(final Tuple update) { | ||
45 | update(update, 1); | ||
46 | } | ||
47 | |||
48 | @Override | ||
49 | public void delete(final Tuple update) { | ||
50 | update(update, -1); | ||
51 | } | ||
52 | |||
53 | @Override | ||
54 | public void update(final Tuple update, final int delta) { | ||
55 | final Integer oldCount = this.indexer.get(update); | ||
56 | final int newCount = (oldCount == null ? 0 : oldCount) + delta; | ||
57 | if (newCount == 0) { | ||
58 | this.indexer.remove(update); | ||
59 | } else { | ||
60 | this.indexer.put(update, newCount); | ||
61 | } | ||
62 | } | ||
63 | |||
64 | @Override | ||
65 | public boolean isEmpty() { | ||
66 | return this.indexer.isEmpty(); | ||
67 | } | ||
68 | |||
69 | @Override | ||
70 | public void clear() { | ||
71 | this.indexer.clear(); | ||
72 | } | ||
73 | |||
74 | } | ||