#include <TauSpinnerCMS.h>

Inheritance diagram for TauSpinnerCMS:

Public Member Functions
virtual void	beginJob ()

virtual void	endJob ()

virtual void	endRun (const edm::Run &, const edm::EventSetup &)

virtual void	produce (edm::Event &, const edm::EventSetup &)

	TauSpinnerCMS (const edm::ParameterSet &)

virtual	~TauSpinnerCMS ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
void	GetLastSelf (const reco::GenParticle *Particle)

void	GetRecoDaughters (const reco::GenParticle *Particle, std::vector< TauSpinner::SimpleParticle > &daughters, int parentpdgid)

bool	isFirst (const reco::GenParticle *Particle)

int	readParticlesfromReco (edm::Event &e, TauSpinner::SimpleParticle &X, TauSpinner::SimpleParticle &tau, TauSpinner::SimpleParticle &tau2, std::vector< TauSpinner::SimpleParticle > &tau_daughters, std::vector< TauSpinner::SimpleParticle > &tau2_daughters)

Private Attributes
double	CMSEnergy_

edm::EDGetTokenT < edm::HepMCProduct >	EvtHandleToken_

edm::InputTag	gensrc_

edm::EDGetTokenT < reco::GenParticleCollection >	gensrcToken_

int	Ipol_

bool	isLHPDFConfigured_

bool	isReco_

bool	isTauolaConfigured_

std::string	LHAPDFname_

int	MotherPDGID_

int	nonSM2_

int	nonSMN_

double	roundOff_

Static Private Attributes
static bool	isTauSpinnerConfigure =false

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 35 of file TauSpinnerCMS.h.

Constructor & Destructor Documentation

TauSpinnerCMS::TauSpinnerCMS ( const edm::ParameterSet & pset )

explicit

Definition at line 33 of file TauSpinnerCMS.cc.

References decayRandomEngine, EvtHandleToken_, edm::hlt::Exception, gensrc_, gensrcToken_, and isReco_.

                                                        :
   isReco_(pset.getParameter<bool>("isReco"))
   ,isTauolaConfigured_(pset.getParameter<bool>("isTauolaConfigured" ))
   ,isLHPDFConfigured_(pset.getParameter<bool>("isLHPDFConfigured" ))
   ,LHAPDFname_(pset.getUntrackedParameter("LHAPDFname",(string)("MSTW2008nnlo90cl.LHgrid")))
   ,CMSEnergy_(pset.getParameter<double>("CMSEnergy"))//GeV
   ,gensrc_(pset.getParameter<edm::InputTag>("gensrc"))
   ,MotherPDGID_(pset.getUntrackedParameter("MotherPDGID",(int)(-1)))
   ,Ipol_(pset.getUntrackedParameter("Ipol",(int)(0)))
   ,nonSM2_(pset.getUntrackedParameter("nonSM2",(int)(0)))
   ,nonSMN_(pset.getUntrackedParameter("nonSMN",(int)(0)))
   ,roundOff_(pset.getUntrackedParameter("roundOff",(double)(0.01)))
 {
   produces<bool>("TauSpinnerWTisValid").setBranchAlias("TauSpinnerWTisValid");
   produces<double>("TauSpinnerWT").setBranchAlias("TauSpinnerWT");
   produces<double>("TauSpinnerWTFlip").setBranchAlias("TauSpinnerWTFlip");
   produces<double>("TauSpinnerWThplus").setBranchAlias("TauSpinnerWThplus");
   produces<double>("TauSpinnerWThminus").setBranchAlias("TauSpinnerWThminus");
 
   if(!isReco_) EvtHandleToken_ = consumes<HepMCProduct>(gensrc_);
   if(isReco_) gensrcToken_= consumes<reco::GenParticleCollection>(gensrc_);
 
   Service<RandomNumberGenerator> rng;
   if(!rng.isAvailable()) {
     throw cms::Exception("Configuration")
       << "The RandomNumberProducer module requires the RandomNumberGeneratorService\n"
           "which appears to be absent.  Please add that service to your configuration\n"
       "or remove the modules that require it." << std::endl;
   }
   decayRandomEngine = &rng->getEngine();
   
 }

virtual TauSpinnerCMS::~TauSpinnerCMS ( )

inlinevirtual

Definition at line 42 of file TauSpinnerCMS.h.

42 {} // no need to delete ROOT stuff

Member Function Documentation

void TauSpinnerCMS::beginJob ( void )

virtual

Reimplemented from edm::EDProducer.

Definition at line 66 of file TauSpinnerCMS.cc.

References CMSEnergy_, Ipol_, isLHPDFConfigured_, isTauolaConfigured_, isTauSpinnerConfigure, LHAPDFname_, nonSM2_, and nonSMN_.

 {
   if(!isTauolaConfigured_){
     Tauolapp::Tauola::initialize();
   }
   if(!isLHPDFConfigured_){
     LHAPDF::initPDFSetByName(LHAPDFname_);   
   }
   if(!isTauSpinnerConfigure){
     isTauSpinnerConfigure=true;
     bool Ipp = true;  // for pp collisions 
     // Initialize TauSpinner
     //Ipol - polarization of input sample
     //nonSM2 - nonstandard model calculations
     //nonSMN
     TauSpinner::initialize_spinner(Ipp,Ipol_,nonSM2_,nonSMN_,CMSEnergy_);
   }
 }

void TauSpinnerCMS::endJob ( void )

virtual

Reimplemented from edm::EDProducer.

Definition at line 179 of file TauSpinnerCMS.cc.

179 {}

void TauSpinnerCMS::endRun	(	const edm::Run &	r,
		const edm::EventSetup &
	)

virtual

Reimplemented from edm::EDProducer.

Definition at line 177 of file TauSpinnerCMS.cc.

177 {}

void TauSpinnerCMS::GetLastSelf ( const reco::GenParticle * Particle )

private

Definition at line 236 of file TauSpinnerCMS.cc.

References reco::CompositeRefCandidateT< D >::daughter(), i, reco::CompositeRefCandidateT< D >::numberOfDaughters(), and reco::LeafCandidate::pdgId().

Referenced by readParticlesfromReco().

                                                               {
   for (unsigned int i=0; i< Particle->numberOfDaughters(); i++){
     const reco::GenParticle *dau=static_cast<const reco::GenParticle*>(Particle->daughter(i));
     if(Particle->pdgId()==dau->pdgId()){
       Particle=dau;
       GetLastSelf(Particle);  
     }
   }
 }

void TauSpinnerCMS::GetRecoDaughters	(	const reco::GenParticle *	Particle,
		std::vector< TauSpinner::SimpleParticle > &	daughters,
		int	parentpdgid
	)

private

Definition at line 256 of file TauSpinnerCMS.cc.

References funct::abs(), reco::CompositeRefCandidateT< D >::daughter(), i, reco::CompositeRefCandidateT< D >::numberOfDaughters(), reco::LeafCandidate::p4(), and reco::LeafCandidate::pdgId().

Referenced by readParticlesfromReco().

                                                                                                                          {
   if( Particle->numberOfDaughters()==0 || abs(Particle->pdgId())==111){
     SimpleParticle tp(Particle->p4().Px(), Particle->p4().Py(), Particle->p4().Pz(), Particle->p4().E(), Particle->pdgId());
     daughters.push_back(tp);
     return;
   }
   for (unsigned int i=0; i< Particle->numberOfDaughters(); i++){
     const reco::Candidate *dau=Particle->daughter(i);
     GetRecoDaughters(static_cast<const reco::GenParticle*>(dau),daughters,Particle->pdgId());
   }
 }

bool TauSpinnerCMS::isFirst ( const reco::GenParticle * Particle )

private

Definition at line 246 of file TauSpinnerCMS.cc.

References i, reco::CompositeRefCandidateT< D >::mother(), reco::CompositeRefCandidateT< D >::numberOfMothers(), and reco::LeafCandidate::pdgId().

Referenced by readParticlesfromReco().

                                                           {
   for (unsigned int i=0; i< Particle->numberOfMothers(); i++){
     const reco::GenParticle *moth=static_cast<const reco::GenParticle*>(Particle->mother(i));
     if(Particle->pdgId()==moth->pdgId()){
       return false;
     }
   }
   return true;
 }

void TauSpinnerCMS::produce	(	edm::Event &	e,
		const edm::EventSetup &	iSetup
	)

virtual

Implements edm::EDProducer.

Definition at line 85 of file TauSpinnerCMS.cc.

References funct::abs(), gather_cfg::cout, EvtHandleToken_, edm::Event::getByToken(), i, isReco_, MotherPDGID_, edm::Event::put(), readParticlesFromHepMC(), readParticlesfromReco(), reco::return(), roundOff_, metsig::tau, and X.

                                                                     {
   double WT=1.0;
   double WTFlip=1.0;
   double polSM=-999; //range [-1,1]
   SimpleParticle X, tau, tau2;
   std::vector<SimpleParticle> tau_daughters, tau_daughters2;
   int stat(0);
   if(isReco_){
     stat=readParticlesfromReco(e,X,tau,tau2,tau_daughters,tau_daughters2);
   }
   else{
     Handle< HepMCProduct > EvtHandle ;
     e.getByToken(EvtHandleToken_, EvtHandle ) ;
     const HepMC::GenEvent* Evt = EvtHandle->GetEvent() ;
     stat=readParticlesFromHepMC(Evt,X,tau,tau2,tau_daughters,tau_daughters2);
   }  
   if(MotherPDGID_<0 || abs(X.pdgid())==MotherPDGID_){
     if(stat!=1){
       // Determine the weight      
       if( abs(X.pdgid())==24 ||  abs(X.pdgid())==37 ){
         TLorentzVector tau_1r(0,0,0,0);
         TLorentzVector tau_1(tau.px(),tau.py(),tau.pz(),tau.e());
         for(unsigned int i=0; i<tau_daughters.size();i++){
           tau_1r+=TLorentzVector(tau_daughters.at(i).px(),tau_daughters.at(i).py(),tau_daughters.at(i).pz(),tau_daughters.at(i).e());
         }
     if(fabs(tau_1r.M()-tau_1.M())<roundOff_){
       WT = TauSpinner::calculateWeightFromParticlesWorHpn(X, tau, tau2, tau_daughters); // note that tau2 is tau neutrino
       polSM=getTauSpin();
       WTFlip=(2.0-WT)/WT;
     }
       }
       else if( X.pdgid()==25 || X.pdgid()==36 || X.pdgid()==22 || X.pdgid()==23 ){
     TLorentzVector tau_1r(0,0,0,0), tau_2r(0,0,0,0);
     TLorentzVector tau_1(tau.px(),tau.py(),tau.pz(),tau.e()), tau_2(tau2.px(),tau2.py(),tau2.pz(),tau2.e());
     for(unsigned int i=0; i<tau_daughters.size();i++){
       tau_1r+=TLorentzVector(tau_daughters.at(i).px(),tau_daughters.at(i).py(),tau_daughters.at(i).pz(),tau_daughters.at(i).e());
     }
     for(unsigned int i=0; i<tau_daughters2.size();i++){
       tau_2r+=TLorentzVector(tau_daughters2.at(i).px(),tau_daughters2.at(i).py(),tau_daughters2.at(i).pz(),tau_daughters2.at(i).e());
     }
 
         if(fabs(tau_1r.M()-tau_1.M())<roundOff_ && fabs(tau_2r.M()-tau_2.M())<roundOff_){
           WT = TauSpinner::calculateWeightFromParticlesH(X, tau, tau2, tau_daughters,tau_daughters2);
       //std::cout << "WT " << WT << std::endl;
           polSM=getTauSpin();
           if(X.pdgid()==25 || X.pdgid()==22 || X.pdgid()==23 ){
             if(X.pdgid()==25) X.setPdgid(23);
             if( X.pdgid()==22 || X.pdgid()==23 ) X.setPdgid(25);
 
             double WTother=TauSpinner::calculateWeightFromParticlesH(X, tau, tau2, tau_daughters,tau_daughters2);
             WTFlip=WTother/WT;
           }
         }
       }
       else{
     cout<<"TauSpinner: WARNING: Unexpected PDG for tau mother: "<<X.pdgid()<<endl;
       }
     }
   }
   bool isValid=true;
   if(!(0<=WT && WT<10)){isValid=false; WT=1.0; WTFlip=1.0;}
   std::auto_ptr<bool> TauSpinnerWeightisValid(new bool);
   *TauSpinnerWeightisValid =isValid;
   e.put(TauSpinnerWeightisValid,"TauSpinnerWTisValid");
 
   // regular weight
   std::auto_ptr<double> TauSpinnerWeight(new double);
   *TauSpinnerWeight =WT;    
   e.put(TauSpinnerWeight,"TauSpinnerWT");  
   
   // flipped weight (ie Z->H or H->Z)
   std::auto_ptr<double> TauSpinnerWeightFlip(new double);
   *TauSpinnerWeightFlip =WTFlip;
   e.put(TauSpinnerWeightFlip,"TauSpinnerWTFlip");
   
   // h+ polarization
   double WThplus=WT;
   if(polSM<0.0 && polSM!=-999 && isValid) WThplus=0; 
   std::auto_ptr<double> TauSpinnerWeighthplus(new double);
   *TauSpinnerWeighthplus = WThplus;
   e.put(TauSpinnerWeighthplus,"TauSpinnerWThplus");
 
   // h- polarization
   double WThminus=WT;
   if(polSM>0.0&& polSM!=-999 && isValid) WThminus=0;
   std::auto_ptr<double> TauSpinnerWeighthminus(new double);
   *TauSpinnerWeighthminus = WThminus;
   e.put(TauSpinnerWeighthminus,"TauSpinnerWThminus");
   
   return ;
 }  

int TauSpinnerCMS::readParticlesfromReco	(	edm::Event &	e,
		TauSpinner::SimpleParticle &	X,
		TauSpinner::SimpleParticle &	tau,
		TauSpinner::SimpleParticle &	tau2,
		std::vector< TauSpinner::SimpleParticle > &	tau_daughters,
		std::vector< TauSpinner::SimpleParticle > &	tau2_daughters
	)

private

Definition at line 181 of file TauSpinnerCMS.cc.

References funct::abs(), reco::Candidate::daughter(), genParticleCandidates2GenParticles_cfi::genParticles, gensrcToken_, edm::Event::getByToken(), GetLastSelf(), GetRecoDaughters(), i, isFirst(), NULL, reco::LeafCandidate::p4(), reco::Candidate::pdgId(), and reco::LeafCandidate::pdgId().

Referenced by produce().

                                                                                                         {
   edm::Handle<reco::GenParticleCollection> genParticles;
   e.getByToken(gensrcToken_, genParticles);
   for(reco::GenParticleCollection::const_iterator itr = genParticles->begin(); itr!= genParticles->end(); ++itr){
     int pdgid=abs(itr->pdgId());
     if(pdgid==24 || pdgid==37 || pdgid ==25 || pdgid==36 || pdgid==22 || pdgid==23 ){
       const reco::GenParticle *hx=&(*itr);
       if(!isFirst(hx)) continue;
       GetLastSelf(hx);
       const reco::GenParticle *recotau1=NULL;
       const reco::GenParticle *recotau2=NULL;
       unsigned int ntau(0),ntauornu(0);
       for(unsigned int i=0; i<itr->numberOfDaughters(); i++){
     const reco::Candidate *dau=itr->daughter(i);
     if(abs(dau->pdgId())!=pdgid){
       if(abs(dau->pdgId())==15 || abs(dau->pdgId())==16){
         if(ntau==0 && abs(dau->pdgId())==15){
           recotau1=static_cast<const reco::GenParticle*>(dau);
           GetLastSelf(recotau1);
           ntau++;
         }
         else if((ntau==1 && abs(dau->pdgId())==15) || abs(dau->pdgId())==16){
           recotau2=static_cast<const reco::GenParticle*>(dau);
           if(abs(dau->pdgId())==15){ntau++;GetLastSelf(recotau2);}
         }
         ntauornu++;
       }
     }
       }
       if((ntau==2 && ntauornu==2) || (ntau==1 && ntauornu==2)){
     X.setPx(itr->p4().Px());
     X.setPy(itr->p4().Py());
     X.setPz(itr->p4().Pz());
     X.setE (itr->p4().E());
     X.setPdgid(itr->pdgId());
     tau.setPx(recotau1->p4().Px());
     tau.setPy(recotau1->p4().Py());
     tau.setPz(recotau1->p4().Pz());
     tau.setE (recotau1->p4().E());
         tau.setPdgid(recotau1->pdgId());
     GetRecoDaughters(recotau1,tau_daughters,recotau1->pdgId());
     tau2.setPx(recotau2->p4().Px());
         tau2.setPy(recotau2->p4().Py());
         tau2.setPz(recotau2->p4().Pz());
         tau2.setE (recotau2->p4().E());
         tau2.setPdgid(recotau2->pdgId());
     if(ntau==2)GetRecoDaughters(recotau2,tau2_daughters,recotau2->pdgId());
     return 0;
       }
     }
   }
   return 1;
 }

Member Data Documentation

double TauSpinnerCMS::CMSEnergy_

private

Definition at line 55 of file TauSpinnerCMS.h.

Referenced by beginJob().

edm::EDGetTokenT<edm::HepMCProduct> TauSpinnerCMS::EvtHandleToken_

private

Definition at line 60 of file TauSpinnerCMS.h.

Referenced by produce(), and TauSpinnerCMS().

edm::InputTag TauSpinnerCMS::gensrc_

private

Definition at line 56 of file TauSpinnerCMS.h.

Referenced by TauSpinnerCMS().

edm::EDGetTokenT<reco::GenParticleCollection> TauSpinnerCMS::gensrcToken_

private

Definition at line 61 of file TauSpinnerCMS.h.

Referenced by readParticlesfromReco(), and TauSpinnerCMS().

int TauSpinnerCMS::Ipol_

private

Definition at line 57 of file TauSpinnerCMS.h.

Referenced by beginJob().

bool TauSpinnerCMS::isLHPDFConfigured_

private

Definition at line 53 of file TauSpinnerCMS.h.

Referenced by beginJob().

bool TauSpinnerCMS::isReco_

private

Definition at line 51 of file TauSpinnerCMS.h.

Referenced by produce(), and TauSpinnerCMS().

bool TauSpinnerCMS::isTauolaConfigured_

private

Definition at line 52 of file TauSpinnerCMS.h.

Referenced by beginJob().

bool TauSpinnerCMS::isTauSpinnerConfigure =false

staticprivate

Definition at line 58 of file TauSpinnerCMS.h.

Referenced by beginJob().

std::string TauSpinnerCMS::LHAPDFname_

private

Definition at line 54 of file TauSpinnerCMS.h.

Referenced by beginJob().

int TauSpinnerCMS::MotherPDGID_

private

Definition at line 57 of file TauSpinnerCMS.h.

Referenced by produce().

int TauSpinnerCMS::nonSM2_

private

Definition at line 57 of file TauSpinnerCMS.h.

Referenced by beginJob().

int TauSpinnerCMS::nonSMN_

private

Definition at line 57 of file TauSpinnerCMS.h.

Referenced by beginJob().

double TauSpinnerCMS::roundOff_

private

Definition at line 75 of file TauSpinnerCMS.h.

Referenced by produce().

Public Member Functions

Private Member Functions

Private Attributes

Static Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation