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];
  }
}

//....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.....

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

  _files.push_back(fileName);

  // Check if file can be opened
  TFile *file = TFile::Open(fileName.data());
  if (file->IsZombie()) {
    COUT(WARNING) << "File " << fileName << " cannot be opened. File discarded." << ENDL;
    _files.pop_back();
    delete file;
    return;
  }

  // Do some consistency checks
  if (!force) {
    // Check if file contains GGS simulation data
    GGSTSimInfo *simInfo = (GGSTSimInfo *)(file->Get("GGSSimInfo"));
    if (simInfo == NULL) {
      COUT(WARNING) << "GGS simulation info not present in file " << fileName << ". File discarded." << ENDL;
      _files.pop_back();
      delete file;
      return;
    }
    // Retrieve geometry and generator parameters (if they exist)
    GGSTGeoParams *geoParams = (GGSTGeoParams *)(file->Get("GGSGeoParams"));
    GGSTParameters *genParams = (GGSTParameters *)(file->Get("GGSGenParams"));
    TGeoManager *geoManager = (TGeoManager *)(file->Get("GGSGeometry"));
    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;
      // Set ROOT geometry object if it exists
      if (geoManager)
        _geoManager = geoManager;
    } else {
      // Check if info from current file is consistent with others
      if (!(simInfo->IsSameSimAs(_simInfo))) {
        COUT(WARNING) << "Simulation info in file " << fileName << " are inconsistent with those in file " << _files[0]
                      << ". File discarded." << ENDL;
        _files.pop_back();
      } else if (!(geoParams->AreSameParamsAs(_geoParams))) {
        COUT(WARNING) << "Geometry parameters in file " << fileName << " are inconsistent with those in file "
                      << _files[0] << ". File discarded." << ENDL;
        _files.pop_back();
      } else if (!(genParams->AreSameParamsAs(_genParams))) {
        COUT(WARNING) << "Generator parameters in file " << fileName << " are inconsistent with those in file "
                      << _files[0] << ". File discarded." << ENDL;
        _files.pop_back();
      }
    }
  }
  delete file;
}

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

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

  if (_chains.size() > 0) {
    COUT(WARNING) << "Can't sort files after chain creation." << ENDL;
    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.....