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:
    struct ObjectParameters {
      double ratioMin        = 0.95;
      double ratioMax        = 1.05;
      double pvalThreshold   = 0.05;
      std::string compMethod = "E";
    };

    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]; }
    const std::pair<std::string, uint32_t>& GetObject(int iFile, int iObject) const
    {
      return fvObjects[iFile][iObject];
    }

    const ObjectParameters& GetObjectParameters(uint32_t paramIndex) const { return fvObjectParameters[paramIndex]; }
    int GetNofDatasets() const { return fvDatasets.size(); }

    int GetNofObjects() const { return fvObjects.size(); }

    int GetNofObjects(int iFile) const { return fvObjects[iFile].size(); }

    int GetNofObjectParameters() const { return fvObjectParameters.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::pair<std::string, uint32_t>>> fvObjects;
 
    std::vector<ObjectParameters> fvObjectParameters;  
    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