TauValidation.h

Go to the documentation of this file.

#ifndef TauValidation_H
#define TauValidation_H

/*class TauValidation
 *  
 *  Class to fill Event Generator dqm monitor elements; works on HepMCProduct
 *
 *
 */

// framework & common header files

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Run.h"

#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

//DQM services
#include "DQMServices/Core/interface/DQMStore.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "DQMServices/Core/interface/MonitorElement.h"
#include "DQMServices/Core/interface/DQMEDAnalyzer.h"

#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"

#include "SimGeneral/HepPDTRecord/interface/ParticleDataTable.h"
#include "TLorentzVector.h"

#include "Validation/EventGenerator/interface/WeightManager.h"

class TauValidation : public DQMEDAnalyzer
{
    public:
    // tau decays
        enum  {undetermined,
               electron,
               muon,
               pi,
               rho,
           a1,
               K,
           Kstar,
           pi1pi0,
               pinpi0,
               tripi,
               tripinpi0,
           stable};
    // tau mother particles 
    enum  {other,
           B,
           D,
           gamma,
           Z,
           W,
           HSM,
           H0,
           A0,
           Hpm};

    public:
    explicit TauValidation(const edm::ParameterSet&);
    virtual ~TauValidation();
    virtual void bookHistograms(DQMStore::IBooker &i, edm::Run const &, edm::EventSetup const &) override;
    virtual void dqmBeginRun(const edm::Run& r, const edm::EventSetup& c) override;
    virtual void analyze(edm::Event const&, edm::EventSetup const&) override;
    private:
      WeightManager wmanager_;

    int tauMother(const HepMC::GenParticle*, double weight);
    int tauProngs(const HepMC::GenParticle*, double weight);
    int tauDecayChannel(const HepMC::GenParticle*, double weight=0.0);
    int findMother(const HepMC::GenParticle*);
    bool isLastTauinChain(const HepMC::GenParticle* tau);
    void rtau(const HepMC::GenParticle*,int,int, double weight);
    void spinEffectsWHpm(const HepMC::GenParticle*,int,int,std::vector<HepMC::GenParticle*> &part,double weight);
    void spinEffectsZH(const HepMC::GenParticle* boson, double weight);
    double leadingPionMomentum(const HepMC::GenParticle*, double weight);
    double visibleTauEnergy(const HepMC::GenParticle*);
    TLorentzVector leadingPionP4(const HepMC::GenParticle*);
    TLorentzVector motherP4(const HepMC::GenParticle*);
    void photons(const HepMC::GenParticle*, double weight);
    void findTauList(const HepMC::GenParticle* tau,std::vector<const HepMC::GenParticle*> &TauList);
    void findFSRandBrem(const HepMC::GenParticle* p, bool doBrem, std::vector<const HepMC::GenParticle*> &ListofFSR,
               std::vector<const HepMC::GenParticle*> &ListofBrem);
    void FindPhotosFSR(const HepMC::GenParticle* p,std::vector<const HepMC::GenParticle*> &ListofFSR,double &BosonScale);
    const HepMC::GenParticle* GetMother(const HepMC::GenParticle* tau);
    const std::vector<HepMC::GenParticle*> GetMothers(const HepMC::GenParticle* boson);
    double Zstoa(double zs);

        edm::InputTag hepmcCollection_;

    double tauEtCut;

    edm::ESHandle<HepPDT::ParticleDataTable> fPDGTable ;
  
        MonitorElement *nTaus, *nPrimeTaus;
    MonitorElement *TauPt, *TauEta, *TauPhi, *TauProngs, *TauDecayChannels, *TauMothers, 
      *TauRtauW, *TauRtauHpm,
      *TauSpinEffectsW_X, *TauSpinEffectsW_UpsilonRho, *TauSpinEffectsW_UpsilonA1,*TauSpinEffectsW_eX,*TauSpinEffectsW_muX,
      *TauSpinEffectsHpm_X, *TauSpinEffectsHpm_UpsilonRho, *TauSpinEffectsHpm_UpsilonA1,*TauSpinEffectsHpm_eX,*TauSpinEffectsHpm_muX, 
      *TauSpinEffectsZ_MVis, *TauSpinEffectsZ_Zs, *TauSpinEffectsZ_Xf, *TauSpinEffectsZ_Xb, 
      *TauSpinEffectsZ_eX, *TauSpinEffectsZ_muX, *TauSpinEffectsZ_X, *TauSpinEffectsH_X,
      *TauSpinEffectsH_MVis, *TauSpinEffectsH_Zs, *TauSpinEffectsH_Xf, *TauSpinEffectsH_Xb,
      *TauSpinEffectsH_eX, *TauSpinEffectsH_muX, *TauSpinEffectsH_rhorhoAcoplanarityplus,  *TauSpinEffectsH_rhorhoAcoplanarityminus,
      *TauBremPhotonsN,*TauBremPhotonsPt,*TauBremPhotonsPtSum,*TauFSRPhotonsN,*TauFSRPhotonsPt,*TauFSRPhotonsPtSum,
      *TauSpinEffectsH_pipiAcoplanarity,*TauSpinEffectsH_pipiAcollinearity,*TauSpinEffectsH_pipiAcollinearityzoom, *DecayLength,
      *LifeTime;

    unsigned int NJAKID;
    MonitorElement *JAKID;
    std::vector<std::vector<MonitorElement *> > JAKInvMass;

    int zsbins;
    double zsmin,zsmax;

    edm::EDGetTokenT<edm::HepMCProduct> hepmcCollectionToken_;
};

#endif