package ca.bc.webarts.widgets.graph;

import java.util.Iterator;
import org.graphstream.graph.*;
import org.graphstream.graph.implementations.*;

import ca.bc.webarts.tools.SqlQuery;

/** NetworkGraph is a class that creates a directed graph using the GraphStream library to
  * represent a Power system distribution network.
  **/
public class NetworkGraph
{
  /** Main method for testing from the commandline. **/
  public static void main( String args[] )
  {
    NetworkGraph instance = null;
    if(args.length<1)
      instance = new NetworkGraph();
   //lse
  }


  /** default constructor that initializes the cals with a simple sample network and displays it. **/
  public NetworkGraph()
  {
    Graph graph = new SingleGraph( "Sample Network" );

    graph.addAttribute( "ui.stylesheet", styleSheet );
    graph.setAutoCreate( true );
    graph.setStrict( false );

    // DISPLAY the graph
    graph.display();

    addSampleEdges(graph);

    for ( Node node : graph )
    {
      node.addAttribute( "ui.label", node.getId() );
    }

    explore( graph.getNode( "Root" ) );
  }


  /** default constructor that initializes the cals with a simple sample network and displays it. **/
  public NetworkGraph(StringBuilder izSb)
  {

    Graph graph = new SingleGraph( "Network" );

    graph.addAttribute( "ui.stylesheet", styleSheet );
    graph.setAutoCreate( true );
    graph.setStrict( false );

    createSubstationCircuits(izSb);

    // DISPLAY the graph
    graph.display();

    //addEdges(graph, sb);

    for ( Node node : graph )
    {
      node.addAttribute( "ui.label", node.getId() );
    }

    explore( graph.getNode( "Root" ) );
  }


  public void explore( Node source )
  {
    Iterator<? extends Node> k = source.getBreadthFirstIterator();

    while ( k.hasNext() )
    {
      Node next = k.next();
      next.setAttribute( "ui.class", "marked" );
      sleep();
    }
  }


  /** simple method to sleep the current thread and hides and interupted exceptions. **/
  protected void sleep()
  {
    try
    { Thread.sleep(1000 ); }
    catch ( Exception e )
    { }
  }


  /** Adds a sample set of edges to the graph for demo purposes. **/
  protected void addSampleEdges(Graph graph)
  {
    graph.addEdge( "RootToB", "Root", "B" );
    graph.addEdge( "BC", "B", "C" );
    graph.addEdge( "BD", "B", "D" );
    graph.addEdge( "CE", "C", "E" );
    graph.addEdge( "EF", "E", "F" );
    graph.addEdge( "FG", "F", "G" );
    graph.addEdge( "FH", "F", "H" );
    graph.addEdge( "EI", "E", "I" );
  }


  /** Addes edges to the graph based on the contents of the passed in StringBuiler that holds a delimeted . **/
  protected void addSampleEdges(Graph graph, StringBuilder sb)
  {
    graph.addEdge( "RootToB", "Root", "B" );
    graph.addEdge( "BC", "B", "C" );
    graph.addEdge( "BD", "B", "D" );
    graph.addEdge( "CE", "C", "E" );
    graph.addEdge( "EF", "E", "F" );
    graph.addEdge( "FG", "F", "G" );
    graph.addEdge( "FH", "F", "H" );
    graph.addEdge( "EI", "E", "I" );
  }


  /**
   * Takes a IZ StringBuffer and creATES THE GRAPH.
   */
  private void createSubstationCircuits(StringBuffer sb)
  {
    boolean debugOut = false;

    int numRows = sb.toString().split("\n").length-1;
    String currFdr = "";
    String currIz = "";
    String deviceType = "";
    String deviceSysID = "";
    String deviceCapacity = "";
    String custCount = "";
    String childVertexName = "";
    String izParent = "";

    if (debugOut) System.out.println("numRows "+numRows);
    if (debugOut) System.out.println(sb.toString());

    String [][] data = SqlQuery.sbToDataRowArray(sb);  // [row][col]

    for (int currRow=0; currRow<numRows; currRow++)
    {
      debugOut = false;
      currFdr = data[currRow][1].trim();
      currIz = data[currRow][2].trim();
      deviceSysID = data[currRow][4].trim();
      deviceType = data[currRow][5].trim();
      if (data[currRow].length>6 && data[currRow][6].trim().length()>0)
        deviceCapacity = data[currRow][6].trim();
      else
        deviceCapacity = "";
      if (data[currRow].length>7 && data[currRow][7].trim().length()>0)
        custCount = data[currRow][7].trim();
      else
        custCount = "0";
      // ************
      childVertexName = createIzoneVertexName(currFdr,currIz,deviceType,deviceSysID,deviceCapacity,custCount);
      // ************
      if (currRow==0) graph.addNode(currFdr.substring(5)); // create substation

      /*
      if (!vertexVector.contains(currFdr) )
      {
        if (debugOut) System.out.println("ADDING: "+currFdr+"   currRow="+currRow);
        if (debugOut) System.out.println("       vertexVector: "+vertexVector);
        vertexVector.addElement(currFdr);
        graph.addVertex(currFdr);
      }
      */

      if (currFdr.equals(data[i][1].trim()))
      {
        if (("1.source".equals(currIz)||"01.source".equals(currIz)))
        {
          graph.addNode(currFdr+".source");
        }
        else
        {
          if(currIz.lastIndexOf(".").lastIndexOf(".")>-1)
          {
            izParent = currIz.substring(0,currIz.lastIndexOf(".").lastIndexOf("."));
            graph.addEdge(childVertexName,izParent,currIz);
          }
          else
          {
            izParent = currFdr+".source";
            graph.addEdge(childVertexName,izParent,currIz);
          }
        }

      }
    }
  }


  /**
   *
   */
  private String createIzoneVertexName(String fdr, String iz, String deviceType, String deviceSysID)
  {
    return fdr+"}"+iz+"_"+deviceType+"-"+deviceSysID;
  }


  /**
   *
   */
  private String createIzoneVertexName(String fdr, String iz, String deviceType, String deviceSysID, String deviceCapacity)
  {
    return fdr+"}"+iz+"_"+deviceType+(deviceType.equalsIgnoreCase("cutout")?"("+deviceCapacity+")":"")+"-"+deviceSysID;
  }


  /**
   *
   */
  private String createIzoneVertexName(String fdr, String iz, String deviceType, String deviceSysID, String deviceCapacity, String custCount)
  {
    return fdr+"}"+iz+"_"+deviceType+(deviceType.equalsIgnoreCase("cutout")?"("+deviceCapacity+")":"")+"-"+deviceSysID+".Custs="+custCount;
  }


  protected String styleSheet =
      "node {" +
      "   fill-color: black;" +
      "}" +
      "node.marked {" +
      "   fill-color: red;" +
      "}";
}