56 lines
1.1 KiB
Java
56 lines
1.1 KiB
Java
|
|
package net.filebot.similarity;
|
|
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.TreeSet;
|
|
|
|
import net.filebot.format.PropertyBindings;
|
|
|
|
public class CrossPropertyMetric implements SimilarityMetric {
|
|
|
|
private SimilarityMetric metric;
|
|
|
|
public CrossPropertyMetric(SimilarityMetric metric) {
|
|
this.metric = metric;
|
|
}
|
|
|
|
public CrossPropertyMetric() {
|
|
this.metric = new StringEqualsMetric();
|
|
}
|
|
|
|
@Override
|
|
public float getSimilarity(Object o1, Object o2) {
|
|
Map<String, Object> m1 = getProperties(o1);
|
|
if (m1.isEmpty())
|
|
return 0;
|
|
|
|
Map<String, Object> m2 = getProperties(o2);
|
|
if (m2.isEmpty())
|
|
return 0;
|
|
|
|
// works with commons keys
|
|
Set<String> keys = new TreeSet<String>(String.CASE_INSENSITIVE_ORDER);
|
|
keys.addAll(m1.keySet());
|
|
keys.retainAll(m2.keySet());
|
|
if (keys.isEmpty())
|
|
return 0;
|
|
|
|
float feedback = 0;
|
|
for (String k : keys) {
|
|
try {
|
|
feedback += metric.getSimilarity(m1.get(k), m2.get(k));
|
|
} catch (Exception e) {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
return feedback / keys.size();
|
|
}
|
|
|
|
protected Map<String, Object> getProperties(Object object) {
|
|
return new PropertyBindings(object);
|
|
}
|
|
|
|
}
|