PedeResult.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 <map>
#include <string>
#include <vector>
 
namespace Belle2 {
  namespace alignment {
    class PedeResult {
    public:
      PedeResult() : data(), eigenNumbers(), valid(false), labelIndices() {}
      explicit PedeResult(const std::string& filename);
      void read(std::string filename = "millepede.res");
      void readEigenFile(std::string filename = "millepede.eve");
      bool isValid() {return valid;}
      void dump();
      int getNoParameters() const {return data.size();}
      int getNoDeterminedParameters() const {return noDeterminedParams;}
      int getParameterIndex(int parameterLabel);
      unsigned int getParameterLabel(unsigned int parameterIndex);
      double getParameterCorrection(unsigned int parameterIndex);
      double getParameterError(unsigned int parameterIndex);
      double getParameterPresigma(unsigned int parameterIndex);
      bool isParameterFixed(unsigned int parameterIndex);
      bool isParameterDetermined(unsigned int parameterIndex);
      int getNoEigenPairs() const {return eigenNumbers.size();}
      double getEigenNumber(unsigned int eigenPairIndex);
      double getEigenVectorElement(unsigned int eigenPairIndex, unsigned int parameterIndex);
 
    private:
      bool outOfBounds(int iparam) {if (iparam < 0) return true; if ((unsigned int)iparam >= data.size()) return true; return false;}
    public:
      struct parameterData {
        parameterData() {}
        parameterData(int index_, int label_, double correction_, double error_, double presigma_)
        {
          index = index_;
          label = label_;
          correction = correction_;
          error = error_;
          presigma = presigma_;
          eigenweights.clear();
        }
        int index;
        int label;
        double correction;
        double error;
        double presigma;
        std::vector<double> eigenweights;
      };
    private:
      std::vector<parameterData> data;
      std::vector<double> eigenNumbers;
      bool valid;
      std::map<int, int> labelIndices;
      int noDeterminedParams{0};
    };
  }
}