include/edm4hep/ReconstructedParticleCollectionData.h

0001 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
0002
0003 #ifndef EDM4HEP_ReconstructedParticle_CollectionData_H
0004 #define EDM4HEP_ReconstructedParticle_CollectionData_H
0005
0006 // datamodel specific includes
0007 #include "edm4hep/Cluster.h"
0008 #include "edm4hep/ReconstructedParticleData.h"
0009 #include "edm4hep/ReconstructedParticleObj.h"
0010 #include "edm4hep/Track.h"
0011 #include "edm4hep/Vertex.h"
0012
0013 // schema evolution specific includes
0014
0015 // podio specific includes
0016 #include "podio/CollectionBuffers.h"
0017 #include "podio/ICollectionProvider.h"
0018
0019 #include <deque>
0020 #include <memory>
0021
0022 namespace edm4hep {
0023
0024 using ReconstructedParticleObjPointerContainer = std::deque<ReconstructedParticleObj*>;
0025 using ReconstructedParticleDataContainer = std::vector<ReconstructedParticleData>;
0026
0027 /**
0028  * Class encapsulating everything related to storage of data that is needed by a
0029  * collection.
0030  */
0031 class ReconstructedParticleCollectionData {
0032 public:
0033   /**
0034    * The Objs of this collection
0035    */
0036   ReconstructedParticleObjPointerContainer entries{};
0037
0038   /**
0039    * Default constructor setting up the necessary buffers
0040    */
0041   ReconstructedParticleCollectionData();
0042
0043   /**
0044    * Constructor from existing I/O buffers
0045    */
0046   ReconstructedParticleCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
0047
0048   /**
0049    * Non copy-able, move-only class
0050    */
0051   ReconstructedParticleCollectionData(const ReconstructedParticleCollectionData&) = delete;
0052   ReconstructedParticleCollectionData& operator=(const ReconstructedParticleCollectionData&) = delete;
0053   ReconstructedParticleCollectionData(ReconstructedParticleCollectionData&& other) = default;
0054   ReconstructedParticleCollectionData& operator=(ReconstructedParticleCollectionData&& other) = default;
0055
0056   /**
0057    * Destructor
0058    */
0059   ~ReconstructedParticleCollectionData() = default;
0060
0061   void clear(bool isSubsetColl);
0062
0063   podio::CollectionWriteBuffers getCollectionBuffers(bool isSubsetColl);
0064
0065   void prepareForWrite(bool isSubsetColl);
0066
0067   void prepareAfterRead(uint32_t collectionID);
0068
0069   void makeSubsetCollection();
0070
0071   void createRelations(ReconstructedParticleObj* obj);
0072
0073   bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
0074
0075 private:
0076   // members to handle 1-to-N-relations
0077   podio::UVecPtr<edm4hep::Cluster> m_rel_clusters;                    ///< Relation buffer for read / write
0078   std::vector<podio::UVecPtr<edm4hep::Cluster>> m_rel_clusters_tmp{}; ///< Relation buffer for internal book-keeping
0079   podio::UVecPtr<edm4hep::Track> m_rel_tracks;                        ///< Relation buffer for read / write
0080   std::vector<podio::UVecPtr<edm4hep::Track>> m_rel_tracks_tmp{};     ///< Relation buffer for internal book-keeping
0081   podio::UVecPtr<edm4hep::ReconstructedParticle> m_rel_particles;     ///< Relation buffer for read / write
0082   std::vector<podio::UVecPtr<edm4hep::ReconstructedParticle>> m_rel_particles_tmp{}; ///< Relation buffer for internal
0083                                                                                      ///< book-keeping
0084   podio::UVecPtr<edm4hep::Vertex> m_rel_decayVertex{nullptr}; ///< Relation buffer for read / write
0085
0086   // members to handle vector members
0087
0088   // I/O related buffers
0089   podio::CollRefCollection m_refCollections{};
0090   podio::VectorMembersInfo m_vecmem_info{};
0091   std::unique_ptr<ReconstructedParticleDataContainer> m_data{nullptr};
0092 };
0093
0094 } // namespace edm4hep
0095
0096 #endif