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