include/edm4eic/ClusterCollectionData.h

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