PdbParameterMapContainer.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PdbParameterMapContainer.cc

#include "PdbParameterMapContainer.h"
#include "PdbBankID.h"
#include "PdbParameterMap.h"

#include <phool/PHTimeStamp.h>
#include <phool/phool.h>

#include <TBufferXML.h>
#include <TFile.h>
#include <TSystem.h>

#include <boost/stacktrace.hpp>

#include <unistd.h>
#include <algorithm>
#include <cctype>
#include <ctime>
#include <iostream>
#include <sstream>

using namespace std;

PdbParameterMapContainer::~PdbParameterMapContainer()
{
  while (parametermap.begin() != parametermap.end())
  {
    delete parametermap.begin()->second;
    parametermap.erase(parametermap.begin());
  }
  return;
}

void PdbParameterMapContainer::print() const
{
  for (map<int, PdbParameterMap *>::const_iterator iter = parametermap.begin();
       iter != parametermap.end(); ++iter)
  {
    cout << "layer " << iter->first << endl;
    iter->second->print();
  }
  return;
}
void PdbParameterMapContainer::Reset()
{
  while (parametermap.begin() != parametermap.end())
  {
    delete parametermap.begin()->second;
    parametermap.erase(parametermap.begin());
  }
  return;
}

void PdbParameterMapContainer::AddPdbParameterMap(const int layer, PdbParameterMap *params)
{
  if (parametermap.find(layer) != parametermap.end())
  {
    cout << PHWHERE << " layer " << layer << " already exists" << endl;
    cout << "Here is the stacktrace: " << endl;
    cout << boost::stacktrace::stacktrace();
    cout << endl
         << "DO NOT PANIC - this is not a segfault" << endl;
    cout << "Check the stacktrace for the guilty party (typically #2)" << endl;
    gSystem->Exit(1);
  }
  parametermap[layer] = params;
}

const PdbParameterMap *
PdbParameterMapContainer::GetParameters(const int layer) const
{
  map<int, PdbParameterMap *>::const_iterator iter = parametermap.find(layer);
  if (iter == parametermap.end())
  {
    return nullptr;
  }
  return iter->second;
}

PdbParameterMap *
PdbParameterMapContainer::GetParametersToModify(const int layer)
{
  map<int, PdbParameterMap *>::iterator iter = parametermap.find(layer);
  if (iter == parametermap.end())
  {
    return nullptr;
  }
  return iter->second;
}

int PdbParameterMapContainer::WriteToFile(const std::string &detector_name,
                                          const string &extension, const string &dir)
{
  //Note the naming convention should be consistent with PHParameters::WriteToFile

  ostringstream fullpath;
  ostringstream fnamestream;
  PdbBankID bankID(0);  // lets start at zero
  PHTimeStamp TStart(0);
  PHTimeStamp TStop(0xffffffff);
  fullpath << dir;
  // add / if directory lacks ending /
  if (*(dir.rbegin()) != '/')
  {
    fullpath << "/";
  }
  fnamestream << detector_name << "_geoparams"
              << "-"
              << bankID.getInternalValue() << "-" << TStart.getTics() << "-"
              << TStop.getTics() << "-" << time(0) << "." << extension;
  string fname = fnamestream.str();
  std::transform(fname.begin(), fname.end(), fname.begin(), ::tolower);
  fullpath << fname;

  cout << "PdbParameterMapContainer::WriteToFile - save to " << fullpath.str()
       << endl;

  TFile *f = TFile::Open(fullpath.str().c_str(), "recreate");

  PdbParameterMapContainer *container = new PdbParameterMapContainer();
  for (std::map<int, PdbParameterMap *>::const_iterator it =
           parametermap.begin();
       it != parametermap.end(); ++it)
  {
    PdbParameterMap *myparm = static_cast<PdbParameterMap *>(it->second->CloneMe());
    container->AddPdbParameterMap(it->first, myparm);
  }

  // force xml file writing to use extended precision shown experimentally
  // to not modify input parameters (.15e)
  string floatformat = TBufferXML::GetFloatFormat();
  TBufferXML::SetFloatFormat("%.17g");  // for IEEE 754 double
  container->Write("PdbParameterMapContainer");
  delete f;
  // restore previous xml float format
  TBufferXML::SetFloatFormat(floatformat.c_str());
  cout << "sleeping 1 second to prevent duplicate inserttimes" << endl;
  sleep(1);
  return 0;
}