BeamParametersFitter.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2021  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Kirill Chilikin                                          *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
/* MDST headers. */
#include <mdst/dbobjects/BeamSpot.h>
#include <mdst/dbobjects/CollisionBoostVector.h>
#include <mdst/dbobjects/CollisionInvariantMass.h>
 
/* Belle 2 headers. */
#include <framework/database/DBObjPtr.h>
#include <framework/database/IntervalOfValidity.h>
#include <framework/dbobjects/BeamParameters.h>
 
namespace Belle2 {
  class BeamParametersFitter {
 
  public:
 
    BeamParametersFitter()
    {
    }
 
    ~BeamParametersFitter()
    {
    }
 
    void setIntervalOfValidity(const IntervalOfValidity& iov)
    {
      m_IntervalOfValidity = iov;
    }
 
    void setAngleHER(double angleHER)
    {
      m_AngleHER = angleHER;
    }
 
    void setAngleLER(double angleLER)
    {
      m_AngleLER = angleLER;
    }
 
    void setAngleError(double angleError)
    {
      m_AngleError = angleError;
    }
 
    void setBoostError(double boostError)
    {
      m_BoostError = boostError;
    }
 
    void setInvariantMassError(double invariantMassError)
    {
      m_InvariantMassError = invariantMassError;
    }
 
    void useMomentum(bool use)
    {
      m_UseMomentum = use;
    }
 
    void setVerbose(bool verbose)
    {
      m_Verbose = verbose;
    }
 
    BeamParameters getBeamParameters() const
    {
      return m_BeamParameters;
    }
 
    void setBeamParameters(const BeamParameters* beamParameters)
    {
      m_BeamParameters = *beamParameters;
    }
 
    void fit();
 
    void fillVertexData(double covarianceXX, double covarianceYY);
 
    void importBeamParameters();
 
  protected:
 
    void setupDatabase();
 
    IntervalOfValidity m_IntervalOfValidity;
 
    double m_AngleHER = 0;
 
    double m_AngleLER = 0;
 
    double m_AngleError = 0;
 
    double m_BoostError = 0.0001;
 
    double m_InvariantMassError = 0.0001;
 
    bool m_UseMomentum = false;
 
    bool m_Verbose = false;
 
    BeamParameters m_BeamParameters;
 
    DBObjPtr<BeamSpot> m_BeamSpot;
 
    DBObjPtr<CollisionBoostVector> m_CollisionBoostVector;
 
    DBObjPtr<CollisionInvariantMass> m_CollisionInvariantMass;
 
  };
 
}