AFPProtonTransportTool.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef TRIGAFPHYPO_AFPPROTONTRANSPORTTOOL_H
#define TRIGAFPHYPO_AFPPROTONTRANSPORTTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "IAFPProtonTransportTool.h"
#include "AFPProtonTransportParam.h"
 
#include <string>
#include <memory>

 
class AFPProtonTransportTool : public AthAlgTool, virtual public IAFPProtonTransportTool {
public:
  AFPProtonTransportTool(const std::string & type, const std::string & name, const IInterface* parent);
 
  virtual ~AFPProtonTransportTool();
  
  virtual StatusCode initialize() override;
  
  virtual StatusCode load() override;
 
  virtual void setParamFile(const std::string& paramFile) override { m_filename = paramFile; }

  virtual double x(double x0, double y0, double z0, double sx0, double sy0, double E) const override {
    return m_x_position->evaluate(x0 + m_xPositionInitIP, y0 + m_yPositionInitIP, z0 + m_zPositionInitIP, sx0 + m_xSlopeInitIP,
                          sy0 + m_ySlopeInitIP, E);
  }

  double y(double x0, double y0, double z0, double sx0, double sy0, double E) const {
    return m_y_position->evaluate(x0 + m_xPositionInitIP, y0 + m_yPositionInitIP, z0 + m_zPositionInitIP, sx0 + m_xSlopeInitIP,
                          sy0 + m_ySlopeInitIP, E);
  }

  double sx(double x0, double y0, double z0, double sx0, double sy0, double E) const {
    return m_x_slope->evaluate(x0 + m_xPositionInitIP, y0 + m_yPositionInitIP, z0 + m_zPositionInitIP, sx0 + m_xSlopeInitIP,
                           sy0 + m_ySlopeInitIP, E);
  }

  double sy(double x0, double y0, double z0, double sx0, double sy0, double E) const {
   return m_y_slope->evaluate(x0 + m_xPositionInitIP, y0 + m_yPositionInitIP, z0 + m_zPositionInitIP, sx0 + m_xSlopeInitIP, sy0 + m_ySlopeInitIP, E);
  }

  double parametrisationPosition() const {return m_parametrisationPosition;}

  double energy() const {return m_energy;};
private:

  std::unique_ptr<AFPProtonTransportParam> m_x_position;
  std::unique_ptr<AFPProtonTransportParam> m_y_position;
  std::unique_ptr<AFPProtonTransportParam> m_x_slope;
  std::unique_ptr<AFPProtonTransportParam> m_y_slope;

  Gaudi::Property<double> m_xSlopeInitIP {this, "xSlopeInitIp", 0, "Initial x slope at IP"};
  Gaudi::Property<double> m_ySlopeInitIP {this, "ySlopeInitIp", 0, "Initial y slope at IP"}; 
  Gaudi::Property<double> m_xPositionInitIP {this, "xPositionInitIp", 0, "Initial x position at IP"};
  Gaudi::Property<double> m_yPositionInitIP {this, "yPositionInitIp", 0, "Initial y position at IP"};
  Gaudi::Property<double> m_zPositionInitIP {this, "zPositionInitIp", 0, "Initial z position at IP"};
  Gaudi::Property<double> m_energy {this, "beamEnergy", 0, "Nominal beam energy"}; 
  Gaudi::Property<double> m_parametrisationPosition {this, "parameterisationPosition", 0, "Detector position for which parameterisation was derived"}; 
  
  Gaudi::Property<std::string> m_filename {this, "filename", "", "Parameterisation file"};
 
};
 
#endif