TRestManager.cxx

 
#include "TRestManager.h"
 
#include "TInterpreter.h"
#include "TRestTask.h"
#include "TSystem.h"
 
using namespace std;
ClassImp(TRestManager);
 
TRestManager::TRestManager() { Initialize(); }
 
TRestManager::~TRestManager() {
    // delete all the added metadata objects(besides self)
    if (fMetaObjects.size() > 1) {
        for (unsigned int i = fMetaObjects.size() - 1; i >= 1; i--) {
            delete fMetaObjects[i];
        }
    }
}
 
void TRestManager::Initialize() {
    SetSectionName(this->ClassName());
    // delete all the added metadata objects(besides self)
    if (fMetaObjects.size() > 1) {
        for (unsigned int i = fMetaObjects.size() - 1; i >= 1; i--) {
            delete fMetaObjects[i];
        }
    }
    // add self to the metadata objects list
    fMetaObjects.clear();
    fMetaObjects.push_back(this);
}
 
int TRestManager::LoadSectionMetadata() { return TRestMetadata::LoadSectionMetadata(); }
 
Int_t TRestManager::ReadConfig(string keydeclare, TiXmlElement* e) {
    // if (keydeclare == "TRestRun") {
    //  TRestRun* fRunInfo = new TRestRun();
    //  fRunInfo->SetHostmgr(this);
    //  fRunInfo->LoadConfigFromFile(e, fElementGlobal);
    //  fMetaObjects.push_back(fRunInfo);
    //  gROOT->Add(fRunInfo);
    //  return 0;
    //}
 
    // else if (keydeclare == "TRestProcessRunner") {
    //  TRestProcessRunner* fProcessRunner = new TRestProcessRunner();
    //  fProcessRunner->SetHostmgr(this);
    //  fProcessRunner->LoadConfigFromFile(e, fElementGlobal);
    //  fMetaObjects.push_back(fProcessRunner);
    //  gROOT->Add(fProcessRunner);
    //  return 0;
    //}
 
    // cout << "----------- " << gROOT->FindObject("SJTU_Proto") << endl;
 
    if (Count(keydeclare, "TRest") > 0) {
        TRestMetadata* meta = REST_Reflection::Assembly(keydeclare);
        if (meta == nullptr) return -1;
        meta->SetHostmgr(this);
        fMetaObjects.push_back(meta);
        meta->SetConfigFile(fConfigFileName);
        meta->LoadConfigFromElement(e, fElementGlobal, fVariables);
 
        return 0;
    }
 
    else if (keydeclare == "addTask") {
        string active = GetParameter("value", e, "");
        if (active != "ON" && active != "On" && active != "on") {
            RESTDebug << "Inactived task : \"" << ElementToString(e) << "\"" << RESTendl;
            return 0;
        }
        RESTDebug << "Loading Task...";
 
        const char* type = e->Attribute("type");
        const char* cmd = e->Attribute("command");
        if (type == nullptr && cmd == nullptr) {
            RESTWarning << "command or type should be given!" << RESTendl;
            return -1;
        }
        if (type != nullptr) {
            RESTDebug << " \"" << type << "\" " << RESTendl;
            if ((string)type == "processEvents") {
                auto pr = GetProcessRunner();
                if (pr != nullptr) pr->RunProcess();
            } else if ((string)type == "analysisPlot") {
                auto ap = GetAnaPlot();
                if (ap != nullptr) ap->PlotCombinedCanvas();
            } else if ((string)type == "saveMetadata") {
                auto ri = GetRunInfo();
                if (ri != nullptr) {
                    ri->FormOutputFile();
                    ri->CloseFile();
                }
            } else {
                TRestTask* tsk = TRestTask::GetTaskFromMacro(type);
                if (tsk == nullptr) {
                    RESTWarning << "REST ERROR. Task : " << type << " not found!!" << RESTendl;
                    RESTWarning << "This task will be skipped." << RESTendl;
                    return -1;
                }
                tsk->LoadConfigFromElement(e, fElementGlobal);
                tsk->RunTask(this);
                return 0;
            }
        } else if (cmd != nullptr) {
            RESTDebug << " \"" << cmd << "\" " << RESTendl;
 
            TRestTask* tsk = TRestTask::GetTaskFromCommand(cmd);
            if (tsk == nullptr) {
                RESTWarning << "REST ERROR. Command : " << cmd << " cannot be parsed!!" << RESTendl;
                RESTWarning << "This task will be skipped." << RESTendl;
                return -1;
            }
            tsk->RunTask(this);
            return 0;
        }
    }
 
    return -1;
}
 
void TRestManager::InitFromTask(string taskName, vector<string> arguments) {
    TRestTask* tsk = TRestTask::GetTaskFromMacro(taskName);
    if (tsk == nullptr) {
        gSystem->Exit(-1);
    }
    tsk->SetArgumentValue(arguments);
    tsk->RunTask(nullptr);
    gSystem->Exit(0);
}
 
TRestMetadata* TRestManager::GetMetadataClass(string type) {
    for (unsigned int i = 0; i < fMetaObjects.size(); i++) {
        if ((string)fMetaObjects[i]->ClassName() == type) {
            return fMetaObjects[i];
        }
    }
    return nullptr;
}
 
TRestMetadata* TRestManager::GetMetadata(string name) {
    for (unsigned int i = 0; i < fMetaObjects.size(); i++) {
        if (fMetaObjects[i]->GetName() == name) {
            return fMetaObjects[i];
        }
    }
    return nullptr;
}
 
void TRestManager::PrintMetadata() { return; }