CbmQaCheckerObjectDB.h

Go to the documentation of this file.

/* Copyright (C) 2023-2024 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Sergei Zharko [committer] */
 
 
#ifndef CbmQaCheckerObjectDB_h
#define CbmQaCheckerObjectDB_h 1
 
#include "CbmQaCheckerTypedefs.h"
 
#include <set>
#include <string>
#include <utility>
#include <vector>
 
class TString;
class TDirectory;
 
namespace cbm::qa::checker
{
  class ObjectDB {
   public:
    ObjectDB() = default;
 
    ~ObjectDB() = default;
 
    ObjectDB(const ObjectDB&) = delete;
 
    ObjectDB(ObjectDB&&) = delete;
 
    ObjectDB& operator=(const ObjectDB&) = delete;
 
    ObjectDB& operator=(ObjectDB&&) = delete;
 
    void AddVersion(const char* label, const char* path);
 
    void AddDataset(const char* dataset);
 
    void Clear();
 
    const std::string& GetDataset(int iDataset) const { return fvDatasets[iDataset]; }
 
    int GetDefaultID() const { return fDefVersionID; }
 
    const std::string& GetFileLabel(int iFile) const { return fvFileLabels[iFile]; }
 
    std::string GetInputFileName(int iVersion, int iFile, int iDataset) const;
 
    const std::string& GetObject(int iFile, int iObject) const { return fvObjects[iFile][iObject]; }
 
    int GetNofDatasets() const { return fvDatasets.size(); }
 
    int GetNofObjects() const { return fvObjects.size(); }
 
    int GetNofObjects(int iFile) const { return fvObjects[iFile].size(); }
 
    int GetNofFiles() const { return fvFiles.size(); }
 
    int GetNofVersions() const { return fvVersionLabels.size(); }
 
    const std::string& GetOutputPath() const { return fsOutputPath; }
 
    double GetPvalThreshold() const { return fPvalThresh; }
 
    double GetRatioRangeMax() const { return fRatioMax; }
 
    double GetRatioRangeMin() const { return fRatioMin; }
 
    const std::string& GetVersionLabel(int iVersion) const { return fvVersionLabels[iVersion]; }
 
    const std::string& GetVersionPath(int iVersion) const { return fvVersionPaths[iVersion]; }
 
    void Init();
 
    void ReadFromYAML(const char* configName);
 
    std::string ToString(int verbose = 1) const;
 
    void SetDefaultLabel(const char* defaultLabel) { fsDefaultLabel = defaultLabel; }
 
    void SetInputRootPath(const char* pathName) { fsInputRootPath = pathName; }
 
    void SetOutputPath(const char* path) { fsOutputPath = path; }
 
    void SetPvalThreshold(double pVal);
 
    void SetRatioRange(double min, double max);
 
   private:
    void ReadObjectList(int iFile);
 
    static void CollectObjectPaths(TDirectory* pDir, const TString& parentPah, std::set<std::string>& paths);
 
    int fDefVersionID = -1;  
 
    std::string fsInputRootPath = "";  
    std::string fsOutputPath    = "";  
    std::string fsDefaultLabel  = "";  
 
    std::vector<std::string> fvDatasets;              
    std::vector<std::string> fvFiles;                 
    std::vector<char> fvbProcessWholeFile;            
    std::vector<int> fvObjectFirstGlobIndex;          
    std::vector<std::string> fvFileLabels;            
    std::vector<std::vector<std::string>> fvObjects;  
    std::vector<std::string> fvVersionLabels;         
    std::vector<std::string> fvVersionPaths;          
 
    double fPvalThresh{0.05};  
    double fRatioMax{1.05};    
    double fRatioMin{0.95};    
  };
}  // namespace cbm::qa::checker
 
#endif  // CbmQaCheckerObjectDB_h