MadUtilities.cxx

Go to the documentation of this file.

#include "Mad/MadUtilities.h"
#include "glob.h"
#include <iostream>
#include <string>

#include "TFile.h"
#include "TChain.h"
#include "TList.h"
#include "TIterator.h"
#include "TObject.h"
#include "TRegexp.h"
#include "TSystem.h"
#include "TROOT.h"

Bool_t MadUtilities::MakeChains(const char* fileName)
{

  TChain *chain[5];
  chain[0] = 0;
  chain[1] = 0;
  chain[2] = 0;
  chain[3] = 0;
  chain[4] = 0;

  TString filnom = MadUtilities::GetOneFile(fileName);
  //Check file for various trees:
  bool foundSR = false;
  bool foundMC = false;
  bool foundTH = false;
  bool foundEM = false;
  bool foundST = false;
  TFile *tmp = new TFile(filnom,"READ");
  TList *list = tmp->GetListOfKeys();
  TIterator *iter = list->MakeIterator();
  for(int i=0;i<list->GetSize();i++){
    TObject *obj = iter->Next();
    std::string nom(obj->GetName());
    if(nom=="NtpSR") foundSR=true;
    else if(nom=="NtpMC") foundMC=true;
    else if(nom=="NtpTH") foundTH=true;
    else if(nom=="NtpEM") foundEM=true;
    else if(nom=="NtpSt") foundST=true;
  }
  delete tmp;

  if(foundSR){
    chain[0] = new TChain("NtpSR");
    chain[0]->Add(fileName,-1);        
    std::cout << "Adding NtpSR Stream to Mad" << std::endl;
  }
  if(foundMC){
    chain[1] = new TChain("NtpMC");
    chain[1]->Add(fileName,-1);
    std::cout << "Adding NtpMC Stream to Mad" << std::endl;
  }
  if(foundTH){
    chain[2] = new TChain("NtpTH");
    chain[2]->Add(fileName,-1);
    std::cout << "Adding NtpTH Stream to Mad" << std::endl;  
  }
  if(foundEM){
    chain[3] = new TChain("NtpEM");
    chain[3]->Add(fileName,-1);
    std::cout << "Adding NtpEM Stream to Mad" << std::endl;
  }
  if(foundST){
    chain[4] = new TChain("NtpSt");
    chain[4]->Add(fileName,-1);
    std::cout << "Adding NtpSt Stream to Mad" << std::endl;
  }
  if(foundSR || foundST) return true;
  return false;
}

TString MadUtilities::GetOneFile(const char* infiles){
  
  // find one of the files specified:
  // in the possibly wildcarded infiles
  TString rval;
  glob_t found_files;
  int flags = 0;
  int n = glob(infiles,flags,NULL,&found_files);
  if(n!=0){
    std::cout<<"MadUtilties::GetOneFile(): error when globbing '"
             <<infiles<<"'"<<std::endl;
    return rval;
  }
  if(found_files.gl_pathc == 0){
    std::cout<<"MadUtilities::GetOneFile(): '"
             <<infiles<<"' fails to expand to any files"<<std::endl;
    return rval;      
  }
  
  rval = found_files.gl_pathv[0];

  return rval;

  
}

int MadUtilities::GetTotEntries(const char* fileName){
  //find the files specified:
  TString basename(fileName);
  Int_t slashpos = basename.Last('/');
  TString directory;
  if (slashpos>=0) {
    directory = basename(0,slashpos); // Copy the directory name
    basename.Remove(0,slashpos+1);      // and remove it from basename
  } 
  else {
    directory = gSystem->WorkingDirectory();
  }
  const char *file;
  int ent=0;
  void *dir=gSystem->OpenDirectory(gSystem->ExpandPathName(directory.Data()));
  if (dir) {
    TRegexp re(basename,kTRUE);
    while ((file = gSystem->GetDirEntry(dir))) {      
      if (!strcmp(file,".") || !strcmp(file,"..")) continue;
      TString s = file;
      if ( (basename!=file) && s.Index(re) == kNPOS) continue;
      TFile *tmp = new TFile(directory+"/"+file,"READ");
      TTree *tree = (TTree*) tmp->Get("DaqSnarl");
      ent+=tree->GetEntries();
      delete tmp;
    }
  }
  return ent;
}

---