TtDecayChannelSelector.h

Go to the documentation of this file.

#include <vector>
#include <cstring>
#include <iostream>

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"

class TtDecayChannelSelector {

 public:

  enum {Elec=0, Muon=1, Tau =2};
  typedef std::vector<int> Decay;

  TtDecayChannelSelector(const edm::ParameterSet&);
  ~TtDecayChannelSelector();
  bool operator()(const reco::GenParticleCollection& parts, std::string inputType) const;

 private:

  unsigned int decayChannel() const;
  // return the check sum of all entries
  unsigned int checkSum(const Decay& vec) const;
  bool search(reco::GenParticleCollection::const_iterator& part, int pdgId, std::string& inputType) const;
  bool search(reco::GenParticle::const_iterator& part, int pdgId, std::string& inputType) const;
  bool tauDecay(const reco::Candidate&) const;
  unsigned int countProngs(const reco::Candidate& part) const;

 private:

  bool  invert_;  
  bool restrictTauDecays_;
  bool allowElectron_;
  bool allowMuon_;
  bool allow1Prong_;
  bool allow3Prong_;
  Decay decayBranchA_;
  Decay decayBranchB_;
  Decay allowedDecays_;
};

inline unsigned int
TtDecayChannelSelector::decayChannel() const
{
  unsigned int channel=0;
  if( std::count(decayBranchA_.begin(), decayBranchA_.end(), 1) > 0 ){ 
    ++channel; 
  }
  if( std::count(decayBranchB_.begin(), decayBranchB_.end(), 1) > 0 ){ 
    ++channel; 
  }
  return channel;
}

inline unsigned int
TtDecayChannelSelector::checkSum(const Decay& vec) const
{
  unsigned int sum=0;
  for(unsigned int d=0; d<vec.size(); ++d){
    sum+=vec[d];
  }
  return sum;
}