source: glemon-0.x/graph_displayer_canvas.cc @ 20:a3bd39d50930

#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);
}