3#ifndef EDM4EIC_ClusterCollection_H
4#define EDM4EIC_ClusterCollection_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"
57 return m_index <=> other.m_index;
61 return m_index == x.m_index;
80 Cluster m_object { podio::utils::MaybeSharedPtr<ClusterObj>{
nullptr} };
106 return m_index <=> other.m_index;
110 return m_index == x.m_index;
129 MutableCluster m_object { podio::utils::MaybeSharedPtr<ClusterObj>{
nullptr} };
159 constexpr static std::string_view
typeName =
"edm4eic::ClusterCollection";
161 constexpr static std::string_view
dataTypeName =
"edm4eic::ClusterData";
166 void print(std::ostream& os=std::cout,
bool flush=true) const final;
176 template<
typename... Args>
180 std::size_t
size() const final;
186 bool empty() const final;
198 return m_isSubsetColl;
204 Cluster operator[](std::
size_t index) const;
227 if (!m_isSubsetColl) {
228 std::for_each(m_storage.entries.begin(), m_storage.entries.end(),
229 [ID] (
ClusterObj* obj) { obj->id = {obj->id.index, static_cast<uint32_t>(ID)}; }
236 return m_collectionID;
243 size_t getDatamodelRegistryIndex() const final;
247 return iterator(0, &m_storage.entries);
256 return iterator(m_storage.entries.size(), &m_storage.entries);
259 return const_iterator(m_storage.entries.size(), &m_storage.entries);
285 std::vector<std::int32_t> type(
const size_t nElem = 0)
const;
286 std::vector<float> energy(
const size_t nElem = 0)
const;
287 std::vector<float> energyError(
const size_t nElem = 0)
const;
288 std::vector<float> time(
const size_t nElem = 0)
const;
289 std::vector<float> timeError(
const size_t nElem = 0)
const;
290 std::vector<std::uint32_t> nhits(
const size_t nElem = 0)
const;
291 std::vector<edm4hep::Vector3f> position(
const size_t nElem = 0)
const;
292 std::vector<edm4eic::Cov3f> positionError(
const size_t nElem = 0)
const;
293 std::vector<float> intrinsicTheta(
const size_t nElem = 0)
const;
294 std::vector<float> intrinsicPhi(
const size_t nElem = 0)
const;
295 std::vector<edm4eic::Cov2f> intrinsicDirectionError(
const size_t nElem = 0)
const;
303 bool m_isValid{
false};
304 mutable bool m_isPrepared{
false};
305 bool m_isSubsetColl{
false};
306 uint32_t m_collectionID{0};
307 mutable std::unique_ptr<std::mutex> m_storageMtx{
nullptr};
311std::ostream&
operator<<(std::ostream& o,
const ClusterCollection& v);
313template<
typename... Args>
315 if (m_isSubsetColl) {
316 throw std::logic_error(
"Cannot create new elements on a subset collection");
318 const int size = m_storage.entries.size();
319 auto obj =
new ClusterObj({
size, m_collectionID}, {std::forward<Args>(args)...});
320 m_storage.entries.push_back(obj);
323 obj->
m_clusters =
new std::vector<edm4eic::Cluster>();
324 obj->m_hits =
new std::vector<edm4eic::CalorimeterHit>();
325 obj->m_particleIDs =
new std::vector<edm4hep::ParticleID>();
326 obj->m_shapeParameters =
new std::vector<float>();
327 obj->m_hitContributions =
new std::vector<float>();
328 obj->m_subdetectorEnergies =
new std::vector<float>();
329 m_storage.createRelations(obj);
333#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
Definition ClusterCollectionData.h:32
Definition ClusterCollection.h:137
bool isSubsetCollection() const final
Definition ClusterCollection.h:197
static constexpr std::string_view dataTypeName
Definition ClusterCollection.h:161
const_iterator cend() const
Definition ClusterCollection.h:261
void prepareAfterRead() final
Definition ClusterCollection.cc:108
void clear() final
Definition ClusterCollection.cc:94
std::size_t max_size() const final
maximal number of elements in the collection
Definition ClusterCollection.cc:57
const_reverse_iterator rend() const
Definition ClusterCollection.h:277
ptrdiff_t difference_type
Definition ClusterCollection.h:143
reverse_iterator rend()
Definition ClusterCollection.h:274
ClusterCollectionIterator const_iterator
Definition ClusterCollection.h:141
std::reverse_iterator< const_iterator > const_reverse_iterator
Definition ClusterCollection.h:145
bool empty() const final
Is the collection empty.
Definition ClusterCollection.cc:67
Cluster at(std::size_t index) const
Returns the const object of given index.
Definition ClusterCollection.cc:41
ClusterCollection & operator=(ClusterCollection &&)=default
std::size_t size() const final
number of elements in the collection
Definition ClusterCollection.cc:53
void setID(uint32_t ID) final
Definition ClusterCollection.h:225
ClusterCollection()
Definition ClusterCollection.cc:26
const_iterator cbegin() const
Definition ClusterCollection.h:252
const_iterator begin() const
Definition ClusterCollection.h:249
static constexpr std::string_view valueTypeName
Definition ClusterCollection.h:160
ClusterCollection(const ClusterCollection &)=delete
MutableCluster mutable_type
Definition ClusterCollection.h:140
const std::string_view getDataTypeName() const final
fully qualified type name of stored POD elements - with namespace
Definition ClusterCollection.h:193
const std::string_view getTypeName() const final
fully qualified type name
Definition ClusterCollection.h:189
iterator end()
Definition ClusterCollection.h:255
friend class ClusterCollectionData
Definition ClusterCollection.h:301
const_iterator end() const
Definition ClusterCollection.h:258
ClusterCollection & operator=(const ClusterCollection &)=delete
ClusterCollection(ClusterCollection &&)=default
MutableCluster create()
Append a new object to the collection, and return this object.
Definition ClusterCollection.cc:82
bool setReferences(const podio::ICollectionProvider *collectionProvider) final
Definition ClusterCollection.cc:123
static constexpr std::string_view typeName
Definition ClusterCollection.h:159
const std::string_view getValueTypeName() const final
fully qualified type name of elements - with namespace
Definition ClusterCollection.h:191
podio::CollectionWriteBuffers getBuffers() final
Get the collection buffers for this collection.
Definition ClusterCollection.cc:160
const_reverse_iterator rbegin() const
Definition ClusterCollection.h:268
const_reverse_iterator crbegin() const
Definition ClusterCollection.h:271
void push_back(const MutableCluster &object)
Append object to the collection.
Definition ClusterCollection.cc:127
reverse_iterator rbegin()
Definition ClusterCollection.h:265
ClusterMutableCollectionIterator iterator
Definition ClusterCollection.h:142
bool isValid() const final
Definition ClusterCollection.h:239
void setSubsetCollection(bool setSubset=true) final
Definition ClusterCollection.cc:71
Cluster value_type
Definition ClusterCollection.h:139
size_t size_type
Definition ClusterCollection.h:144
iterator begin()
Definition ClusterCollection.h:246
podio::SchemaVersionT getSchemaVersion() const final
schema version
Definition ClusterCollection.cc:279
const_reverse_iterator crend() const
Definition ClusterCollection.h:280
~ClusterCollection() override
Definition ClusterCollection.cc:32
void print(std::ostream &os=std::cout, bool flush=true) const final
Print this collection to the passed stream.
Definition ClusterCollection.cc:583
void prepareForWrite() const final
Definition ClusterCollection.cc:99
uint32_t getID() const final
Definition ClusterCollection.h:235
std::reverse_iterator< iterator > reverse_iterator
Definition ClusterCollection.h:146
Definition ClusterCollection.h:36
ClusterCollectionIterator & operator+=(difference_type n)
Definition ClusterCollection.cc:422
ClusterCollectionIterator()=default
ClusterCollectionIterator & operator=(ClusterCollectionIterator &&)=default
Cluster value_type
Definition ClusterCollection.h:38
ptrdiff_t difference_type
Definition ClusterCollection.h:39
std::input_iterator_tag iterator_category
Definition ClusterCollection.h:42
ClusterCollectionIterator(const ClusterCollectionIterator &)=default
ClusterCollectionIterator & operator=(const ClusterCollectionIterator &)=default
bool operator==(const ClusterCollectionIterator &x) const
Definition ClusterCollection.h:60
reference operator[](difference_type n) const
Definition ClusterCollection.cc:448
ClusterCollectionIterator & operator-=(difference_type n)
Definition ClusterCollection.cc:437
ClusterCollectionIterator & operator--()
Definition ClusterCollection.cc:411
Cluster * pointer
Definition ClusterCollection.h:41
~ClusterCollectionIterator()=default
std::random_access_iterator_tag iterator_concept
Definition ClusterCollection.h:45
auto operator<=>(const ClusterCollectionIterator &other) const
Definition ClusterCollection.h:56
ClusterCollectionIterator operator+(difference_type n) const
Definition ClusterCollection.cc:427
Cluster reference
Definition ClusterCollection.h:40
ClusterCollectionIterator(ClusterCollectionIterator &&)=default
ClusterCollectionIterator operator-(difference_type n) const
Definition ClusterCollection.cc:442
pointer operator->()
Definition ClusterCollection.cc:395
ClusterCollectionIterator(size_t index, const ClusterObjPointerContainer *collection)
Definition ClusterCollection.h:47
ClusterCollectionIterator & operator++()
Definition ClusterCollection.cc:400
reference operator*() const
Definition ClusterCollection.cc:391
Definition ClusterCollection.h:85
auto operator<=>(const ClusterMutableCollectionIterator &other) const
Definition ClusterCollection.h:105
pointer operator->()
Definition ClusterCollection.cc:462
ClusterMutableCollectionIterator(size_t index, const ClusterObjPointerContainer *collection)
Definition ClusterCollection.h:96
~ClusterMutableCollectionIterator()=default
ptrdiff_t difference_type
Definition ClusterCollection.h:88
ClusterMutableCollectionIterator & operator=(const ClusterMutableCollectionIterator &)=default
ClusterMutableCollectionIterator & operator-=(difference_type n)
Definition ClusterCollection.cc:504
MutableCluster * pointer
Definition ClusterCollection.h:90
std::random_access_iterator_tag iterator_concept
Definition ClusterCollection.h:94
bool operator==(const ClusterMutableCollectionIterator &x) const
Definition ClusterCollection.h:109
reference operator*() const
Definition ClusterCollection.cc:458
std::input_iterator_tag iterator_category
Definition ClusterCollection.h:91
MutableCluster reference
Definition ClusterCollection.h:89
ClusterMutableCollectionIterator operator+(difference_type n) const
Definition ClusterCollection.cc:494
ClusterMutableCollectionIterator & operator+=(difference_type n)
Definition ClusterCollection.cc:489
ClusterMutableCollectionIterator operator-(difference_type n) const
Definition ClusterCollection.cc:509
ClusterMutableCollectionIterator()=default
reference operator[](difference_type n) const
Definition ClusterCollection.cc:515
ClusterMutableCollectionIterator & operator++()
Definition ClusterCollection.cc:467
ClusterMutableCollectionIterator & operator=(ClusterMutableCollectionIterator &&)=default
Cluster value_type
Definition ClusterCollection.h:87
ClusterMutableCollectionIterator(const ClusterMutableCollectionIterator &)=default
ClusterMutableCollectionIterator(ClusterMutableCollectionIterator &&)=default
ClusterMutableCollectionIterator & operator--()
Definition ClusterCollection.cc:478
Definition ClusterObj.h:21
std::vector< edm4eic::Cluster > * m_clusters
Definition ClusterObj.h:37
Definition MutableCluster.h:40
Definition CalorimeterHit.cc:17
std::deque< ClusterObj * > ClusterObjPointerContainer
Definition ClusterCollectionData.h:24
std::ostream & operator<<(std::ostream &o, const CalorimeterHit &value)
Definition CalorimeterHit.cc:103
Definition CalorimeterHit.h:31