LocalRailway.C

Go to the documentation of this file.

#ifndef LOCALHELPER_C
#define LOCALHELPER_C
#ifndef __CINT__
# include "Railway.C"
# include "ChainBuilder.C"
# include <TUrl.h>
# include <TString.h>
# include <TSystem.h>
# include <AliAnalysisManager.h>
#else
class TChain;
class Railway;
class TUrl;
#endif

// ===================================================================
struct LocalRailway : public Railway
{
  LocalRailway(const TUrl& url, Int_t verbose)
    : Railway(url, verbose), fChain(0)
  {
    fOptions.Add("recursive","Scan recursive");
    fOptions.Add("trackref", "For MC input, check TrackRef.root presence");
    fOptions.Add("scan",     "Scan for number of events in chain");
    fOptions.Add("clean",    "Clean chain elements");
    fOptions.Add("pattern",  "GLOB", "File name pattern", "*.root");
  }
  LocalRailway(const LocalRailway& o) 
    : Railway(o), fChain(o.fChain)
  {}
  LocalRailway& operator=(const LocalRailway& o) 
  {
    if (&o == this) return *this;
    Railway::operator=(o);
    fChain = o.fChain;
    return *this;
  }
  virtual ~LocalRailway() {}
  virtual Bool_t LoadLibrary(const TString& name, Bool_t, Bool_t)
  {
    Int_t ret = gSystem->Load(MakeLibraryName(name));
    return ret >= 0;
  }
  virtual UShort_t Mode() const { return kLocal; }
  virtual const char* ModeString() const { return "local"; }
  virtual Bool_t PreSetup() 
  {
    return true;
  }
  virtual Bool_t PostSetup() 
  {
    // --- Check that the manager was made ---------------------------
    AliAnalysisManager* mgr = AliAnalysisManager::GetAnalysisManager();
    if (!mgr) { 
      Error("PostSetup", "No analysis manager defined");
      return false;
    }

    // --- Create the chain ------------------------------------------
    fChain = LocalChain();
    if (!fChain) return false;
    
    return true;
  };
  virtual Long64_t Run(Long64_t nEvents=-1) 
  {
    AliAnalysisManager* mgr = AliAnalysisManager::GetAnalysisManager();
    
    if (nEvents < 0) nEvents = fChain->GetEntries();
    return mgr->StartAnalysis(fUrl.GetProtocol(), fChain, nEvents);
  }
  virtual const Char_t* UrlHelp() const 
  {
    return "local://<datadir or list>[?<options>][#<treeName>]";
  }
  virtual const char* Desc() const { return "Local"; }
  TChain* fChain; // Our chain
};
#endif
//
// EOF
//