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;
}