package tr.gov.ulakbim.jDenetX.cluster;

import tr.gov.ulakbim.jDenetX.AbstractMOAObject;
import weka.core.Instance;

import java.util.*;

public abstract class Cluster extends AbstractMOAObject {
    private double id = -1;
    private double gtLabel = -1;

    private HashMap<String, String> measure_values;


    public Cluster() {
        this.measure_values = new HashMap<String, String>();
    }

    /**
     * @return the center of the cluster
     */
    public abstract double[] getCenter();

    /**
     * Returns the weight of this cluster, not neccessarily normalized.
     * It could, for instance, simply return the number of points contined
     * in this cluster.
     *
     * @return the weight
     */
    public abstract double getWeight();

    /**
     * Returns the probability of the given point belonging to
     * this cluster.
     *
     * @param instance
     * @return a value between 0 and 1
     */
    public abstract double getInclusionProbability(Instance instance);


    //TODO: for non sphere cluster sample points, find out MIN MAX neighbours within cluster
    //and return the relative distance
    //public abstract double getRelativeHullDistance(Instance instance);

    @Override
    public void getDescription(StringBuilder sb, int i) {
        sb.append("Cluster Object");
    }

    public void setId(double id) {
        this.id = id;
    }

    public double getId() {
        return id;
    }

    public boolean isGroundTruth() {
        return gtLabel != -1;
    }

    public void setGroundTruth(double truth) {
        gtLabel = truth;
    }

    public double getGroundTruth() {
        return gtLabel;
    }


    /**
     * Samples this cluster by returning a point from inside it.
     *
     * @param random a random number source
     * @return an Instance that lies inside this cluster
     */
    public abstract Instance sample(Random random);


    public void setMeasureValue(String measureKey, String value) {
        measure_values.put(measureKey, value);
    }

    public void setMeasureValue(String measureKey, double value) {
        measure_values.put(measureKey, Double.toString(value));
    }


    public String getMeasureValue(String measureKey) {
        if (measure_values.containsKey(measureKey))
            return measure_values.get(measureKey);
        else
            return "";
    }


    protected void getClusterSpecificInfo(ArrayList<String> infoTitle, ArrayList<String> infoValue) {
        infoTitle.add("ClusterID");
        infoValue.add(Integer.toString((int) getId()));

        infoTitle.add("Type");
        infoValue.add(getClass().getSimpleName());

        double c[] = getCenter();
        if (c != null)
            for (int i = 0; i < c.length; i++) {
                infoTitle.add("Dim" + i);
                infoValue.add(Double.toString(c[i]));
            }

        infoTitle.add("Weight");
        infoValue.add(Double.toString(getWeight()));

    }

    /**
     *
     */
    public String getInfo() {
        ArrayList<String> infoTitle = new ArrayList<String>();
        ArrayList<String> infoValue = new ArrayList<String>();
        getClusterSpecificInfo(infoTitle, infoValue);

        StringBuffer sb = new StringBuffer();

        //Cluster properties
        sb.append("<html>");
        sb.append("<table>");
        int i = 0;
        while (i < infoTitle.size() && i < infoValue.size()) {
            sb.append("<tr><td>" + infoTitle.get(i) + "</td><td>" + infoValue.get(i) + "</td></tr>");
            i++;
        }
        sb.append("</table>");

        //Evaluation info
        sb.append("<br>");
        sb.append("<b>Evaluation</b><br>");
        sb.append("<table>");
        Iterator miterator = measure_values.entrySet().iterator();
        while (miterator.hasNext()) {
            Map.Entry e = (Map.Entry) miterator.next();
            sb.append("<tr><td>" + e.getKey() + "</td><td>" + e.getValue() + "</td></tr>");
        }
        sb.append("</table>");
        sb.append("</html>");
        return sb.toString();
    }

}