TensorCollectionData.h

Go to the documentation of this file.

// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
 
#ifndef EDM4EIC_Tensor_CollectionData_H
#define EDM4EIC_Tensor_CollectionData_H
 
// datamodel specific includes
#include "edm4eic/TensorData.h"
#include "edm4eic/TensorObj.h"
#include "edm4eic/Tensor.h"
 
// schema evolution specific includes
 
// podio specific includes
#include "podio/CollectionBuffers.h"
#include "podio/ICollectionProvider.h"
 
#include <deque>
#include <memory>
 
namespace edm4eic {
 
 
using TensorObjPointerContainer = std::deque<TensorObj*>;
using TensorDataContainer = std::vector<TensorData>;
 
 
/**
 * Class encapsulating everything related to storage of data that is needed by a
 * collection.
 */
class TensorCollectionData {
public:
  /**
   * The Objs of this collection
   */
  TensorObjPointerContainer entries{};
 
  /**
   * Default constructor setting up the necessary buffers
   */
  TensorCollectionData();
 
  /**
   * Constructor from existing I/O buffers
   */
  TensorCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
 
  /**
   * Non copy-able, move-only class
   */
  TensorCollectionData(const TensorCollectionData&) = delete;
  TensorCollectionData& operator=(const TensorCollectionData&) = delete;
  TensorCollectionData(TensorCollectionData&& other) = default;
  TensorCollectionData& operator=(TensorCollectionData&& other) = default;
 
  /**
   * Destructor
   */
  ~TensorCollectionData() = default;
 
  void clear(bool isSubsetColl);
 
  podio::CollectionWriteBuffers getCollectionBuffers(bool isSubsetColl);
 
  void prepareForWrite(bool isSubsetColl);
 
  void prepareAfterRead(uint32_t collectionID);
 
  void makeSubsetCollection();
 
  void createRelations(TensorObj* obj);
 
  bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
 
private:
  // members to handle 1-to-N-relations
 
  // members to handle vector members
  podio::UVecPtr<std::int64_t> m_vec_shape{nullptr}; /// combined vector of all objects in collection
  std::vector<podio::UVecPtr<std::int64_t>> m_vecs_shape{}; /// pointers to individual member vectors
  podio::UVecPtr<float> m_vec_floatData{nullptr}; /// combined vector of all objects in collection
  std::vector<podio::UVecPtr<float>> m_vecs_floatData{}; /// pointers to individual member vectors
  podio::UVecPtr<std::int64_t> m_vec_int64Data{nullptr}; /// combined vector of all objects in collection
  std::vector<podio::UVecPtr<std::int64_t>> m_vecs_int64Data{}; /// pointers to individual member vectors
 
  // I/O related buffers
  podio::CollRefCollection m_refCollections{};
  podio::VectorMembersInfo m_vecmem_info{};
  std::unique_ptr<TensorDataContainer> m_data{nullptr};
};
 
 
} // namespace edm4eic
 
 
#endif