BoostVectorAlgorithm.cc

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2021 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Radek Zlebcik                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#include <mdst/dbobjects/CollisionBoostVector.h>
#include <tracking/calibration/BoostVectorAlgorithm.h>
#include <tracking/calibration/BoostVectorStandAlone.h>
#include <tracking/calibration/calibTools.h>
 
#include <Eigen/Dense>
 
using Eigen::Vector3d;
using Eigen::Matrix3d;
 
using namespace Belle2;
 
using Belle2::BoostVectorCalib::getEvents;
using Belle2::BoostVectorCalib::runBoostVectorAnalysis;
 
 
BoostVectorAlgorithm::BoostVectorAlgorithm() : CalibrationAlgorithm("BoostVectorCollector")
{
  setDescription("BoostVector calibration algorithm");
}
 
 
static TObject* getBoostVectorObj(Vector3d vBoost, Matrix3d  vBoostUnc, Matrix3d /*vBoostSpread*/)
{
  auto payload = new CollisionBoostVector();
  payload->setBoost(toTVector3(vBoost), toTMatrixDSym(vBoostUnc));
  TObject* obj  = static_cast<TObject*>(payload);
  return obj;
}
 
 
 
/* Main calibration method calling dedicated functions */
CalibrationAlgorithm::EResult BoostVectorAlgorithm::calibrate()
{
  TTree* tracks = getObjectPtr<TTree>("events").get();
  return runCalibration(tracks,  "CollisionBoostVector", getEvents,
                        runBoostVectorAnalysis, getBoostVectorObj,
                        m_lossFunctionOuter, m_lossFunctionInner);
}