KVDataSet.h

/*
$Id: KVDataSet.h,v 1.26 2009/01/27 08:06:56 franklan Exp $
$Revision: 1.26 $
$Date: 2009/01/27 08:06:56 $
$Author: franklan $
*/
 
#ifndef __KVDATASET_H
#define __KVDATASET_H
 
#include "KVString.h"
#include "KVBase.h"
#include "KVUniqueNameList.h"
#include "KVNumberList.h"
#include "TDatime.h"
#include "TFile.h"
#include "KVDataAnalysisTask.h"
#include "KVExpDB.h"
#include "run_index.h"
 
class KVDataSetManager;
class KVDBSystem;
class TObjArray;
class KVDBRun;
class KVAvailableRunsFile;
class KVDataSetRepository;
 
class KVDataSet: public KVBase {
 
   friend class KVAvailableRunsFile;
   friend class KVDataSetManager;
 
   bool make_analysis_class(const KVDataAnalysisTask*, const Char_t* classname);
 
   TObject* file_opener(const TString&, const TString&, Option_t* = "");
 
   KVDataSetRepository* fRepository = nullptr;
    TString fDatatypes;
    KVUniqueNameList fAllTasks;
    KVUniqueNameList fTasks;
    TString fCalibDir;
    Bool_t fDBBuild;
    mutable std::unique_ptr<TFile> fDBase;
    TString fDBName;
   mutable KVExpDB* fDataBase = nullptr;       //pointer to dataset's database
    mutable KVList fAvailableRuns;
   enum {
      kAvailable = BIT(14)     //flag set if this dataset is physically present on local machine
   };
    TString fUserGroups;
 
   virtual const Char_t* GetBaseFileName(const Char_t* type, const run_index_t& run) const;
 
   void OpenDataBase(const TString& = "") const;
   void OpenDBFile(const Char_t* full_path_to_dbfile) const;
   void WriteDBFile(const Char_t* full_path_to_dbfile) const;
   TString GetDBFileName() const;
   void SetDBName(const Char_t* name)
   {
      fDBName = name;
   }
   TString GetDBName() const;
   TString GetFullPathToDB() const;
   void SetDataSetSpecificTaskParameters(KVDataAnalysisTask*) const;
   TObject* open_runfile(const KVString& type, const run_index_t& run);
 
   mutable Bool_t fDataBaseUpdateInProgress = false;
public:
   Bool_t DataBaseUpdateInProgress() const
   {
      return fDataBaseUpdateInProgress;
   }
 
   KVAvailableRunsFile* GetAvailableRunsFile(const TString& type) const;
 
   KVDataSet();
 
   void SetDataPathSubdir(const TString& s)
   {
      SetLabel(s);
   }
   KVString GetDataPathSubdir() const
   {
      return GetLabel();
   }
   KVString GetDataTypeSubdir(const TString& type) const
   {
      if (type.IsNull())
         return type;
      KVString snom;
      snom.Form("KVDataSet.DataType.Subdir.%s", type.Data());
      return GetDataSetEnv(snom.Data(), type);
   }
   const Char_t* GetAvailableDataTypes() const
   {
      return fDatatypes.Data();
   }
   void AddAvailableDataType(const Char_t*);
   void SetUserGroups(const Char_t* groups)
   {
      fUserGroups = groups;
   }
   Bool_t IsAvailable() const
   {
      return TestBit(kAvailable);
   }
   void SetAvailable(Bool_t yes = kTRUE)
   {
      SetBit(kAvailable, yes);
   }
   void CheckAvailable();
 
   Bool_t HasDataType(const Char_t* data_type) const
   {
 
      KVString _dt = data_type;
      _dt.Remove(TString::kBoth, ' ');
      return (_dt == "*" || fDatatypes.Contains(_dt));
   }
 
   void ls(Option_t* opt = "") const override;
   void Print(Option_t* opt = "") const override;
 
   void SetAnalysisTasks(const KVSeqCollection*);
   KVDataAnalysisTask* GetAnalysisTask(Int_t) const;
   KVDataAnalysisTask* GetAnalysisTask(const Char_t* keywords) const;
   KVDataAnalysisTask* GetAnalysisTaskAny(const Char_t* keywords) const;
   Int_t GetNtasks() const;
 
   KVUnownedList GetListOfAvailableSystems(const TString& datatype);
   KVUnownedList GetListOfAvailableSystems(KVDataAnalysisTask* datan);
   KVList GetListOfAvailableRunFilesForSystem(const TString& datatype, KVDBSystem* systol);
   KVList GetListOfAvailableRunFilesForSystem(KVDataAnalysisTask* datan, KVDBSystem* systol);
   KVUniqueNameList GetSystemsForRunFiles(const TString& datatype, const run_index_list&) const;
 
   void cd() const;
 
   const Char_t* GetDataSetDir() const;
   void SetName(const char* name) override;
 
 
   KVExpDB* GetDataBase(const TString& opt = "") const;
   void SaveDataBase() const;
 
   template<typename ValType>
   ValType GetDataSetEnv(const Char_t* type, const ValType& defval) const
   {
 
      return KVBase::GetDataSetEnv(GetName(), type, defval);
   }
   KVString GetDataSetEnv(const Char_t* type, const Char_t* defval = "") const
   {
      return GetDataSetEnv(type, KVString(defval));
   }
   KVString GetDataSetEnv(const Char_t* type, Char_t* defval) const
   {
      return GetDataSetEnv(type, KVString(defval));
   }
 
   TString GetFullPathToRunfile(const KVString& type, const run_index_t& run) const;
   KVString GetRunfileName(const KVString& type, const run_index_t& run) const;
   std::optional<TDatime> GetRunfileDate(const KVString& type, const run_index_t& run);
   std::optional<run_index_t> GetRunNumberFromFileName(const TString& datatype, const TString& filename);
   template <typename FileType>
   FileType* OpenRunfile(const KVString& type, const run_index_t& run)
   {
      return dynamic_cast<FileType*>(open_runfile(type, run));
   }
   TObject* OpenRunfile(const KVString& type, const run_index_t& run)
   {
      return open_runfile(type, run);
   }
 
   Bool_t CheckRunfileAvailable(const KVString& type, const run_index_t& run);
   void UpdateAvailableRuns(const KVString& type);
   TFile* NewRunfile(const KVString& type, const run_index_t& run);
   void CommitRunfile(const KVString& type, const run_index_t& run, TFile* file);
 
   void CopyRunfilesFromRepository(const Char_t* type, const run_index_list& runs, const Char_t* destdir);
 
   void DeleteRunfile(const KVString& type, const run_index_t& run, Bool_t confirm = kTRUE);
   void DeleteRunfiles(const Char_t* type, const run_index_list& lrun = {}, Bool_t confirm = kTRUE);
   run_index_list GetRunList_DateSelection(const TString& type, const KVUnownedList& systems, std::optional<KVDatime> min_date = std::nullopt, std::optional<KVDatime> max_date = std::nullopt);
   run_index_list GetRunList_StageSelection(const TString& other_type, const TString& base_type, KVDBSystem* sys = 0, Bool_t OnlyCol = kFALSE);
   run_index_list GetRunList_StageSelection(const TString& other_type, const TString& base_type, const KVUnownedList& systems, Bool_t OnlyCol = kFALSE);
   run_index_list GetRunList_VersionSelection(const TString& type, const TString& version, KVDBSystem* sys = 0);
   void SetRepository(KVDataSetRepository*);
   KVDataSetRepository* GetRepository() const;
 
   void CheckMultiRunfiles(const Char_t* data_type);
   void CleanMultiRunfiles(const Char_t* data_type, Bool_t confirm = kTRUE);
   void CleanRunfileDataBase(const Char_t* data_type);
 
   Bool_t CheckRunfileUpToDate(const KVString& data_type, const run_index_t& run,
                               KVDataSetRepository* other_repos);
   Bool_t CheckRunfileUpToDate(const KVString& data_type, const run_index_t& run,
                               const KVString& other_repos);
   void CheckUpToDate(const Char_t* data_type,
                      const Char_t* other_repos);
   run_index_list GetUpdatableRuns(const Char_t* data_type,
                                   const Char_t* other_repos);
 
   run_index_list GetRunList(const Char_t* data_type,
                             const KVDBSystem* sys = 0) const;
 
   bool MakeAnalysisClass(const Char_t* task, const Char_t* classname);
   bool MakeAnalysisClass(int task, const Char_t* classname);
 
   Bool_t OpenDataSetFile(const Char_t* filename, std::ifstream& file);
   static Bool_t OpenDataSetFile(const TString& dataset, const Char_t* filename, std::ifstream& file);
   TString GetFullPathToDataSetFile(const Char_t* filename) const;
   static TString GetFullPathToDataSetFile(const TString& dataset, const Char_t* filename);
   static Bool_t FindDataSetFile(const TString& dataset, const Char_t* filename);
   Bool_t FindDataSetFile(const Char_t* filename) const;
 
   Bool_t DataBaseNeedsUpdate() const;
 
   KVString GetReconstructedEventClassName() const
   {
 
      return GetDataSetEnv("ReconstructedEventClassName", "KVReconstructedEvent");
   }
 
   TString GetOutputRepository(const Char_t* taskname) const;
   void CopyRunfilesToRepository(const Char_t* type, const run_index_list& runs, const Char_t* destrepo);
 
   Bool_t HasCalibIdentInfos() const
   {
      return GetDataSetEnv("HasCalibIdentInfos", kTRUE);
   }
 
   ClassDefOverride(KVDataSet, 3)       //Describes a set of experimental data which may be analysed with KaliVeda
};
 
R__EXTERN KVDataSet* gDataSet;
 
#endif