GlobalDerivatives.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Tadeas Bilka                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <root/TMatrixD.h>
 
#include <vector>
 
namespace Belle2 {
  namespace alignment {
    class GlobalDerivatives {
    public:
      explicit GlobalDerivatives(int dim = 2) {m_globals.second.ResizeTo(dim, 0);}
      explicit GlobalDerivatives(const std::pair<std::vector<int>, TMatrixD>& globals);
      GlobalDerivatives(const std::vector<int>& labels, const TMatrixD& derivs);
      operator std::pair<std::vector<int>, TMatrixD>() {return passGlobals(m_globals);}
      operator std::vector<int>() {return passGlobals(m_globals).first;}
      operator TMatrixD() {return passGlobals(m_globals).second;}
      const std::vector<int>& getLabels() const {return m_globals.first;}
      const TMatrixD& getDerivatives() const {return m_globals.second;}
      void add(const std::pair<std::vector<int>, TMatrixD>& globals);
      void add(int paramLabel, std::vector<double> dResiduals_dParam);
      void add(int paramLabel, double drudp);
      static std::pair<std::vector<int>, TMatrixD> passGlobals(std::pair<std::vector<int>, TMatrixD> globals);
    private:
      std::pair<std::vector<int>, TMatrixD> m_globals {{}, TMatrixD()};
    };
  }
}