TtEventPartons.h

Go to the documentation of this file.

#ifndef TtEventPartons_h
#define TtEventPartons_h

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

#include <vector>

namespace reco {
  class Candidate;
}
class TtGenEvent;

class TtEventPartons {
public:
  TtEventPartons(){};
  virtual ~TtEventPartons(){};

  virtual std::vector<const reco::Candidate*> vec(const TtGenEvent& genEvt) = 0;

  void expand(std::vector<int>& vec);

protected:
  reco::Candidate* dummyCandidatePtr() const {
    return new reco::GenParticle(0, reco::Particle::LorentzVector(), reco::Particle::Point(), 0, 0, false);
  };

  void prune(std::vector<const reco::Candidate*>& vec);

  std::vector<bool> ignorePartons_;
};

#endif