ITS/AliITSBeamTestDigitizer.cxx

   1 ////////////////////////////////////////////////////
   2 //  Class to manage the                           //
   3 //  ITS beam test conversion from rawdata         //
   4 //  to digits. It executes the digitization for   //
   5 //  SPD, SDD and SSD.                             //
   6 //                                                //
   7 //                                                //
   8 //  Origin:  E. Crescio crescio@to.infn.it        //
   9 //           J. Conrad  Jan.Conrad@cern.ch        //
  10 ////////////////////////////////////////////////////
  11 #include "AliHeader.h"
  12 #include "AliRun.h"
  13 #include "AliRunLoader.h"
  14 #include "AliITSEventHeader.h"
  15 #include "AliITSLoader.h"
  16 #include "AliITSBeamTestDigSDD.h"
  17 #include "AliITSBeamTestDigSPD.h"
  18 #include "AliITSBeamTestDigSSD.h"
  19 #include "AliITSBeamTestDigitizer.h"
  20 #include "AliRawReaderDate.h"
  21 #include "AliRawReaderRoot.h"
  22
  23 const TString AliITSBeamTestDigitizer::fgkDefaultDigitsFileName="ITS.Digits.root";
  24
  25 ClassImp(AliITSBeamTestDigitizer)
  26
  27
  28 //_____________________________________________________________
  29 AliITSBeamTestDigitizer::AliITSBeamTestDigitizer():TTask(),
  30 fEvIn(0),
  31 fEvFin(0),
  32 fRunNumber(-1),
  33 fDATEEvType(7),
  34 fFlagHeader(kTRUE),
  35 fFlagInit(kFALSE),
  36 fOptDate(kFALSE),
  37 fDigitsFileName(0),
  38 fRawdataFileName(0),
  39 fPeriod(kNov04),
  40 fRunLoader(0),
  41 fLoader(0),
  42 fHeader(0){
  43   //
  44   // Default constructor
  45   //
  46   SetFlagInit();
  47   SetOptDate();
  48 }
  49
  50 //_____________________________________________________________
  51 AliITSBeamTestDigitizer::AliITSBeamTestDigitizer(const Text_t* name, const Text_t* title, Char_t* opt,const char* filename):TTask(name,title),
  52 fEvIn(0),
  53 fEvFin(0),
  54 fRunNumber(-1),
  55 fDATEEvType(7),
  56 fFlagHeader(kTRUE),
  57 fFlagInit(kFALSE),
  58 fOptDate(kFALSE),
  59 fDigitsFileName(0),
  60 fRawdataFileName(0),
  61 fPeriod(kNov04),
  62 fRunLoader(0),
  63 fLoader(0),
  64 fHeader(0)
  65 {
  66   //
  67   // Standard constructor
  68   //
  69   SetOptDate();
  70
  71   TString choice(opt);
  72   Bool_t aug04 = choice.Contains("Aug04");
  73   Bool_t nov04 = choice.Contains("Nov04");
  74   if(aug04) fPeriod=kAug04;
  75   if(nov04) fPeriod=kNov04;
  76   Init(filename);
  77  }
  78
  79 //_____________________________________________________________
  80 AliITSBeamTestDigitizer::AliITSBeamTestDigitizer(const Text_t* name, const Text_t* title, Int_t run, Char_t* opt,const char* filename):TTask(name,title),
  81 fEvIn(0),
  82 fEvFin(0),
  83 fRunNumber(run),
  84 fDATEEvType(7),
  85 fFlagHeader(kTRUE),
  86 fFlagInit(kFALSE),
  87 fOptDate(kFALSE),
  88 fDigitsFileName(0),
  89 fRawdataFileName(0),
  90 fPeriod(kNov04),
  91 fRunLoader(0),
  92 fLoader(0),
  93 fHeader(0){
  94   //
  95   // Constructor
  96   //
  97   SetOptDate();
  98   TString choice(opt);
  99   Bool_t aug04 = choice.Contains("Aug04");
 100   Bool_t nov04 = choice.Contains("Nov04");
 101   if(aug04) fPeriod=kAug04;
 102   if(nov04) fPeriod=kNov04;
 103
 104   Init(filename);
 105  }
 106
 107 //___________________________________________________________
 108 void AliITSBeamTestDigitizer::Init(const char* filename){
 109
 110   //
 111   //Initialization of run loader and its loader
 112   //creation of galice.root
 113   //
 114
 115
 116   fRunLoader = AliRunLoader::Open(filename,AliConfig::GetDefaultEventFolderName(),"update");
 117   if (fRunLoader == 0x0)
 118     {
 119       Error("AliITSBeamTestDigitizer","Can not load the session",filename);
 120       return;
 121     }
 122   fRunLoader->LoadgAlice();
 123   gAlice = fRunLoader->GetAliRun();
 124
 125   if(!gAlice) {
 126     Error("AliITSBeamTestDigitizer","gAlice not found on file. Aborting.");
 127     return;
 128   }
 129   fRunLoader->MakeTree("E");
 130   fLoader = (AliITSLoader*)fRunLoader->GetLoader("ITSLoader");
 131
 132   fDigitsFileName=fgkDefaultDigitsFileName;
 133   this->Add(new AliITSBeamTestDigSPD("DigSPD","SPD Digitization"));
 134   this->Add(new AliITSBeamTestDigSDD("DigSDD","SDD Digitization"));
 135   this->Add(new AliITSBeamTestDigSSD("DigSSD","SSD Digitization"));
 136
 137   SetFlagInit(kTRUE);
 138 }
 139
 140
 141 //______________________________________________________________________
 142 AliITSBeamTestDigitizer::AliITSBeamTestDigitizer(const AliITSBeamTestDigitizer &bt):TTask(bt),
 143 fEvIn(bt.fEvIn),
 144 fEvFin(bt.fEvFin),
 145 fRunNumber(bt.fRunNumber),
 146 fDATEEvType(bt.fDATEEvType),
 147 fFlagHeader(bt.fFlagHeader),
 148 fFlagInit(bt.fFlagInit),
 149 fOptDate(bt.fOptDate),
 150 fDigitsFileName(bt.fDigitsFileName),
 151 fRawdataFileName(bt.fRawdataFileName),
 152 fPeriod(bt.fPeriod),
 153 fRunLoader(bt.fRunLoader),
 154 fLoader(bt.fLoader),
 155 fHeader(bt.fHeader){
 156   // Copy constructor.
 157 }
 158 //______________________________________________________________________
 159 AliITSBeamTestDigitizer& AliITSBeamTestDigitizer::operator=(const AliITSBeamTestDigitizer &source){
 160     // Assignment operator.
 161   this->~AliITSBeamTestDigitizer();
 162   new(this) AliITSBeamTestDigitizer(source);
 163   return *this;
 164 }
 165
 166
 167 //______________________________________________________________
 168 AliITSBeamTestDigitizer::~AliITSBeamTestDigitizer(){
 169
 170   //Destructor
 171   //  if(fBt) delete fBt;
 172   if(fLoader) delete fLoader;
 173   if(fHeader) delete fHeader;
 174 }
 175
 176
 177 //_____________________________________________________________
 178 void AliITSBeamTestDigitizer::SetNumberOfEventsPerFile(Int_t nev)
 179 {
 180   //Sets number of events per file
 181
 182   if(fRunLoader) fRunLoader->SetNumberOfEventsPerFile(nev);
 183   else Warning("SetNumberOfEventsPerFile","fRunLoader is 0");
 184 }
 185
 186
 187 //____________________________________________________
 188 void AliITSBeamTestDigitizer::ExecDigitization(){
 189
 190   // Execution of digitisation for SPD,SDD and SSD
 191
 192   if(!GetFlagInit()){
 193     Warning("ExecDigitization()","Run Init() please..");
 194     return;
 195   }
 196   fLoader->SetDigitsFileName(fDigitsFileName);
 197   fLoader->LoadDigits("recreate");
 198
 199   AliRawReader* rd;
 200
 201   if(GetOptDate()) rd = new AliRawReaderDate(fRawdataFileName,fEvIn);
 202   else rd = new AliRawReaderRoot(fRawdataFileName,fEvIn);
 203
 204   AliHeader* header = fRunLoader->GetHeader();
 205   Int_t iev=fEvIn-1;
 206
 207
 208   AliITSBeamTestDigSDD* digSDD = (AliITSBeamTestDigSDD*)fTasks->FindObject("DigSDD");
 209   AliITSBeamTestDigSPD* digSPD = (AliITSBeamTestDigSPD*)fTasks->FindObject("DigSPD");
 210   AliITSBeamTestDigSSD* digSSD = (AliITSBeamTestDigSSD*)fTasks->FindObject("DigSSD");
 211
 212
 213   do{
 214     iev++;
 215     if(fEvFin!=0){
 216       if(iev>fEvFin) break;
 217     }
 218     AliITSEventHeader* itsh = new AliITSEventHeader("ITSHeader");
 219     fRunLoader->SetEventNumber(iev);
 220
 221     rd->RequireHeader(fFlagHeader);
 222     rd->SelectEvents(fDATEEvType);
 223
 224     digSDD->SetRawReader(rd);
 225     digSPD->SetRawReader(rd);
 226     digSSD->SetRawReader(rd);
 227
 228     if(fLoader->TreeD() == 0x0) fLoader->MakeTree("D");
 229
 230     TTree* treeD = (TTree*)fLoader->TreeD();
 231
 232     // Make branches outside the dig-classes
 233
 234     TClonesArray* digitsSPD = new TClonesArray("AliITSdigitSPD",1000);
 235     treeD->Branch("ITSDigitsSPD",&digitsSPD);
 236
 237     TClonesArray* digitsSDD = new TClonesArray("AliITSdigitSDD",1000);
 238     treeD->Branch("ITSDigitsSDD",&digitsSDD);
 239
 240     TClonesArray* digitsSSD = new TClonesArray("AliITSdigitSSD",1000);
 241     treeD->Branch("ITSDigitsSSD",&digitsSSD);
 242
 243
 244     digSSD->SetTree(treeD);
 245     digSDD->SetTree(treeD);
 246     digSPD->SetTree(treeD);
 247
 248     AliITSgeom* geom = fLoader->GetITSgeom();
 249
 250     digSSD->SetITSgeom(geom);
 251     digSDD->SetITSgeom(geom);
 252     digSPD->SetITSgeom(geom);
 253
 254     digSSD->SetITSEventHeader(itsh);
 255     digSDD->SetITSEventHeader(itsh);
 256     digSPD->SetITSEventHeader(itsh);
 257
 258     digSDD->SetBtPeriod(GetBeamTestPeriod());
 259     if(GetBeamTestPeriod()==1)digSDD->SetThreshold(16);
 260     else digSDD->SetThreshold(0);
 261     ExecuteTask(0);
 262
 263     header->SetEventNrInRun(iev);
 264     header->SetEvent(iev);
 265     header->SetRun(fRunNumber);
 266     fRunLoader->GetHeader()->AddDetectorEventHeader(itsh);
 267     fRunLoader->TreeE()->Fill();
 268     header->Reset(fRunNumber,iev);
 269
 270     delete digitsSPD;
 271     delete digitsSDD;
 272     delete digitsSSD;
 273
 274    }while(rd->NextEvent());
 275
 276
 277   fRunLoader->WriteHeader("OVERWRITE");
 278   fRunLoader->WriteRunLoader("OVERWRITE");
 279
 280   delete rd;
 281   fLoader->UnloadDigits();
 282   fLoader->UnloadRawClusters();
 283   fRunLoader->UnloadHeader();
 284
 285
 286 }
 287
 288
 289
 290 //_______________________________________________
 291 void AliITSBeamTestDigitizer:: SetActive(const TString& subdet,Bool_t value){
 292
 293   //Sets active sub-tasks (detectors)
 294
 295   Bool_t sdd = subdet.Contains("SDD");
 296   Bool_t spd = subdet.Contains("SPD");
 297   Bool_t ssd = subdet.Contains("SSD");
 298
 299   if(sdd){
 300   AliITSBeamTestDigSDD* digSDD = (AliITSBeamTestDigSDD*)fTasks->FindObject("DigSDD");
 301   digSDD->SetActive(value);
 302   }
 303
 304  if(spd){
 305   AliITSBeamTestDigSPD* digSPD = (AliITSBeamTestDigSPD*)fTasks->FindObject("DigSPD");
 306   digSPD->SetActive(value);
 307
 308   }
 309
 310   if(ssd){
 311   AliITSBeamTestDigSSD* digSSD = (AliITSBeamTestDigSSD*)fTasks->FindObject("DigSSD");
 312   digSSD->SetActive(value);
 313
 314   }
 315
 316
 317
 318 }
 319