GGSTFilesHandler.cpp

Go to the documentation of this file.

/*
 * GGSTFilesHandler.cpp
 *
 *  Created on: 31 Jul 2014
 *      Author: Nicola mori
 */

// C++ headers
#include <algorithm>
#include <iostream>

// ROOT headers
#include "TFile.h"

// GGS headers
#include "montecarlo/readers/GGSTFilesHandler.h"
#include "utils/GGSSmartLog.h"

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

GGSTFilesHandler::GGSTFilesHandler() : _geoManager{nullptr} {}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

GGSTFilesHandler::~GGSTFilesHandler() {
  for (unsigned int iChain = 0; iChain < _chains.size(); iChain++) {
    delete _chains[iChain];
  }
  delete _geoManager;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

const std::string &GGSTFilesHandler::GetFileName(unsigned int iFile) {
  static const std::string nullFile("");
  if (iFile < GetNFiles())
    return _files[iFile];
  else
    return nullFile;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

bool GGSTFilesHandler::_AddFile(const std::string &fileName, bool force) {
  static const std::string routineName("GGSTFilesHandler::_AddFile");

  _files.push_back(fileName);

  // Check if file exists and can be opened
  TFile *file = TFile::Open(fileName.data());
  if (!file) {
    GGSCOUT(WARNING) << "File " << fileName << " not found. File discarded." << GGSENDL;
    _files.pop_back();
    return false;
  }

  if (file->IsZombie()) {
    GGSCOUT(WARNING) << "File " << fileName << " cannot be opened. File discarded." << GGSENDL;
    _files.pop_back();
    delete file;
    return false;
  }

  // Do some consistency checks
  if (!force) {
    // Check if file contains GGS simulation data
    std::unique_ptr<GGSTSimInfo> simInfo((GGSTSimInfo *)(file->Get("GGSSimInfo")));
    if (simInfo == NULL) {
      GGSCOUT(WARNING) << "GGS simulation info not present in file " << fileName << ". File discarded." << GGSENDL;
      _files.pop_back();
      delete file;
      return false;
    }
    // Retrieve geometry and generator parameters (if they exist)
    std::unique_ptr<GGSTGeoParams> geoParams((GGSTGeoParams *)(file->Get("GGSGeoParams")));
    std::unique_ptr<GGSTParameters> genParams((GGSTParameters *)(file->Get("GGSGenParams")));
    if (_files.size() == 1) {
      // Save simulation info
      _simInfo = *simInfo;
      // Set geometry parameters info object if it exists
      if (geoParams)
        _geoParams = *geoParams;
      // Set generator parameters info object if it exists
      if (genParams)
        _genParams = *genParams;
    } else {
      // Check if info from current file is consistent with others
      if (!(simInfo->IsSameSimAs(_simInfo))) {
        GGSCOUT(WARNING) << "Simulation info in file " << fileName << " are inconsistent with those in file "
                         << _files[0] << ". File discarded." << GGSENDL;
        _files.pop_back();
      } else if (!(geoParams->AreSameParamsAs(_geoParams))) {
        GGSCOUT(WARNING) << "Geometry parameters in file " << fileName << " are inconsistent with those in file "
                         << _files[0] << ". File discarded." << GGSENDL;
        _files.pop_back();
      } else if (!(genParams->AreSameParamsAs(_genParams))) {
        GGSCOUT(WARNING) << "Generator parameters in file " << fileName << " are inconsistent with those in file "
                         << _files[0] << ". File discarded." << GGSENDL;
        _files.pop_back();
      }
    }
  }
  delete file;
  return true;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....

void GGSTFilesHandler::_SortFiles() {
  static const std::string routineName("GGSTFilesHandler::_SortFiles");

  if (_chains.size() > 0) {
    GGSCOUT(WARNING) << "Can't sort files after chain creation." << GGSENDL;
    return;
  }

  sort(_files.begin(), _files.end());
  // Reset the simulation info to that of the first file
  TFile *file0 = TFile::Open(_files[0].data());
  _simInfo = *((GGSTSimInfo *)(file0->Get("GGSSimInfo")));
  delete file0;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo.....