3#ifndef EDM4EIC_TensorCollection_H
4#define EDM4EIC_TensorCollection_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::TensorCollection";
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 Tensor operator[](std::
size_t index) const;
177 if (!m_isSubsetColl) {
179 [ID] (
TensorObj* 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> elementType(
const size_t nElem = 0)
const;
217 bool m_isValid{
false};
218 mutable bool m_isPrepared{
false};
219 bool m_isSubsetColl{
false};
220 uint32_t m_collectionID{0};
221 mutable std::unique_ptr<std::mutex> m_storageMtx{
nullptr};
222 mutable TensorCollectionData m_storage{};
225std::ostream&
operator<<(std::ostream& o,
const TensorCollection& v);
227template<
typename... Args>
229 if (m_isSubsetColl) {
230 throw std::logic_error(
"Cannot create new elements on a subset collection");
232 const int size = m_storage.entries.size();
233 auto obj =
new TensorObj({size, m_collectionID}, {std::forward<Args>(args)...});
234 m_storage.entries.push_back(obj);
237 obj->
m_shape =
new std::vector<std::int64_t>();
238 obj->m_floatData =
new std::vector<float>();
239 obj->m_int64Data =
new std::vector<std::int64_t>();
240 m_storage.createRelations(obj);
244#if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
Definition MutableTensor.h:32
Definition TensorCollectionData.h:30
TensorObjPointerContainer entries
Definition TensorCollectionData.h:35
Definition TensorCollection.h:95
TensorCollection & operator=(TensorCollection &&)=default
static constexpr auto typeName
Definition TensorCollection.h:112
TensorCollection()
Definition TensorCollection.cc:23
bool isSubsetCollection() const final
Definition TensorCollection.h:147
static constexpr auto dataTypeName
Definition TensorCollection.h:114
void prepareAfterRead() final
Definition TensorCollection.cc:107
uint32_t getID() const final
Definition TensorCollection.h:185
MutableTensor create()
Append a new object to the collection, and return this object.
Definition TensorCollection.cc:69
const std::string_view getTypeName() const final
fully qualified type name
Definition TensorCollection.h:139
const_iterator end() const
Definition TensorCollection.h:205
const std::string_view getValueTypeName() const final
fully qualified type name of elements - with namespace
Definition TensorCollection.h:141
~TensorCollection()
Definition TensorCollection.cc:29
void setID(uint32_t ID) final
Definition TensorCollection.h:175
void push_back(MutableTensor object)
Append object to the collection.
Definition TensorCollection.cc:126
void clear() final
Definition TensorCollection.cc:81
iterator end()
Definition TensorCollection.h:202
TensorCollection(TensorCollection &&)=default
std::size_t size() const final
number of elements in the collection
Definition TensorCollection.cc:50
TensorCollection & operator=(const TensorCollection &)=delete
podio::CollectionWriteBuffers getBuffers() final
Get the collection buffers for this collection.
Definition TensorCollection.cc:159
bool isValid() const final
Definition TensorCollection.h:189
static constexpr auto valueTypeName
Definition TensorCollection.h:113
podio::SchemaVersionT getSchemaVersion() const final
schema version
Definition TensorCollection.cc:178
bool setReferences(const podio::ICollectionProvider *collectionProvider) final
Definition TensorCollection.cc:122
TensorCollection(const TensorCollection &)=delete
void prepareForWrite() const final
Definition TensorCollection.cc:86
const std::string_view getDataTypeName() const final
fully qualified type name of stored POD elements - with namespace
Definition TensorCollection.h:143
const_iterator begin() const
Definition TensorCollection.h:199
void setSubsetCollection(bool setSubset=true) final
Definition TensorCollection.cc:58
Tensor at(std::size_t index) const
Returns the const object of given index.
Definition TensorCollection.cc:38
void print(std::ostream &os=std::cout, bool flush=true) const final
Print this collection to the passed stream.
Definition TensorCollection.cc:352
bool empty() const final
Is the collection empty.
Definition TensorCollection.cc:54
Definition TensorCollection.h:40
Tensor * operator->()
Definition TensorCollection.cc:290
bool operator!=(const TensorCollectionIterator &x) const
Definition TensorCollection.h:47
TensorCollectionIterator(const TensorCollectionIterator &)=delete
TensorCollectionIterator(size_t index, const TensorObjPointerContainer *collection)
Definition TensorCollection.h:42
TensorCollectionIterator & operator++()
Definition TensorCollection.cc:295
bool operator==(const TensorCollectionIterator &x) const
Definition TensorCollection.h:51
Tensor operator*()
Definition TensorCollection.cc:285
TensorCollectionIterator & operator=(const TensorCollectionIterator &)=delete
Definition TensorCollection.h:66
TensorMutableCollectionIterator & operator++()
Definition TensorCollection.cc:312
TensorMutableCollectionIterator & operator=(const TensorMutableCollectionIterator &)=delete
TensorMutableCollectionIterator(size_t index, const TensorObjPointerContainer *collection)
Definition TensorCollection.h:68
bool operator!=(const TensorMutableCollectionIterator &x) const
Definition TensorCollection.h:73
MutableTensor operator*()
Definition TensorCollection.cc:302
TensorMutableCollectionIterator(const TensorMutableCollectionIterator &)=delete
MutableTensor * operator->()
Definition TensorCollection.cc:307
bool operator==(const TensorMutableCollectionIterator &x) const
Definition TensorCollection.h:77
Definition TensorObj.h:18
std::vector< std::int64_t > * m_shape
Definition TensorObj.h:34
Definition CalorimeterHit.cc:18
std::deque< TensorObj * > TensorObjPointerContainer
Definition TensorCollectionData.h:22
std::ostream & operator<<(std::ostream &o, const CalorimeterHit &value)
Definition CalorimeterHit.cc:97
Definition CalorimeterHitCollection.h:32