ConstraintConfiguration.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 <vector>
#include <Eigen/Core>
 
namespace TreeFitter {
 
  class ConstraintConfiguration {
  public:
 
    ConstraintConfiguration() :
      m_massConstraintType(false),
      m_massConstraintListPDG({}),
                            m_fixedToMotherVertexListPDG({}),
                            m_geoConstraintListPDG({}),
                            m_removeConstraintList({}),
                            m_automatic_vertex_constraining(false),
                            m_ipConstraint(false),
                            m_customOrigin(false),
                            m_customOriginVertex({}),
                            m_customOriginCovariance({}),
                            m_originDimension(3),
                            m_headOfTreePDG(0),
                            m_beamConstraintPDG(0),
                            m_beamMomE(Eigen::Matrix<double, 4, 1>()),
                            m_beamCovariance(Eigen::Matrix<double, 4, 4>()),
                            m_inflationFactorCovZ(1)
    {};
 
    ConstraintConfiguration(const bool& massConstraintType,
                            const std::vector<int>& massConstraintListPDG,
                            const std::vector<int>& fixedToMotherVertexListPDG,
                            const std::vector<int>& geoConstraintListPDG,
                            const std::vector<std::string>& removeConstraintList,
                            const bool& automatic_vertex_constraining,
                            const bool& ipConstraint,
                            const bool& customOrigin,
                            const std::vector<double>& customOriginVertex,
                            const std::vector<double>& customOriginCovariance,
                            const int& originDimension,
                            const int& beamConstraintPDG,
                            const Eigen::Matrix<double, 4, 1>& beamMomE,
                            const Eigen::Matrix<double, 4, 4>& beamCovariance,
                            const int& inflationFactorCovZ = 1
                           ) :
      m_massConstraintType(massConstraintType),
      m_massConstraintListPDG(massConstraintListPDG),
      m_fixedToMotherVertexListPDG(fixedToMotherVertexListPDG),
      m_geoConstraintListPDG(geoConstraintListPDG),
      m_removeConstraintList(removeConstraintList),
      m_automatic_vertex_constraining(automatic_vertex_constraining),
      m_ipConstraint(ipConstraint),
      m_customOrigin(customOrigin),
      m_customOriginVertex(customOriginVertex),
      m_customOriginCovariance(customOriginCovariance),
      m_originDimension(originDimension),
      m_headOfTreePDG(0),
      m_beamConstraintPDG(beamConstraintPDG),
      m_beamMomE(beamMomE),
      m_beamCovariance(beamCovariance),
      m_inflationFactorCovZ(inflationFactorCovZ)
    {}
 
    const bool  m_massConstraintType;
 
    const std::vector<int> m_massConstraintListPDG;
 
    const std::vector<int>  m_fixedToMotherVertexListPDG;
 
    const std::vector<int>  m_geoConstraintListPDG;
 
    const std::vector<std::string>  m_removeConstraintList;
 
    const bool  m_automatic_vertex_constraining;
 
    const bool m_ipConstraint;
 
    const bool m_customOrigin;
 
    const std::vector<double> m_customOriginVertex;
 
    const std::vector<double> m_customOriginCovariance;
 
    const int m_originDimension;
 
    mutable int m_headOfTreePDG;
 
    const int m_beamConstraintPDG;
 
    Eigen::Matrix<double, 4, 1> m_beamMomE;
 
    Eigen::Matrix<double, 4, 4> m_beamCovariance;
 
    const int m_inflationFactorCovZ;
 
  };
}