BPHOniaToMuMuBuilder.h

Go to the documentation of this file.

#ifndef HeavyFlavorAnalysis_SpecificDecay_BPHOniaToMuMuBuilder_h
#define HeavyFlavorAnalysis_SpecificDecay_BPHOniaToMuMuBuilder_h

//----------------------
// Base Class Headers --
//----------------------

//------------------------------------
// 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 BPHMuonPtSelect;
class BPHMuonEtaSelect;
class BPHChi2Select;
class BPHMassSelect;
class BPHRecoSelect;
class BPHMomentumSelect;
class BPHVertexSelect;

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

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

class BPHOniaToMuMuBuilder {
public:
  enum oniaType { Phi, Psi1, Psi2, Ups, Ups1, Ups2, Ups3 };

  BPHOniaToMuMuBuilder(const edm::EventSetup& es,
                       const BPHRecoBuilder::BPHGenericCollection* muPosCollection,
                       const BPHRecoBuilder::BPHGenericCollection* muNegCollection);

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

  virtual ~BPHOniaToMuMuBuilder();

  std::vector<BPHPlusMinusConstCandPtr> build();

  std::vector<BPHPlusMinusConstCandPtr> getList(oniaType type,
                                                BPHRecoSelect* dSel = nullptr,
                                                BPHMomentumSelect* mSel = nullptr,
                                                BPHVertexSelect* vSel = nullptr,
                                                BPHFitSelect* kSel = nullptr);

  BPHPlusMinusConstCandPtr getOriginalCandidate(const BPHRecoCandidate& cand);

  void setPtMin(oniaType type, double pt);
  void setEtaMax(oniaType type, double eta);
  void setMassMin(oniaType type, double m);
  void setMassMax(oniaType type, double m);
  void setProbMin(oniaType type, double p);
  void setConstr(oniaType type, double mass, double sigma);

  double getPtMin(oniaType type) const;
  double getEtaMax(oniaType type) const;
  double getMassMin(oniaType type) const;
  double getMassMax(oniaType type) const;
  double getProbMin(oniaType type) const;
  double getConstrMass(oniaType type) const;
  double getConstrSigma(oniaType type) const;

private:
  std::string muPosName;
  std::string muNegName;

  const edm::EventSetup* evSetup;
  const BPHRecoBuilder::BPHGenericCollection* posCollection;
  const BPHRecoBuilder::BPHGenericCollection* negCollection;

  struct OniaParameters {
    BPHMuonPtSelect* ptSel;
    BPHMuonEtaSelect* etaSel;
    BPHMassSelect* massSel;
    BPHChi2Select* chi2Sel;
    double mass;
    double sigma;
    bool updated;
  };
  bool updated;

  std::map<oniaType, OniaParameters> oniaPar;
  std::map<oniaType, std::vector<BPHPlusMinusConstCandPtr>> oniaList;
  std::vector<BPHPlusMinusConstCandPtr> fullList;

  void setNotUpdated();
  void setParameters(oniaType type,
                     double ptMin,
                     double etaMax,
                     double massMin,
                     double massMax,
                     double probMin,
                     double mass,
                     double sigma);
  void extractList(oniaType type);
};

#endif