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