CDCSVGPlotter.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: dschneider, Oliver Frost                                 *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/display/EventDataPlotter.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    template <class T>
    class Styling;
 
    class CDCSVGPlotter {
 
    public:
      CDCSVGPlotter(bool animate = false, bool forwardFade = false);
 
      CDCSVGPlotter* clone();
 
      void drawWires(const CDCWireTopology& cdcWireTopology);
 
      void drawInteractionPoint();
 
      void drawSuperLayerBoundaries(const std::string& stroke);
 
      void drawOuterCDCWall(const std::string& stroke);
 
      void drawInnerCDCWall(const std::string& stroke);
 
      void drawHits(const std::string& storeArrayName,
                    const std::string& stroke,
                    const std::string& strokeWidth);
 
      void drawSimHits(const std::string& storeArrayName,
                       const std::string& stroke,
                       const std::string& strokeWidth);
 
      void drawClusters(const std::string& storeObjName,
                        const std::string& stroke,
                        const std::string& strokeWidth);
 
      void drawSegments(const std::string& storeObjName,
                        const std::string& stroke,
                        const std::string& strokeWidth);
 
      void drawSegmentTrajectories(const std::string& storeObjName,
                                   const std::string& stroke,
                                   const std::string& strokeWidth);
 
      void drawSegmentPairs(const std::string& storeObjName,
                            const std::string& stroke,
                            const std::string& strokeWidth);
 
      void drawAxialSegmentPairs(const std::string& storeObjName,
                                 const std::string& stroke,
                                 const std::string& strokeWidth);
 
      void drawSegmentTriples(const std::string& storeObjName,
                              const std::string& stroke,
                              const std::string& strokeWidth);
 
      void drawSegmentTripleTrajectories(const std::string& storeObjName,
                                         const std::string& stroke,
                                         const std::string& strokeWidth);
 
      void drawTracks(const std::string& storeObjName,
                      const std::string& stroke,
                      const std::string& strokeWidth);
 
      void drawTrackTrajectories(const std::string& storeObjName,
                                 const std::string& stroke,
                                 const std::string& strokeWidth);
 
      void drawRecoTracks(const std::string& storeArrayName,
                          const std::string& stroke,
                          const std::string& strokeWidth);
 
      void drawRecoTrackTrajectories(const std::string& storeArrayName,
                                     const std::string& stroke,
                                     const std::string& strokeWidth);
 
      void drawMCParticleTrajectories(const std::string& storeArrayName,
                                      const std::string& stroke,
                                      const std::string& strokeWidth);
 
    public:
      /* *** Section with some more complex operations *** */
 
      void drawWrongRLHitsInSegments(const std::string& segmentsStoreObjName);
 
      void drawWrongRLHitsInTracks(const std::string& tracksStoreObjName);
 
    private:
      template<class ACDCHitCollection>
      void drawWrongRLHits(const std::string& storeObjName);
 
    public:
      void drawSimHitsConnectByToF(const std::string& simHitStoreArrayName,
                                   const std::string& stroke,
                                   const std::string& strokeWidth);
 
      void drawMCSegmentPairs(const std::string& segmentsStoreObjName,
                              const std::string& stroke,
                              const std::string& strokeWidth);
 
      void drawMCAxialSegmentPairs(const std::string& segmentsStoreObjName,
                                   const std::string& stroke,
                                   const std::string& strokeWidth);
 
      void drawMCSegmentTriples(const std::string& segmentsStoreObjName,
                                const std::string& stroke,
                                const std::string& strokeWidth);
 
    public:
      std::string saveFile(const std::string& fileName = "display.svg");
 
    private:
      template<class AItem, bool a_drawTrajectories = false>
      void drawStoreArray(const std::string& storeArrayName, Styling<AItem>& styling);
 
      template<class AItem, bool a_drawTrajectories = false>
      void drawStoreVector(const std::string& storeObjName, Styling<AItem>& styling);
 
      template <bool a_drawTrajectory = false, class AIterable, class AStyling>
      void drawIterable(const AIterable& items, AStyling& styling);
 
      template <bool a_drawTrajectory = false, class AObject>
      void draw(const AObject& object, const AttributeMap& attributeMap);
 
    private:
      EventDataPlotter m_eventdataPlotter;
    };
  }
}