MassResolution.h

#ifndef MassResolution_h
#define MassResolution_h

//----------------------------------------------------------//
// Project:   MassResolution
// Author:    Matt Kenzie (matthew.william.kenzie@cern.ch)
// Modified:  25/08/2011
// Admins:    Matth Kenzie (matthew.william.kenzie@cern.ch)
//---------------------------------------------------------//

/*
  Calculates the mass resolution for a higgs to two photon event.
  Designed to work solely inside the h2gglobe framework.

  Define a MassResolution object as public data member of StatAnalysis.h
  (or equivalent) with 
    MassResolution massResFinder;
  
  In the Init method of StatAnalysis create a new object with 
    massResFinder = new MassResolution();

  Make sure to clean this in the Term method of StatAnalysis with
    delete massResFinder;

  To calculate the mass resolution event by event insert the following in 
  the Analysis method of StatAnalysis (after the smearing has been done and HiggsM andHiggs Pt have been calculated)

    massResFinder.Setup(l,diphoton_index.first,diphton_index.second,diphoton_id,higgsPt,higgsMass,eSmearPars,nR9Categories,nEtaCategories);

  and then access the mass resolution with
    
    massResFinder.getMassResolution();

  there are various other methods to get more specialised information such getAngleResolution(), getLeadResolution() etc.

  You can also use printInfo() or dumpInfo(filename) to print or write the information in Setup.

*/
  

#include <iostream>
#include <string>
#include <cassert>

#include "TROOT.h"
#include "TMath.h"
#include "TLorentzVector.h"
#include "TVector3.h"
#include "TFile.h"
#include "TF1.h"
#include "math.h"

#include "BaseAnalysis.h"
#include "BaseSmearer.h"
//#include "PhotonAnalysis.h"
#include "LoopAll.h"
#include "EnergySmearer.h"
#include "RooContainer.h"
#include "PhotonReducedInfo.h"

class MassResolution {

  public:
    MassResolution();

    void Setup(LoopAll&,PhotonReducedInfo*,PhotonReducedInfo*,int,EnergySmearer::energySmearingParameters,int,int,double);
    void Setup(LoopAll&,PhotonReducedInfo*,PhotonReducedInfo*,int,EnergySmearer::energySmearingParameters,int,int,double,bool);

    double beamspotSigma;

   // double relMassResolution();
    double relMassResolutionCorrVtx();
    double relMassResolutionWrongVtx();
    double relMassResolutionEonly();
    double relMassResolutionAonly();
    double relMassResolutionEonlyNoSmear();
    double relMassResolutionCorrVtxNoSmear();
    double relMassResolutionWrongVtxNoSmear();

    double leadRelPhotonResolution();
    double subleadRelPhotonResolution();
    
    double leadRelPhotonResolutionNoSmear();
    double subleadRelPhotonResolutionNoSmear();

    double leadPhotonResolution() { return leadRelPhotonResolution() * leadPhoton->corrEnergy(); };
    double subleadPhotonResolution() { return subleadRelPhotonResolution() * subleadPhoton->corrEnergy(); };
    
    double leadPhotonResolutionNoSmear() { return leadRelPhotonResolutionNoSmear() * leadPhoton->corrEnergy(); };
    double subleadPhotonResolutionNoSmear() { return subleadRelPhotonResolutionNoSmear() * subleadPhoton->corrEnergy(); };
  
  private:
    double getRelPhotonResolution(double, const PhotonReducedInfo &);
    
    double angleResolution();
    double angleResolutionCorrVtx();
    double angleResolutionWrongVtx();
    
    double dzResolution();
    double dzResolutionCorrVtx();
    double dzResolutionWrongVtx();
    
    //double getEffectiveSigma(TF1*, int);
    double propagateDz(double);

    double SecH(double);
    double TanH(double);
    
   // TLorentzVector *lead_p4;
   // TLorentzVector *sublead_p4;
    PhotonReducedInfo *leadPhoton;
    PhotonReducedInfo *subleadPhoton;

    TVector3 lead_sc_pos;
    TVector3 sublead_sc_pos;

    TVector3 *vertex;
    TVector3 *vtx_dxdydz;
    double lead_Eres;
    double sublead_Eres;
    double lead_r9;
    double sublead_r9;
    bool lead_iDet;
    bool sublead_iDet;
    EnergySmearer::energySmearingParameters _eSmearPars;

    double dz;
    double higgsMass;

    TFile *dz_file;
    TGraph *dz_plot;

};

#endif