NuFCConfig.cxx

Go to the documentation of this file.

#include <fstream>
#include <sstream>

// General includes
#include <string>

#include "MessageService/MsgService.h"
#include "NtupleUtils/NuFCConfig.h"

CVSID("$Id: NuFCConfig.cxx,v 1.6 2009/06/03 10:29:09 nickd Exp $");

using namespace std;

// NuFCConfig

// Initialise the base Instance
NuFCConfig NuFCConfig::fInstance;

NuFCConfig::NuFCConfig()
{
  // Initialise everything with defaults

  fHelperFile = "CombinedHelpersRun1LE_bravo.root";
  fNDMC = "NDMCRun1LE_bravo.root";

  // FC Systematics variables
  fFCSystematics = true;
  fFCevents_Near = "/minos/data/analysis/NuMuBar/Helpers/Bravo/near_events.root";
  fFCevents_Far  = "/minos/data/analysis/NuMuBar/Helpers/Bravo/far_events.root";
  fFCevents_Tau  = "/minos/data/analysis/NuMuBar/Helpers/Bravo/tau_events.root";
  
  fPOT = 3.2e20;
  
  fFixedMixingAngle = false;
}

void NuFCConfig::PrintStatus()
{
  // Make a nicified string for the systematics status
  string syststatus = "No";
  if (fFCSystematics) syststatus = "Yes";
  
  string fixedsin = "No";
  if (fFixedMixingAngle) fixedsin = "Yes";
  
  cout << "***************************************************" << endl;
  cout  << "Configuration:" << endl
        << "POT:         " << fPOT << endl
        << "Helper File: " << fHelperFile << endl
        << "ND MC:       " << fNDMC << endl
        << "Systematics? " << syststatus << endl
        << "Fixed Sin?   " << fixedsin << endl;
  
  // Only show the systematics only vars if we want systematics
  cout  << "FC Events" << endl
        << "Near Events: " << fFCevents_Near << endl
        << "Far Events:  " << fFCevents_Far << endl
        << "Tau Events:  " << fFCevents_Tau << endl;
  cout << "***************************************************" << endl;
}

// Grabs hold of (or creates) the sole instance
NuFCConfig &NuFCConfig::GetConfig()
{
  // if (fInstance == 0) {
  //   // We need to create a configuration
  //   fInstance = new NuFCConfig;
  // }
  // return fInstance;
  
  return fInstance;
}

void NuFCConfig::LoadConfig(std::string filename)
{
  ConfigFile cfg(filename);
  
  // Load all the variables out of it
  
  // Generation sources
  fHelperFile = cfg.GetString("Helper", "MC", fHelperFile);
  fNDMC = cfg.GetString("Neardet", "MC", fNDMC);
  
  // Systematics variables
  fFCSystematics = cfg.GetBool("Systematics", "FC_Systematics", fFCSystematics);
  fFCevents_Near = cfg.GetString("Near_Events", "FC_Systematics", fFCevents_Near);
  fFCevents_Far = cfg.GetString("Far_Events", "FC_Systematics", fFCevents_Far);
  fFCevents_Tau = cfg.GetString("Tau_Events", "FC_Systematics", fFCevents_Tau);
  
  fPOT = cfg.GetFloat("POT", "Generation", fPOT);

  // Fitting variables
  fFixedMixingAngle = cfg.GetBool("OneParameter", "Fitting", fFixedMixingAngle);
}

// Configuration File Reader

ConfigFile::ConfigFile(string filename)
{
  ifstream cfg(filename.c_str());
  if(cfg.is_open())
  {
    // All is well, read the file in!
    ReadFile(cfg);
    
  } else {
    // Print an error message if we were trying to open a file
    if (filename != "") {
      MSG("ConfigFile",Msg::kFatal) << "Could not read configuration file " << filename << endl;
      throw runtime_error("Could not read configuration file.");
    }
  }
  
}

string ConfigFile::Decomment(const string &source)
{
  // Strips comments from a line
  
  // Find any instances of '#', and then remove them
  int comment = source.find_first_of("#%");
  if (comment == -1) return source;
  return source.substr(0, comment);
}
string ConfigFile::Trim(const string &source)
{
  // Find the first non-whitespace
  int first = source.find_first_not_of("\n\r\t ");
  int last = source.find_last_not_of("\n\r\t ");
  if (last == -1) return "";
  return source.substr(first, last - first + 1);
}

void ConfigFile::ReadFile(std::ifstream &file)
{
  int lineNo = 0;
  string line, section = "";
  
  // Loop over all lines in the file
  while(std::getline(file, line))
  {
    ++lineNo;
    
    // Clean up the line
    line = Decomment(line);
    line = Trim(line);
    
    // Ignore blank lines
    if (line.size() == 0) continue;
    
    // Several paths here: Section header, Key-Value pair or invalid.
    // First test for section header
    if (IsSectionChange(line))
    {
      section = Section(line);
      continue;
    }
    
    // Must be a key pair, or invalid
    if (!ReadKeyPair(section, line))
    {
      MSG("ConfigFile",Msg::kWarning) << "Invalid line " << lineNo << ": \"" << line << "\"" << endl; 
    }
    
  } // Go to next line
}

bool ConfigFile::IsSectionChange(const string &line)
{
  // Work out if this is a change of section by checking the first char
  if (line[0] == '[') {
    // Check the last char too
    if (line[line.size()-1] != ']')
    {
      // cout << "Warning Parsing file " << lineNo << ": Invalid section change!" << endl;
      MSG("ConfigFile",Msg::kWarning) << "Invalid section change: " << line << endl;
    } else {
      // Both start and ends are valid... is a section change line
      // Make sure it is valid
      string newsec = line.substr(1, line.size()-2);
      if (newsec.size() == 0) {
        MSG("ConfigFile", Msg::kWarning) << "Invalid Section Change: Section cannot be blank" << endl;
      } else {
        return true;
      }
    }
  }
  
  return false;
}

string ConfigFile::Section(const string &line)
{
  // Extracts the section from a line
  string newsec = line.substr(1, line.size()-2);
  
  return newsec;
}

bool ConfigFile::ReadKeyPair(const string &section, const string &line)
{
  // Check to see if this is a key pair
  int split = line.find_first_of("=:");
  
  // what if we don't find a split?
  if (split == -1)
  {
    // MSG("ConfigFile",Msg::kWarning) << "Invalid key-value pair: \"" << line << "\"" << endl;
    return false;
  }
  
  // Read out the values
  string key = Trim(line.substr(0, split));
  string keyval = Trim(line.substr(split+1, line.size()-split+2));
  fStore[section][key] = keyval;
  
  return true;
}

string ConfigFile::GetString(string key, string section, string defaultval)
{
  string value = fStore[section][key];

  if (value.size() == 0)
    return defaultval;
  else
    return value;    
}

int ConfigFile::GetInt(string key, string section, int defaultval)
{
  string value = fStore[section][key];
  if (value.size() == 0) return defaultval;
  
  // Attempt to read an integer out of this stream
  istringstream iss (value);
  int ival = 0;
  iss >> ival;
  
  if (!iss.fail())
  {
    return ival;
  } else {
    MAXMSG("ConfigFile",5,Msg::kWarning) << "Failed to read [" << section << "]." << key << " = " << value << " as integer" << endl;
    return defaultval;
  }
}

double ConfigFile::GetFloat(string key, string section, double defaultval)
{
  string value = fStore[section][key];
  if (value.size() == 0) return defaultval;
  
  // Attempt to read an integer out of this stream
  istringstream iss (value);
  double ival = 0.0;
  iss >> ival;
  
  if (!iss.fail())
  {
    return ival;
  } else {
    MAXMSG("ConfigFile",5,Msg::kWarning) << "Failed to read [" << section << "]." << key << " = " << value << " as integer" << endl;
    return defaultval;
  }
}

bool ConfigFile::GetBool(string key, string section, bool defaultval)
{
  string value = fStore[section][key];
  if (value.size() == 0) return defaultval;
  
  // Now try to work out what it is
  if (value == "True" || value == "true") return true;
  if (value == "False" || value == "false") return false;
  
  // Try to convert to an integer, and see if it is non-zero
  istringstream iss (value);
  int ival = 0;
  iss >> ival;
  
  // It worked - just return the value
  if (!iss.fail())
  {
    return ival;
  }
  
  // It failed..... give up here for now
  MAXMSG("ConfigFile",5,Msg::kWarning) << "Failed to read [" << section << "]." << key << " = " << value << " as boolean" << endl;
  return defaultval;
}

---