import mascoptLib.gui.*;
import mascoptLib.graphs.*;
import java.util.Iterator;
import java.awt.Color;
import mascoptLib.io.graph.MGLWriter;


public class Premier{

  public static void main (String args[]){

    GraphChooser gc = new GraphChooser();
    Graph GG = new Graph();

    GG=(Graph)gc.getGraphMGL();

    Graph sousrouge = new Graph (GG);

    Graph sousvert = new Graph();


    VertexSet vertvv = new VertexSet();
    EdgeSet vertee = new EdgeSet(vertvv);


    Vertex[] pepe;

    VertexSet vv = GG.getVertexSet();
    EdgeSet ee = GG.getEdgeSet();

    VertexSet rougevv =  sousrouge.getVertexSet();
    EdgeSet rougeee = sousrouge.getEdgeSet();

    sousvert.setNodeSet(vertvv);
    sousvert.setEdgeSet(vertee);


    Iterator vit= vv.iterator();
    Iterator eit= ee.iterator();

    while(vit.hasNext())
    {
      Vertex aux = (Vertex) vit.next();
      String couleurp = aux.getValue("couleur");
      if(couleurp.equals("rouge"))
      { rougevv.add(aux);
        aux.setIntValue("color",(Color.red).getRGB());}
        if(couleurp.equals("vert")) 
        {vertvv.add(aux);
          aux.setIntValue("color",(Color.green).getRGB());}
    }


    while(eit.hasNext()){
      Edge aex = (Edge) eit.next(); 
      pepe = aex.getVertices();
      if(pepe[0].getValue("couleur").equals("rouge")&&pepe[1].getValue("couleur").equals("rouge"))
        rougeee.add(aex);


      if(pepe[0].getValue("couleur").equals("vert")&&pepe[1].getValue("couleur").equals("vert"))
        vertee.add(aex);
    }


    System.out.println("Graphe rouge = " + sousrouge);

    System.out.println("Graphe vert = " + sousvert);




    MascoptViewer mv = new MascoptViewer();
    GView gv = mv.newView("toma castanya");
    GLayer gl= mv.newLayer(sousrouge.getName());
    GLayer gll= mv.newLayer(sousvert.getName());
    mv.addGraphInLayer(sousrouge,gl,true);
    mv.addGraphInLayer(sousvert,gll,true);
    mv.addLayerInView(gl,gv);
    mv.addLayerInView(gll,gv);



    try
    {
      MGLWriter writer = new MGLWriter(args[1]);
      writer.add(sousrouge);
      writer.add(sousvert);
      writer.write();
    }
    catch(java.io.FileNotFoundException fe){fe.printStackTrace();}



  }





}