TruthVertex_v1.h

Go to the documentation of this file.

// -*- C++ -*-
 
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef XAODTRUTH_VERSIONS_TRUTHVERTEX_V1_H
#define XAODTRUTH_VERSIONS_TRUTHVERTEX_V1_H
 
// ROOT include(s):
#include <TLorentzVector.h>
 
// EDM include(s):
#include "AthContainers/AuxElement.h"
#include "AthLinks/ElementLink.h"
 
// xAOD include(s):
#include "xAODBase/ObjectType.h"
 
// Local include(s):
#include "xAODTruth/TruthParticleContainerFwd.h"
 
// ROOT include(s):
#include "Math/Vector4D.h"
 
namespace xAOD {
 
   class TruthVertex_v1 : public SG::AuxElement {
 
   public:
      TruthVertex_v1();
 
 
      void setId( int value );
      int id() const;
 
      int status() const { return id(); }
 
      void setStatus( int value ) { setId(value); }
 
      void setBarcode( int value );
      int barcode() const;
 
 
 
      typedef ElementLink< TruthParticleContainer > TPLink_t;
      typedef std::vector< TPLink_t > TPLinks_t;
 
      const TPLinks_t& incomingParticleLinks() const;
      void setIncomingParticleLinks( const TPLinks_t& links );
      size_t nIncomingParticles() const;
      const TruthParticle_v1* incomingParticle( size_t index ) const;
      void addIncomingParticleLink( const TPLink_t& link );
      void clearIncomingParticleLinks();
 
      const TPLinks_t& outgoingParticleLinks() const;
      void setOutgoingParticleLinks( const TPLinks_t& links );
      size_t nOutgoingParticles() const;
      const TruthParticle_v1* outgoingParticle( size_t index ) const;
      std::vector<const TruthParticle*> particles_in() const;
      std::vector<const TruthParticle*> particles_out() const;
      void addOutgoingParticleLink( const TPLink_t& link );
      void clearOutgoingParticleLinks();
 
 
 
      float x() const;
      void setX( float value );
 
      float y() const;
      void setY( float value );
 
      float z() const;
      void setZ( float value );
 
      float perp() const;
      float eta() const;
      float phi() const;
 
      float t() const;
      void setT( float value );
 
      typedef TLorentzVector FourVec_t;
 
      FourVec_t v4() const;
 
      typedef ROOT::Math::LorentzVector<ROOT::Math::PxPyPzE4D<double> > GenVecFourVec_t;
 
      GenVecFourVec_t genvecV4() const;
 
 
      Type::ObjectType type() const;
 
      void toPersistent();
 
   }; // class TruthVertex_v1
  inline std::ostream& operator<<(std::ostream& os, const TruthVertex_v1* v) {
    if (!v) { os << "Vtx: Empty vertex" << std::endl; return os;}
    os << "Vtx: id=";
    os << v->id() << " status=";
    os << v->status();
    os << " (x,y,z,t)=" << v->x() << "," << v->y() << "," << v->z() << "," << v->t();
    /* os << std::endl; AV:Not clear if we need a new line here */
    return os;
  }
} // namespace xAOD
 
// Declare the inheritance of the type to StoreGate:
#include "xAODCore/BaseInfo.h"
SG_BASE( xAOD::TruthVertex_v1, SG::AuxElement );
 
#endif // XAODTRUTH_VERSIONS_TRUTHVERTEX_V1_H