1 /***************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
20 /////////////////////////////////////////////////////////////////////
21 // Base simulation functions for ITS //
24 /////////////////////////////////////////////////////////////////////
26 #include "TClonesArray.h"
27 #include "TObjArray.h"
31 #include "AliITSdigit.h"
32 #include "AliITSdigitSPD.h"
33 #include "AliITSdigitSDD.h"
34 #include "AliITSdigitSSD.h"
35 #include "AliITSDetTypeSim.h"
36 #include "AliITSgeom.h"
37 #include "AliITSpListItem.h"
38 #include "AliITSresponseSPD.h"
39 #include "AliITSresponseSDD.h"
40 #include "AliITSresponseSSD.h"
41 #include "AliITSsegmentationSPD.h"
42 #include "AliITSsegmentationSDD.h"
43 #include "AliITSsegmentationSSD.h"
44 #include "AliITSsimulation.h"
45 #include "AliITSsimulationSPD.h"
46 #include "AliITSsimulationSDD.h"
47 #include "AliITSsimulationSSD.h"
50 const Int_t AliITSDetTypeSim::fgkNdettypes = 3;
52 ClassImp(AliITSDetTypeSim)
54 //----------------------------------------------------------------------
55 AliITSDetTypeSim::AliITSDetTypeSim():
58 fSimulation(), // [NDet]
59 fSegmentation(), // [NDet]
60 fResponse(), // [NMod]
61 fPreProcess(), // [] e.g. Fill fHitModule with hits
62 fPostProcess(), // [] e.g. Wright Raw data
63 fNSDigits(0), //! number of SDigits
64 fSDigits(), //! [NMod][NSDigits]
65 fNDigits(0), //! number of Digits
66 fDigits(), //! [NMod][NDigits]
67 fHitClassName(), // String with Hit class name.
68 fSDigClassName(),// String with SDigit class name.
69 fDigClassName(){ // String with digit class name.
70 // Default Constructor
76 // A properly zero-ed AliITSDetTypeSim class.
78 fSimulation = new TObjArray(fgkNdettypes);
79 fSegmentation = new TObjArray(fgkNdettypes);
84 fSDigits = new TClonesArray("AliITSpListItem",1000);
85 fDigits = new TObjArray(fgkNdettypes);
86 fNDigits = new Int_t[fgkNdettypes];
89 //----------------------------------------------------------------------
90 AliITSDetTypeSim::~AliITSDetTypeSim(){
99 if(fGeom) delete fGeom;
101 fSimulation->Delete();
107 fSegmentation->Delete();
108 delete fSegmentation;
119 fPreProcess->Delete();
125 fPostProcess->Delete();
132 fLoader->GetModulesFolder()->Remove(this);
148 //----------------------------------------------------------------------
149 AliITSDetTypeSim::AliITSDetTypeSim(const AliITSDetTypeSim &source) : TObject(source){
150 // Copy Constructor for object AliITSDetTypeSim not allowed
151 if(this==&source) return;
152 Error("Copy constructor",
153 "You are not allowed to make a copy of the AliITSDetTypeSim");
158 //----------------------------------------------------------------------
159 AliITSDetTypeSim& AliITSDetTypeSim::operator=(const AliITSDetTypeSim &source){
160 // The = operator for object AliITSDetTypeSim
162 if(&source==this) return *this;
163 Error("operator=","You are not allowed to make a copy of the AliITSDetTypeSIm");
169 //______________________________________________________________________
170 void AliITSDetTypeSim::SetSimulationModel(Int_t dettype,AliITSsimulation *sim){
172 //Set simulation model for detector type
174 if(fSimulation==0) fSimulation = new TObjArray(fgkNdettypes);
175 fSimulation->AddAt(sim,dettype);
177 //______________________________________________________________________
178 AliITSsimulation* AliITSDetTypeSim::GetSimulationModel(Int_t dettype){
180 //Get simulation model for detector type
182 Warning("GetSimulationModel","fSimulation is 0!");
185 return (AliITSsimulation*)(fSimulation->At(dettype));
187 //______________________________________________________________________
188 AliITSsimulation* AliITSDetTypeSim::GetSimulationModelByModule(Int_t module){
190 //Get simulation model by module number
192 Warning("GetSimulationModelByModule","fGeom is 0!");
196 return GetSimulationModel(fGeom->GetModuleType(module));
198 //______________________________________________________________________
199 void AliITSDetTypeSim::SetSegmentationModel(Int_t dettype,AliITSsegmentation *seg){
201 //Set segmentation model for detector type
202 if(fSegmentation==0x0) fSegmentation = new TObjArray(fgkNdettypes);
203 fSegmentation->AddAt(seg,dettype);
206 //______________________________________________________________________
207 AliITSsegmentation* AliITSDetTypeSim::GetSegmentationModel(Int_t dettype){
209 //Get segmentation model for detector type
211 if(fSegmentation==0) {
212 Warning("GetSegmentationModel","fSegmentation is 0!");
215 return (AliITSsegmentation*)(fSegmentation->At(dettype));
218 //_______________________________________________________________________
219 AliITSsegmentation* AliITSDetTypeSim::GetSegmentationModelByModule(Int_t module){
221 //Get segmentation model by module number
223 Warning("GetSegmentationModelByModule","fGeom is 0!");
226 return GetSegmentationModel(fGeom->GetModuleType(module));
229 //_______________________________________________________________________
230 void AliITSDetTypeSim::SetResponseModel(Int_t dettype,AliITSresponse *resp){
233 //Set segmentation model for module number
234 if(fResponse==0) fResponse = new TObjArray(fgkNdettypes);
235 fResponse->AddAt(resp,dettype);
237 //______________________________________________________________________
238 void AliITSDetTypeSim::ResetResponse(){
240 //resets response array
242 for(Int_t i=0;i<fResponse->GetEntries();i++){
243 if(fResponse->At(i)) delete (AliITSresponse*)fResponse->At(i);
247 //______________________________________________________________________
248 void AliITSDetTypeSim::ResetSegmentation(){
250 //Resets segmentation array
252 for(Int_t i=0;i<fgkNdettypes;i++){
253 if(fSegmentation->At(i)) delete (AliITSsegmentation*)fSegmentation->At(i);
258 //_______________________________________________________________________
259 AliITSresponse* AliITSDetTypeSim::GetResponseModel(Int_t dettype){
261 //Get segmentation model for module number
264 Warning("GetResponseModel","fResponse is 0!");
267 return (AliITSresponse*)(fResponse->At(dettype));
269 //_______________________________________________________________________
270 void AliITSDetTypeSim::SetDefaults(){
272 //Set defaults for segmentation and response
276 Warning("SetDefaults","fGeom is 0!");
280 if(!fResponse) fResponse = new TObjArray(fgkNdettypes);
282 AliITSsegmentation* seg;
287 for(Int_t idet=0;idet<fgkNdettypes;idet++){
290 if(!GetSegmentationModel(idet)){
291 seg = new AliITSsegmentationSPD(fGeom);
292 SetSegmentationModel(idet,seg);
294 if(!GetResponseModel(idet)){
295 res = new AliITSresponseSPD();
296 SetResponseModel(idet,res);
298 const char *kData0=(GetResponseModel(idet))->DataType();
299 if (strstr(kData0,"real")) {
300 SetDigitClassName(idet,"AliITSdigit");
301 } else SetDigitClassName(idet,"AliITSdigitSPD");
305 if(!GetResponseModel(idet)){
306 SetResponseModel(idet,new AliITSresponseSDD("simulated"));
308 if(!GetSegmentationModel(idet)){
309 res = GetResponseModel(idet);
310 seg = new AliITSsegmentationSDD(fGeom,res);
311 SetSegmentationModel(idet,seg);
313 const char *kopt = GetResponseModel(idet)->ZeroSuppOption();
314 if((!strstr(kopt,"2D"))&&(!strstr(kopt,"1D"))) SetDigitClassName(idet,"AliITSdigit");
315 else SetDigitClassName(idet,"AliITSdigitSDD");
319 if(!GetSegmentationModel(idet)){
320 seg = new AliITSsegmentationSSD(fGeom);
321 SetSegmentationModel(idet,seg);
323 if(!GetResponseModel(idet)){
324 SetResponseModel(idet,new AliITSresponseSSD("simulated"));
326 const char *kData2 = (GetResponseModel(idet))->DataType();
327 if (strstr(kData2,"real")) {
328 SetDigitClassName(idet,"AliITSdigit");
329 } else SetDigitClassName(idet,"AliITSdigitSSD");
337 //_______________________________________________________________________
338 void AliITSDetTypeSim::SetDefaultSimulation(){
340 //Set default simulation for detector type
344 Warning("SetDefaults","fGeom is 0!");
348 if(!fResponse) fResponse = new TObjArray(fgkNdettypes);
350 AliITSsegmentation* seg;
352 AliITSsimulation* sim;
354 for(Int_t idet=0;idet<fgkNdettypes;idet++){
357 sim = GetSimulationModel(idet);
359 seg = (AliITSsegmentationSPD*)GetSegmentationModel(idet);
360 res = (AliITSresponseSPD*)GetResponseModel(idet);
361 sim = new AliITSsimulationSPD(seg,res);
362 SetSimulationModel(idet,sim);
364 sim->SetResponseModel(GetResponseModel(idet));
365 sim->SetSegmentationModel((AliITSsegmentationSPD*)GetSegmentationModel(idet));
371 sim = GetSimulationModel(idet);
373 seg = (AliITSsegmentationSDD*)GetSegmentationModel(idet);
374 res = (AliITSresponseSDD*)GetResponseModel(idet);
375 sim = new AliITSsimulationSDD(seg,res);
376 SetSimulationModel(idet,sim);
378 sim->SetResponseModel((AliITSresponseSDD*)GetResponseModel(idet));
379 sim->SetSegmentationModel((AliITSsegmentationSDD*)GetSegmentationModel(idet));
386 sim = GetSimulationModel(idet);
388 seg = (AliITSsegmentationSSD*)GetSegmentationModel(idet);
389 res = (AliITSresponseSSD*)GetResponseModel(idet);
390 sim = new AliITSsimulationSSD(seg,res);
391 SetSimulationModel(idet,sim);
393 sim->SetResponseModel((AliITSresponseSSD*)GetResponseModel(idet));
394 sim->SetSegmentationModel((AliITSsegmentationSSD*)GetSegmentationModel(idet));
403 //___________________________________________________________________
404 void AliITSDetTypeSim::SetTreeAddressS(TTree* treeS, Char_t* name){
405 // Set branch address for the ITS summable digits Trees.
412 if (fSDigits == 0x0){
413 fSDigits = new TClonesArray("AliITSpListItem",1000);
416 sprintf(branchname,"%s",name);
417 branch = treeS->GetBranch(branchname);
418 if (branch) branch->SetAddress(&fSDigits);
421 //___________________________________________________________________
422 void AliITSDetTypeSim::SetTreeAddressD(TTree* treeD, Char_t* name){
423 // Set branch address for the digit Trees.
425 const char *det[3] = {"SPD","SDD","SSD"};
434 fDigits = new TObjArray(fgkNdettypes);
436 for(Int_t i=0;i<fgkNdettypes;i++){
437 Char_t* digclass = GetDigitClassName(i);
439 if(i==0) SetDigitClassName(i,"AliITSdigitSPD");
440 if(i==1) SetDigitClassName(i,"AliITSdigitSDD");
441 if(i==2) SetDigitClassName(i,"AliITSdigitSSD");
442 digclass = GetDigitClassName(i);
444 TString classn = digclass;
445 if(!(fDigits->At(i))){
446 fDigits->AddAt(new TClonesArray(classn.Data(),1000),i);
451 if(fgkNdettypes==3) sprintf(branchname,"%sDigits%s",name,det[i]);
452 else sprintf(branchname,"%sDigits%d",name,i+1);
454 branch = treeD->GetBranch(branchname);
455 if(branch) branch->SetAddress(&((*fDigits)[i]));
460 //___________________________________________________________________
461 void AliITSDetTypeSim::ResetDigits(){
462 // Reset number of digits and the digits array for the ITS detector.
466 Error("ResetDigits","fDigits is null!");
469 for(Int_t i=0;i<fgkNdettypes;i++){
473 //___________________________________________________________________
474 void AliITSDetTypeSim::ResetDigits(Int_t branch){
475 // Reset number of digits and the digits array for this branch.
477 if(fDigits->At(branch)){
478 ((TClonesArray*)fDigits->At(branch))->Clear();
480 if(fNDigits) fNDigits[branch]=0;
486 //_______________________________________________________________________
487 void AliITSDetTypeSim::SDigitsToDigits(Option_t* opt, Char_t* name){
488 // Standard Summable digits to Digits function.
490 Warning("SDigitsToDigits","fGeom is null!!");
494 const char *all = strstr(opt,"All");
495 const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),
497 if( !det[0] && !det[1] && !det[2] ) all = "All";
499 static Bool_t setDef = kTRUE;
500 if(setDef) SetDefaultSimulation();
503 AliITSsimulation *sim =0;
504 TTree* trees = fLoader->TreeS();
505 if( !(trees && GetSDigits()) ){
506 Error("SDigits2Digits","Error: No trees or SDigits. Returning.");
509 sprintf(name,"%s",name);
510 TBranch* brchSDigits = trees->GetBranch(name);
513 for(Int_t module=0;module<fGeom->GetIndexMax();module++){
514 id = fGeom->GetModuleType(module);
515 if (!all && !det[id]) continue;
516 sim = (AliITSsimulation*)GetSimulationModel(id);
518 Error("SDigit2Digits","The simulation class was not "
519 "instanciated for module %d type %s!",module,
520 fGeom->GetModuleTypeName(module));
523 sim->InitSimulationModule(module,gAlice->GetEvNumber());
526 brchSDigits->GetEvent(module);
527 sim->AddSDigitsToModule(fSDigits,0);
528 sim->FinishSDigitiseModule();
529 fLoader->TreeD()->Fill();
532 fLoader->TreeD()->GetEntries();
533 fLoader->TreeD()->AutoSave();
534 fLoader->TreeD()->Reset();
539 //_________________________________________________________
540 void AliITSDetTypeSim::AddSumDigit(AliITSpListItem &sdig){
542 //Adds the module full of summable digits to the summable digits tree.
543 TClonesArray &lsdig = *fSDigits;
544 new(lsdig[fNSDigits++]) AliITSpListItem(sdig);
546 //__________________________________________________________
547 void AliITSDetTypeSim::AddRealDigit(Int_t branch, Int_t *digits){
548 // Add a real digit - as coming from data.
549 TClonesArray &ldigits = *((TClonesArray*)fDigits->At(branch));
550 new(ldigits[fNDigits[branch]++]) AliITSdigit(digits);
552 //__________________________________________________________
553 void AliITSDetTypeSim::AddSimDigit(Int_t branch, AliITSdigit* d){
555 // Add a simulated digit.
556 TClonesArray &ldigits = *((TClonesArray*)fDigits->At(branch));
559 new(ldigits[fNDigits[branch]++]) AliITSdigitSPD(*((AliITSdigitSPD*)d));
562 new(ldigits[fNDigits[branch]++]) AliITSdigitSDD(*((AliITSdigitSDD*)d));
565 new(ldigits[fNDigits[branch]++]) AliITSdigitSSD(*((AliITSdigitSSD*)d));
572 //______________________________________________________________________
573 void AliITSDetTypeSim::AddSimDigit(Int_t branch,Float_t phys,Int_t *digits,
574 Int_t *tracks,Int_t *hits,Float_t *charges){
575 // Add a simulated digit to the list.
577 TClonesArray &ldigits = *((TClonesArray*)fDigits->At(branch));
578 AliITSresponseSDD *resp = 0;
581 new(ldigits[fNDigits[branch]++]) AliITSdigitSPD(digits,tracks,hits);
584 resp = (AliITSresponseSDD*)GetResponseModel(branch);
585 new(ldigits[fNDigits[branch]++]) AliITSdigitSDD(phys,digits,tracks,
589 new(ldigits[fNDigits[branch]++]) AliITSdigitSSD(digits,tracks,hits);