ParametricMuonProfile.h

Go to the documentation of this file.

#ifndef __ParametricMuonProfile__
#define __ParametricMuonProfile__

#include <tls/VMuonProfile.h>
#include <tls/MuonProfileUtilities.h>

#include <boost/multi_array.hpp>

#include <vector>

namespace utl { class Branch; }

namespace ParametricMuonProfileNS {
  class ParametricMuonProfile : public tls::VMuonProfile {
  public:

    enum ProfileType { eNMuTank, eCosTheta, eLogEnergy };

    ParametricMuonProfile(utl::Branch);
    ~ParametricMuonProfile() {}

    inline
    void CachedSetProfile(double theta, double phi, double energy)
    {
      if (theta < GetThetaMin()) theta = GetThetaMin();
      else if (theta > GetThetaMax()) theta = GetThetaMax();

      if (theta == fTheta && phi == fPhi && energy == fEnergy) return;

      fEnergy           = energy;
      fTheta            = theta;
      fPhi              = phi;
      fBPsi             = tls::GetBpsi(theta, phi);
      fNMuonNumber      = TotalMuonNumber(theta,energy);

      BuildLocalModel();
    }

    double NMuon(double xpos, double ypos, double theta, double phi, double energy)
    { return EvaluateProfile(eNMuTank, xpos, ypos, theta, phi, energy); }

    double ThetaMuon(double xpos, double ypos, double theta, double phi, double energy)
    { return EvaluateProfile(eCosTheta, xpos, ypos, theta, phi, energy); }

    double EnergyMuon(double xpos, double ypos, double theta, double phi, double energy)
    { return EvaluateProfile(eLogEnergy, xpos, ypos, theta, phi, energy); }

    double GetThetaMax();
    double GetThetaMin();

  private:

    double D(double theta) const;

    double TotalMuonNumber(double theta, double energy) const;

    void BuildLocalModel();

    double EvaluateProfile(ProfileType type, double x, double y, double theta, double phi, double energy);

    double EvaluateLocalProfile(ProfileType type, double sqr, double zeta) const;

    typedef boost::multi_array<double,6> ModelDataType;
    ModelDataType fModelData[3];

    typedef boost::multi_array<double,3> LocalDataType;
    LocalDataType fLocalData[3];

    void ReadModelData(const utl::Branch& modelBranch,
                       ModelDataType& tensor,
                       LocalDataType& tensorLocal) const;

    double fBiasCorrection[3];

    typedef std::vector<double> ArrayType;

    ArrayType fDMaxData;

    ArrayType fNMuData;

    ArrayType fThetaRange;
    double fSqrtRadiusMax;

    double fTheta;
    double fPhi;
    double fEnergy;
    double fBPsi;
    double fNMuonNumber;
  };
} // NS ParametricMuonProfileNS

#endif