MutableReconstructedParticle.h

Go to the documentation of this file.

// AUTOMATICALLY GENERATED FILE - DO NOT EDIT
 
#ifndef EICD_MutableReconstructedParticle_H
#define EICD_MutableReconstructedParticle_H
 
#include "eicd/ReconstructedParticleObj.h"
// Make the immutable class available from its mutable version but not vice versa
#include "eicd/ReconstructedParticle.h"
 
#include "eicd/Cluster.h"
#include "eicd/Cov4f.h"
#include "eicd/Track.h"
#include "edm4hep/ParticleID.h"
#include "edm4hep/Vector3f.h"
#include "podio/RelationRange.h"
#include <cstdint>
#include <vector>
#include "podio/ObjectID.h"
#include <ostream>
 
#ifdef PODIO_JSON_OUTPUT
#include "nlohmann/json.hpp"
#endif
 
// forward declarations
namespace eicd {
class Vertex;
class MutableVertex;
}
namespace edm4hep {
class ParticleID;
class MutableParticleID;
}
 
 
namespace eicd {
 
 
/** @class MutableReconstructedParticle
 *  EIC Reconstructed Particle
 *  @author: W. Armstrong, S. Joosten, F. Gaede
 */
class MutableReconstructedParticle {
 
  friend class ReconstructedParticleCollection;
  friend class ReconstructedParticleMutableCollectionIterator;
  friend class ReconstructedParticle;
 
public:
 
  /// default constructor
  MutableReconstructedParticle();
  MutableReconstructedParticle(std::int32_t type, float energy, edm4hep::Vector3f momentum, edm4hep::Vector3f referencePoint, float charge, float mass, float goodnessOfPID, eicd::Cov4f covMatrix, std::int32_t PDG);
 
  /// constructor from existing ReconstructedParticleObj
  MutableReconstructedParticle(ReconstructedParticleObj* obj);
 
  /// copy constructor
  MutableReconstructedParticle(const MutableReconstructedParticle& other);
 
  /// copy-assignment operator
  MutableReconstructedParticle& operator=(MutableReconstructedParticle other);
 
  /// create a mutable deep-copy of the object with identical relations
  MutableReconstructedParticle clone() const;
 
  /// destructor
  ~MutableReconstructedParticle();
 
  /// conversion to const object
  operator ReconstructedParticle() const;
 
public:
 
  /// Access the type of reconstructed particle. Check/set collection parameters ReconstructedParticleTypeNames and ReconstructedParticleTypeValues.
  const std::int32_t& getType() const;
 
  /// Access the [GeV] energy of the reconstructed particle. Four momentum state is not kept consistent internally.
  const float& getEnergy() const;
 
  /// Access the [GeV] particle momentum. Four momentum state is not kept consistent internally.
  const edm4hep::Vector3f& getMomentum() const;
 
  /// Access the [mm] reference, i.e. where the particle has been measured
  const edm4hep::Vector3f& getReferencePoint() const;
 
  /// Access the charge of the reconstructed particle.
  const float& getCharge() const;
 
  /// Access the [GeV] mass of the reconstructed particle, set independently from four vector. Four momentum state is not kept consistent internally.
  const float& getMass() const;
 
  /// Access the overall goodness of the PID on a scale of [0;1]
  const float& getGoodnessOfPID() const;
 
  /// Access the cvariance matrix of the reconstructed particle 4vector (10 parameters).
  const eicd::Cov4f& getCovMatrix() const;
 
  /// Access the PDG code for this particle
  const std::int32_t& getPDG() const;
 
 
  /// Access the Start vertex associated to this particle
  const eicd::Vertex getStartVertex() const;
  /// Access the particle ID used for the kinematics of this particle
  const edm4hep::ParticleID getParticleIDUsed() const;
 
  /// Set the type of reconstructed particle. Check/set collection parameters ReconstructedParticleTypeNames and ReconstructedParticleTypeValues.
  void setType(std::int32_t value);
 
  /// Set the [GeV] energy of the reconstructed particle. Four momentum state is not kept consistent internally.
  void setEnergy(float value);
 
  /// Set the [GeV] particle momentum. Four momentum state is not kept consistent internally.
  void setMomentum(edm4hep::Vector3f value);
  /// Get reference to [GeV] particle momentum. Four momentum state is not kept consistent internally.
  edm4hep::Vector3f& momentum();
 
  /// Set the [mm] reference, i.e. where the particle has been measured
  void setReferencePoint(edm4hep::Vector3f value);
  /// Get reference to [mm] reference, i.e. where the particle has been measured
  edm4hep::Vector3f& referencePoint();
 
  /// Set the charge of the reconstructed particle.
  void setCharge(float value);
 
  /// Set the [GeV] mass of the reconstructed particle, set independently from four vector. Four momentum state is not kept consistent internally.
  void setMass(float value);
 
  /// Set the overall goodness of the PID on a scale of [0;1]
  void setGoodnessOfPID(float value);
 
  /// Set the cvariance matrix of the reconstructed particle 4vector (10 parameters).
  void setCovMatrix(eicd::Cov4f value);
  /// Get reference to cvariance matrix of the reconstructed particle 4vector (10 parameters).
  eicd::Cov4f& covMatrix();
 
  /// Set the PDG code for this particle
  void setPDG(std::int32_t value);
 
 
  /// Set the Start vertex associated to this particle
  void setStartVertex(eicd::Vertex value);
  /// Set the particle ID used for the kinematics of this particle
  void setParticleIDUsed(edm4hep::ParticleID value);
 
  void addToClusters(eicd::Cluster);
  unsigned int clusters_size() const;
  eicd::Cluster getClusters(unsigned int) const;
  std::vector<eicd::Cluster>::const_iterator clusters_begin() const;
  std::vector<eicd::Cluster>::const_iterator clusters_end() const;
  podio::RelationRange<eicd::Cluster> getClusters() const;
  void addToTracks(eicd::Track);
  unsigned int tracks_size() const;
  eicd::Track getTracks(unsigned int) const;
  std::vector<eicd::Track>::const_iterator tracks_begin() const;
  std::vector<eicd::Track>::const_iterator tracks_end() const;
  podio::RelationRange<eicd::Track> getTracks() const;
  void addToParticles(eicd::ReconstructedParticle);
  unsigned int particles_size() const;
  eicd::ReconstructedParticle getParticles(unsigned int) const;
  std::vector<eicd::ReconstructedParticle>::const_iterator particles_begin() const;
  std::vector<eicd::ReconstructedParticle>::const_iterator particles_end() const;
  podio::RelationRange<eicd::ReconstructedParticle> getParticles() const;
  void addToParticleIDs(edm4hep::ParticleID);
  unsigned int particleIDs_size() const;
  edm4hep::ParticleID getParticleIDs(unsigned int) const;
  std::vector<edm4hep::ParticleID>::const_iterator particleIDs_begin() const;
  std::vector<edm4hep::ParticleID>::const_iterator particleIDs_end() const;
  podio::RelationRange<edm4hep::ParticleID> getParticleIDs() const;
 
 bool isCompound() const {return particles_size() > 0;}
 
 
 
  /// check whether the object is actually available
  bool isAvailable() const;
  /// disconnect from ReconstructedParticleObj instance
  void unlink() { m_obj = nullptr; }
 
  bool operator==(const MutableReconstructedParticle& other) const { return m_obj == other.m_obj; }
  bool operator==(const ReconstructedParticle& other) const;
 
  // less comparison operator, so that objects can be e.g. stored in sets.
  bool operator<(const MutableReconstructedParticle& other) const { return m_obj < other.m_obj; }
 
  unsigned int id() const { return getObjectID().collectionID * 10000000 + getObjectID().index; }
 
  const podio::ObjectID getObjectID() const;
 
  friend void swap(MutableReconstructedParticle& a, MutableReconstructedParticle& b) {
    using std::swap;
    swap(a.m_obj, b.m_obj); // swap out the internal pointers
  }
 
private:
  ReconstructedParticleObj* m_obj;
};
 
#ifdef PODIO_JSON_OUTPUT
void to_json(nlohmann::json& j, const MutableReconstructedParticle& value);
#endif
 
 
} // namespace eicd
 
 
#endif