TruthinessCollectionData.h

Go to the documentation of this file.

// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
 
#ifndef EDM4EIC_Truthiness_CollectionData_H
#define EDM4EIC_Truthiness_CollectionData_H
 
// datamodel specific includes
#include "edm4eic/TruthinessData.h"
#include "edm4eic/TruthinessObj.h"
#include "edm4eic/Truthiness.h"
#include "edm4eic/MCRecoParticleAssociation.h"
#include "edm4eic/ReconstructedParticle.h"
#include "edm4hep/MCParticle.h"
 
// schema evolution specific includes
 
// podio specific includes
#include "podio/CollectionBuffers.h"
#include "podio/ICollectionProvider.h"
 
#include <deque>
#include <memory>
 
namespace edm4eic {
 
 
using TruthinessObjPointerContainer = std::deque<TruthinessObj*>;
using TruthinessDataContainer = std::vector<TruthinessData>;
 
 
/**
 * Class encapsulating everything related to storage of data that is needed by a
 * collection.
 */
class TruthinessCollectionData {
public:
  /**
   * The Objs of this collection
   */
  TruthinessObjPointerContainer entries{};
 
  /**
   * Default constructor setting up the necessary buffers
   */
  TruthinessCollectionData();
 
  /**
   * Constructor from existing I/O buffers
   */
  TruthinessCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
 
  /**
   * Non copy-able, move-only class
   */
  TruthinessCollectionData(const TruthinessCollectionData&) = delete;
  TruthinessCollectionData& operator=(const TruthinessCollectionData&) = delete;
  TruthinessCollectionData(TruthinessCollectionData&& other) = default;
  TruthinessCollectionData& operator=(TruthinessCollectionData&& other) = default;
 
  /**
   * Destructor
   */
  ~TruthinessCollectionData() = default;
 
  void clear(bool isSubsetColl);
 
  podio::CollectionWriteBuffers getCollectionBuffers(bool isSubsetColl);
 
  void prepareForWrite(bool isSubsetColl);
 
  void prepareAfterRead(uint32_t collectionID);
 
  void makeSubsetCollection();
 
  void createRelations(TruthinessObj* obj);
 
  bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
 
private:
  // members to handle 1-to-N-relations
  podio::UVecPtr<edm4eic::MCRecoParticleAssociation> m_rel_associations{std::make_unique<std::vector<edm4eic::MCRecoParticleAssociation>>()}; ///< Relation buffer for read / write
  std::vector<podio::UVecPtr<edm4eic::MCRecoParticleAssociation>> m_rel_associations_tmp{}; ///< Relation buffer for internal book-keeping
  podio::UVecPtr<edm4hep::MCParticle> m_rel_unassociatedMCParticles{std::make_unique<std::vector<edm4hep::MCParticle>>()}; ///< Relation buffer for read / write
  std::vector<podio::UVecPtr<edm4hep::MCParticle>> m_rel_unassociatedMCParticles_tmp{}; ///< Relation buffer for internal book-keeping
  podio::UVecPtr<edm4eic::ReconstructedParticle> m_rel_unassociatedRecoParticles{std::make_unique<std::vector<edm4eic::ReconstructedParticle>>()}; ///< Relation buffer for read / write
  std::vector<podio::UVecPtr<edm4eic::ReconstructedParticle>> m_rel_unassociatedRecoParticles_tmp{}; ///< Relation buffer for internal book-keeping
 
  // members to handle vector members
  podio::UVecPtr<edm4eic::TruthinessContribution> m_vec_associationContributions{nullptr}; /// combined vector of all objects in collection
  std::vector<podio::UVecPtr<edm4eic::TruthinessContribution>> m_vecs_associationContributions{}; /// pointers to individual member vectors
 
  // I/O related buffers
  podio::CollRefCollection m_refCollections{};
  podio::VectorMembersInfo m_vecmem_info{};
  std::unique_ptr<TruthinessDataContainer> m_data{nullptr};
};
 
 
} // namespace edm4eic
 
 
#endif