3#ifndef EDM4EIC_TrajectoryCollection_H
4#define EDM4EIC_TrajectoryCollection_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::TrajectoryCollection";
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 Trajectory operator[](std::
size_t index) const;
177 if (!m_isSubsetColl) {
179 [ID] (
TrajectoryObj* 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::uint32_t> type(
const size_t nElem = 0)
const;
210 std::vector<std::uint32_t> nStates(
const size_t nElem = 0)
const;
211 std::vector<std::uint32_t> nMeasurements(
const size_t nElem = 0)
const;
212 std::vector<std::uint32_t> nOutliers(
const size_t nElem = 0)
const;
213 std::vector<std::uint32_t> nHoles(
const size_t nElem = 0)
const;
214 std::vector<std::uint32_t> nSharedHits(
const size_t nElem = 0)
const;
222 bool m_isValid{
false};
223 mutable bool m_isPrepared{
false};
224 bool m_isSubsetColl{
false};
225 uint32_t m_collectionID{0};
226 mutable std::unique_ptr<std::mutex> m_storageMtx{
nullptr};
227 mutable TrajectoryCollectionData m_storage{};
230std::ostream&
operator<<(std::ostream& o,
const TrajectoryCollection& v);
232template<
typename... Args>
234 if (m_isSubsetColl) {
235 throw std::logic_error(
"Cannot create new elements on a subset collection");
237 const int size = m_storage.entries.size();
238 auto obj =
new TrajectoryObj({size, m_collectionID}, {std::forward<Args>(args)...});
239 m_storage.entries.push_back(obj);
243 obj->m_measurements_deprecated =
new std::vector<edm4eic::Measurement2D>();
244 obj->m_outliers_deprecated =
new std::vector<edm4eic::Measurement2D>();
245 obj->m_measurementChi2 =
new std::vector<float>();
246 obj->m_outlierChi2 =
new std::vector<float>();
247 m_storage.createRelations(obj);
251#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
Definition MutableTrajectory.h:39
Definition TrajectoryCollectionData.h:33
TrajectoryObjPointerContainer entries
Definition TrajectoryCollectionData.h:38
Definition TrajectoryCollection.h:95
TrajectoryCollection & operator=(TrajectoryCollection &&)=default
TrajectoryCollection & operator=(const TrajectoryCollection &)=delete
bool isSubsetCollection() const final
Definition TrajectoryCollection.h:147
podio::SchemaVersionT getSchemaVersion() const final
schema version
Definition TrajectoryCollection.cc:231
void push_back(MutableTrajectory object)
Append object to the collection.
Definition TrajectoryCollection.cc:129
bool isValid() const final
Definition TrajectoryCollection.h:189
static constexpr auto dataTypeName
Definition TrajectoryCollection.h:114
static constexpr auto typeName
Definition TrajectoryCollection.h:112
void prepareAfterRead() final
Definition TrajectoryCollection.cc:110
TrajectoryCollection(const TrajectoryCollection &)=delete
MutableTrajectory create()
Append a new object to the collection, and return this object.
Definition TrajectoryCollection.cc:72
static constexpr auto valueTypeName
Definition TrajectoryCollection.h:113
bool empty() const final
Is the collection empty.
Definition TrajectoryCollection.cc:57
void setSubsetCollection(bool setSubset=true) final
Definition TrajectoryCollection.cc:61
const_iterator begin() const
Definition TrajectoryCollection.h:199
void print(std::ostream &os=std::cout, bool flush=true) const final
Print this collection to the passed stream.
Definition TrajectoryCollection.cc:419
std::size_t size() const final
number of elements in the collection
Definition TrajectoryCollection.cc:53
const_iterator end() const
Definition TrajectoryCollection.h:205
iterator end()
Definition TrajectoryCollection.h:202
~TrajectoryCollection()
Definition TrajectoryCollection.cc:32
uint32_t getID() const final
Definition TrajectoryCollection.h:185
TrajectoryCollection(TrajectoryCollection &&)=default
bool setReferences(const podio::ICollectionProvider *collectionProvider) final
Definition TrajectoryCollection.cc:125
void prepareForWrite() const final
Definition TrajectoryCollection.cc:89
const std::string_view getValueTypeName() const final
fully qualified type name of elements - with namespace
Definition TrajectoryCollection.h:141
const std::string_view getTypeName() const final
fully qualified type name
Definition TrajectoryCollection.h:139
TrajectoryCollection()
Definition TrajectoryCollection.cc:26
void clear() final
Definition TrajectoryCollection.cc:84
void setID(uint32_t ID) final
Definition TrajectoryCollection.h:175
Trajectory at(std::size_t index) const
Returns the const object of given index.
Definition TrajectoryCollection.cc:41
podio::CollectionWriteBuffers getBuffers() final
Get the collection buffers for this collection.
Definition TrajectoryCollection.cc:162
const std::string_view getDataTypeName() const final
fully qualified type name of stored POD elements - with namespace
Definition TrajectoryCollection.h:143
Definition TrajectoryCollection.h:40
TrajectoryCollectionIterator(size_t index, const TrajectoryObjPointerContainer *collection)
Definition TrajectoryCollection.h:42
bool operator==(const TrajectoryCollectionIterator &x) const
Definition TrajectoryCollection.h:51
bool operator!=(const TrajectoryCollectionIterator &x) const
Definition TrajectoryCollection.h:47
TrajectoryCollectionIterator & operator=(const TrajectoryCollectionIterator &)=delete
TrajectoryCollectionIterator & operator++()
Definition TrajectoryCollection.cc:345
Trajectory operator*()
Definition TrajectoryCollection.cc:335
Trajectory * operator->()
Definition TrajectoryCollection.cc:340
TrajectoryCollectionIterator(const TrajectoryCollectionIterator &)=delete
Definition Trajectory.h:40
Definition TrajectoryCollection.h:66
bool operator==(const TrajectoryMutableCollectionIterator &x) const
Definition TrajectoryCollection.h:77
MutableTrajectory * operator->()
Definition TrajectoryCollection.cc:357
bool operator!=(const TrajectoryMutableCollectionIterator &x) const
Definition TrajectoryCollection.h:73
TrajectoryMutableCollectionIterator(size_t index, const TrajectoryObjPointerContainer *collection)
Definition TrajectoryCollection.h:68
MutableTrajectory operator*()
Definition TrajectoryCollection.cc:352
TrajectoryMutableCollectionIterator & operator++()
Definition TrajectoryCollection.cc:362
TrajectoryMutableCollectionIterator & operator=(const TrajectoryMutableCollectionIterator &)=delete
TrajectoryMutableCollectionIterator(const TrajectoryMutableCollectionIterator &)=delete
Definition TrajectoryObj.h:24
std::vector< edm4eic::TrackParameters > * m_trackParameters
Definition TrajectoryObj.h:41
Definition CalorimeterHit.cc:18
std::deque< TrajectoryObj * > TrajectoryObjPointerContainer
Definition TrajectoryCollectionData.h:25
std::ostream & operator<<(std::ostream &o, const CalorimeterHit &value)
Definition CalorimeterHit.cc:97
Definition CalorimeterHitCollection.h:32