File indexing completed on 2025-12-25 09:45:59
0001
0002
0003 #ifndef EDM4EIC_CherenkovParticleID_CollectionData_H
0004 #define EDM4EIC_CherenkovParticleID_CollectionData_H
0005
0006
0007 #include "edm4eic/CherenkovParticleIDData.h"
0008 #include "edm4eic/CherenkovParticleIDObj.h"
0009 #include "edm4eic/CherenkovParticleID.h"
0010 #include "edm4eic/MCRecoTrackerHitAssociation.h"
0011 #include "edm4eic/TrackSegment.h"
0012
0013
0014
0015
0016 #include "podio/CollectionBuffers.h"
0017 #include "podio/ICollectionProvider.h"
0018
0019 #include <deque>
0020 #include <memory>
0021
0022 namespace edm4eic {
0023
0024
0025 using CherenkovParticleIDObjPointerContainer = std::deque<CherenkovParticleIDObj*>;
0026 using CherenkovParticleIDDataContainer = std::vector<CherenkovParticleIDData>;
0027
0028
0029
0030
0031
0032
0033 class CherenkovParticleIDCollectionData {
0034 public:
0035
0036
0037
0038 CherenkovParticleIDObjPointerContainer entries{};
0039
0040
0041
0042
0043 CherenkovParticleIDCollectionData();
0044
0045
0046
0047
0048 CherenkovParticleIDCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
0049
0050
0051
0052
0053 CherenkovParticleIDCollectionData(const CherenkovParticleIDCollectionData&) = delete;
0054 CherenkovParticleIDCollectionData& operator=(const CherenkovParticleIDCollectionData&) = delete;
0055 CherenkovParticleIDCollectionData(CherenkovParticleIDCollectionData&& other) = default;
0056 CherenkovParticleIDCollectionData& operator=(CherenkovParticleIDCollectionData&& other) = default;
0057
0058
0059
0060
0061 ~CherenkovParticleIDCollectionData() = default;
0062
0063 void clear(bool isSubsetColl);
0064
0065 podio::CollectionWriteBuffers getCollectionBuffers(bool isSubsetColl);
0066
0067 void prepareForWrite(bool isSubsetColl);
0068
0069 void prepareAfterRead(uint32_t collectionID);
0070
0071 void makeSubsetCollection();
0072
0073 void createRelations(CherenkovParticleIDObj* obj);
0074
0075 bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
0076
0077 private:
0078
0079 podio::UVecPtr<edm4eic::MCRecoTrackerHitAssociation> m_rel_rawHitAssociations{std::make_unique<std::vector<edm4eic::MCRecoTrackerHitAssociation>>()};
0080 std::vector<podio::UVecPtr<edm4eic::MCRecoTrackerHitAssociation>> m_rel_rawHitAssociations_tmp{};
0081 podio::UVecPtr<edm4eic::TrackSegment> m_rel_chargedParticle{std::make_unique<std::vector<edm4eic::TrackSegment>>()};
0082
0083
0084 podio::UVecPtr<edm4eic::CherenkovParticleIDHypothesis> m_vec_hypotheses{nullptr};
0085 std::vector<podio::UVecPtr<edm4eic::CherenkovParticleIDHypothesis>> m_vecs_hypotheses{};
0086 podio::UVecPtr<edm4hep::Vector2f> m_vec_thetaPhiPhotons{nullptr};
0087 std::vector<podio::UVecPtr<edm4hep::Vector2f>> m_vecs_thetaPhiPhotons{};
0088
0089
0090 podio::CollRefCollection m_refCollections{};
0091 podio::VectorMembersInfo m_vecmem_info{};
0092 std::unique_ptr<CherenkovParticleIDDataContainer> m_data{nullptr};
0093 };
0094
0095
0096 }
0097
0098
0099 #endif