[u/mrichter/AliRoot.git] / PWG3 / AliQuarkoniaEfficiency.cxx


// Root 
#include "TAxis.h"
#include "TFile.h"
#include "TH2.h"
#include "TString.h"
#include "TDirectory.h"

// AliRoot includes
#include "AliQuarkoniaEfficiency.h"
#include "AliLog.h"


AliQuarkoniaEfficiency::AliQuarkoniaEfficiency(Int_t quarkoniaResonance, Int_t decayChannel,
					       Int_t simParameterization):
  fEfficiencyFileName("QuarkoniaEfficiency.root"),
  fQuarkoniaResonance(quarkoniaResonance),     
  fDecayChannel(decayChannel),           
  fParameterization(simParameterization),  
  fTriggerType(kPairUnlikeApt),
  fTrigger(kFALSE),
  fEfficiency(0x0)
{

}

AliQuarkoniaEfficiency::~AliQuarkoniaEfficiency()
{
  delete fEfficiency;
}

void AliQuarkoniaEfficiency::Init()
{
  switch (fQuarkoniaResonance)  {  
  case kJpsi:
    SetTitle("Jpsi");
    break;
  case kPsiP:
    SetTitle("PsiP");
    break;
  case kUpsilon:
    SetTitle("Upsilon");
    break;
  case kUpsilonP:
    SetTitle("UpsilonP");
    break;
  case kUpsilonPP:
    SetTitle("UpsilonPP");
    break;
  case kOmega:
    SetTitle("Omega");
    break;
  case kPhi:
    SetTitle("Phi");
    break;
  }

  switch (fDecayChannel) {
  case kDimuon:
    SetName("Dimuon");
    break;
  case kDielectron:
    SetName("Dielectron");
    break;
  }

  char *param=0;
  switch (fParameterization){
  case kFlat:
    param = "Flat";
    break;
  case kCDFscaled:
    param = "CDFscaled";
    break;
  case kCDFscaledPP:
    param = "CDFscaledPP";
    break;
  }
  
  char *trig=0;
  switch (fTriggerType){
  case kSinglePlusLpt:
    trig = "SinglePlusLpt";
    break;
  case kSinglePlusHpt: 
    trig = "SinglePlusHpt";
    break;
  case kSinglePlusApt:
    trig = "SinglePlusApt";
    break;    
  case kSingleMinusLpt:
    trig = "SingleMinusLpt";
    break;    
  case kSingleMinusHpt:
    trig = "SingleMinusHpt";
    break;    
  case kSingleMinusApt:
    trig = "SingleMinusApt";
    break;    
  case kSingleUndefLpt:
    trig = "SingleUndefLpt";
    break;    
  case kSingleUndefHpt:
    trig = "SingleUndefHpt";
    break;    
  case kSingleUndefApt:
    trig = "SingleUndefApt";
    break;    
  case kPairUnlikeLpt:
    trig = "PairUnlikeLpt";
    break;    
  case kPairUnlikeHpt:
    trig = "PairUnlikeHpt";
    break;    
  case kPairUnlikeApt:
    trig = "PairUnlikeApt";
    break;    
  case kPairLikeLpt:
    trig = "PairLikeLpt";
    break;    
  case kPairLikeHpt:
    trig = "PairLikeHpt";
    break;    
  case kPairLikeApt:
    trig = "PairLikeApt";
    break;    
  }


  if(!fEfficiency) delete fEfficiency; 

  TFile efficiencyFile(fEfficiencyFileName);
  if ( efficiencyFile.IsOpen() ) {

    char quarkoniaDir[15];
    sprintf(quarkoniaDir,"%s",GetTitle());
    if (! efficiencyFile.cd(quarkoniaDir) ){
      AliError(Form("Directory %s not found in file %s \n Efficiency data for quarkonia %s and channel %s not found ",
		    quarkoniaDir,fEfficiencyFileName.Data(),GetTitle(),GetName() ));
      return;
    }
    
    char histosDir[30];
    sprintf(histosDir,"%s/%s_%s_%s",quarkoniaDir,GetTitle(),GetName(),param);
    if(! efficiencyFile.cd(histosDir) ){
      AliError(Form("Subdirectory %s/%s not found in file %s \n Efficiency data for quarkonia %s and channel %s not found ",
		    quarkoniaDir,histosDir,fEfficiencyFileName.Data(),GetTitle(),GetName() ));
      return;
    }

    char histoname[50];
    if(fTrigger) sprintf(histoname,"h%sEfficiencyPtRap_%s",GetTitle(),trig);
    else sprintf(histoname,"h%sEfficiencyPtRap",GetTitle());
    char histonameposition[99];
    sprintf(histonameposition,"%s/%s",histosDir,histoname);
    fEfficiency = (TH2F*)efficiencyFile.Get(histonameposition);

    if ( !fEfficiency ) {
      AliError(Form("Histo %s not found in file %s \n Efficiency data for quarkonia %s and channel %s not found",
		    histoname,GetTitle(), GetName() ));
    }
    else {
      fEfficiency->SetDirectory(0);
    }
    efficiencyFile.Close();

  }
  else {
    AliError(Form("File %s not found",fEfficiencyFileName.Data()));
  }

}

TH2F*  AliQuarkoniaEfficiency::GetEfficiencyHisto() const
{
  if (fEfficiency) return fEfficiency;
  else {
    AliError(Form("Efficiency data for quarkonia %s and channel %s not found",GetTitle(),GetName()));
    return 0x0;
  }
}

void  AliQuarkoniaEfficiency::GetEfficiency(Float_t rap, Float_t pT, Double_t &eff, Double_t &error)
{
  Int_t binx=0;
  Int_t biny=0;
  
  if (!fEfficiency) {
    AliError(Form("Efficiency data for quarkonia %s and channel %s not found",GetTitle(),GetName()));
  }
  else {
    if ( rap < (fEfficiency->GetXaxis())->GetXmin()  ||  
	 rap > (fEfficiency->GetXaxis())->GetXmax()  ||
	 pT  < (fEfficiency->GetYaxis())->GetXmin()  ||  
         pT  > (fEfficiency->GetYaxis())->GetXmax()   ) {
      AliInfo("Values out of range");
      eff   = 0.;
      error = 0.;
    }
    else  { 
      binx  = fEfficiency->GetXaxis()->FindBin(rap);  
      biny  = fEfficiency->GetYaxis()->FindBin(pT);
      eff   = fEfficiency->GetBinContent(binx,biny);
      error = fEfficiency->GetBinError(binx,biny);
    }
  } 
}
Commit	Line	Data
977e08ed	1
	2	// Root
	3	#include "TAxis.h"
	4	#include "TFile.h"
	5	#include "TH2.h"
	6	#include "TString.h"
	7	#include "TDirectory.h"
	8
	9	// AliRoot includes
	10	#include "AliQuarkoniaEfficiency.h"
	11	#include "AliLog.h"
	12
	13
	14	AliQuarkoniaEfficiency::AliQuarkoniaEfficiency(Int_t quarkoniaResonance, Int_t decayChannel,
	15	Int_t simParameterization):
	16	fEfficiencyFileName("QuarkoniaEfficiency.root"),
	17	fQuarkoniaResonance(quarkoniaResonance),
	18	fDecayChannel(decayChannel),
	19	fParameterization(simParameterization),
	20	fTriggerType(kPairUnlikeApt),
	21	fTrigger(kFALSE),
	22	fEfficiency(0x0)
	23	{
	24
	25	}
	26
	27	AliQuarkoniaEfficiency::~AliQuarkoniaEfficiency()
	28	{
	29	delete fEfficiency;
	30	}
	31
	32	void AliQuarkoniaEfficiency::Init()
	33	{
	34	switch (fQuarkoniaResonance) {
	35	case kJpsi:
	36	SetTitle("Jpsi");
	37	break;
	38	case kPsiP:
	39	SetTitle("PsiP");
	40	break;
	41	case kUpsilon:
	42	SetTitle("Upsilon");
	43	break;
	44	case kUpsilonP:
	45	SetTitle("UpsilonP");
	46	break;
	47	case kUpsilonPP:
	48	SetTitle("UpsilonPP");
	49	break;
	50	case kOmega:
	51	SetTitle("Omega");
	52	break;
	53	case kPhi:
	54	SetTitle("Phi");
	55	break;
	56	}
	57
	58	switch (fDecayChannel) {
	59	case kDimuon:
	60	SetName("Dimuon");
	61	break;
	62	case kDielectron:
	63	SetName("Dielectron");
	64	break;
65	}
66
67	char *param=0;
68	switch (fParameterization){
69	case kFlat:
70	param = "Flat";
71	break;
72	case kCDFscaled:
73	param = "CDFscaled";
74	break;
75	case kCDFscaledPP:
76	param = "CDFscaledPP";
77	break;
78	}
79
80	char *trig=0;
81	switch (fTriggerType){
82	case kSinglePlusLpt:
83	trig = "SinglePlusLpt";
84	break;
85	case kSinglePlusHpt:
86	trig = "SinglePlusHpt";
87	break;
88	case kSinglePlusApt:
89	trig = "SinglePlusApt";
90	break;
91	case kSingleMinusLpt:
92	trig = "SingleMinusLpt";
93	break;
94	case kSingleMinusHpt:
95	trig = "SingleMinusHpt";
96	break;
97	case kSingleMinusApt:
98	trig = "SingleMinusApt";
99	break;
100	case kSingleUndefLpt:
101	trig = "SingleUndefLpt";
102	break;
103	case kSingleUndefHpt:
104	trig = "SingleUndefHpt";
105	break;
106	case kSingleUndefApt:
107	trig = "SingleUndefApt";
108	break;
109	case kPairUnlikeLpt:
110	trig = "PairUnlikeLpt";
111	break;
112	case kPairUnlikeHpt:
113	trig = "PairUnlikeHpt";
114	break;
115	case kPairUnlikeApt:
116	trig = "PairUnlikeApt";
117	break;
118	case kPairLikeLpt:
119	trig = "PairLikeLpt";
120	break;
121	case kPairLikeHpt:
122	trig = "PairLikeHpt";
123	break;
124	case kPairLikeApt:
125	trig = "PairLikeApt";
126	break;
127	}
128
129
130	if(!fEfficiency) delete fEfficiency;
131
132	TFile efficiencyFile(fEfficiencyFileName);
133	if ( efficiencyFile.IsOpen() ) {
134
135	char quarkoniaDir[15];
136	sprintf(quarkoniaDir,"%s",GetTitle());
137	if (! efficiencyFile.cd(quarkoniaDir) ){
138	AliError(Form("Directory %s not found in file %s \n Efficiency data for quarkonia %s and channel %s not found ",
139	quarkoniaDir,fEfficiencyFileName.Data(),GetTitle(),GetName() ));
140	return;
141	}
142
143	char histosDir[30];
144	sprintf(histosDir,"%s/%s_%s_%s",quarkoniaDir,GetTitle(),GetName(),param);
145	if(! efficiencyFile.cd(histosDir) ){
146	AliError(Form("Subdirectory %s/%s not found in file %s \n Efficiency data for quarkonia %s and channel %s not found ",
147	quarkoniaDir,histosDir,fEfficiencyFileName.Data(),GetTitle(),GetName() ));
148	return;
149	}
150
151	char histoname[50];
152	if(fTrigger) sprintf(histoname,"h%sEfficiencyPtRap_%s",GetTitle(),trig);
153	else sprintf(histoname,"h%sEfficiencyPtRap",GetTitle());
154	char histonameposition[99];
155	sprintf(histonameposition,"%s/%s",histosDir,histoname);
156	fEfficiency = (TH2F*)efficiencyFile.Get(histonameposition);
157
158	if ( !fEfficiency ) {
159	AliError(Form("Histo %s not found in file %s \n Efficiency data for quarkonia %s and channel %s not found",
160	histoname,GetTitle(), GetName() ));
161	}
162	else {
163	fEfficiency->SetDirectory(0);
164	}
165	efficiencyFile.Close();
166
167	}
168	else {
169	AliError(Form("File %s not found",fEfficiencyFileName.Data()));
170	}
171
172	}
173
174	TH2F* AliQuarkoniaEfficiency::GetEfficiencyHisto() const
175	{
176	if (fEfficiency) return fEfficiency;
177	else {
178	AliError(Form("Efficiency data for quarkonia %s and channel %s not found",GetTitle(),GetName()));
179	return 0x0;
180	}
181	}
182
183	void AliQuarkoniaEfficiency::GetEfficiency(Float_t rap, Float_t pT, Double_t &eff, Double_t &error)
184	{
185	Int_t binx=0;
186	Int_t biny=0;
187
188	if (!fEfficiency) {
189	AliError(Form("Efficiency data for quarkonia %s and channel %s not found",GetTitle(),GetName()));
190	}
191	else {
192	if ( rap < (fEfficiency->GetXaxis())->GetXmin() \|\|
193	rap > (fEfficiency->GetXaxis())->GetXmax() \|\|
194	pT < (fEfficiency->GetYaxis())->GetXmin() \|\|
195	pT > (fEfficiency->GetYaxis())->GetXmax() ) {
196	AliInfo("Values out of range");
197	eff = 0.;
198	error = 0.;
199	}
200	else {
201	binx = fEfficiency->GetXaxis()->FindBin(rap);
202	biny = fEfficiency->GetYaxis()->FindBin(pT);
203	eff = fEfficiency->GetBinContent(binx,biny);
204	error = fEfficiency->GetBinError(binx,biny);
205	}
206	}
207	}