Back to home page

EIC code displayed by LXR

 
 

    


File indexing completed on 2026-08-31 09:08:31

0001 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
0002 
0003 #ifndef EDM4EIC_MutableVertex_H
0004 #define EDM4EIC_MutableVertex_H
0005 
0006 #include "edm4eic/VertexObj.h"
0007 // Make the immutable class available from its mutable version but not vice versa
0008 #include "edm4eic/Vertex.h"
0009 
0010 #include "edm4eic/Cov4f.h"
0011 #include "edm4eic/ReconstructedParticle.h"
0012 #include "edm4hep/Vector4f.h"
0013 #include "podio/RelationRange.h"
0014 #include <cstdint>
0015 #include <vector>
0016 
0017 #include "podio/utilities/MaybeSharedPtr.h"
0018 
0019 #include <cstdint>
0020 
0021 #if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
0022 #include "nlohmann/json_fwd.hpp"
0023 #endif
0024 
0025 // forward declarations
0026 namespace edm4eic {
0027 class VertexCollection;
0028 }
0029 
0030 
0031 namespace edm4eic {
0032 
0033 
0034 /** @class MutableVertex
0035  *  EIC vertex
0036  *  @author: J. Osborn
0037  */
0038 class MutableVertex {
0039 
0040   friend class VertexCollection;
0041   friend class VertexMutableCollectionIterator;
0042   friend class Vertex;
0043 
0044 public:
0045   using object_type = Vertex;
0046   using collection_type = VertexCollection;
0047 
0048   /// default constructor
0049   MutableVertex() = default;
0050 
0051   /// Constructor initializing all members
0052   MutableVertex(const std::int32_t type, const float chi2, const int ndf, const edm4hep::Vector4f& position, const edm4eic::Cov4f& positionError);
0053 
0054   /// copy constructor
0055   MutableVertex(const MutableVertex& other) = default;
0056 
0057   /// copy-assignment operator
0058   MutableVertex& operator=(MutableVertex other) &; // Rebind this to other's internal object
0059   MutableVertex& operator=(MutableVertex other) && = delete; // Prevent rebinding temporary as the changes wouldn't persist
0060 
0061   /// create a mutable deep-copy of the object with identical relations
0062   /// if cloneRelations=false, the relations are not cloned and will be empty
0063   MutableVertex clone(bool cloneRelations=true) const;
0064 
0065   /// destructor
0066   ~MutableVertex() = default;
0067 
0068 
0069 public:
0070 
0071   /// Access the Type flag, to identify what type of vertex it is (e.g. primary, secondary, generated, etc.)
0072   std::int32_t getType() const;
0073 
0074   /// Access the Chi-squared of the vertex fit
0075   float getChi2() const;
0076 
0077   /// Access the NDF of the vertex fit
0078   int getNdf() const;
0079 
0080   /// Access the position [mm] + time t0 [ns] of the vertex. Time is 4th component in vector
0081   const edm4hep::Vector4f& getPosition() const;
0082 
0083   /// Access the Covariance matrix of the position+time. Time is 4th component, similarly to 4vector
0084   const edm4eic::Cov4f& getPositionError() const;
0085 
0086 
0087 
0088   /// Set the Type flag, to identify what type of vertex it is (e.g. primary, secondary, generated, etc.)
0089   void setType(const std::int32_t type);
0090   /// Get mutable reference to Type flag, to identify what type of vertex it is (e.g. primary, secondary, generated, etc.)
0091   std::int32_t& getType();
0092 
0093   /// Set the Chi-squared of the vertex fit
0094   void setChi2(const float chi2);
0095   /// Get mutable reference to Chi-squared of the vertex fit
0096   float& getChi2();
0097 
0098   /// Set the NDF of the vertex fit
0099   void setNdf(const int ndf);
0100   /// Get mutable reference to NDF of the vertex fit
0101   int& getNdf();
0102 
0103   /// Set the position [mm] + time t0 [ns] of the vertex. Time is 4th component in vector
0104   void setPosition(const edm4hep::Vector4f& position);
0105   /// Get mutable reference to position [mm] + time t0 [ns] of the vertex. Time is 4th component in vector
0106   edm4hep::Vector4f& getPosition();
0107 
0108   /// Set the Covariance matrix of the position+time. Time is 4th component, similarly to 4vector
0109   void setPositionError(const edm4eic::Cov4f& positionError);
0110   /// Get mutable reference to Covariance matrix of the position+time. Time is 4th component, similarly to 4vector
0111   edm4eic::Cov4f& getPositionError();
0112 
0113 
0114 
0115   void addToAssociatedParticles(const edm4eic::ReconstructedParticle&);
0116   std::size_t associatedParticles_size() const;
0117   edm4eic::ReconstructedParticle getAssociatedParticles(std::size_t) const;
0118   std::vector<edm4eic::ReconstructedParticle>::const_iterator associatedParticles_begin() const;
0119   std::vector<edm4eic::ReconstructedParticle>::const_iterator associatedParticles_end() const;
0120   podio::RelationRange<edm4eic::ReconstructedParticle> getAssociatedParticles() const;
0121 
0122 
0123 
0124   /// check whether the object is actually available
0125   bool isAvailable() const;
0126   /// disconnect from VertexObj instance
0127   void unlink() { m_obj = podio::utils::MaybeSharedPtr<VertexObj>{nullptr}; }
0128 
0129   constexpr bool operator==(const MutableVertex& other) const { return m_obj == other.m_obj; }
0130   bool operator==(const Vertex& other) const;
0131 
0132   constexpr bool operator!=(const MutableVertex& other) const { return !(*this == other); }
0133   bool operator!=(const Vertex& other) const { return !(*this == other); }
0134 
0135   // less comparison operator, so that objects can be e.g. stored in sets.
0136   bool operator<(const MutableVertex& other) const { return podio::detail::getOrderKey(*this) < podio::detail::getOrderKey(other); }
0137 
0138   podio::ObjectID id() const { return getObjectID(); }
0139 
0140   const podio::ObjectID getObjectID() const;
0141 
0142   friend std::hash<MutableVertex>;
0143 
0144   friend void swap(MutableVertex& a, MutableVertex& b) {
0145     using std::swap;
0146     swap(a.m_obj, b.m_obj); // swap out the internal pointers
0147   }
0148 
0149 private:
0150   /// constructor from existing VertexObj
0151   explicit MutableVertex(podio::utils::MaybeSharedPtr<VertexObj> obj);
0152 
0153   podio::utils::MaybeSharedPtr<VertexObj> m_obj{new VertexObj{}, podio::utils::MarkOwned};
0154 };
0155 
0156 #if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
0157 void to_json(nlohmann::json& j, const MutableVertex& value);
0158 #endif
0159 
0160 
0161 } // namespace edm4eic
0162 
0163 
0164 
0165 template<>
0166 struct std::hash<edm4eic::MutableVertex> {
0167   std::size_t operator()(const edm4eic::MutableVertex& obj) const {
0168     return std::hash<edm4eic::VertexObj*>{}(obj.m_obj.get());
0169   }
0170 };
0171 
0172 
0173 #endif