EnergyLossExtrapolationValidation.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// EnergyLossValidation.h, (c) ATLAS Detector software
 
#ifndef TRKEXALGS_ENERGYLOSSEXTRAPOLATIONVALIDATION_H
#define TRKEXALGS_ENERGYLOSSEXTRAPOLATIONVALIDATION_H
 
// Gaudi includes
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/IRndmGenSvc.h"
#include "GaudiKernel/RndmGenerators.h"
#include <string>
#include "AthContainers/DataVector.h"
// CylinderSurfaces
#include "TrkSurfaces/CylinderSurface.h"
#include "TrkSurfaces/DiscSurface.h"
//
#include <memory>
class TTree;
 
#ifndef TRKEXALGS_MAXPARAMETERS
#define TRKEXALGS_MAXPARAMETERS 10
#endif
 
 
namespace Trk 
{
 
  class IExtrapolator;
  class Surface;
  class TrackingGeometry;
  class TrackingVolume;
 
  class EnergyLossExtrapolationValidation : public AthAlgorithm
    {
    public:
 
       EnergyLossExtrapolationValidation(const std::string& name, ISvcLocator* pSvcLocator);
       ~EnergyLossExtrapolationValidation();
 
       StatusCode          initialize();
       StatusCode          execute();
       StatusCode          finalize();
 
    private:
      static std::unique_ptr<Amg::Transform3D> createTransform(double x,
                                                        double y,
                                                        double z,
                                                        double phi = 0.,
                                                        double theta = 0.,
                                                        double alphaZ = 0.);
 
      const TrackingVolume*     m_highestVolume;
 
      ToolHandle<IExtrapolator> m_extrapolator;
 
      Rndm::Numbers*            m_gaussDist;
      Rndm::Numbers*            m_flatDist;
 
      bool                      m_materialCollectionValidation; 
 
      TTree*                    m_validationTree;            
      TTree*                    m_validationRunTree;         
 
      std::string               m_validationTreeFolder;      
      std::string               m_validationTreeName;        
      std::string               m_validationTreeDescription; 
      std::string               m_validationRunTreeFolder;      
      std::string               m_validationRunTreeName;        
      std::string               m_validationRunTreeDescription; 
 
      double                    m_maximumR;                  
      double                    m_maximumZ;                  
 
 
      size_t                    m_cylinders;      
      bool                      m_onion;          
      float                     m_momentum;       
      bool                      m_usePt;          
      float                     m_minEta;         
      float                     m_maxEta;         
 
      size_t                    m_events;         
      int                       m_totalRecordedLayers; 
      float                     m_avgRecordedLayers;   
 
      int                       m_pdg;            
      int                       m_particleType;   
 
      // ---- variables for the ROOT event tree
      size_t            m_entries;         
      float             m_energy[TRKEXALGS_MAXPARAMETERS];           
      float             m_energyLoss[TRKEXALGS_MAXPARAMETERS];       
      float             m_parameterX0[TRKEXALGS_MAXPARAMETERS];      
      float             m_radius[TRKEXALGS_MAXPARAMETERS];           
//      float             m_parameterLoc1[TRKEXALGS_MAXPARAMETERS];    //!< local 1
//      float             m_parameterLoc2[TRKEXALGS_MAXPARAMETERS];    //!< local 2
      float             m_positionX[TRKEXALGS_MAXPARAMETERS];        
      float             m_positionY[TRKEXALGS_MAXPARAMETERS];        
      float             m_positionZ[TRKEXALGS_MAXPARAMETERS];        
      float             m_parameterPhi[TRKEXALGS_MAXPARAMETERS];     
      float             m_parameterEta[TRKEXALGS_MAXPARAMETERS];     
      float             m_parameterTheta[TRKEXALGS_MAXPARAMETERS];   
      float             m_parameterQoverP[TRKEXALGS_MAXPARAMETERS];  
      float             m_parameterP[TRKEXALGS_MAXPARAMETERS];       
      size_t            m_layer[TRKEXALGS_MAXPARAMETERS];            
 
      // ---- output statistics
      size_t            m_triesForward;      
      size_t            m_breaksForward;     
      size_t            m_triesBack;         
      size_t            m_breaksBack;        
 
      size_t            m_collectedLayerForward;   
      size_t            m_collectedLayerBack;    
 
 
      // ---- cylinders and discs
      float                     m_cylinderR[TRKEXALGS_MAXPARAMETERS];   
      float                     m_cylinderZ[TRKEXALGS_MAXPARAMETERS];   
      DataVector<const Trk::CylinderSurface>* m_theCylinders;
      DataVector<const Trk::DiscSurface>* m_theDiscs1;
      DataVector<const Trk::DiscSurface>* m_theDiscs2;
      std::vector<float>        m_cylinderVR;       
      std::vector<float>        m_cylinderVZ;       
    }; 
} // end of namespace
 
#endif