FileMapper.hh

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2023.
 
This file is part of BDSIM.
 
BDSIM is free software: you can redistribute it and/or modify 
it under the terms of the GNU General Public License as published 
by the Free Software Foundation version 3 of the License.
 
BDSIM is distributed in the hope that it will be useful, but 
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with BDSIM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef FILEMAPPER_H
#define FILEMAPPER_H
 
#include <string>
#include <vector>
 
class HistogramAccumulator;
class TDirectory;
class TFile;
class TH1;
 
namespace RBDS
{
  bool GetFileType(TFile* file,
                   std::string& fileType,
                   int*   dataVersion = nullptr);
 
  bool IsBDSIMOutputFile(TFile* file,
                         int*   dataVersion = nullptr);
 
  bool IsBDSIMOutputFile(const std::string& filePath,
                         int* dataVersion = nullptr);
 
  bool IsREBDSIMOutputFile(TFile* file);
 
  bool IsREBDSIMOutputFile(const std::string& filePath);
 
  bool IsREBDSIMOrCombineOutputFile(TFile* file);
 
  bool IsREBDSIMOrCombineOutputFile(const std::string& filePath);
 
  int DetermineDimensionality(TH1* h);
 
  void WarningMissingHistogram(const std::string& histName,
                               const std::string& fileName);
 
  struct HistogramPath
  {
    std::string path; 
    std::string name; 
    HistogramAccumulator* accumulator;
    TDirectory* outputDir;
    bool BDSBH4Dtype;
  };
 
  enum class MergeType {none, meanmerge, sum};
 
  MergeType DetermineMergeType(const std::string& parentDir);
}
 
class HistogramMap
{
public:
  HistogramMap() = delete; 
  HistogramMap(TFile* file,
               TFile* output,
               bool   debugIn = false);
  ~HistogramMap(){;}
 
  void MapDirectory(TDirectory* dir,
                    TFile*      output,
                    const std::string& parentDir);
 
  inline const std::vector<RBDS::HistogramPath>& Histograms() const {return histograms;}
  
private:
  bool debug;                                   
  std::vector<RBDS::HistogramPath> histograms;  
};
 
#endif