packages/trackbase/TrkrHitTruthAssocv1.h

0001 #ifndef TRACKBASE_TRKRHITTRUTHASSOCV1_H
0002 #define TRACKBASE_TRKRHITTRUTHASSOCV1_H
0003
0004 /**
0005  * @file trackbase/TrkrHitCellAssoc
0006  * @author D. McGlinchey, H. PEREIRA DA COSTA
0007  * @date June 2018
0008  * @brief Association object for PHG4Hits contributiong to TrkrHits
0009  */
0010
0011 #include "TrkrDefs.h"
0012 #include "TrkrHitTruthAssoc.h"
0013
0014 #include <iostream>  // for cout, ostream
0015 #include <map>
0016 #include <utility>  // for pair
0017
0018 /**
0019  * @brief Association object for PHG4Cells contributiong to TrkrHits
0020  *
0021  * Association object holding a multimap of PHG4Cells associated with a given TrkrHit
0022  */
0023 class TrkrHitTruthAssocv1 : public TrkrHitTruthAssoc
0024 {
0025  public:
0026   TrkrHitTruthAssocv1() = default;
0027
0028   void Reset() override;
0029
0030   void identify(std::ostream &os = std::cout) const override;
0031
0032   void addAssoc(const TrkrDefs::hitsetkey, const TrkrDefs::hitkey, const PHG4HitDefs::keytype) override;
0033
0034   void findOrAddAssoc(const TrkrDefs::hitsetkey, const TrkrDefs::hitkey, const PHG4HitDefs::keytype) override;
0035
0036   void removeAssoc(const TrkrDefs::hitsetkey, const TrkrDefs::hitkey) override;
0037
0038   void getG4Hits(const TrkrDefs::hitsetkey hitsetkey, const unsigned int hidx, MMap &temp_map) const override;
0039
0040  private:
0041   MMap m_map;
0042
0043   ClassDefOverride(TrkrHitTruthAssocv1, 1);
0044 };
0045
0046 #endif  // TRACKBASE_TRKRHITTRUTHASSOC_H