Changeset 1086:caa13d291528 in lemon-0.x

Timestamp:

01/18/05 13:02:27 (20 years ago)

Author:

Alpar Juttner

Branch:

default

Phase:

public

Convert:

svn:c9d7d8f5-90d6-0310-b91f-818b3a526b0e/lemon/trunk@1482

Message:

In graphToEps(), nodes may have different shapes (circles or squares).

Location:

src

Files:

• demo/graph_to_eps_demo.cc (modified) (6 diffs)
• lemon/graph_to_eps.h (modified) (8 diffs)

src/demo/graph_to_eps_demo.cc

@@ -68 +68 @@
   ListGraph::NodeMap<double> sizes(g);
   ListGraph::NodeMap<int> colors(g);
+  ListGraph::NodeMap<int> shapes(g);
   ListGraph::EdgeMap<int> ecolors(g);
   ListGraph::EdgeMap<int> widths(g);
   
-  coords[n1]=Xy(50,50);  sizes[n1]=1; colors[n1]=1;
-  coords[n2]=Xy(50,70);  sizes[n2]=2; colors[n2]=2;
-  coords[n3]=Xy(70,70);  sizes[n3]=1; colors[n3]=3;
-  coords[n4]=Xy(70,50);  sizes[n4]=2; colors[n4]=4;
-  coords[n5]=Xy(85,60);  sizes[n5]=3; colors[n5]=5;
+  coords[n1]=Xy(50,50);  sizes[n1]=1; colors[n1]=1; shapes[n1]=0;
+  coords[n2]=Xy(50,70);  sizes[n2]=2; colors[n2]=2; shapes[n2]=0;
+  coords[n3]=Xy(70,70);  sizes[n3]=1; colors[n3]=3; shapes[n3]=0;
+  coords[n4]=Xy(70,50);  sizes[n4]=2; colors[n4]=4; shapes[n4]=1;
+  coords[n5]=Xy(85,60);  sizes[n5]=3; colors[n5]=5; shapes[n5]=0;
   
   Edge e;
@@ -91 +92 @@
   graphToEps(g,"graph_to_eps_demo_out.eps").scale(10).coords(coords).
     nodeScale(2).nodeSizes(sizes).
+    nodeShapes(shapes).
     nodeColors(composeMap(colorSet,colors)).
     edgeColors(composeMap(colorSet,ecolors)).
@@ -98 +100 @@
   graphToEps(g,"graph_to_eps_demo_out_arr.eps").scale(10).coords(coords).
     nodeScale(2).nodeSizes(sizes).
+    nodeShapes(shapes).
     nodeColors(composeMap(colorSet,colors)).
     edgeColors(composeMap(colorSet,ecolors)).
@@ -117 +120 @@
   graphToEps(g,"graph_to_eps_demo_out_par.eps").scale(10).coords(coords).
     nodeScale(2).nodeSizes(sizes).
+    nodeShapes(shapes).
     nodeColors(composeMap(colorSet,colors)).
     edgeColors(composeMap(colorSet,ecolors)).
@@ -125 +129 @@
   graphToEps(g,"graph_to_eps_demo_out_par_arr.eps").scale(10).coords(coords).
     nodeScale(2).nodeSizes(sizes).
+    nodeShapes(shapes).
     nodeColors(composeMap(colorSet,colors)).
     edgeColors(composeMap(colorSet,ecolors)).
@@ -130 +135 @@
     nodeTexts(id).nodeTextSize(3).
     enableParallel().parEdgeDist(1).
+    //    hideNodes().
     drawArrows().arrowWidth(1).arrowLength(1);
 }

src/lemon/graph_to_eps.h

@@ -81 +81 @@
   ConstMap<typename Graph::Node,xy<double> > _coords;
   ConstMap<typename Graph::Node,double > _nodeSizes;
+  ConstMap<typename Graph::Node,int > _nodeShapes;
 
   ConstMap<typename Graph::Node,Color > _nodeColors;
@@ -124 +125 @@
                           bool _pros=false) :
     g(_g), os(_os),
-    _coords(xy<double>(1,1)), _nodeSizes(1.0),
+    _coords(xy<double>(1,1)), _nodeSizes(1.0), _nodeShapes(0),
     _nodeColors(Color(1,1,1)), _edgeColors(Color(0,0,0)),
     _edgeWidths(1), _edgeWidthScale(0.3),
@@ -155 +156 @@
   bool dontPrint;
 
+  enum NodeShapes { CIRCLE=0, SQUARE=1 };
+                   
   class edgeLess {
     const Graph &g;
@@ -215 +218 @@
     return GraphToEps<NodeSizesTraits<X> >(NodeSizesTraits<X>(*this,x));
   }
+  template<class X> struct NodeShapesTraits : public T {
+    const X &_nodeShapes;
+    NodeShapesTraits(const T &t,const X &x) : T(t), _nodeShapes(x) {}
+  };
+  ///Sets the map of the node shapes
+
+  ///Sets the map of the node shapes
+  ///\param x must be a node map with \c int (or convertible) values. 
+  ///\todo Incomplete doc.
+  template<class X> GraphToEps<NodeShapesTraits<X> > nodeShapes(const X &x)
+  {
+    dontPrint=true;
+    return GraphToEps<NodeShapesTraits<X> >(NodeShapesTraits<X>(*this,x));
+  }
   template<class X> struct NodeTextsTraits : public T {
     const X &_nodeTexts;
@@ -387 +404 @@
   ///Use it to show the undirected edges as a pair of directed ones.
   GraphToEps<T> &bidir(bool b=true) {_undir=!b;return *this;}
-  
+
+protected:
+  bool isInsideNode(xy<double> p, double r,int t) 
+  {
+    switch(t) {
+    case CIRCLE:
+      return p.normSquare()<=r*r;
+    case SQUARE:
+      return p.x<=r&&p.x>=-r&&p.y<=r&&p.y>=-r;
+    }
+    return false;
+  }
+
+public:
   ~GraphToEps() 
   {
@@ -410 +440 @@
        << "      4 2 roll 1 index 1 index curveto stroke } bind def\n";
     os << "/l { setlinewidth setrgbcolor newpath moveto lineto stroke } bind def\n";
+    //x y r
     os << "/c { newpath dup 3 index add 2 index moveto 0 360 arc closepath } bind def\n";
+    //x y r
+    os << "/sq { newpath 2 index 1 index add 2 index 2 index add moveto\n"
+       << "      2 index 1 index sub 2 index 2 index add lineto\n"
+       << "      2 index 1 index sub 2 index 2 index sub lineto\n"
+       << "      2 index 1 index add 2 index 2 index sub lineto\n"
+       << "      closepath pop pop pop} bind def\n";
     // x y r cr cg cb
-    os << "/n { setrgbcolor 2 index 2 index 2 index c fill\n"
+    os << "/nc { setrgbcolor 2 index 2 index 2 index c fill\n"
        << "     0 0 0 setrgbcolor dup "
        << _nodeBorderQuotient << " mul setlinewidth "
        << 1+_nodeBorderQuotient/2 << " div c stroke\n"
+       << "   } bind def\n";
+    os << "/nsq { setrgbcolor 2 index 2 index 2 index sq fill\n"
+       << "     0 0 0 setrgbcolor dup "
+       << _nodeBorderQuotient << " mul setlinewidth "
+       << 1+_nodeBorderQuotient/2 << " div sq stroke\n"
        << "   } bind def\n";
     os << "/arrl " << _arrowLength << " def\n";
@@ -475 +517 @@
             xy<double> mm=m+rot(d)*sw/.75;
             if(_drawArrows) {
+              int node_shape;
               const int INERPOL_PREC=20;
               xy<double> s=_coords[g.source(*e)];
               xy<double> t=_coords[g.target(*e)];
               double rn=_nodeSizes[g.target(*e)]*_nodeScale;
-              rn*=rn;
+              node_shape=_nodeShapes[g.target(*e)];
               Bezier3 bez(s,mm,mm,t);
               double t1=0,t2=1;
               for(int i=0;i<INERPOL_PREC;++i)
-                if((bez((t1+t2)/2)-t).normSquare()>rn) t1=(t1+t2)/2;
-                else t2=(t1+t2)/2;
+                if(isInsideNode(bez((t1+t2)/2)-t,rn,node_shape)) t2=(t1+t2)/2;
+                else t1=(t1+t2)/2;
               xy<double> apoint=bez((t1+t2)/2);
-              rn = _nodeSizes[g.target(*e)]*_nodeScale+_arrowLength+
-                _edgeWidths[*e]*_edgeWidthScale;
+              rn = _arrowLength+_edgeWidths[*e]*_edgeWidthScale;
               rn*=rn;
-              t1=0;t2=1;
+              t2=(t1+t2)/2;t1=0;
               for(int i=0;i<INERPOL_PREC;++i)
-                if((bez((t1+t2)/2)-t).normSquare()>rn) t1=(t1+t2)/2;
+                if((bez((t1+t2)/2)-apoint).normSquare()>rn) t1=(t1+t2)/2;
                 else t2=(t1+t2)/2;
               xy<double> linend=bez((t1+t2)/2);       
               bez=bez.before((t1+t2)/2);
-              rn=_nodeSizes[g.source(*e)]*_nodeScale; rn*=rn;
-              t1=0;t2=1;
-              for(int i=0;i<INERPOL_PREC;++i)
-                if((bez((t1+t2)/2)-s).normSquare()>rn) t2=(t1+t2)/2;
-                else t1=(t1+t2)/2;
-              bez=bez.after((t1+t2)/2);
+//            rn=_nodeSizes[g.source(*e)]*_nodeScale;
+//            node_shape=_nodeShapes[g.source(*e)];
+//            t1=0;t2=1;
+//            for(int i=0;i<INERPOL_PREC;++i)
+//              if(isInsideNode(bez((t1+t2)/2)-t,rn,node_shape)) t1=(t1+t2)/2;
+//              else t2=(t1+t2)/2;
+//            bez=bez.after((t1+t2)/2);
               os << _edgeWidths[*e]*_edgeWidthScale << " setlinewidth "
                  << _edgeColors[*e].getR() << ' '
@@ -560 +603 @@
     if(_showNodes) {
       os << "%Nodes:\ngsave\n";
-      for(NodeIt n(g);n!=INVALID;++n)
+      for(NodeIt n(g);n!=INVALID;++n) {
         os << _coords[n].x << ' ' << _coords[n].y << ' '
            << _nodeSizes[n]*_nodeScale << ' '
            << _nodeColors[n].getR() << ' '
            << _nodeColors[n].getG() << ' '
-           << _nodeColors[n].getB() << " n\n";
+           << _nodeColors[n].getB() << ' ';
+        switch(_nodeShapes[n]) {
+        case CIRCLE:
+          os<< "nc";break;
+        case SQUARE:
+          os<< "nsq";break;
+        }
+        os<<'\n';
+      }
       os << "grestore\n";
     }