[u/mrichter/AliRoot.git] / FMD / AliFMDBackgroundCorrection.h

// -*- mode: C++ -*- 

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights
 * reserved. 
 *
 * See cxx source for full Copyright notice                               
 */
// Thil class computes background corrections for the FMD. The correction is computed 
// in eta,phi cells and the objects stored can be put into alien to use with analysis.
//
// Author: Hans Hjersing Dalsgaard, NBI, hans.dalsgaard@cern.ch
//
// 

#ifndef ALIFMDBACKGROUNDCORRECTION_H
#define ALIFMDBACKGROUNDCORRECTION_H

#include "AliFMDInput.h"
#include "TObjArray.h"


class AliFMDBackgroundCorrection : public TNamed {
  
public:
  
  AliFMDBackgroundCorrection() ;
  ~AliFMDBackgroundCorrection() {};
  void GenerateBackgroundCorrection(Int_t nvtxbins=10,
				    Float_t zvtxcut=10,
				    Int_t nBinsEta=100, 
				    Bool_t storeInAlien = kFALSE, 
				    Int_t runNo =0, 
				    Int_t endRunNo=999999999, 
				    const Char_t* filename="background.root", 
				    Bool_t simulate = kFALSE, 
				    Int_t nEvents=10);
  
  class AliFMDInputBG : public AliFMDInput {
    
  public :
    AliFMDInputBG() ; 
   
    Bool_t Init();
    
    Int_t GetNprim() {return fPrim;}
    Int_t GetNhits() {return fHits;}
    void  SetVtxCutZ(Double_t vtxCut) { fZvtxCut = vtxCut;}
    void  SetNvtxBins(Int_t nBins) { fNvtxBins = nBins;}
    void  SetNbinsEta(Int_t nBins) { fNbinsEta = nBins;}
    TObjArray*  GetHits() {return &fHitArray;}
    TObjArray*  GetPrimaries() {return &fPrimaryArray;}
  private:
    Bool_t ProcessTrack(Int_t i, TParticle* p, AliFMDHit* h );
    TObjArray fPrimaryArray;
    TObjArray fHitArray;
    Int_t fPrim;
    Int_t fHits;
    Double_t fZvtxCut;
    Int_t fNvtxBins;
    Int_t fPrevTrack;
    Int_t fPrevDetector;
    Char_t fPrevRing;
    Int_t fNbinsEta;
  };
  
private:
  
  void Simulate(Int_t);
  void ProcessPrimaries(AliRunLoader*);
  void ProcessHits();
  TObjArray fCorrectionArray;
   
  
  ClassDef(AliFMDBackgroundCorrection,0)
  
};
#endif
// EOF
Commit	Line	Data
df0d5480	1	// -- mode: C++ --
	2
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights
	4	* reserved.
	5	*
	6	* See cxx source for full Copyright notice
	7	*/
	8	// Thil class computes background corrections for the FMD. The correction is computed
	9	// in eta,phi cells and the objects stored can be put into alien to use with analysis.
	10	//
	11	// Author: Hans Hjersing Dalsgaard, NBI, hans.dalsgaard@cern.ch
	12	//
	13	//
	14
	15	#ifndef ALIFMDBACKGROUNDCORRECTION_H
	16	#define ALIFMDBACKGROUNDCORRECTION_H
	17
	18	#include "AliFMDInput.h"
	19	#include "TObjArray.h"
	20
	21
	22	class AliFMDBackgroundCorrection : public TNamed {
	23
	24	public:
	25
	26	AliFMDBackgroundCorrection() ;
	27	~AliFMDBackgroundCorrection() {};
	28	void GenerateBackgroundCorrection(Int_t nvtxbins=10,
	29	Float_t zvtxcut=10,
	30	Int_t nBinsEta=100,
	31	Bool_t storeInAlien = kFALSE,
	32	Int_t runNo =0,
	33	Int_t endRunNo=999999999,
	34	const Char_t* filename="background.root",
	35	Bool_t simulate = kFALSE,
	36	Int_t nEvents=10);
	37
	38	class AliFMDInputBG : public AliFMDInput {
	39
	40	public :
	41	AliFMDInputBG() ;
	42
	43	Bool_t Init();
	44
	45	Int_t GetNprim() {return fPrim;}
	46	Int_t GetNhits() {return fHits;}
	47	void SetVtxCutZ(Double_t vtxCut) { fZvtxCut = vtxCut;}
	48	void SetNvtxBins(Int_t nBins) { fNvtxBins = nBins;}
	49	void SetNbinsEta(Int_t nBins) { fNbinsEta = nBins;}
	50	TObjArray* GetHits() {return &fHitArray;}
	51	TObjArray* GetPrimaries() {return &fPrimaryArray;}
	52	private:
	53	Bool_t ProcessTrack(Int_t i, TParticle* p, AliFMDHit* h );
	54	TObjArray fPrimaryArray;
	55	TObjArray fHitArray;
	56	Int_t fPrim;
	57	Int_t fHits;
	58	Double_t fZvtxCut;
	59	Int_t fNvtxBins;
	60	Int_t fPrevTrack;
	61	Int_t fPrevDetector;
	62	Char_t fPrevRing;
	63	Int_t fNbinsEta;
	64	};
65
66	private:
67
68	void Simulate(Int_t);
69	void ProcessPrimaries(AliRunLoader*);
70	void ProcessHits();
71	TObjArray fCorrectionArray;
72
73
74	ClassDef(AliFMDBackgroundCorrection,0)
75
76	};
77	#endif
78	// EOF