3#ifndef EDM4EIC_TrackCollection_H
4#define EDM4EIC_TrackCollection_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::TrackCollection";
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;
155 Track operator[](std::
size_t index) const;
159 Track at(std::
size_t index) const;
178 if (!m_isSubsetColl) {
180 [ID] (
TrackObj* obj) { obj->id = {obj->id.index, static_cast<uint32_t>(ID)}; }
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<edm4hep::Vector3f> position(
const size_t nElem = 0)
const;
218 std::vector<edm4hep::Vector3f> momentum(
const size_t nElem = 0)
const;
219 std::vector<edm4eic::Cov6f> positionMomentumCovariance(
const size_t nElem = 0)
const;
220 std::vector<float> time(
const size_t nElem = 0)
const;
221 std::vector<float> timeError(
const size_t nElem = 0)
const;
222 std::vector<float> charge(
const size_t nElem = 0)
const;
223 std::vector<float> chi2(
const size_t nElem = 0)
const;
224 std::vector<std::uint32_t> ndf(
const size_t nElem = 0)
const;
225 std::vector<std::int32_t> pdg(
const size_t nElem = 0)
const;
233 bool m_isValid{
false};
234 mutable bool m_isPrepared{
false};
235 bool m_isSubsetColl{
false};
236 uint32_t m_collectionID{0};
237 mutable std::unique_ptr<std::mutex> m_storageMtx{
nullptr};
238 mutable TrackCollectionData m_storage{};
241std::ostream&
operator<<(std::ostream& o,
const TrackCollection& v);
243template<
typename... Args>
245 if (m_isSubsetColl) {
246 throw std::logic_error(
"Cannot create new elements on a subset collection");
248 const int size = m_storage.entries.size();
249 auto obj =
new TrackObj({size, m_collectionID}, {std::forward<Args>(args)...});
250 m_storage.entries.push_back(obj);
254 obj->m_tracks =
new std::vector<edm4eic::Track>();
255 m_storage.createRelations(obj);
259#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
Definition MutableTrack.h:40
Definition TrackCollectionData.h:32
TrackObjPointerContainer entries
Definition TrackCollectionData.h:37
Definition TrackCollection.h:91
size_t size_type
Definition TrackCollection.h:97
~TrackCollection() override
Definition TrackCollection.cc:30
uint32_t getID() const final
Definition TrackCollection.h:186
const std::string_view getTypeName() const final
fully qualified type name
Definition TrackCollection.h:140
static constexpr auto dataTypeName
Definition TrackCollection.h:112
podio::CollectionWriteBuffers getBuffers() final
Get the collection buffers for this collection.
Definition TrackCollection.cc:164
TrackCollection & operator=(TrackCollection &&)=default
const std::string_view getDataTypeName() const final
fully qualified type name of stored POD elements - with namespace
Definition TrackCollection.h:144
void setSubsetCollection(bool setSubset=true) final
Definition TrackCollection.cc:63
const_iterator cbegin() const
Definition TrackCollection.h:203
bool empty() const final
Is the collection empty.
Definition TrackCollection.cc:59
iterator end()
Definition TrackCollection.h:206
const std::string_view getValueTypeName() const final
fully qualified type name of elements - with namespace
Definition TrackCollection.h:142
bool setReferences(const podio::ICollectionProvider *collectionProvider) final
Definition TrackCollection.cc:127
static constexpr auto typeName
Definition TrackCollection.h:110
Track at(std::size_t index) const
Returns the const object of given index.
Definition TrackCollection.cc:39
void prepareAfterRead() final
Definition TrackCollection.cc:112
const_iterator begin() const
Definition TrackCollection.h:200
ptrdiff_t difference_type
Definition TrackCollection.h:96
TrackCollection(const TrackCollection &)=delete
void prepareForWrite() const final
Definition TrackCollection.cc:91
void clear() final
Definition TrackCollection.cc:86
void setID(uint32_t ID) final
Definition TrackCollection.h:176
TrackCollection()
Definition TrackCollection.cc:24
MutableTrack create()
Append a new object to the collection, and return this object.
Definition TrackCollection.cc:74
bool isSubsetCollection() const final
Definition TrackCollection.h:148
static constexpr auto valueTypeName
Definition TrackCollection.h:111
podio::SchemaVersionT getSchemaVersion() const final
schema version
Definition TrackCollection.cc:273
void print(std::ostream &os=std::cout, bool flush=true) const final
Print this collection to the passed stream.
Definition TrackCollection.cc:449
std::size_t size() const final
number of elements in the collection
Definition TrackCollection.cc:51
TrackCollection(TrackCollection &&)=default
std::size_t max_size() const final
maximal number of elements in the collection
Definition TrackCollection.cc:55
void push_back(const MutableTrack &object)
Append object to the collection.
Definition TrackCollection.cc:131
bool isValid() const final
Definition TrackCollection.h:190
const_iterator cend() const
Definition TrackCollection.h:212
TrackCollection & operator=(const TrackCollection &)=delete
const_iterator end() const
Definition TrackCollection.h:209
Definition TrackCollection.h:36
TrackCollectionIterator(const TrackCollectionIterator &)=delete
bool operator==(const TrackCollectionIterator &x) const
Definition TrackCollection.h:47
TrackCollectionIterator & operator=(const TrackCollectionIterator &)=delete
TrackCollectionIterator & operator++()
Definition TrackCollection.cc:386
Track * operator->()
Definition TrackCollection.cc:381
Track operator*()
Definition TrackCollection.cc:376
TrackCollectionIterator(size_t index, const TrackObjPointerContainer *collection)
Definition TrackCollection.h:38
bool operator!=(const TrackCollectionIterator &x) const
Definition TrackCollection.h:43
Definition TrackCollection.h:62
TrackMutableCollectionIterator(size_t index, const TrackObjPointerContainer *collection)
Definition TrackCollection.h:64
TrackMutableCollectionIterator & operator=(const TrackMutableCollectionIterator &)=delete
TrackMutableCollectionIterator(const TrackMutableCollectionIterator &)=delete
bool operator==(const TrackMutableCollectionIterator &x) const
Definition TrackCollection.h:73
MutableTrack * operator->()
Definition TrackCollection.cc:398
bool operator!=(const TrackMutableCollectionIterator &x) const
Definition TrackCollection.h:69
TrackMutableCollectionIterator & operator++()
Definition TrackCollection.cc:403
MutableTrack operator*()
Definition TrackCollection.cc:393
std::vector< edm4eic::Measurement2D > * m_measurements
Definition TrackObj.h:41
Definition CalorimeterHit.cc:17
std::ostream & operator<<(std::ostream &o, const CalorimeterHit &value)
Definition CalorimeterHit.cc:103
std::deque< TrackObj * > TrackObjPointerContainer
Definition TrackCollectionData.h:24
Definition CalorimeterHit.h:31