-
Notifications
You must be signed in to change notification settings - Fork 14
/
IntegrityConstraintSet.java
124 lines (104 loc) · 4.96 KB
/
IntegrityConstraintSet.java
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/**
* Copyright (C) 2022 Czech Technical University in Prague
*
* This program is free software: you can redistribute it and/or modify it under
* the terms of the GNU General Public License as published by the Free Software
* Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
* details. You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package cz.cvut.kbss.jopa.owl2java;
import cz.cvut.kbss.jopa.ic.api.*;
import org.semanticweb.owlapi.model.OWLClass;
import org.semanticweb.owlapi.model.OWLDataProperty;
import org.semanticweb.owlapi.model.OWLObjectProperty;
import java.util.*;
public class IntegrityConstraintSet {
private final Map<OWLClass, List<IntegrityConstraint>> cConstraints = new HashMap<>();
private final Map<OWLClass, Map<OWLObjectProperty, Collection<IntegrityConstraint>>> opConstraints = new HashMap<>();
private final Map<OWLClass, Map<OWLDataProperty, Collection<IntegrityConstraint>>> dpConstraints = new HashMap<>();
private static <K, V> void addToMap(final K key, final V value, final Map<K, Collection<V>> map) {
if (!map.containsKey(key)) {
map.put(key, new HashSet<>());
}
map.get(key).add(value);
}
public void addIntegrityConstraint(final IntegrityConstraint ic) {
ic.accept(new IntegrityConstraintVisitor() {
@Override
public void visit(AtomicSubClassConstraint cpc) {
List<IntegrityConstraint> setC =
cConstraints.computeIfAbsent(cpc.getSubClass(), k -> new ArrayList<>());
setC.add(cpc);
}
@Override
public void visit(DataParticipationConstraint cpc) {
addToClassDataIntegrityConstraintSet(cpc.getSubject(), cpc.getPredicate(), cpc);
}
@Override
public void visit(ObjectParticipationConstraint cpc) {
addToClassObjectIntegrityConstraintSet(cpc.getSubject(), cpc.getPredicate(), cpc);
}
@Override
public void visit(ObjectDomainConstraint cpc) {
addToClassObjectIntegrityConstraintSet(cpc.getDomain(), cpc.getProperty(), cpc);
}
@Override
public void visit(ObjectRangeConstraint cpc) {
addToClassObjectIntegrityConstraintSet(cpc.getOWLClass(), cpc.getProperty(), cpc);
}
@Override
public void visit(DataDomainConstraint cpc) {
addToClassDataIntegrityConstraintSet(cpc.getDomain(), cpc.getProperty(), cpc);
}
@Override
public void visit(DataRangeConstraint cpc) {
addToClassDataIntegrityConstraintSet(cpc.getOWLClass(), cpc.getProperty(), cpc);
}
});
}
private Map<OWLObjectProperty, Collection<IntegrityConstraint>> addToClassObjectIntegrityConstraintSet(
final OWLClass subjClass, final OWLObjectProperty p, final IntegrityConstraint ic) {
Map<OWLObjectProperty, Collection<IntegrityConstraint>> map = opConstraints
.computeIfAbsent(subjClass, k -> new HashMap<>());
addToMap(p, ic, map);
return map;
}
private Map<OWLDataProperty, Collection<IntegrityConstraint>> addToClassDataIntegrityConstraintSet(
final OWLClass subjClass, final OWLDataProperty p, final IntegrityConstraint ic) {
Map<OWLDataProperty, Collection<IntegrityConstraint>> map = dpConstraints
.computeIfAbsent(subjClass, k -> new HashMap<>());
addToMap(p, ic, map);
return map;
}
public List<IntegrityConstraint> getClassIntegrityConstraints(final OWLClass cls) {
return cConstraints.getOrDefault(cls, Collections.emptyList());
}
public Collection<IntegrityConstraint> getClassObjectIntegrityConstraints(
final OWLClass clazz,
final OWLObjectProperty prop) {
final Map<OWLObjectProperty, Collection<IntegrityConstraint>> constraints = opConstraints
.get(clazz);
if (constraints != null && constraints.containsKey(prop)) {
return constraints.get(prop);
} else {
return Collections.emptyList();
}
}
public Collection<IntegrityConstraint> getClassDataIntegrityConstraints(
final OWLClass clazz,
final OWLDataProperty prop) {
final Map<OWLDataProperty, Collection<IntegrityConstraint>> constraints = dpConstraints
.get(clazz);
if (constraints != null && constraints.containsKey(prop)) {
return constraints.get(prop);
} else {
return Collections.emptyList();
}
}
}