mp103_processSelector.C

Go to the documentation of this file.

/// \file
/// \ingroup tutorial_multicore
/// \notebook -nodraw
/// Illustrate the usage of the multiproc TSelector interfaces with the h1 analysis
/// example.
///
/// \macro_code
///
/// \authors Anda Chelba, Gerardo Ganis

#include "TString.h"
#include "TROOT.h"
#include "TChain.h"
#include "TFileCollection.h"
#include "TH1F.h"
#include "TTreeReader.h"
#include "ROOT/TTreeProcessorMP.hxx"

const char *fh1[] = {"http://root.cern.ch/files/h1/dstarmb.root", "http://root.cern.ch/files/h1/dstarp1a.root",
                     "http://root.cern.ch/files/h1/dstarp1b.root", "http://root.cern.ch/files/h1/dstarp2.root"};

int mp103_processSelector()
{

   // MacOSX may generate connection to WindowServer errors
   gROOT->SetBatch(kTRUE);

   TString selectorPath = gROOT->GetTutorialDir();
   selectorPath += "/tree/h1analysis.C+";
   std::cout << "selector used is: " << selectorPath << "\n";
   TSelector *sel = TSelector::GetSelector(selectorPath);

// The following code generates a crash when Davix is used for HTTP
// Davix does not seem fork-safe; the problem has been reported to the
// Davix developers. For the time being we disable this part.
// To repoduce the problem, uncomment the next line.
//
// #define __reproduce_davix
#if defined(__reproduce_davix)
   TFile *fp = TFile::Open(fh1[0]);
   TTree *tree = (TTree *)fp->Get("h42");
#endif

   ROOT::TTreeProcessorMP pool(3);

   TList *out = nullptr;
#if defined(__reproduce_davix)
   // TTreeProcessorMP::Process with a single tree
   out = pool.Process(*tree, *sel);
   sel->GetOutputList()->Delete();
#endif

   // TTreeProcessorMP::Process with single file name and tree name
   // Note: we have less files than workers here
   out = pool.Process(fh1[0], *sel, "h42");
   sel->GetOutputList()->Delete();

   // Prepare datasets: vector of files, TFileCollection
   TChain ch;
   TFileCollection fc;
   std::vector<std::string> files;
   for (int i = 0; i < 4; i++) {
      files.push_back(fh1[i]);
      fc.Add(new TFileInfo(fh1[i]));
      ch.Add(fh1[i]);
   }

   // TTreeProcessorMP::Process with vector of files and tree name
   // Note: we have more files than workers here (different behaviour)
   out = pool.Process(files, *sel, "h42");
   sel->GetOutputList()->Delete();

   // TTreeProcessorMP::Process with TFileCollection, no tree name
   out = pool.Process(fc, *sel);
   sel->GetOutputList()->Delete();

   // TTreeProcessorMP::Process with TChain, no tree name
   out = pool.Process(ch, *sel);
   sel->GetOutputList()->Delete();

   return 0;
}