[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliJetEmbeddingFromPYTHIATask.h

#ifndef ALIJETEMBEDDINGFROMPYTHIATASK_H
#define ALIJETEMBEDDINGFROMPYTHIATASK_H

// $Id: AliJetEmbeddingFromPYTHIATask.h  $

#include "AliJetEmbeddingFromAODTask.h"
#include <TArrayD.h>

template<class T> 
class TParameter;

class TString;

class AliJetEmbeddingFromPYTHIATask : public AliJetEmbeddingFromAODTask {
 public:
  AliJetEmbeddingFromPYTHIATask();
  AliJetEmbeddingFromPYTHIATask(const char *name, Bool_t drawqa=kFALSE); 
  virtual ~AliJetEmbeddingFromPYTHIATask();

  Bool_t         UserNotify();
  void           UserCreateOutputObjects();

  void           SetPYTHIAPath(const char* p)                      { fPYTHIAPath                                = p ; }
  void           SetPtHardBinScaling(Int_t n, Double_t *scaling)   { new (&fPtHardBinScaling) TArrayD(n, scaling)   ; }
  void           SetAnchorRun(Int_t r)                             { fAnchorRun                                 = r ; }
  void           SetLHC11hAnchorRuns(Bool_t a=kTRUE)               { fLHC11hAnchorRun                           = a ; }

 protected:
  Bool_t         ExecOnce()           ;// intialize task
  Bool_t         GetNextEntry()       ;// get next entry in current tree
  Int_t          GetRandomPtHardBin() ;// get a radnom pt hard bin according to fPtHardBinScaling
  TString        GetNextFileName()    ;// get next file name

  TString        fPYTHIAPath          ;// Path of the PYTHIA production
  TArrayD        fPtHardBinScaling    ;// Pt hard bin scaling
  Bool_t         fLHC11hAnchorRun     ;// LHC11h anchor runs
  Int_t          fAnchorRun           ;// Anchor run
  Int_t          fCurrentPtHardBin    ;//!Pt hard bin of the current open file
  TParameter<int> *fPtHardBinParam    ;//!Pt hard bin param

  TH1           *fHistPtHardBins      ;//!Embeded pt hard bin distribution

 private:
  AliJetEmbeddingFromPYTHIATask(const AliJetEmbeddingFromPYTHIATask&);            // not implemented
  AliJetEmbeddingFromPYTHIATask &operator=(const AliJetEmbeddingFromPYTHIATask&); // not implemented

  ClassDef(AliJetEmbeddingFromPYTHIATask, 1) // Jet embedding from PYTHIA task
};
#endif
Commit	Line	Data
a6e51e61	1	#ifndef ALIJETEMBEDDINGFROMPYTHIATASK_H
	2	#define ALIJETEMBEDDINGFROMPYTHIATASK_H
	3
	4	// $Id: AliJetEmbeddingFromPYTHIATask.h $
	5
	6	#include "AliJetEmbeddingFromAODTask.h"
	7	#include <TArrayD.h>
	8
ca5c29fa	9	template<class T>
	10	class TParameter;
	11
a6e51e61	12	class TString;
	13
	14	class AliJetEmbeddingFromPYTHIATask : public AliJetEmbeddingFromAODTask {
	15	public:
	16	AliJetEmbeddingFromPYTHIATask();
ca5c29fa	17	AliJetEmbeddingFromPYTHIATask(const char *name, Bool_t drawqa=kFALSE);
a6e51e61	18	virtual ~AliJetEmbeddingFromPYTHIATask();
	19
	20	Bool_t UserNotify();
ca5c29fa	21	void UserCreateOutputObjects();
a6e51e61	22
	23	void SetPYTHIAPath(const char* p) { fPYTHIAPath = p ; }
	24	void SetPtHardBinScaling(Int_t n, Double_t *scaling) { new (&fPtHardBinScaling) TArrayD(n, scaling) ; }
	25	void SetAnchorRun(Int_t r) { fAnchorRun = r ; }
	26	void SetLHC11hAnchorRuns(Bool_t a=kTRUE) { fLHC11hAnchorRun = a ; }
	27
	28	protected:
	29	Bool_t ExecOnce() ;// intialize task
	30	Bool_t GetNextEntry() ;// get next entry in current tree
	31	Int_t GetRandomPtHardBin() ;// get a radnom pt hard bin according to fPtHardBinScaling
	32	TString GetNextFileName() ;// get next file name
	33
	34	TString fPYTHIAPath ;// Path of the PYTHIA production
	35	TArrayD fPtHardBinScaling ;// Pt hard bin scaling
	36	Bool_t fLHC11hAnchorRun ;// LHC11h anchor runs
	37	Int_t fAnchorRun ;// Anchor run
	38	Int_t fCurrentPtHardBin ;//!Pt hard bin of the current open file
ca5c29fa	39	TParameter<int> *fPtHardBinParam ;//!Pt hard bin param
a6e51e61	40
ca5c29fa	41	TH1 *fHistPtHardBins ;//!Embeded pt hard bin distribution
a6e51e61	42
	43	private:
	44	AliJetEmbeddingFromPYTHIATask(const AliJetEmbeddingFromPYTHIATask&); // not implemented
	45	AliJetEmbeddingFromPYTHIATask &operator=(const AliJetEmbeddingFromPYTHIATask&); // not implemented
	46
	47	ClassDef(AliJetEmbeddingFromPYTHIATask, 1) // Jet embedding from PYTHIA task
	48	};
	49	#endif