[u/mrichter/AliRoot.git] / PHOS / AliPHOSDA1.cxx

#include "AliPHOSDA1.h"
#include "TString.h"

ClassImp(AliPHOSDA1)

//----------------------------------------------------------------
AliPHOSDA1::AliPHOSDA1(int module) : TNamed(),
	   fHistoFile(0),fMod(module),fWriteToFile(kTRUE),fHistoArray()

{
  // Create DA1 ("Calibration DA") object.
  // module is the PHOS module number (0..4).
  // Checks existence of histograms which might have been left
  // from the previous runs to continue their filling.
  // Histogram names: module_iX_iZ_gain for TH2 and  module_iX_iZ for TH1.
  // Root file name: PHOS_ModuleX_Calib.root, where X - module number.
  
  char name[128];
  sprintf(name,"PHOS_Module%d_Calib",fMod);
  SetName(name);

  SetTitle("Calibration Detector Algorithm");

  char rootname[128];
  sprintf(rootname,"%s.root",GetName());

  fHistoFile =  new TFile(rootname,"update");
  fHistoArray.SetName("histo_container");

  char hname[128];
  TH1F* hist1=0;
  TH2F* hist2=0;

  for(Int_t iX=0; iX<64; iX++) {
    for(Int_t iZ=0; iZ<56; iZ++) {

      sprintf(hname,"%d_%d_%d",fMod,iX,iZ);
      hist1 = (TH1F*)fHistoFile->Get(hname);
      if(hist1) { 
	fHgLgRatio[iX][iZ] = hist1;
	fHistoArray.Add(hist1);
      }
      else
	fHgLgRatio[iX][iZ] = 0;

      for(Int_t iGain=0; iGain<2; iGain++) {
	sprintf(hname,"%d_%d_%d_%d",fMod,iX,iZ,iGain);
	hist2 = (TH2F*)fHistoFile->Get(hname);
	if(hist2) {
	  fTimeEnergy[iX][iZ][iGain] = hist2;
	  fHistoArray.Add(hist2);
	}
	else
	  fTimeEnergy[iX][iZ][iGain] = 0;
      }

    }
  }
 
}

//-------------------------------------------------------------------
AliPHOSDA1::AliPHOSDA1(Int_t module, TH2F oldTimeEnergy[64][56][2]) :
  TNamed(),fHistoFile(0),fMod(module),fWriteToFile(kFALSE),fHistoArray()
{
  // Constructor. 
  // oldTimeEnergy is an array of histograms kept from the previous run.
  // By default the final histograms will not be saved to the root file.

  char name[128];
  sprintf(name,"PHOS_Module%d_Calib",fMod);
  SetName(name);
  
  SetTitle("Calibration Detector Algorithm");
  fHistoArray.SetName("histo_container");

  if(oldTimeEnergy) {

    for(Int_t iX=0; iX<64; iX++) {
      for(Int_t iZ=0; iZ<56; iZ++) {

	fHgLgRatio[iX][iZ] = 0;
	for(Int_t iGain=0; iGain<2; iGain++) {

	  if((&(oldTimeEnergy[iX][iZ][iGain]))->GetEntries()) { 
	    fTimeEnergy[iX][iZ][iGain] = &(oldTimeEnergy[iX][iZ][iGain]);
	    fHistoArray.Add(&(oldTimeEnergy[iX][iZ][iGain]));
	  }
	  else
	    fTimeEnergy[iX][iZ][iGain] = 0;
	}
      }
    }
  
  }

  else {
    for(Int_t iX=0; iX<64; iX++) {
      for(Int_t iZ=0; iZ<56; iZ++) {
	fHgLgRatio[iX][iZ] = 0;
	for(Int_t iGain=0; iGain<2; iGain++) {
	  fTimeEnergy[iX][iZ][iGain] = 0;
	}
      }
    }
  }


}

//-------------------------------------------------------------------
AliPHOSDA1::AliPHOSDA1(const AliPHOSDA1& da) : TNamed(da),
 fHistoFile(0),fMod(da.fMod),fWriteToFile(da.fWriteToFile),fHistoArray(da.fHistoArray)
{
  // Copy constructor.

  fHistoFile = new TFile(da.GetName(),"update");

  char hname[128];
  TH1F* hist1=0;
  TH2F* hist2=0;

  for(Int_t iX=0; iX<64; iX++) {
    for(Int_t iZ=0; iZ<56; iZ++) {

      sprintf(hname,"%d_%d_%d",fMod,iX,iZ);
      hist1 = (TH1F*)da.fHistoFile->Get(hname);
      if(hist1) fHgLgRatio[iX][iZ] = hist1;
      else
	fHgLgRatio[iX][iZ] = 0;

      for(Int_t iGain=0; iGain<2; iGain++) {
	sprintf(hname,"%d_%d_%d_%d",fMod,iX,iZ,iGain);
	hist2 = (TH2F*)da.fHistoFile->Get(hname);
	if(hist2) fTimeEnergy[iX][iZ][iGain] = hist2;
	else
	  fTimeEnergy[iX][iZ][iGain] = 0;
      }

    }
  }
  
}

//-------------------------------------------------------------------
AliPHOSDA1& AliPHOSDA1::operator= (const AliPHOSDA1& da)
{
  //Assignment operator.

  if(this != &da) {

    TString oldname(fHistoFile->GetName());
    TString newname(da.fHistoFile->GetName());

    if(oldname != newname) {
      delete fHistoFile;
      fHistoFile = new TFile(da.fHistoFile->GetName(),"update");
    }

    fMod = da.fMod;

    SetName(da.GetName());
    SetTitle(da.GetTitle());

    for(Int_t iX=0; iX<64; iX++) {
      for(Int_t iZ=0; iZ<56; iZ++) {

	if(fHgLgRatio[iX][iZ]) delete fHgLgRatio[iX][iZ];
	fHgLgRatio[iX][iZ] = da.fHgLgRatio[iX][iZ];

	for(Int_t iGain=0; iGain<2; iGain++) {
	  if (fTimeEnergy[iX][iZ][iGain]) delete fTimeEnergy[iX][iZ][iGain];
	  fTimeEnergy[iX][iZ][iGain] = da.fTimeEnergy[iX][iZ][iGain];
	}

      }
    }

    fHistoArray = da.fHistoArray;
    
  }

  return *this;
}


//-------------------------------------------------------------------
AliPHOSDA1::~AliPHOSDA1()
{
  // Destructor
  
  UpdateHistoFile();
  if(fHistoFile) delete fHistoFile;
  
}

//-------------------------------------------------------------------
void AliPHOSDA1::FillHistograms(Float_t e[64][56][2], Float_t t[64][56][2]) 
{
  // Fill energy vs time-of-flight and HG/LG ratio histograms in one event.
  // Energy and TOF data are encoded as e[X][Z][gain] and t[X][Z][gain], 
  // where X(0..63) and Z(0..55) - crystal position in the module, 
  // gain=0 - low gain, gain=1 - high gain.
  // Energy in ADC counts, time in samples.
  // If no energy or time read for particular channel, 
  // the correspondent array entry should be filled by zero.
  // WARNING: this function should be called once per event!

  char hname[128];
  char htitl[128];

  for(Int_t iX=0; iX<64; iX++) {
    for (Int_t iZ=0; iZ<56; iZ++) {

      // HG/LG
      if(e[iX][iZ][0] && e[iX][iZ][1]) {
	if(fHgLgRatio[iX][iZ]) {
// 	  printf("iX=%d iZ=%d,e[iX][iZ][1]=%.3f,e[iX][iZ][0]=%.3f, t1=%.3f,t0=%.3f\n",
// 		 iX,iZ,e[iX][iZ][1],e[iX][iZ][0], t[iX][iZ][1],t[iX][iZ][0]);
	  fHgLgRatio[iX][iZ]->Fill(e[iX][iZ][1]/e[iX][iZ][0]); 
	}
	else
	  {
	    sprintf(hname,"%d_%d_%d",fMod,iX,iZ);
	    sprintf(htitl,"HG/LG ratio for crystal %d_%d_%d",fMod,iX,iZ);
	    fHgLgRatio[iX][iZ] = new TH1F(hname,htitl,400,14.,18.);
// 	    printf("iX=%d iZ=%d,e[iX][iZ][1]=%.3f,e[iX][iZ][0]=%.3f\n",
// 		   iX,iZ,e[iX][iZ][1],e[iX][iZ][0]);
	    fHgLgRatio[iX][iZ]->Fill(e[iX][iZ][1]/e[iX][iZ][0]);
	    fHistoArray.Add(fHgLgRatio[iX][iZ]);
	  }
      }
	  
      // Energy vs TOF
      for(Int_t iGain=0; iGain<2; iGain++) {
	if(e[iX][iZ][iGain]<10) continue;

	if(fTimeEnergy[iX][iZ][iGain]) 
	  fTimeEnergy[iX][iZ][iGain]->Fill(e[iX][iZ][iGain],t[iX][iZ][iGain]);
	else {
	  sprintf(hname,"%d_%d_%d_%d",fMod,iX,iZ,iGain);
	  sprintf(htitl,"Energy vs TOF for crystal %d_%d_%d and gain %d",fMod,iX,iZ,iGain);
	  fTimeEnergy[iX][iZ][iGain] = new TH2F(hname,htitl,1024,0.,1024.,100,0.,100);
	  fTimeEnergy[iX][iZ][iGain]->Fill(e[iX][iZ][iGain],t[iX][iZ][iGain]);
	  fHistoArray.Add(fTimeEnergy[iX][iZ][iGain]);
	}
      }

    }
  }

}

//-------------------------------------------------------------------
void AliPHOSDA1::UpdateHistoFile()
{
  // Write histograms to file

  if(!fWriteToFile) return;
  if(!fHistoFile) return;
  if(!fHistoFile->IsOpen()) return;

  TH1F* hist1=0;
  TH2F* hist2=0;

  for(Int_t iX=0; iX<64; iX++) {
    for(Int_t iZ=0; iZ<56; iZ++) {

      hist1 = fHgLgRatio[iX][iZ];
      if(hist1) hist1->Write(hist1->GetName(),TObject::kWriteDelete);

      for(Int_t iGain=0; iGain<2; iGain++) {
	hist2 = fTimeEnergy[iX][iZ][iGain];
	if(hist2) hist2->Write(hist2->GetName(),TObject::kWriteDelete);
      } 

    }
  }

}

//-----------------------------------------------------------------------
void AliPHOSDA1::SetWriteToFile(Bool_t write)
{
  if(!write) { 
    fWriteToFile = write;
    return;
  }

  if(!fHistoFile) {
    char rootname[128];
    sprintf(rootname,"%s.root",GetName());
    fHistoFile =  new TFile(rootname,"update");
  } 
  
  fWriteToFile = write;
}
Commit	Line	Data
c810e10b	1	#include "AliPHOSDA1.h"
	2	#include "TString.h"
	3
	4	ClassImp(AliPHOSDA1)
	5
	6	//----------------------------------------------------------------
	7	AliPHOSDA1::AliPHOSDA1(int module) : TNamed(),
ea542e87	8	fHistoFile(0),fMod(module),fWriteToFile(kTRUE),fHistoArray()
c810e10b	9
	10	{
	11	// Create DA1 ("Calibration DA") object.
	12	// module is the PHOS module number (0..4).
	13	// Checks existence of histograms which might have been left
	14	// from the previous runs to continue their filling.
	15	// Histogram names: module_iX_iZ_gain for TH2 and module_iX_iZ for TH1.
	16	// Root file name: PHOS_ModuleX_Calib.root, where X - module number.
	17
	18	char name[128];
	19	sprintf(name,"PHOS_Module%d_Calib",fMod);
	20	SetName(name);
	21
	22	SetTitle("Calibration Detector Algorithm");
	23
	24	char rootname[128];
	25	sprintf(rootname,"%s.root",GetName());
	26
	27	fHistoFile = new TFile(rootname,"update");
ea542e87	28	fHistoArray.SetName("histo_container");
c810e10b	29
	30	char hname[128];
	31	TH1F* hist1=0;
	32	TH2F* hist2=0;
	33
	34	for(Int_t iX=0; iX<64; iX++) {
	35	for(Int_t iZ=0; iZ<56; iZ++) {
	36
	37	sprintf(hname,"%d_%d_%d",fMod,iX,iZ);
	38	hist1 = (TH1F*)fHistoFile->Get(hname);
ea542e87	39	if(hist1) {
	40	fHgLgRatio[iX][iZ] = hist1;
	41	fHistoArray.Add(hist1);
	42	}
c810e10b	43	else
	44	fHgLgRatio[iX][iZ] = 0;
	45
	46	for(Int_t iGain=0; iGain<2; iGain++) {
	47	sprintf(hname,"%d_%d_%d_%d",fMod,iX,iZ,iGain);
	48	hist2 = (TH2F*)fHistoFile->Get(hname);
ea542e87	49	if(hist2) {
	50	fTimeEnergy[iX][iZ][iGain] = hist2;
	51	fHistoArray.Add(hist2);
	52	}
c810e10b	53	else
	54	fTimeEnergy[iX][iZ][iGain] = 0;
	55	}
	56
	57	}
	58	}
	59
	60	}
	61
b3efbd4c	62	//-------------------------------------------------------------------
b3efbd4c	63	AliPHOSDA1::AliPHOSDA1(Int_t module, TH2F oldTimeEnergy[64][56][2]) :
ea542e87	64	TNamed(),fHistoFile(0),fMod(module),fWriteToFile(kFALSE),fHistoArray()
b3efbd4c	65	{
	66	// Constructor.
	67	// oldTimeEnergy is an array of histograms kept from the previous run.
	68	// By default the final histograms will not be saved to the root file.
	69
	70	char name[128];
	71	sprintf(name,"PHOS_Module%d_Calib",fMod);
	72	SetName(name);
	73
	74	SetTitle("Calibration Detector Algorithm");
ea542e87	75	fHistoArray.SetName("histo_container");
ea542e87	76
b3efbd4c	77	if(oldTimeEnergy) {
	78
	79	for(Int_t iX=0; iX<64; iX++) {
	80	for(Int_t iZ=0; iZ<56; iZ++) {
ea542e87	81
ea542e87	82	fHgLgRatio[iX][iZ] = 0;
b3efbd4c	83	for(Int_t iGain=0; iGain<2; iGain++) {
b3efbd4c	84
ea542e87	85	if((&(oldTimeEnergy[iX][iZ][iGain]))->GetEntries()) {
b3efbd4c	86	fTimeEnergy[iX][iZ][iGain] = &(oldTimeEnergy[iX][iZ][iGain]);
ea542e87	87	fHistoArray.Add(&(oldTimeEnergy[iX][iZ][iGain]));
ea542e87	88	}
b3efbd4c	89	else
	90	fTimeEnergy[iX][iZ][iGain] = 0;
	91	}
	92	}
	93	}
	94
	95	}
	96
ea542e87	97	else {
	98	for(Int_t iX=0; iX<64; iX++) {
	99	for(Int_t iZ=0; iZ<56; iZ++) {
	100	fHgLgRatio[iX][iZ] = 0;
	101	for(Int_t iGain=0; iGain<2; iGain++) {
	102	fTimeEnergy[iX][iZ][iGain] = 0;
	103	}
	104	}
	105	}
	106	}
	107
	108
b3efbd4c	109	}
b3efbd4c	110
c810e10b	111	//-------------------------------------------------------------------
c810e10b	112	AliPHOSDA1::AliPHOSDA1(const AliPHOSDA1& da) : TNamed(da),
ea542e87	113	fHistoFile(0),fMod(da.fMod),fWriteToFile(da.fWriteToFile),fHistoArray(da.fHistoArray)
c810e10b	114	{
	115	// Copy constructor.
	116
	117	fHistoFile = new TFile(da.GetName(),"update");
	118
	119	char hname[128];
	120	TH1F* hist1=0;
	121	TH2F* hist2=0;
	122
	123	for(Int_t iX=0; iX<64; iX++) {
	124	for(Int_t iZ=0; iZ<56; iZ++) {
	125
	126	sprintf(hname,"%d_%d_%d",fMod,iX,iZ);
	127	hist1 = (TH1F*)da.fHistoFile->Get(hname);
	128	if(hist1) fHgLgRatio[iX][iZ] = hist1;
	129	else
	130	fHgLgRatio[iX][iZ] = 0;
	131
	132	for(Int_t iGain=0; iGain<2; iGain++) {
	133	sprintf(hname,"%d_%d_%d_%d",fMod,iX,iZ,iGain);
	134	hist2 = (TH2F*)da.fHistoFile->Get(hname);
	135	if(hist2) fTimeEnergy[iX][iZ][iGain] = hist2;
	136	else
	137	fTimeEnergy[iX][iZ][iGain] = 0;
	138	}
	139
	140	}
	141	}
	142
	143	}
	144
	145	//-------------------------------------------------------------------
	146	AliPHOSDA1& AliPHOSDA1::operator= (const AliPHOSDA1& da)
	147	{
	148	//Assignment operator.
	149
	150	if(this != &da) {
	151
	152	TString oldname(fHistoFile->GetName());
	153	TString newname(da.fHistoFile->GetName());
	154
	155	if(oldname != newname) {
	156	delete fHistoFile;
	157	fHistoFile = new TFile(da.fHistoFile->GetName(),"update");
	158	}
	159
	160	fMod = da.fMod;
	161
	162	SetName(da.GetName());
	163	SetTitle(da.GetTitle());
	164
	165	for(Int_t iX=0; iX<64; iX++) {
	166	for(Int_t iZ=0; iZ<56; iZ++) {
	167
	168	if(fHgLgRatio[iX][iZ]) delete fHgLgRatio[iX][iZ];
	169	fHgLgRatio[iX][iZ] = da.fHgLgRatio[iX][iZ];
	170
	171	for(Int_t iGain=0; iGain<2; iGain++) {
	172	if (fTimeEnergy[iX][iZ][iGain]) delete fTimeEnergy[iX][iZ][iGain];
	173	fTimeEnergy[iX][iZ][iGain] = da.fTimeEnergy[iX][iZ][iGain];
	174	}
	175
	176	}
	177	}
ea542e87	178
ea542e87	179	fHistoArray = da.fHistoArray;
c810e10b	180
	181	}
	182
	183	return *this;
	184	}
	185
	186
	187	//-------------------------------------------------------------------
	188	AliPHOSDA1::~AliPHOSDA1()
	189	{
	190	// Destructor
	191
	192	UpdateHistoFile();
	193	if(fHistoFile) delete fHistoFile;
	194
	195	}
	196
	197	//-------------------------------------------------------------------
	198	void AliPHOSDA1::FillHistograms(Float_t e[64][56][2], Float_t t[64][56][2])
	199	{
	200	// Fill energy vs time-of-flight and HG/LG ratio histograms in one event.
	201	// Energy and TOF data are encoded as e[X][Z][gain] and t[X][Z][gain],
	202	// where X(0..63) and Z(0..55) - crystal position in the module,
	203	// gain=0 - low gain, gain=1 - high gain.
	204	// Energy in ADC counts, time in samples.
	205	// If no energy or time read for particular channel,
	206	// the correspondent array entry should be filled by zero.
	207	// WARNING: this function should be called once per event!
	208
	209	char hname[128];
	210	char htitl[128];
	211
	212	for(Int_t iX=0; iX<64; iX++) {
	213	for (Int_t iZ=0; iZ<56; iZ++) {
	214
	215	// HG/LG
	216	if(e[iX][iZ][0] && e[iX][iZ][1]) {
	217	if(fHgLgRatio[iX][iZ]) {
	218	// printf("iX=%d iZ=%d,e[iX][iZ][1]=%.3f,e[iX][iZ][0]=%.3f, t1=%.3f,t0=%.3f\n",
	219	// iX,iZ,e[iX][iZ][1],e[iX][iZ][0], t[iX][iZ][1],t[iX][iZ][0]);
	220	fHgLgRatio[iX][iZ]->Fill(e[iX][iZ][1]/e[iX][iZ][0]);
	221	}
	222	else
	223	{
	224	sprintf(hname,"%d_%d_%d",fMod,iX,iZ);
	225	sprintf(htitl,"HG/LG ratio for crystal %d_%d_%d",fMod,iX,iZ);
	226	fHgLgRatio[iX][iZ] = new TH1F(hname,htitl,400,14.,18.);
	227	// printf("iX=%d iZ=%d,e[iX][iZ][1]=%.3f,e[iX][iZ][0]=%.3f\n",
	228	// iX,iZ,e[iX][iZ][1],e[iX][iZ][0]);
	229	fHgLgRatio[iX][iZ]->Fill(e[iX][iZ][1]/e[iX][iZ][0]);
ea542e87	230	fHistoArray.Add(fHgLgRatio[iX][iZ]);
c810e10b	231	}
	232	}
	233
	234	// Energy vs TOF
	235	for(Int_t iGain=0; iGain<2; iGain++) {
ea542e87	236	if(e[iX][iZ][iGain]<10) continue;
c810e10b	237
	238	if(fTimeEnergy[iX][iZ][iGain])
	239	fTimeEnergy[iX][iZ][iGain]->Fill(e[iX][iZ][iGain],t[iX][iZ][iGain]);
	240	else {
	241	sprintf(hname,"%d_%d_%d_%d",fMod,iX,iZ,iGain);
	242	sprintf(htitl,"Energy vs TOF for crystal %d_%d_%d and gain %d",fMod,iX,iZ,iGain);
	243	fTimeEnergy[iX][iZ][iGain] = new TH2F(hname,htitl,1024,0.,1024.,100,0.,100);
	244	fTimeEnergy[iX][iZ][iGain]->Fill(e[iX][iZ][iGain],t[iX][iZ][iGain]);
ea542e87	245	fHistoArray.Add(fTimeEnergy[iX][iZ][iGain]);
c810e10b	246	}
	247	}
	248
	249	}
	250	}
	251
	252	}
	253
	254	//-------------------------------------------------------------------
	255	void AliPHOSDA1::UpdateHistoFile()
	256	{
	257	// Write histograms to file
	258
b3efbd4c	259	if(!fWriteToFile) return;
c810e10b	260	if(!fHistoFile) return;
	261	if(!fHistoFile->IsOpen()) return;
	262
	263	TH1F* hist1=0;
	264	TH2F* hist2=0;
	265
	266	for(Int_t iX=0; iX<64; iX++) {
	267	for(Int_t iZ=0; iZ<56; iZ++) {
	268
	269	hist1 = fHgLgRatio[iX][iZ];
	270	if(hist1) hist1->Write(hist1->GetName(),TObject::kWriteDelete);
	271
	272	for(Int_t iGain=0; iGain<2; iGain++) {
	273	hist2 = fTimeEnergy[iX][iZ][iGain];
	274	if(hist2) hist2->Write(hist2->GetName(),TObject::kWriteDelete);
	275	}
	276
	277	}
	278	}
	279
	280	}
	281
b3efbd4c	282	//-----------------------------------------------------------------------
	283	void AliPHOSDA1::SetWriteToFile(Bool_t write)
	284	{
	285	if(!write) {
	286	fWriteToFile = write;
	287	return;
	288	}
	289
	290	if(!fHistoFile) {
	291	char rootname[128];
	292	sprintf(rootname,"%s.root",GetName());
	293	fHistoFile = new TFile(rootname,"update");
	294	}
	295
	296	fWriteToFile = write;
	297	}