packages/trackbase/ClusHitsVerbosev1.h

0001 #ifndef G4TRACKING_CLUSHITSVERBOSEV1_H
0002 #define G4TRACKING_CLUSHITSVERBOSEV1_H
0003
0004 /**
0005  * @file trackbase/ClusHitsVerbosev1.h
0006  * @author D. Stewart
0007  * @date May 2023
0008  * @brief Ojbect to show hit locations in given clusters
0009  */
0010
0011 #include "ClusHitsVerbose.h"
0012
0013 /**
0014  * @brief Cluster container object
0015  */
0016 class ClusHitsVerbosev1 : public ClusHitsVerbose
0017 {
0018  public:
0019   ClusHitsVerbosev1() = default;
0020
0021   void Reset() override;
0022
0023   bool hasClusKey(TrkrDefs::cluskey) const override;
0024   Vector& phiBins(TrkrDefs::cluskey) override;
0025   Vector& zBins(TrkrDefs::cluskey) override;
0026   Vector& phiCutBins(TrkrDefs::cluskey) override;
0027   Vector& zCutBins(TrkrDefs::cluskey) override;
0028   Map& getMap() override { return m_data; };
0029
0030   PairVector phiBins_pvecIE(TrkrDefs::cluskey) override;
0031   PairVector phiCutBins_pvecIE(TrkrDefs::cluskey) override;
0032   PairVector zBins_pvecIE(TrkrDefs::cluskey) override;
0033   PairVector zCutBins_pvecIE(TrkrDefs::cluskey) override;
0034
0035   void addPhiHit(int _i, int _v) override { m_stage_phi.push_back({_i, _v}); };
0036   void addPhiCutHit(int _i, int _v) override { m_stage_phiCut.push_back({_i, _v}); };
0037   void addZHit(int _i, int _v) override { m_stage_z.push_back({_i, _v}); };
0038   void addZCutHit(int _i, int _v) override { m_stage_zCut.push_back({_i, _v}); };
0039   void push_hits(TrkrDefs::cluskey) override;
0040
0041   void identify(std::ostream& os = std::cout) const override
0042   {
0043     os << "ClusHitsVerbosev1 class" << std::endl;
0044   };
0045
0046  private:
0047   // the data
0048   Map m_data{};
0049   Vector m_stage_phi{};
0050   Vector m_stage_z{};
0051   Vector m_stage_phiCut{};
0052   Vector m_stage_zCut{};
0053
0054   PairVector pvecIE(TrkrDefs::cluskey, int which);
0055   Vector& vecBins(TrkrDefs::cluskey, int which);
0056
0057   ClassDefOverride(ClusHitsVerbosev1, 1)
0058 };
0059
0060 #endif  // G4TRACKING_CLUSHITSVERBOSEV1_H