forked from D-Net/dnet-hadoop
53 lines
1.3 KiB
Java
53 lines
1.3 KiB
Java
package eu.dnetlib.pace.condition;
|
|
|
|
import java.util.List;
|
|
import eu.dnetlib.pace.common.AbstractPaceFunctions;
|
|
import eu.dnetlib.pace.config.Cond;
|
|
import eu.dnetlib.pace.distance.eval.ConditionEval;
|
|
import eu.dnetlib.pace.distance.eval.ConditionEvalMap;
|
|
import eu.dnetlib.pace.model.Document;
|
|
import eu.dnetlib.pace.model.Field;
|
|
import eu.dnetlib.pace.model.FieldDef;
|
|
|
|
/**
|
|
* Abstract conditions needs a list of field names.
|
|
*
|
|
* @author claudio
|
|
*
|
|
*/
|
|
public abstract class AbstractCondition extends AbstractPaceFunctions implements ConditionAlgo {
|
|
|
|
protected Cond cond;
|
|
|
|
protected List<FieldDef> fields;
|
|
|
|
public AbstractCondition(final Cond cond, final List<FieldDef> fields) {
|
|
this.cond = cond;
|
|
this.fields = fields;
|
|
}
|
|
|
|
protected abstract ConditionEval verify(FieldDef fd, Field a, Field b);
|
|
|
|
@Override
|
|
public ConditionEvalMap verify(final Document a, final Document b) {
|
|
final ConditionEvalMap res = new ConditionEvalMap();
|
|
for (final FieldDef fd : getFields()) {
|
|
|
|
final Field va = a.values(fd.getName());
|
|
final Field vb = b.values(fd.getName());
|
|
|
|
if ((va.isEmpty() || vb.isEmpty()) && fd.isIgnoreMissing()) {
|
|
res.put(fd.getName(), new ConditionEval(cond, va, vb, 0));
|
|
} else {
|
|
res.put(fd.getName(), verify(fd, va, vb));
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
public List<FieldDef> getFields() {
|
|
return fields;
|
|
}
|
|
|
|
}
|