SVGPrimitivePlotter.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Oliver Frost                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/display/PrimitivePlotter.h>
#include <sstream>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class SVGPrimitivePlotter : public PrimitivePlotter {
 
    private:
      static const int s_defaultNIndentationSpaces;
 
      static const int s_addtionalNIndentationSpaces;
 
    public:
      SVGPrimitivePlotter();
 
      explicit SVGPrimitivePlotter(const AttributeMap& svgAttributes);
 
      SVGPrimitivePlotter(const SVGPrimitivePlotter& plotter);
 
    public:
      std::unique_ptr<PrimitivePlotter> clone() const override;
 
    public:
      void drawLine(float startX,
                    float startY,
                    float endX,
                    float endY,
                    const AttributeMap& attributeMap = AttributeMap()) override;
 
      void drawArrow(float startX,
                     float startY,
                     float endX,
                     float endY,
                     const AttributeMap& attributeMap = AttributeMap()) override;
 
      void drawCircle(float centerX,
                      float centerY,
                      float radius,
                      const AttributeMap& attributeMap = AttributeMap()) override;
 
      void drawCircleArc(float startX,
                         float startY,
                         float endX,
                         float endY,
                         float radius,
                         bool longArc,
                         bool sweepFlag,
                         const AttributeMap& attributeMap = AttributeMap()) override;
 
      void drawCurve(const std::vector<std::array<float, 2>>& points,
                     const std::vector<std::array<float, 2>>& tangents,
                     const AttributeMap& attributeMap = AttributeMap()) override;
 
      void startGroup(const AttributeMap& attributeMap = AttributeMap()) override;
 
      void endGroup() override;
 
      const std::string save(const std::string& fileName) override;
 
      void clear() override;
 
    private:
      void indent();
 
      void dedent();
 
      void writeSVGHeader(std::ostream& outputStream);
 
      void writeSVGDefs(std::ostream& outputStream);
 
      void writeOpeningTag(std::ostream& outputStream,
                           const std::string& tagName,
                           const AttributeMap& geometryAttributeMap = AttributeMap(),
                           const AttributeMap& styleAttributeMap = AttributeMap());
 
      void writeStandAloneTag(std::ostream& outputStream,
                              const std::string& tagName,
                              const AttributeMap& geometryAttributeMap = AttributeMap(),
                              const AttributeMap& styleAttributeMap = AttributeMap());
 
      void writeTagIntern(std::ostream& outputStream,
                          const std::string& tagName,
                          const AttributeMap& geometryAttributeMap = AttributeMap(),
                          const AttributeMap& styleAttributeMap = AttributeMap());
 
      void writeAttributes(std::ostream& outputStream,
                           const AttributeMap& attributeMap);
 
      void writeClosingTag(std::ostream& outputStream,
                           const std::string& tagName);
 
    private:
      std::ostringstream m_svgContentStream;
 
      int m_nIndentationSpaces;
 
      AttributeMap m_svgAttributes;
 
    };
 
  }
}