include/edm4hep/EventHeaderCollectionData.h

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