BPHDecayToV0Builder.h

Go to the documentation of this file.

#ifndef HeavyFlavorAnalysis_SpecificDecay_BPHDecayToV0Builder_h
#define HeavyFlavorAnalysis_SpecificDecay_BPHDecayToV0Builder_h

//----------------------
// Base Class Headers --
//----------------------
#include "HeavyFlavorAnalysis/SpecificDecay/interface/BPHDecayGenericBuilder.h"

//------------------------------------
// Collaborating Class Declarations --
//------------------------------------
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoBuilder.h"
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHRecoCandidate.h"
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHPlusMinusCandidate.h"

#include "DataFormats/Candidate/interface/VertexCompositeCandidate.h"
#include "HeavyFlavorAnalysis/RecoDecay/interface/BPHVertexCompositePtrCandidate.h"

#include "FWCore/Framework/interface/Event.h"

//---------------
// C++ Headers --
//---------------
#include <string>
#include <vector>
#include <map>

//              ---------------------
//              -- Class Interface --
//              ---------------------

class BPHDecayToV0Builder : public BPHDecayGenericBuilder {
public:
  enum v0Type { VertexCompositeCandidate, VertexCompositePtrCandidate };
  struct V0Info {
    v0Type type;
    const void* v0;
  };

  BPHDecayToV0Builder(const edm::EventSetup& es,
                      const std::string& d1Name,
                      const std::string& d2Name,
                      const BPHRecoBuilder::BPHGenericCollection* d1Collection,
                      const BPHRecoBuilder::BPHGenericCollection* d2Collection);
  BPHDecayToV0Builder(const edm::EventSetup& es,
                      const std::string& d1Name,
                      const std::string& d2Name,
                      const std::vector<reco::VertexCompositeCandidate>* v0Collection,
                      const std::string& searchList = "cfp");
  BPHDecayToV0Builder(const edm::EventSetup& es,
                      const std::string& d1Name,
                      const std::string& d2Name,
                      const std::vector<reco::VertexCompositePtrCandidate>* vpCollection,
                      const std::string& searchList = "cfp");

  // deleted copy constructor and assignment operator
  BPHDecayToV0Builder(const BPHDecayToV0Builder& x) = delete;
  BPHDecayToV0Builder& operator=(const BPHDecayToV0Builder& x) = delete;

  ~BPHDecayToV0Builder() override;

  std::vector<BPHPlusMinusConstCandPtr> build();

  void setPtMin(double pt);
  void setEtaMax(double eta);

  double getPtMin() const;
  double getEtaMax() const;

protected:
  std::vector<BPHPlusMinusConstCandPtr> cList;

  std::string p1Name;
  std::string p2Name;

  const BPHRecoBuilder::BPHGenericCollection* p1Collection;
  const BPHRecoBuilder::BPHGenericCollection* p2Collection;
  const std::vector<reco::VertexCompositeCandidate>* vCollection;
  const std::vector<reco::VertexCompositePtrCandidate>* rCollection;
  std::string sList;

  double ptMin;
  double etaMax;

  std::map<const BPHRecoCandidate*, const V0Info*> v0Map;

  virtual void buildFromBPHGenericCollection() = 0;
  template <class T>
  void buildFromV0(const T* v0Collection, v0Type type);
  virtual BPHPlusMinusCandidatePtr buildCandidate(const reco::Candidate* c1,
                                                  const reco::Candidate* c2,
                                                  const void* v0,
                                                  v0Type type) = 0;
  void v0Clear();
};

#endif