-
Notifications
You must be signed in to change notification settings - Fork 5
/
ComputeAllLevDistances.java
100 lines (85 loc) · 3.45 KB
/
ComputeAllLevDistances.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
package org.genericsystem.cv.comparator;
import java.util.List;
import org.genericsystem.api.core.Snapshot;
import org.genericsystem.common.Generic;
import org.genericsystem.cv.Levenshtein;
import org.genericsystem.cv.model.Doc;
import org.genericsystem.cv.model.Doc.DocInstance;
import org.genericsystem.cv.model.DocClass;
import org.genericsystem.cv.model.ImgFilter;
import org.genericsystem.cv.model.ImgFilter.ImgFilterInstance;
import org.genericsystem.cv.model.LevDistance;
import org.genericsystem.cv.model.ZoneGeneric;
import org.genericsystem.cv.model.ZoneGeneric.ZoneInstance;
import org.genericsystem.cv.model.ZoneText;
import org.genericsystem.cv.model.ZoneText.ZoneTextInstance;
import org.genericsystem.kernel.Engine;
import org.opencv.core.Core;
/**
* The ComputeAllLevDistances class computes the {@link LevDistance} between two
* {@link ZoneText}.
*
* The data is retrieved from GS, and stored in GS.
*
* @author Pierrik Lassalas
*
*/
public class ComputeAllLevDistances {
private final static String docType = "id-fr-front";
private static final String gsPath = System.getenv("HOME") + "/genericsystem/gs-cv_model/";
private final static Engine engine = new Engine(gsPath, Doc.class,
ImgFilter.class, ZoneGeneric.class, ZoneText.class, LevDistance.class);
static {
System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
}
public static void main(String[] mainArgs) {
engine.newCache().start();
compute();
engine.close();
}
@SuppressWarnings({ "unchecked", "rawtypes" })
public static void compute() {
Generic currentDocClass = engine.find(DocClass.class).getInstance(docType);
ImgFilter imgFilter = engine.find(ImgFilter.class);
ZoneText zoneText = engine.find(ZoneText.class);
LevDistance levDistance = engine.find(LevDistance.class);
System.out.println("Current doc class : " + currentDocClass);
Snapshot<DocInstance> docInstances = (Snapshot) currentDocClass.getHolders(engine.find(Doc.class));
Snapshot<ZoneInstance> zoneInstances = (Snapshot) currentDocClass.getHolders(engine.find(ZoneGeneric.class));
List<ImgFilterInstance> imgFilterInstances = (List) imgFilter.getInstances()
.filter(f -> !"reality".equals(f.getValue())).toList();
// Loop over all documents in this class
docInstances.forEach(docInstance -> {
System.out.println("=> Document " + docInstance);
// Loop over all zone instances
zoneInstances.forEach(zoneInstance -> {
System.out.println(" => Zone " + zoneInstance);
// Loop over all filters
for (int i = 0; i < imgFilterInstances.size(); ++i) {
System.out.println(" => Filter " + imgFilterInstances.get(i));
ZoneTextInstance zoneText1 = zoneText.getZoneText(docInstance, zoneInstance,
imgFilterInstances.get(i));
for (int j = i; j < imgFilterInstances.size(); ++j) {
int dist = 0;
if (i == j) {
levDistance.setLevDistance(0, zoneText1, zoneText1);
break;
}
ZoneTextInstance zoneText2 = zoneText.getZoneText(docInstance, zoneInstance,
imgFilterInstances.get(j));
String text1 = (String) zoneText1.getValue();
String text2 = (String) zoneText2.getValue();
dist = Levenshtein.distance(text1.trim(), text2.trim());
levDistance.setLevDistance(dist, zoneText1, zoneText2);
levDistance.setLevDistance(dist, zoneText2, zoneText1);
}
// Free some resources
System.gc();
System.runFinalization();
}
engine.getCurrentCache().flush();
});
});
engine.getCurrentCache().flush();
}
}