NcJob.cxx

Go to the documentation of this file.

 
#include "NcJob.h"
#include "Riostream.h"
 
ClassImp(NcJob); // Class implementation to enable ROOT I/O

NcJob::NcJob(const char* name,const char* title) : TTask(name,title)
{
 
 fMakefolder=0;
 fMainObject=0;
 fFolder=0;
 fObjects=0;
 fSelect=0;
 
 // Introduce this NcJob based instance into the ROOT task list
 TSeqCollection* tasks=gROOT->GetListOfTasks();
 if (tasks) tasks->Add(this);
 
 NcVersion version;
 version.Data();
}

NcJob::~NcJob()
{
 
 // Remove this NcJob based instance into the ROOT task list
 TSeqCollection* tasks=gROOT->GetListOfTasks();
 if (tasks) tasks->Remove(this);
 
 if (fObjects)
 {
  delete fObjects;
  fObjects=0;
 }
 if (fFolder)
 {
  TList* list=gROOT->GetListOfBrowsables();
  if (list)
  {
   TFolder* top=(TFolder*)list->FindObject("NcJob-folders");
   if (top) RecursiveRemove(fFolder);
  }
  delete fFolder;
  fFolder=0;
 }
 if (fSelect)
 {
  delete fSelect;
  fSelect=0;
 }
 
 // Remove the subtasks from the internal TTask list without deleting them
 if (fTasks) fTasks->Clear();
}

void NcJob::ListEnvironment()
{
 
 cout << " ***" << endl;
 cout << " *** Environment of job " << GetName() << " ***" << endl;
 cout << " ***" << endl;
 cout << " === Available (sub)tasks : " << endl;
 ls();
 if (fFolder)
 {
  cout << " === Current job-folder contents : " << endl;
  fFolder->ls();
 }
 cout << endl;
}

void NcJob::ExecuteJob(Int_t mode)
{
 
 if (mode<0) fMakefolder=-1;
 if (mode==1 || mode==11) fMakefolder=1;
 
 gROOT->cd(); 
 Exec(GetName());
 
 if (abs(mode)>9)
 {
  CleanTasks();
  ExecuteTasks(GetName());
 }
}

void NcJob::MakeFolder()
{
 
 if (!fMakefolder) fMakefolder=1;
}

TFolder* NcJob::GetFolder() const
{
 
 return fFolder;
}

TObject* NcJob::GetMainObject() const
{
 
 return fMainObject;
}

void NcJob::SetMainObject(TObject* obj)
{
 
 if (obj)
 {
  fMainObject=obj;
  AddObject(obj);
 }
}

void NcJob::AddObject(TObject* obj)
{
 
 if (!obj) return;
 
 if (!fObjects) fObjects=new TObjArray();
 
 if (fMakefolder>0 && !fFolder)
 {
  // Create the top level environment folder for all NcJobs if needed 
  TList* list=gROOT->GetListOfBrowsables();
  if (list)
  {
   TFolder* top=(TFolder*)list->FindObject("NcJob-folders");
   if (!top)
   {
    top=new TFolder("NcJob-folders","Environment for all NcJob derived tasks");
    list->Add(top,"NcJob-folders");
   }
   // Create the task-specific folder as a sub-folder in the top folder 
   fFolder=top->AddFolder(GetName(),GetTitle());
  }
 }
 
 // Add object pointer to array and folder if it doesn't already exist 
 Int_t exist=0;
 for (Int_t i=0; i<fObjects->GetEntries(); i++)
 {
  if (obj==fObjects->At(i))
  {
   exist=1;
   break;
  }
 }
 if (!exist)
 {
  fObjects->Add(obj);
  if (fFolder) fFolder->Add(obj);
 }
}

void NcJob::AddObjects(TObjArray* arr)
{
 
 if (!arr) return;
 
 TObject* obj=0;
 for (Int_t i=0; i<arr->GetSize(); i++)
 {
  obj=arr->At(i);
  if (obj) AddObject(obj);
 }
}

void NcJob::RemoveObject(TObject* obj)
{
 
 if (!obj) return;
 
 if (fObjects)
 {
  TObject* test=fObjects->Remove(obj);
  if (test)
  {
   fObjects->Compress();
   if (test==fMainObject) fMainObject=0;
  }
 }
 
 if (fFolder) fFolder->Remove(obj);
}

void NcJob::RemoveObjects(const char* classname)
{
 
 if (!fObjects) return;
 
 Int_t remove=0;
 for (Int_t i=0; i<fObjects->GetEntries(); i++)
 {
  TObject* obj=fObjects->At(i);
  if (obj->InheritsFrom(classname))
  {
   TObject* test=fObjects->Remove(obj);
   if (test)
   {
    remove=1;
    if (test==fMainObject) fMainObject=0;
   }
   if (fFolder) fFolder->Remove(obj);
  }
 }
 if (remove) fObjects->Compress();
}

TObject* NcJob::GetObject(Int_t j) const
{
 
 if (!fObjects || j<1) return 0;
 
 TObject* obj=0;
 
 if (j<=fObjects->GetEntries()) obj=fObjects->At(j-1);
 return obj;
}

TObject* NcJob::GetObject(const char* classname) const
{
 
 if (!fObjects) return 0;
 
 TObject* obj=0;
 for (Int_t i=0; i<fObjects->GetEntries(); i++)
 {
  TObject* obx=fObjects->At(i);
  if (obx->InheritsFrom(classname))
  {
   obj=obx;
   break;
  }
 }
 return obj;
}

TObjArray* NcJob::GetObjects() const
{
 
 return fObjects;
}

TObjArray* NcJob::GetObjects(const char* classname)
{
 
 if (!fObjects) return 0;
 
 if (fSelect)
 {
  fSelect->Clear();
 }
 else
 {
  fSelect=new TObjArray();
 }
 
 for (Int_t i=0; i<fObjects->GetEntries(); i++)
 {
  TObject* obj=fObjects->At(i);
  if (obj->InheritsFrom(classname)) fSelect->Add(obj);
 }
 return fSelect;
}

void NcJob::ProcessObject(TObject* obj)
{
 
 if (!obj) return;
 
 SetMainObject(obj);
 
 gROOT->cd(); 
 CleanTasks();
 ExecuteTasks(GetName());
 
 RemoveObject(obj);
}