78 lines
2.1 KiB
Java
78 lines
2.1 KiB
Java
|
|
package eu.dnetlib.dhp.broker.oa.util;
|
|
|
|
import java.io.IOException;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
|
import eu.dnetlib.broker.objects.OaBrokerMainEntity;
|
|
import eu.dnetlib.pace.config.DedupConfig;
|
|
import eu.dnetlib.pace.model.MapDocument;
|
|
import eu.dnetlib.pace.tree.support.TreeProcessor;
|
|
import eu.dnetlib.pace.util.MapDocumentUtil;
|
|
|
|
public class TrustUtils {
|
|
|
|
private static final Logger log = LoggerFactory.getLogger(TrustUtils.class);
|
|
|
|
private static DedupConfig dedupConfig;
|
|
|
|
static {
|
|
final ObjectMapper mapper = new ObjectMapper();
|
|
try {
|
|
dedupConfig = mapper
|
|
.readValue(
|
|
DedupConfig.class.getResourceAsStream("/eu/dnetlib/dhp/broker/oa/dedupConfig/dedupConfig.json"),
|
|
DedupConfig.class);
|
|
} catch (final IOException e) {
|
|
log.error("Error loading dedupConfig, e");
|
|
}
|
|
|
|
}
|
|
|
|
protected static float calculateTrust(final OaBrokerMainEntity r1, final OaBrokerMainEntity r2) {
|
|
|
|
if (dedupConfig == null) {
|
|
return BrokerConstants.MIN_TRUST;
|
|
}
|
|
|
|
try {
|
|
final ObjectMapper objectMapper = new ObjectMapper();
|
|
final MapDocument doc1 = MapDocumentUtil
|
|
.asMapDocumentWithJPath(dedupConfig, objectMapper.writeValueAsString(r1));
|
|
final MapDocument doc2 = MapDocumentUtil
|
|
.asMapDocumentWithJPath(dedupConfig, objectMapper.writeValueAsString(r2));
|
|
|
|
final double score = new TreeProcessor(dedupConfig).computeScore(doc1, doc2);
|
|
|
|
final double threshold = dedupConfig.getWf().getThreshold();
|
|
|
|
return TrustUtils.rescale(score, threshold);
|
|
} catch (final Exception e) {
|
|
log.error("Error computing score between results", e);
|
|
return BrokerConstants.MIN_TRUST;
|
|
}
|
|
}
|
|
|
|
public static float rescale(final double score, final double threshold) {
|
|
if (score >= BrokerConstants.MAX_TRUST) {
|
|
return BrokerConstants.MAX_TRUST;
|
|
}
|
|
|
|
final double val = (score - threshold) * (BrokerConstants.MAX_TRUST - BrokerConstants.MIN_TRUST)
|
|
/ (BrokerConstants.MAX_TRUST - threshold);
|
|
|
|
if (val < BrokerConstants.MIN_TRUST) {
|
|
return BrokerConstants.MIN_TRUST;
|
|
}
|
|
if (val > BrokerConstants.MAX_TRUST) {
|
|
return BrokerConstants.MAX_TRUST;
|
|
}
|
|
|
|
return (float) val;
|
|
}
|
|
}
|