added max events to process

[u/mrichter/AliRoot.git] / PWG1 / AliGenInfoTask.h

#ifndef ALIGENINFOTASK_H\r
#define ALIGENINFOTASK_H\r
\r
// ROOT includes\r
#include <TList.h>\r
#include <TH1.h>\r
#include <TH2.h>\r
\r
// AliRoot includes\r
#include <AliAnalysisTask.h>\r
#include <AliESDEvent.h>\r
#include <AliESDfriend.h>\r
#include <AliESDtrack.h>\r
#include <AliESDfriendTrack.h>\r
#include <AliTPCseed.h>\r
class AliGenInfoMaker;\r
\r
class AliGenInfoTask : public AliAnalysisTask {\r
 public:\r
 AliGenInfoTask();\r
 AliGenInfoTask(const char *name);\r
  virtual ~AliGenInfoTask() {};\r
  \r
  virtual void   ConnectInputData(Option_t *);\r
  virtual void   CreateOutputObjects();\r
  virtual void   Exec(Option_t *option);\r
  virtual void   Terminate(Option_t *);\r
  virtual void   SetDebugLevel(Int_t level) {fDebug = level;}\r
  virtual void   SetMaxTracks(Int_t max=10) {fMaxTracks = max;}\r
  \r
 protected:\r
  virtual Int_t FillTrackHistograms(Int_t nTracks, AliESDtrack* track, \r
				    AliESDfriendTrack* friendTrack, \r
				    AliTPCseed* seed);\r
  AliGenInfoMaker *fGenMaker;    // gen Maker\r
  Int_t         fDebug;          //  Debug flag\r
  AliESDEvent*  fESD;            //! ESD\r
  AliESDfriend* fESDfriend;      //! ESD friend\r
  TList*        fListOfHists;    //! Output list of histograms\r
  \r
  Int_t         fMaxTracks;      // Max tracks in histogram\r
  TH1F*         hESDTracks;      //! N ESD tracks\r
  TH1F*         hGoodTracks;     //! GOOD tracks\r
\r
  ClassDef(AliGenInfoTask, 1); // Analysis task base class for TPC tracks and clusters\r
};\r
\r
#endif\r