PrimitivePlotter.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/BoundingBox.h>
#include <tracking/trackFindingCDC/display/AttributeMap.h>
 
#include <vector>
#include <array>
#include <memory>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class PrimitivePlotter {
 
    public:
      using AttributeMap = Belle2::TrackFindingCDC::AttributeMap;
 
    public:
      static const float s_defaultCanvasWidth;
 
      static const float s_defaultCanvasHeight;
 
    public:
      PrimitivePlotter();
 
      virtual ~PrimitivePlotter();
 
    public:
      virtual std::unique_ptr<PrimitivePlotter> clone() const;
 
    public:
      virtual void drawLine(float startX,
                            float startY,
                            float endX,
                            float endY,
                            const AttributeMap& attributeMap = AttributeMap());
 
      virtual void drawArrow(float startX,
                             float startY,
                             float endX,
                             float endY,
                             const AttributeMap& attributeMap = AttributeMap());
 
      virtual void drawCircle(float centerX,
                              float centerY,
                              float radius,
                              const AttributeMap& attributeMap = AttributeMap());
 
      virtual void drawCircleArc(float startX,
                                 float startY,
                                 float endX,
                                 float endY,
                                 float radius,
                                 bool longArc,
                                 bool sweepFlag,
                                 const AttributeMap& attributeMap = AttributeMap());
 
      virtual void drawCurve(const std::vector<std::array<float, 2>>& points,
                             const std::vector<std::array<float, 2>>& tangents,
                             const AttributeMap& attributeMap = AttributeMap());
 
      virtual void startGroup(const AttributeMap& attributeMap = AttributeMap());
 
      virtual void endGroup();
 
      virtual const std::string save(const std::string& fileName);
 
      virtual void clear();
 
      void clearBoundingBox();
 
      const BoundingBox& getBoundingBox() const
      {
        return m_boundingBox;
      }
 
      void setBoundingBox(const BoundingBox& boundingBox)
      {
        m_boundingBox = boundingBox;
      }
 
    public:
      float getCanvasWidth()
      {
        return m_canvasWidth;
      }
 
      float getCanvasHeight()
      {
        return m_canvasHeight;
      }
 
      void setCanvasWidth(float width)
      {
        m_canvasWidth = width;
      }
 
      void setCanvasHeight(float height)
      {
        m_canvasHeight = height;
      }
 
    private:
      BoundingBox m_boundingBox;
 
      float m_canvasWidth;
 
      float m_canvasHeight;
    };
  }
}