dnet-hadoop/dhp-pace-core/src/main/java/eu/dnetlib/pace/tree/support/TreeNodeStats.java

package eu.dnetlib.pace.tree.support;

import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;

public class TreeNodeStats implements Serializable {

	private Map<String, FieldStats> results; // this is an accumulator for the results of the node

	public TreeNodeStats() {
		this.results = new HashMap<>();
	}

	public Map<String, FieldStats> getResults() {
		return results;
	}

	public void addFieldStats(String id, FieldStats fieldStats) {
		this.results.put(id, fieldStats);
	}

	public int fieldsCount() {
		return this.results.size();
	}

	public int undefinedCount() {
		int undefinedCount = 0;
		for (FieldStats fs : this.results.values()) {
			if (fs.getResult() == -1)
				undefinedCount++;
		}
		return undefinedCount;
	}

	public double scoreSum() {
		double scoreSum = 0.0;
		for (FieldStats fs : this.results.values()) {
			if (fs.getResult() >= 0.0) {
				scoreSum += fs.getResult();
			}
		}
		return scoreSum;
	}

	// return the sum of the weights without considering the fields with countIfMissing=false && result=-1
	public double weightSum() {
		double weightSum = 0.0;
		for (FieldStats fs : this.results.values()) {
			if (fs.getResult() >= 0.0 || (fs.getResult() < 0.0 && fs.isCountIfUndefined())) {
				weightSum += fs.getWeight();
			}
		}
		return weightSum;
	}

	public double weightedScoreSum() {
		double weightedScoreSum = 0.0;
		for (FieldStats fs : this.results.values()) {
			if (fs.getResult() >= 0.0) {
				weightedScoreSum += fs.getResult() * fs.getWeight();
			}
		}
		return weightedScoreSum;
	}

	public double max() {
		double max = -1.0;
		for (FieldStats fs : this.results.values()) {
			if (fs.getResult() > max)
				max = fs.getResult();
		}
		return max;
	}

	public double min() {
		double min = 100.0; // random high value
		for (FieldStats fs : this.results.values()) {
			if (fs.getResult() < min) {
				if (fs.getResult() >= 0.0 || (fs.getResult() == -1 && fs.isCountIfUndefined()))
					min = fs.getResult();
			}
		}
		return min;
	}

	// if at least one is true, return 1.0
	public double or() {
		for (FieldStats fieldStats : this.results.values()) {
			if (fieldStats.getResult() >= fieldStats.getThreshold())
				return 1.0;
		}
		return 0.0;
	}

	// if at least one is false, return 0.0
	public double and() {
		for (FieldStats fieldStats : this.results.values()) {

			if (fieldStats.getResult() == -1) {
				if (fieldStats.isCountIfUndefined())
					return 0.0;
			} else {
				if (fieldStats.getResult() < fieldStats.getThreshold())
					return 0.0;
			}

		}
		return 1.0;
	}

	public double getFinalScore(AggType aggregation) {

		switch (aggregation) {
			case AVG:
				return scoreSum() / fieldsCount();
			case SUM:
				return scoreSum();
			case MAX:
				return max();
			case MIN:
				return min();
			case W_MEAN:
				return weightedScoreSum() / weightSum();
			case OR:
				return or();
			case AND:
				return and();
			default:
				return 0.0;
		}
	}
}