RunFinalQA.C

Go to the documentation of this file.

//____________________________________________________________________
Bool_t ScanDirectory(TSystemDirectory* dir, 
         TList* list, 
         const char* pattern, 
         bool recursive=false)
{
  TString fnPattern = pattern;

  // Assume failure 
  Bool_t ret = false;

  // Get list of files, and go back to old working directory
  TString oldDir(gSystem->WorkingDirectory());
  TList*  files = dir->GetListOfFiles();
  if (!gSystem->ChangeDirectory(oldDir)) { 
    Error("ScanDirectory", "Failed to go back to %s", oldDir.Data());
    return false;
  }
  if (!files) {
    Warning("ScanDirectory", "No files found in %s", dir->GetName());
    return false;
  }
  // files->ls();

  TList toAdd;
    
  // Sort list of files and check if we should add it 
  // files->Sort();
  TIter next(files);
  TSystemFile* file = 0;
  while ((file = static_cast<TSystemFile*>(next()))) {
    TString name(file->GetName());
    TString title(file->GetTitle());
    TString full(gSystem->ConcatFileName(file->GetTitle(), name.Data()));
    if (file->IsA() == TSystemDirectory::Class()) full = title;
    if (name == "." || name == "..") { 
      continue;
    }

    FileStat_t fs;
    if (gSystem->GetPathInfo(full.Data(), fs)) {
      Warning("ScanDirectory", "Cannot stat %s (%s)", full.Data(),
        gSystem->WorkingDirectory());
      continue;
    }
    // Check if this is a directory 
    if (file->IsDirectory(full)) { 
      if (recursive) {
  TSystemDirectory* d = new TSystemDirectory(file->GetName(),
               full.Data());
  if (ScanDirectory(d,list,pattern,recursive))
    ret = true;
  delete d;
      }
      continue;
    }
    
    // If this is not a root file, ignore 
    if (!name.EndsWith(".root")) {
      continue;
    }

    // If this file does not contain AliESDs, ignore 
    if (!name.Contains(fnPattern)) { 
      continue;
    }
    
    // Add 
    toAdd.Add(new TObjString(full));
  }

  TIter nextAdd(&toAdd);
  TObjString* s = 0;
  while ((s = static_cast<TObjString*>(nextAdd()))) {
    list->Add(s);
  }
  if (toAdd.GetEntries() > 0) ret = true;

  gSystem->ChangeDirectory(oldDir);
  return ret;
}
//____________________________________________________________________
TList*
GetListOfFiles(const char* input=".")
{
  TList* ret = new TList;
  TString dir(input);
  // if (dir == ".") dir = "";

  TString savdir(gSystem->WorkingDirectory());
  TSystemDirectory d(gSystem->BaseName(dir.Data()), dir.Data());
  if (!ScanDirectory(&d, ret, "tree_", false)) { 
    delete ret;
    ret = 0;
  }
  gSystem->ChangeDirectory(savdir);  
  if (ret) ret->Sort();
  return ret;
}

//____________________________________________________________________
void 
RunFinalQA(const char* dir, Int_t prodYear=0, const char* prodLetter="",
     Bool_t useVar=false)
{
   int ret = 0;
   gROOT->SetMacroPath(Form(".:%s",gROOT->GetMacroPath()));
   gSystem->Load("libGpad");
   gSystem->Load("libTree");

   gROOT->LoadMacro("QABase.h+g");
   gROOT->LoadMacro("QAPlotter.C+g");

   Info("RunFinalQA", "Final QA: %d%c (variance: %s)", 
  prodYear, prodLetter[0], (useVar ? "true" : "false"));
   QAPlotter p(prodYear, prodLetter[0], useVar);
  
   TList* l = GetListOfFiles(dir);
   TIter next(l);
   TObject* o = 0;
   while ((o = next())) {
     p.AddFile(o->GetName());
   }

   p.Run();
}
//
// EOF
//