BeamParametersFitter.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
 
/* MDST headers. */
#include <mdst/dbobjects/BeamSpot.h>
#include <mdst/dbobjects/CollisionBoostVector.h>
#include <mdst/dbobjects/CollisionInvariantMass.h>
 
/* Basf2 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 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_Verbose = false;
 
    BeamParameters m_BeamParameters;
 
    DBObjPtr<BeamSpot> m_BeamSpot;
 
    DBObjPtr<CollisionBoostVector> m_CollisionBoostVector;
 
    DBObjPtr<CollisionInvariantMass> m_CollisionInvariantMass;
 
  };
 
}