analysis/common/analysis_tree_converter/steer/Run.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Frederic Linz [committer], Volker Friese */
 
#ifndef CBM_ATCONVERTER_STEER_RUN_H
#define CBM_ATCONVERTER_STEER_RUN_H 1
 
#include "CbmDefs.h"
#include "Config.h"
 
#include <FairRunAna.h>
 
#include <TNamed.h>
#include <TString.h>
 
#include <string>
 
class CbmSetup;
class FairTask;
class FairFileSource;
class TTree;
 
namespace cbm::atconverter
{
 
  class Run : public TNamed {
 
  public:
    Run();
 
 
    virtual ~Run();
 
 
    void AddTask(FairTask* task);
 
 
    void AllowOverwrite() { fOverwrite = true; }
 
 
    void Exec();
 
 
    const Config& GetConfig() const { return fConfig; }
 
 
    bool IsDataPresent(ECbmModuleId detector) const { return (fDataPresent.count(detector) == 0 ? false : true); }
 
 
    void LoadConfig(const char* fileName);
 
 
    void SetConfig(const Config& config) { fConfig = config; }
 
 
    void SetGeoSetupTag(const char* tag) { fSetupTag = tag; }
 
 
    void SetOutput(const char* fileName) { fOutput = fileName; }
 
 
    void SetTraFiles(const std::vector<std::string> files);
 
 
    void SetRawFile(const char* fileName) { fRaw = fileName; }
 
 
    void SetParFile(const char* fileName) { fPar = fileName; }
 
 
    void SetRecoFile(const char* fileName) { fReco = fileName; }
 
 
  private:
    Run(const Run&) = delete;
 
 
    Run operator=(const Run&) = delete;
 
 
    bool CheckFile(const char* fileName);
 
 
    void CheckRecoBranch(TTree* tree, ECbmModuleId detector);
 
 
    void CheckInputBranches(FairFileSource* source);
 
 
    void CreateTopology();
 
 
  private:
    FairRunAna fRun {};
    TString fOutput   = "";
    std::vector<TString> fTra;
    TString fRaw      = "";
    TString fPar      = "";
    TString fReco     = "";
    TString fSetupTag = "";
    CbmSetup* fSetup  = nullptr;
    bool fOverwrite   = false;
 
  public:
    Config fConfig                      = {};
    std::set<ECbmModuleId> fDataPresent = {};
 
 
    ClassDef(cbm::atconverter::Run, 1);
  };
 
}  // namespace cbm::atconverter
 
#endif /* CBM_ATCONVERTER_STEER_RUN_H */