[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectronCF.h

#ifndef ALIDIELECTRONCF_H
#define ALIDIELECTRONCF_H
/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//#############################################################
//#                                                           #
//#             Class AliDielectronCF                         #
//#       Dielectron Correction Framework Manager             #
//#                                                           #
//#  Authors:                                                 #
//#   Anton     Andronic, GSI / A.Andronic@gsi.de             #
//#   Ionut C.  Arsene,   GSI / I.C.Arsene@gsi.de             #
//#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
//#   Frederick Kramer,   Uni Ffm, / Frederick.Kramer@cern.ch #
//#   Magnus    Mager,    CERN / Magnus.Mager@cern.ch         #
//#   WooJin J. Park,     GSI / W.J.Park@gsi.de               #
//#   Jens      Wiechula, Uni HD / Jens.Wiechula@cern.ch      #
//#                                                           #
//#############################################################


#include <TNamed.h>
#include "AliDielectronVarManager.h"

class AliAnalysisCuts;
class AliAnalysisFilter;
class AliCFContainer;

class AliDielectronCF : public TNamed {
public:
  enum {kNmaxAddSteps=50};
  
  AliDielectronCF();
  AliDielectronCF(const char* name, const char* title);
  virtual ~AliDielectronCF();

  void SetStepForMCtruth(Bool_t steps=kTRUE)           { fStepForMCtruth=steps;           }
  void SetStepForNoCutsMCmotherPid(Bool_t steps=kTRUE) { fStepForNoCutsMCmotherPid=steps; }
  void SetStepForAfterAllCuts(Bool_t steps=kTRUE)      { fStepForAfterAllCuts=steps;      }
  void SetStepsForEachCut(Bool_t steps=kTRUE)          { fStepsForEachCut=steps;          }
  void SetStepsForCutsIncreasing(Bool_t steps=kTRUE)   { fStepsForCutsIncreasing=steps;   }
  
  void SetPdgMother(Int_t pdg) { fPdgMother=pdg; }
  
  void AddStepMask(UInt_t mask)                  { fStepMasks[fNStepMasks++]=mask; }
  
  void AddVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins, Double_t min, Double_t max);

  void InitialiseContainer(const AliAnalysisFilter& filter);
  
  void Fill(UInt_t mask, const TObject *particle);
  void FillMC(const TObject *particle);
  
  AliCFContainer* GetContainer() const { return fCfContainer; }
  
// private:
  UInt_t          fVariables[AliDielectronVarManager::kNMaxValues]; //configured variables
  
  Int_t           fNSteps;                     // number of selection steps
  Int_t           fNVars;                      // number of variables
  Int_t           fNBins[kNmaxAddSteps];       // array of numbers ob bins of the vars
  Double_t        fVarLoLimit[kNmaxAddSteps];  // array of the lower limits of the vars
  Double_t        fVarUpLimit[kNmaxAddSteps];  // array of the upper limits of the vars

  Int_t           fNCuts;                      // Number of cuts in the filter concerned

  Bool_t fStepForMCtruth;               //create a step for the MC truth
  Bool_t fStepForNoCutsMCmotherPid;     //create a step for before cuts, but with MC truth of the mother
  Bool_t fStepForAfterAllCuts;          //create a step for before cuts, but with MC truth of the mother
  Bool_t fStepsForEachCut;              //create steps for each cut?
  Bool_t fStepsForCutsIncreasing;       //create steps for increasing cut combinatons?
                                        //e.g. cut1&cut2, cut1&cut2&cut3 ...

  UInt_t fStepMasks[kNmaxAddSteps];      //steps for additional cut combinatons
  UInt_t fNStepMasks;                    //number of configured step masks

  Int_t fPdgMother;                      //Pdg code of MCtruth validation
  AliCFContainer* fCfContainer;          //the CF container
  
  AliDielectronCF(const AliDielectronCF &c);
  AliDielectronCF &operator=(const AliDielectronCF &c);
  
  ClassDef(AliDielectronCF,2)  //Dielectron Correction Framework handler
};

#endif
Commit	Line	Data
b2a297fa	1	#ifndef ALIDIELECTRONCF_H
	2	#define ALIDIELECTRONCF_H
	3	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	//#############################################################
	7	//# #
	8	//# Class AliDielectronCF #
	9	//# Dielectron Correction Framework Manager #
	10	//# #
	11	//# Authors: #
	12	//# Anton Andronic, GSI / A.Andronic@gsi.de #
	13	//# Ionut C. Arsene, GSI / I.C.Arsene@gsi.de #
	14	//# Julian Book, Uni Ffm / Julian.Book@cern.ch #
	15	//# Frederick Kramer, Uni Ffm, / Frederick.Kramer@cern.ch #
	16	//# Magnus Mager, CERN / Magnus.Mager@cern.ch #
	17	//# WooJin J. Park, GSI / W.J.Park@gsi.de #
	18	//# Jens Wiechula, Uni HD / Jens.Wiechula@cern.ch #
	19	//# #
	20	//#############################################################
	21
	22
	23
	24	#include <TNamed.h>
	25	#include "AliDielectronVarManager.h"
	26
	27	class AliAnalysisCuts;
	28	class AliAnalysisFilter;
	29	class AliCFContainer;
	30
	31	class AliDielectronCF : public TNamed {
	32	public:
	33	enum {kNmaxAddSteps=50};
	34
	35	AliDielectronCF();
	36	AliDielectronCF(const char* name, const char* title);
	37	virtual ~AliDielectronCF();
	38
e123f993	39	void SetStepForMCtruth(Bool_t steps=kTRUE) { fStepForMCtruth=steps; }
	40	void SetStepForNoCutsMCmotherPid(Bool_t steps=kTRUE) { fStepForNoCutsMCmotherPid=steps; }
	41	void SetStepForAfterAllCuts(Bool_t steps=kTRUE) { fStepForAfterAllCuts=steps; }
	42	void SetStepsForEachCut(Bool_t steps=kTRUE) { fStepsForEachCut=steps; }
	43	void SetStepsForCutsIncreasing(Bool_t steps=kTRUE) { fStepsForCutsIncreasing=steps; }
b2a297fa	44
	45	void SetPdgMother(Int_t pdg) { fPdgMother=pdg; }
	46
	47	void AddStepMask(UInt_t mask) { fStepMasks[fNStepMasks++]=mask; }
	48
	49	void AddVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins, Double_t min, Double_t max);
	50
	51	void InitialiseContainer(const AliAnalysisFilter& filter);
	52
	53	void Fill(UInt_t mask, const TObject *particle);
	54	void FillMC(const TObject *particle);
	55
	56	AliCFContainer* GetContainer() const { return fCfContainer; }
	57
e123f993	58	// private:
b2a297fa	59	UInt_t fVariables[AliDielectronVarManager::kNMaxValues]; //configured variables
	60
	61	Int_t fNSteps; // number of selection steps
	62	Int_t fNVars; // number of variables
	63	Int_t fNBins[kNmaxAddSteps]; // array of numbers ob bins of the vars
	64	Double_t fVarLoLimit[kNmaxAddSteps]; // array of the lower limits of the vars
	65	Double_t fVarUpLimit[kNmaxAddSteps]; // array of the upper limits of the vars
	66
	67	Int_t fNCuts; // Number of cuts in the filter concerned
	68
e123f993	69	Bool_t fStepForMCtruth; //create a step for the MC truth
	70	Bool_t fStepForNoCutsMCmotherPid; //create a step for before cuts, but with MC truth of the mother
	71	Bool_t fStepForAfterAllCuts; //create a step for before cuts, but with MC truth of the mother
	72	Bool_t fStepsForEachCut; //create steps for each cut?
	73	Bool_t fStepsForCutsIncreasing; //create steps for increasing cut combinatons?
	74	//e.g. cut1&cut2, cut1&cut2&cut3 ...
b2a297fa	75
	76	UInt_t fStepMasks[kNmaxAddSteps]; //steps for additional cut combinatons
	77	UInt_t fNStepMasks; //number of configured step masks
	78
	79	Int_t fPdgMother; //Pdg code of MCtruth validation
	80	AliCFContainer* fCfContainer; //the CF container
	81
	82	AliDielectronCF(const AliDielectronCF &c);
	83	AliDielectronCF &operator=(const AliDielectronCF &c);
	84
e123f993	85	ClassDef(AliDielectronCF,2) //Dielectron Correction Framework handler
b2a297fa	86	};
	87
	88	#endif