ForwardSimplifiedGeometry.h

Go to the documentation of this file.

#ifndef FASTSIM_FORWARDSIMPLIFIEDGEOMETRY_H
#define FASTSIM_FORWARDSIMPLIFIEDGEOMETRY_H

#include "FastSimulation/SimplifiedGeometryPropagator/interface/SimplifiedGeometry.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include <TH1F.h>
#include "FWCore/Utilities/interface/Exception.h"

// Author: L. Vanelderen, S. Kurz
// Date: 29 May 2017

namespace fastsim {


  class ForwardSimplifiedGeometry : public SimplifiedGeometry {
  public:

    ForwardSimplifiedGeometry(double z) : SimplifiedGeometry(z) {}

    ForwardSimplifiedGeometry(ForwardSimplifiedGeometry &&) = default;

    ~ForwardSimplifiedGeometry() override{};


    const double getZ() const { return geomProperty_; }


    const double getThickness(const math::XYZTLorentzVector &position) const override {
      return thicknessHist_->GetBinContent(thicknessHist_->GetXaxis()->FindBin(position.Pt()));
    }


    const double getThickness(const math::XYZTLorentzVector &position,
                              const math::XYZTLorentzVector &momentum) const override {
      return getThickness(position) / fabs(momentum.Pz()) * momentum.P();
    }


    const double getMagneticFieldZ(const math::XYZTLorentzVector &position) const override {
      return magneticFieldHist_->GetBinContent(magneticFieldHist_->GetXaxis()->FindBin(position.Pt()));
    }


    bool isForward() const override { return true; }
  };

}  // namespace fastsim

#endif