VisualizationHelpers.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONR4_MUONVISUALIZATIONHELPERS_H
#define MUONR4_MUONVISUALIZATIONHELPERS_H
 
 
#include <GeoPrimitives/GeoPrimitives.h>
#include <memory>
#include <vector>
#include <MuonPatternEvent/SegmentFitterEventData.h>
 
 
#include "TEllipse.h"
#include "TBox.h"
#include "TLatex.h"
#include "TLine.h"
#include "TArrow.h"
 
namespace MuonValR4 {
    constexpr int hollowFilling = 0;
    constexpr int fullFilling = 1001;
    constexpr int hatchedFilling = 3344;
    
    constexpr int objViewEta = Acts::toUnderlying(MuonR4::SegmentFit::AxisDefs::etaCov);
    constexpr int objViewPhi = Acts::toUnderlying(MuonR4::SegmentFit::AxisDefs::phiCov);
    
    
    std::vector<std::unique_ptr<TObject>> clone(const std::vector<std::unique_ptr<TObject>>& cloneMe);
    std::unique_ptr<TEllipse> drawDriftCircle(const Amg::Vector3D& center,
                                              const double radius,
                                              const int color = kViolet,
                                              const int fillStyle = hollowFilling);
    std::unique_ptr<TBox> drawBox(const Amg::Vector3D& boxCenter,
                                  const double boxWidth,
                                  const double boxHeight,
                                  const int color = kGreen +2,
                                  const int fillStyle = hollowFilling,
                                  const int view = objViewEta);
    std::unique_ptr<TBox> drawBox(const double x1,
                                  const double y1,
                                  const double x2,
                                  const double y2,
                                  const int color = kGreen +2,
                                  const int fillStyle = hollowFilling);
    std::unique_ptr<TLine> drawLine(const MuonR4::SegmentFit::Parameters& pars,
                                    const double lowEnd, const double highEnd,
                                    const int color = kRed +1,
                                    const int lineStyle = kDashed,
                                    const int view = objViewEta);
    std::unique_ptr<TLine> drawLine(const Amg::Vector3D& lowEnd,
                                    const Amg::Vector3D& highEnd,
                                    const int color = kRed + 1,
                                    const int lineStyle = kSolid,
                                    const int view = objViewEta);
    std::unique_ptr<TArrow> drawArrow(const Amg::Vector3D& start,
                                      const Amg::Vector3D& dir,
                                      const int color = kRed +1,
                                      const int lineStyle = kDashed,
                                      const int view = objViewEta);
    std::unique_ptr<TLatex> drawLabel(const std::string& text, 
                                      const double xPos, 
                                      const double yPos,
                                      const unsigned int fontSize = 18);
    std::unique_ptr<TLatex> drawLabelNDC(const std::string& text, 
                                      const double xPos, 
                                      const double yPos,
                                      const unsigned int fontSize = 18);
    std::unique_ptr<TLatex> drawAtlasLabel(const double xPos,
                                           const double yPos,
                                           const std::string& status = "Internal");
    std::unique_ptr<TLatex> drawLumiSqrtS(const double xPos,
                                          const double yPos,
                                          const std::string_view sqrtS="14",
                                          const std::string_view lumi = "");
}
#endif