RhodeCode

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

    _autoNodeScale=b;return *this;

  }

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

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

  ///

  ///\sa nodeScale()

  ///

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

    _absoluteNodeSizes=b;return *this;

  }

  ///Negates the Y coordinates.

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

    _negY=b;return *this;

  }

  ///Turn on/off pre-scaling

  ///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.

  ///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

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

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

  GraphToEps<T> &border(double b=10) {_xBorder=_yBorder=b;return *this;}

  ///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

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

  ///Sets the length of the arrowheads

  GraphToEps<T> &arrowLength(double d=1.0) {_arrowLength*=d;return *this;}

  ///Sets the width of the arrowheads

  GraphToEps<T> &arrowWidth(double d=.3) {_arrowWidth*=d;return *this;}

  ///Scales the drawing to fit to A4 page

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

  ///Enables parallel arcs

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

  ///Sets the distance between parallel arcs

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

  ///Hides the arcs

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

  ///Hides the nodes

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

  ///Sets the size of the node texts

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

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

  ///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 graph is undirected

  ///Sets whether the graph is undirected.

  ///

  ///This setting is the default for undirected graphs.

  ///

  ///\sa directed()

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

  ///Sets whether the graph is directed

  ///Sets whether the graph is directed.

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

  ///

  ///This setting is the default for digraphs.

  ///

  ///\sa undirected()

  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",

  ///this function calls the algorithm itself, i.e. in this case

  ///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';

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

    {

#ifndef WIN32

      timeval tv;

      gettimeofday(&tv, 0);

      char cbuf[26];

      ctime_r(&tv.tv_sec,cbuf);

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

#else

      SYSTEMTIME time;

      char buf1[11], buf2[9], buf3[5];

      if (GetDateFormat(LOCALE_USER_DEFAULT, 0, &time,

                        "ddd MMM dd", buf1, 11) &&

          GetTimeFormat(LOCALE_USER_DEFAULT, 0, &time,

                        "HH':'mm':'ss", buf2, 9) &&

          GetDateFormat(LOCALE_USER_DEFAULT, 0, &time,

                                "yyyy", buf3, 5)) {

        os << "%%CreationDate: " << buf1 << ' '

           << buf2 << ' ' << buf3 << std::endl;

      }

#endif

    }

    if (_autoArcWidthScale) {

      double max_w=0;

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

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

      if(max_w>EPSILON) {

        _arcWidthScale/=max_w;

      }

    }

    if (_autoNodeScale) {

      double max_s=0;

      for(NodeIt n(g);n!=INVALID;++n)

        max_s=std::max(double(_nodeSizes[n]),max_s);

      if(max_s>EPSILON) {

        _nodeScale/=max_s;

      }

    }

    double diag_len = 1;

    if(!(_absoluteNodeSizes&&_absoluteArcWidths)) {

      dim2::Box<double> bb;

      for(NodeIt n(g);n!=INVALID;++n) bb.add(mycoords[n]);

      if (bb.empty()) {

        bb = dim2::Box<double>(dim2::Point<double>(0,0));

      }

      diag_len = std::sqrt((bb.bottomLeft()-bb.topRight()).normSquare());

      if(diag_len<EPSILON) diag_len = 1;

      if(!_absoluteNodeSizes) _nodeScale*=diag_len;

      if(!_absoluteArcWidths) _arcWidthScale*=diag_len;

    }

    dim2::Box<double> bb;

    for(NodeIt n(g);n!=INVALID;++n) {

      double ns=_nodeSizes[n]*_nodeScale;

      dim2::Point<double> p(ns,ns);

      switch(_nodeShapes[n]) {

      case CIRCLE:

      case SQUARE:

      case DIAMOND:

        bb.add(p+mycoords[n]);

        bb.add(-p+mycoords[n]);

        break;

      case MALE:

        bb.add(-p+mycoords[n]);

        bb.add(dim2::Point<double>(1.5*ns,1.5*std::sqrt(3.0)*ns)+mycoords[n]);

        break;

      case FEMALE:

        bb.add(p+mycoords[n]);

        bb.add(dim2::Point<double>(-ns,-3.01*ns)+mycoords[n]);

        break;

      }

    }

    if (bb.empty()) {

      bb = dim2::Box<double>(dim2::Point<double>(0,0));

    }

    if(_scaleToA4)

      os <<"%%BoundingBox: 0 0 596 842\n%%DocumentPaperSizes: a4\n";

    else {

      if(_preScale) {

        //Rescale so that BoundingBox won't be neither to big nor too small.

        while(bb.height()*_scale>1000||bb.width()*_scale>1000) _scale/=10;

        while(bb.height()*_scale<100||bb.width()*_scale<100) _scale*=10;

      }

      os << "%%BoundingBox: "

         << int(floor(bb.left()   * _scale - _xBorder)) << ' '

         << int(floor(bb.bottom() * _scale - _yBorder)) << ' '

         << int(ceil(bb.right()  * _scale + _xBorder)) << ' '

         << int(ceil(bb.top()    * _scale + _yBorder)) << '\n';

    }

    os << "%%EndComments\n";

    //x1 y1 x2 y2 x3 y3 cr cg cb w

    os << "/lb { setlinewidth setrgbcolor newpath moveto\n"

       << "      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

    os << "/di { newpath 2 index 1 index add 2 index moveto\n"

       << "      2 index             2 index 2 index add lineto\n"

       << "      2 index 1 index sub 2 index             lineto\n"

       << "      2 index             2 index 2 index sub lineto\n"

       << "      closepath pop pop pop} bind def\n";

    // x y r cr cg cb

    os << "/nc { 0 0 0 setrgbcolor 5 index 5 index 5 index c fill\n"

       << "     setrgbcolor " << 1+_nodeBorderQuotient << " div c fill\n"

       << "   } bind def\n";

    os << "/nsq { 0 0 0 setrgbcolor 5 index 5 index 5 index sq fill\n"

       << "     setrgbcolor " << 1+_nodeBorderQuotient << " div sq fill\n"

       << "   } bind def\n";

    os << "/ndi { 0 0 0 setrgbcolor 5 index 5 index 5 index di fill\n"

       << "     setrgbcolor " << 1+_nodeBorderQuotient << " div di fill\n"

       << "   } bind def\n";

    os << "/nfemale { 0 0 0 setrgbcolor 3 index "

       << _nodeBorderQuotient/(1+_nodeBorderQuotient)

       << " 1.5 mul mul setlinewidth\n"

       << "  newpath 5 index 5 index moveto "

       << "5 index 5 index 5 index 3.01 mul sub\n"

       << "  lineto 5 index 4 index .7 mul sub 5 index 5 index 2.2 mul sub"

       << " moveto\n"

       << "  5 index 4 index .7 mul add 5 index 5 index 2.2 mul sub lineto "

       << "stroke\n"

       << "  5 index 5 index 5 index c fill\n"

       << "  setrgbcolor " << 1+_nodeBorderQuotient << " div c fill\n"

       << "  } bind def\n";

    os << "/nmale {\n"

       << "  0 0 0 setrgbcolor 3 index "

       << _nodeBorderQuotient/(1+_nodeBorderQuotient)

       <<" 1.5 mul mul setlinewidth\n"

       << "  newpath 5 index 5 index moveto\n"

       << "  5 index 4 index 1 mul 1.5 mul add\n"

       << "  5 index 5 index 3 sqrt 1.5 mul mul add\n"

       << "  1 index 1 index lineto\n"

       << "  1 index 1 index 7 index sub moveto\n"

       << "  1 index 1 index lineto\n"

       << "  exch 5 index 3 sqrt .5 mul mul sub exch 5 index .5 mul sub"

       << " lineto\n"

       << "  stroke\n"

       << "  5 index 5 index 5 index c fill\n"

       << "  setrgbcolor " << 1+_nodeBorderQuotient << " div c fill\n"

       << "  } bind def\n";

    os << "/arrl " << _arrowLength << " def\n";

    os << "/arrw " << _arrowWidth << " def\n";

    // l dx_norm dy_norm

    os << "/lrl { 2 index mul exch 2 index mul exch rlineto pop} bind def\n";

    //len w dx_norm dy_norm x1 y1 cr cg cb

    os << "/arr { setrgbcolor /y1 exch def /x1 exch def /dy exch def /dx "

       << "exch def\n"

       << "       /w exch def /len exch def\n"

      //<< "0.1 setlinewidth x1 y1 moveto dx len mul dy len mul rlineto stroke"

       << "       newpath x1 dy w 2 div mul add y1 dx w 2 div mul sub moveto\n"

       << "       len w sub arrl sub dx dy lrl\n"

       << "       arrw dy dx neg lrl\n"

       << "       dx arrl w add mul dy w 2 div arrw add mul sub\n"

       << "       dy arrl w add mul dx w 2 div arrw add mul add rlineto\n"

       << "       dx arrl w add mul neg dy w 2 div arrw add mul sub\n"

       << "       dy arrl w add mul neg dx w 2 div arrw add mul add rlineto\n"

       << "       arrw dy dx neg lrl\n"

       << "       len w sub arrl sub neg dx dy lrl\n"

       << "       closepath fill } bind def\n";

    os << "/cshow { 2 index 2 index moveto dup stringwidth pop\n"

       << "         neg 2 div fosi .35 mul neg rmoveto show pop pop} def\n";

    os << "\ngsave\n";

    if(_scaleToA4)

      if(bb.height()>bb.width()) {

        double sc= std::min((A4HEIGHT-2*A4BORDER)/bb.height(),

                  (A4WIDTH-2*A4BORDER)/bb.width());

        os << ((A4WIDTH -2*A4BORDER)-sc*bb.width())/2 + A4BORDER << ' '

           << ((A4HEIGHT-2*A4BORDER)-sc*bb.height())/2 + A4BORDER

           << " translate\n"

           << sc << " dup scale\n"

           << -bb.left() << ' ' << -bb.bottom() << " translate\n";

      }

      else {

        double sc= std::min((A4HEIGHT-2*A4BORDER)/bb.width(),

                  (A4WIDTH-2*A4BORDER)/bb.height());

        os << ((A4WIDTH -2*A4BORDER)-sc*bb.height())/2 + A4BORDER << ' '

           << ((A4HEIGHT-2*A4BORDER)-sc*bb.width())/2 + A4BORDER

           << " translate\n"

           << sc << " dup scale\n90 rotate\n"

           << -bb.left() << ' ' << -bb.top() << " translate\n";

        }

    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) {