3#ifndef EDM4EIC_ClusterCollection_H
4#define EDM4EIC_ClusterCollection_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::ClusterCollection";
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;
154 Cluster operator[](std::
size_t index) const;
177 if (!m_isSubsetColl) {
179 [ID] (
ClusterObj* obj) { obj->id = {obj->id.index, static_cast<uint32_t>(ID)}; }
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<float> energyError(
const size_t nElem = 0)
const;
212 std::vector<float> time(
const size_t nElem = 0)
const;
213 std::vector<float> timeError(
const size_t nElem = 0)
const;
214 std::vector<std::uint32_t> nhits(
const size_t nElem = 0)
const;
215 std::vector<edm4hep::Vector3f> position(
const size_t nElem = 0)
const;
216 std::vector<edm4eic::Cov3f> positionError(
const size_t nElem = 0)
const;
217 std::vector<float> intrinsicTheta(
const size_t nElem = 0)
const;
218 std::vector<float> intrinsicPhi(
const size_t nElem = 0)
const;
219 std::vector<edm4eic::Cov2f> intrinsicDirectionError(
const size_t nElem = 0)
const;
227 bool m_isValid{
false};
228 mutable bool m_isPrepared{
false};
229 bool m_isSubsetColl{
false};
230 uint32_t m_collectionID{0};
231 mutable std::unique_ptr<std::mutex> m_storageMtx{
nullptr};
232 mutable ClusterCollectionData m_storage{};
235std::ostream&
operator<<(std::ostream& o,
const ClusterCollection& v);
237template<
typename... Args>
239 if (m_isSubsetColl) {
240 throw std::logic_error(
"Cannot create new elements on a subset collection");
242 const int size = m_storage.entries.size();
243 auto obj =
new ClusterObj({size, m_collectionID}, {std::forward<Args>(args)...});
244 m_storage.entries.push_back(obj);
247 obj->
m_clusters =
new std::vector<edm4eic::Cluster>();
248 obj->m_hits =
new std::vector<edm4eic::CalorimeterHit>();
249 obj->m_particleIDs =
new std::vector<edm4hep::ParticleID>();
250 obj->m_shapeParameters =
new std::vector<float>();
251 obj->m_hitContributions =
new std::vector<float>();
252 obj->m_subdetectorEnergies =
new std::vector<float>();
253 m_storage.createRelations(obj);
257#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
Definition ClusterCollectionData.h:32
ClusterObjPointerContainer entries
Definition ClusterCollectionData.h:37
Definition ClusterCollection.h:95
bool isSubsetCollection() const final
Definition ClusterCollection.h:147
void prepareAfterRead() final
Definition ClusterCollection.cc:109
void clear() final
Definition ClusterCollection.cc:83
bool empty() const final
Is the collection empty.
Definition ClusterCollection.cc:56
Cluster at(std::size_t index) const
Returns the const object of given index.
Definition ClusterCollection.cc:40
ClusterCollection & operator=(ClusterCollection &&)=default
std::size_t size() const final
number of elements in the collection
Definition ClusterCollection.cc:52
void setID(uint32_t ID) final
Definition ClusterCollection.h:175
ClusterCollection()
Definition ClusterCollection.cc:25
const_iterator begin() const
Definition ClusterCollection.h:199
ClusterCollection(const ClusterCollection &)=delete
const std::string_view getDataTypeName() const final
fully qualified type name of stored POD elements - with namespace
Definition ClusterCollection.h:143
const std::string_view getTypeName() const final
fully qualified type name
Definition ClusterCollection.h:139
iterator end()
Definition ClusterCollection.h:202
const_iterator end() const
Definition ClusterCollection.h:205
ClusterCollection & operator=(const ClusterCollection &)=delete
ClusterCollection(ClusterCollection &&)=default
void push_back(MutableCluster object)
Append object to the collection.
Definition ClusterCollection.cc:128
MutableCluster create()
Append a new object to the collection, and return this object.
Definition ClusterCollection.cc:71
bool setReferences(const podio::ICollectionProvider *collectionProvider) final
Definition ClusterCollection.cc:124
static constexpr auto valueTypeName
Definition ClusterCollection.h:113
const std::string_view getValueTypeName() const final
fully qualified type name of elements - with namespace
Definition ClusterCollection.h:141
podio::CollectionWriteBuffers getBuffers() final
Get the collection buffers for this collection.
Definition ClusterCollection.cc:161
bool isValid() const final
Definition ClusterCollection.h:189
void setSubsetCollection(bool setSubset=true) final
Definition ClusterCollection.cc:60
static constexpr auto dataTypeName
Definition ClusterCollection.h:114
static constexpr auto typeName
Definition ClusterCollection.h:112
podio::SchemaVersionT getSchemaVersion() const final
schema version
Definition ClusterCollection.cc:280
void print(std::ostream &os=std::cout, bool flush=true) const final
Print this collection to the passed stream.
Definition ClusterCollection.cc:479
void prepareForWrite() const final
Definition ClusterCollection.cc:88
~ClusterCollection()
Definition ClusterCollection.cc:31
uint32_t getID() const final
Definition ClusterCollection.h:185
Definition ClusterCollection.h:40
ClusterCollectionIterator & operator=(const ClusterCollectionIterator &)=delete
Cluster * operator->()
Definition ClusterCollection.cc:392
bool operator==(const ClusterCollectionIterator &x) const
Definition ClusterCollection.h:51
bool operator!=(const ClusterCollectionIterator &x) const
Definition ClusterCollection.h:47
Cluster operator*()
Definition ClusterCollection.cc:387
ClusterCollectionIterator(const ClusterCollectionIterator &)=delete
ClusterCollectionIterator(size_t index, const ClusterObjPointerContainer *collection)
Definition ClusterCollection.h:42
ClusterCollectionIterator & operator++()
Definition ClusterCollection.cc:397
Definition ClusterCollection.h:66
ClusterMutableCollectionIterator(size_t index, const ClusterObjPointerContainer *collection)
Definition ClusterCollection.h:68
ClusterMutableCollectionIterator(const ClusterMutableCollectionIterator &)=delete
MutableCluster * operator->()
Definition ClusterCollection.cc:409
bool operator!=(const ClusterMutableCollectionIterator &x) const
Definition ClusterCollection.h:73
bool operator==(const ClusterMutableCollectionIterator &x) const
Definition ClusterCollection.h:77
ClusterMutableCollectionIterator & operator=(const ClusterMutableCollectionIterator &)=delete
ClusterMutableCollectionIterator & operator++()
Definition ClusterCollection.cc:414
MutableCluster operator*()
Definition ClusterCollection.cc:404
Definition ClusterObj.h:20
std::vector< edm4eic::Cluster > * m_clusters
Definition ClusterObj.h:36
Definition MutableCluster.h:37
Definition CalorimeterHit.cc:18
std::deque< ClusterObj * > ClusterObjPointerContainer
Definition ClusterCollectionData.h:24
std::ostream & operator<<(std::ostream &o, const CalorimeterHit &value)
Definition CalorimeterHit.cc:97
Definition CalorimeterHitCollection.h:32