include/edm4eic/CherenkovParticleIDCollectionData.h

0001 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
0002
0003 #ifndef EDM4EIC_CherenkovParticleID_CollectionData_H
0004 #define EDM4EIC_CherenkovParticleID_CollectionData_H
0005
0006 // datamodel specific includes
0007 #include "edm4eic/CherenkovParticleIDData.h"
0008 #include "edm4eic/CherenkovParticleIDObj.h"
0009 #include "edm4eic/MCRecoTrackerHitAssociation.h"
0010 #include "edm4eic/TrackSegment.h"
0011
0012 // schema evolution specific includes
0013
0014 // podio specific includes
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 CherenkovParticleIDObjPointerContainer = std::deque<CherenkovParticleIDObj*>;
0025 using CherenkovParticleIDDataContainer = std::vector<CherenkovParticleIDData>;
0026
0027
0028 /**
0029  * Class encapsulating everything related to storage of data that is needed by a
0030  * collection.
0031  */
0032 class CherenkovParticleIDCollectionData {
0033 public:
0034   /**
0035    * The Objs of this collection
0036    */
0037   CherenkovParticleIDObjPointerContainer entries{};
0038
0039   /**
0040    * Default constructor setting up the necessary buffers
0041    */
0042   CherenkovParticleIDCollectionData();
0043
0044   /**
0045    * Constructor from existing I/O buffers
0046    */
0047   CherenkovParticleIDCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
0048
0049   /**
0050    * Non copy-able, move-only class
0051    */
0052   CherenkovParticleIDCollectionData(const CherenkovParticleIDCollectionData&) = delete;
0053   CherenkovParticleIDCollectionData& operator=(const CherenkovParticleIDCollectionData&) = delete;
0054   CherenkovParticleIDCollectionData(CherenkovParticleIDCollectionData&& other) = default;
0055   CherenkovParticleIDCollectionData& operator=(CherenkovParticleIDCollectionData&& other) = default;
0056
0057   /**
0058    * Destructor
0059    */
0060   ~CherenkovParticleIDCollectionData() = 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   void createRelations(CherenkovParticleIDObj* obj);
0073
0074   bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
0075
0076 private:
0077   // members to handle 1-to-N-relations
0078   podio::UVecPtr<edm4eic::MCRecoTrackerHitAssociation> m_rel_rawHitAssociations;  ///< Relation buffer for read / write
0079   std::vector<podio::UVecPtr<edm4eic::MCRecoTrackerHitAssociation>> m_rel_rawHitAssociations_tmp{}; ///< Relation buffer for internal book-keeping
0080   podio::UVecPtr<edm4eic::TrackSegment> m_rel_chargedParticle{nullptr}; ///< Relation buffer for read / write
0081
0082   // members to handle vector members
0083   podio::UVecPtr<edm4eic::CherenkovParticleIDHypothesis> m_vec_hypotheses{nullptr}; /// combined vector of all objects in collection
0084   std::vector<podio::UVecPtr<edm4eic::CherenkovParticleIDHypothesis>> m_vecs_hypotheses{}; /// pointers to individual member vectors
0085   podio::UVecPtr<edm4hep::Vector2f> m_vec_thetaPhiPhotons{nullptr}; /// combined vector of all objects in collection
0086   std::vector<podio::UVecPtr<edm4hep::Vector2f>> m_vecs_thetaPhiPhotons{}; /// pointers to individual member vectors
0087
0088   // I/O related buffers
0089   podio::CollRefCollection m_refCollections{};
0090   podio::VectorMembersInfo m_vecmem_info{};
0091   std::unique_ptr<CherenkovParticleIDDataContainer> m_data{nullptr};
0092 };
0093
0094
0095 } // namespace edm4eic
0096
0097
0098 #endif