com.github.TKnudsen.ComplexDataObject.model.distanceMeasure.Double.WeightedEuclideanDistanceMeasure Maven / Gradle / Ivy

Go to download
package com.github.TKnudsen.ComplexDataObject.model.distanceMeasure.Double;

import java.util.List;

import com.github.TKnudsen.ComplexDataObject.model.distanceMeasure.WeightedDistanceMeasure;

/**
 * 
 * Title: WeightedEuclideanDistanceMeasure
 * 
 * 
 * 
 * Description:
 * 
 * 
 * 
 * Copyright: (c) 2016-2017 Juergen Bernard
 * 
 * 
 * @author Christian Ritter
 * @version 1.01
 */
public class WeightedEuclideanDistanceMeasure extends WeightedDistanceMeasure {

	public WeightedEuclideanDistanceMeasure(List weights) {
		super(weights);
	}

	public WeightedEuclideanDistanceMeasure(List weights, double nullValue) {
		super(weights, nullValue);
	}

	@Override
	public String getDescription() {
		return "Calculated the weighted Euclidean distance.";
	}

	@Override
	public double getDistance(double[] o1, double[] o2) {
		if (o1.length != o2.length || o1.length != getWeights().size()) {
			try {
				throw new Exception("The Arrays have different Sizes.");
			} catch (Exception e) {
				e.printStackTrace();
			}
			return Double.NaN;
		}

		int length = o1.length;
		double result = 0;

		for (int i = 0; i < length; i++) {
			if (!Double.isNaN(o1[i] + o2[i]))
				result += getWeights().get(i) * Math.pow((o1[i] - o2[i]), 2);
			else
				result += getNullValue() * getWeights().get(i);
		}

		return Math.sqrt(result);
	}

	@Override
	public String getName() {
		return "WeightedEuclidean";
	}

}