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 );

  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:

  // private copy and assigment constructors
  BPHOniaToMuMuBuilder           ( const BPHOniaToMuMuBuilder& x ) = delete;
  BPHOniaToMuMuBuilder& operator=( const BPHOniaToMuMuBuilder& x ) = delete;

  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