import choco.kernel.solver.variables.integer.IntDomainVar;
import choco.kernel.solver.variables.AbstractVar;
import choco.kernel.memory.IStateInt;
import java.util.*;
import java.io.*;

//
// for monitoring the search process
//

public class MonitorObject {

    private IntDomainVar[] vars;           // 0/1 decision variables
    private int n;                         // number of decision variables
    private int[] saturation;              // saturation of vertex i
    private int[] colour;                  // colour of a vertex
    private int[] solution;                // best solution so far
    private IStateInt depth;               // current depth of search
    private IStateInt[] deg;               // current degree
    private Object[] saturationAtDepth;    // average saturation at depth
    private Object[] colourAtDepth;        // chromatic number at depth
    private Object[] degAtDepth;           // average degree at depth
    private Object[] pAtDepth;             // size of candidate set P
    private Object[] kAtDepth;             // size of clique
    private int[] visits;                  // visits at depth
    private Object[] densityAtDepth;       // essentially, edge probability
    private ArrayList<Integer> kMaxLog;    // trace of max clique size
    private ArrayList<Integer> pLog;       // trace of candidate set size
    private ArrayList<Integer> kLog;       // trace of current clique size
    private ArrayList<Integer> depthLog;   // trace of current depth
    private int kMax;                      // size of largest clique

    public MonitorObject(IntDomainVar[] vars,int[] saturation,int[] colour,
			 IStateInt[] deg,int[] solution,IStateInt depth){
	this.vars         = vars; 
	n                 = vars.length; 
	kMax              = 0;
	this.saturation   = saturation; 
	this.colour       = colour;
	this.deg          = deg;
	this.solution     = solution;
	this.depth        = depth;
	saturationAtDepth = new Object[n];
	colourAtDepth     = new Object[n];
	degAtDepth        = new Object[n];
	pAtDepth          = new Object[n];
	kAtDepth          = new Object[n];
	densityAtDepth    = new Object[n];
	kMaxLog           = new ArrayList<Integer>();
	pLog              = new ArrayList<Integer>();
	kLog              = new ArrayList<Integer>();
	depthLog          = new ArrayList<Integer>();
	visits            = new int[n];
	for (int i=0;i<n;i++){
	    saturationAtDepth[i] = new ArrayList<Float>();
	    colourAtDepth[i]     = new ArrayList<Integer>();
	    degAtDepth[i]        = new ArrayList<Float>();
	    pAtDepth[i]          = new ArrayList<Integer>();
	    kAtDepth[i]          = new ArrayList<Integer>();
	    densityAtDepth[i]    = new ArrayList<Float>();
	    visits[i]            = 0;
	}
    }

    public void monitor(){
	depth.add(1);
	int i,K,P,sumSat,maxCol,sumDeg;
	i = depth.get();
	kMax = K = P = sumSat = maxCol = sumDeg = 0;
	visits[i]++;
	for (int j=0;j<n;j++){
	    kMax = kMax + solution[j];
	    if (vars[j].isInstantiated() && vars[j].getVal() == 1) 
		K++;                                 // size of current clique
	    if (!vars[j].isInstantiated()){          // future subproblem
		P++;                                 // size of candidate set
		sumSat = sumSat + saturation[j];     // sum of saturations
		maxCol = Math.max(maxCol,colour[j]); // chromatic number
		sumDeg = sumDeg + deg[j].get();      // number of edges in P
	    }
	}
	ArrayList<Float> x1 = (ArrayList<Float>)saturationAtDepth[i];
	x1.add((float)sumSat/(float)P);
	ArrayList<Integer> i1 = (ArrayList<Integer>)colourAtDepth[i];
	i1.add(maxCol);
	ArrayList<Float> x2 = (ArrayList<Float>)degAtDepth[i];
	x2.add((float)sumDeg/(float)P);
	ArrayList<Integer> i2 = (ArrayList<Integer>)pAtDepth[i];
	i2.add(P);
	ArrayList<Integer> i3 = (ArrayList<Integer>)kAtDepth[i];
	i3.add(K);
	ArrayList<Float> x3 = (ArrayList<Float>)densityAtDepth[i];
	if (P > 1) x3.add((float)sumDeg/(float)(P*(P-1)));
	kMaxLog.add(kMax);
	kLog.add(K);
	pLog.add(P);
	depthLog.add(i);
    }

    private float iAverage(ArrayList<Integer>x){
	int sum = 0;
	for(int i : x) sum = sum + i;
	if (x.size() == 0) return 0;
	return (float)sum/(float)x.size();
    }

    private float fAverage(ArrayList<Float>x){
	float sum = (float)0.0;
	for(float y : x) sum = sum + y;
	if (x.size() == 0) return 0;
	return sum/(float)x.size();
    }
    
    public String toString(){
	String s = "";
	for (int i=0;i<n;i++){
	    s = s + i +" ";
	    s = s + visits[i] +" ";
	    s = s + iAverage((ArrayList<Integer>)pAtDepth[i]) +" ";
	    s = s + iAverage((ArrayList<Integer>)kAtDepth[i]) +" ";
	    s = s + fAverage((ArrayList<Float>)degAtDepth[i]) +" ";
	    s = s + fAverage((ArrayList<Float>)saturationAtDepth[i]) +" ";
	    s = s + iAverage((ArrayList<Integer>)colourAtDepth[i]) +" ";
	    s = s + fAverage((ArrayList<Float>)densityAtDepth[i]) +"\n";
	}
	return s;
    }

    public void printLog(){
	try{
	    BufferedWriter out = new BufferedWriter(new FileWriter("logFile"));
	    for (int i=0;i<kMaxLog.size();i++)
		out.write(i +" "+ kMaxLog.get(i) +" "+ kLog.get(i) +" "+ pLog.get(i) +" "+ depthLog.get(i) +"\n");
	    out.close();
	} catch (IOException e){}
    }
	    
}