include/edm4hep/TrackCollectionData.h

0001 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
0002
0003 #ifndef EDM4HEP_Track_CollectionData_H
0004 #define EDM4HEP_Track_CollectionData_H
0005
0006 // datamodel specific includes
0007 #include "edm4hep/Track.h"
0008 #include "edm4hep/TrackData.h"
0009 #include "edm4hep/TrackObj.h"
0010 #include "edm4hep/TrackerHit.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 edm4hep {
0022
0023 using TrackObjPointerContainer = std::deque<TrackObj*>;
0024 using TrackDataContainer = std::vector<TrackData>;
0025
0026 /**
0027  * Class encapsulating everything related to storage of data that is needed by a
0028  * collection.
0029  */
0030 class TrackCollectionData {
0031 public:
0032   /**
0033    * The Objs of this collection
0034    */
0035   TrackObjPointerContainer entries{};
0036
0037   /**
0038    * Default constructor setting up the necessary buffers
0039    */
0040   TrackCollectionData();
0041
0042   /**
0043    * Constructor from existing I/O buffers
0044    */
0045   TrackCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
0046
0047   /**
0048    * Non copy-able, move-only class
0049    */
0050   TrackCollectionData(const TrackCollectionData&) = delete;
0051   TrackCollectionData& operator=(const TrackCollectionData&) = delete;
0052   TrackCollectionData(TrackCollectionData&& other) = default;
0053   TrackCollectionData& operator=(TrackCollectionData&& other) = default;
0054
0055   /**
0056    * Destructor
0057    */
0058   ~TrackCollectionData() = default;
0059
0060   void clear(bool isSubsetColl);
0061
0062   podio::CollectionWriteBuffers getCollectionBuffers(bool isSubsetColl);
0063
0064   void prepareForWrite(bool isSubsetColl);
0065
0066   void prepareAfterRead(uint32_t collectionID);
0067
0068   void makeSubsetCollection();
0069
0070   void createRelations(TrackObj* obj);
0071
0072   bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
0073
0074 private:
0075   // members to handle 1-to-N-relations
0076   podio::UVecPtr<edm4hep::TrackerHit> m_rel_trackerHits{
0077       std::make_unique<std::vector<edm4hep::TrackerHit>>()}; ///< Relation buffer for read / write
0078   std::vector<podio::UVecPtr<edm4hep::TrackerHit>>
0079       m_rel_trackerHits_tmp{}; ///< Relation buffer for internal book-keeping
0080   podio::UVecPtr<edm4hep::Track> m_rel_tracks{
0081       std::make_unique<std::vector<edm4hep::Track>>()};           ///< Relation buffer for read / write
0082   std::vector<podio::UVecPtr<edm4hep::Track>> m_rel_tracks_tmp{}; ///< Relation buffer for internal book-keeping
0083
0084   // members to handle vector members
0085   podio::UVecPtr<std::int32_t> m_vec_subdetectorHitNumbers{nullptr}; /// combined vector of all objects in collection
0086   std::vector<podio::UVecPtr<std::int32_t>> m_vecs_subdetectorHitNumbers{}; /// pointers to individual member vectors
0087   podio::UVecPtr<std::int32_t> m_vec_subdetectorHoleNumbers{nullptr}; /// combined vector of all objects in collection
0088   std::vector<podio::UVecPtr<std::int32_t>> m_vecs_subdetectorHoleNumbers{}; /// pointers to individual member vectors
0089   podio::UVecPtr<edm4hep::TrackState> m_vec_trackStates{nullptr}; /// combined vector of all objects in collection
0090   std::vector<podio::UVecPtr<edm4hep::TrackState>> m_vecs_trackStates{}; /// pointers to individual member vectors
0091
0092   // I/O related buffers
0093   podio::CollRefCollection m_refCollections{};
0094   podio::VectorMembersInfo m_vecmem_info{};
0095   std::unique_ptr<TrackDataContainer> m_data{nullptr};
0096 };
0097
0098 } // namespace edm4hep
0099
0100 #endif