3#ifndef EDM4EIC_ReconstructedParticleCollection_H
4#define EDM4EIC_ReconstructedParticleCollection_H
13#include "podio/ICollectionProvider.h"
14#include "podio/CollectionBase.h"
16#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
17#include "nlohmann/json_fwd.hpp"
44 return m_index != x.m_index;
48 return m_index == x.m_index;
70 return m_index != x.m_index;
74 return m_index == x.m_index;
110 constexpr static auto typeName =
"edm4eic::ReconstructedParticleCollection";
112 constexpr static auto dataTypeName =
"edm4eic::ReconstructedParticleData";
117 void print(std::ostream& os=std::cout,
bool flush=true) const final;
127 template<
typename... Args>
131 std::size_t
size() const final;
137 bool empty() const final;
149 return m_isSubsetColl;
178 if (!m_isSubsetColl) {
187 return m_collectionID;
194 size_t getDatamodelRegistryIndex() const final;
198 return iterator(0, &m_storage.entries);
207 return iterator(m_storage.entries.size(), &m_storage.entries);
210 return const_iterator(m_storage.entries.size(), &m_storage.entries);
216 std::vector<std::int32_t> type(
const size_t nElem = 0)
const;
217 std::vector<float> energy(
const size_t nElem = 0)
const;
218 std::vector<edm4hep::Vector3f> momentum(
const size_t nElem = 0)
const;
219 std::vector<edm4hep::Vector3f> referencePoint(
const size_t nElem = 0)
const;
220 std::vector<float> charge(
const size_t nElem = 0)
const;
221 std::vector<float> mass(
const size_t nElem = 0)
const;
222 std::vector<float> goodnessOfPID(
const size_t nElem = 0)
const;
223 std::vector<edm4eic::Cov4f> covMatrix(
const size_t nElem = 0)
const;
224 std::vector<std::int32_t> PDG(
const size_t nElem = 0)
const;
232 bool m_isValid{
false};
233 mutable bool m_isPrepared{
false};
234 bool m_isSubsetColl{
false};
235 uint32_t m_collectionID{0};
236 mutable std::unique_ptr<std::mutex> m_storageMtx{
nullptr};
237 mutable ReconstructedParticleCollectionData m_storage{};
240std::ostream&
operator<<(std::ostream& o,
const ReconstructedParticleCollection& v);
242template<
typename... Args>
244 if (m_isSubsetColl) {
245 throw std::logic_error(
"Cannot create new elements on a subset collection");
247 const int size = m_storage.entries.size();
249 m_storage.entries.push_back(obj);
252 obj->
m_clusters =
new std::vector<edm4eic::Cluster>();
253 obj->m_tracks =
new std::vector<edm4eic::Track>();
254 obj->m_particles =
new std::vector<edm4eic::ReconstructedParticle>();
255 obj->m_particleIDs =
new std::vector<edm4hep::ParticleID>();
256 m_storage.createRelations(obj);
260#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
Definition MutableReconstructedParticle.h:46
Definition ReconstructedParticleCollectionData.h:34
ReconstructedParticleObjPointerContainer entries
Definition ReconstructedParticleCollectionData.h:39
Definition ReconstructedParticleCollection.h:91
void clear() final
Definition ReconstructedParticleCollection.cc:88
static constexpr auto typeName
Definition ReconstructedParticleCollection.h:110
void setSubsetCollection(bool setSubset=true) final
Definition ReconstructedParticleCollection.cc:65
const std::string_view getDataTypeName() const final
fully qualified type name of stored POD elements - with namespace
Definition ReconstructedParticleCollection.h:144
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:76
iterator end()
Definition ReconstructedParticleCollection.h:206
ptrdiff_t difference_type
Definition ReconstructedParticleCollection.h:96
const_iterator begin() const
Definition ReconstructedParticleCollection.h:200
ReconstructedParticleCollection(ReconstructedParticleCollection &&)=default
void setID(uint32_t ID) final
Definition ReconstructedParticleCollection.h:176
bool empty() const final
Is the collection empty.
Definition ReconstructedParticleCollection.cc:61
std::size_t size() const final
number of elements in the collection
Definition ReconstructedParticleCollection.cc:53
ReconstructedParticleCollection & operator=(ReconstructedParticleCollection &&)=default
size_t size_type
Definition ReconstructedParticleCollection.h:97
void push_back(const MutableReconstructedParticle &object)
Append object to the collection.
Definition ReconstructedParticleCollection.cc:133
ReconstructedParticleCollection & operator=(const ReconstructedParticleCollection &)=delete
void prepareAfterRead() final
Definition ReconstructedParticleCollection.cc:114
const_iterator cend() const
Definition ReconstructedParticleCollection.h:212
bool isSubsetCollection() const final
Definition ReconstructedParticleCollection.h:148
const_iterator cbegin() const
Definition ReconstructedParticleCollection.h:203
void prepareForWrite() const final
Definition ReconstructedParticleCollection.cc:93
uint32_t getID() const final
Definition ReconstructedParticleCollection.h:186
~ReconstructedParticleCollection() override
Definition ReconstructedParticleCollection.cc:32
ReconstructedParticleCollection(const ReconstructedParticleCollection &)=delete
static constexpr auto dataTypeName
Definition ReconstructedParticleCollection.h:112
podio::SchemaVersionT getSchemaVersion() const final
schema version
Definition ReconstructedParticleCollection.cc:265
bool setReferences(const podio::ICollectionProvider *collectionProvider) final
Definition ReconstructedParticleCollection.cc:129
static constexpr auto valueTypeName
Definition ReconstructedParticleCollection.h:111
std::size_t max_size() const final
maximal number of elements in the collection
Definition ReconstructedParticleCollection.cc:57
podio::CollectionWriteBuffers getBuffers() final
Get the collection buffers for this collection.
Definition ReconstructedParticleCollection.cc:166
void print(std::ostream &os=std::cout, bool flush=true) const final
Print this collection to the passed stream.
Definition ReconstructedParticleCollection.cc:452
const std::string_view getValueTypeName() const final
fully qualified type name of elements - with namespace
Definition ReconstructedParticleCollection.h:142
bool isValid() const final
Definition ReconstructedParticleCollection.h:190
const std::string_view getTypeName() const final
fully qualified type name
Definition ReconstructedParticleCollection.h:140
const_iterator end() const
Definition ReconstructedParticleCollection.h:209
Definition ReconstructedParticleCollection.h:36
bool operator==(const ReconstructedParticleCollectionIterator &x) const
Definition ReconstructedParticleCollection.h:47
ReconstructedParticleCollectionIterator & operator=(const ReconstructedParticleCollectionIterator &)=delete
ReconstructedParticleCollectionIterator & operator++()
Definition ReconstructedParticleCollection.cc:378
ReconstructedParticle operator*()
Definition ReconstructedParticleCollection.cc:368
ReconstructedParticle * operator->()
Definition ReconstructedParticleCollection.cc:373
ReconstructedParticleCollectionIterator(const ReconstructedParticleCollectionIterator &)=delete
ReconstructedParticleCollectionIterator(size_t index, const ReconstructedParticleObjPointerContainer *collection)
Definition ReconstructedParticleCollection.h:38
bool operator!=(const ReconstructedParticleCollectionIterator &x) const
Definition ReconstructedParticleCollection.h:43
Definition ReconstructedParticle.h:54
Definition ReconstructedParticleCollection.h:62
ReconstructedParticleMutableCollectionIterator(size_t index, const ReconstructedParticleObjPointerContainer *collection)
Definition ReconstructedParticleCollection.h:64
bool operator!=(const ReconstructedParticleMutableCollectionIterator &x) const
Definition ReconstructedParticleCollection.h:69
ReconstructedParticleMutableCollectionIterator(const ReconstructedParticleMutableCollectionIterator &)=delete
ReconstructedParticleMutableCollectionIterator & operator=(const ReconstructedParticleMutableCollectionIterator &)=delete
ReconstructedParticleMutableCollectionIterator & operator++()
Definition ReconstructedParticleCollection.cc:395
MutableReconstructedParticle operator*()
Definition ReconstructedParticleCollection.cc:385
MutableReconstructedParticle * operator->()
Definition ReconstructedParticleCollection.cc:390
bool operator==(const ReconstructedParticleMutableCollectionIterator &x) const
Definition ReconstructedParticleCollection.h:73
Definition ReconstructedParticleObj.h:29
std::vector< edm4eic::Cluster > * m_clusters
Definition ReconstructedParticleObj.h:47
Definition CalorimeterHit.cc:17
std::deque< ReconstructedParticleObj * > ReconstructedParticleObjPointerContainer
Definition ReconstructedParticleCollectionData.h:26
std::ostream & operator<<(std::ostream &o, const CalorimeterHit &value)
Definition CalorimeterHit.cc:103
Definition CalorimeterHit.h:31