MaterialEffectsEngine.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 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;     
      BooleanProperty m_eLossCorrection{this, "EnergyLossCorrection", true,
    "apply the energy loss correction"};
      BooleanProperty m_eLossMpv{this, "MostProbableEnergyLoss", true,
    "apply the energy loss correction as most probable value"};
      BooleanProperty m_mscCorrection{this, "MultipleScatteringCorrection", true,
    "apply the multiple (coulomb) scattering correction"};
 
      StringProperty m_sopPrefix_prop{this, "OutputPrefix", ""};
      StringProperty m_sopPostfix_prop{this, "OutputPostfix", ""};
 
  };
      
 
} // end of namespace
 
#endif // TRKEXINTERFACES_MATERIAKEFFECTSENGINE_H