33 lines
646 B
Java
33 lines
646 B
Java
|
package eu.dnetlib.pace.distance.eval;
|
||
|
|
||
|
import java.util.HashMap;
|
||
|
|
||
|
/**
|
||
|
* Created by claudio on 10/03/16.
|
||
|
*/
|
||
|
public class DistanceEvalMap extends HashMap<String, DistanceEval> {
|
||
|
|
||
|
private double sumWeights;
|
||
|
|
||
|
private double sumDistances = 0.0;
|
||
|
|
||
|
public DistanceEvalMap(final double sumWeights) {
|
||
|
this.sumWeights = sumWeights;
|
||
|
}
|
||
|
|
||
|
public void updateDistance(final DistanceEval d) {
|
||
|
|
||
|
put(d.getFieldDef().getName(), d);
|
||
|
if (d.getDistance() >= 0) {
|
||
|
sumDistances += d.getDistance();
|
||
|
} else {
|
||
|
sumWeights -= d.getFieldDef().getWeight();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public double distance() {
|
||
|
return sumWeights == 0 ? 0 : sumDistances / sumWeights;
|
||
|
}
|
||
|
|
||
|
}
|