[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis / AliFMDAnaCalibEventSelectionEfficiency.cxx


#include "AliFMDAnaCalibEventSelectionEfficiency.h"

#include <TH1F.h>
#include <TBrowser.h>
#include "AliLog.h"
#include "iostream"

ClassImp(AliFMDAnaCalibEventSelectionEfficiency)
#if 0
; // For Emacs
#endif 

//____________________________________________________________________
AliFMDAnaCalibEventSelectionEfficiency::AliFMDAnaCalibEventSelectionEfficiency() : TObject(),
										   fCorrection(),
										   fCorrectionList(),
										   fIsInit(kFALSE),
										   fVtxEff(1)
{
  
  
}


//____________________________________________________________________
AliFMDAnaCalibEventSelectionEfficiency::
AliFMDAnaCalibEventSelectionEfficiency(const AliFMDAnaCalibEventSelectionEfficiency& o) : TObject(o),			     
											  fCorrection(o.fCorrection),
											  fCorrectionList(),
											  fIsInit(o.fIsInit),
											  fVtxEff(o.fVtxEff)
{
  // Copy ctor 
}
//____________________________________________________________________
AliFMDAnaCalibEventSelectionEfficiency&
AliFMDAnaCalibEventSelectionEfficiency::operator=(const AliFMDAnaCalibEventSelectionEfficiency& /*o*/) 
{
  // Assignment operator 
  
  return (*this);
}
//____________________________________________________________________
void AliFMDAnaCalibEventSelectionEfficiency::Init() {

  fCorrection.SetName("EventSelectionEffCorrection");
  
  fIsInit = kTRUE;

}
//____________________________________________________________________
void AliFMDAnaCalibEventSelectionEfficiency::SetCorrection(TH1F* hCorrection) {
  
  fCorrection.SetBins(hCorrection->GetNbinsX(),
		      hCorrection->GetXaxis()->GetXmin(),
		      hCorrection->GetXaxis()->GetXmax());
  for(Int_t i=1; i<=hCorrection->GetNbinsX(); i++) {
    fCorrection.SetBinContent(i,hCorrection->GetBinContent(i));
    fCorrection.SetBinError(i,hCorrection->GetBinError(i));
  }
  

}
//____________________________________________________________________
void AliFMDAnaCalibEventSelectionEfficiency::SetCorrection(TString trig,
							   Int_t vtxbin, 
							   Char_t ring,
							   TH2F* hCorrection) {
  //TString test = trig;
  if(!trig.Contains("INEL") && !trig.Contains("NSD"))
    AliWarning("Please choose NSD or INEL!");
  
  if(trig.Contains("INEL"))
    hCorrection->SetName(Form("correction_%c_%d",ring,vtxbin));
  if(trig.Contains("NSD")) 
    hCorrection->SetName(Form("correction%s_%c_%d","NSD",ring,vtxbin));
  
  fCorrectionList.Add(hCorrection);
  
}
//____________________________________________________________________
TH2F* AliFMDAnaCalibEventSelectionEfficiency::GetCorrection(TString name,
							    Int_t vtxbin, 
							    Char_t ring) {
  
  //TString name = trig;
  
  if(name.Contains("INEL"))
    name.Form("correction_%c_%d",ring,vtxbin);
  if(name.Contains("NSD")) 
    name.Form("correction%s_%c_%d","NSD",ring,vtxbin);
  
  TH2F* hCorrection = (TH2F*)fCorrectionList.FindObject(name);
   
  return hCorrection;

}

//____________________________________________________________________
Float_t AliFMDAnaCalibEventSelectionEfficiency::GetCorrection(Int_t vtxbin) {

  if( (vtxbin-1) > fCorrection.GetNbinsX() || vtxbin < 0)
    return 0;
  
  Float_t correction = fCorrection.GetBinContent(vtxbin+1);
  
  return correction;

}

//____________________________________________________________________
void AliFMDAnaCalibEventSelectionEfficiency::Browse(TBrowser* /*b*/)
{
  
}

//____________________________________________________________________
//
// EOF
//
Commit	Line	Data
c3ac94c6	1
c3ac94c6	2	#include "AliFMDAnaCalibEventSelectionEfficiency.h"
ab3e0abc	3
c3ac94c6	4	#include <TH1F.h>
c3ac94c6	5	#include <TBrowser.h>
059c7c6b	6	#include "AliLog.h"
059c7c6b	7	#include "iostream"
c3ac94c6	8
	9	ClassImp(AliFMDAnaCalibEventSelectionEfficiency)
	10	#if 0
	11	; // For Emacs
	12	#endif
	13
	14	//____________________________________________________________________
	15	AliFMDAnaCalibEventSelectionEfficiency::AliFMDAnaCalibEventSelectionEfficiency() : TObject(),
daedf077	16	fCorrection(),
059c7c6b	17	fCorrectionList(),
541c19ed	18	fIsInit(kFALSE),
daedf077	19	fVtxEff(1)
c3ac94c6	20	{
	21
	22
	23
	24	}
	25
	26
	27	//____________________________________________________________________
059c7c6b	28	AliFMDAnaCalibEventSelectionEfficiency::
	29	AliFMDAnaCalibEventSelectionEfficiency(const AliFMDAnaCalibEventSelectionEfficiency& o) : TObject(o),
	30	fCorrection(o.fCorrection),
059c7c6b	31	fCorrectionList(),
541c19ed	32	fIsInit(o.fIsInit),
059c7c6b	33	fVtxEff(o.fVtxEff)
c3ac94c6	34	{
	35	// Copy ctor
	36	}
	37	//____________________________________________________________________
	38	AliFMDAnaCalibEventSelectionEfficiency&
ae26bdd7	39	AliFMDAnaCalibEventSelectionEfficiency::operator=(const AliFMDAnaCalibEventSelectionEfficiency& /o/)
c3ac94c6	40	{
	41	// Assignment operator
	42
	43	return (*this);
	44	}
	45	//____________________________________________________________________
	46	void AliFMDAnaCalibEventSelectionEfficiency::Init() {
	47
	48	fCorrection.SetName("EventSelectionEffCorrection");
	49
	50	fIsInit = kTRUE;
	51
	52	}
	53	//____________________________________________________________________
	54	void AliFMDAnaCalibEventSelectionEfficiency::SetCorrection(TH1F* hCorrection) {
	55
	56	fCorrection.SetBins(hCorrection->GetNbinsX(),
	57	hCorrection->GetXaxis()->GetXmin(),
	58	hCorrection->GetXaxis()->GetXmax());
	59	for(Int_t i=1; i<=hCorrection->GetNbinsX(); i++) {
	60	fCorrection.SetBinContent(i,hCorrection->GetBinContent(i));
	61	fCorrection.SetBinError(i,hCorrection->GetBinError(i));
	62	}
	63
	64
	65	}
ab3e0abc	66	//____________________________________________________________________
df2a9c32	67	void AliFMDAnaCalibEventSelectionEfficiency::SetCorrection(TString trig,
059c7c6b	68	Int_t vtxbin,
ab3e0abc	69	Char_t ring,
ab3e0abc	70	TH2F* hCorrection) {
df2a9c32	71	//TString test = trig;
df2a9c32	72	if(!trig.Contains("INEL") && !trig.Contains("NSD"))
059c7c6b	73	AliWarning("Please choose NSD or INEL!");
059c7c6b	74
df2a9c32	75	if(trig.Contains("INEL"))
059c7c6b	76	hCorrection->SetName(Form("correction_%c_%d",ring,vtxbin));
df2a9c32	77	if(trig.Contains("NSD"))
059c7c6b	78	hCorrection->SetName(Form("correction%s_%c_%d","NSD",ring,vtxbin));
059c7c6b	79
ab3e0abc	80	fCorrectionList.Add(hCorrection);
	81
	82	}
	83	//____________________________________________________________________
df2a9c32	84	TH2F* AliFMDAnaCalibEventSelectionEfficiency::GetCorrection(TString name,
059c7c6b	85	Int_t vtxbin,
ab3e0abc	86	Char_t ring) {
ab3e0abc	87
df2a9c32	88	//TString name = trig;
059c7c6b	89
df2a9c32	90	if(name.Contains("INEL"))
059c7c6b	91	name.Form("correction_%c_%d",ring,vtxbin);
df2a9c32	92	if(name.Contains("NSD"))
059c7c6b	93	name.Form("correction%s_%c_%d","NSD",ring,vtxbin);
	94
	95	TH2F* hCorrection = (TH2F*)fCorrectionList.FindObject(name);
	96
ab3e0abc	97	return hCorrection;
	98
	99	}
	100
c3ac94c6	101	//____________________________________________________________________
	102	Float_t AliFMDAnaCalibEventSelectionEfficiency::GetCorrection(Int_t vtxbin) {
	103
	104	if( (vtxbin-1) > fCorrection.GetNbinsX() \|\| vtxbin < 0)
	105	return 0;
	106
	107	Float_t correction = fCorrection.GetBinContent(vtxbin+1);
	108
	109	return correction;
	110
	111	}
	112
	113	//____________________________________________________________________
ae26bdd7	114	void AliFMDAnaCalibEventSelectionEfficiency::Browse(TBrowser* /b/)
c3ac94c6	115	{
	116
	117	}
	118
	119	//____________________________________________________________________
	120	//
	121	// EOF
	122	//