PhysicsValidationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ISF_FATRASTOOLS_PHYSICSVALIDATIONTOOL_H
#define ISF_FATRASTOOLS_PHYSICSVALIDATIONTOOL_H
 
// Athena Base
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ITHistSvc.h"
#include "AthenaKernel/IAtRndmGenSvc.h"
#include "CxxUtils/checker_macros.h"
 
// iFatras
#include "ISF_FatrasInterfaces/IPhysicsValidationTool.h"
 
// Tracking
#include "TrkEventPrimitives/PdgToParticleHypothesis.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
 
#ifndef MAXCHILDREN
#define MAXCHILDREN 40
#endif
 
// ROOT forward declarations
class TTree;
 
namespace ISF {
    class ISFParticle;
}
 
namespace iFatras 
{  
  
  class ATLAS_NOT_THREAD_SAFE PhysicsValidationTool : public extends<AthAlgTool, IPhysicsValidationTool>  // deprecated: ATLASSIM-6020
  {
  public:
    PhysicsValidationTool(const std::string&,const std::string&,const IInterface*);
    
    virtual ~PhysicsValidationTool ();
    
    virtual StatusCode initialize() override;

    virtual
    void saveISFParticleInfo(const ISF::ISFParticle& isp, int endProcess, const Trk::TrackParameters* ePar, double time, double dX0 ) const override;
    
    virtual
    void saveISFParticleInfo(const ISF::ISFParticle& isp, const Trk::ExtrapolationCell<Trk::TrackParameters>& ec,
                     Trk::ExtrapolationCode ecode ) const override;

    virtual
    void saveISFParticleInfo(const ISF::ISFParticle& isp, const Trk::ExtrapolationCell<Trk::NeutralParameters>& ec,
                     Trk::ExtrapolationCode ecode ) const override;
 
 
    virtual
    void saveISFVertexInfo(int process,Amg::Vector3D vertex,const ISF::ISFParticle& isp,Amg::Vector3D primIn,
               Amg::Vector3D* primOut, const ISF::ISFParticleVector& children) const override;
 
    
  private:
     template <class T> StatusCode retrieveTool(ToolHandle<T>& thandle){
       if (!thandle.empty() && thandle.retrieve().isFailure()){
         ATH_MSG_FATAL( "[ fatras setup ] Cannot retrieve " << thandle << ". Abort.");
         return StatusCode::FAILURE;
       }
       else {
         ATH_MSG_DEBUG("[ fatras setup ] Successfully retrieved " << thandle);
       }
       return StatusCode::SUCCESS;
     }
     
     void saveInfo(const ISF::ISFParticle& isp) const; 
      
    /*---------------------------------------------------------------------
     *  Private members
     *---------------------------------------------------------------------*/
 
    ServiceHandle<ITHistSvc>                                     m_thistSvc;         
    std::string                                                  m_validationStream; 
 
    // FIXME mutable variables are not thread-safe.
    TTree                                                                *m_particles = nullptr;    
    mutable int                                                           m_pdg = 0;
    mutable int                                                           m_scIn = 0;  
    mutable int                                                           m_scEnd = 0;
    mutable int                                                           m_gen = 0;
    mutable int                                                           m_geoID = 0;  
    mutable float                                                         m_pth = 0.0F;
    mutable float                                                         m_pph = 0.0F;
    mutable float                                                         m_p = 0.0F;
    mutable float                                                         m_eloss;
    mutable float                                                         m_ionloss;
    mutable float                                                         m_radloss;
    mutable float                                                         m_zOaTr;
    mutable float                                                         m_wZ;
    mutable float                                                         m_thIn;
    mutable float                                                         m_phIn;
    mutable float                                                         m_dIn;
    mutable float                                                         m_thEnd;
    mutable float                                                         m_phEnd;
    mutable float                                                         m_dEnd;
    mutable float                                                         m_X0;
    mutable float                                                         m_dt;
 
    TTree                                                        *m_interactions;   
    mutable int                                                   m_process;
    mutable int                                                   m_pdg_mother;
    mutable int                                                   m_gen_mother;
    mutable int                                                   m_nChild;
    mutable  float                                                m_vtx_dist;
    mutable float                                                 m_vtx_theta;
    mutable float                                                 m_vtx_phi;
    mutable float                                                 m_vtx_e_diff;
    mutable float                                                 m_vtx_p_diff;
    mutable float                                                 m_vtx_plong_diff;
    mutable float                                                 m_vtx_pperp_diff;
    mutable float                                                 m_p_mother;
    mutable int                                                   m_pdg_child[MAXCHILDREN]{};     // decay product pdg code
    mutable float                                                 m_fp_child[MAXCHILDREN]{};      // fraction of incoming momentum 
    mutable float                                                 m_oa_child[MAXCHILDREN]{};      // opening angle wrt the mother
        
  }; 
}
 
#endif // FATRASTOOLS_TRANSPORTTOOL_H