TSelEventGen.h

Go to the documentation of this file.

// @(#)root/proof:$Id$
// Author: Sangsu Ryu 22/06/2010
 
/*************************************************************************
 * Copyright (C) 1995-2005, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#ifndef ROOT_TSelEventGen
#define ROOT_TSelEventGen
 
//////////////////////////////////////////////////////////////////////////
//                                                                      //
// TSelEventGen                                                         //
//                                                                      //
// PROOF selector for event file generation.                            //
// List of files to be generated for each node is provided by client.   //
// And list of files generated is sent back.                            //
// Existing files are reused if not forced to be regenerated.           //
//                                                                      //
//////////////////////////////////////////////////////////////////////////
 
#include <TSelector.h>
#include <TTree.h>
#include <TString.h>
 
class TList;
class TMacro;
 
class TSelEventGen : public TSelector {
 
private:
 
   TString fBaseDir;                          // URL where the files will be written
   //Int_t fMaxNWorkers;
   Long64_t fNEvents;                         //number of events in a file
   Int_t fNTracks;                            //avg or min-avg number of tracks in an event
   Int_t fNTracksMax;                         //max-avg number of tracks in an event
   Int_t fRegenerate;                         //force generation of cleanup files
 
   TObject* fTotalGen;                        //events generated on this worker
   TList* fFilesGenerated;                    //list of files generated
 
   TMacro  *fGenerateFun;                     //Macro with the function to generate the files
 
protected:
 
   Long64_t GenerateFiles(const char *filename, Long64_t sizenevents);
 
public :
 
   TTree          *fChain;   //!pointer to the analyzed TTree or TChain
 
//   TSelEventGen(TTree *);
   TSelEventGen();
   ~TSelEventGen() override { }
   Int_t   Version() const override {return 1;}
   void    Begin(TTree *) override;
   void    SlaveBegin(TTree *tree) override;
   void    Init(TTree *tree) override;
   Bool_t  Notify() override;
   Bool_t  Process(Long64_t entry) override;
   void    SetOption(const char *option) override { fOption = option; }
   void    SetObject(TObject *obj) override { fObject = obj; }
   void    SetInputList(TList *input) override {fInput = input;}
   TList  *GetOutputList() const override { return fOutput; }
   void    SlaveTerminate() override;
   void    Terminate() override;
   void    Print(Option_t *option="") const override;
 
   ClassDefOverride(TSelEventGen,0)     //PROOF selector for event file generation
};
 
#endif
 
#ifdef TSelEventGen_cxx
 
void TSelEventGen::Init(TTree *tree)
{
   // The Init() function is called when the selector needs to initialize
   // a new tree or chain. Typically here the branch addresses of the tree
   // will be set. It is normaly not necessary to make changes to the
   // generated code, but the routine can be extended by the user if needed.
   // Init() will be called many times when running with PROOF.
 
   if (tree == 0) return;
   fChain = tree;
   fChain->SetMakeClass(1);
}
 
Bool_t TSelEventGen::Notify()
{
   // The Notify() function is called when a new file is opened. This
   // can be either for a new TTree in a TChain or when when a new TTree
   // is started when using PROOF. Typically here the branch pointers
   // will be retrieved. It is normaly not necessary to make changes
   // to the generated code, but the routine can be extended by the
   // user if needed.
 
   return kTRUE;
}
 
#endif // #ifdef TSelEventGen_cxx