3#ifndef EDM4EIC_ReconstructedParticleCollection_H
4#define EDM4EIC_ReconstructedParticleCollection_H
14#include "podio/ICollectionProvider.h"
15#include "podio/CollectionBase.h"
16#include "podio/CollectionIDTable.h"
18#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
19#include "nlohmann/json_fwd.hpp"
48 return m_index != x.m_index;
52 return m_index == x.m_index;
74 return m_index != x.m_index;
78 return m_index == x.m_index;
112 constexpr static auto typeName =
"edm4eic::ReconstructedParticleCollection";
114 constexpr static auto dataTypeName =
"edm4eic::ReconstructedParticleData";
119 void print(std::ostream& os=std::cout,
bool flush=true) const final;
129 template<
typename... Args>
133 std::size_t
size() const final;
136 bool empty() const final;
148 return m_isSubsetColl;
177 if (!m_isSubsetColl) {
186 return m_collectionID;
193 size_t getDatamodelRegistryIndex() const final;
197 return iterator(0, &m_storage.entries);
203 return iterator(m_storage.entries.size(), &m_storage.entries);
206 return const_iterator(m_storage.entries.size(), &m_storage.entries);
209 std::vector<std::int32_t> type(
const size_t nElem = 0)
const;
210 std::vector<float> energy(
const size_t nElem = 0)
const;
211 std::vector<edm4hep::Vector3f> momentum(
const size_t nElem = 0)
const;
212 std::vector<edm4hep::Vector3f> referencePoint(
const size_t nElem = 0)
const;
213 std::vector<float> charge(
const size_t nElem = 0)
const;
214 std::vector<float> mass(
const size_t nElem = 0)
const;
215 std::vector<float> goodnessOfPID(
const size_t nElem = 0)
const;
216 std::vector<edm4eic::Cov4f> covMatrix(
const size_t nElem = 0)
const;
217 std::vector<std::int32_t> PDG(
const size_t nElem = 0)
const;
225 bool m_isValid{
false};
226 mutable bool m_isPrepared{
false};
227 bool m_isSubsetColl{
false};
228 uint32_t m_collectionID{0};
229 mutable std::unique_ptr<std::mutex> m_storageMtx{
nullptr};
230 mutable ReconstructedParticleCollectionData m_storage{};
233std::ostream&
operator<<(std::ostream& o,
const ReconstructedParticleCollection& v);
235template<
typename... Args>
237 if (m_isSubsetColl) {
238 throw std::logic_error(
"Cannot create new elements on a subset collection");
240 const int size = m_storage.entries.size();
242 m_storage.entries.push_back(obj);
245 obj->
m_clusters =
new std::vector<edm4eic::Cluster>();
246 obj->m_tracks =
new std::vector<edm4eic::Track>();
247 obj->m_particles =
new std::vector<edm4eic::ReconstructedParticle>();
248 obj->m_particleIDs =
new std::vector<edm4hep::ParticleID>();
249 m_storage.createRelations(obj);
253#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
Definition MutableReconstructedParticle.h:46
Definition ReconstructedParticleCollectionData.h:34
ReconstructedParticleObjPointerContainer entries
Definition ReconstructedParticleCollectionData.h:39
Definition ReconstructedParticleCollection.h:95
void clear() final
Definition ReconstructedParticleCollection.cc:84
static constexpr auto typeName
Definition ReconstructedParticleCollection.h:112
void setSubsetCollection(bool setSubset=true) final
Definition ReconstructedParticleCollection.cc:61
const std::string_view getDataTypeName() const final
fully qualified type name of stored POD elements - with namespace
Definition ReconstructedParticleCollection.h:143
ReconstructedParticle at(std::size_t index) const
Returns the const object of given index.
Definition ReconstructedParticleCollection.cc:41
ReconstructedParticleCollection()
Definition ReconstructedParticleCollection.cc:26
MutableReconstructedParticle create()
Append a new object to the collection, and return this object.
Definition ReconstructedParticleCollection.cc:72
iterator end()
Definition ReconstructedParticleCollection.h:202
const_iterator begin() const
Definition ReconstructedParticleCollection.h:199
ReconstructedParticleCollection(ReconstructedParticleCollection &&)=default
void setID(uint32_t ID) final
Definition ReconstructedParticleCollection.h:175
bool empty() const final
Is the collection empty.
Definition ReconstructedParticleCollection.cc:57
std::size_t size() const final
number of elements in the collection
Definition ReconstructedParticleCollection.cc:53
ReconstructedParticleCollection & operator=(ReconstructedParticleCollection &&)=default
ReconstructedParticleCollection & operator=(const ReconstructedParticleCollection &)=delete
void prepareAfterRead() final
Definition ReconstructedParticleCollection.cc:110
bool isSubsetCollection() const final
Definition ReconstructedParticleCollection.h:147
void prepareForWrite() const final
Definition ReconstructedParticleCollection.cc:89
uint32_t getID() const final
Definition ReconstructedParticleCollection.h:185
ReconstructedParticleCollection(const ReconstructedParticleCollection &)=delete
static constexpr auto dataTypeName
Definition ReconstructedParticleCollection.h:114
podio::SchemaVersionT getSchemaVersion() const final
schema version
Definition ReconstructedParticleCollection.cc:261
void push_back(MutableReconstructedParticle object)
Append object to the collection.
Definition ReconstructedParticleCollection.cc:129
bool setReferences(const podio::ICollectionProvider *collectionProvider) final
Definition ReconstructedParticleCollection.cc:125
static constexpr auto valueTypeName
Definition ReconstructedParticleCollection.h:113
podio::CollectionWriteBuffers getBuffers() final
Get the collection buffers for this collection.
Definition ReconstructedParticleCollection.cc:162
~ReconstructedParticleCollection()
Definition ReconstructedParticleCollection.cc:32
void print(std::ostream &os=std::cout, bool flush=true) const final
Print this collection to the passed stream.
Definition ReconstructedParticleCollection.cc:443
const std::string_view getValueTypeName() const final
fully qualified type name of elements - with namespace
Definition ReconstructedParticleCollection.h:141
bool isValid() const final
Definition ReconstructedParticleCollection.h:189
const std::string_view getTypeName() const final
fully qualified type name
Definition ReconstructedParticleCollection.h:139
const_iterator end() const
Definition ReconstructedParticleCollection.h:205
Definition ReconstructedParticleCollection.h:40
bool operator==(const ReconstructedParticleCollectionIterator &x) const
Definition ReconstructedParticleCollection.h:51
ReconstructedParticleCollectionIterator & operator=(const ReconstructedParticleCollectionIterator &)=delete
ReconstructedParticleCollectionIterator & operator++()
Definition ReconstructedParticleCollection.cc:369
ReconstructedParticle operator*()
Definition ReconstructedParticleCollection.cc:359
ReconstructedParticle * operator->()
Definition ReconstructedParticleCollection.cc:364
ReconstructedParticleCollectionIterator(const ReconstructedParticleCollectionIterator &)=delete
ReconstructedParticleCollectionIterator(size_t index, const ReconstructedParticleObjPointerContainer *collection)
Definition ReconstructedParticleCollection.h:42
bool operator!=(const ReconstructedParticleCollectionIterator &x) const
Definition ReconstructedParticleCollection.h:47
Definition ReconstructedParticle.h:47
Definition ReconstructedParticleCollection.h:66
ReconstructedParticleMutableCollectionIterator(size_t index, const ReconstructedParticleObjPointerContainer *collection)
Definition ReconstructedParticleCollection.h:68
bool operator!=(const ReconstructedParticleMutableCollectionIterator &x) const
Definition ReconstructedParticleCollection.h:73
ReconstructedParticleMutableCollectionIterator(const ReconstructedParticleMutableCollectionIterator &)=delete
ReconstructedParticleMutableCollectionIterator & operator=(const ReconstructedParticleMutableCollectionIterator &)=delete
ReconstructedParticleMutableCollectionIterator & operator++()
Definition ReconstructedParticleCollection.cc:386
MutableReconstructedParticle operator*()
Definition ReconstructedParticleCollection.cc:376
MutableReconstructedParticle * operator->()
Definition ReconstructedParticleCollection.cc:381
bool operator==(const ReconstructedParticleMutableCollectionIterator &x) const
Definition ReconstructedParticleCollection.h:77
Definition ReconstructedParticleObj.h:28
std::vector< edm4eic::Cluster > * m_clusters
Definition ReconstructedParticleObj.h:46
Definition CalorimeterHit.cc:18
std::deque< ReconstructedParticleObj * > ReconstructedParticleObjPointerContainer
Definition ReconstructedParticleCollectionData.h:26
std::ostream & operator<<(std::ostream &o, const CalorimeterHit &value)
Definition CalorimeterHit.cc:97
Definition CalorimeterHitCollection.h:32