#include <graph_displayer_canvas.h>

#include <broken_edge.h>

#include <math.h>

GraphDisplayerCanvas::GraphDisplayerCanvas(Graph & gr, CoordinatesMap & cm, MapStorage & ms):g(gr),nodesmap(g),edgesmap(g),edgetextmap(g),displayed_graph(*(root()), 0, 0),mapstorage(ms),isbutton(0),active_item(NULL),target_item(NULL)

{

  actual_handler=displayed_graph.signal_event().connect(sigc::mem_fun(*this, &GraphDisplayerCanvas::create_edge_event_handler), false);

  active_node=INVALID;

  active_edge=INVALID;

  //set_center_scroll_region(true);

  //first edges are drawn, to hide joining with nodes later

  for (EdgeIt i(g); i!=INVALID; ++i)

  {

    //drawing green lines, coordinates are from cm

    Gnome::Canvas::Points coos;

    coos.push_back(Gnome::Art::Point(cm[g.source(i)].x,cm[g.source(i)].y));

    coos.push_back(Gnome::Art::Point(cm[g.target(i)].x,cm[g.target(i)].y));

    edgesmap[i]=new BrokenEdge(displayed_graph, coos);

    *(edgesmap[i]) << Gnome::Canvas::Properties::fill_color("green");

    edgesmap[i]->property_width_pixels().set_value(10);    

    //initializing edge-text as well, to empty string

    double x1, x2, y1, y2;

    edgesmap[i]->get_bounds(x1, y1, x2, y2);

    edgetextmap[i]=new Gnome::Canvas::Text(displayed_graph,(x1+x2)/2, (y1+y2)/2, "");

    edgetextmap[i]->property_fill_color().set_value("black");

  }

  //afterwards nodes come to be drawn

  NodeIt i(g);

  int maxx=0, maxy=0, minx=(int)cm[i].x, miny=(int)cm[i].y;

  for (; i!=INVALID; ++i)

  {

    //minimum and maximum is gathered to be able to zoom to the graph correctly (whole figure should be seen)

    if(cm[i].x>maxx)maxx=(int)cm[i].x;

    if(cm[i].y>maxy)maxy=(int)cm[i].y;

    if(cm[i].x<minx)minx=(int)cm[i].x;

    if(cm[i].y<miny)miny=(int)cm[i].y;

    //drawing bule nodes, with black line around them

    nodesmap[i]=new Gnome::Canvas::Ellipse(displayed_graph, cm[i].x-20, cm[i].y-20, cm[i].x+20, cm[i].y+20);

    *(nodesmap[i]) << Gnome::Canvas::Properties::fill_color("blue");

    *(nodesmap[i]) << Gnome::Canvas::Properties::outline_color("black");

    //!!!!!!! (nodesmap[i])->signal_event().connect(sigc::bind(sigc::mem_fun(*this, &GraphDisplayerCanvas::event_handler),i));

  }

  updateScrollRegion();

}

GraphDisplayerCanvas::~GraphDisplayerCanvas()

{

  //writing out the end state of the graph

  //\todo all the maps has to be write out!

  Graph::NodeMap <int> id(g);

  Graph::NodeMap <double> xc(g);

  Graph::NodeMap <double> yc(g);

  int j=1;

  for (NodeIt i(g); i!=INVALID; ++i)

  {

    double x1,y1,x2,y2;

    nodesmap[i]->get_bounds(x1, y1, x2, y2);

    id[i]=j++;

    xc[i]=(x1+x2)/2;

    yc[i]=(y1+y2)/2;

  }

  GraphWriter<Graph> writer(std::cout,g);

  writer.writeNodeMap("id", id);

  writer.writeNodeMap("coordinates_x", xc);

  writer.writeNodeMap("coordinates_y", yc);

  writer.run();

}

int GraphDisplayerCanvas::changeLineWidth (std::string mapname)

{

  for (EdgeIt i(g); i!=INVALID; ++i)

  {

    int w=(int)(*(mapstorage.edgemap_storage)[mapname])[i];

    edgesmap[i]->property_width_pixels().set_value(w);

  }

  return 0;

};

int GraphDisplayerCanvas::changeColor (std::string mapname)

{  

  //function maps the range of the maximum and

  //the minimum of the nodemap to the range of

  //green in RGB

  for (EdgeIt i(g); i!=INVALID; ++i)

  {

    double w=(*(mapstorage.edgemap_storage)[mapname])[i];

    double max=mapstorage.maxOfEdgeMap(mapname);

    double min=mapstorage.minOfEdgeMap(mapname);

    //std::cout<<w<<" "<<max<<" "<<min<<" "<<100*(w-min)/(max-min)<<std::endl;

    Gdk::Color color;

    if(max!=min)

    {

      color.set_rgb_p (0, 100*(w-min)/(max-min), 0);

    }

    else

    {

      color.set_rgb_p (0, 100, 0);

    }

    edgesmap[i]->property_fill_color_gdk().set_value(color);

  }

  return 0;

};

int GraphDisplayerCanvas::changeText (std::string mapname)

{

  //the number in the map will be written on the edge

  //EXCEPT when the name of the map is Text, because

  //in that case empty string will be written, because

  //that is the deleter map

  //\todo isn't it a bit woodcutter?

  for (EdgeIt i(g); i!=INVALID; ++i)

  {

    if(mapname!="Text")

    {

      double number=(*(mapstorage.edgemap_storage)[mapname])[i];

      int length=(int)(floor(log(number)/log(10)))+1;

      int maxpos=(int)(pow(10,length-1));

      int strl=length+1+RANGE;

      char * str=new char[strl];

      str[length]='.';

      str[strl]='\0';

      for(int j=0;j<strl;j++)

      {

	if(j!=length)

        {

	  int digit=(int)(number/maxpos);

	  str[j]=(digit+'0');

	  number-=digit*maxpos;

	  number*=10;

        }

      }

      edgetextmap[i]->property_text().set_value(str);

    }

    else

    {

      edgetextmap[i]->property_text().set_value("");

    }

  }

  return 0;

};

bool GraphDisplayerCanvas::event_handler(GdkEvent* e, Node n)

{

  switch(e->type)

  {

    case GDK_BUTTON_PRESS:

      //we mark the location of the event to be able to calculate parameters of dragging

      clicked_x=e->button.x;

      clicked_y=e->button.y;

      active_item=(get_item_at(e->button.x, e->button.y));

      isbutton=1;

      break;

    case GDK_BUTTON_RELEASE:

      isbutton=0;

      active_item=NULL;

      updateScrollRegion();

      break;

    case GDK_MOTION_NOTIFY:

      //we only have to do sg. if the mouse button is pressed

      if(isbutton)

      {

	//new coordinates will be the old values,

	//because the item will be moved to the

	//new coordinate therefore the new movement

	//has to be calculated from here

        double dx=e->motion.x-clicked_x;

        double dy=e->motion.y-clicked_y;

        active_item->move(dx, dy);

        clicked_x=e->motion.x;

        clicked_y=e->motion.y;

	//all the edges connected to the moved point has to be redrawn

        EdgeIt e;

        g.firstOut(e,n);

        for(;e!=INVALID;g.nextOut(e))

        {

            Gnome::Canvas::Points coos;

            double x1, x2, y1, y2;

            nodesmap[g.source(e)]->get_bounds(x1, y1, x2, y2);

            coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

            nodesmap[g.target(e)]->get_bounds(x1, y1, x2, y2);

            coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

            edgesmap[e]->property_points().set_value(coos);

	    edgesmap[e]->get_bounds(x1, y1, x2, y2);

	    edgetextmap[e]->property_x().set_value((x1+x2)/2);

	    edgetextmap[e]->property_y().set_value((y1+y2)/2);

        }

        g.firstIn(e,n);

        for(;e!=INVALID;g.nextIn(e))

        {

            Gnome::Canvas::Points coos;

            double x1, x2, y1, y2;

            nodesmap[g.source(e)]->get_bounds(x1, y1, x2, y2);

            coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

            nodesmap[g.target(e)]->get_bounds(x1, y1, x2, y2);

            coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

            edgesmap[e]->property_points().set_value(coos);

	    edgesmap[e]->get_bounds(x1, y1, x2, y2);

	    edgetextmap[e]->property_x().set_value((x1+x2)/2);

	    edgetextmap[e]->property_y().set_value((y1+y2)/2);

        }

      }

    default: break;

  }

  return true;

}

bool GraphDisplayerCanvas::on_expose_event(GdkEventExpose *event)

{

  Gnome::Canvas::CanvasAA::on_expose_event(event);

  //usleep(10000);

  //rezoom();

  return true;

}

void GraphDisplayerCanvas::zoomIn()

{

  set_pixels_per_unit(

      (1.0 + (double) zoom_step / 100.0) * get_pixels_per_unit());

}

void GraphDisplayerCanvas::zoomOut()

{

  set_pixels_per_unit(

      (1.0 - (double) zoom_step / 100.0) * get_pixels_per_unit());

}

void GraphDisplayerCanvas::zoomFit()

{

  // get the height and width of the canvas

  Gtk::Allocation a = get_allocation();

  int aw = a.get_width();

  int ah = a.get_height();

  // add some space

  aw -= 5; if (aw < 0) aw = 0;

  ah -= 5; if (ah < 0) ah = 0;

  // get the bounding box of the graph

  double wx1, wy1, wx2, wy2;

  Gnome::Canvas::Item* pCanvasItem = root();

  pCanvasItem->get_bounds(wx1, wy1, wx2, wy2);

  // fit the graph to the window

  double ppu1 = (double) aw / fabs(wx2 - wx1);

  double ppu2 = (double) ah / fabs(wy2 - wy1);

  set_pixels_per_unit((ppu1 < ppu2) ? ppu1 : ppu2);

}

void GraphDisplayerCanvas::zoom100()

{

  set_pixels_per_unit(1.0);

}

void GraphDisplayerCanvas::updateScrollRegion()

{

  double wx1, wy1, wx2, wy2;

  Gnome::Canvas::Item* pCanvasItem = root();

  pCanvasItem->get_bounds(wx1, wy1, wx2, wy2);

  set_scroll_region(wx1, wy1, wx2, wy2);

}

void GraphDisplayerCanvas::changeEditorialTool(int newtool)

{

  actual_handler.disconnect();

  switch(newtool)

    {

    case MOVE:

      actual_handler=displayed_graph.signal_event().connect(sigc::mem_fun(*this, &GraphDisplayerCanvas::move_event_handler), false);

      break;

      //it has to assigned to canvas, because all the canvas has to be monitored, not only the elements of the already drawn group

    case CREATE_NODE:

      actual_handler=signal_event().connect(sigc::mem_fun(*this, &GraphDisplayerCanvas::create_node_event_handler), false);

      break;

    case CREATE_EDGE:

      actual_handler=displayed_graph.signal_event().connect(sigc::mem_fun(*this, &GraphDisplayerCanvas::create_edge_event_handler), false);

      break;

    case ERASER:

      actual_handler=displayed_graph.signal_event().connect(sigc::mem_fun(*this, &GraphDisplayerCanvas::eraser_event_handler), false);

      break;

    default:

      break;

    }

}

bool GraphDisplayerCanvas::move_event_handler(GdkEvent* e)

{

  switch(e->type)

  {

    case GDK_BUTTON_PRESS:

      //we mark the location of the event to be able to calculate parameters of dragging

      clicked_x=e->button.x;

      clicked_y=e->button.y;

      active_item=(get_item_at(e->button.x, e->button.y));

      active_node=INVALID;

      for (NodeIt i(g); i!=INVALID; ++i)

	{

	  if(nodesmap[i]==active_item)

	    {

	      active_node=i;

	    }

	}

      switch(e->button.button)

	{

	case 3:      

	  isbutton=3;

	  break;

	default:

	  isbutton=1;

	  break;

	}

      break;

    case GDK_BUTTON_RELEASE:

      isbutton=0;

      active_item=NULL;

      active_node=INVALID;

      updateScrollRegion();

      break;

    case GDK_MOTION_NOTIFY:

      //we only have to do sg. if the mouse button is pressed AND the click was on a node that was found in the set of nodes

      if(active_node!=INVALID)

      {

	//new coordinates will be the old values,

	//because the item will be moved to the

	//new coordinate therefore the new movement

	//has to be calculated from here

        double dx=e->motion.x-clicked_x;

        double dy=e->motion.y-clicked_y;

        active_item->move(dx, dy);

        clicked_x=e->motion.x;

        clicked_y=e->motion.y;

	//all the edges connected to the moved point has to be redrawn

        EdgeIt ei;

        g.firstOut(ei,active_node);

        for(;ei!=INVALID;g.nextOut(ei))

        {

            Gnome::Canvas::Points coos;

            double x1, x2, y1, y2;

            nodesmap[g.source(ei)]->get_bounds(x1, y1, x2, y2);

            coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

            nodesmap[g.target(ei)]->get_bounds(x1, y1, x2, y2);

            coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

	    if(isbutton==3)

	      {

		edgesmap[ei]->set_points(coos);

	      }

	    else

	      {

		edgesmap[ei]->set_points(coos,true);

	      }

	    edgesmap[ei]->get_bounds(x1, y1, x2, y2);

	    edgetextmap[ei]->property_x().set_value((x1+x2)/2);

	    edgetextmap[ei]->property_y().set_value((y1+y2)/2);

        }

        g.firstIn(ei,active_node);

        for(;ei!=INVALID;g.nextIn(ei))

        {

            Gnome::Canvas::Points coos;

            double x1, x2, y1, y2;

            nodesmap[g.source(ei)]->get_bounds(x1, y1, x2, y2);

            coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

            nodesmap[g.target(ei)]->get_bounds(x1, y1, x2, y2);

            coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

	    if(isbutton==3)

	      {

		edgesmap[ei]->set_points(coos);

	      }

	    else

	      {

		edgesmap[ei]->set_points(coos,true);

	      }

	    edgesmap[ei]->get_bounds(x1, y1, x2, y2);

	    edgetextmap[ei]->property_x().set_value((x1+x2)/2);

	    edgetextmap[ei]->property_y().set_value((y1+y2)/2);

        }

      }

    default: break;

  }

  return true;

}

bool GraphDisplayerCanvas::create_node_event_handler(GdkEvent* e)

{

  switch(e->type)

    {

      //draw the new node in red at the clicked place

    case GDK_BUTTON_PRESS:

      isbutton=1;

      active_node=NodeIt(g,g.addNode());

      window_to_world (e->button.x, e->button.y, clicked_x, clicked_y);

      nodesmap[active_node]=new Gnome::Canvas::Ellipse(displayed_graph, clicked_x-20, clicked_y-20, clicked_x+20, clicked_y+20);

      active_item=(Gnome::Canvas::Item *)(nodesmap[active_node]);

      *(nodesmap[active_node]) << Gnome::Canvas::Properties::fill_color("red");

      *(nodesmap[active_node]) << Gnome::Canvas::Properties::outline_color("black");

      (nodesmap[active_node])->show();

      break;

      //move the new node

    case GDK_MOTION_NOTIFY:

      {

	double world_motion_x, world_motion_y;

	GdkEvent * generated=new GdkEvent();

	window_to_world (e->motion.x, e->motion.y, world_motion_x, world_motion_y);

	generated->motion.x=world_motion_x;

	generated->motion.y=world_motion_y;

	generated->type=GDK_MOTION_NOTIFY;

	move_event_handler(generated);      

	break;

      }

      //finalize the new node

    case GDK_BUTTON_RELEASE:

      isbutton=0;

      *active_item << Gnome::Canvas::Properties::fill_color("blue");

      active_item=NULL;

      active_node=INVALID;

      updateScrollRegion();

      break;

    default:

      break;

    }

  return false;

}

bool GraphDisplayerCanvas::create_edge_event_handler(GdkEvent* e)

{

  switch(e->type)

    {

    case GDK_BUTTON_PRESS:

      //in edge creatino right button has special meaning

      if(e->button.button!=3)

	{

	  //there is not yet selected node

	  if(active_node==INVALID)

	    {

	      //we mark the location of the event to be able to calculate parameters of dragging

	      clicked_x=e->button.x;

	      clicked_y=e->button.y;

	      active_item=(get_item_at(e->button.x, e->button.y));

	      active_node=INVALID;

	      for (NodeIt i(g); i!=INVALID; ++i)

		{

		  if(nodesmap[i]==active_item)

		    {

		      active_node=i;

		    }

		}

	      //the clicked item is really a node

	      if(active_node!=INVALID)

		{

		  *(nodesmap[active_node]) << Gnome::Canvas::Properties::fill_color("red");

		  isbutton=1;

		}

	      //clicked item was not a node. It could be e.g. edge.

	      else

		{

		  active_item=NULL;

		}

	    }

	  //we only have to do sg. if the mouse button

	  // is pressed already once AND the click was

	  // on a node that was found in the set of 

	  //nodes, and now we only search for the second 

	  //node

	  else

	    {

	      target_item=(get_item_at(e->button.x, e->button.y));

	      Graph::NodeIt target_node=INVALID;

	      for (NodeIt i(g); i!=INVALID; ++i)

		{

		  if(nodesmap[i]==target_item)

		    {

		      target_node=i;

		    }

		}

	      //the clicked item is a node, the edge can be drawn

	      if(target_node!=INVALID)

		{

		  *(nodesmap[target_node]) << Gnome::Canvas::Properties::fill_color("red");

		  //creating new edge

		  active_edge=EdgeIt(g,g.addEdge(active_node, target_node));

		  //calculating coordinates of new edge

		  Gnome::Canvas::Points coos;

		  double x1, x2, y1, y2;

		  active_item->get_bounds(x1, y1, x2, y2);

		  coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

		  target_item->get_bounds(x1, y1, x2, y2);

		  coos.push_back(Gnome::Art::Point((x1+x2)/2,(y1+y2)/2));

		  //drawing new edge

		  edgesmap[active_edge]=new BrokenEdge(displayed_graph, coos);

		  *(edgesmap[active_edge]) << Gnome::Canvas::Properties::fill_color("green");

		  edgesmap[active_edge]->property_width_pixels().set_value(10);

		  //redraw nodes to blank terminations of the new edge

		  target_item->raise_to_top();

		  active_item->raise_to_top();

		  //initializing edge-text as well, to empty string

		  edgesmap[active_edge]->get_bounds(x1, y1, x2, y2);

		  edgetextmap[active_edge]=new Gnome::Canvas::Text(displayed_graph,(x1+x2)/2, (y1+y2)/2, "");

		  edgetextmap[active_edge]->property_fill_color().set_value("black");

		}

	      //clicked item was not a node. it could be an e.g. edge. we do not deal with it furthermore.

	      else

		{

		  target_item=NULL;

		}

	    }

	}

      break;

    case GDK_BUTTON_RELEASE:

      isbutton=0;

      //we clear settings in two cases

      //1: the edge is ready (target_item has valid value)

      //2: the edge creation is cancelled with right button

      if((target_item)||(e->button.button==3))

	{

	  if(active_item)

	    {

	      *active_item << Gnome::Canvas::Properties::fill_color("blue");

	      active_item=NULL;

	    }

	  if(target_item)

	    {

	      *target_item << Gnome::Canvas::Properties::fill_color("blue");

	      target_item=NULL;

	    }

	  active_node=INVALID;

	  active_edge=INVALID;

	}

      break;

    default:

      break;

    }

  return false;

}

bool GraphDisplayerCanvas::eraser_event_handler(GdkEvent* e)

{

  switch(e->type)

    {

    case GDK_BUTTON_PRESS:

      active_item=(get_item_at(e->button.x, e->button.y));

      active_node=INVALID;

      active_edge=INVALID;

      for (NodeIt i(g); i!=INVALID; ++i)

	{

	  if(nodesmap[i]==active_item)

	    {

	      active_node=i;

	    }

	}

      if(active_node==INVALID)

	{

	  for (EdgeIt i(g); i!=INVALID; ++i)

	    {

	      if(edgesmap[i]==active_item)

		{

		  active_edge=i;

		}

	    }

	}

    *active_item << Gnome::Canvas::Properties::fill_color("red");

      break;

    case GDK_BUTTON_RELEASE:

      if(active_item==(get_item_at(e->button.x, e->button.y)))

	{

	  if(active_node!=INVALID)

	    {

	      //collecting edges to delete

	      EdgeIt e;

	      std::set<Graph::Edge> edges_to_delete;

	      g.firstOut(e,active_node);

	      for(;e!=INVALID;g.nextOut(e))

		{

		      edges_to_delete.insert(e);

		}

	      g.firstIn(e,active_node);

	      for(;e!=INVALID;g.nextIn(e))

		{

		      edges_to_delete.insert(e);

		}

	      //deleting collected edges

	      for(std::set<Graph::Edge>::iterator edge_set_it=edges_to_delete.begin();edge_set_it!=edges_to_delete.end();edge_set_it++)

		{

		  delete_item(*edge_set_it);

		}

	      delete_item(active_node);

	    }

	  //a simple edge was chosen

	  else

	    {

	      delete_item(active_edge);

	    }

	}

      //pointer was moved, deletion is cancelled

      else

	{

	  if(active_node!=INVALID)

	    {

	      *active_item << Gnome::Canvas::Properties::fill_color("blue");

	    }

	  else

	    {

	      *active_item << Gnome::Canvas::Properties::fill_color("green");

	    }

	}

      //reseting datas

      active_item=NULL;

      active_edge=INVALID;

      active_node=INVALID;

      break;

    case GDK_MOTION_NOTIFY:

      break;

    default:

      break;

    }

  return true;

}

void GraphDisplayerCanvas::delete_item(NodeIt node_to_delete)

{

  delete(nodesmap[node_to_delete]);

  g.erase(node_to_delete);

}

void GraphDisplayerCanvas::delete_item(EdgeIt edge_to_delete)

{

  delete(edgesmap[edge_to_delete]);

  g.erase(edge_to_delete);

}

void GraphDisplayerCanvas::delete_item(Graph::Edge edge_to_delete)

{

  delete(edgesmap[edge_to_delete]);

  g.erase(edge_to_delete);

}