diff options
Diffstat (limited to 'subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/GroupBasedMessageIndexer.java')
-rw-r--r-- | subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/GroupBasedMessageIndexer.java | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/GroupBasedMessageIndexer.java b/subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/GroupBasedMessageIndexer.java new file mode 100644 index 00000000..80271252 --- /dev/null +++ b/subprojects/viatra-runtime-rete/src/main/java/tools/refinery/viatra/runtime/rete/network/indexer/GroupBasedMessageIndexer.java | |||
@@ -0,0 +1,95 @@ | |||
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 | import java.util.Set; | ||
14 | |||
15 | import tools.refinery.viatra.runtime.matchers.tuple.Tuple; | ||
16 | import tools.refinery.viatra.runtime.matchers.tuple.TupleMask; | ||
17 | import tools.refinery.viatra.runtime.matchers.util.CollectionsFactory; | ||
18 | |||
19 | /** | ||
20 | * @author Tamas Szabo | ||
21 | * @since 2.0 | ||
22 | */ | ||
23 | public class GroupBasedMessageIndexer implements MessageIndexer { | ||
24 | |||
25 | protected final Map<Tuple, DefaultMessageIndexer> indexer; | ||
26 | protected final TupleMask groupMask; | ||
27 | |||
28 | public GroupBasedMessageIndexer(final TupleMask groupMask) { | ||
29 | this.indexer = CollectionsFactory.createMap(); | ||
30 | this.groupMask = groupMask; | ||
31 | } | ||
32 | |||
33 | public Map<Tuple, Integer> getTuplesByGroup(final Tuple group) { | ||
34 | final DefaultMessageIndexer values = this.indexer.get(group); | ||
35 | if (values == null) { | ||
36 | return Collections.emptyMap(); | ||
37 | } else { | ||
38 | return Collections.unmodifiableMap(values.getTuples()); | ||
39 | } | ||
40 | } | ||
41 | |||
42 | @Override | ||
43 | public int getCount(final Tuple update) { | ||
44 | final Tuple group = this.groupMask.transform(update); | ||
45 | final Integer count = getTuplesByGroup(group).get(update); | ||
46 | if (count == null) { | ||
47 | return 0; | ||
48 | } else { | ||
49 | return count; | ||
50 | } | ||
51 | } | ||
52 | |||
53 | public Set<Tuple> getGroups() { | ||
54 | return Collections.unmodifiableSet(this.indexer.keySet()); | ||
55 | } | ||
56 | |||
57 | @Override | ||
58 | public void insert(final Tuple update) { | ||
59 | update(update, 1); | ||
60 | } | ||
61 | |||
62 | @Override | ||
63 | public void delete(final Tuple update) { | ||
64 | update(update, -1); | ||
65 | } | ||
66 | |||
67 | @Override | ||
68 | public void update(final Tuple update, final int delta) { | ||
69 | final Tuple group = this.groupMask.transform(update); | ||
70 | DefaultMessageIndexer valueIndexer = this.indexer.get(group); | ||
71 | |||
72 | if (valueIndexer == null) { | ||
73 | valueIndexer = new DefaultMessageIndexer(); | ||
74 | this.indexer.put(group, valueIndexer); | ||
75 | } | ||
76 | |||
77 | valueIndexer.update(update, delta); | ||
78 | |||
79 | // it may happen that the indexer becomes empty as a result of the update | ||
80 | if (valueIndexer.isEmpty()) { | ||
81 | this.indexer.remove(group); | ||
82 | } | ||
83 | } | ||
84 | |||
85 | @Override | ||
86 | public boolean isEmpty() { | ||
87 | return this.indexer.isEmpty(); | ||
88 | } | ||
89 | |||
90 | @Override | ||
91 | public void clear() { | ||
92 | this.indexer.clear(); | ||
93 | } | ||
94 | |||
95 | } | ||