ReconstructedParticleCollectionData.h

Go to the documentation of this file.

// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
 
#ifndef EICD_ReconstructedParticle_CollectionData_H
#define EICD_ReconstructedParticle_CollectionData_H
 
// datamodel specific includes
#include "eicd/ReconstructedParticleData.h"
#include "eicd/ReconstructedParticleObj.h"
#include "eicd/Cluster.h"
#include "eicd/Track.h"
#include "eicd/Vertex.h"
#include "edm4hep/ParticleID.h"
 
// podio specific includes
#include "podio/CollectionBuffers.h"
#include "podio/ICollectionProvider.h"
 
#include <deque>
#include <memory>
 
namespace eicd {
 
 
using ReconstructedParticleObjPointerContainer = std::deque<ReconstructedParticleObj*>;
using ReconstructedParticleDataContainer = std::vector<ReconstructedParticleData>;
 
 
/**
 * Class encapsulating everything related to storage of data that is needed by a
 * collection.
 */
class ReconstructedParticleCollectionData {
public:
  /**
   * The Objs of this collection
   */
  ReconstructedParticleObjPointerContainer entries{};
 
  /**
   * Default constructor setting up the necessary buffers
   */
  ReconstructedParticleCollectionData();
 
  /**
   * Non copy-able, move-only class
   */
  ReconstructedParticleCollectionData(const ReconstructedParticleCollectionData&) = delete;
  ReconstructedParticleCollectionData& operator=(const ReconstructedParticleCollectionData&) = delete;
  ReconstructedParticleCollectionData(ReconstructedParticleCollectionData&& other) = default;
  ReconstructedParticleCollectionData& operator=(ReconstructedParticleCollectionData&& other) = default;
 
  /**
   * Deconstructor
   */
  ~ReconstructedParticleCollectionData() = default;
 
  void clear(bool isSubsetColl);
 
  podio::CollectionBuffers getCollectionBuffers(bool isSubsetColl);
 
  void prepareForWrite(bool isSubsetColl);
 
  void prepareAfterRead(int collectionID);
 
  void makeSubsetCollection();
 
  void createRelations(ReconstructedParticleObj* obj);
 
  bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
 
private:
  // members to handle 1-to-N-relations
  podio::UVecPtr<eicd::Cluster> m_rel_clusters;  ///< Relation buffer for read / write
  std::vector<podio::UVecPtr<eicd::Cluster>> m_rel_clusters_tmp{}; ///< Relation buffer for internal book-keeping
  podio::UVecPtr<eicd::Track> m_rel_tracks;  ///< Relation buffer for read / write
  std::vector<podio::UVecPtr<eicd::Track>> m_rel_tracks_tmp{}; ///< Relation buffer for internal book-keeping
  podio::UVecPtr<eicd::ReconstructedParticle> m_rel_particles;  ///< Relation buffer for read / write
  std::vector<podio::UVecPtr<eicd::ReconstructedParticle>> m_rel_particles_tmp{}; ///< Relation buffer for internal book-keeping
  podio::UVecPtr<edm4hep::ParticleID> m_rel_particleIDs;  ///< Relation buffer for read / write
  std::vector<podio::UVecPtr<edm4hep::ParticleID>> m_rel_particleIDs_tmp{}; ///< Relation buffer for internal book-keeping
  podio::UVecPtr<eicd::Vertex> m_rel_startVertex; ///< Relation buffer for read / write
  podio::UVecPtr<edm4hep::ParticleID> m_rel_particleIDUsed; ///< Relation buffer for read / write
 
  // members to handle vector members
 
  // I/O related buffers
  podio::CollRefCollection m_refCollections{};
  podio::VectorMembersInfo m_vecmem_info{};
  std::unique_ptr<ReconstructedParticleDataContainer> m_data{nullptr};
};
 
 
} // namespace eicd
 
 
#endif