File indexing completed on 2025-01-18 09:55:30
0001
0002
0003 #ifndef EDM4EIC_MCRecoVertexParticleAssociation_CollectionData_H
0004 #define EDM4EIC_MCRecoVertexParticleAssociation_CollectionData_H
0005
0006
0007 #include "edm4eic/MCRecoVertexParticleAssociationData.h"
0008 #include "edm4eic/MCRecoVertexParticleAssociationObj.h"
0009 #include "edm4eic/Vertex.h"
0010 #include "edm4hep/MCParticle.h"
0011
0012
0013
0014
0015 #include "podio/CollectionBuffers.h"
0016 #include "podio/ICollectionProvider.h"
0017
0018 #include <deque>
0019 #include <memory>
0020
0021 namespace edm4eic {
0022
0023
0024 using MCRecoVertexParticleAssociationObjPointerContainer = std::deque<MCRecoVertexParticleAssociationObj*>;
0025 using MCRecoVertexParticleAssociationDataContainer = std::vector<MCRecoVertexParticleAssociationData>;
0026
0027
0028
0029
0030
0031
0032 class MCRecoVertexParticleAssociationCollectionData {
0033 public:
0034
0035
0036
0037 MCRecoVertexParticleAssociationObjPointerContainer entries{};
0038
0039
0040
0041
0042 MCRecoVertexParticleAssociationCollectionData();
0043
0044
0045
0046
0047 MCRecoVertexParticleAssociationCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
0048
0049
0050
0051
0052 MCRecoVertexParticleAssociationCollectionData(const MCRecoVertexParticleAssociationCollectionData&) = delete;
0053 MCRecoVertexParticleAssociationCollectionData& operator=(const MCRecoVertexParticleAssociationCollectionData&) = delete;
0054 MCRecoVertexParticleAssociationCollectionData(MCRecoVertexParticleAssociationCollectionData&& other) = default;
0055 MCRecoVertexParticleAssociationCollectionData& operator=(MCRecoVertexParticleAssociationCollectionData&& other) = default;
0056
0057
0058
0059
0060 ~MCRecoVertexParticleAssociationCollectionData() = default;
0061
0062 void clear(bool isSubsetColl);
0063
0064 podio::CollectionWriteBuffers getCollectionBuffers(bool isSubsetColl);
0065
0066 void prepareForWrite(bool isSubsetColl);
0067
0068 void prepareAfterRead(uint32_t collectionID);
0069
0070 void makeSubsetCollection();
0071
0072
0073 bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
0074
0075 private:
0076
0077 podio::UVecPtr<edm4eic::Vertex> m_rel_rec{nullptr};
0078 podio::UVecPtr<edm4hep::MCParticle> m_rel_sim{nullptr};
0079
0080
0081
0082
0083 podio::CollRefCollection m_refCollections{};
0084 podio::VectorMembersInfo m_vecmem_info{};
0085 std::unique_ptr<MCRecoVertexParticleAssociationDataContainer> m_data{nullptr};
0086 };
0087
0088
0089 }
0090
0091
0092 #endif