[u/mrichter/AliRoot.git] / ZDC / AliZDCCalibTask.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

#include <Riostream.h> 
#include <TROOT.h>
#include <TSystem.h>
#include <TInterpreter.h>
#include <TTree.h>
#include <TCanvas.h>
#include <TList.h>
#include <TH1F.h>
#include <TFile.h>

#include "AliZDCCalibTask.h"
#include "AliESDEvent.h"
#include "AliLog.h"

ClassImp(AliZDCCalibTask)

//_______________________________________________________
AliZDCCalibTask::AliZDCCalibTask(const char *name):
  AliAnalysisTaskSE(name),
  fESD(0),	
  fListOfHistos(0x0),
  fZNCHisto(0x0),	
  fZPCHisto(0x0),
  fZNAHisto(0x0),
  fZPAHisto(0x0)	
{
  // Constructor
  DefineOutput(1, TList::Class());

}

//_______________________________________________________
AliZDCCalibTask::AliZDCCalibTask(const AliZDCCalibTask &calibTask):
  AliAnalysisTaskSE("AliZDCCalibTask"),
  fESD(calibTask.fESD),	
  fListOfHistos(calibTask.fListOfHistos),
  fZNCHisto(calibTask.fZNCHisto),	
  fZPCHisto(calibTask.fZPCHisto),
  fZNAHisto(calibTask.fZNAHisto),
  fZPAHisto(calibTask.fZPAHisto)	
{
  // Copy constructor
 
}

//______________________________________________________________________________
AliZDCCalibTask:: ~AliZDCCalibTask() 
{
  // destructor
  delete fESD;        
  delete fListOfHistos;
  delete fZNCHisto;   
  delete fZPCHisto;   
  delete fZNAHisto;   
  delete fZPAHisto;   
}

//______________________________________________________________________________
AliZDCCalibTask& AliZDCCalibTask::operator=(const AliZDCCalibTask &calibtask)  
{ 
  //assignment operator
  
  fESD = calibtask.fESD;	
  fListOfHistos = calibtask.fListOfHistos;
  fZNCHisto = calibtask.fZNCHisto;	
  fZPCHisto = calibtask.fZPCHisto;	
  fZNAHisto = calibtask.fZNAHisto;	
  fZPAHisto = calibtask.fZPAHisto; 
  
  return *this;  
}

//--------------------------------------------------------------------------
void AliZDCCalibTask::BookHistos(){

  //booking histos

  AliInfo(Form("***  Booking Histograms %s", GetName())) ; 
  
  fZNCHisto = new TH1F("fZNCHisto", "Energy in ZNC",100,0.,MAXENVALUE);
  fZNCHisto->SetXTitle("E (GeV)");
  fZPCHisto = new TH1F("fZPCHisto", "Energy in ZPC",100,0.,MAXENVALUE);
  fZPCHisto->SetXTitle("E (GeV)");
  fZNAHisto = new TH1F("fZNAHisto", "Energy in ZNA",100,0.,MAXENVALUE);
  fZNAHisto->SetXTitle("E (GeV)");
  fZPAHisto = new TH1F("fZPAHisto", "Energy in ZPA",100,0.,MAXENVALUE);
  fZPAHisto->SetXTitle("E (GeV)");
  
  if(!fListOfHistos) fListOfHistos = new TList();
  fListOfHistos->SetOwner();
  fListOfHistos->AddAt(fZNCHisto, 0);
  fListOfHistos->AddAt(fZPCHisto, 1);
  fListOfHistos->AddAt(fZNAHisto, 2);
  fListOfHistos->AddAt(fZPAHisto, 3);
  
}

//--------------------------------------------------------------------------
void AliZDCCalibTask::DrawHistos(){

  //drawing output histos
  
  AliInfo(Form("*** Drawing Histograms %s", GetName())) ; 

  if(!gROOT->IsBatch()){
    fListOfHistos = dynamic_cast<TList*>(GetOutputData(1));
    fZNCHisto = (TH1F*) fListOfHistos->At(0);
    fZPCHisto = (TH1F*) fListOfHistos->At(1);
    fZNAHisto = (TH1F*) fListOfHistos->At(2);
    fZPAHisto = (TH1F*) fListOfHistos->At(3);
    
    TCanvas * canvPlot = new TCanvas("canvPlot","ZDC energy",0,0,600,600);
    canvPlot->Divide(2,2);
    canvPlot->cd(1);
    fZNCHisto->SetLineColor(kTeal+1);
    fZNCHisto->DrawCopy("hist");
    canvPlot->cd(2);
    fZPCHisto->SetLineColor(kAzure+1);
    fZPCHisto->DrawCopy("hist");
    canvPlot->cd(3);
    fZNCHisto->SetLineColor(kGreen+1);
    fZNCHisto->DrawCopy("hist");
    canvPlot->cd(4);
    fZPCHisto->SetLineColor(kBlue+1);
    fZPCHisto->DrawCopy("hist");
    
  }
  
}

//_______________________________________________________
void AliZDCCalibTask::UserCreateOutputObjects()
{
  // Create histograms

  if(fDebug>1) printf("ZDCCalibTask::UserCreateOutputObjects() \n");

  //Open histograms
  OpenFile(1);
  BookHistos();
  
}

//_______________________________________________________
void AliZDCCalibTask::UserExec(Option_t */*option*/)
{
    // Execute analysis    
    AliVEvent *event = InputEvent();
    if (!event) {
      printf("ERROR: Could not retrieve event\n");
      return;
    }
    AliESDEvent* esd = dynamic_cast<AliESDEvent*>(event);
    AliESDZDC *esdZDC = esd->GetESDZDC();

    Float_t *znctow = (Float_t *) (esdZDC->GetZN1TowerEnergyLR());    
    Float_t *zpctow = (Float_t *) (esdZDC->GetZP1TowerEnergyLR());    
    Float_t *znatow = (Float_t *) (esdZDC->GetZN2TowerEnergyLR());    
    Float_t *zpatow = (Float_t *) (esdZDC->GetZP2TowerEnergyLR());    
    
    Float_t energyZNC=0., energyZPC=0.;
    Float_t energyZNA=0., energyZPA=0.;
    for(Int_t j=0; j<5; j++){
      energyZNC += znctow[j];
      energyZPC += zpctow[j];
      energyZNA += znatow[j];
      energyZPA += zpatow[j];
    }
    
    //printf(" Filling histos with values: %f  %f  %f  %f\n",
    //	energyZNC,energyZPC,energyZNA,energyZPA);
    //
    fZNCHisto->Fill(energyZNC);
    fZPCHisto->Fill(energyZPC);
    fZNAHisto->Fill(energyZNA);
    fZPAHisto->Fill(energyZPA);
    
    PostData(1, fListOfHistos);
}

//_______________________________________________________
void AliZDCCalibTask::Terminate(Option_t *)
{
  // Draw results
  TH1::AddDirectory(0);

  fListOfHistos = dynamic_cast<TList*>(GetOutputData(1));
  fZNCHisto = (TH1F*) fListOfHistos->At(0);
  fZPCHisto = (TH1F*) fListOfHistos->At(1);
  fZNAHisto = (TH1F*) fListOfHistos->At(2);
  fZPAHisto = (TH1F*) fListOfHistos->At(3);
  
  Int_t binMax[4], nBinsx[4];
  Float_t yMax[4], meanFitVal[4]={0.,0.,0.,0.};
  TF1 *fitfun[4];

  /*printf("\n # of entries in histos: %d  %d  %d  %d \n",
  	(Int_t) fZNCHisto->GetEntries(),(Int_t) fZPCHisto->GetEntries(),
	(Int_t) fZNAHisto->GetEntries(),(Int_t) fZPAHisto->GetEntries());*/

  if(fZNCHisto->GetEntries() != 0){ 
    binMax[0] = fZNCHisto->GetMaximumBin();
    yMax[0]   = (fZNCHisto->GetXaxis())->GetXmax();
    nBinsx[0] = (fZNCHisto->GetXaxis())->GetNbins();
    fZNCHisto->Fit("gaus","Q","",binMax[0]*yMax[0]/nBinsx[0]*0.7,binMax[0]*yMax[0]/nBinsx[0]*1.25);
    fitfun[0] = fZNCHisto->GetFunction("gaus");
    if(fitfun[0]) meanFitVal[0] = (Float_t) (fitfun[0]->GetParameter(1));
    else printf(" !!! No fit on fZNCHisto !!!\n");
  } 
  else AliWarning(" !!! ZNC empty histo !!!\n");

  if(fZPCHisto->GetEntries() != 0){ 
    binMax[1] = fZPCHisto->GetMaximumBin();
    yMax[1]   = (fZPCHisto->GetXaxis())->GetXmax();
    nBinsx[1] = (fZPCHisto->GetXaxis())->GetNbins();
    fZPCHisto->Fit("gaus","Q","",binMax[1]*yMax[1]/nBinsx[1]*0.7,binMax[1]*yMax[1]/nBinsx[1]*1.25);
    fitfun[1] = fZPCHisto->GetFunction("gaus");
    if(fitfun[1]) meanFitVal[1] = (Float_t) (fitfun[1]->GetParameter(1));
    else printf(" !!! No fit on fZPCHisto !!!\n");
  } 
  else AliWarning(" !!! ZPC empty histo !!!\n");

  if(fZNAHisto->GetEntries() != 0){ 
    binMax[2] = fZNAHisto->GetMaximumBin();
    yMax[2]   = (fZNAHisto->GetXaxis())->GetXmax();
    nBinsx[2] = (fZNAHisto->GetXaxis())->GetNbins();
    fZNAHisto->Fit("gaus","Q","",binMax[2]*yMax[2]/nBinsx[2]*0.7,binMax[2]*yMax[2]/nBinsx[2]*1.25);
    fitfun[2] = fZNAHisto->GetFunction("gaus");
    if(fitfun[2]) meanFitVal[2] = (Float_t) (fitfun[2]->GetParameter(1));
    else printf(" !!! No fit on fZNAHisto !!!\n");
  } 
  else AliWarning(" !!! ZNA empty histo !!!\n");

  if(fZPAHisto->GetEntries() != 0){ 
    binMax[3] = fZPAHisto->GetMaximumBin();
    yMax[3]   = (fZPAHisto->GetXaxis())->GetXmax();
    nBinsx[3] = (fZPAHisto->GetXaxis())->GetNbins();
    fZPAHisto->Fit("gaus","Q","",binMax[3]*yMax[3]/nBinsx[3]*0.7,binMax[3]*yMax[3]/nBinsx[3]*1.25);
    fitfun[3] = fZPAHisto->GetFunction("gaus");
    if(fitfun[3]) meanFitVal[3] = (Float_t) (fitfun[3]->GetParameter(1));
    else printf(" !!! No fit on fZPAHisto !!!\n");
  } 
  else AliWarning(" !!! ZPA empty histo !!!\n");
   
  FILE *fileOut = fopen("EMDCalibData.dat","w");
  for(Int_t j=0; j<6; j++){
     if(j<4) fprintf(fileOut,"\t%f\n",meanFitVal[j]);
     else    fprintf(fileOut,"\t1.0000\n");
  }
  fclose(fileOut);

  DrawHistos();
  
    

}
Commit	Line	Data
b81fcf27	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	#include <Riostream.h>
	17	#include <TROOT.h>
	18	#include <TSystem.h>
	19	#include <TInterpreter.h>
	20	#include <TTree.h>
	21	#include <TCanvas.h>
	22	#include <TList.h>
	23	#include <TH1F.h>
	24	#include <TFile.h>
	25
	26	#include "AliZDCCalibTask.h"
	27	#include "AliESDEvent.h"
	28	#include "AliLog.h"
	29
	30	ClassImp(AliZDCCalibTask)
	31
	32	//_______________________________________________________
	33	AliZDCCalibTask::AliZDCCalibTask(const char *name):
	34	AliAnalysisTaskSE(name),
	35	fESD(0),
	36	fListOfHistos(0x0),
	37	fZNCHisto(0x0),
	38	fZPCHisto(0x0),
	39	fZNAHisto(0x0),
	40	fZPAHisto(0x0)
	41	{
	42	// Constructor
	43	DefineOutput(1, TList::Class());
	44
	45	}
	46
	47	//_______________________________________________________
	48	AliZDCCalibTask::AliZDCCalibTask(const AliZDCCalibTask &calibTask):
	49	AliAnalysisTaskSE("AliZDCCalibTask"),
	50	fESD(calibTask.fESD),
	51	fListOfHistos(calibTask.fListOfHistos),
	52	fZNCHisto(calibTask.fZNCHisto),
	53	fZPCHisto(calibTask.fZPCHisto),
	54	fZNAHisto(calibTask.fZNAHisto),
	55	fZPAHisto(calibTask.fZPAHisto)
	56	{
	57	// Copy constructor
	58
	59	}
	60
	61	//______________________________________________________________________________
	62	AliZDCCalibTask:: ~AliZDCCalibTask()
	63	{
	64	// destructor
65	delete fESD;
66	delete fListOfHistos;
67	delete fZNCHisto;
68	delete fZPCHisto;
69	delete fZNAHisto;
70	delete fZPAHisto;
71	}
72
73	//______________________________________________________________________________
74	AliZDCCalibTask& AliZDCCalibTask::operator=(const AliZDCCalibTask &calibtask)
75	{
76	//assignment operator
77
78	fESD = calibtask.fESD;
79	fListOfHistos = calibtask.fListOfHistos;
80	fZNCHisto = calibtask.fZNCHisto;
81	fZPCHisto = calibtask.fZPCHisto;
82	fZNAHisto = calibtask.fZNAHisto;
83	fZPAHisto = calibtask.fZPAHisto;
84
85	return *this;
86	}
87
88	//--------------------------------------------------------------------------
89	void AliZDCCalibTask::BookHistos(){
90
91	//booking histos
92
93	AliInfo(Form("*** Booking Histograms %s", GetName())) ;
94
95	fZNCHisto = new TH1F("fZNCHisto", "Energy in ZNC",100,0.,MAXENVALUE);
96	fZNCHisto->SetXTitle("E (GeV)");
97	fZPCHisto = new TH1F("fZPCHisto", "Energy in ZPC",100,0.,MAXENVALUE);
98	fZPCHisto->SetXTitle("E (GeV)");
99	fZNAHisto = new TH1F("fZNAHisto", "Energy in ZNA",100,0.,MAXENVALUE);
100	fZNAHisto->SetXTitle("E (GeV)");
101	fZPAHisto = new TH1F("fZPAHisto", "Energy in ZPA",100,0.,MAXENVALUE);
102	fZPAHisto->SetXTitle("E (GeV)");
103
104	if(!fListOfHistos) fListOfHistos = new TList();
105	fListOfHistos->SetOwner();
106	fListOfHistos->AddAt(fZNCHisto, 0);
107	fListOfHistos->AddAt(fZPCHisto, 1);
108	fListOfHistos->AddAt(fZNAHisto, 2);
109	fListOfHistos->AddAt(fZPAHisto, 3);
110
111	}
112
113	//--------------------------------------------------------------------------
114	void AliZDCCalibTask::DrawHistos(){
115
116	//drawing output histos
117
118	AliInfo(Form("*** Drawing Histograms %s", GetName())) ;
119
120	if(!gROOT->IsBatch()){
121	fListOfHistos = dynamic_cast<TList*>(GetOutputData(1));
122	fZNCHisto = (TH1F*) fListOfHistos->At(0);
123	fZPCHisto = (TH1F*) fListOfHistos->At(1);
124	fZNAHisto = (TH1F*) fListOfHistos->At(2);
125	fZPAHisto = (TH1F*) fListOfHistos->At(3);
126
127	TCanvas * canvPlot = new TCanvas("canvPlot","ZDC energy",0,0,600,600);
128	canvPlot->Divide(2,2);
129	canvPlot->cd(1);
130	fZNCHisto->SetLineColor(kTeal+1);
131	fZNCHisto->DrawCopy("hist");
132	canvPlot->cd(2);
133	fZPCHisto->SetLineColor(kAzure+1);
134	fZPCHisto->DrawCopy("hist");
135	canvPlot->cd(3);
136	fZNCHisto->SetLineColor(kGreen+1);
137	fZNCHisto->DrawCopy("hist");
138	canvPlot->cd(4);
139	fZPCHisto->SetLineColor(kBlue+1);
140	fZPCHisto->DrawCopy("hist");
141
142	}
143
144	}
145
146	//_______________________________________________________
147	void AliZDCCalibTask::UserCreateOutputObjects()
148	{
149	// Create histograms
150
151	if(fDebug>1) printf("ZDCCalibTask::UserCreateOutputObjects() \n");
152
153	//Open histograms
154	OpenFile(1);
155	BookHistos();
156
157	}
158
159	//_______________________________________________________
160	void AliZDCCalibTask::UserExec(Option_t /option*/)
161	{
162	// Execute analysis
163	AliVEvent *event = InputEvent();
164	if (!event) {
165	printf("ERROR: Could not retrieve event\n");
166	return;
167	}
168	AliESDEvent* esd = dynamic_cast<AliESDEvent*>(event);
169	AliESDZDC *esdZDC = esd->GetESDZDC();
170
171	Float_t znctow = (Float_t ) (esdZDC->GetZN1TowerEnergyLR());
172	Float_t zpctow = (Float_t ) (esdZDC->GetZP1TowerEnergyLR());
173	Float_t znatow = (Float_t ) (esdZDC->GetZN2TowerEnergyLR());
174	Float_t zpatow = (Float_t ) (esdZDC->GetZP2TowerEnergyLR());
175
176	Float_t energyZNC=0., energyZPC=0.;
177	Float_t energyZNA=0., energyZPA=0.;
178	for(Int_t j=0; j<5; j++){
179	energyZNC += znctow[j];
180	energyZPC += zpctow[j];
181	energyZNA += znatow[j];
182	energyZPA += zpatow[j];
183	}
184
185	//printf(" Filling histos with values: %f %f %f %f\n",
186	// energyZNC,energyZPC,energyZNA,energyZPA);
187	//
188	fZNCHisto->Fill(energyZNC);
189	fZPCHisto->Fill(energyZPC);
190	fZNAHisto->Fill(energyZNA);
191	fZPAHisto->Fill(energyZPA);
192
193	PostData(1, fListOfHistos);
194	}
195
196	//_______________________________________________________
197	void AliZDCCalibTask::Terminate(Option_t *)
198	{
199	// Draw results
200	TH1::AddDirectory(0);
201
202	fListOfHistos = dynamic_cast<TList*>(GetOutputData(1));
203	fZNCHisto = (TH1F*) fListOfHistos->At(0);
204	fZPCHisto = (TH1F*) fListOfHistos->At(1);
205	fZNAHisto = (TH1F*) fListOfHistos->At(2);
206	fZPAHisto = (TH1F*) fListOfHistos->At(3);
207
208	Int_t binMax[4], nBinsx[4];
209	Float_t yMax[4], meanFitVal[4]={0.,0.,0.,0.};
210	TF1 *fitfun[4];
211
212	/*printf("\n # of entries in histos: %d %d %d %d \n",
213	(Int_t) fZNCHisto->GetEntries(),(Int_t) fZPCHisto->GetEntries(),
214	(Int_t) fZNAHisto->GetEntries(),(Int_t) fZPAHisto->GetEntries());*/
215
216	if(fZNCHisto->GetEntries() != 0){
217	binMax[0] = fZNCHisto->GetMaximumBin();
218	yMax[0] = (fZNCHisto->GetXaxis())->GetXmax();
219	nBinsx[0] = (fZNCHisto->GetXaxis())->GetNbins();
220	fZNCHisto->Fit("gaus","Q","",binMax[0]yMax[0]/nBinsx[0]0.7,binMax[0]yMax[0]/nBinsx[0]1.25);
221	fitfun[0] = fZNCHisto->GetFunction("gaus");
222	if(fitfun[0]) meanFitVal[0] = (Float_t) (fitfun[0]->GetParameter(1));
223	else printf(" !!! No fit on fZNCHisto !!!\n");
224	}
225	else AliWarning(" !!! ZNC empty histo !!!\n");
226
227	if(fZPCHisto->GetEntries() != 0){
228	binMax[1] = fZPCHisto->GetMaximumBin();
229	yMax[1] = (fZPCHisto->GetXaxis())->GetXmax();
230	nBinsx[1] = (fZPCHisto->GetXaxis())->GetNbins();
231	fZPCHisto->Fit("gaus","Q","",binMax[1]yMax[1]/nBinsx[1]0.7,binMax[1]yMax[1]/nBinsx[1]1.25);
232	fitfun[1] = fZPCHisto->GetFunction("gaus");
233	if(fitfun[1]) meanFitVal[1] = (Float_t) (fitfun[1]->GetParameter(1));
234	else printf(" !!! No fit on fZPCHisto !!!\n");
235	}
236	else AliWarning(" !!! ZPC empty histo !!!\n");
237
238	if(fZNAHisto->GetEntries() != 0){
239	binMax[2] = fZNAHisto->GetMaximumBin();
240	yMax[2] = (fZNAHisto->GetXaxis())->GetXmax();
241	nBinsx[2] = (fZNAHisto->GetXaxis())->GetNbins();
242	fZNAHisto->Fit("gaus","Q","",binMax[2]yMax[2]/nBinsx[2]0.7,binMax[2]yMax[2]/nBinsx[2]1.25);
243	fitfun[2] = fZNAHisto->GetFunction("gaus");
244	if(fitfun[2]) meanFitVal[2] = (Float_t) (fitfun[2]->GetParameter(1));
245	else printf(" !!! No fit on fZNAHisto !!!\n");
246	}
247	else AliWarning(" !!! ZNA empty histo !!!\n");
248
249	if(fZPAHisto->GetEntries() != 0){
250	binMax[3] = fZPAHisto->GetMaximumBin();
251	yMax[3] = (fZPAHisto->GetXaxis())->GetXmax();
252	nBinsx[3] = (fZPAHisto->GetXaxis())->GetNbins();
253	fZPAHisto->Fit("gaus","Q","",binMax[3]yMax[3]/nBinsx[3]0.7,binMax[3]yMax[3]/nBinsx[3]1.25);
254	fitfun[3] = fZPAHisto->GetFunction("gaus");
255	if(fitfun[3]) meanFitVal[3] = (Float_t) (fitfun[3]->GetParameter(1));
256	else printf(" !!! No fit on fZPAHisto !!!\n");
257	}
258	else AliWarning(" !!! ZPA empty histo !!!\n");
259
260	FILE *fileOut = fopen("EMDCalibData.dat","w");
261	for(Int_t j=0; j<6; j++){
262	if(j<4) fprintf(fileOut,"\t%f\n",meanFitVal[j]);
263	else fprintf(fileOut,"\t1.0000\n");
264	}
265	fclose(fileOut);
266
267	DrawHistos();
268
269
270
271	}