RhodeCode

  /// Dark yellow color constant

  extern const Color DARK_YELLOW;

  /// Dark magenta color constant

  extern const Color DARK_MAGENTA;

  /// Dark cyan color constant

  extern const Color DARK_CYAN;

  ///Map <tt>int</tt>s to different \ref Color "Color"s

  ///This map assigns one of the predefined \ref Color "Color"s to

  ///each <tt>int</tt>. It is possible to change the colors as well as

  ///their number. The integer range is cyclically mapped to the

  ///provided set of colors.

  ///

  ///This is a true \ref concepts::ReferenceMap "reference map", so

  ///you can also change the actual colors.

  class Palette : public MapBase<int,Color>

  {

    std::vector<Color> colors;

  public:

    ///Constructor

    ///Constructor 

    ///\param num the number of the allocated colors. If it is \c -1,

    ///white).  If \c num is less then 26/27 then the default color

    ///list is cut. Otherwise the color list is filled repeatedly with

    ///the default color list.  (The colors can be changed later on.)

    Palette(bool have_white=false,int num=-1)

    {

      if (num==0) return;

      do {

        if(have_white) colors.push_back(Color(1,1,1));

        colors.push_back(Color(0,0,0));

        colors.push_back(Color(1,0,0));

        colors.push_back(Color(0,1,0));

        colors.push_back(Color(0,0,1));

        colors.push_back(Color(1,1,0));

        colors.push_back(Color(1,0,1));

        colors.push_back(Color(0,1,1));

        colors.push_back(Color(.5,0,0));

        colors.push_back(Color(0,.5,0));

        colors.push_back(Color(0,0,.5));

        colors.push_back(Color(.5,.5,0));

        colors.push_back(Color(.5,0,.5));

        colors.push_back(Color(0,.5,.5));

        colors.push_back(Color(.5,.5,.5));

        colors.push_back(Color(1,.5,.5));

        colors.push_back(Color(.5,1,.5));

        colors.push_back(Color(.5,1,0));

        colors.push_back(Color(1,0,.5));

        colors.push_back(Color(0,1,.5));

        colors.push_back(Color(0,.5,1));

        colors.push_back(Color(.5,0,1));

      } while(int(colors.size())<num);

      //    colors.push_back(Color(1,1,1));

      if(num>=0) colors.resize(num);

    }

    ///\e

    Color &operator[](int i)

    {

      return colors[i%colors.size()];

    }

    ///\e

    const Color &operator[](int i) const

    {

      return colors[i%colors.size()];

    }

    ///\e

    void set(int i,const Color &c)

    {

      colors[i%colors.size()]=c;

    }

    void add(const Color &c) 

    {

      colors.push_back(c);

    }

    ///Sets the number of the exiting colors.

    void resize(int s) { colors.resize(s);}

    ///Returns the number of the existing colors.

    int size() const { return int(colors.size());}

  };

  ///Returns a \ref Color which is as different from the given parameter

  ///as it is possible.

  inline Color distantColor(const Color &c) 

  {

    return Color(c.red()<.5?1:0,c.green()<.5?1:0,c.blue()<.5?1:0);

  }

  ///Returns black for light colors and white for the dark ones.

  ///Returns black for light colors and white for the dark ones.

  inline Color distantBW(const Color &c){

    return (.2125*c.red()+.7154*c.green()+.0721*c.blue())<.5 ? WHITE : BLACK;

  }

  /// @}

} //END OF NAMESPACE LEMON

#endif // LEMON_COLOR_H

#include <sys/time.h>

#ifdef WIN32

#include <lemon/bits/mingw32_time.h>

#endif

#include<iostream>

#include<fstream>

#include<sstream>

#include<algorithm>

#include<vector>

#include<ctime>

#include<lemon/math.h>

#include<lemon/bits/invalid.h>

#include<lemon/dim2.h>

#include<lemon/maps.h>

#include<lemon/color.h>

#include<lemon/bits/bezier.h>

///\ingroup eps_io

///\file

namespace lemon {

  namespace _graph_to_eps_bits {

    template<class MT>

    class _NegY {

    public:

      typedef typename MT::Key Key;

      typedef typename MT::Value Value;

      const MT ↦

      int yscale;

      _NegY(const MT &m,bool b) : map(m), yscale(1-b*2) {}

      Value operator[](Key n) { return Value(map[n].x,map[n].y*yscale);}

    };

  }

///Default traits class of \ref GraphToEps

///Default traits class of \ref GraphToEps

///

///\c G is the type of the underlying graph.

template<class G>

struct DefaultGraphToEpsTraits

{

			  bool _pros=false) :

    g(_g), os(_os),

    _coords(dim2::Point<double>(1,1)), _nodeSizes(1), _nodeShapes(0),

    _nodeColors(WHITE), _arcColors(BLACK),

    _arcWidths(1.0), _arcWidthScale(0.003),

    _nodeScale(.01), _xBorder(10), _yBorder(10), _scale(1.0),

    _nodeBorderQuotient(.1),

    _drawArrows(false), _arrowLength(1), _arrowWidth(0.3),

    _showNodes(true), _showArcs(true),

    _enableParallel(false), _parArcDist(1),

    _showNodeText(false), _nodeTexts(false), _nodeTextSize(1),

    _showNodePsText(false), _nodePsTexts(false), _nodePsTextsPreamble(0),

    _undirected(lemon::UndirectedTagIndicator<G>::value),

    _pleaseRemoveOsStream(_pros), _scaleToA4(false),

    _nodeTextColorType(SAME_COL), _nodeTextColors(BLACK),

    _autoNodeScale(false),

    _autoArcWidthScale(false),

    _absoluteNodeSizes(false),

    _absoluteArcWidths(false),

    _negY(false),

    _preScale(true)

  {}

};

template<class T> class GraphToEps : public T 

{

  // Can't believe it is required by the C++ standard

  using T::g;

  using T::os;

  using T::_coords;

  using T::_nodeSizes;

  using T::_nodeShapes;

  using T::_nodeColors;

  using T::_arcColors;

  using T::_arcWidths;

  using T::_arcWidthScale;

  using T::_nodeScale;

  using T::_xBorder;

  using T::_yBorder;

  using T::_scale;

  using T::_nodeBorderQuotient;

  using T::_drawArrows;

  using T::_arrowLength;

  using T::_arrowWidth;

  template<class X> GraphToEps<CoordsTraits<X> > coords(const X &x) {

    dontPrint=true;

    return GraphToEps<CoordsTraits<X> >(CoordsTraits<X>(*this,x));

  }

  template<class X> struct NodeSizesTraits : public T {

    const X &_nodeSizes;

    NodeSizesTraits(const T &t,const X &x) : T(t), _nodeSizes(x) {}

  };

  ///Sets the map of the node sizes

  ///Sets the map of the node sizes

  ///\param x must be a node map with \c double (or convertible) values. 

  template<class X> GraphToEps<NodeSizesTraits<X> > nodeSizes(const X &x)

  {

    dontPrint=true;

    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.

  ///can be found in \ref NodeShapes "enum NodeShapes".

  ///\param x must be a node map with \c int (or convertible) values. 

  ///\sa NodeShapes

  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;

    NodeTextsTraits(const T &t,const X &x) : T(t), _nodeTexts(x) {}

  };

  ///Sets the text printed on the nodes

  ///Sets the text printed on the nodes

  ///\param x must be a node map with type that can be pushed to a standard

  ///ostream. 

  template<class X> GraphToEps<NodeTextsTraits<X> > nodeTexts(const X &x)

  {

    dontPrint=true;

    _showNodeText=true;

    return GraphToEps<NodeTextsTraits<X> >(NodeTextsTraits<X>(*this,x));

  }

  template<class X> struct NodePsTextsTraits : public T {

    const X &_nodePsTexts;

    NodePsTextsTraits(const T &t,const X &x) : T(t), _nodePsTexts(x) {}

  };

  ///Inserts a PostScript block to the nodes

  ///With this command it is possible to insert a verbatim PostScript

  ///block to the nodes.

  ///The PS current point will be moved to the centre of the node before

  ///the PostScript block inserted.

  ///

  ///Before and after the block a newline character is inserted so you

  ///don't have to bother with the separators.

  ///

  ///\param x must be a node map with type that can be pushed to a standard

  ///ostream.

  ///

  ///\sa nodePsTextsPreamble()

  template<class X> GraphToEps<NodePsTextsTraits<X> > nodePsTexts(const X &x)

  {

    dontPrint=true;

    _showNodePsText=true;

    return GraphToEps<NodePsTextsTraits<X> >(NodePsTextsTraits<X>(*this,x));

  }

  template<class X> struct ArcWidthsTraits : public T {

    const X &_arcWidths;

    ArcWidthsTraits(const T &t,const X &x) : T(t), _arcWidths(x) {}

  };

  ///Sets the map of the arc widths

  ///Sets the map of the arc widths

  ///\param x must be a arc map with \c double (or convertible) values. 

  template<class X> GraphToEps<ArcWidthsTraits<X> > arcWidths(const X &x)

  {

    dontPrint=true;

    return GraphToEps<ArcWidthsTraits<X> >(ArcWidthsTraits<X>(*this,x));

  }

  template<class X> struct NodeColorsTraits : public T {

    const X &_nodeColors;

    NodeColorsTraits(const T &t,const X &x) : T(t), _nodeColors(x) {}

  };

  ///Turns on/off the automatic node width scaling.

  ///Turns on/off the automatic node width scaling.

  ///

  ///\sa nodeScale()

  ///

  GraphToEps<T> &autoNodeScale(bool b=true) {

    _autoNodeScale=b;return *this;

  }

  ///Turns on/off the absolutematic node width scaling.

  ///Turns on/off the absolutematic node width scaling.

  ///

  ///\sa nodeScale()

  ///

  GraphToEps<T> &absoluteNodeSizes(bool b=true) {

    _absoluteNodeSizes=b;return *this;

  }

  ///Negates the Y coordinates.

  ///Negates the Y coordinates.

  ///

  GraphToEps<T> &negateY(bool b=true) {

    _negY=b;return *this;

  }

  ///By default graphToEps() rescales the whole image in order to avoid

  ///very big or very small bounding boxes.

  ///

  ///This (p)rescaling can be turned off with this function.

  ///

  GraphToEps<T> &preScale(bool b=true) {

    _preScale=b;return *this;

  }

  ///Sets a global scale factor for arc widths

  /// Sets a global scale factor for arc widths.

  ///

  /// If arcWidths() is not given, this function simply sets the arc

  /// widths to \c d.  If arcWidths() is given, but

  /// autoArcWidthScale() is not, then the arc withs given by

  /// arcWidths() will be multiplied by the value \c d.

  /// If both arcWidths() and autoArcWidthScale() are used, then the

  /// arc withs will be scaled in such a way that the greatest width will be

  /// equal to \c d.

  GraphToEps<T> &arcWidthScale(double d=.003) {_arcWidthScale=d;return *this;}

  ///Turns on/off the automatic arc width scaling.

  ///\sa arcWidthScale()

  ///

  GraphToEps<T> &autoArcWidthScale(bool b=true) {

    _autoArcWidthScale=b;return *this;

  }

  ///Turns on/off the absolutematic arc width scaling.

  ///Turns on/off the absolutematic arc width scaling.

  ///

  ///\sa arcWidthScale()

  ///

  GraphToEps<T> &absoluteArcWidths(bool b=true) {

    _absoluteArcWidths=b;return *this;

  }

  ///Sets a global scale factor for the whole picture

  ///Sets a global scale factor for the whole picture

  ///

  GraphToEps<T> &scale(double d) {_scale=d;return *this;}

  ///Sets the width of the border around the picture

  ///Sets the width of the border around the picture

  ///

  ///Sets the width of the border around the picture

  ///Sets the width of the border around the picture

  ///

  GraphToEps<T> &border(double x, double y) {

    _xBorder=x;_yBorder=y;return *this;

  }

  ///Sets whether to draw arrows

  ///Sets whether to draw arrows

  ///

  GraphToEps<T> &drawArrows(bool b=true) {_drawArrows=b;return *this;}

  ///Sets the length of the arrowheads

  ///Sets the length of the arrowheads

  ///

  ///Sets the width of the arrowheads

  ///Sets the width of the arrowheads

  ///

  ///Scales the drawing to fit to A4 page

  ///Scales the drawing to fit to A4 page

  ///

  GraphToEps<T> &scaleToA4() {_scaleToA4=true;return *this;}

  ///Enables parallel arcs

  ///Enables parallel arcs

  GraphToEps<T> &enableParallel(bool b=true) {_enableParallel=b;return *this;}

  ///Sets the distance 

  ///Sets the distance 

  ///

  GraphToEps<T> &parArcDist(double d) {_parArcDist*=d;return *this;}

  ///Hides the arcs

  ///Hides the arcs

  ///

  GraphToEps<T> &hideArcs(bool b=true) {_showArcs=!b;return *this;}

  ///Hides the nodes

  ///Sets the color of the node texts to be as different from the node color

  ///as it is possible

  ///

  GraphToEps<T> &distantColorNodeTexts()

  {_nodeTextColorType=DIST_COL;return *this;}

  ///Sets the color of the node texts to be black or white and always visible.

  ///Sets the color of the node texts to be black or white according to

  ///which is more 

  ///different from the node color

  ///

  GraphToEps<T> &distantBWNodeTexts()

  {_nodeTextColorType=DIST_BW;return *this;}

  ///Gives a preamble block for node Postscript block.

  ///Gives a preamble block for node Postscript block.

  ///

  ///\sa nodePsTexts()

  GraphToEps<T> & nodePsTextsPreamble(const char *str) {

    _nodePsTextsPreamble=str ;return *this;

  }

  ///Sets whether the the graph is undirected

  ///

  ///Sets whether the the graph is directed

  ///Sets whether the the graph is directed.

  ///Use it to show the edges as a pair of directed ones.

  GraphToEps<T> &directed(bool b=true) {_undirected=!b;return *this;}

  ///Sets the title.

  ///Sets the title of the generated image,

  ///namely it inserts a <tt>%%Title:</tt> DSC field to the header of

  ///the EPS file.

  GraphToEps<T> &title(const std::string &t) {_title=t;return *this;}

  ///Sets the copyright statement.

  ///Sets the copyright statement of the generated image,

  ///namely it inserts a <tt>%%Copyright:</tt> DSC field to the header of

  ///the EPS file.

  GraphToEps<T> &copyright(const std::string &t) {_copyright=t;return *this;}

protected:

  bool isInsideNode(dim2::Point<double> p, double r,int t) 

  {

    switch(t) {

    case CIRCLE:

    case MALE:

    case FEMALE:

      return p.normSquare()<=r*r;

    case SQUARE:

      return p.x<=r&&p.x>=-r&&p.y<=r&&p.y>=-r;

    case DIAMOND:

      return p.x+p.y<=r && p.x-p.y<=r && -p.x+p.y<=r && -p.x-p.y<=r;

    }

    return false;

  }

public:

  ~GraphToEps() { }

  ///Draws the graph.

  ///Like other functions using

  ///\ref named-templ-func-param "named template parameters",

  ///it draws the graph.

  void run() {

    const double EPSILON=1e-9;

    if(dontPrint) return;

    _graph_to_eps_bits::_NegY<typename T::CoordsMapType>

      mycoords(_coords,_negY);

    os << "%!PS-Adobe-2.0 EPSF-2.0\n";

    if(_title.size()>0) os << "%%Title: " << _title << '\n';

     if(_copyright.size()>0) os << "%%Copyright: " << _copyright << '\n';

//        << "%%Copyright: XXXX\n"

    os << "%%Creator: LEMON, graphToEps()\n";

    {

      char cbuf[50];

      timeval tv;

      gettimeofday(&tv, 0);

      ctime_r(&tv.tv_sec,cbuf);

      os << "%%CreationDate: " << cbuf;

    }

    if (_autoArcWidthScale) {

      double max_w=0;

      for(ArcIt e(g);e!=INVALID;++e)

	max_w=std::max(double(_arcWidths[e]),max_w);

    else if(_scale!=1.0) os << _scale << " dup scale\n";

    if(_showArcs) {

      os << "%Arcs:\ngsave\n";      

      if(_enableParallel) {

	std::vector<Arc> el;

	for(ArcIt e(g);e!=INVALID;++e)

	  if((!_undirected||g.source(e)<g.target(e))&&_arcWidths[e]>0

	     &&g.source(e)!=g.target(e))

	    el.push_back(e);

	std::sort(el.begin(),el.end(),arcLess(g));

	typename std::vector<Arc>::iterator j;

	for(typename std::vector<Arc>::iterator i=el.begin();i!=el.end();i=j) {

	  for(j=i+1;j!=el.end()&&isParallel(*i,*j);++j) ;

	  double sw=0;

	  for(typename std::vector<Arc>::iterator e=i;e!=j;++e)

	    sw+=_arcWidths[*e]*_arcWidthScale+_parArcDist;

	  sw-=_parArcDist;

	  sw/=-2.0;

	  dim2::Point<double>

	    dvec(mycoords[g.target(*i)]-mycoords[g.source(*i)]);

	  double l=std::sqrt(dvec.normSquare()); 

	  dim2::Point<double> d(dvec/std::max(l,EPSILON));

 	  dim2::Point<double> m;

// 	  m=dim2::Point<double>(mycoords[g.target(*i)]+mycoords[g.source(*i)])/2.0;

//  	  m=dim2::Point<double>(mycoords[g.source(*i)])+

// 	    dvec*(double(_nodeSizes[g.source(*i)])/

// 	       (_nodeSizes[g.source(*i)]+_nodeSizes[g.target(*i)]));

 	  m=dim2::Point<double>(mycoords[g.source(*i)])+

	    d*(l+_nodeSizes[g.source(*i)]-_nodeSizes[g.target(*i)])/2.0;

	  for(typename std::vector<Arc>::iterator e=i;e!=j;++e) {

	    sw+=_arcWidths[*e]*_arcWidthScale/2.0;

	    dim2::Point<double> mm=m+rot90(d)*sw/.75;

	    if(_drawArrows) {

	      int node_shape;

	      dim2::Point<double> s=mycoords[g.source(*e)];

	      dim2::Point<double> t=mycoords[g.target(*e)];

	      double rn=_nodeSizes[g.target(*e)]*_nodeScale;

	      node_shape=_nodeShapes[g.target(*e)];

	      dim2::Bezier3 bez(s,mm,mm,t);

	      double t1=0,t2=1;

	      for(int ii=0;ii<INTERPOL_PREC;++ii)

		if(isInsideNode(bez((t1+t2)/2)-t,rn,node_shape)) t2=(t1+t2)/2;