d4/dad/BPHDecayToTkpTknSymChargeBuilder_8h_source.html

#ifndef HeavyFlavorAnalysis_SpecificDecay_BPHDecayToTkpTknSymChargeBuilder_h

#define HeavyFlavorAnalysis_SpecificDecay_BPHDecayToTkpTknSymChargeBuilder_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 "FWCore/Framework/interface/Event.h"


class BPHParticlePtSelect;

class BPHParticleEtaSelect;

class BPHChi2Select;

class BPHMassSelect;


//---------------

// C++ Headers --

//---------------

#include <string>

#include <vector>


//              ---------------------

//              -- Class Interface --

//              ---------------------


class BPHDecayToTkpTknSymChargeBuilder : public BPHDecayGenericBuilder {

public:

  BPHDecayToTkpTknSymChargeBuilder(const edm::EventSetup& es,

                                   const std::string& daug1Name,

                                   double daug1Mass,

                                   double daug1Sigma,

                                   const std::string& daug2Name,

                                   double daug2Mass,

                                   double daug2Sigma,

                                   const BPHRecoBuilder::BPHGenericCollection* posCollection,

                                   const BPHRecoBuilder::BPHGenericCollection* negCollection,

                                   double expectedMass);


  // deleted copy constructor and assignment operator

  BPHDecayToTkpTknSymChargeBuilder(const BPHDecayToTkpTknSymChargeBuilder& x) = delete;

  BPHDecayToTkpTknSymChargeBuilder& operator=(const BPHDecayToTkpTknSymChargeBuilder& x) = delete;


  ~BPHDecayToTkpTknSymChargeBuilder() override;


  std::vector<BPHPlusMinusConstCandPtr> build();


  void setTrk1PtMin(double pt);

  void setTrk2PtMin(double pt);

  void setTrk1EtaMax(double eta);

  void setTrk2EtaMax(double eta);

  void setDzMax(double dz);


  double getTrk1PtMin() const { return pt1Min; }

  double getTrk2PtMin() const { return pt2Min; }

  double getTrk1EtaMax() const { return eta1Max; }

  double getTrk2EtaMax() const { return eta2Max; }

  double getDzMax() const { return dzMax; }


private:

  std::string d1Name;

  double d1Mass;

  double d1Sigma;

  std::string d2Name;

  double d2Mass;

  double d2Sigma;

  double eMass;


  const BPHRecoBuilder::BPHGenericCollection* pCollection;

  const BPHRecoBuilder::BPHGenericCollection* nCollection;


  double pt1Min;

  double pt2Min;

  double eta1Max;

  double eta2Max;

  double dzMax;


  std::vector<BPHPlusMinusConstCandPtr> recList;


  class Particle {

  public:

    Particle(const reco::Candidate* c, const reco::Track* tk, double x, double y, double z, double f, double g)

        : cand(c), track(tk), px(x), py(y), pz(z), e1(f), e2(g) {}

    const reco::Candidate* cand;

    const reco::Track* track;

    double px;

    double py;

    double pz;

    double e1;

    double e2;

  };

  void addParticle(const BPHRecoBuilder::BPHGenericCollection* collection, int charge, std::vector<Particle*>& list);

};


#endif