KVSimDataManager.h

#pragma once
#include <optional>
#include "KVBase.h"
#include "SQLiteDB.h"
#include "KV2Body.h"
#include "KVDataRepository.h"
 
class KVSimFile;
 
class KVSimDataManager : public KVBase {
public:
   using dataset_id = int;
   using model_id = int;
   using reaction_id = int;
   using file_id = int;
   using repository_id = int;
   using filter_id = int;
 
   enum class primary_or_secondary {
      primary,
      secondary
   };
 
private:
   mutable KVSQLite::database fDB;
   std::map<int, std::unique_ptr<KVDataRepository>> fRepos;
 
   Bool_t import_data_into_repository = false;
   repository_id import_repo = 0;
 
   std::optional<int> find_index_in_table(const TString& table_name, const KVNameValueList& params) const
   {
      fDB.select_data_in_list(table_name, params, "id");
      std::optional<int> id;
      while (fDB.get_next_result())
         id = fDB[table_name]["id"].get_data<int>();
      return id;
   }
   void enter_data_row(const TString& table_name, const KVNameValueList& params)
   {
      fDB.prepare_data_insertion(table_name);
      fDB[table_name].prepare_data(params);
      fDB.insert_data_row();
      fDB.end_data_insertion();
   }
   KVNameValueList get_data_for_unique_index(const TString& table_name, int id) const
   {
      KVNameValueList result;
      auto callback = [&](const KVNameValueList & l) {
         result = l;
         return false;
      };
      fDB.get_name_value_list_for_each_row(table_name, callback, Form("id = %d", id));
      return result;
   }
   KVNameValueList get_params_for_reaction(std::optional<double> opt, bool reaction, const KV2Body& R, const TString& ip_units) const;
   const KVDataRepository& get_data_repository(repository_id id);
 
   void initialize_repository(repository_id id, const TString& root_dir, const TString& access = "local",
                              std::optional<TString> xrdsrv = std::nullopt, std::optional<TString> xrdroot = std::nullopt);
 
   std::pair<TString, TString> import_file_to_repository(const TString& filepath, repository_id repo, dataset_id dset, reaction_id reac, std::optional<filter_id> filt_id = {});
 
   void add_file_to_collection(const TString& method_name, const TString& table_name, KVNameValueList& params, const TString& filepath, dataset_id dataset,
                               reaction_id reaction, std::optional<filter_id> filt_id = {});
 
public:
   KVSimDataManager() = default;
 
   void OpenDataBase(const TString&);
 
   repository_id AddDataRepository(const TString& root_dir, const TString& access = "local",
                                   std::optional<TString> xrdsrv = std::nullopt,
                                   std::optional<TString> xrdroot = std::nullopt);
 
   model_id   AddModel(const TString&, const KVNameValueList&);
   dataset_id AddDataSet(const TString& short_description, const TString& model, const KVNameValueList& model_params);
   dataset_id AddDataSet(const TString& short_description, model_id modid);
   reaction_id AddReaction(const KV2Body&, std::optional<double> = std::nullopt, const TString& ip_units = "fm");
   filter_id AddFilter(reaction_id kinematics, const TString& geometry, const TString& filter_type, bool random_phi,
                       std::optional<int> run_number = {}, std::optional<TString> data_quality_audit = {}, std::optional<int> gem_decay_per_event = {},
                       std::optional<bool> gem_add_rot_energy = {});
   void AddUnfilteredSimulation(const TString& filepath, Int_t events, primary_or_secondary P, dataset_id dataset, reaction_id reaction);
   void AddFilteredSimulation(const KVSimFile& f, dataset_id dataset, reaction_id reaction);
 
   std::optional<model_id> GetModelId(const TString&, const KVNameValueList&) const;
   std::optional<reaction_id> GetReactionId(const KV2Body&, std::optional<double> = std::nullopt, const TString& ip_units = "fm") const;
 
   void AddFilesFromDirectory(dataset_id dataset, reaction_id reaction, const TString& dirpath);
 
   void ImportFilesToRepository(repository_id = 0);
 
   TString GetReactionName(reaction_id id) const;
   TString GetModelName(model_id id) const;
 
   ClassDefOverride(KVSimDataManager, 0)
 
};