include/edm4hep/TimeSeriesCollectionData.h

0001 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
0002
0003 #ifndef EDM4HEP_TimeSeries_CollectionData_H
0004 #define EDM4HEP_TimeSeries_CollectionData_H
0005
0006 // datamodel specific includes
0007 #include "edm4hep/TimeSeriesData.h"
0008 #include "edm4hep/TimeSeriesObj.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 TimeSeriesObjPointerContainer = std::deque<TimeSeriesObj*>;
0022 using TimeSeriesDataContainer = std::vector<TimeSeriesData>;
0023
0024 /**
0025  * Class encapsulating everything related to storage of data that is needed by a
0026  * collection.
0027  */
0028 class TimeSeriesCollectionData {
0029 public:
0030   /**
0031    * The Objs of this collection
0032    */
0033   TimeSeriesObjPointerContainer entries{};
0034
0035   /**
0036    * Default constructor setting up the necessary buffers
0037    */
0038   TimeSeriesCollectionData();
0039
0040   /**
0041    * Constructor from existing I/O buffers
0042    */
0043   TimeSeriesCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
0044
0045   /**
0046    * Non copy-able, move-only class
0047    */
0048   TimeSeriesCollectionData(const TimeSeriesCollectionData&) = delete;
0049   TimeSeriesCollectionData& operator=(const TimeSeriesCollectionData&) = delete;
0050   TimeSeriesCollectionData(TimeSeriesCollectionData&& other) = default;
0051   TimeSeriesCollectionData& operator=(TimeSeriesCollectionData&& other) = default;
0052
0053   /**
0054    * Destructor
0055    */
0056   ~TimeSeriesCollectionData() = 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(TimeSeriesObj* 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<float> m_vec_amplitude{nullptr};        /// combined vector of all objects in collection
0077   std::vector<podio::UVecPtr<float>> m_vecs_amplitude{}; /// 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<TimeSeriesDataContainer> m_data{nullptr};
0083 };
0084
0085 } // namespace edm4hep
0086
0087 #endif