aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/store-dse/src/main/java/tools/refinery/store/dse/modification/internal/ModificationAdapterImpl.java
blob: 4e77c4620198adebda8f3e17bfb057fe077aa0e4 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
 * SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package tools.refinery.store.dse.modification.internal;

import tools.refinery.store.adapter.ModelStoreAdapter;
import tools.refinery.store.dse.modification.DanglingEdges;
import tools.refinery.store.dse.modification.ModificationAdapter;
import tools.refinery.store.model.Interpretation;
import tools.refinery.store.model.Model;
import tools.refinery.store.representation.Symbol;
import tools.refinery.store.tuple.Tuple;
import tools.refinery.store.tuple.Tuple1;

import java.util.HashSet;

public class ModificationAdapterImpl implements ModificationAdapter {
	static final Symbol<Integer> NEXT_ID = Symbol.of("NEXT_ID", 0, Integer.class, 0);

	final ModelStoreAdapter storeAdapter;
	final Model model;
	Interpretation<Integer> nodeCountInterpretation;

	ModificationAdapterImpl(ModelStoreAdapter storeAdapter, Model model) {
		this.storeAdapter = storeAdapter;
		this.model = model;
		this.nodeCountInterpretation = model.getInterpretation(NEXT_ID);
	}

	@Override
	public Model getModel() {
		return model;
	}

	@Override
	public ModelStoreAdapter getStoreAdapter() {
		return storeAdapter;
	}

	@Override
	public int getModelSize() {
		return nodeCountInterpretation.get(Tuple.of());
	}

	@Override
	public Tuple1 createObject() {
		var newNodeId = getModelSize();
		nodeCountInterpretation.put(Tuple.of(), newNodeId + 1);
		return Tuple.of(newNodeId);
	}

	@Override
	public boolean deleteObject(Tuple tuple, DanglingEdges danglingEdges) {
		if (tuple.getSize() != 1) {
			throw new IllegalArgumentException("Tuple size must be 1");
		}
		int objectId = tuple.get(0);
		if (danglingEdges == DanglingEdges.DELETE) {
			deleteDanglingEdges(objectId);
		} else if (danglingEdges == DanglingEdges.FAIL && hasDanglingEdges(objectId)) {
			return false;

		}
		int modelSize = getModelSize();
		if (objectId == modelSize - 1) {
			nodeCountInterpretation.put(Tuple.of(), modelSize - 1);
		}
		return true;
	}

	private void deleteDanglingEdges(int objectId) {
		for (var symbol : model.getStore().getSymbols()) {
			deleteDanglingEdges(objectId, (Symbol<?>) symbol);
		}
	}

	private <T> void deleteDanglingEdges(int objectId, Symbol<T> symbol) {
		var interpretation = model.getInterpretation(symbol);
		var toDelete = new HashSet<Tuple>();
		int arity = symbol.arity();
		for (int i = 0; i < arity; i++) {
			var cursor = interpretation.getAdjacent(i, objectId);
			while (cursor.move()) {
				toDelete.add(cursor.getKey());
			}
		}
		var defaultValue = symbol.defaultValue();
		for (var tuple : toDelete) {
			interpretation.put(tuple, defaultValue);
		}
	}

	private boolean hasDanglingEdges(int objectId) {
		for (var symbol : model.getStore().getSymbols()) {
			var interpretation = model.getInterpretation(symbol);
			int arity = symbol.arity();
			for (int i = 0; i < arity; i++) {
				if (interpretation.getAdjacentSize(i, objectId) > 0) {
					return true;
				}
			}
		}
		return false;
	}
}