#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 ()

virtual	~EDProducer ()

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

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

virtual	~ProducerBase ()

Static Public Member Functions
static double	flat ()

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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::InputTag	gensrc_

int	Ipol_

bool	isLHPDFConfigured_

bool	isReco_

bool	isTauolaConfigured_

std::string	LHAPDFname_

int	MotherPDGID_

int	nonSM2_

int	nonSMN_

double	roundOff_

Static Private Attributes
static CLHEP::HepRandomEngine *	fRandomEngine = nullptr

static bool	isTauSpinnerConfigure =false

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

typedef WorkerT< EDProducer >	WorkerType

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

Protected Member Functions inherited from edm::EDProducer
CurrentProcessingContext const *	currentContext () const

Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)

Detailed Description

Definition at line 37 of file TauSpinnerCMS.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 22 of file TauSpinnerCMS.cc.

                                                        :
   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");
 
 }

virtual TauSpinnerCMS::~TauSpinnerCMS ( )

inlinevirtual

Definition at line 44 of file TauSpinnerCMS.h.

44 {}

Member Function Documentation

void TauSpinnerCMS::beginJob ( void )

virtual

Reimplemented from edm::EDProducer.

Definition at line 43 of file TauSpinnerCMS.cc.

References CMSEnergy_, edm::hlt::Exception, flat(), fRandomEngine, edm::RandomNumberGenerator::getEngine(), Ipol_, edm::Service< T >::isAvailable(), isLHPDFConfigured_, isTauolaConfigured_, isTauSpinnerConfigure, LHAPDFname_, nonSM2_, and nonSMN_.

 {
   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;
   }
   fRandomEngine = &rng->getEngine();
 
   if(!isTauolaConfigured_){
     Tauolapp::Tauola::setRandomGenerator(TauSpinnerCMS::flat);
     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 166 of file TauSpinnerCMS.cc.

166 {}

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

virtual

Definition at line 164 of file TauSpinnerCMS.cc.

164 {}

double TauSpinnerCMS::flat ( void )

static

Definition at line 255 of file TauSpinnerCMS.cc.

References edm::hlt::Exception, and fRandomEngine.

Referenced by beginJob().

                           {
   if ( !fRandomEngine ) {
     throw cms::Exception("LogicError")
       << "TauSpinnerCMS::flat: Attempt to generate random number when engine pointer is null\n"
       << "This might mean that the code was modified to generate a random number outside the\n"
       << "event and beginLuminosityBlock methods, which is not allowed.\n";
   }
   return fRandomEngine->flat();
 }

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

private

Definition at line 223 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 243 of file TauSpinnerCMS.cc.

References 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 233 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 72 of file TauSpinnerCMS.cc.

References abs, gather_cfg::cout, edm::Event::getByLabel(), i, isReco_, MotherPDGID_, edm::Event::put(), readParticlesFromHepMC(), readParticlesfromReco(), hitfit::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.getByLabel( "generator", 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 168 of file TauSpinnerCMS.cc.

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

Referenced by produce().

                                                                                                         {
   edm::Handle<reco::GenParticleCollection> genParticles;
   e.getByLabel(gensrc_, 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 57 of file TauSpinnerCMS.h.

Referenced by beginJob().

CLHEP::HepRandomEngine * TauSpinnerCMS::fRandomEngine = nullptr

staticprivate

Definition at line 75 of file TauSpinnerCMS.h.

Referenced by beginJob(), and flat().

edm::InputTag TauSpinnerCMS::gensrc_

private

Definition at line 58 of file TauSpinnerCMS.h.

Referenced by readParticlesfromReco().

int TauSpinnerCMS::Ipol_

private

Definition at line 59 of file TauSpinnerCMS.h.

Referenced by beginJob().

bool TauSpinnerCMS::isLHPDFConfigured_

private

Definition at line 55 of file TauSpinnerCMS.h.

Referenced by beginJob().

bool TauSpinnerCMS::isReco_

private

Definition at line 53 of file TauSpinnerCMS.h.

Referenced by produce().

bool TauSpinnerCMS::isTauolaConfigured_

private

Definition at line 54 of file TauSpinnerCMS.h.

Referenced by beginJob().

bool TauSpinnerCMS::isTauSpinnerConfigure =false

staticprivate

Definition at line 60 of file TauSpinnerCMS.h.

Referenced by beginJob().

std::string TauSpinnerCMS::LHAPDFname_

private

Definition at line 56 of file TauSpinnerCMS.h.

Referenced by beginJob().

int TauSpinnerCMS::MotherPDGID_

private

Definition at line 59 of file TauSpinnerCMS.h.

Referenced by produce().

int TauSpinnerCMS::nonSM2_

private

Definition at line 59 of file TauSpinnerCMS.h.

Referenced by beginJob().

int TauSpinnerCMS::nonSMN_

private

Definition at line 59 of file TauSpinnerCMS.h.

Referenced by beginJob().

double TauSpinnerCMS::roundOff_

private

Definition at line 74 of file TauSpinnerCMS.h.

Referenced by produce().

Public Member Functions

Static 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