File indexing completed on 2025-12-27 10:49:06
0001
0002
0003 #ifndef EDM4EIC_TrackSegment_CollectionData_H
0004 #define EDM4EIC_TrackSegment_CollectionData_H
0005
0006
0007 #include "edm4eic/TrackSegmentData.h"
0008 #include "edm4eic/TrackSegmentObj.h"
0009 #include "edm4eic/TrackSegment.h"
0010 #include "edm4eic/Track.h"
0011
0012
0013
0014
0015 #include "podio/CollectionBuffers.h"
0016 #include "podio/ICollectionProvider.h"
0017
0018 #include <deque>
0019 #include <memory>
0020
0021 namespace edm4eic {
0022
0023
0024 using TrackSegmentObjPointerContainer = std::deque<TrackSegmentObj*>;
0025 using TrackSegmentDataContainer = std::vector<TrackSegmentData>;
0026
0027
0028
0029
0030
0031
0032 class TrackSegmentCollectionData {
0033 public:
0034
0035
0036
0037 TrackSegmentObjPointerContainer entries{};
0038
0039
0040
0041
0042 TrackSegmentCollectionData();
0043
0044
0045
0046
0047 TrackSegmentCollectionData(podio::CollectionReadBuffers buffers, bool isSubsetColl);
0048
0049
0050
0051
0052 TrackSegmentCollectionData(const TrackSegmentCollectionData&) = delete;
0053 TrackSegmentCollectionData& operator=(const TrackSegmentCollectionData&) = delete;
0054 TrackSegmentCollectionData(TrackSegmentCollectionData&& other) = default;
0055 TrackSegmentCollectionData& operator=(TrackSegmentCollectionData&& other) = default;
0056
0057
0058
0059
0060 ~TrackSegmentCollectionData() = default;
0061
0062 void clear(bool isSubsetColl);
0063
0064 podio::CollectionWriteBuffers getCollectionBuffers(bool isSubsetColl);
0065
0066 void prepareForWrite(bool isSubsetColl);
0067
0068 void prepareAfterRead(uint32_t collectionID);
0069
0070 void makeSubsetCollection();
0071
0072 void createRelations(TrackSegmentObj* obj);
0073
0074 bool setReferences(const podio::ICollectionProvider* collectionProvider, bool isSubsetColl);
0075
0076 private:
0077
0078 podio::UVecPtr<edm4eic::Track> m_rel_track{std::make_unique<std::vector<edm4eic::Track>>()};
0079
0080
0081 podio::UVecPtr<edm4eic::TrackPoint> m_vec_points{nullptr};
0082 std::vector<podio::UVecPtr<edm4eic::TrackPoint>> m_vecs_points{};
0083
0084
0085 podio::CollRefCollection m_refCollections{};
0086 podio::VectorMembersInfo m_vecmem_info{};
0087 std::unique_ptr<TrackSegmentDataContainer> m_data{nullptr};
0088 };
0089
0090
0091 }
0092
0093
0094 #endif