InvariantMassAlgorithm.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <calibration/CalibrationAlgorithm.h>
#include <reconstruction/calibration/BeamSpotBoostInvMass/InvariantMassMuMuStandAlone.h>
#include <reconstruction/calibration/BeamSpotBoostInvMass/InvariantMassBhadStandAlone.h>
#include <reconstruction/calibration/BeamSpotBoostInvMass/calibTools.h>
 
namespace Belle2 {
  class InvariantMassAlgorithm : public CalibrationAlgorithm {
  public:
 
    InvariantMassAlgorithm();
 
    virtual ~InvariantMassAlgorithm() {}
 
    void setOuterLoss(const std::string& loss) { m_lossFunctionOuter = loss; }
 
    void setInnerLoss(const std::string& loss) { m_lossFunctionInner = loss; }
 
    void includeHadBcalib(bool state = true) {m_runHadB = state;}
 
    void setMuMuEcmsSpread(double spread) {m_eCMSmumuSpread = spread;}
 
    void setMuMuEcmsOffset(double shift) {m_eCMSmumuShift = shift;}
 
    std::vector<InvariantMassMuMuCalib::Event> getDataMuMu(const std::vector<std::string>& files, bool is4S);
 
    std::vector<InvariantMassBhadCalib::Event> getDataHadB(const std::vector<std::string>& files);
 
  protected:
 
    virtual EResult calibrate() override;
 
  private:
 
    std::vector<std::vector<CalibrationData>> adjustOffResonanceEnergy(std::vector<std::vector<CalibrationData>>  CalResultsBlocks,
                                           const std::vector<std::vector<InvariantMassMuMuCalib::Event>>&  evtsMuMuBlocks);
 
 
    TString m_lossFunctionOuter = "pow(0.000010e0*rawTime, 2) +  1./nEv";
 
    TString m_lossFunctionInner = "pow(0.000120e0*rawTime, 2) +  1./nEv";
 
    bool m_runHadB = true;
 
    double m_eCMSmumuSpread = 5.2e-3;
 
    double m_eCMSmumuShift  = 10e-3;
 
  };
} // namespace Belle2