RunAncillaryFilter.C

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <fstream>
#include <sstream>
#include <cstdio>

#include "TSystemDirectory.h"
#include "TString.h"
#include "TList.h"
#include "TObject.h"
#include "TFile.h"
#include "TChain.h"
#include "TTree.h"
#include "TROOT.h"

using namespace std;

void RunAncillaryFilter(TString SourceDir, TString TreeName)
{
    TSystemDirectory Dir;
    Dir.SetDirectory(SourceDir);

    TChain *MyChain = new TChain(TreeName);

    TList *ListOfSubFolders = Dir.GetListOfFiles();
    for(int i = 0 ; i<ListOfSubFolders->GetSize() ; i++)
    {
        TObject *obj = ListOfSubFolders->At(i);

        TString Name = obj->GetName();

        if(Name.EndsWith(".root"))
        {
            TString Path = Form("%s/%s",SourceDir.Data(),Name.Data());

            TFile *file = new TFile(Path);
            TTree *t = (TTree*)file->Get(TreeName.Data());
            if(t && !((TObject*)t)->InheritsFrom("TChain"))
            {
                Path = Form("%s/%s/%s",SourceDir.Data(),Name.Data(),TreeName.Data());
                MyChain->Add(Path.Data());
            }
        }
    }

    cout<<"TChain build from files:"<<endl;

    MyChain->GetListOfFiles()->ls();

    MyChain->Process("AncillaryFilter.C+g");

    gROOT->ProcessLine(".q");
}