HCALResponse.h

Go to the documentation of this file.

#ifndef HCALResponse_h
#define HCALResponse_h

#include "FastSimulation/Utilities/interface/DoubleCrystalBallGenerator.h"
#include <vector>
#include <string>

//define multidimensional vector types
typedef std::vector<double> vec1;
typedef std::vector<vec1>   vec2;
typedef std::vector<vec2>   vec3;
typedef std::vector<vec3>   vec4;
typedef std::vector<vec4>   vec5;
enum part{hcbarrel=0, hcendcap=1, hcforward=2};
enum type{ECAL=0, HCAL=1, VFCAL=2};

class RandomEngineAndDistribution;

namespace edm { 
  class ParameterSet;
}

class HCALResponse
{
 public:
  HCALResponse(const edm::ParameterSet& pset);
  ~HCALResponse(){ } 

  // Get the response smearing factor
  // for  e/gamma = 0, hadron = 1, mu = 2, mip: 0/1/2
  // mip = 2 means "mean" response regardless actual mip
  double responseHCAL(int _mip, double energy, double eta, int partype, RandomEngineAndDistribution const*);

  //Get the energy and eta dependent mip fraction
  double getMIPfraction(double energy, double eta);

  // legacy methods using simple formulae
  double getHCALEnergyResponse(double e, int hit, RandomEngineAndDistribution const*);

  // correct HF response for SL
  void correctHF(double e, int type);
  vec1 & getCorrHFem()  {return corrHFem;}   
  vec1 & getCorrHFhad() {return corrHFhad;}   
  
 private:

  // calculates interpolated-extrapolated response smearing factors
  // for hadrons, muons, and e/gamma (the last in HF specifically)
  double interHD(int mip, double e, int ie, int ieta, int det, RandomEngineAndDistribution const*);
  double interEM(double e, int ie, int ieta, RandomEngineAndDistribution const*);
  double interMU(double e, int ie, int ieta, RandomEngineAndDistribution const*);
  
  //random shooting functions w/ protection from negative energies
  double gaussShootNoNegative(double e, double sigma,RandomEngineAndDistribution const*);
  double cballShootNoNegative(double mu, double sigma, double aL, double nL, double aR, double nR,RandomEngineAndDistribution const*);
  double PoissonShootNoNegative(double e, double sigma,RandomEngineAndDistribution const*);

  //find subdet
  int getDet(int ieta);
  
  //debugging and mip toggles
  bool debug, usemip;
  
  //Default values for resolution parametrisation:
  //stochastic, constant and noise.
  //in the barrel and in the endcap
  //in the ECAL, HCAL, VFCAL
  double RespPar[3][2][3];

  //HCAL response parameters
  double eResponseScale[3];
  double eResponsePlateau[3];
  double eResponseExponent;
  double eResponseCoefficient;

  //max values
  int maxMUe, maxMUeta, maxMUbin, maxEMe, maxEMeta;
  int maxHDe[4];
  // eta step for eta index calc
  double etaStep;
  // eta index for different regions
  int HDeta[4], maxHDetas[3], barrelMUeta, endcapMUeta;
  // energy step of the tabulated muon data
  double muStep;
  // correction factor for HF EM
  double respFactorEM;

  // Tabulated energy, et/pt and eta points
  vec1 eGridHD[4];
  vec1 eGridEM;
  vec1 eGridMU;
  vec1 etaGridMU;

  // Tabulated response and mean for hadrons normalized to the energy
  // indices: parameters[par][mip][det][energy][eta]
  int nPar;
  std::vector<std::string> parNames;
  vec5 parameters;
  
  // Tabulated response and mean for e/gamma in HF specifically (normalized)
  // indices: meanEM[energy][eta]
  vec2 meanEM, sigmaEM;

  // muon histos
  // indices: responseMU[energy][eta][bin]
  vec3 responseMU; 
  vec3 mipfraction;
  vec3 PoissonParameters;

  // crystal ball generator
  DoubleCrystalBallGenerator cball;

  // HF correction for SL
  int maxEta, maxEne;
  vec1 energyHF;
  vec2 corrHFgEm, corrHFgHad;
  vec2 corrHFhEm, corrHFhHad;
  vec1 corrHFem, corrHFhad;
};
#endif