PHOS/AliPHOSDA1.cxx

   1 #include "AliPHOSDA1.h"
   2 #include "TString.h"
   3
   4 ClassImp(AliPHOSDA1)
   5
   6 //----------------------------------------------------------------
   7 AliPHOSDA1::AliPHOSDA1(int module) : TNamed(),
   8            fHistoFile(0),fMod(module),fWriteToFile(kTRUE),fHistoArray()
   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");
  28   fHistoArray.SetName("histo_container");
  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);
  39       if(hist1) {
  40         fHgLgRatio[iX][iZ] = hist1;
  41         fHistoArray.Add(hist1);
  42       }
  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);
  49         if(hist2) {
  50           fTimeEnergy[iX][iZ][iGain] = hist2;
  51           fHistoArray.Add(hist2);
  52         }
  53         else
  54           fTimeEnergy[iX][iZ][iGain] = 0;
  55       }
  56
  57     }
  58   }
  59
  60 }
  61
  62 //-------------------------------------------------------------------
  63 AliPHOSDA1::AliPHOSDA1(Int_t module, TH2F oldTimeEnergy[64][56][2]) :
  64   TNamed(),fHistoFile(0),fMod(module),fWriteToFile(kFALSE),fHistoArray()
  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");
  75   fHistoArray.SetName("histo_container");
  76
  77   if(oldTimeEnergy) {
  78
  79     for(Int_t iX=0; iX<64; iX++) {
  80       for(Int_t iZ=0; iZ<56; iZ++) {
  81
  82         fHgLgRatio[iX][iZ] = 0;
  83         for(Int_t iGain=0; iGain<2; iGain++) {
  84
  85           if((&(oldTimeEnergy[iX][iZ][iGain]))->GetEntries()) {
  86             fTimeEnergy[iX][iZ][iGain] = &(oldTimeEnergy[iX][iZ][iGain]);
  87             fHistoArray.Add(&(oldTimeEnergy[iX][iZ][iGain]));
  88           }
  89           else
  90             fTimeEnergy[iX][iZ][iGain] = 0;
  91         }
  92       }
  93     }
  94
  95   }
  96
  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
 109 }
 110
 111 //-------------------------------------------------------------------
 112 AliPHOSDA1::AliPHOSDA1(const AliPHOSDA1& da) : TNamed(da),
 113  fHistoFile(0),fMod(da.fMod),fWriteToFile(da.fWriteToFile),fHistoArray(da.fHistoArray)
 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     }
 178
 179     fHistoArray = da.fHistoArray;
 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]);
 230             fHistoArray.Add(fHgLgRatio[iX][iZ]);
 231           }
 232       }
 233
 234       // Energy vs TOF
 235       for(Int_t iGain=0; iGain<2; iGain++) {
 236         if(e[iX][iZ][iGain]<10) continue;
 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]);
 245           fHistoArray.Add(fTimeEnergy[iX][iZ][iGain]);
 246         }
 247       }
 248
 249     }
 250   }
 251
 252 }
 253
 254 //-------------------------------------------------------------------
 255 void AliPHOSDA1::UpdateHistoFile()
 256 {
 257   // Write histograms to file
 258
 259   if(!fWriteToFile) return;
 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
 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 }