---

GeneratorTau.h

Go to the documentation of this file.

#ifndef RECOTAUTAG_TAUMVADISCRIMINATOR_GENERATORTAU
#define RECOTAUTAG_TAUMVADISCRIMINATOR_GENERATORTAU


/*
 * Class GeneratorTau
 *
 * Tool for retrieving visible decay products and determining PDG-style tau decay mode 
 * from Pythia output (in CMS genParticle format)
 *
 * Author: Evan K. Friis, UC Davis; friis@physics.ucdavis.edu
 *
 * with code and contributions from Ricardo Vasquez Sierra and Christian Veelken, UC Davis
 *
 */

#include "FWCore/ServiceRegistry/interface/Service.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/Candidate/interface/Particle.h"
#include "PhysicsTools/Utilities/interface/Angle.h"
#include "PhysicsTools/Utilities/interface/deltaR.h"
#include <vector>

using namespace std;

typedef math::XYZTLorentzVector LorentzVector;

class GeneratorTau : public reco::GenParticle {
   public:

      enum tauDecayModeEnum {kElectron, kMuon, 
         kOneProng0pi0, kOneProng1pi0, kOneProng2pi0,
         kThreeProng0pi0, kThreeProng1pi0,
         kOther, kUndefined};

      //default constructor
      GeneratorTau(const reco::GenParticle& input):GenParticle(input){};
      GeneratorTau();

      ~GeneratorTau(){};

      vector<const reco::Candidate*>         getGenChargedPions() const;
      vector<const reco::Candidate*>         getGenGammas() const;
      vector<const reco::Candidate*>         getGenNeutralPions() const;
      vector<const reco::Candidate*>         getStableDecayProducts() const;
      vector<const reco::Candidate*>         getGenNu() const;

      tauDecayModeEnum                          getDecayType() const {return theDecayMode_;};
      bool                                      isFinalStateTau() const {return aFinalStateTau_;};

      void                                      init(); //called to update class after downcasting

      vector<LorentzVector>                     getChargedPions() const;
      vector<LorentzVector>                     getGammas() const;
      LorentzVector                             getVisibleFourVector() const;
      vector<LorentzVector>                     getVisibleFourVectors() const;
      const reco::Candidate*                    getLeadTrack() const;
      const reco::GenParticle*         findLeadTrack();

      float                                    getVisNuAngle() const;
      float                                    getOpeningAngle(const vector<const reco::GenParticle*>& aCollection) const;
      float                                    getChargedOpeningAngle() const;
      float                                    getGammaOpeningAngle() const;

      void                                      decayToPDGClassification(const reco::GenParticle*, std::vector<const reco::GenParticle *>&);
      void                                      computeStableDecayProducts(const reco::GenParticle*, std::vector<const reco::GenParticle *>&);
      tauDecayModeEnum                          computeDecayMode(const reco::GenParticle*);
      LorentzVector                             convertHepMCFourVec(const reco::GenParticle* theParticle);
      vector<LorentzVector>                     convertMCVectorToLorentzVectors(const vector<const reco::GenParticle*>& theList) const;

   private:

      Angle<LorentzVector>                              angleFinder;
      DeltaR<LorentzVector>                             deltaRFinder;


      //only fill these with stable particles
      vector<const reco::GenParticle*>         visibleDecayProducts_;
      vector<const reco::GenParticle*>         genChargedPions_;
      vector<const reco::GenParticle*>         genNeutralPions_;
      vector<const reco::GenParticle*>         genGammas_;
      vector<const reco::GenParticle*>         stableDecayProducts_;
      vector<const reco::GenParticle*>         genNus_; 
      const reco::GenParticle*                 theLeadTrack_;

      tauDecayModeEnum                                  theDecayMode_;
      int                                               aFinalStateTau_;

};


#endif

---