#include "graph_displayer_canvas.h"

 #include <cmath>

 GraphDisplayerCanvas::EdgeBase::EdgeBase(Gnome::Canvas::Group& _group, Edge _edge, GraphDisplayerCanvas& _canvas) : 

   Gnome::Canvas::Group(_group), edge(_edge), canvas(_canvas), arrow(*this)

 {

   arrow.property_fill_color().set_value("red");

   arrow.lower_to_bottom();

 }

 GraphDisplayerCanvas::EdgeBase::~EdgeBase()

 {

 }

 void GraphDisplayerCanvas::EdgeBase::drawArrow(XY unit_vector_in_dir)

 {

   MapStorage& ms = canvas.mytab.mapstorage;

   XY center(ms.arrow_pos[edge]);

   XY unit_norm_vector(0-unit_vector_in_dir.y, unit_vector_in_dir.x);

   //       /\       // top

   //      /  \      //

   //      -  -      // c(enter)l(eft), ccl, ccr, cr

   //       ||       //

   //       ||       // b(ottom)l, br

   double size=3;

   XY bl (center - unit_vector_in_dir * 3 * size + unit_norm_vector * size );

   XY br (center - unit_vector_in_dir * 3 * size - unit_norm_vector * size );

   XY ccl(center + unit_vector_in_dir *  size + unit_norm_vector * size );

   XY ccr(center + unit_vector_in_dir *  size - unit_norm_vector * size );

   XY cl (center + unit_vector_in_dir *  size + unit_norm_vector * 2 * size );

   XY cr (center + unit_vector_in_dir *  size - unit_norm_vector * 2 * size );

   XY top(center + unit_vector_in_dir * 3 * size);

   Gnome::Canvas::Points arrow_points;

   arrow_points.push_back(Gnome::Art::Point( bl.x , bl.y  ) );

   arrow_points.push_back(Gnome::Art::Point( br.x , br.y  ) );

   arrow_points.push_back(Gnome::Art::Point( ccr.x, ccr.y ) );

   arrow_points.push_back(Gnome::Art::Point( cr.x , cr.y  ) );

   arrow_points.push_back(Gnome::Art::Point( top.x, top.y ) );

   arrow_points.push_back(Gnome::Art::Point( cl.x , cl.y  ) );

   arrow_points.push_back(Gnome::Art::Point( ccl.x, ccl.y ) );

   arrow.property_points().set_value(arrow_points);

 }

 GraphDisplayerCanvas::BrokenEdge::BrokenEdge(Gnome::Canvas::Group & g,

     Edge _edge, GraphDisplayerCanvas & gc) : EdgeBase(g, _edge, gc),

   isbutton(false), line(*this)

 {

   arrow.signal_event().connect(sigc::mem_fun(*this, &GraphDisplayerCanvas::BrokenEdge::edgeFormerEventHandler));

   line.property_fill_color().set_value("green");

   line.property_width_units().set_value(10);    

   line.lower_to_bottom();

   draw();

 }

 GraphDisplayerCanvas::BrokenEdge::~BrokenEdge()

 {

 }

 void GraphDisplayerCanvas::BrokenEdge::draw()

 {

   MapStorage& ms = canvas.mytab.mapstorage;

   //calculating coordinates of the direction indicator arrow

   XY head(ms.coords[ms.graph.target(edge)]);

   XY center(ms.arrow_pos[edge]);

   XY unit_vector_in_dir(head-center);

   double length=sqrt( unit_vector_in_dir.normSquare() );

   unit_vector_in_dir/=length;

   // update the arrow

   drawArrow(unit_vector_in_dir);

   // update the edge

   Gnome::Canvas::Points points;

   Node source = ms.graph.source(edge);

   Node target = ms.graph.target(edge);

   points.push_back(Gnome::Art::Point(ms.coords[source].x,

         ms.coords[source].y));

   points.push_back(Gnome::Art::Point(ms.arrow_pos[edge].x,

         ms.arrow_pos[edge].y));

   points.push_back(Gnome::Art::Point(ms.coords[target].x,

         ms.coords[target].y));

   line.property_points().set_value(points);

 }

 bool GraphDisplayerCanvas::BrokenEdge::edgeFormerEventHandler(GdkEvent* e)

 {

   switch(e->type)

   {

     case GDK_BUTTON_PRESS:

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

       //of dragging

       if(canvas.getActualTool()!=CREATE_NODE)

       {

         canvas.toggleEdgeActivity(this, true);

         clicked_x=e->button.x;

         clicked_y=e->button.y;

         isbutton=true;

       }

       break;

     case GDK_BUTTON_RELEASE:

       if(canvas.getActualTool()!=CREATE_NODE)

       {

         canvas.toggleEdgeActivity(this, false);

         isbutton=false;

       }

       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;

         Gnome::Canvas::Points points_new;

         canvas.mytab.mapstorage.arrow_pos.set(edge, canvas.mytab.mapstorage.arrow_pos[edge] + XY(dx, dy));

         draw();

         canvas.textReposition(canvas.mytab.mapstorage.arrow_pos[edge]);

         clicked_x=e->motion.x;

         clicked_y=e->motion.y;

       }

     default: break;

   }

   return true;

 }

 void GraphDisplayerCanvas::BrokenEdge::setLineWidth(int w)

 {

   line.property_width_units().set_value(w);

 }

 void GraphDisplayerCanvas::BrokenEdge::setFillColor(Gdk::Color c)

 {

   line.property_fill_color_gdk().set_value(c);

 }

 GraphDisplayerCanvas::LoopEdge::LoopEdge(Gnome::Canvas::Group& _group,

     Edge _edge, GraphDisplayerCanvas& _canvas) :

   EdgeBase(_group, _edge, _canvas), line(*this), isbutton(false)

 {

   arrow.signal_event().connect(sigc::mem_fun(*this, &GraphDisplayerCanvas::LoopEdge::edgeFormerEventHandler));

   line.property_outline_color().set_value("green");

   line.property_width_units().set_value(10);

   line.lower_to_bottom();

   draw();

 }

 GraphDisplayerCanvas::LoopEdge::~LoopEdge()

 {

 }

 void GraphDisplayerCanvas::LoopEdge::draw()

 {

   MapStorage& ms = canvas.mytab.mapstorage;

   Node node = ms.graph.source(edge);

   XY center = (ms.coords[node] + ms.arrow_pos[edge]) / 2.0;

   XY unit_vector_in_dir(rot90(center - ms.arrow_pos[edge]));

   double length = sqrt(unit_vector_in_dir.normSquare());

   unit_vector_in_dir /= length;

   drawArrow(unit_vector_in_dir);

   double radius =

     sqrt((ms.arrow_pos[edge] - ms.coords[node]).normSquare()) / 2.0;

   XY p1 = center + XY(-radius,  radius);

   XY p2 = center + XY( radius, -radius);

   line.property_x1().set_value(p1.x);

   line.property_y1().set_value(p1.y);

   line.property_x2().set_value(p2.x);

   line.property_y2().set_value(p2.y);

 }

 void GraphDisplayerCanvas::LoopEdge::setLineWidth(int w)

 {

   line.property_width_units().set_value(w);

 }

 void GraphDisplayerCanvas::LoopEdge::setFillColor(Gdk::Color c)

 {

   line.property_outline_color_gdk().set_value(c);

 }

 bool GraphDisplayerCanvas::LoopEdge::edgeFormerEventHandler(GdkEvent* e)

 {

   switch(e->type)

   {

     case GDK_BUTTON_PRESS:

       if(canvas.getActualTool()!=CREATE_NODE)

       {

         canvas.toggleEdgeActivity(this, true);

         isbutton=true;

       }

       break;

     case GDK_BUTTON_RELEASE:

       if(canvas.getActualTool()!=CREATE_NODE)

       {

         canvas.toggleEdgeActivity(this, false);

         isbutton=false;

       }

       break;

     case GDK_MOTION_NOTIFY:

       if(isbutton)

       {

         canvas.mytab.mapstorage.arrow_pos.set(edge, XY(e->motion.x, e->motion.y));

         draw();

         canvas.textReposition(canvas.mytab.mapstorage.arrow_pos[edge]);

       }

     default: break;

   }

   return true;

 }