[1073] | 1 | /* -*- C++ -*- |
---|
| 2 | * src/lemon/graph_to_eps.h - Part of LEMON, a generic C++ optimization library |
---|
| 3 | * |
---|
| 4 | * Copyright (C) 2004 Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport |
---|
| 5 | * (Egervary Combinatorial Optimization Research Group, EGRES). |
---|
| 6 | * |
---|
| 7 | * Permission to use, modify and distribute this software is granted |
---|
| 8 | * provided that this copyright notice appears in all copies. For |
---|
| 9 | * precise terms see the accompanying LICENSE file. |
---|
| 10 | * |
---|
| 11 | * This software is provided "AS IS" with no warranty of any kind, |
---|
| 12 | * express or implied, and with no claim as to its suitability for any |
---|
| 13 | * purpose. |
---|
| 14 | * |
---|
| 15 | */ |
---|
| 16 | |
---|
| 17 | #ifndef LEMON_GRAPH_TO_EPS_H |
---|
| 18 | #define LEMON_GRAPH_TO_EPS_H |
---|
| 19 | |
---|
| 20 | #include<iostream> |
---|
| 21 | #include<fstream> |
---|
| 22 | #include<sstream> |
---|
| 23 | #include<algorithm> |
---|
| 24 | #include<vector> |
---|
| 25 | |
---|
| 26 | #include<lemon/xy.h> |
---|
| 27 | #include<lemon/maps.h> |
---|
| 28 | #include<lemon/bezier.h> |
---|
| 29 | |
---|
| 30 | ///\ingroup misc |
---|
| 31 | ///\file |
---|
| 32 | ///\brief Simple graph drawer |
---|
| 33 | /// |
---|
| 34 | ///\author Alpar Juttner |
---|
| 35 | |
---|
| 36 | namespace lemon { |
---|
| 37 | |
---|
| 38 | ///Data structure representing RGB colors. |
---|
| 39 | |
---|
| 40 | ///Data structure representing RGB colors. |
---|
| 41 | ///\ingroup misc |
---|
| 42 | class Color |
---|
| 43 | { |
---|
| 44 | double _r,_g,_b; |
---|
| 45 | public: |
---|
| 46 | ///Default constructor |
---|
| 47 | Color() {} |
---|
| 48 | ///Constructor |
---|
| 49 | Color(double r,double g,double b) :_r(r),_g(g),_b(b) {}; |
---|
| 50 | ///Returns the red component |
---|
| 51 | double getR() {return _r;} |
---|
| 52 | ///Returns the green component |
---|
| 53 | double getG() {return _g;} |
---|
| 54 | ///Returns the blue component |
---|
| 55 | double getB() {return _b;} |
---|
| 56 | ///Set the color components |
---|
| 57 | void set(double r,double g,double b) { _r=r;_g=g;_b=b; }; |
---|
| 58 | }; |
---|
| 59 | |
---|
| 60 | ///Default traits class of \ref GraphToEps |
---|
| 61 | |
---|
| 62 | ///Default traits class of \ref GraphToEps |
---|
| 63 | /// |
---|
| 64 | ///\c G is the type of the underlying graph. |
---|
| 65 | template<class G> |
---|
| 66 | struct DefaultGraphToEpsTraits |
---|
| 67 | { |
---|
| 68 | typedef G Graph; |
---|
| 69 | typedef typename Graph::Node Node; |
---|
| 70 | typedef typename Graph::NodeIt NodeIt; |
---|
| 71 | typedef typename Graph::Edge Edge; |
---|
| 72 | typedef typename Graph::EdgeIt EdgeIt; |
---|
| 73 | typedef typename Graph::InEdgeIt InEdgeIt; |
---|
| 74 | typedef typename Graph::OutEdgeIt OutEdgeIt; |
---|
| 75 | |
---|
| 76 | |
---|
| 77 | const Graph &g; |
---|
| 78 | |
---|
| 79 | std::ostream& os; |
---|
| 80 | |
---|
| 81 | ConstMap<typename Graph::Node,xy<double> > _coords; |
---|
| 82 | ConstMap<typename Graph::Node,double > _nodeSizes; |
---|
[1086] | 83 | ConstMap<typename Graph::Node,int > _nodeShapes; |
---|
[1073] | 84 | |
---|
| 85 | ConstMap<typename Graph::Node,Color > _nodeColors; |
---|
| 86 | ConstMap<typename Graph::Edge,Color > _edgeColors; |
---|
| 87 | |
---|
| 88 | ConstMap<typename Graph::Edge,double > _edgeWidths; |
---|
| 89 | |
---|
| 90 | double _edgeWidthScale; |
---|
| 91 | |
---|
| 92 | double _nodeScale; |
---|
| 93 | double _xBorder, _yBorder; |
---|
| 94 | double _scale; |
---|
| 95 | double _nodeBorderQuotient; |
---|
| 96 | |
---|
| 97 | bool _drawArrows; |
---|
| 98 | double _arrowLength, _arrowWidth; |
---|
| 99 | |
---|
| 100 | bool _showNodes, _showEdges; |
---|
| 101 | |
---|
| 102 | bool _enableParallel; |
---|
| 103 | double _parEdgeDist; |
---|
| 104 | |
---|
| 105 | bool _showNodeText; |
---|
| 106 | ConstMap<typename Graph::Node,bool > _nodeTexts; |
---|
| 107 | double _nodeTextSize; |
---|
| 108 | |
---|
[1085] | 109 | bool _showNodePsText; |
---|
| 110 | ConstMap<typename Graph::Node,bool > _nodePsTexts; |
---|
| 111 | char *_nodePsTextsPreamble; |
---|
| 112 | |
---|
[1073] | 113 | bool _undir; |
---|
| 114 | bool _pleaseRemoveOsStream; |
---|
| 115 | ///Constructor |
---|
| 116 | |
---|
| 117 | ///Constructor |
---|
| 118 | ///\param _g is a reference to the graph to be printed |
---|
| 119 | ///\param _os is a reference to the output stream. |
---|
| 120 | ///\param _os is a reference to the output stream. |
---|
| 121 | ///\param _pros If it is \c true, then the \c ostream referenced by \c _os |
---|
| 122 | ///will be explicitly deallocated by the destructor. |
---|
| 123 | ///By default it is <tt>std::cout</tt> |
---|
| 124 | DefaultGraphToEpsTraits(const G &_g,std::ostream& _os=std::cout, |
---|
| 125 | bool _pros=false) : |
---|
| 126 | g(_g), os(_os), |
---|
[1086] | 127 | _coords(xy<double>(1,1)), _nodeSizes(1.0), _nodeShapes(0), |
---|
[1073] | 128 | _nodeColors(Color(1,1,1)), _edgeColors(Color(0,0,0)), |
---|
| 129 | _edgeWidths(1), _edgeWidthScale(0.3), |
---|
| 130 | _nodeScale(1.0), _xBorder(10), _yBorder(10), _scale(1.0), |
---|
| 131 | _nodeBorderQuotient(.1), |
---|
| 132 | _drawArrows(false), _arrowLength(1), _arrowWidth(0.3), |
---|
| 133 | _showNodes(true), _showEdges(true), |
---|
| 134 | _enableParallel(false), _parEdgeDist(1), |
---|
| 135 | _showNodeText(false), _nodeTexts(false), _nodeTextSize(1), |
---|
[1085] | 136 | _showNodePsText(false), _nodePsTexts(false), _nodePsTextsPreamble(0), |
---|
[1073] | 137 | _undir(false), |
---|
| 138 | _pleaseRemoveOsStream(_pros) {} |
---|
| 139 | }; |
---|
| 140 | |
---|
| 141 | ///Helper class to implement the named parameters of \ref graphToEps() |
---|
| 142 | |
---|
| 143 | ///Helper class to implement the named parameters of \ref graphToEps() |
---|
| 144 | ///\todo Is 'helper class' a good name for this? |
---|
| 145 | /// |
---|
| 146 | template<class T> class GraphToEps : public T |
---|
| 147 | { |
---|
| 148 | typedef typename T::Graph Graph; |
---|
| 149 | typedef typename Graph::Node Node; |
---|
| 150 | typedef typename Graph::NodeIt NodeIt; |
---|
| 151 | typedef typename Graph::Edge Edge; |
---|
| 152 | typedef typename Graph::EdgeIt EdgeIt; |
---|
| 153 | typedef typename Graph::InEdgeIt InEdgeIt; |
---|
| 154 | typedef typename Graph::OutEdgeIt OutEdgeIt; |
---|
| 155 | |
---|
[1087] | 156 | static const int INTERPOL_PREC=20; |
---|
| 157 | |
---|
[1073] | 158 | bool dontPrint; |
---|
| 159 | |
---|
[1086] | 160 | enum NodeShapes { CIRCLE=0, SQUARE=1 }; |
---|
| 161 | |
---|
[1073] | 162 | class edgeLess { |
---|
| 163 | const Graph &g; |
---|
| 164 | public: |
---|
| 165 | edgeLess(const Graph &_g) : g(_g) {} |
---|
| 166 | bool operator()(Edge a,Edge b) const |
---|
| 167 | { |
---|
| 168 | Node ai=min(g.source(a),g.target(a)); |
---|
| 169 | Node aa=max(g.source(a),g.target(a)); |
---|
| 170 | Node bi=min(g.source(b),g.target(b)); |
---|
| 171 | Node ba=max(g.source(b),g.target(b)); |
---|
| 172 | return ai<bi || |
---|
| 173 | (ai==bi && (aa < ba || |
---|
| 174 | (aa==ba && ai==g.source(a) && bi==g.target(b)))); |
---|
| 175 | } |
---|
| 176 | }; |
---|
| 177 | bool isParallel(Edge e,Edge f) const |
---|
| 178 | { |
---|
| 179 | return (g.source(e)==g.source(f)&&g.target(e)==g.target(f))|| |
---|
| 180 | (g.source(e)==g.target(f)&&g.target(e)==g.source(f)); |
---|
| 181 | } |
---|
| 182 | static xy<double> rot(xy<double> v) |
---|
| 183 | { |
---|
| 184 | return xy<double>(v.y,-v.x); |
---|
| 185 | } |
---|
| 186 | template<class xy> |
---|
| 187 | static std::string psOut(const xy &p) |
---|
| 188 | { |
---|
| 189 | std::ostringstream os; |
---|
| 190 | os << p.x << ' ' << p.y; |
---|
| 191 | return os.str(); |
---|
| 192 | } |
---|
| 193 | |
---|
| 194 | public: |
---|
| 195 | GraphToEps(const T &t) : T(t), dontPrint(false) {}; |
---|
| 196 | |
---|
| 197 | template<class X> struct CoordsTraits : public T { |
---|
| 198 | const X &_coords; |
---|
| 199 | CoordsTraits(const T &t,const X &x) : T(t), _coords(x) {} |
---|
| 200 | }; |
---|
| 201 | ///Sets the map of the node coordinates |
---|
| 202 | |
---|
| 203 | ///Sets the map of the node coordinates. |
---|
| 204 | ///\param x must be a node map with xy<double> or xy<int> values. |
---|
| 205 | template<class X> GraphToEps<CoordsTraits<X> > coords(const X &x) { |
---|
| 206 | dontPrint=true; |
---|
| 207 | return GraphToEps<CoordsTraits<X> >(CoordsTraits<X>(*this,x)); |
---|
| 208 | } |
---|
| 209 | template<class X> struct NodeSizesTraits : public T { |
---|
| 210 | const X &_nodeSizes; |
---|
| 211 | NodeSizesTraits(const T &t,const X &x) : T(t), _nodeSizes(x) {} |
---|
| 212 | }; |
---|
| 213 | ///Sets the map of the node sizes |
---|
| 214 | |
---|
| 215 | ///Sets the map of the node sizes |
---|
| 216 | ///\param x must be a node map with \c double (or convertible) values. |
---|
| 217 | template<class X> GraphToEps<NodeSizesTraits<X> > nodeSizes(const X &x) |
---|
| 218 | { |
---|
| 219 | dontPrint=true; |
---|
| 220 | return GraphToEps<NodeSizesTraits<X> >(NodeSizesTraits<X>(*this,x)); |
---|
| 221 | } |
---|
[1086] | 222 | template<class X> struct NodeShapesTraits : public T { |
---|
| 223 | const X &_nodeShapes; |
---|
| 224 | NodeShapesTraits(const T &t,const X &x) : T(t), _nodeShapes(x) {} |
---|
| 225 | }; |
---|
| 226 | ///Sets the map of the node shapes |
---|
| 227 | |
---|
| 228 | ///Sets the map of the node shapes |
---|
| 229 | ///\param x must be a node map with \c int (or convertible) values. |
---|
| 230 | ///\todo Incomplete doc. |
---|
| 231 | template<class X> GraphToEps<NodeShapesTraits<X> > nodeShapes(const X &x) |
---|
| 232 | { |
---|
| 233 | dontPrint=true; |
---|
| 234 | return GraphToEps<NodeShapesTraits<X> >(NodeShapesTraits<X>(*this,x)); |
---|
| 235 | } |
---|
[1073] | 236 | template<class X> struct NodeTextsTraits : public T { |
---|
| 237 | const X &_nodeTexts; |
---|
| 238 | NodeTextsTraits(const T &t,const X &x) : T(t), _nodeTexts(x) {} |
---|
| 239 | }; |
---|
| 240 | ///Sets the text printed on the nodes |
---|
| 241 | |
---|
| 242 | ///Sets the text printed on the nodes |
---|
| 243 | ///\param x must be a node map with type that can be pushed to a standard |
---|
| 244 | ///ostream. |
---|
| 245 | template<class X> GraphToEps<NodeTextsTraits<X> > nodeTexts(const X &x) |
---|
| 246 | { |
---|
| 247 | dontPrint=true; |
---|
| 248 | _showNodeText=true; |
---|
| 249 | return GraphToEps<NodeTextsTraits<X> >(NodeTextsTraits<X>(*this,x)); |
---|
| 250 | } |
---|
[1085] | 251 | template<class X> struct NodePsTextsTraits : public T { |
---|
| 252 | const X &_nodePsTexts; |
---|
| 253 | NodePsTextsTraits(const T &t,const X &x) : T(t), _nodePsTexts(x) {} |
---|
| 254 | }; |
---|
| 255 | ///Inserts a PostScript block to the nodes |
---|
| 256 | |
---|
| 257 | ///With this command it is possible to insert a verbatim PostScript |
---|
| 258 | ///block to the nodes. |
---|
| 259 | ///The PS current point will be moved to the centre of the node before |
---|
| 260 | ///the PostScript block inserted. |
---|
| 261 | /// |
---|
| 262 | ///Before and after the block a newline character is inserted to you |
---|
| 263 | ///don't have to bother with the separators. |
---|
| 264 | /// |
---|
| 265 | ///\param x must be a node map with type that can be pushed to a standard |
---|
| 266 | ///ostream. |
---|
| 267 | /// |
---|
| 268 | ///\sa nodePsTextsPreamble() |
---|
| 269 | ///\todo Offer the choise not to move to the centre but pass the coordinates |
---|
| 270 | ///to the Postscript block inserted. |
---|
| 271 | template<class X> GraphToEps<NodePsTextsTraits<X> > nodePsTexts(const X &x) |
---|
| 272 | { |
---|
| 273 | dontPrint=true; |
---|
| 274 | _showNodePsText=true; |
---|
| 275 | return GraphToEps<NodePsTextsTraits<X> >(NodePsTextsTraits<X>(*this,x)); |
---|
| 276 | } |
---|
| 277 | template<class X> struct EdgeWidthsTraits : public T { |
---|
[1073] | 278 | const X &_edgeWidths; |
---|
| 279 | EdgeWidthsTraits(const T &t,const X &x) : T(t), _edgeWidths(x) {} |
---|
| 280 | }; |
---|
| 281 | ///Sets the map of the edge widths |
---|
| 282 | |
---|
| 283 | ///Sets the map of the edge widths |
---|
| 284 | ///\param x must be a edge map with \c double (or convertible) values. |
---|
| 285 | template<class X> GraphToEps<EdgeWidthsTraits<X> > edgeWidths(const X &x) |
---|
| 286 | { |
---|
| 287 | dontPrint=true; |
---|
| 288 | return GraphToEps<EdgeWidthsTraits<X> >(EdgeWidthsTraits<X>(*this,x)); |
---|
| 289 | } |
---|
| 290 | |
---|
| 291 | template<class X> struct NodeColorsTraits : public T { |
---|
| 292 | const X &_nodeColors; |
---|
| 293 | NodeColorsTraits(const T &t,const X &x) : T(t), _nodeColors(x) {} |
---|
| 294 | }; |
---|
| 295 | ///Sets the map of the node colors |
---|
| 296 | |
---|
| 297 | ///Sets the map of the node colors |
---|
| 298 | ///\param x must be a node map with \ref Color values. |
---|
| 299 | template<class X> GraphToEps<NodeColorsTraits<X> > |
---|
| 300 | nodeColors(const X &x) |
---|
| 301 | { |
---|
| 302 | dontPrint=true; |
---|
| 303 | return GraphToEps<NodeColorsTraits<X> >(NodeColorsTraits<X>(*this,x)); |
---|
| 304 | } |
---|
| 305 | template<class X> struct EdgeColorsTraits : public T { |
---|
| 306 | const X &_edgeColors; |
---|
| 307 | EdgeColorsTraits(const T &t,const X &x) : T(t), _edgeColors(x) {} |
---|
| 308 | }; |
---|
| 309 | ///Sets the map of the edge colors |
---|
| 310 | |
---|
| 311 | ///Sets the map of the edge colors |
---|
| 312 | ///\param x must be a edge map with \ref Color values. |
---|
| 313 | template<class X> GraphToEps<EdgeColorsTraits<X> > |
---|
| 314 | edgeColors(const X &x) |
---|
| 315 | { |
---|
| 316 | dontPrint=true; |
---|
| 317 | return GraphToEps<EdgeColorsTraits<X> >(EdgeColorsTraits<X>(*this,x)); |
---|
| 318 | } |
---|
| 319 | ///Sets a global scale factor for node sizes |
---|
| 320 | |
---|
| 321 | ///Sets a global scale factor for node sizes |
---|
| 322 | /// |
---|
| 323 | GraphToEps<T> &nodeScale(double d) {_nodeScale=d;return *this;} |
---|
| 324 | ///Sets a global scale factor for edge widths |
---|
| 325 | |
---|
| 326 | ///Sets a global scale factor for edge widths |
---|
| 327 | /// |
---|
| 328 | GraphToEps<T> &edgeWidthScale(double d) {_edgeWidthScale=d;return *this;} |
---|
| 329 | ///Sets a global scale factor for the whole picture |
---|
| 330 | |
---|
| 331 | ///Sets a global scale factor for the whole picture |
---|
| 332 | /// |
---|
| 333 | GraphToEps<T> &scale(double d) {_scale=d;return *this;} |
---|
| 334 | ///Sets the width of the border around the picture |
---|
| 335 | |
---|
| 336 | ///Sets the width of the border around the picture |
---|
| 337 | /// |
---|
| 338 | GraphToEps<T> &border(double b) {_xBorder=_yBorder=b;return *this;} |
---|
| 339 | ///Sets the width of the border around the picture |
---|
| 340 | |
---|
| 341 | ///Sets the width of the border around the picture |
---|
| 342 | /// |
---|
| 343 | GraphToEps<T> &border(double x, double y) { |
---|
| 344 | _xBorder=x;_yBorder=y;return *this; |
---|
| 345 | } |
---|
| 346 | ///Sets whether to draw arrows |
---|
| 347 | |
---|
| 348 | ///Sets whether to draw arrows |
---|
| 349 | /// |
---|
| 350 | GraphToEps<T> &drawArrows(bool b=true) {_drawArrows=b;return *this;} |
---|
| 351 | ///Sets the length of the arrowheads |
---|
| 352 | |
---|
| 353 | ///Sets the length of the arrowheads |
---|
| 354 | /// |
---|
| 355 | GraphToEps<T> &arrowLength(double d) {_arrowLength*=d;return *this;} |
---|
| 356 | ///Sets the width of the arrowheads |
---|
| 357 | |
---|
| 358 | ///Sets the width of the arrowheads |
---|
| 359 | /// |
---|
| 360 | GraphToEps<T> &arrowWidth(double d) {_arrowWidth*=d;return *this;} |
---|
| 361 | |
---|
| 362 | ///Enables parallel edges |
---|
| 363 | |
---|
| 364 | ///Enables parallel edges |
---|
| 365 | ///\todo Partially implemented |
---|
| 366 | GraphToEps<T> &enableParallel(bool b=true) {_enableParallel=b;return *this;} |
---|
| 367 | |
---|
| 368 | ///Sets the distance |
---|
| 369 | |
---|
| 370 | ///Sets the distance |
---|
| 371 | /// |
---|
| 372 | GraphToEps<T> &parEdgeDist(double d) {_parEdgeDist*=d;return *this;} |
---|
| 373 | |
---|
| 374 | ///Hides the edges |
---|
| 375 | |
---|
| 376 | ///Hides the edges |
---|
| 377 | /// |
---|
| 378 | GraphToEps<T> &hideEdges(bool b=true) {_showEdges=!b;return *this;} |
---|
| 379 | ///Hides the nodes |
---|
| 380 | |
---|
| 381 | ///Hides the nodes |
---|
| 382 | /// |
---|
| 383 | GraphToEps<T> &hideNodes(bool b=true) {_showNodes=!b;return *this;} |
---|
| 384 | |
---|
| 385 | ///Sets the size of the node texts |
---|
| 386 | |
---|
| 387 | ///Sets the size of the node texts |
---|
| 388 | /// |
---|
| 389 | GraphToEps<T> &nodeTextSize(double d) {_nodeTextSize=d;return *this;} |
---|
[1085] | 390 | ///Gives a preamble block for node Postscript block. |
---|
| 391 | |
---|
| 392 | ///Gives a preamble block for node Postscript block. |
---|
| 393 | /// |
---|
| 394 | ///\sa nodePsTexts() |
---|
| 395 | GraphToEps<T> & nodePsTextsPreamble(const char *str) { |
---|
| 396 | _nodePsTextsPreamble=s ;return *this; |
---|
| 397 | } |
---|
[1073] | 398 | ///Sets whether the the graph is undirected |
---|
| 399 | |
---|
| 400 | ///Sets whether the the graph is undirected |
---|
| 401 | /// |
---|
| 402 | GraphToEps<T> &undir(bool b=true) {_undir=b;return *this;} |
---|
| 403 | ///Sets whether the the graph is directed |
---|
| 404 | |
---|
| 405 | ///Sets whether the the graph is directed. |
---|
| 406 | ///Use it to show the undirected edges as a pair of directed ones. |
---|
| 407 | GraphToEps<T> &bidir(bool b=true) {_undir=!b;return *this;} |
---|
[1086] | 408 | |
---|
| 409 | protected: |
---|
| 410 | bool isInsideNode(xy<double> p, double r,int t) |
---|
| 411 | { |
---|
| 412 | switch(t) { |
---|
| 413 | case CIRCLE: |
---|
| 414 | return p.normSquare()<=r*r; |
---|
| 415 | case SQUARE: |
---|
| 416 | return p.x<=r&&p.x>=-r&&p.y<=r&&p.y>=-r; |
---|
| 417 | } |
---|
| 418 | return false; |
---|
| 419 | } |
---|
| 420 | |
---|
| 421 | public: |
---|
[1073] | 422 | ~GraphToEps() |
---|
| 423 | { |
---|
| 424 | if(dontPrint) return; |
---|
| 425 | |
---|
| 426 | os << "%!PS-Adobe-2.0 EPSF-2.0\n"; |
---|
| 427 | //\todo: Chech whether the graph is empty. |
---|
| 428 | BoundingBox<double> bb; |
---|
| 429 | for(NodeIt n(g);n!=INVALID;++n) { |
---|
| 430 | double ns=_nodeSizes[n]*_nodeScale; |
---|
| 431 | xy<double> p(ns,ns); |
---|
| 432 | bb+=p+_coords[n]; |
---|
| 433 | bb+=-p+_coords[n]; |
---|
| 434 | } |
---|
| 435 | os << "%%BoundingBox: " |
---|
| 436 | << bb.left()* _scale-_xBorder << ' ' |
---|
| 437 | << bb.bottom()*_scale-_yBorder << ' ' |
---|
| 438 | << bb.right()* _scale+_xBorder << ' ' |
---|
| 439 | << bb.top()* _scale+_yBorder << '\n'; |
---|
| 440 | //x1 y1 x2 y2 x3 y3 cr cg cb w |
---|
| 441 | os << "/lb { setlinewidth setrgbcolor newpath moveto\n" |
---|
| 442 | << " 4 2 roll 1 index 1 index curveto stroke } bind def\n"; |
---|
| 443 | os << "/l { setlinewidth setrgbcolor newpath moveto lineto stroke } bind def\n"; |
---|
[1086] | 444 | //x y r |
---|
[1073] | 445 | os << "/c { newpath dup 3 index add 2 index moveto 0 360 arc closepath } bind def\n"; |
---|
[1086] | 446 | //x y r |
---|
| 447 | os << "/sq { newpath 2 index 1 index add 2 index 2 index add moveto\n" |
---|
| 448 | << " 2 index 1 index sub 2 index 2 index add lineto\n" |
---|
| 449 | << " 2 index 1 index sub 2 index 2 index sub lineto\n" |
---|
| 450 | << " 2 index 1 index add 2 index 2 index sub lineto\n" |
---|
| 451 | << " closepath pop pop pop} bind def\n"; |
---|
[1073] | 452 | // x y r cr cg cb |
---|
[1086] | 453 | os << "/nc { setrgbcolor 2 index 2 index 2 index c fill\n" |
---|
[1073] | 454 | << " 0 0 0 setrgbcolor dup " |
---|
| 455 | << _nodeBorderQuotient << " mul setlinewidth " |
---|
| 456 | << 1+_nodeBorderQuotient/2 << " div c stroke\n" |
---|
| 457 | << " } bind def\n"; |
---|
[1086] | 458 | os << "/nsq { setrgbcolor 2 index 2 index 2 index sq fill\n" |
---|
| 459 | << " 0 0 0 setrgbcolor dup " |
---|
| 460 | << _nodeBorderQuotient << " mul setlinewidth " |
---|
| 461 | << 1+_nodeBorderQuotient/2 << " div sq stroke\n" |
---|
| 462 | << " } bind def\n"; |
---|
[1073] | 463 | os << "/arrl " << _arrowLength << " def\n"; |
---|
| 464 | os << "/arrw " << _arrowWidth << " def\n"; |
---|
| 465 | // l dx_norm dy_norm |
---|
| 466 | os << "/lrl { 2 index mul exch 2 index mul exch rlineto pop} bind def\n"; |
---|
| 467 | //len w dx_norm dy_norm x1 y1 cr cg cb |
---|
| 468 | os << "/arr { setrgbcolor /y1 exch def /x1 exch def /dy exch def /dx exch def\n" |
---|
| 469 | << " /w exch def /len exch def\n" |
---|
| 470 | // << " 0.1 setlinewidth x1 y1 moveto dx len mul dy len mul rlineto stroke" |
---|
| 471 | << " newpath x1 dy w 2 div mul add y1 dx w 2 div mul sub moveto\n" |
---|
| 472 | << " len w sub arrl sub dx dy lrl\n" |
---|
| 473 | << " arrw dy dx neg lrl\n" |
---|
| 474 | << " dx arrl w add mul dy w 2 div arrw add mul sub\n" |
---|
| 475 | << " dy arrl w add mul dx w 2 div arrw add mul add rlineto\n" |
---|
| 476 | << " dx arrl w add mul neg dy w 2 div arrw add mul sub\n" |
---|
| 477 | << " dy arrl w add mul neg dx w 2 div arrw add mul add rlineto\n" |
---|
| 478 | << " arrw dy dx neg lrl\n" |
---|
| 479 | << " len w sub arrl sub neg dx dy lrl\n" |
---|
| 480 | << " closepath fill } bind def\n"; |
---|
| 481 | os << "/cshow { 2 index 2 index moveto dup stringwidth pop\n" |
---|
| 482 | << " neg 2 div fosi .35 mul neg rmoveto show pop pop} def\n"; |
---|
| 483 | |
---|
| 484 | os << "\ngsave\n"; |
---|
| 485 | if(_scale!=1.0) os << _scale << " dup scale\n"; |
---|
| 486 | |
---|
[1085] | 487 | if(_showEdges) { |
---|
| 488 | os << "%Edges:\ngsave\n"; |
---|
[1073] | 489 | if(_enableParallel) { |
---|
| 490 | std::vector<Edge> el; |
---|
| 491 | for(EdgeIt e(g);e!=INVALID;++e) |
---|
| 492 | if(!_undir||g.source(e)<g.target(e)) el.push_back(e); |
---|
| 493 | sort(el.begin(),el.end(),edgeLess(g)); |
---|
| 494 | |
---|
| 495 | typename std::vector<Edge>::iterator j; |
---|
| 496 | for(typename std::vector<Edge>::iterator i=el.begin();i!=el.end();i=j) { |
---|
| 497 | for(j=i+1;j!=el.end()&&isParallel(*i,*j);++j) ; |
---|
| 498 | |
---|
| 499 | double sw=0; |
---|
| 500 | for(typename std::vector<Edge>::iterator e=i;e!=j;++e) |
---|
| 501 | sw+=_edgeWidths[*e]*_edgeWidthScale+_parEdgeDist; |
---|
| 502 | sw-=_parEdgeDist; |
---|
| 503 | sw/=-2.0; |
---|
[1085] | 504 | xy<double> dvec(_coords[g.target(*i)]-_coords[g.source(*i)]); |
---|
| 505 | double l=sqrt(dvec.normSquare()); |
---|
| 506 | xy<double> d(dvec/l); |
---|
| 507 | xy<double> m; |
---|
| 508 | // m=xy<double>(_coords[g.target(*i)]+_coords[g.source(*i)])/2.0; |
---|
| 509 | |
---|
| 510 | // m=xy<double>(_coords[g.source(*i)])+ |
---|
| 511 | // dvec*(double(_nodeSizes[g.source(*i)])/ |
---|
| 512 | // (_nodeSizes[g.source(*i)]+_nodeSizes[g.target(*i)])); |
---|
| 513 | |
---|
| 514 | m=xy<double>(_coords[g.source(*i)])+ |
---|
| 515 | d*(l+_nodeSizes[g.source(*i)]-_nodeSizes[g.target(*i)])/2.0; |
---|
| 516 | |
---|
[1073] | 517 | for(typename std::vector<Edge>::iterator e=i;e!=j;++e) { |
---|
| 518 | sw+=_edgeWidths[*e]*_edgeWidthScale/2.0; |
---|
[1085] | 519 | xy<double> mm=m+rot(d)*sw/.75; |
---|
[1073] | 520 | if(_drawArrows) { |
---|
[1086] | 521 | int node_shape; |
---|
[1073] | 522 | xy<double> s=_coords[g.source(*e)]; |
---|
| 523 | xy<double> t=_coords[g.target(*e)]; |
---|
| 524 | double rn=_nodeSizes[g.target(*e)]*_nodeScale; |
---|
[1086] | 525 | node_shape=_nodeShapes[g.target(*e)]; |
---|
[1085] | 526 | Bezier3 bez(s,mm,mm,t); |
---|
[1073] | 527 | double t1=0,t2=1; |
---|
[1087] | 528 | for(int i=0;i<INTERPOL_PREC;++i) |
---|
[1086] | 529 | if(isInsideNode(bez((t1+t2)/2)-t,rn,node_shape)) t2=(t1+t2)/2; |
---|
| 530 | else t1=(t1+t2)/2; |
---|
[1073] | 531 | xy<double> apoint=bez((t1+t2)/2); |
---|
[1086] | 532 | rn = _arrowLength+_edgeWidths[*e]*_edgeWidthScale; |
---|
[1073] | 533 | rn*=rn; |
---|
[1086] | 534 | t2=(t1+t2)/2;t1=0; |
---|
[1087] | 535 | for(int i=0;i<INTERPOL_PREC;++i) |
---|
[1086] | 536 | if((bez((t1+t2)/2)-apoint).normSquare()>rn) t1=(t1+t2)/2; |
---|
[1073] | 537 | else t2=(t1+t2)/2; |
---|
| 538 | xy<double> linend=bez((t1+t2)/2); |
---|
| 539 | bez=bez.before((t1+t2)/2); |
---|
[1086] | 540 | // rn=_nodeSizes[g.source(*e)]*_nodeScale; |
---|
| 541 | // node_shape=_nodeShapes[g.source(*e)]; |
---|
| 542 | // t1=0;t2=1; |
---|
[1087] | 543 | // for(int i=0;i<INTERPOL_PREC;++i) |
---|
[1086] | 544 | // if(isInsideNode(bez((t1+t2)/2)-t,rn,node_shape)) t1=(t1+t2)/2; |
---|
| 545 | // else t2=(t1+t2)/2; |
---|
| 546 | // bez=bez.after((t1+t2)/2); |
---|
[1073] | 547 | os << _edgeWidths[*e]*_edgeWidthScale << " setlinewidth " |
---|
| 548 | << _edgeColors[*e].getR() << ' ' |
---|
| 549 | << _edgeColors[*e].getG() << ' ' |
---|
| 550 | << _edgeColors[*e].getB() << " setrgbcolor newpath\n" |
---|
| 551 | << bez.p1.x << ' ' << bez.p1.y << " moveto\n" |
---|
| 552 | << bez.p2.x << ' ' << bez.p2.y << ' ' |
---|
| 553 | << bez.p3.x << ' ' << bez.p3.y << ' ' |
---|
| 554 | << bez.p4.x << ' ' << bez.p4.y << " curveto stroke\n"; |
---|
| 555 | xy<double> dd(rot(linend-apoint)); |
---|
| 556 | dd*=(_edgeWidths[*e]*_edgeWidthScale+_arrowWidth)/ |
---|
| 557 | sqrt(dd.normSquare()); |
---|
| 558 | os << "newpath " << psOut(apoint) << " moveto " |
---|
| 559 | << psOut(linend+dd) << " lineto " |
---|
| 560 | << psOut(linend-dd) << " lineto closepath fill\n"; |
---|
| 561 | } |
---|
| 562 | else { |
---|
| 563 | os << _coords[g.source(*e)].x << ' ' |
---|
| 564 | << _coords[g.source(*e)].y << ' ' |
---|
[1085] | 565 | << mm.x << ' ' << mm.y << ' ' |
---|
[1073] | 566 | << _coords[g.target(*e)].x << ' ' |
---|
| 567 | << _coords[g.target(*e)].y << ' ' |
---|
| 568 | << _edgeColors[*e].getR() << ' ' |
---|
| 569 | << _edgeColors[*e].getG() << ' ' |
---|
| 570 | << _edgeColors[*e].getB() << ' ' |
---|
| 571 | << _edgeWidths[*e]*_edgeWidthScale << " lb\n"; |
---|
| 572 | } |
---|
| 573 | sw+=_edgeWidths[*e]*_edgeWidthScale/2.0+_parEdgeDist; |
---|
| 574 | } |
---|
| 575 | } |
---|
| 576 | } |
---|
| 577 | else for(EdgeIt e(g);e!=INVALID;++e) |
---|
| 578 | if(!_undir||g.source(e)<g.target(e)) |
---|
| 579 | if(_drawArrows) { |
---|
| 580 | xy<double> d(_coords[g.target(e)]-_coords[g.source(e)]); |
---|
[1087] | 581 | double rn=_nodeSizes[g.target(e)]*_nodeScale; |
---|
| 582 | int node_shape=_nodeShapes[g.target(e)]; |
---|
| 583 | double t1=0,t2=1; |
---|
| 584 | for(int i=0;i<INTERPOL_PREC;++i) |
---|
| 585 | if(isInsideNode((-(t1+t2)/2)*d,rn,node_shape)) t1=(t1+t2)/2; |
---|
| 586 | else t2=(t1+t2)/2; |
---|
[1073] | 587 | double l=sqrt(d.normSquare()); |
---|
| 588 | d/=l; |
---|
[1087] | 589 | |
---|
| 590 | os << l*(1-(t1+t2)/2) << ' ' |
---|
[1073] | 591 | << _edgeWidths[e]*_edgeWidthScale << ' ' |
---|
| 592 | << d.x << ' ' << d.y << ' ' |
---|
[1087] | 593 | << _coords[g.source(e)].x << ' ' |
---|
| 594 | << _coords[g.source(e)].y << ' ' |
---|
[1073] | 595 | << _edgeColors[e].getR() << ' ' |
---|
| 596 | << _edgeColors[e].getG() << ' ' |
---|
| 597 | << _edgeColors[e].getB() << " arr\n"; |
---|
| 598 | } |
---|
| 599 | else os << _coords[g.source(e)].x << ' ' |
---|
| 600 | << _coords[g.source(e)].y << ' ' |
---|
| 601 | << _coords[g.target(e)].x << ' ' |
---|
| 602 | << _coords[g.target(e)].y << ' ' |
---|
| 603 | << _edgeColors[e].getR() << ' ' |
---|
| 604 | << _edgeColors[e].getG() << ' ' |
---|
| 605 | << _edgeColors[e].getB() << ' ' |
---|
| 606 | << _edgeWidths[e]*_edgeWidthScale << " l\n"; |
---|
[1085] | 607 | os << "grestore\n"; |
---|
| 608 | } |
---|
| 609 | if(_showNodes) { |
---|
| 610 | os << "%Nodes:\ngsave\n"; |
---|
[1086] | 611 | for(NodeIt n(g);n!=INVALID;++n) { |
---|
[1073] | 612 | os << _coords[n].x << ' ' << _coords[n].y << ' ' |
---|
| 613 | << _nodeSizes[n]*_nodeScale << ' ' |
---|
| 614 | << _nodeColors[n].getR() << ' ' |
---|
| 615 | << _nodeColors[n].getG() << ' ' |
---|
[1086] | 616 | << _nodeColors[n].getB() << ' '; |
---|
| 617 | switch(_nodeShapes[n]) { |
---|
| 618 | case CIRCLE: |
---|
| 619 | os<< "nc";break; |
---|
| 620 | case SQUARE: |
---|
| 621 | os<< "nsq";break; |
---|
| 622 | } |
---|
| 623 | os<<'\n'; |
---|
| 624 | } |
---|
[1085] | 625 | os << "grestore\n"; |
---|
| 626 | } |
---|
[1073] | 627 | if(_showNodeText) { |
---|
[1085] | 628 | os << "%Node texts:\ngsave\n"; |
---|
[1073] | 629 | os << "/fosi " << _nodeTextSize << " def\n"; |
---|
| 630 | os << "(Helvetica) findfont fosi scalefont setfont\n"; |
---|
| 631 | os << "0 0 0 setrgbcolor\n"; |
---|
| 632 | for(NodeIt n(g);n!=INVALID;++n) |
---|
| 633 | os << _coords[n].x << ' ' << _coords[n].y |
---|
| 634 | << " (" << _nodeTexts[n] << ") cshow\n"; |
---|
[1085] | 635 | os << "grestore\n"; |
---|
[1073] | 636 | } |
---|
[1085] | 637 | if(_showNodePsText) { |
---|
| 638 | os << "%Node PS blocks:\ngsave\n"; |
---|
| 639 | for(NodeIt n(g);n!=INVALID;++n) |
---|
| 640 | os << _coords[n].x << ' ' << _coords[n].y |
---|
| 641 | << " moveto\n" << _nodePsTexts[n] << "\n"; |
---|
| 642 | os << "grestore\n"; |
---|
| 643 | } |
---|
| 644 | |
---|
| 645 | os << "grestore\n"; |
---|
[1073] | 646 | |
---|
| 647 | //CleanUp: |
---|
| 648 | if(_pleaseRemoveOsStream) {delete &os;} |
---|
| 649 | } |
---|
| 650 | }; |
---|
| 651 | |
---|
| 652 | |
---|
| 653 | ///Generates an EPS file from a graph |
---|
| 654 | |
---|
| 655 | ///\ingroup misc |
---|
| 656 | ///Generates an EPS file from a graph. |
---|
| 657 | ///\param g is a reference to the graph to be printed |
---|
| 658 | ///\param os is a reference to the output stream. |
---|
| 659 | ///By default it is <tt>std::cout</tt> |
---|
| 660 | /// |
---|
| 661 | ///This function also has a lot of \ref named-templ-param "named parameters", |
---|
| 662 | ///they are declared as the members of class \ref GraphToEps. The following |
---|
| 663 | ///example shows how to use these parameters. |
---|
| 664 | ///\code |
---|
| 665 | /// graphToEps(g).scale(10).coords(coords) |
---|
| 666 | /// .nodeScale(2).nodeSizes(sizes) |
---|
| 667 | /// .edgeWidthScale(.4); |
---|
| 668 | ///\endcode |
---|
| 669 | ///\sa GraphToEps |
---|
| 670 | ///\sa graphToEps(G &g, char *file_name) |
---|
| 671 | template<class G> |
---|
| 672 | GraphToEps<DefaultGraphToEpsTraits<G> > |
---|
| 673 | graphToEps(G &g, std::ostream& os=std::cout) |
---|
| 674 | { |
---|
| 675 | return |
---|
| 676 | GraphToEps<DefaultGraphToEpsTraits<G> >(DefaultGraphToEpsTraits<G>(g,os)); |
---|
| 677 | } |
---|
| 678 | |
---|
| 679 | ///Generates an EPS file from a graph |
---|
| 680 | |
---|
| 681 | //\ingroup misc |
---|
| 682 | ///This function does the same as |
---|
| 683 | ///\ref graphToEps(G &g,std::ostream& os) |
---|
| 684 | ///but it writes its output into the file \c file_name |
---|
| 685 | ///instead of a stream. |
---|
| 686 | ///\sa graphToEps(G &g, std::ostream& os) |
---|
| 687 | template<class G> |
---|
| 688 | GraphToEps<DefaultGraphToEpsTraits<G> > |
---|
| 689 | graphToEps(G &g,char *file_name) |
---|
| 690 | { |
---|
| 691 | return GraphToEps<DefaultGraphToEpsTraits<G> > |
---|
| 692 | (DefaultGraphToEpsTraits<G>(g,*new std::ofstream(file_name),true)); |
---|
| 693 | } |
---|
| 694 | |
---|
| 695 | } //END OF NAMESPACE LEMON |
---|
| 696 | |
---|
| 697 | #endif // LEMON_GRAPH_TO_EPS_H |
---|