ZMassHistogrammer.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "TH1.h"

class ZMassHistogrammer : public edm::EDAnalyzer {
public:
  ZMassHistogrammer(const edm::ParameterSet& pset);
private:
  void analyze(const edm::Event& event, const edm::EventSetup& setup) override;
  edm::EDGetTokenT<reco::CandidateView>  zToken_;
  edm::EDGetTokenT<reco::CandidateView>  genToken_;
  TH1F *h_mZ_, *h_mZMC_;
};

#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include <iostream>

using namespace std;
using namespace reco;
using namespace edm;

ZMassHistogrammer::ZMassHistogrammer(const ParameterSet& pset) :
  zToken_(consumes<reco::CandidateView>(pset.getParameter<InputTag>("z"))),
  genToken_(consumes<reco::CandidateView>(pset.getParameter<InputTag>("gen"))) {
  cout << ">>> Z Mass constructor" << endl;
  Service<TFileService> fs;
  h_mZ_ = fs->make<TH1F>("ZMass", "Z mass (GeV/c^{2})", 100,  0, 200);
  h_mZMC_ = fs->make<TH1F>("ZMCMass", "Z MC mass (GeV/c^{2})", 100,  0, 200);
}

void ZMassHistogrammer::analyze(const edm::Event& event, const edm::EventSetup& setup) {
  cout << ">>> Z Mass analyze" << endl;
  Handle<CandidateCollection> z;
  Handle<CandidateCollection> gen;
  event.getByToken(zToken_, z);
  event.getByToken(genToken_, gen);
  for(unsigned int i = 0; i < z->size(); ++i) {
    const Candidate &zCand = (*z)[i];
    h_mZ_->Fill(zCand.mass());
  }
  for(unsigned int i = 0; i < gen->size(); ++i) {
    const Candidate &genCand = (*gen)[i];
    if((genCand.pdgId() == 23) && (genCand.status() == 2)) //this is an intermediate Z0
      cout << ">>> intermediate Z0 found, with " << genCand.numberOfDaughters()
       << " daughters" << endl;
    if((genCand.pdgId() == 23)&&(genCand.status() == 3)) { //this is a Z0
      cout << ">>> Z0 found, with " << genCand.numberOfDaughters()
       << " daughters" << endl;
      h_mZMC_->Fill(genCand.mass());
      if(genCand.numberOfDaughters() == 3) {//Z0 decays in mu+ mu-, the 3rd daughter is the same Z0
    const Candidate * dauGen0 = genCand.daughter(0);
    const Candidate * dauGen1 = genCand.daughter(1);
    const Candidate * dauGen2 = genCand.daughter(2);
    cout << ">>> daughter MC 0 PDG Id " << dauGen0->pdgId()
         << ", status " << dauGen0->status()
         << ", charge " << dauGen0->charge()
         << endl;
    cout << ">>> daughter MC 1 PDG Id " << dauGen1->pdgId()
         << ", status " << dauGen1->status()
         << ", charge " << dauGen1->charge()
         << endl;
    cout << ">>> daughter MC 2 PDG Id " << dauGen2->pdgId()
         << ", status " << dauGen2->status()
         << ", charge " << dauGen2->charge() << endl;
      }
    }
  }
}

#include "FWCore/Framework/interface/MakerMacros.h"

DEFINE_FWK_MODULE(ZMassHistogrammer);