source/designer/svgexporter.cpp

   1 #include <cmath>
   2 #include <msp/strings/format.h>
   3 #include "libr2c2/catalogue.h"
   4 #include "libr2c2/track.h"
   5 #include "libr2c2/tracktype.h"
   6 #include "svgexporter.h"
   7
   8 using namespace std;
   9 using namespace Msp;
  10 using namespace R2C2;
  11
  12 SvgExporter::SvgExporter(const Layout &l):
  13         layout(l),
  14         gauge(0),
  15         rail_width(0)
  16 { }
  17
  18 void SvgExporter::save(const string &fn)
  19 {
  20         gauge = layout.get_catalogue().get_gauge()*1000;
  21
  22         const Profile &rail_profile = layout.get_catalogue().get_rail_profile();
  23         const Vector &rail_min = rail_profile.get_min_coords();
  24         const Vector &rail_max = rail_profile.get_max_coords();
  25         rail_width = (rail_max.x-rail_min.x)*1000;
  26
  27         xmlpp::Document *doc = new xmlpp::Document;
  28         xmlpp::Element *root = doc->create_root_node("svg", "http://www.w3.org/2000/svg");
  29
  30         xmlpp::Element *style = root->add_child("style");
  31         style->set_attribute("type", "text/css");
  32         style->set_child_text(format("\n.rail { fill: none; stroke: #000000; stroke-width: %.3f; }\n"
  33                 ".endpoint { fill: none; stroke: #808080; stroke-width: %.3f; }\n"
  34                 ".artnr { text-anchor: middle; font-size: %.3f; }\n",
  35                 rail_width, rail_width, gauge*0.75));
  36
  37         Vector minp;
  38         Vector maxp;
  39         const set<Track *> &tracks = layout.get_all<Track>();
  40         for(set<Track *>::const_iterator i=tracks.begin(); i!=tracks.end(); ++i)
  41         {
  42                 xmlpp::Element *elem = root->add_child("g");
  43                 save_track(**i, *elem);
  44
  45                 unsigned n_endpoints = (*i)->get_type().get_endpoints().size();
  46                 for(unsigned j=0; j<n_endpoints; ++j)
  47                 {
  48                         Vector pos = (*i)->get_snap_node(j).position;
  49                         if(i==tracks.begin() && j==0)
  50                                 minp = maxp = pos;
  51                         else
  52                         {
  53                                 minp.x = min(minp.x, pos.x);
  54                                 minp.y = min(minp.y, pos.y);
  55                                 maxp.x = max(maxp.x, pos.x);
  56                                 maxp.y = max(maxp.y, pos.y);
  57                         }
  58                 }
  59         }
  60
  61         root->set_attribute("viewBox", format("%.3f %.3f %.3f %.3f",
  62                 minp.x*1000-gauge*3, -maxp.y*1000-gauge*3, (maxp.x-minp.x)*1000+gauge*6, (maxp.y-minp.y)*1000+gauge*6));
  63
  64         doc->write_to_file_formatted(fn);
  65 }
  66
  67 void SvgExporter::save_track(const Track &track, xmlpp::Element &group)
  68 {
  69         const Vector &pos = track.get_position();
  70         const Angle &rot = track.get_rotation();
  71         string transform = format("translate(%.3f %.3f) rotate(%.3f)", pos.x*1000, -pos.y*1000, -rot.degrees());
  72         group.set_attribute("transform", transform);
  73
  74         const TrackType &type = track.get_type();
  75
  76         const vector<TrackType::Endpoint> &endpoints = type.get_endpoints();
  77         for(vector<TrackType::Endpoint>::const_iterator i=endpoints.begin(); i!=endpoints.end(); ++i)
  78         {
  79                 xmlpp::Element *elem = group.add_child("path");
  80                 elem->set_attribute("class", "endpoint");
  81
  82                 Vector delta = rotated_vector(Vector(0, gauge, 0), i->dir);
  83
  84                 string data = format("M %.3f %.3f L %.3f %.3f",
  85                         i->pos.x*1000+delta.x, -i->pos.y*1000-delta.y,
  86                         i->pos.x*1000-delta.x, -i->pos.y*1000+delta.y);
  87                 elem->set_attribute("d", data);
  88         }
  89
  90         const vector<TrackPart> &parts = type.get_parts();
  91         for(vector<TrackPart>::const_iterator i=parts.begin(); i!=parts.end(); ++i)
  92         {
  93                 OrientedPoint start = i->get_point(0);
  94                 OrientedPoint end = i->get_point(i->get_length());
  95                 if(i->is_curved())
  96                 {
  97                         xmlpp::Element *elem = group.add_child("path");
  98                         elem->set_attribute("class", "rail");
  99
 100                         Vector delta1 = rotated_vector(Vector(0, (gauge+rail_width)*0.5, 0), start.rotation);
 101                         Vector delta2 = rotated_vector(Vector(0, (gauge+rail_width)*0.5, 0), end.rotation);
 102                         // Largely an educated guess, but seems to be accurate enough
 103                         float clen = i->get_length()*1000/2.9;
 104                         Vector ctrl1 = rotated_vector(Vector(clen, 0, 0), start.rotation);
 105                         Vector ctrl2 = rotated_vector(Vector(clen, 0, 0), end.rotation);
 106
 107                         string data = format("M %.3f %.3f C %.3f %.3f %.3f %.3f %.3f %.3f")
 108                                 (start.position.x*1000+delta1.x)(-start.position.y*1000-delta1.y)
 109                                 (start.position.x*1000+delta1.x+ctrl1.x)(-start.position.y*1000-delta1.y-ctrl1.y)
 110                                 (end.position.x*1000+delta2.x-ctrl2.x)(-end.position.y*1000-delta2.y+ctrl2.y)
 111                                 (end.position.x*1000+delta2.x)(-end.position.y*1000-delta2.y).str();
 112                         data += format(" M %.3f %.3f C %.3f %.3f %.3f %.3f %.3f %.3f")
 113                                 (start.position.x*1000-delta1.x)(-start.position.y*1000+delta1.y)
 114                                 (start.position.x*1000-delta1.x+ctrl1.x)(-start.position.y*1000+delta1.y-ctrl1.y)
 115                                 (end.position.x*1000-delta2.x-ctrl2.x)(-end.position.y*1000+delta2.y+ctrl2.y)
 116                                 (end.position.x*1000-delta2.x)(-end.position.y*1000+delta2.y).str();
 117                         elem->set_attribute("d", data);
 118                 }
 119                 else
 120                 {
 121                         xmlpp::Element *elem = group.add_child("path");
 122                         elem->set_attribute("class", "rail");
 123
 124                         Vector delta = rotated_vector(Vector(0, (gauge+rail_width)*0.5, 0), start.rotation);
 125
 126                         string data = format("M %.3f %.3f L %.3f %.3f",
 127                                 start.position.x*1000+delta.x, -start.position.y*1000-delta.y,
 128                                 end.position.x*1000+delta.x, -end.position.y*1000-delta.y);
 129                         data += format(" M %.3f %.3f L %.3f %.3f",
 130                                 start.position.x*1000-delta.x, -start.position.y*1000+delta.y,
 131                                 end.position.x*1000-delta.x, -end.position.y*1000+delta.y);
 132                         elem->set_attribute("d", data);
 133                 }
 134         }
 135
 136         OrientedPoint label_pt = parts.front().get_point(parts.front().get_length()/2);
 137
 138         label_pt.rotation.wrap_with_base(-Angle::quarter_turn());
 139         if(label_pt.rotation>Angle::quarter_turn())
 140                 label_pt.rotation -= Angle::half_turn();
 141
 142         label_pt.position *= 1000;
 143         label_pt.position += rotated_vector(Vector(0, -gauge*0.25, 0), label_pt.rotation);
 144
 145         xmlpp::Element *elem = group.add_child("text");
 146         elem->set_attribute("class", "artnr");
 147         elem->set_attribute("transform", format("translate(%.3f %.3f) rotate(%.3f)",
 148                 label_pt.position.x, -label_pt.position.y, -label_pt.rotation.degrees()));
 149         elem->set_child_text(track.get_type().get_article_number().str());
 150 }