[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronHF.h

#ifndef ALIDIELECTRONHF_H
#define ALIDIELECTRONHF_H

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

//#############################################################
//#                                                           # 
//#         Class AliDielectronHF                     #
//#                                                           #
//#  Authors:                                                 #
//#   Julian Book, Uni-Frankfurt / Julian.Book@cern.ch        #
//#                                                           #
//#############################################################

#include <TNamed.h>
#include <TObjArray.h>
#include <TBits.h>
#include <THnBase.h>

#include "AliDielectronVarManager.h"
#include "AliDielectronHistos.h"

class AliDielectronHF : public TNamed {
public:
  enum { kMaxCuts=20 };
  enum EBinType {
    kStdBin=0,
    kBinToMax,
    kBinFromMin,
    kSymBin
  };
  enum EPairType {
    //    kOSonly=0,  kOSandLS, kOSandROT, kOSandMIX, kAll, kMConly
    kAll=0, kMConly,
    kSeAll,   kSeOnlyOS,
    kMeAll,   kMeOnlyOS,
    kSeMeAll, kSeMeOnlyOS,
    kSeReAll, kSeReOnlyOS,
  };

  AliDielectronHF();
  AliDielectronHF(const char*name, const char* title);

  virtual ~AliDielectronHF();

  void Init();
  void SetSignalsMC(TObjArray* array)    {fSignalsMC = array;}
  void SetStepForMCGenerated(Bool_t switcher=kTRUE)    {fStepGenerated = switcher;}
  void SetPairTypes(EPairType ptype) { fPairType=ptype; }
  void SetEventArray(Bool_t switcher=kTRUE) {fEventArray=switcher;}

  // functions to add 1-dimensional objects
  void UserProfile(const char* histClass, UInt_t valTypeP,
		   const TVectorD * const binsX, UInt_t valTypeX, TString option="",
		   UInt_t valTypeW=AliDielectronHistos::kNoWeights);
  void UserHistogram(const char* histClass,
		     const TVectorD * const binsX, UInt_t valTypeX, UInt_t valTypeW=AliDielectronHistos::kNoWeights)
  { UserProfile(histClass,AliDielectronHistos::kNoProfile,binsX,valTypeX,"",valTypeW); }

  // functions to add 2-dimensional objects
  void UserProfile(const char* histClass, UInt_t valTypeP,
		   const TVectorD * const binsX, const TVectorD * const binsY,
		   UInt_t valTypeX, UInt_t valTypeY, TString option="", UInt_t valTypeW=AliDielectronHistos::kNoWeights);
  void UserHistogram(const char* histClass,
                     const TVectorD * const binsX, const TVectorD * const binsY,
		     UInt_t valTypeX, UInt_t valTypeY, UInt_t valTypeW=AliDielectronHistos::kNoWeights)
  { UserProfile(histClass,AliDielectronHistos::kNoProfile,binsX,binsY,valTypeX,valTypeY,"",valTypeW); }

  // functions to add 3-dimensional objects
  void UserProfile(const char* histClass, UInt_t valTypeP,
		   const TVectorD * const binsX, const TVectorD * const binsY, const TVectorD * const binsZ,
		   UInt_t valTypeX, UInt_t valTypeY, UInt_t valTypeZ, TString option="",
		   UInt_t valTypeW=AliDielectronHistos::kNoWeights);
  void UserHistogram(const char* histClass,
                     const TVectorD * const binsX, const TVectorD * const binsY, const TVectorD * const binsZ,
                     UInt_t valTypeX, UInt_t valTypeY, UInt_t valTypeZ, UInt_t valTypeW=AliDielectronHistos::kNoWeights)
  { UserProfile(histClass,AliDielectronHistos::kNoProfile,binsX,binsY,binsZ,valTypeX,valTypeY,valTypeZ,"",valTypeW); }

  // functions to add multidimensional stuff
  void UserSparse(const char* histClass,
		  Int_t ndim, TObjArray *limits, UInt_t *vars, UInt_t valTypeW=AliDielectronHistos::kNoWeights);


  // functions to define the grid
  void AddCutVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins,
		      Double_t min, Double_t max, Bool_t log=kFALSE, Bool_t leg=kFALSE, EBinType btype=kStdBin);
  void AddCutVariable(AliDielectronVarManager::ValueTypes type, 
		      const char* binLimitStr, Bool_t leg=kFALSE, EBinType btype=kStdBin);
  void AddCutVariable(AliDielectronVarManager::ValueTypes type, 
		      TVectorD * binLimits, Bool_t leg=kFALSE, EBinType btype=kStdBin);

  void Fill(Int_t pairIndex, const AliDielectronPair *particle);
  void Fill(Int_t label1, Int_t label2, Int_t nSignal);
  void Fill(Int_t Index, Double_t * const valuesPair, Double_t * const valuesLeg1, Double_t * const valuesLeg2);

  Bool_t IsPairTypeSelected(Int_t itype);

  Int_t GetNumberOfBins() const;
  const TObjArray * GetHistArray() const { return &fArrPairType; }
  Bool_t GetStepForMCGenerated()   const { return fStepGenerated; }
  Bool_t IsEventArray()           const { return fEventArray; }
  
  
private:
  TBits     *fUsedVars;             // list of used variables

  TObjArray fArrPairType;           //-> array of pair types
  EPairType fPairType;              // which pair combinations to include
  TObjArray* fSignalsMC;            //! array of MC signals to be studied

  UShort_t  fVarCuts[kMaxCuts];     // cut variables
  TBits     *fVarCutType;           // array to store leg booleans
  //  Bool_t    fVarCutType[kMaxCuts];  // array to store leg booleans
  TObjArray fAxes;                  // Axis descriptions of the cut binning
  UShort_t  fBinType[kMaxCuts];     // binning type of the axes
  
  Bool_t    fHasMC;                 // is mc array
  Bool_t    fStepGenerated;         // switcher for generated particles
  Bool_t    fEventArray;            // switch OFF pair types and ON event array
  TObjArray fRefObj;               // reference object

  AliDielectronHF(const AliDielectronHF &c);
  AliDielectronHF &operator=(const AliDielectronHF &c);

  
  ClassDef(AliDielectronHF,6)         // Dielectron HF
};


#endif
Commit	Line	Data
5e2cf960	1	#ifndef ALIDIELECTRONHF_H
	2	#define ALIDIELECTRONHF_H
	3
	4	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//#############################################################
	8	//# #
	9	//# Class AliDielectronHF #
	10	//# #
	11	//# Authors: #
	12	//# Julian Book, Uni-Frankfurt / Julian.Book@cern.ch #
	13	//# #
	14	//#############################################################
	15
	16	#include <TNamed.h>
	17	#include <TObjArray.h>
a94c2e7e	18	#include <TBits.h>
775970eb	19	#include <THnBase.h>
5e2cf960	20
5e2cf960	21	#include "AliDielectronVarManager.h"
d4619886	22	#include "AliDielectronHistos.h"
5e2cf960	23
5e2cf960	24	class AliDielectronHF : public TNamed {
5e2cf960	25	public:
187de573	26	enum { kMaxCuts=20 };
5e2cf960	27	enum EBinType {
	28	kStdBin=0,
	29	kBinToMax,
	30	kBinFromMin,
	31	kSymBin
	32	};
	33	enum EPairType {
7957061b	34	// kOSonly=0, kOSandLS, kOSandROT, kOSandMIX, kAll, kMConly
	35	kAll=0, kMConly,
	36	kSeAll, kSeOnlyOS,
	37	kMeAll, kMeOnlyOS,
	38	kSeMeAll, kSeMeOnlyOS,
	39	kSeReAll, kSeReOnlyOS,
5e2cf960	40	};
	41
	42	AliDielectronHF();
	43	AliDielectronHF(const charname, const char title);
	44
	45	virtual ~AliDielectronHF();
443a091c	46
	47	void Init();
	48	void SetSignalsMC(TObjArray* array) {fSignalsMC = array;}
4d7704c5	49	void SetStepForMCGenerated(Bool_t switcher=kTRUE) {fStepGenerated = switcher;}
7957061b	50	void SetPairTypes(EPairType ptype) { fPairType=ptype; }
a9510cf1	51	void SetEventArray(Bool_t switcher=kTRUE) {fEventArray=switcher;}
5e2cf960	52
8d42b5b9	53	// functions to add 1-dimensional objects
8d42b5b9	54	void UserProfile(const char* histClass, UInt_t valTypeP,
d4619886	55	const TVectorD * const binsX, UInt_t valTypeX, TString option="",
d4619886	56	UInt_t valTypeW=AliDielectronHistos::kNoWeights);
8d42b5b9	57	void UserHistogram(const char* histClass,
d4619886	58	const TVectorD * const binsX, UInt_t valTypeX, UInt_t valTypeW=AliDielectronHistos::kNoWeights)
d4619886	59	{ UserProfile(histClass,AliDielectronHistos::kNoProfile,binsX,valTypeX,"",valTypeW); }
8d42b5b9	60
	61	// functions to add 2-dimensional objects
	62	void UserProfile(const char* histClass, UInt_t valTypeP,
	63	const TVectorD * const binsX, const TVectorD * const binsY,
d4619886	64	UInt_t valTypeX, UInt_t valTypeY, TString option="", UInt_t valTypeW=AliDielectronHistos::kNoWeights);
8d42b5b9	65	void UserHistogram(const char* histClass,
8d42b5b9	66	const TVectorD * const binsX, const TVectorD * const binsY,
d4619886	67	UInt_t valTypeX, UInt_t valTypeY, UInt_t valTypeW=AliDielectronHistos::kNoWeights)
d4619886	68	{ UserProfile(histClass,AliDielectronHistos::kNoProfile,binsX,binsY,valTypeX,valTypeY,"",valTypeW); }
8d42b5b9	69
	70	// functions to add 3-dimensional objects
	71	void UserProfile(const char* histClass, UInt_t valTypeP,
	72	const TVectorD * const binsX, const TVectorD * const binsY, const TVectorD * const binsZ,
d4619886	73	UInt_t valTypeX, UInt_t valTypeY, UInt_t valTypeZ, TString option="",
d4619886	74	UInt_t valTypeW=AliDielectronHistos::kNoWeights);
8d42b5b9	75	void UserHistogram(const char* histClass,
8d42b5b9	76	const TVectorD * const binsX, const TVectorD * const binsY, const TVectorD * const binsZ,
d4619886	77	UInt_t valTypeX, UInt_t valTypeY, UInt_t valTypeZ, UInt_t valTypeW=AliDielectronHistos::kNoWeights)
d4619886	78	{ UserProfile(histClass,AliDielectronHistos::kNoProfile,binsX,binsY,binsZ,valTypeX,valTypeY,valTypeZ,"",valTypeW); }
8d42b5b9	79
775970eb	80	// functions to add multidimensional stuff
	81	void UserSparse(const char* histClass,
	82	Int_t ndim, TObjArray limits, UInt_t vars, UInt_t valTypeW=AliDielectronHistos::kNoWeights);
	83
8d42b5b9	84
8d42b5b9	85	// functions to define the grid
5e2cf960	86	void AddCutVariable(AliDielectronVarManager::ValueTypes type, Int_t nbins,
	87	Double_t min, Double_t max, Bool_t log=kFALSE, Bool_t leg=kFALSE, EBinType btype=kStdBin);
	88	void AddCutVariable(AliDielectronVarManager::ValueTypes type,
	89	const char* binLimitStr, Bool_t leg=kFALSE, EBinType btype=kStdBin);
	90	void AddCutVariable(AliDielectronVarManager::ValueTypes type,
	91	TVectorD * binLimits, Bool_t leg=kFALSE, EBinType btype=kStdBin);
	92
443a091c	93	void Fill(Int_t pairIndex, const AliDielectronPair *particle);
	94	void Fill(Int_t label1, Int_t label2, Int_t nSignal);
	95	void Fill(Int_t Index, Double_t * const valuesPair, Double_t * const valuesLeg1, Double_t * const valuesLeg2);
	96
5e2cf960	97	Bool_t IsPairTypeSelected(Int_t itype);
443a091c	98
5e2cf960	99	Int_t GetNumberOfBins() const;
5e2cf960	100	const TObjArray * GetHistArray() const { return &fArrPairType; }
a9510cf1	101	Bool_t GetStepForMCGenerated() const { return fStepGenerated; }
	102	Bool_t IsEventArray() const { return fEventArray; }
	103
	104
443a091c	105
5e2cf960	106	private:
a94c2e7e	107	TBits *fUsedVars; // list of used variables
a94c2e7e	108
5e2cf960	109	TObjArray fArrPairType; //-> array of pair types
5e2cf960	110	EPairType fPairType; // which pair combinations to include
443a091c	111	TObjArray* fSignalsMC; //! array of MC signals to be studied
5e2cf960	112
5e2cf960	113	UShort_t fVarCuts[kMaxCuts]; // cut variables
775970eb	114	TBits *fVarCutType; // array to store leg booleans
775970eb	115	// Bool_t fVarCutType[kMaxCuts]; // array to store leg booleans
5e2cf960	116	TObjArray fAxes; // Axis descriptions of the cut binning
	117	UShort_t fBinType[kMaxCuts]; // binning type of the axes
	118
d4619886	119	Bool_t fHasMC; // is mc array
4d7704c5	120	Bool_t fStepGenerated; // switcher for generated particles
a9510cf1	121	Bool_t fEventArray; // switch OFF pair types and ON event array
8d42b5b9	122	TObjArray fRefObj; // reference object
35363fb9	123
5e2cf960	124	AliDielectronHF(const AliDielectronHF &c);
	125	AliDielectronHF &operator=(const AliDielectronHF &c);
	126
	127
a9510cf1	128	ClassDef(AliDielectronHF,6) // Dielectron HF
5e2cf960	129	};
	130
	131
	132
	133	#endif