File indexing completed on 2025-10-13 09:14:39
0001
0002
0003 #ifndef EDM4EIC_RawHGCROCHitCollection_H
0004 #define EDM4EIC_RawHGCROCHitCollection_H
0005
0006
0007 #include "edm4eic/RawHGCROCHit.h"
0008 #include "edm4eic/MutableRawHGCROCHit.h"
0009 #include "edm4eic/RawHGCROCHitObj.h"
0010 #include "edm4eic/RawHGCROCHitCollectionData.h"
0011
0012
0013 #include "podio/ICollectionProvider.h"
0014 #include "podio/CollectionBase.h"
0015
0016 #if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
0017 #include "nlohmann/json_fwd.hpp"
0018 #endif
0019
0020 #include <string_view>
0021 #include <vector>
0022 #include <algorithm>
0023 #include <ostream>
0024 #include <mutex>
0025 #include <memory>
0026 #include <cstddef>
0027
0028 namespace podio {
0029 struct RelationNames;
0030 }
0031
0032 namespace edm4eic {
0033
0034
0035
0036 class RawHGCROCHitCollectionIterator {
0037 public:
0038 using value_type = RawHGCROCHit;
0039 using difference_type = ptrdiff_t;
0040 using reference = RawHGCROCHit;
0041 using pointer = RawHGCROCHit*;
0042 using iterator_category = std::input_iterator_tag;
0043
0044
0045 using iterator_concept = std::random_access_iterator_tag;
0046
0047 RawHGCROCHitCollectionIterator(size_t index, const RawHGCROCHitObjPointerContainer* collection) : m_index(index), m_object(podio::utils::MaybeSharedPtr<RawHGCROCHitObj>{nullptr}), m_collection(collection) {}
0048 RawHGCROCHitCollectionIterator() = default;
0049
0050 RawHGCROCHitCollectionIterator(const RawHGCROCHitCollectionIterator&) = default;
0051 RawHGCROCHitCollectionIterator(RawHGCROCHitCollectionIterator&&) = default;
0052 RawHGCROCHitCollectionIterator& operator=(const RawHGCROCHitCollectionIterator&) = default;
0053 RawHGCROCHitCollectionIterator& operator=(RawHGCROCHitCollectionIterator&&) = default;
0054 ~RawHGCROCHitCollectionIterator() = default;
0055
0056 auto operator<=>(const RawHGCROCHitCollectionIterator& other) const {
0057 return m_index <=> other.m_index;
0058 }
0059
0060 bool operator==(const RawHGCROCHitCollectionIterator& x) const {
0061 return m_index == x.m_index;
0062 }
0063
0064 reference operator*() const;
0065 pointer operator->();
0066 RawHGCROCHitCollectionIterator& operator++();
0067 RawHGCROCHitCollectionIterator operator++(int);
0068 RawHGCROCHitCollectionIterator& operator--();
0069 RawHGCROCHitCollectionIterator operator--(int);
0070 RawHGCROCHitCollectionIterator& operator+=(difference_type n);
0071 RawHGCROCHitCollectionIterator operator+(difference_type n) const;
0072 friend RawHGCROCHitCollectionIterator operator+(difference_type n, const RawHGCROCHitCollectionIterator& it);
0073 RawHGCROCHitCollectionIterator& operator-=(difference_type n);
0074 RawHGCROCHitCollectionIterator operator-(difference_type n) const;
0075 reference operator[](difference_type n) const;
0076 difference_type operator-(const RawHGCROCHitCollectionIterator& other) const;
0077
0078 private:
0079 size_t m_index{0};
0080 RawHGCROCHit m_object { podio::utils::MaybeSharedPtr<RawHGCROCHitObj>{nullptr} };
0081 const RawHGCROCHitObjPointerContainer* m_collection{nullptr};
0082 };
0083
0084
0085 class RawHGCROCHitMutableCollectionIterator {
0086 public:
0087 using value_type = RawHGCROCHit;
0088 using difference_type = ptrdiff_t;
0089 using reference = MutableRawHGCROCHit;
0090 using pointer = MutableRawHGCROCHit*;
0091 using iterator_category = std::input_iterator_tag;
0092
0093
0094 using iterator_concept = std::random_access_iterator_tag;
0095
0096 RawHGCROCHitMutableCollectionIterator(size_t index, const RawHGCROCHitObjPointerContainer* collection) : m_index(index), m_object(podio::utils::MaybeSharedPtr<RawHGCROCHitObj>{nullptr}), m_collection(collection) {}
0097 RawHGCROCHitMutableCollectionIterator() = default;
0098
0099 RawHGCROCHitMutableCollectionIterator(const RawHGCROCHitMutableCollectionIterator&) = default;
0100 RawHGCROCHitMutableCollectionIterator(RawHGCROCHitMutableCollectionIterator&&) = default;
0101 RawHGCROCHitMutableCollectionIterator& operator=(const RawHGCROCHitMutableCollectionIterator&) = default;
0102 RawHGCROCHitMutableCollectionIterator& operator=(RawHGCROCHitMutableCollectionIterator&&) = default;
0103 ~RawHGCROCHitMutableCollectionIterator() = default;
0104
0105 auto operator<=>(const RawHGCROCHitMutableCollectionIterator& other) const {
0106 return m_index <=> other.m_index;
0107 }
0108
0109 bool operator==(const RawHGCROCHitMutableCollectionIterator& x) const {
0110 return m_index == x.m_index;
0111 }
0112
0113 reference operator*() const;
0114 pointer operator->();
0115 RawHGCROCHitMutableCollectionIterator& operator++();
0116 RawHGCROCHitMutableCollectionIterator operator++(int);
0117 RawHGCROCHitMutableCollectionIterator& operator--();
0118 RawHGCROCHitMutableCollectionIterator operator--(int);
0119 RawHGCROCHitMutableCollectionIterator& operator+=(difference_type n);
0120 RawHGCROCHitMutableCollectionIterator operator+(difference_type n) const;
0121 friend RawHGCROCHitMutableCollectionIterator operator+(difference_type n, const RawHGCROCHitMutableCollectionIterator& it);
0122 RawHGCROCHitMutableCollectionIterator& operator-=(difference_type n);
0123 RawHGCROCHitMutableCollectionIterator operator-(difference_type n) const;
0124 reference operator[](difference_type n) const;
0125 difference_type operator-(const RawHGCROCHitMutableCollectionIterator& other) const;
0126
0127 private:
0128 size_t m_index{0};
0129 MutableRawHGCROCHit m_object { podio::utils::MaybeSharedPtr<RawHGCROCHitObj>{nullptr} };
0130 const RawHGCROCHitObjPointerContainer* m_collection{nullptr};
0131 };
0132
0133
0134
0135
0136
0137 class RawHGCROCHitCollection : public podio::CollectionBase {
0138 public:
0139 using value_type = RawHGCROCHit;
0140 using mutable_type = MutableRawHGCROCHit;
0141 using const_iterator = RawHGCROCHitCollectionIterator;
0142 using iterator = RawHGCROCHitMutableCollectionIterator;
0143 using difference_type = ptrdiff_t;
0144 using size_type = size_t;
0145 using const_reverse_iterator = std::reverse_iterator<const_iterator>;
0146 using reverse_iterator = std::reverse_iterator<iterator>;
0147
0148 RawHGCROCHitCollection();
0149 RawHGCROCHitCollection(RawHGCROCHitCollectionData&& data, bool isSubsetColl);
0150
0151 RawHGCROCHitCollection(const RawHGCROCHitCollection& ) = delete;
0152 RawHGCROCHitCollection& operator=(const RawHGCROCHitCollection& ) = delete;
0153 RawHGCROCHitCollection(RawHGCROCHitCollection&&) = default;
0154 RawHGCROCHitCollection& operator=(RawHGCROCHitCollection&&) = default;
0155
0156
0157 ~RawHGCROCHitCollection() override;
0158
0159 constexpr static std::string_view typeName = "edm4eic::RawHGCROCHitCollection";
0160 constexpr static std::string_view valueTypeName = "edm4eic::RawHGCROCHit";
0161 constexpr static std::string_view dataTypeName = "edm4eic::RawHGCROCHitData";
0162
0163 void clear() final;
0164
0165
0166 void print(std::ostream& os=std::cout, bool flush=true) const final;
0167
0168
0169 RawHGCROCHitCollection* operator->() { return static_cast<RawHGCROCHitCollection*>(this); }
0170
0171
0172 MutableRawHGCROCHit create();
0173
0174
0175
0176 template<typename... Args>
0177 MutableRawHGCROCHit create(Args&&... args);
0178
0179
0180 std::size_t size() const final;
0181
0182
0183 std::size_t max_size() const final;
0184
0185
0186 bool empty() const final;
0187
0188
0189 const std::string_view getTypeName() const final { return typeName; }
0190
0191 const std::string_view getValueTypeName() const final { return valueTypeName; }
0192
0193 const std::string_view getDataTypeName() const final { return dataTypeName; }
0194
0195 podio::SchemaVersionT getSchemaVersion() const final;
0196
0197 bool isSubsetCollection() const final {
0198 return m_isSubsetColl;
0199 }
0200
0201 void setSubsetCollection(bool setSubset=true) final;
0202
0203
0204 RawHGCROCHit operator[](std::size_t index) const;
0205
0206 MutableRawHGCROCHit operator[](std::size_t index);
0207
0208 RawHGCROCHit at(std::size_t index) const;
0209
0210 MutableRawHGCROCHit at(std::size_t index);
0211
0212
0213
0214 void push_back(const MutableRawHGCROCHit& object);
0215
0216 void push_back(const RawHGCROCHit& object);
0217
0218 void prepareForWrite() const final;
0219 void prepareAfterRead() final;
0220 bool setReferences(const podio::ICollectionProvider* collectionProvider) final;
0221
0222
0223 podio::CollectionWriteBuffers getBuffers() final;
0224
0225 void setID(uint32_t ID) final {
0226 m_collectionID = ID;
0227 if (!m_isSubsetColl) {
0228 std::for_each(m_storage.entries.begin(), m_storage.entries.end(),
0229 [ID] (RawHGCROCHitObj* obj) { obj->id = {obj->id.index, static_cast<uint32_t>(ID)}; }
0230 );
0231 }
0232 m_isValid = true;
0233 }
0234
0235 uint32_t getID() const final {
0236 return m_collectionID;
0237 }
0238
0239 bool isValid() const final {
0240 return m_isValid;
0241 }
0242
0243 size_t getDatamodelRegistryIndex() const final;
0244
0245
0246 iterator begin() {
0247 return iterator(0, &m_storage.entries);
0248 }
0249 const_iterator begin() const {
0250 return const_iterator(0, &m_storage.entries);
0251 }
0252 const_iterator cbegin() const {
0253 return begin();
0254 }
0255 iterator end() {
0256 return iterator(m_storage.entries.size(), &m_storage.entries);
0257 }
0258 const_iterator end() const {
0259 return const_iterator(m_storage.entries.size(), &m_storage.entries);
0260 }
0261 const_iterator cend() const {
0262 return end();
0263 }
0264
0265 reverse_iterator rbegin() {
0266 return reverse_iterator(end());
0267 }
0268 const_reverse_iterator rbegin() const {
0269 return const_reverse_iterator(end());
0270 }
0271 const_reverse_iterator crbegin() const {
0272 return rbegin();
0273 }
0274 reverse_iterator rend() {
0275 return reverse_iterator(begin());
0276 }
0277 const_reverse_iterator rend() const {
0278 return const_reverse_iterator(begin());
0279 }
0280 const_reverse_iterator crend() const {
0281 return rend();
0282 }
0283
0284
0285 std::vector<std::uint64_t> cellID(const size_t nElem = 0) const;
0286 std::vector<std::int32_t> samplePhase(const size_t nElem = 0) const;
0287 std::vector<std::int32_t> timeStamp(const size_t nElem = 0) const;
0288
0289 private:
0290
0291
0292
0293 friend class RawHGCROCHitCollectionData;
0294
0295 bool m_isValid{false};
0296 mutable bool m_isPrepared{false};
0297 bool m_isSubsetColl{false};
0298 uint32_t m_collectionID{0};
0299 mutable std::unique_ptr<std::mutex> m_storageMtx{nullptr};
0300 mutable RawHGCROCHitCollectionData m_storage{};
0301 };
0302
0303 std::ostream& operator<<(std::ostream& o, const RawHGCROCHitCollection& v);
0304
0305 template<typename... Args>
0306 MutableRawHGCROCHit RawHGCROCHitCollection::create(Args&&... args) {
0307 if (m_isSubsetColl) {
0308 throw std::logic_error("Cannot create new elements on a subset collection");
0309 }
0310 const int size = m_storage.entries.size();
0311 auto obj = new RawHGCROCHitObj({size, m_collectionID}, {std::forward<Args>(args)...});
0312 m_storage.entries.push_back(obj);
0313
0314
0315 obj->m_samples = new std::vector<edm4eic::HGCROCSample>();
0316 m_storage.createRelations(obj);
0317 return MutableRawHGCROCHit(podio::utils::MaybeSharedPtr(obj));
0318 }
0319
0320 #if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
0321 void to_json(nlohmann::json& j, const RawHGCROCHitCollection& collection);
0322 #endif
0323
0324 }
0325
0326
0327 #endif