MillepedeAlgorithm.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 <alignment/dataobjects/PedeSteering.h>
#include <alignment/GlobalTimeLine.h>
#include <alignment/PedeResult.h>
#include <alignment/PedeApplication.h>
#include <calibration/CalibrationAlgorithm.h>
 
namespace Belle2 {
  class MillepedeAlgorithm : public CalibrationAlgorithm {
  public:
 
    MillepedeAlgorithm();
 
    virtual ~MillepedeAlgorithm() {}
 
    PedeSteering& steering() {return m_steering;}
 
    alignment::PedeResult& result() {return m_result;}
 
    alignment::PedeApplication& pede() {return m_pede;}
 
    void invertSign(bool use_subtraction = true) {m_invertSign = use_subtraction;}
 
    void setComponents(const std::vector<std::string>& components) {m_components = components;}
 
    void ignoreUndeterminedParams(bool ignore = true) {m_ignoreUndeterminedParams = ignore;}
 
    void setEvents(const std::vector<EventMetaData>& events) {m_events = events;}
 
    void setMinEntries(int minEntries) {m_minEntries = minEntries;}
 
    void setTimedepConfig(PyObject* config) {setEvents(alignment::timeline::setupTimedepGlobalLabels(config));}
 
    void setTimedepConfig(std::vector< std::tuple< std::vector<int>, std::vector< std::tuple<int, int, int> > > >& config)
    {
      setEvents(alignment::timeline::setupTimedepGlobalLabels(config));
    }
 
  protected:
 
    virtual EResult calibrate() override;
 
  private:
    std::vector<std::string> m_components{};
    bool m_invertSign{false};
    PedeSteering m_steering{"PedeSteering.txt"};
    alignment::PedeResult m_result{};
    alignment::PedeApplication m_pede{};
    bool m_ignoreUndeterminedParams{false};
    std::vector<EventMetaData> m_events{};
 
    void prepareMilleBinary();
 
    int m_minEntries{ -1};
 
  };
} // namespace Belle2