EDM4eic
EIC data model
Loading...
Searching...
No Matches
ReconstructedParticleCollection.h
Go to the documentation of this file.
1// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
2
3#ifndef EDM4EIC_ReconstructedParticleCollection_H
4#define EDM4EIC_ReconstructedParticleCollection_H
5
6// datamodel specific includes
12
13// podio specific includes
14#include "podio/ICollectionProvider.h"
15#include "podio/CollectionBase.h"
16#include "podio/CollectionIDTable.h"
17
18#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
19#include "nlohmann/json_fwd.hpp"
20#endif
21
22#include <string>
23#include <vector>
24#include <deque>
25#include <array>
26#include <algorithm>
27#include <ostream>
28#include <mutex>
29#include <memory>
30#include <cstddef>
31
32namespace podio {
33 struct RelationNames;
34}
35
36namespace edm4eic {
37
38
39
41public:
42 ReconstructedParticleCollectionIterator(size_t index, const ReconstructedParticleObjPointerContainer* collection) : m_index(index), m_object(podio::utils::MaybeSharedPtr<ReconstructedParticleObj>{nullptr}), m_collection(collection) {}
43
46
48 return m_index != x.m_index; // TODO: may not be complete
49 }
50
52 return m_index == x.m_index; // TODO: may not be complete
53 }
54
58
59private:
60 size_t m_index;
61 ReconstructedParticle m_object;
63};
64
65
67public:
68 ReconstructedParticleMutableCollectionIterator(size_t index, const ReconstructedParticleObjPointerContainer* collection) : m_index(index), m_object(podio::utils::MaybeSharedPtr<ReconstructedParticleObj>{nullptr}), m_collection(collection) {}
69
72
74 return m_index != x.m_index; // TODO: may not be complete
75 }
76
78 return m_index == x.m_index; // TODO: may not be complete
79 }
80
84
85private:
86 size_t m_index;
89};
90
91
92/**
93A Collection is identified by an ID.
94*/
95class ReconstructedParticleCollection : public podio::CollectionBase {
96public:
100
103 // This is a move-only type
108
109// ReconstructedParticleCollection(ReconstructedParticleVector* data, uint32_t collectionID);
111
112 constexpr static auto typeName = "edm4eic::ReconstructedParticleCollection";
113 constexpr static auto valueTypeName = "edm4eic::ReconstructedParticle";
114 constexpr static auto dataTypeName = "edm4eic::ReconstructedParticleData";
115
116 void clear() final;
117
118 /// Print this collection to the passed stream
119 void print(std::ostream& os=std::cout, bool flush=true) const final;
120
121 /// operator to allow pointer like calling of members a la LCIO
123
124 /// Append a new object to the collection, and return this object.
126
127 /// Append a new object to the collection, and return this object.
128 /// Initialized with the parameters given
129 template<typename... Args>
130 MutableReconstructedParticle create(Args&&... args);
131
132 /// number of elements in the collection
133 std::size_t size() const final;
134
135 /// Is the collection empty
136 bool empty() const final;
137
138 /// fully qualified type name
139 const std::string_view getTypeName() const final { return typeName; }
140 /// fully qualified type name of elements - with namespace
141 const std::string_view getValueTypeName() const final { return valueTypeName; }
142 /// fully qualified type name of stored POD elements - with namespace
143 const std::string_view getDataTypeName() const final { return dataTypeName; }
144 /// schema version
145 podio::SchemaVersionT getSchemaVersion() const final;
146
147 bool isSubsetCollection() const final {
148 return m_isSubsetColl;
149 }
150
151 void setSubsetCollection(bool setSubset=true) final;
152
153 /// Returns the const object of given index
154 ReconstructedParticle operator[](std::size_t index) const;
155 /// Returns the object of a given index
156 MutableReconstructedParticle operator[](std::size_t index);
157 /// Returns the const object of given index
158 ReconstructedParticle at(std::size_t index) const;
159 /// Returns the object of given index
160 MutableReconstructedParticle at(std::size_t index);
161
162
163 /// Append object to the collection
165 /// Append an object to the (subset) collection
166 void push_back(ReconstructedParticle object);
167
168 void prepareForWrite() const final;
169 void prepareAfterRead() final;
170 bool setReferences(const podio::ICollectionProvider* collectionProvider) final;
171
172 /// Get the collection buffers for this collection
173 podio::CollectionWriteBuffers getBuffers() final;
174
175 void setID(uint32_t ID) final {
176 m_collectionID = ID;
177 if (!m_isSubsetColl) {
178 std::for_each(m_storage.entries.begin(), m_storage.entries.end(),
179 [ID] (ReconstructedParticleObj* obj) { obj->id = {obj->id.index, static_cast<uint32_t>(ID)}; }
180 );
181 }
182 m_isValid = true;
183 };
184
185 uint32_t getID() const final {
186 return m_collectionID;
187 }
188
189 bool isValid() const final {
190 return m_isValid;
191 }
192
193 size_t getDatamodelRegistryIndex() const final;
194
195 // support for the iterator protocol
196 iterator begin() {
197 return iterator(0, &m_storage.entries);
198 }
200 return const_iterator(0, &m_storage.entries);
201 }
203 return iterator(m_storage.entries.size(), &m_storage.entries);
204 }
206 return const_iterator(m_storage.entries.size(), &m_storage.entries);
207 }
208
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;
218
219private:
220 // For setReferences, we need to give our own CollectionData access to our
221 // private entries. Otherwise we would need to expose a public member function
222 // that gives access to the Obj* which is definitely not what we want
224
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{};
231};
232
233std::ostream& operator<<(std::ostream& o, const ReconstructedParticleCollection& v);
234
235template<typename... Args>
236MutableReconstructedParticle ReconstructedParticleCollection::create(Args&&... args) {
237 if (m_isSubsetColl) {
238 throw std::logic_error("Cannot create new elements on a subset collection");
239 }
240 const int size = m_storage.entries.size();
241 auto obj = new ReconstructedParticleObj({size, m_collectionID}, {std::forward<Args>(args)...});
242 m_storage.entries.push_back(obj);
243
244 // Need to initialize the relation vectors manually for the {ObjectID, ReconstructedParticleData} constructor
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);
250 return MutableReconstructedParticle(podio::utils::MaybeSharedPtr(obj));
251}
252
253#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
254void to_json(nlohmann::json& j, const ReconstructedParticleCollection& collection);
255#endif
256
257} // namespace edm4eic
258
259
260#endif
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