package de.ugoe.cs.autoquest.tasktrees.alignment.matrix;


import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;

import de.ugoe.cs.autoquest.eventcore.guimodel.IGUIElement;
import de.ugoe.cs.autoquest.tasktrees.alignment.algorithms.AlignmentHelpers;
import de.ugoe.cs.autoquest.tasktrees.alignment.algorithms.Constants;
import de.ugoe.cs.autoquest.tasktrees.treeifc.IEventTaskInstance;
import de.ugoe.cs.autoquest.tasktrees.treeifc.ITask;
import de.ugoe.cs.autoquest.tasktrees.treeifc.ITaskInstance;


public class ObjectDistanceSubstitionMatrix implements SubstitutionMatrix {

	HashMap<Integer, Integer> idmapping;
	private TriangleMatrix matrix;
	private HashSet<ITask> uniqueTasks;
	private double gapPenalty; 
	private int index = 0;
	
	private double positiveThreshold;
	
	public ObjectDistanceSubstitionMatrix(
			HashSet<ITask> uniqueTasks,float positiveThreshold, int gapPenalty) {
		this.uniqueTasks = uniqueTasks;
		this.positiveThreshold = positiveThreshold;
		idmapping = new HashMap<Integer, Integer>();
		matrix = new TriangleMatrix(uniqueTasks.size()+1);
		this.gapPenalty = gapPenalty;
		
	}


	public double getGapPenalty() {
		return gapPenalty;
	}
	
	public void setGapPenalty(double gapPenalty) {
		this.gapPenalty = gapPenalty;
	}

	public void generate() {
		
		int index1=-1;
		int index2=-1;
		float distance=0;
		for (Iterator<ITask> it = uniqueTasks.iterator(); it.hasNext();) {
			ITask task1 = it.next();
			for (Iterator<ITask> jt = uniqueTasks.iterator(); jt.hasNext();) {
				ITask task2 = jt.next();
				ITaskInstance ti1 = null;
				ITaskInstance ti2 = null;
				//TODO This seems ugly
				if(task1.getInstances().size() > 0) {
					ti1 = (ITaskInstance) task1.getInstances().iterator().next();
				}
				if(task2.getInstances().size() > 0) {
					ti2 = (ITaskInstance) task2.getInstances().iterator().next();
				}
				IEventTaskInstance eti1 = null;
				IEventTaskInstance eti2 = null;
				
				
				if (ti1 instanceof IEventTaskInstance && ti2 instanceof IEventTaskInstance) {
					eti1 = (IEventTaskInstance) ti1;
					index1 = getIndex(eti1);
					eti2 = (IEventTaskInstance) ti2;
					index2 = getIndex(eti2);
					distance = distanceBetweenInstances(eti1,eti2);
				}
				else if(ti1 instanceof IEventTaskInstance &&  !(ti2 instanceof IEventTaskInstance)) {
					task1 = ((ITaskInstance) ti1).getTask();
					index2 = getIndex(task2);
					eti1 = (IEventTaskInstance) ti1;
					index1 = getIndex(eti1);
					distance = distanceBetweenTaskAndInstance(task2,eti1);
				}
				else if(!(ti1 instanceof IEventTaskInstance) && ti2 instanceof IEventTaskInstance) {
					index1 = getIndex(task1);
					eti2 = (IEventTaskInstance) ti2;
					index2 = getIndex(eti2);
					distance = distanceBetweenTaskAndInstance(task1,eti2);
				}
				else if(!(ti1 instanceof IEventTaskInstance) && !(ti2 instanceof IEventTaskInstance)) {
					index1 = getIndex(task1);
					index2 = getIndex(task2);
					distance = distanceBetweenTasks(task1,task2);
				}
				else {
					System.out.println("Unknown error");
				}
				
				matrix.set(index1, index2,distance);
	
			}
		}
	}

	//TODO: Calculate distance here
	private float distanceBetweenTaskAndInstance(ITask task1, IEventTaskInstance eti) {
		return 0;
	}
	
	//TODO: Calculate distance here
	private float distanceBetweenTasks(ITask task1, ITask task2) {
		return 0;
	}


	private int getIndex(ITask task) {
		int tempindex =-1;
	
		if(!idmapping.containsKey(task.getId())) {
			idmapping.put(task.getId(), index);
			tempindex = index;
			index++;
		}
		else {
			tempindex = idmapping.get(task.getId());
		}
		
		return tempindex;
	}
	
	private int getIndex(IEventTaskInstance eti) {
		int tempindex = -1;
		if(!idmapping.containsKey(eti.getTask().getId())) {
			idmapping.put(eti.getTask().getId(), index);
			tempindex = index;
			index++;
		}	
		else {
			tempindex = idmapping.get(eti.getTask().getId());
		}
		return tempindex;
	};
	
	private float distanceBetweenInstances(IEventTaskInstance eti1, IEventTaskInstance eti2)  {
		IGUIElement first = (IGUIElement) eti1.getEvent().getTarget();
		IGUIElement second = (IGUIElement) eti2.getEvent().getTarget();
		float distance = -1*AlignmentHelpers.distanceBetween(first, second);
		distance += positiveThreshold;
		return distance;
	}
	
	public String toString(){
		return matrix.toString();
	}

	public double getScore(int taskId1, int taskId2) {
		if(taskId1 == Constants.GAP_SYMBOL || taskId1 == Constants.UNMATCHED_SYMBOL || taskId2 == Constants.GAP_SYMBOL || taskId2 == Constants.UNMATCHED_SYMBOL ) {
			return 0.0; 
		} 
		else {
			
			Integer first = idmapping.get(taskId1);
			Integer second = idmapping.get(taskId2);
			return matrix.get(first,second);	
		}
		
	}

}