include/edm4hep/RecDqdxCollectionData.h

0001 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
0002
0003 #ifndef EDM4HEP_RecDqdx_CollectionData_H
0004 #define EDM4HEP_RecDqdx_CollectionData_H
0005
0006 // datamodel specific includes
0007 #include "edm4hep/RecDqdxData.h"
0008 #include "edm4hep/RecDqdxObj.h"
0009 #include "edm4hep/Track.h"
0010
0011 // schema evolution specific includes
0012
0013 // podio specific includes
0014 #include "podio/CollectionBuffers.h"
0015 #include "podio/ICollectionProvider.h"
0016
0017 #include <deque>
0018 #include <memory>
0019
0020 namespace edm4hep {
0021
0022 using RecDqdxObjPointerContainer = std::deque<RecDqdxObj*>;
0023 using RecDqdxDataContainer = std::vector<RecDqdxData>;
0024
0025 /**
0026  * Class encapsulating everything related to storage of data that is needed by a
0027  * collection.
0028  */
0029 class RecDqdxCollectionData {
0030 public:
0031   /**
0032    * The Objs of this collection
0033    */
0034   RecDqdxObjPointerContainer entries{};
0035
0036   /**
0037    * Default constructor setting up the necessary buffers
0038    */
0039   RecDqdxCollectionData();
0040
0041   /**
0042    * Constructor from existing I/O buffers
0043    */
0044   RecDqdxCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
0045
0046   /**
0047    * Non copy-able, move-only class
0048    */
0049   RecDqdxCollectionData(const RecDqdxCollectionData&) = delete;
0050   RecDqdxCollectionData& operator=(const RecDqdxCollectionData&) = delete;
0051   RecDqdxCollectionData(RecDqdxCollectionData&& other) = default;
0052   RecDqdxCollectionData& operator=(RecDqdxCollectionData&& other) = default;
0053
0054   /**
0055    * Destructor
0056    */
0057   ~RecDqdxCollectionData() = 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   void createRelations(RecDqdxObj* obj);
0070
0071   bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
0072
0073 private:
0074   // members to handle 1-to-N-relations
0075   podio::UVecPtr<edm4hep::Track> m_rel_track{nullptr}; ///< Relation buffer for read / write
0076
0077   // members to handle vector members
0078   podio::UVecPtr<edm4hep::HitLevelData> m_vec_hitData{nullptr};        /// combined vector of all objects in collection
0079   std::vector<podio::UVecPtr<edm4hep::HitLevelData>> m_vecs_hitData{}; /// pointers to individual member vectors
0080
0081   // I/O related buffers
0082   podio::CollRefCollection m_refCollections{};
0083   podio::VectorMembersInfo m_vecmem_info{};
0084   std::unique_ptr<RecDqdxDataContainer> m_data{nullptr};
0085 };
0086
0087 } // namespace edm4hep
0088
0089 #endif