PedeSteering.cc

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 
#include <alignment/dataobjects/PedeSteering.h>
 
#include <fstream>
 
using namespace Belle2;
using namespace std;
 
string PedeSteering::make(string filename)
{
  command("end");
  if (filename == "")
    filename = name;
  ofstream file(filename.c_str());
  file << "! Steerig file for Pede generated by PedeSteering class !" << endl;
  file << "Cfiles" << endl;
  for (const string& cfile : files)
    file << cfile << endl;
 
  file << endl;
 
  for (const string& cmd : commands)
    file << cmd << endl;
 
  file.close();
  return filename;
}
void PedeSteering::command(const std::string& line)
{
  commands.push_back(line);
}
void PedeSteering::import(string filename)
{
  commands.clear();
  ifstream file(filename);
  if (!file.is_open())
    return;
 
  string commandLine;
  while (getline(file, commandLine))
    command(commandLine);
}
 
void PedeSteering::fixParameters(const vector< int >& labels,
                                 const vector< double >& values,
                                 const vector< double >& presigmas)
{
  command("Parameters");
  bool useValues = (labels.size() == values.size());
  bool usePresigmas = (labels.size() == presigmas.size());
 
  for (unsigned int i = 0; i < labels.size(); i++) {
    double value = 0.;
    double presigma = -1.;
    if (useValues)
      value = values[i];
    if (usePresigmas)
      presigma = presigmas[i];
 
    command(to_string(labels[i]) + " " + to_string(value) + " " + to_string(presigma));
  }
}
 
void PedeSteering::addFile(std::string filename, double weight)
{
  if (weight != 1.)
    filename = filename + " -- " + std::to_string(weight);
  bool exists = false;
  for (const auto& file : files)
    if (file == filename)
      exists = true;
  if (!exists)
    files.push_back(filename);
}
 
/*
void PedeSteering::addConstraint(double constraint, vector< int > labels, vector< double > coefficients) {
  command("Constraint " + to_string(constraint));
  if (labels.size() != coefficients.size())
    return;
 
  for (unsigned int i = 0; i < labels.size(); i++) {
    command(to_string(labels[i]) + " " + to_string(coefficients[i]));
  }
}
*/