include/edm4hep/MutableParticleID.h

0001 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
0002
0003 #ifndef EDM4HEP_MutableParticleID_H
0004 #define EDM4HEP_MutableParticleID_H
0005
0006 #include "edm4hep/ParticleIDObj.h"
0007 // Make the immutable class available from its mutable version but not vice versa
0008 #include "edm4hep/ParticleID.h"
0009
0010 #include "podio/RelationRange.h"
0011 #include <cstdint>
0012 #include <vector>
0013
0014 #include "podio/utilities/MaybeSharedPtr.h"
0015
0016 #include <cstdint>
0017
0018 #if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
0019 #include "nlohmann/json_fwd.hpp"
0020 #endif
0021
0022 // forward declarations
0023 namespace edm4hep {
0024 class ParticleIDCollection;
0025 class ReconstructedParticle;
0026 class MutableReconstructedParticle;
0027 } // namespace edm4hep
0028
0029 namespace edm4hep {
0030
0031 /** @class MutableParticleID
0032  *  ParticleID
0033  *  @author: EDM4hep authors
0034  */
0035 class MutableParticleID {
0036
0037   friend class ParticleIDCollection;
0038   friend class ParticleIDMutableCollectionIterator;
0039   friend class ParticleID;
0040
0041 public:
0042   using object_type = ParticleID;
0043   using collection_type = ParticleIDCollection;
0044
0045   /// default constructor
0046   MutableParticleID() = default;
0047
0048   /// Constructor initializing all members
0049   MutableParticleID(const std::int32_t type, const std::int32_t PDG, const std::int32_t algorithmType,
0050                     const float likelihood);
0051
0052   /// copy constructor
0053   MutableParticleID(const MutableParticleID& other) = default;
0054
0055   /// copy-assignment operator
0056   MutableParticleID& operator=(MutableParticleID other) &; // Rebind this to other's internal object
0057   MutableParticleID&
0058   operator=(MutableParticleID other) && = delete; // Prevent rebinding temporary as the changes wouldn't persist
0059
0060   /// create a mutable deep-copy of the object with identical relations
0061   /// if cloneRelations=false, the relations are not cloned and will be empty
0062   MutableParticleID clone(bool cloneRelations = true) const;
0063
0064   /// destructor
0065   ~MutableParticleID() = default;
0066
0067 public:
0068   /// Access the userdefined type
0069   std::int32_t getType() const;
0070
0071   /// Access the PDG code of this id - ( 999999 ) if unknown
0072   std::int32_t getPDG() const;
0073
0074   /// Access the type of the algorithm/module that created this hypothesis
0075   std::int32_t getAlgorithmType() const;
0076
0077   /// Access the likelihood of this hypothesis - in a user defined normalization
0078   float getLikelihood() const;
0079
0080   /// Access the the particle from which this PID has been computed
0081   const edm4hep::ReconstructedParticle getParticle() const;
0082
0083   /// Set the userdefined type
0084   void setType(const std::int32_t type);
0085   /// Get mutable reference to userdefined type
0086   std::int32_t& getType();
0087
0088   /// Set the PDG code of this id - ( 999999 ) if unknown
0089   void setPDG(const std::int32_t PDG);
0090   /// Get mutable reference to PDG code of this id - ( 999999 ) if unknown
0091   std::int32_t& getPDG();
0092
0093   /// Set the type of the algorithm/module that created this hypothesis
0094   void setAlgorithmType(const std::int32_t algorithmType);
0095   /// Get mutable reference to type of the algorithm/module that created this hypothesis
0096   std::int32_t& getAlgorithmType();
0097
0098   /// Set the likelihood of this hypothesis - in a user defined normalization
0099   void setLikelihood(const float likelihood);
0100   /// Get mutable reference to likelihood of this hypothesis - in a user defined normalization
0101   float& getLikelihood();
0102
0103   /// Set the the particle from which this PID has been computed
0104   void setParticle(const edm4hep::ReconstructedParticle& value);
0105
0106   void addToParameters(const float&);
0107   std::size_t parameters_size() const;
0108   float getParameters(std::size_t) const;
0109   std::vector<float>::const_iterator parameters_begin() const;
0110   std::vector<float>::const_iterator parameters_end() const;
0111   podio::RelationRange<float> getParameters() const;
0112
0113   /// check whether the object is actually available
0114   bool isAvailable() const;
0115   /// disconnect from ParticleIDObj instance
0116   void unlink() { m_obj = podio::utils::MaybeSharedPtr<ParticleIDObj>{nullptr}; }
0117
0118   constexpr bool operator==(const MutableParticleID& other) const { return m_obj == other.m_obj; }
0119   bool operator==(const ParticleID& other) const;
0120
0121   constexpr bool operator!=(const MutableParticleID& other) const { return !(*this == other); }
0122   bool operator!=(const ParticleID& other) const { return !(*this == other); }
0123
0124   // less comparison operator, so that objects can be e.g. stored in sets.
0125   bool operator<(const MutableParticleID& other) const {
0126     return podio::detail::getOrderKey(*this) < podio::detail::getOrderKey(other);
0127   }
0128
0129   podio::ObjectID id() const { return getObjectID(); }
0130
0131   const podio::ObjectID getObjectID() const;
0132
0133   friend std::hash<MutableParticleID>;
0134
0135   friend void swap(MutableParticleID& a, MutableParticleID& b) {
0136     using std::swap;
0137     swap(a.m_obj, b.m_obj); // swap out the internal pointers
0138   }
0139
0140 private:
0141   /// constructor from existing ParticleIDObj
0142   explicit MutableParticleID(podio::utils::MaybeSharedPtr<ParticleIDObj> obj);
0143
0144   podio::utils::MaybeSharedPtr<ParticleIDObj> m_obj{new ParticleIDObj{}, podio::utils::MarkOwned};
0145 };
0146
0147 #if defined(PODIO_JSON_OUTPUT) && !defined(__CLING__)
0148 void to_json(nlohmann::json& j, const MutableParticleID& value);
0149 #endif
0150
0151 } // namespace edm4hep
0152
0153 template <>
0154 struct std::hash<edm4hep::MutableParticleID> {
0155   std::size_t operator()(const edm4hep::MutableParticleID& obj) const {
0156     return std::hash<edm4hep::ParticleIDObj*>{}(obj.m_obj.get());
0157   }
0158 };
0159
0160 #endif