BackgroundHandler.h

Go to the documentation of this file.

#ifndef BackgroundHandler_h
#define BackgroundHandler_h

#include "MuonAnalysis/MomentumScaleCalibration/interface/Functions.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include <vector>
#include "MuonAnalysis/MomentumScaleCalibration/interface/MassWindow.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "SimDataFormats/Track/interface/SimTrack.h"
#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include <CLHEP/Vector/LorentzVector.h>
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

// Unit test for testing BackgroundHandler
class TestBackgroundHandler;

class BackgroundHandler {
  // For tests
  friend class TestBackgroundHandler;

public:
  BackgroundHandler(const std::vector<int>& identifiers,
                    const std::vector<double>& leftWindowBorders,
                    const std::vector<double>& rightWindowBorders,
                    const double* ResMass,
                    const double* massWindowHalfWidth);
  ~BackgroundHandler();

  void initializeParNums();

  inline int regionsParNum() { return parNumsResonances_[0]; }

  bool checkBackgroundWindow(const double& mass, const int iRegion) { return backgroundWindow_[iRegion].isIn(mass); }

  void countEventsInAllWindows(
      const std::vector<std::pair<reco::Particle::LorentzVector, reco::Particle::LorentzVector> >& muonPairs,
      const double& weight);

  void setParameters(double* Start,
                     double* Step,
                     double* Mini,
                     double* Maxi,
                     int* ind,
                     TString* parname,
                     const std::vector<double>& parBgr,
                     const std::vector<int>& parBgrOrder,
                     const int muonType);

  bool unlockParameter(const std::vector<int>& resfind, const unsigned int ipar);

  std::pair<double, double> windowBorders(const bool doBackgroundFit, const int ires);

  double resMass(const bool doBackgroundFit, const int ires);

  void rescale(std::vector<double>& parBgr,
               const double* ResMass,
               const double* massWindowHalfWidth,
               const std::vector<std::pair<reco::Particle::LorentzVector, reco::Particle::LorentzVector> >& muonPairs,
               const double& weight = 1.);

  std::pair<double, double> backgroundFunction(const bool doBackgroundFit,
                                               const double* parval,
                                               const int resTotNum,
                                               const int ires,
                                               const bool* resConsidered,
                                               const double* ResMass,
                                               const double ResHalfWidth[],
                                               /* const int MuonType, const double & mass, const double & resEta ); */
                                               const int MuonType,
                                               const double& mass,
                                               const double& eta1,
                                               const double& eta2);

private:
  void consistencyCheck(const std::vector<int>& identifiers,
                        const std::vector<double>& leftWindowBorders,
                        const std::vector<double>& rightWindowBorders) const noexcept(false);

  // Correspondence between regions and halfWidths used:
  // - for the Upsilons region we use the Upsilon
  // - for the J/Psi and Psi2S region we use the J/Psi
  int regToResHW_[3];

  // Correspondence between resonances and regions:
  // - Z -> region 0
  // - Uspilons -> region 1
  // - J/Psi and Psi2S -> region 2
  int resToReg_[6];

  // Used in the shifts of the parval
  // Contains 0 as the first number and Sum_0^(ires-1)(parNum(i)) for the rest,
  // so that calling parNums[ires] returns the sum of the number of parameters
  // of the previous functions (0 if none) and allows to shift the parval to the
  // parameters of the actual function.
  int parNumsRegions_[3];
  // These start from the correct parameters (take into account that the parRegions are
  // before the parResonances).
  int parNumsResonances_[6];

  // std::vector<double> leftWindowFactors_;
  // std::vector<double> rightWindowFactors_;

  std::vector<MassWindow> resonanceWindow_;
  std::vector<MassWindow> backgroundWindow_;
};

#endif  // BackgroundHandler_h