| 1 |
fgolf |
1.1 |
#include "TList.h"
|
| 2 |
|
|
#include "TObjArray.h"
|
| 3 |
|
|
#include "TH1.h"
|
| 4 |
|
|
#include "TH2.h"
|
| 5 |
|
|
#include "TROOT.h"
|
| 6 |
|
|
#include "TClass.h"
|
| 7 |
|
|
#include "TObjString.h"
|
| 8 |
|
|
#include <iostream>
|
| 9 |
|
|
|
| 10 |
|
|
#include "getMyHistosNames.h"
|
| 11 |
|
|
//--------------------------------------------------------
|
| 12 |
|
|
// Our histogram are named XX_YY_ZZ
|
| 13 |
|
|
// where XX=tt, WW, WZ, etc
|
| 14 |
|
|
// YY refers to what is actually plotted
|
| 15 |
|
|
// ZZ=em, ee, mm, all
|
| 16 |
|
|
//
|
| 17 |
|
|
// It is useful to get a list of all the YY's
|
| 18 |
|
|
//
|
| 19 |
|
|
// We can get this list by looking at all the existing hostograms.
|
| 20 |
|
|
//
|
| 21 |
|
|
// This list is returned here as TObjArray*
|
| 22 |
|
|
//
|
| 23 |
|
|
// The passed prefix should be, eg, "tt" and the passed postfix
|
| 24 |
|
|
// should be, eg, "ee", so that the list is built only
|
| 25 |
|
|
// from the "tt_YY_ee" histograms
|
| 26 |
|
|
//
|
| 27 |
|
|
// Claudio 4 Sep 2007
|
| 28 |
|
|
//
|
| 29 |
|
|
// Added feature to skip 2D histograms
|
| 30 |
|
|
//--------------------------------------------------------
|
| 31 |
|
|
|
| 32 |
|
|
TObjArray* getMyHistosNames (const char* prefix, const char* postfix, bool keep2D)
|
| 33 |
|
|
{
|
| 34 |
|
|
TObjArray* array = new TObjArray();
|
| 35 |
|
|
bool skip2D = !keep2D;
|
| 36 |
|
|
|
| 37 |
|
|
// Get a list of object and their iterator
|
| 38 |
|
|
TList* list = gDirectory->GetList() ;
|
| 39 |
|
|
TIterator* iter = list->MakeIterator();
|
| 40 |
|
|
|
| 41 |
|
|
// Loop over objects
|
| 42 |
|
|
TObject* obj;
|
| 43 |
|
|
while ((obj = iter->Next()))
|
| 44 |
|
|
{
|
| 45 |
|
|
// Only look at objects beginning with 'prefix' and ending with 'postfix'
|
| 46 |
|
|
TString name = obj->GetName();
|
| 47 |
|
|
if (name.BeginsWith(prefix) && name.EndsWith(postfix))
|
| 48 |
|
|
{
|
| 49 |
|
|
if (skip2D && obj->IsA()->InheritsFrom(TH2::Class()))
|
| 50 |
|
|
continue;
|
| 51 |
|
|
|
| 52 |
|
|
// Only look at objects that inherit from TH1 or TH2
|
| 53 |
|
|
if (obj->IsA()->InheritsFrom(TH1::Class()) || obj->IsA()->InheritsFrom(TH2::Class()))
|
| 54 |
|
|
{
|
| 55 |
|
|
// Find the central key, ie, the YY
|
| 56 |
|
|
TObjArray* t = name.Tokenize("_");
|
| 57 |
|
|
TString z = TString(t->At(1)->GetName());
|
| 58 |
|
|
if (t->GetEntries() == 4)
|
| 59 |
|
|
z = TString(Form("%s_%s", t->At(1)->GetName(), t->At(2)->GetName()));
|
| 60 |
|
|
|
| 61 |
|
|
// add to the output array
|
| 62 |
|
|
TObjString* zobj = new TObjString(z);
|
| 63 |
|
|
array->Add(zobj);
|
| 64 |
|
|
}
|
| 65 |
|
|
}
|
| 66 |
|
|
}
|
| 67 |
|
|
|
| 68 |
|
|
return array;
|
| 69 |
|
|
}
|
