diff options
Diffstat (limited to 'subprojects/viatra-runtime/src/main/java/tools/refinery/viatra/runtime/matchers/memories/timely/TimelyUnaryMaskedTupleMemory.java')
-rw-r--r-- | subprojects/viatra-runtime/src/main/java/tools/refinery/viatra/runtime/matchers/memories/timely/TimelyUnaryMaskedTupleMemory.java | 133 |
1 files changed, 133 insertions, 0 deletions
diff --git a/subprojects/viatra-runtime/src/main/java/tools/refinery/viatra/runtime/matchers/memories/timely/TimelyUnaryMaskedTupleMemory.java b/subprojects/viatra-runtime/src/main/java/tools/refinery/viatra/runtime/matchers/memories/timely/TimelyUnaryMaskedTupleMemory.java new file mode 100644 index 00000000..178193af --- /dev/null +++ b/subprojects/viatra-runtime/src/main/java/tools/refinery/viatra/runtime/matchers/memories/timely/TimelyUnaryMaskedTupleMemory.java | |||
@@ -0,0 +1,133 @@ | |||
1 | /******************************************************************************* | ||
2 | * Copyright (c) 2010-2018, Gabor Bergmann, IncQuery Labs Ltd. | ||
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.matchers.memories.timely; | ||
10 | |||
11 | import java.util.Collection; | ||
12 | import java.util.Collections; | ||
13 | import java.util.Iterator; | ||
14 | import java.util.Map; | ||
15 | import java.util.Set; | ||
16 | |||
17 | import tools.refinery.viatra.runtime.matchers.tuple.ITuple; | ||
18 | import tools.refinery.viatra.runtime.matchers.tuple.Tuple; | ||
19 | import tools.refinery.viatra.runtime.matchers.tuple.TupleMask; | ||
20 | import tools.refinery.viatra.runtime.matchers.tuple.Tuples; | ||
21 | import tools.refinery.viatra.runtime.matchers.util.CollectionsFactory; | ||
22 | import tools.refinery.viatra.runtime.matchers.util.TimelyMemory; | ||
23 | import tools.refinery.viatra.runtime.matchers.util.timeline.Diff; | ||
24 | import tools.refinery.viatra.runtime.matchers.util.timeline.Timeline; | ||
25 | |||
26 | /** | ||
27 | * Timely specialization for unary mask. | ||
28 | * | ||
29 | * @author Tamas Szabo | ||
30 | * @since 2.3 | ||
31 | */ | ||
32 | public final class TimelyUnaryMaskedTupleMemory<Timestamp extends Comparable<Timestamp>> | ||
33 | extends AbstractTimelyMaskedMemory<Timestamp, Object> { | ||
34 | |||
35 | protected final int keyPosition; | ||
36 | |||
37 | public TimelyUnaryMaskedTupleMemory(final TupleMask mask, final Object owner, final boolean isLazy) { | ||
38 | super(mask, owner, isLazy); | ||
39 | if (1 != mask.getSize()) | ||
40 | throw new IllegalArgumentException(mask.toString()); | ||
41 | this.keyPosition = mask.indices[0]; | ||
42 | } | ||
43 | |||
44 | @Override | ||
45 | public Iterable<Tuple> getSignatures() { | ||
46 | return () -> { | ||
47 | final Iterator<Object> wrapped = this.memoryMap.keySet().iterator(); | ||
48 | return new Iterator<Tuple>() { | ||
49 | @Override | ||
50 | public boolean hasNext() { | ||
51 | return wrapped.hasNext(); | ||
52 | } | ||
53 | |||
54 | @Override | ||
55 | public Tuple next() { | ||
56 | final Object key = wrapped.next(); | ||
57 | return Tuples.staticArityFlatTupleOf(key); | ||
58 | } | ||
59 | }; | ||
60 | }; | ||
61 | } | ||
62 | |||
63 | @Override | ||
64 | public Diff<Timestamp> removeWithTimestamp(final Tuple tuple, final Tuple signature, final Timestamp timestamp) { | ||
65 | final Object key = tuple.get(keyPosition); | ||
66 | return removeInternal(key, tuple, timestamp); | ||
67 | } | ||
68 | |||
69 | @Override | ||
70 | public Diff<Timestamp> addWithTimestamp(final Tuple tuple, final Tuple signature, final Timestamp timestamp) { | ||
71 | final Object key = tuple.get(keyPosition); | ||
72 | return addInternal(key, tuple, timestamp); | ||
73 | } | ||
74 | |||
75 | @Override | ||
76 | public Collection<Tuple> get(final ITuple signature) { | ||
77 | return getInternal(signature.get(0)); | ||
78 | } | ||
79 | |||
80 | @Override | ||
81 | public Map<Tuple, Timeline<Timestamp>> getWithTimeline(final ITuple signature) { | ||
82 | return getWithTimestampInternal(signature.get(0)); | ||
83 | } | ||
84 | |||
85 | @Override | ||
86 | public boolean isPresentAtInfinity(ITuple signature) { | ||
87 | return isPresentAtInfinityInteral(signature.get(0)); | ||
88 | } | ||
89 | |||
90 | @Override | ||
91 | public Iterable<Tuple> getResumableSignatures() { | ||
92 | if (this.foldingStates == null || this.foldingStates.isEmpty()) { | ||
93 | return Collections.emptySet(); | ||
94 | } else { | ||
95 | return () -> { | ||
96 | final Iterator<Object> wrapped = this.foldingStates.firstEntry().getValue().iterator(); | ||
97 | return new Iterator<Tuple>() { | ||
98 | @Override | ||
99 | public boolean hasNext() { | ||
100 | return wrapped.hasNext(); | ||
101 | } | ||
102 | |||
103 | @Override | ||
104 | public Tuple next() { | ||
105 | final Object key = wrapped.next(); | ||
106 | return Tuples.staticArityFlatTupleOf(key); | ||
107 | } | ||
108 | }; | ||
109 | }; | ||
110 | } | ||
111 | } | ||
112 | |||
113 | @Override | ||
114 | public Map<Tuple, Map<Tuple, Diff<Timestamp>>> resumeAt(final Timestamp timestamp) { | ||
115 | final Map<Tuple, Map<Tuple, Diff<Timestamp>>> result = CollectionsFactory.createMap(); | ||
116 | final Timestamp resumableTimestamp = this.getResumableTimestamp(); | ||
117 | if (resumableTimestamp == null) { | ||
118 | throw new IllegalStateException("There is nothing to fold!"); | ||
119 | } else if (resumableTimestamp.compareTo(timestamp) != 0) { | ||
120 | throw new IllegalStateException("Expected to continue folding at " + resumableTimestamp + "!"); | ||
121 | } | ||
122 | |||
123 | final Set<Object> signatures = this.foldingStates.remove(timestamp); | ||
124 | for (final Object signature : signatures) { | ||
125 | final TimelyMemory<Timestamp> memory = this.memoryMap.get(signature); | ||
126 | final Map<Tuple, Diff<Timestamp>> diffMap = memory.resumeAt(resumableTimestamp); | ||
127 | result.put(Tuples.staticArityFlatTupleOf(signature), diffMap); | ||
128 | registerFoldingState(memory.getResumableTimestamp(), signature); | ||
129 | } | ||
130 | return result; | ||
131 | } | ||
132 | |||
133 | } | ||