MaterialEffectsEngine.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// MaterialEffectsEngine.h, (c) ATLAS Detector software
 
#ifndef TRKEXINTERFACES_MATERIAKEFFECTSENGINE_H
#define TRKEXINTERFACES_MATERIAKEFFECTSENGINE_H
 
// Gaudi
#include "AthenaBaseComps/AthAlgTool.h"
// Trk
#include "TrkExInterfaces/IMaterialEffectsEngine.h"
#include "TrkExInterfaces/ExtrapolationMacros.h"
#include "TrkExUtils/ExtrapolationCell.h"
#include "TrkExUtils/MaterialInteraction.h"
#include "TrkExUtils/MaterialUpdateMode.h"
#include "TrkEventPrimitives/PropDirection.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkNeutralParameters/NeutralParameters.h"
#include "CxxUtils/checker_macros.h"
namespace Trk {
  
  class Layer;
    
  class MaterialEffectsEngine : public AthAlgTool, virtual public IMaterialEffectsEngine {
    public:
 
      MaterialEffectsEngine(const std::string&,const std::string&,const IInterface*);
      
      ~MaterialEffectsEngine();
 
      StatusCode initialize();
      
      StatusCode finalize();
      
      virtual ExtrapolationCode handleMaterial(ExCellCharged& ecCharged,
                                               PropDirection dir=alongMomentum,
                                               MaterialUpdateStage matupstage=fullUpdate) const;
 
      virtual ExtrapolationCode handleMaterial(ExCellNeutral& ecNeutral,
                                               PropDirection dir=alongMomentum,
                                               MaterialUpdateStage matupstage=fullUpdate) const;
      
    protected:
      TrackParameters* updateTrackParameters(Trk::TrackParameters& parameters,
                                             Trk::ExCellCharged& eCell,
                                             Trk::PropDirection dir,
                                             Trk::MaterialUpdateStage matupstage) const;
        
      MaterialInteraction                          m_interactionFormulae;     
      bool                                         m_eLossCorrection;         
      bool                                         m_eLossMpv;                
      bool                                         m_mscCorrection;           
 
  };
      
 
} // end of namespace
 
#endif // TRKEXINTERFACES_MATERIAKEFFECTSENGINE_H