CMSSW_4_4_3_patch1/src/Validation/EventGenerator/plugins/DrellYanValidation.cc

Go to the documentation of this file.

/*class DrellYanValidation
 *  
 *  Class to fill dqm monitor elements from existing EDM file
 *
 *  $Date: 2011/01/24 18:27:40 $
 *  $Revision: 1.4 $
 */
 
#include "Validation/EventGenerator/interface/DrellYanValidation.h"
#include "Validation/EventGenerator/interface/HepMCValidationHelper.h"
#include "TLorentzVector.h"

#include "CLHEP/Units/defs.h"
#include "CLHEP/Units/PhysicalConstants.h"

#include "DataFormats/Math/interface/LorentzVector.h"

using namespace edm;

DrellYanValidation::DrellYanValidation(const edm::ParameterSet& iPSet):  
  hepmcCollection_(iPSet.getParameter<edm::InputTag>("hepmcCollection")),
  _flavor(iPSet.getParameter<int>("decaysTo")),
  _name(iPSet.getParameter<std::string>("name")) 
{    
  dbe = 0;
  dbe = edm::Service<DQMStore>().operator->();
}

DrellYanValidation::~DrellYanValidation() {}

void DrellYanValidation::beginJob()
{
  if(dbe){
    std::string folderName = "Generator/DrellYan";
    folderName+=_name;
    dbe->setCurrentFolder(folderName.c_str());
    
    // Number of analyzed events
    nEvt = dbe->book1D("nEvt", "n analyzed Events", 1, 0., 1.);
    
    //Kinematics
    Zmass = dbe->book1D("Zmass","inv. Mass Z", 70 ,0,140);
    ZmassPeak = dbe->book1D("ZmassPeak","inv. Mass Z", 80 ,80 ,100);
    Zpt = dbe->book1D("Zpt","Z pt",100,0,200);
    ZptLog = dbe->book1D("ZptLog","log(Z pt)",100,0.,5.);
    Zrap = dbe->book1D("Zrap", "Z y", 100, -5, 5);
    Zdaughters = dbe->book1D("Zdaughters", "Z daughters", 60, -30, 30);

    dilep_mass = dbe->book1D("dilep_mass","inv. Mass dilepton", 70 ,0,140);
    dilep_massPeak = dbe->book1D("dilep_massPeak","inv. Mass dilepton", 80 ,80 ,100);
    dilep_pt = dbe->book1D("dilep_pt","dilepton pt",100,0,200);
    dilep_ptLog = dbe->book1D("dilep_ptLog","log(dilepton pt)",100,0.,5.);
    dilep_rap = dbe->book1D("dilep_rap", "dilepton y", 100, -5, 5);

    gamma_energy = dbe->book1D("gamma_energy", "photon energy in Z rest frame", 200, 0., 100.);
    cos_theta_gamma_lepton = dbe->book1D("cos_theta_gamma_lepton",      "cos_theta_gamma_lepton in Z rest frame",      200, -1, 1);

    leadpt = dbe->book1D("leadpt","leading lepton pt", 200, 0., 200.);    
    secpt  = dbe->book1D("secpt","second lepton pt", 200, 0., 200.);    
    leadeta = dbe->book1D("leadeta","leading lepton eta", 100, -5., 5.);
    seceta  = dbe->book1D("seceta","second lepton eta", 100, -5., 5.);

  }
  return;
}

void DrellYanValidation::endJob(){return;}
void DrellYanValidation::beginRun(const edm::Run& iRun,const edm::EventSetup& iSetup)
{
  iSetup.getData( fPDGTable );
  return;
}
void DrellYanValidation::endRun(const edm::Run& iRun,const edm::EventSetup& iSetup){return;}
void DrellYanValidation::analyze(const edm::Event& iEvent,const edm::EventSetup& iSetup)
{ 
  
  // we *DO NOT* rely on a Z entry in the particle listings!

  edm::Handle<HepMCProduct> evt;
  iEvent.getByLabel(hepmcCollection_, evt);

  //Get EVENT
  const HepMC::GenEvent *myGenEvent = evt->GetEvent();

  nEvt->Fill(0.5);

  std::vector<const HepMC::GenParticle*> allproducts; 

  //requires status 1 for leptons and neutrinos (except tau)
  int requiredstatus = (abs(_flavor) == 11 || abs(_flavor) == 12 || abs(_flavor) ==13 || abs(_flavor) ==14 || abs(_flavor) ==16) ? 1 : 3;

  bool vetotau = true; //(abs(_flavor) == 11 || abs(_flavor) == 12 || abs(_flavor) ==13 || abs(_flavor) ==14 || abs(_flavor) ==16) ? true : false;  

  for(HepMC::GenEvent::particle_const_iterator iter = myGenEvent->particles_begin(); iter != myGenEvent->particles_end(); ++iter) {
    if (vetotau) {
      if ((*iter)->status()==3 && abs((*iter)->pdg_id() == 15) ) return; 
    }
    if((*iter)->status()==requiredstatus) {
      if(abs((*iter)->pdg_id())==_flavor)
        allproducts.push_back(*iter);
    }
  }
 
  //nothing to do if we don't have 2 particles
  if (allproducts.size() < 2) return; 

  //sort them in pt
  std::sort(allproducts.begin(), allproducts.end(), HepMCValidationHelper::sortByPt); 

  //get the first element and the first following element with opposite charge 
  std::vector<const HepMC::GenParticle*> products;
  products.push_back(allproducts.front());
  const HepPDT::ParticleData* PData1 = fPDGTable->particle(HepPDT::ParticleID(allproducts.front()->pdg_id()));
  double charge1 = PData1->charge(); 
  for (unsigned int i = 1; i < allproducts.size(); ++i ){
    const HepPDT::ParticleData* PData2 = fPDGTable->particle(HepPDT::ParticleID(allproducts[i]->pdg_id()));
    double charge2 = PData2->charge();
    if (charge1*charge2 < 0) products.push_back(allproducts[i]);
  }

  //if we did not find any opposite charge pair there is nothing to do
  if (products.size() < 2) return; 

  assert(products[0]->momentum().perp() > products[1]->momentum().perp()); 

  //leading lepton with pt > 20.
  if (products[0]->momentum().perp() < 20.) return;

  //assemble FourMomenta
  TLorentzVector lep1(products[0]->momentum().x(), products[0]->momentum().y(), products[0]->momentum().z(), products[0]->momentum().t()); 
  TLorentzVector lep2(products[1]->momentum().x(), products[1]->momentum().y(), products[1]->momentum().z(), products[1]->momentum().t()); 
  TLorentzVector dilepton_mom = lep1 + lep2;
  TLorentzVector dilepton_andphoton_mom = dilepton_mom;

  //mass > 60.
  if (dilepton_mom.M() < 60.) return;

  //find possible qed fsr photons
  std::vector<const HepMC::GenParticle*> fsrphotons;
  HepMCValidationHelper::findFSRPhotons(products, myGenEvent, 0.1, fsrphotons);
  
  Zdaughters->Fill(products[0]->pdg_id()); 
  Zdaughters->Fill(products[1]->pdg_id()); 

  std::vector<TLorentzVector> gammasMomenta;
  for (unsigned int ipho = 0; ipho < fsrphotons.size(); ++ipho){
    TLorentzVector phomom(fsrphotons[ipho]->momentum().x(), fsrphotons[ipho]->momentum().y(), fsrphotons[ipho]->momentum().z(), fsrphotons[ipho]->momentum().t()); 
    dilepton_andphoton_mom += phomom;
    Zdaughters->Fill(fsrphotons[ipho]->pdg_id());
    gammasMomenta.push_back(phomom);
  }  
  //Fill Z histograms
  Zmass->Fill(dilepton_andphoton_mom.M());
  ZmassPeak->Fill(dilepton_andphoton_mom.M());
  Zpt->Fill(dilepton_andphoton_mom.Pt());
  ZptLog->Fill(log10(dilepton_andphoton_mom.Pt())); 
  Zrap->Fill(dilepton_andphoton_mom.Rapidity());

  //Fill dilepton histograms
  dilep_mass->Fill(dilepton_mom.M());
  dilep_massPeak->Fill(dilepton_mom.M());
  dilep_pt->Fill(dilepton_mom.Pt());
  dilep_ptLog->Fill(log10(dilepton_mom.Pt()));
  dilep_rap->Fill(dilepton_mom.Rapidity()); 

  //Fill lepton histograms 
  leadpt->Fill(lep1.Pt());
  secpt->Fill(lep2.Pt());
  leadeta->Fill(lep1.Eta());
  seceta->Fill(lep2.Eta());

  //boost everything in the Z frame
  TVector3 boost = dilepton_andphoton_mom.BoostVector();
  boost*=-1.;
  lep1.Boost(boost);
  lep2.Boost(boost);
  for (unsigned int ipho = 0; ipho < gammasMomenta.size(); ++ipho){
    gammasMomenta[ipho].Boost(boost);
  }
  std::sort(gammasMomenta.begin(), gammasMomenta.end(), HepMCValidationHelper::GreaterByE<TLorentzVector>);

  //fill gamma histograms
  if (gammasMomenta.size() != 0 && dilepton_andphoton_mom.M() > 50.) {
    gamma_energy->Fill(gammasMomenta.front().E());
    double dphi = lep1.DeltaR(gammasMomenta.front()) <  lep2.DeltaR(gammasMomenta.front()) ?
                  lep1.DeltaPhi(gammasMomenta.front()) : lep2.DeltaPhi(gammasMomenta.front());
    cos_theta_gamma_lepton->Fill(cos(dphi));
  } 

}//analyze