X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITS.cxx;h=40be112668bb8dc9d8736acd2e5feb876e5f9784;hb=b68c91e660030717bb2f3969f64906a98515ebeb;hp=fcb1c2767248abbb52fc1c2185c4b78a936654f0;hpb=235841d0997f116b2c67db0545f83a1cf8f536fb;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITS.cxx b/ITS/AliITS.cxx index fcb1c276724..40be112668b 100644 --- a/ITS/AliITS.cxx +++ b/ITS/AliITS.cxx @@ -13,50 +13,15 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* -$Log$ -Revision 1.15 2000/06/28 14:41:12 fca -Corrections to the custom Streamer - -Revision 1.14 2000/06/15 09:27:52 barbera -Problems with the HP compiler fixed - -Revision 1.13 2000/06/13 15:32:44 nilsen -fix compilation error on HP and DEC unix. - -Revision 1.12 2000/06/12 23:43:16 nilsen -New ITS code replacing the old structure and simulations code. - -Revision 1.9.2.8 2000/06/12 18:05:59 barbera -fixed posible compilation errors on HP unix - -Revision 1.9.2.7 2000/06/11 20:20:18 barbera -New AliITS base clase for the new structure. - -Revision 1.9.2.3 2000/02/02 13:42:09 barbera -fixed AliITS.cxx for new AliRun structure. Added ITS hits list to list of hits which will have their track numbers updated - -Revision 1.9.2.2 2000/01/23 03:03:13 nilsen -//fixed FillModule. Removed fi(fabs(xl)
@@ -99,1129 +64,1207 @@ the AliITS class. // futher information. // /////////////////////////////////////////////////////////////////////////////// - -#include -#include -#include -#include -#include -#include -#include - - -#include "AliRun.h" +#include +#include +#include +#include +#include +#include +#include +#include "AliDetector.h" #include "AliITS.h" -#include "AliITSMap.h" -#include "AliITSClusterFinder.h" +#include "AliITSDetTypeSim.h" +#include "AliITSDDLRawData.h" +#include "AliITSLoader.h" +#include "AliITShit.h" +#include "AliITSmodule.h" +#include "AliITSpListItem.h" #include "AliITSsimulation.h" #include "AliITSsimulationFastPoints.h" +#include "AliMC.h" +#include "AliITSDigitizer.h" +#include "AliITSRecPoint.h" #include "AliITSsegmentationSPD.h" -#include "AliITSresponseSPD.h" #include "AliITSsegmentationSDD.h" -#include "AliITSresponseSDD.h" #include "AliITSsegmentationSSD.h" -#include "AliITSresponseSSD.h" -//#include "AliITStrack.h" - +#include "AliITSRawStreamSPD.h" +#include "AliITSRawStreamSSD.h" +#include "AliITSRawStreamSDD.h" +#include "AliRawReader.h" +#include "AliRun.h" +#include "AliLog.h" +#include "AliITSInitGeometry.h" ClassImp(AliITS) - -//_____________________________________________________________________________ -AliITS::AliITS() : AliDetector() { - // - // Default initialiser for ITS - // The default constructor of the AliITS class. In addition to + +//______________________________________________________________________ +AliITS::AliITS() : AliDetector(), +fDetTypeSim(0), +fEuclidOut(0), +fOpt("All"), +fIdN(0), +fIdSens(0), +fIdName(0), +fITSmodules(0), +fTiming(kFALSE) +{ + // Default initializer for ITS + // The default constructor of the AliITS class. In addition to // creating the AliITS class it zeros the variables fIshunt (a member // of AliDetector class), fEuclidOut, and fIdN, and zeros the pointers - // fITSpoints, fIdSens, and fIdName. The AliDetector default constructor + // fIdSens, and fIdName. The AliDetector default constructor // is also called. - // - - fIshunt = 0; - fEuclidOut = 0; - fIdN = 0; - fIdName = 0; - fIdSens = 0; - fITSmodules = 0; - // - fDetTypes = 0; - SetNDetTypes(); - // - fDtype = 0; - fNdtype = 0; - fCtype = 0; - fNctype = 0; - fRecPoints = 0; - fNRecPoints = 0; - // - fITSgeom=0; + +// SetDetectors(); // default to fOpt="All". This variable not written out. +//PH SetMarkerColor(kRed); } +//______________________________________________________________________ +AliITS::AliITS(const Char_t *title):AliDetector("ITS",title), +fDetTypeSim(0), +fEuclidOut(0), +fOpt("All"), +fIdN(0), +fIdSens(0), +fIdName(0), +fITSmodules(0), +fTiming(kFALSE) +{ + // The standard Constructor for the ITS class. + // It also zeros the variables + // fIshunt (a member of AliDetector class), fEuclidOut, and zeros + // the pointers fIdSens and fIdName. To help in displaying hits via the + // ROOT macro display.C AliITS also sets the marker color to red. The + // variables passes with this constructor, const char *name and *title, + // are used by the constructor of AliDetector class. See AliDetector + // class for a description of these parameters and its constructor + // functions. + // Inputs: + // Char_t *title Simulation title for the ITS + // Outputs: + // none. + // Return: + // none. + + fHits = new TClonesArray("AliITShit",1560); // from AliDetector + if(gAlice->GetMCApp()) gAlice->GetMCApp()->AddHitList(fHits); + //fNhits=0; //done in AliDetector(name,title) + SetDetectors(); // default to fOpt="All". This variable not written out. + fDetTypeSim = new AliITSDetTypeSim(); + //PH SetMarkerColor(kRed); + if(!fLoader) MakeLoader(AliConfig::GetDefaultEventFolderName()); + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); +} +//______________________________________________________________________ +AliITS::AliITS(const char *name, const char *title):AliDetector(name,title), +fDetTypeSim(0), +fEuclidOut(0), +fOpt("All"), +fIdN(0), +fIdSens(0), +fIdName(0), +fITSmodules(0), +fTiming(kFALSE) +{ + // The standard Constructor for the ITS class. + // It also zeros the variables + // fIshunt (a member of AliDetector class), fEuclidOut, and zeros + // the pointers fIdSens and fIdName. To help in displaying hits via the + // ROOT macro display.C AliITS also sets the marker color to red. The + // variables passes with this constructor, const char *name and *title, + // are used by the constructor of AliDetector class. See AliDetector + // class for a description of these parameters and its constructor + // functions. + + fHits = new TClonesArray("AliITShit",1560); + if(gAlice->GetMCApp()) gAlice->GetMCApp()->AddHitList(fHits); + //fNhits=0; //done in AliDetector(name,title) -//_____________________________________________________________________________ -AliITS::AliITS(const char *name, const char *title):AliDetector(name,title){ - // - // Default initialiser for ITS - // The constructor of the AliITS class. In addition to creating the - // AliITS class, it allocates memory for the TClonesArrays fHits and - // fDigits, and for the TObjArray fITSpoints. It also zeros the variables - // fIshunt (a member of AliDetector class), fEuclidOut, and fIdN, and zeros - // the pointers fIdSens and fIdName. To help in displaying hits via the ROOT - // macro display.C AliITS also sets the marker color to red. The variables - // passes with this constructor, const char *name and *title, are used by - // the constructor of AliDetector class. See AliDetector class for a - // description of these parameters and its constructor functions. - // - - fHits = new TClonesArray("AliITShit", 1560); - gAlice->AddHitList(fHits); - - SetNDetTypes(); - - fNdtype = 0; - fDtype = 0; - fCtype = 0; - fNctype = 0; - - fRecPoints = 0; - fNRecPoints = 0; - - - fITSmodules = 0; - - fIshunt = 0; - fEuclidOut = 0; - fIdN = 0; - fIdName = 0; - fIdSens = 0; - - fDetTypes = new TObjArray(fNDetTypes); - - Int_t i; - for(i=0;iSetLoader((AliITSLoader*)fLoader); - fITSgeom=0; } -//___________________________________________________________________________ -AliITS::AliITS(AliITS &source){ - if(this==&source) return; - printf("Error: You are not allowed to make a copy of the AliITS\n"); - exit(1); +//______________________________________________________________________ +AliITS::~AliITS(){ + // Default destructor for ITS. + // The default destructor of the AliITS class. In addition to deleting + // the AliITS class it deletes the memory pointed to by + // fIdSens, fIdName, fDetTypeSim and it's contents. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + + if (fHits) { + fHits->Delete(); + delete fHits; + fHits=0; + } + if(fITSmodules) { + this->ClearModules(); + delete fITSmodules; + fITSmodules = 0; + }// end if fITSmodules!=0 + + delete[] fIdName; // Array of TStrings + delete[] fIdSens; + + if (fDetTypeSim){ + delete fDetTypeSim; + fDetTypeSim = 0; + } } -//____________________________________________________________________________ -AliITS& AliITS::operator=(AliITS &source){ - if(this==&source) return *this; - printf("Error: You are not allowed to make a copy of the AliITS\n"); - exit(1); +//______________________________________________________________________ +AliDigitizer* AliITS::CreateDigitizer(AliRunDigitizer* manager)const{ + // Creates the AliITSDigitizer in a standard way for use via AliModule. + // This function can not be included in the .h file because of problems + // with the order of inclusion (recursive). + // Inputs: + // AliRunDigitizer *manager The Manger class for Digitization + // Output: + // none. + // Return: + // A new AliITSRunDigitizer (cast as a AliDigitizer). + + return new AliITSDigitizer(manager); } -//____________________________________________________________________________ -void AliITS::ClearModules(){ - //clear the modules TObjArray - Int_t i; - - if(fITSmodules!=0) { - Int_t indSPD = fITSgeom->GetModuleIndex(2,fITSgeom->GetNladders(2), - fITSgeom->GetNdetectors(2)); - Int_t indSDD = fITSgeom->GetModuleIndex(4,fITSgeom->GetNladders(4), - fITSgeom->GetNdetectors(4)); - for(i=0;iGetEntriesFast();i++){ - if(iAt(i); - else if(iAt(i); - else - delete (AliITSmodule *) fITSmodules->At(i); - } // end for i - }// end if fITSmodules!=0 - +//______________________________________________________________________ +void AliITS::Init(){ + // Initializer ITS after it has been built + // This routine initializes the AliITS class. It is intended to be + // called from the Init function in AliITSv?. Besides displaying a banner + // indicating that it has been called it initializes the array fIdSens + // and sets the default segmentation, response, digit and raw cluster + // classes therefore it should be called after a call to CreateGeometry. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + Int_t i; + // Array of TStrings + if(gMC) for(i=0;iVolId(fIdName[i]); + } -//_____________________________________________________________________________ -AliITS::~AliITS(){ - // - // Default distructor for ITS - // The default destructor of the AliITS class. In addition to deleting - // the AliITS class it deletes the memory pointed to by the fHits, fDigits, - // fIdSens, fIdName, and fITSpoints. - // - - delete fHits; - delete fDigits; - delete fRecPoints; - if(fIdName!=0) delete[] fIdName; - if(fIdSens!=0) delete[] fIdSens; - if(fITSmodules!=0) { - this->ClearModules(); - delete fITSmodules; - }// end if fITSmodules!=0 - // - Int_t i; - if(fDtype) { - for(i=0;iSetDefaults(); - // - if (fDetTypes) { - fDetTypes->Delete(); - delete fDetTypes; - } - - if (fTreeC) delete fTreeC; - } +//______________________________________________________________________ +void AliITS::SetDefaultSimulation(){ + // sets the default simulation. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + if(!fDetTypeSim) { + Error("SetDefaultSimulation()","fDetTypeSim is 0!"); + return; + } -//___________________________________________ -AliITSDetType* AliITS::DetType(Int_t id) -{ - //return pointer to id detector type - return ((AliITSDetType*) (*fDetTypes)[id]); - -} -//___________________________________________ -void AliITS::SetClasses(Int_t id, char* digit, char* cluster) -{ - //set the digit and cluster classes to be used for the id detector type - ((AliITSDetType*) (*fDetTypes)[id])->ClassNames(digit,cluster); + fDetTypeSim->SetDefaultSimulation(); } -//___________________________________________ -void AliITS::SetResponseModel(Int_t id, AliITSresponse *response) -{ - //set the response model for the id detector type - ((AliITSDetType*) (*fDetTypes)[id])->ResponseModel(response); -} +//______________________________________________________________________ +void AliITS::MakeBranch(Option_t* option){ + // Creates Tree branches for the ITS. + // Inputs: + // Option_t *option String of Tree types S,D, and/or R. + // const char *file String of the file name where these branches + // are to be stored. If blank then these branches + // are written to the same tree as the Hits were + // read from. + // Outputs: + // none. + // Return: + // none. + if(!fDetTypeSim) { + Error("MakeBranch","fDetTypeSim is 0!"); + return; + } -//___________________________________________ -void AliITS::SetSegmentationModel(Int_t id, AliITSsegmentation *seg) -{ - //set the segmentation model for the id detector type + Bool_t cH = (strstr(option,"H")!=0); + Bool_t cS = (strstr(option,"S")!=0); + Bool_t cD = (strstr(option,"D")!=0); + + if(cH && (fHits == 0x0)) fHits = new TClonesArray("AliITShit", 1560); + AliDetector::MakeBranch(option); + + if(cS) MakeBranchS(0); + if(cD) MakeBranchD(0); - ((AliITSDetType*) (*fDetTypes)[id])->SegmentationModel(seg); } +//___________________________________________________________________ +void AliITS::MakeBranchS(const char* fl){ -//___________________________________________ -void AliITS::SetSimulationModel(Int_t id, AliITSsimulation *sim) -{ - //set the simulation model for the id detector type - - ((AliITSDetType*) (*fDetTypes)[id])->SimulationModel(sim); - -} -//___________________________________________ -void AliITS::SetReconstructionModel(Int_t id, AliITSClusterFinder *reconst) -{ - //set the cluster finder model for the id detector type + // Creates Tree Branch for the ITS summable digits. + // Inputs: + // cont char *fl File name where SDigits branch is to be written + // to. If blank it write the SDigits to the same + // file in which the Hits were found. - ((AliITSDetType*) (*fDetTypes)[id])->ReconstructionModel(reconst); + + if(!fDetTypeSim){ + Error("MakeBranchS","fDetTypeSim is 0!"); + } + Int_t buffersize = 4000; + char branchname[30]; -} + // only one branch for SDigits. + sprintf(branchname,"%s",GetName()); -//_____________________________________________________________________________ -void AliITS::AddHit(Int_t track, Int_t *vol, Float_t *hits){ - // - // Add an ITS hit - // The function to add information to the AliITShit class. See the - // AliITShit class for a full description. This function allocates the - // necessary new space for the hit information and passes the variable - // track, and the pointers *vol and *hits to the AliITShit constructor - // function. - // - TClonesArray &lhits = *fHits; - new(lhits[fNhits++]) AliITShit(fIshunt,track,vol,hits); + if(fLoader->TreeS()){ + TClonesArray* sdig = (TClonesArray*)fDetTypeSim->GetSDigits(); + MakeBranchInTree(fLoader->TreeS(),branchname,&sdig,buffersize,fl); + } } -//_____________________________________________________________________________ -void AliITS::AddRealDigit(Int_t id, Int_t *digits) -{ - // add a real digit - as coming from data - - TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]); - new(ldigits[fNdtype[id]++]) AliITSdigit(digits); +//______________________________________________________________________ +void AliITS::MakeBranchD(const char* file){ + //Make branch for digits + if(!fDetTypeSim) { + Warning("MakeBranchD","fDetTypeSim is 0!"); + return; + } + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); + MakeBranchInTreeD(fLoader->TreeD(),file); } -//_____________________________________________________________________________ -void AliITS::AddDigit(Int_t id, AliITSdigit *d) -{ - - // add a simulated digit - // should have ctors of type AliITSdigitSDD(const AliITSdigitSDD &) +//___________________________________________________________________ +void AliITS:: MakeBranchInTreeD(TTree* treeD, const char* file){ + // Creates Tree branches for the ITS. - TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]); + if(!fDetTypeSim){ + Error("MakeBranchS","fDetTypeSim is 0!"); + } + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); - switch(id) - { - case 0: - new(ldigits[fNdtype[id]++]) AliITSdigitSPD(*((AliITSdigitSPD*)d)); - break; - case 1: - new(ldigits[fNdtype[id]++]) AliITSdigitSDD(*((AliITSdigitSDD*)d)); - break; - case 2: - new(ldigits[fNdtype[id]++]) AliITSdigitSSD(*((AliITSdigitSSD*)d)); - break; + const Char_t *det[3] = {"SPD","SDD","SSD"}; + Char_t* digclass; + Int_t buffersize = 4000; + Char_t branchname[30]; + + if(!fDetTypeSim->GetDigits()){ + fDetTypeSim->SetDigits(new TObjArray(fgkNTYPES)); + } + for(Int_t i=0;iGetDigitClassName(i); + TString classn = digclass; + if(!((fDetTypeSim->GetDigits())->At(i))){ + (fDetTypeSim->GetDigits())->AddAt(new TClonesArray(classn.Data(),1000),i); + } + else ResetDigits(i); + if(fgkNTYPES==3) sprintf(branchname,"%sDigits%s",GetName(),det[i]); + else sprintf(branchname,"%sDigits%d",GetName(),i+1); + TObjArray* dig = DigitsAddress(i); + if(GetDigits() && treeD) AliDetector::MakeBranchInTree(treeD,branchname, &dig,buffersize,file); } } - -//_____________________________________________________________________________ -void AliITS::AddDigit(Int_t id,Float_t phys,Int_t *digits,Int_t *tracks,Float_t *charges){ - - // add a simulated digit to the list - - TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]); - switch(id) - { - case 0: - new(ldigits[fNdtype[id]++]) AliITSdigitSPD(digits,tracks); - break; - case 1: - new(ldigits[fNdtype[id]++]) AliITSdigitSDD(phys,digits,tracks,charges); - break; - case 2: - new(ldigits[fNdtype[id]++]) AliITSdigitSSD(digits,tracks); - break; +//______________________________________________________________________ +void AliITS::SetTreeAddress(){ + // Set branch address for the Trees. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + + if(!fDetTypeSim) { + Error("SetTreeAddress","fDetTypeSim is 0!"); + return; } - -} - -//_____________________________________________________________________________ -void AliITS::AddCluster(Int_t id, AliITSRawCluster *c) -{ - - // add a cluster to the list - - // should have ctors of type AliITSRawClusterSDD(const AliITSRawClusterSDD &) - TClonesArray &lcl = *((TClonesArray*)(*fCtype)[id]); + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); - switch(id) - { - case 0: - new(lcl[fNctype[id]++]) AliITSRawClusterSPD(*((AliITSRawClusterSPD*)c)); - break; - case 1: - new(lcl[fNctype[id]++]) AliITSRawClusterSDD(*((AliITSRawClusterSDD*)c)); - break; - case 2: - new(lcl[fNctype[id]++]) AliITSRawClusterSSD(*((AliITSRawClusterSSD*)c)); - break; + TTree *treeS = fLoader->TreeS(); + TTree *treeD = fLoader->TreeD(); + if (fLoader->TreeH() && (fHits == 0x0)) { + fHits = new TClonesArray("AliITShit", 1560); } + AliDetector::SetTreeAddress(); + fDetTypeSim->SetTreeAddressS(treeS, (Char_t*)GetName()); + fDetTypeSim->SetTreeAddressD(treeD, (Char_t*)GetName()); } - - -//_____________________________________________________________________________ -void AliITS::AddRecPoint(const AliITSRecPoint &r) -{ - // - // Add a reconstructed space point to the list - // - TClonesArray &lrecp = *fRecPoints; - new(lrecp[fNRecPoints++]) AliITSRecPoint(r); +//______________________________________________________________________ +void AliITS::AddHit(Int_t track, Int_t *vol, Float_t *hits){ + // Add an ITS hit + // The function to add information to the AliITShit class. See the + // AliITShit class for a full description. This function allocates the + // necessary new space for the hit information and passes the variable + // track, and the pointers *vol and *hits to the AliITShit constructor + // function. + // Inputs: + // Int_t track Track number which produced this hit. + // Int_t *vol Array of Integer Hit information. See AliITShit.h + // Float_t *hits Array of Floating Hit information. see AliITShit.h + // Outputs: + // none. + // Return: + // none. + TClonesArray &lhits = *fHits; + new(lhits[fNhits++]) AliITShit(fIshunt,track,vol,hits); } - -//____________________________________________ -void AliITS::ResetDigits() -{ - // - // Reset number of digits and the digits array for thE ITS detector - // - - if (!fDtype) return; +//______________________________________________________________________ +void AliITS::FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules, + Option_t *option, const char *filename){ + // fill the modules with the sorted by module hits; add hits from + // background if option=Add. - Int_t i; - for(i=0;iClear(); - if (fNdtype) fNdtype[i]=0; - } + static TTree *trH1; //Tree with background hits + static Bool_t first=kTRUE; + static TFile *file; + const char *addBgr = strstr(option,"Add"); + + evnt = nmodules; // Dummy use of variables to remove warnings + if (addBgr ) { + if(first) { + file=new TFile(filename); + } // end if first + first=kFALSE; + file->cd(); + file->ls(); + // Get Hits Tree header from file + if(trH1) delete trH1; + trH1=0; + + char treeName[20]; + sprintf(treeName,"TreeH%d",bgrev); + trH1 = (TTree*)gDirectory->Get(treeName); + if (!trH1) { + Error("FillModules","cannot find Hits Tree for event:%d",bgrev); + } // end if !trH1 + // Set branch addresses + } // end if addBgr + + FillModules(fLoader->TreeH(),0); // fill from this file's tree. + + if (addBgr ) { + FillModules(trH1,10000000); // Default mask 10M. + TTree *fAli=fLoader->GetRunLoader()->TreeK(); + TFile *fileAli=0; + if (fAli) fileAli =fAli->GetCurrentFile(); + fileAli->cd(); + } // end if add + + } -//____________________________________________ -void AliITS::ResetDigits(Int_t i) -{ - // - // Reset number of digits and the digits array for this branch - // - if ((*fDtype)[i]) ((TClonesArray*)(*fDtype)[i])->Clear(); - if (fNdtype) fNdtype[i]=0; +//______________________________________________________________________ +void AliITS::FillModules(TTree *treeH, Int_t mask) { + // fill the modules with the sorted by module hits; + // can be called many times to do a merging + // Inputs: + // TTree *treeH The tree containing the hits to be copied into + // the modules. + // Int_t mask The track number mask to indecate which file + // this hits came from. + // Outputs: + // none. + // Return: + // none. + + if (treeH == 0x0) + { + Error("FillModules","Tree is NULL"); + } + Int_t lay,lad,det,index; + AliITShit *itsHit=0; + AliITSmodule *mod=0; + char branchname[20]; + sprintf(branchname,"%s",GetName()); + TBranch *branch = treeH->GetBranch(branchname); + if (!branch) { + Error("FillModules","%s branch in TreeH not found",branchname); + return; + } // end if !branch + branch->SetAddress(&fHits); + Int_t nTracks =(Int_t) treeH->GetEntries(); + Int_t iPrimTrack,h; + for(iPrimTrack=0; iPrimTrackGetEvent(iPrimTrack); + if (nBytes <= 0) continue; + Int_t nHits = fHits->GetEntriesFast(); + for(h=0; hUncheckedAt(h); + itsHit->GetDetectorID(lay,lad,det); + if (GetITSgeom()) { + index = GetITSgeom()->GetModuleIndex(lay,lad,det); + } else { + index=det-1; // This should not be used. + } // end if [You must have fITSgeom for this to work!] + mod = GetModule(index); + itsHit->SetTrack(itsHit->GetTrack()+mask); // Set track mask. + mod->AddHit(itsHit,iPrimTrack,h); + } // end loop over hits + } // end loop over tracks } -//____________________________________________ -void AliITS::ResetClusters() -{ - // - // Reset number of clusters and the clusters array for ITS - // - Int_t i; - for(i=0;iClear(); - if (fNctype) fNctype[i]=0; +//______________________________________________________________________ +Bool_t AliITS::InitModules(Int_t size,Int_t &nmodules){ + // Initialize the modules array. + // Inputs: + // Int_t size Size of array of the number of modules to be + // created. If size <=0 then the number of modules + // is gotten from AliITSgeom class kept in fITSgeom. + // Outputs: + // Int_t &nmodules The number of modules existing. + // Return: + // none. + + if(fITSmodules){ + fITSmodules->Delete(); + delete fITSmodules; + } // end fir fITSmoudles + + if(!fDetTypeSim) { + Error("InitModules","fDetTypeSim is null!"); + return kFALSE; } -} -//____________________________________________ -void AliITS::ResetClusters(Int_t i) -{ - // - // Reset number of clusters and the clusters array for this branch - // - if ((*fCtype)[i]) ((TClonesArray*)(*fCtype)[i])->Clear(); - if (fNctype) fNctype[i]=0; - -} - + Int_t nl,indexMAX,index; -//____________________________________________ -void AliITS::ResetRecPoints() -{ - // - // Reset number of rec points and the rec points array - // - fNRecPoints = 0; - if (fRecPoints) fRecPoints->Clear(); + if(size<=0){ // default to using data stored in AliITSgeom + if(fDetTypeSim->GetITSgeom()==0) { + Error("InitModules","fITSgeom not defined"); + return kFALSE; + } // end if fITSgeom==0 + nl = fDetTypeSim->GetITSgeom()->GetNlayers(); + indexMAX = fDetTypeSim->GetITSgeom()->GetIndexMax(); + nmodules = indexMAX; + fITSmodules = new TObjArray(indexMAX); + for(index=0;indexAddAt( new AliITSmodule(index),index); + } // end for index + }else{ + fITSmodules = new TObjArray(size); + for(index=0;indexAddAt( new AliITSmodule(index),index); + } // end for index + nmodules = size; + } // end i size<=0 + return kTRUE; } +//______________________________________________________________________ +void AliITS::Hits2SDigits(){ + // Standard Hits to summable Digits function. + // Inputs: + // none. + // Outputs: + // none. + -//_____________________________________________________________________________ -Int_t AliITS::DistancetoPrimitive(Int_t , Int_t ){ - // - // Distance from mouse to ITS on the screen. Dummy routine - // A dummy routine used by the ROOT macro display.C to allow for the - // use of the mouse (pointing device) in the macro. In general this should - // never be called. If it is it returns the number 9999 for any value of - // x and y. - // - return 9999; + if(!fDetTypeSim) { + Error("Hits2SDigits","fDetTypeSim is null!"); + return; + + } + + SetDefaults(); + fLoader->LoadHits("read"); + fLoader->LoadSDigits("recreate"); + AliRunLoader* rl = fLoader->GetRunLoader(); + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); + for (Int_t iEvent = 0; iEvent < rl->GetNumberOfEvents(); iEvent++) { + // Do the Hits to Digits operation. Use Standard input values. + // Event number from file, no background hit merging , use size from + // AliITSgeom class, option="All", input from this file only. + rl->GetEvent(iEvent); + if (!fLoader->TreeS()) fLoader->MakeTree("S"); + MakeBranch("S"); + SetTreeAddress(); + HitsToPreDigits(iEvent,0,-1," ",fOpt," "); + } // end for iEvent + + fLoader->UnloadHits(); + fLoader->UnloadSDigits(); + } +//______________________________________________________________________ +void AliITS::Hits2Digits(){ -//_____________________________________________________________________________ -void AliITS::Init(){ - // - // Initialise ITS after it has been built - // This routine initializes the AliITS class. It is intended to be called - // from the Init function in AliITSv?. Besides displaying a banner - // indicating that it has been called it initializes the array fIdSens - // and sets the default segmentation, response, digit and raw cluster classes - // Therefore it should be called after a call to CreateGeometry. - // - - + //Conversion from hits to digits + if(!fDetTypeSim) { + Error("Hits2SDigits","fDetTypeSim is 0!"); + return; + } + + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); SetDefaults(); - Int_t i; - - printf("\n"); - for(i=0;i<35;i++) printf("*"); - printf(" ITS_INIT "); - for(i=0;i<35;i++) printf("*"); - printf("\n"); - // - // - for(i=0;iVolId(fIdName[i]); - // - for(i=0;i<80;i++) printf("*"); - printf("\n"); -} - -//_____________________________________________________________________________ -void AliITS::SetDefaults() -{ - // sets the default segmentation, response, digit and raw cluster classes - - - AliITSDetType *iDetType; - - //SPD - - AliITSsegmentationSPD *seg0=new AliITSsegmentationSPD(fITSgeom); - AliITSresponseSPD *resp0=new AliITSresponseSPD(); - iDetType=DetType(0); - if (!iDetType->GetSegmentationModel()) SetSegmentationModel(0,seg0); - if (!iDetType->GetResponseModel()) SetResponseModel(0,resp0); - // set digit and raw cluster classes to be used - const char *kData=resp0->DataType(); - if (strstr(kData,"real")) { - iDetType->ClassNames("AliITSdigit","AliITSRawClusterSPD"); - } else iDetType->ClassNames("AliITSdigitSPD","AliITSRawClusterSPD"); - - // SDD // - AliITSresponseSDD *resp1=new AliITSresponseSDD(); - AliITSsegmentationSDD *seg1=new AliITSsegmentationSDD(fITSgeom,resp1); - iDetType=DetType(1); - if (!iDetType->GetSegmentationModel()) SetSegmentationModel(1,seg1); - if (!iDetType->GetResponseModel()) SetResponseModel(1,resp1); - kData=resp1->DataType(); - Option_t *opt=resp1->ZeroSuppOption(); - if ((!strstr(opt,"2D")) && (!strstr(opt,"1D")) || strstr(kData,"real") ) { - iDetType->ClassNames("AliITSdigit","AliITSRawClusterSDD"); - } else iDetType->ClassNames("AliITSdigitSDD","AliITSRawClusterSDD"); - - // SSD - AliITSsegmentationSSD *seg2=new AliITSsegmentationSSD(fITSgeom); - AliITSresponseSSD *resp2=new AliITSresponseSSD(); - iDetType=DetType(2); - if (!iDetType->GetSegmentationModel()) SetSegmentationModel(2,seg2); - if (!iDetType->GetResponseModel()) SetResponseModel(2,resp2); - kData=resp2->DataType(); - if (strstr(kData,"real")) { - iDetType->ClassNames("AliITSdigit","AliITSRawClusterSSD"); - } else iDetType->ClassNames("AliITSdigitSSD","AliITSRawClusterSSD"); - - if (fNDetTypes>3) { - Warning("SetDefaults","Only the three basic detector types are initialised!"); + fLoader->LoadHits("read"); + fLoader->LoadDigits("recreate"); + AliRunLoader* rl = fLoader->GetRunLoader(); + for (Int_t iEvent = 0; iEvent < rl->GetNumberOfEvents(); iEvent++) { + rl->GetEvent(iEvent); + if (!fLoader->TreeD()) fLoader->MakeTree("D"); + MakeBranch("D"); + SetTreeAddress(); + HitsToDigits(iEvent,0,-1," ",fOpt," "); } - + + fLoader->UnloadHits(); + fLoader->UnloadDigits(); + } - -//_____________________________________________________________________________ - -void AliITS::MakeTreeC(Option_t *option) -{ - // create a separate tree to store the clusters - - char *optC = strstr(option,"C"); - if (optC && !fTreeC) fTreeC = new TTree("TC","Clusters in ITS"); - - Int_t buffersize = 4000; - char branchname[30]; - - char *det[3] = {"SPD","SDD","SSD"}; - - // one branch for Clusters per type of detector - Int_t i; - for(i=0; iBranch(branchname,&((*fCtype)[i]), buffersize); - printf("Making Branch %s for Clusters of detector type %d\n",branchname,i+1); - } - } - +//______________________________________________________________________ +void AliITS::HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size, + Option_t *option,Option_t *opt, + const char *filename){ + // Keep galice.root for signal and name differently the file for + // background when add! otherwise the track info for signal will be lost ! + // the condition below will disappear when the geom class will be + // initialized for all versions - for the moment it is only for v5 ! + // 7 is the SDD beam test version. + // Inputs: + // Int_t evnt Event to be processed. + // Int_t bgrev Background Hit tree number. + // Int_t nmodules Not used. + // Option_t *option String indicating if merging hits or not. To + // merge hits set equal to "Add". Otherwise no + // background hits are considered. + // Test_t *filename File name containing the background hits.. + // Outputs: + // none. + // Return: + // none. + + if(!fDetTypeSim) { + Error("HitsToDigits","fDetTypeSim is null!"); + return; + } + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); + if(!GetITSgeom()) return; // need transformations to do digitization. + AliITSgeom *geom = GetITSgeom(); + + const char *all = strstr(opt,"All"); + const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"), + strstr(opt,"SSD")}; + static Bool_t setDef=kTRUE; + if (setDef) SetDefaultSimulation(); + setDef=kFALSE; + + Int_t nmodules; + InitModules(size,nmodules); + FillModules(evNumber,bgrev,nmodules,option,filename); + + AliITSsimulation *sim = 0; + AliITSmodule *mod = 0; + Int_t id; + for(Int_t module=0;moduleGetIndexMax();module++){ + id = geom->GetModuleType(module); + if (!all && !det[id]) continue; + sim = (AliITSsimulation*)fDetTypeSim->GetSimulationModel(id); + if (!sim) { + Error("HitsToDigits","The simulation class was not " + "instanciated for module %d type %s!",module, + geom->GetModuleTypeName(module)); + exit(1); + } // end if !sim + mod = (AliITSmodule *)fITSmodules->At(module); + sim->DigitiseModule(mod,module,evNumber); + // fills all branches - wasted disk space + fLoader->TreeD()->Fill(); + ResetDigits(); + } // end for module + + ClearModules(); + + fLoader->TreeD()->GetEntries(); + fLoader->TreeD()->AutoSave(); + // reset tree + fLoader->TreeD()->Reset(); } +//_____________________________________________________________________ +void AliITS::Hits2PreDigits(){ + // Turn hits into SDigits -//_____________________________________________________________________________ -void AliITS::GetTreeC(Int_t event) -{ - - // get the clusters tree for this event and set the branch address - char treeName[20]; - char branchname[30]; - - char *det[3] = {"SPD","SDD","SSD"}; + if(!fDetTypeSim) { + Error("Hits2SDigits","fDetTypeSim is 0!"); + return; + } + + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); + SetDefaults(); + + HitsToPreDigits(fLoader->GetRunLoader()->GetEventNumber(), + 0,-1," ",fOpt," "); +} - ResetClusters(); - if (fTreeC) { - delete fTreeC; - } +//______________________________________________________________________ +void AliITS::HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size, + Option_t *option,Option_t *opt, + const char *filename){ + // Keep galice.root for signal and name differently the file for + // background when add! otherwise the track info for signal will be lost ! + // the condition below will disappear when the geom class will be + // initialized for all versions - for the moment it is only for v5 ! + // 7 is the SDD beam test version. + // Inputs: + // Int_t evnt Event to be processed. + // Int_t bgrev Background Hit tree number. + // Int_t nmodules Not used. + // Option_t *option String indicating if merging hits or not. To + // merge hits set equal to "Add". Otherwise no + // background hits are considered. + // Test_t *filename File name containing the background hits.. + // Outputs: + // none. + // Return: + // none. - sprintf(treeName,"TreeC%d",event); - fTreeC = (TTree*)gDirectory->Get(treeName); + + if(!fDetTypeSim) { + Error("HitsToPreDigits","fDetTypeSim is null!"); + return; + } + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); + if(!GetITSgeom()){ + Error("HitsToPreDigits","fGeom is null!"); + return; // need transformations to do digitization. + } + AliITSgeom *geom = GetITSgeom(); + + const char *all = strstr(opt,"All"); + const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"), + strstr(opt,"SSD")}; + static Bool_t setDef=kTRUE; + if (setDef) SetDefaultSimulation(); + setDef=kFALSE; + + Int_t nmodules; + InitModules(size,nmodules); + FillModules(evNumber,bgrev,nmodules,option,filename); + - TBranch *branch; - if (fTreeC) { - Int_t i; - for(i=0; iGetBranch(branchname); - if (branch) branch->SetAddress(&((*fCtype)[i])); - } - } - } else { - printf("ERROR: cannot find Clusters Tree for event:%d\n",event); - } + AliITSsimulation *sim = 0; + AliITSmodule *mod = 0; + Int_t id,module; + for(module=0;moduleGetIndexMax();module++){ + id = geom->GetModuleType(module); + if (!all && !det[id]) continue; + sim = (AliITSsimulation*)GetSimulationModel(id); + if (!sim) { + Error("HitsToPreDigits","The simulation class was not " + "instanciated for module %d type %s!",module, + geom->GetModuleTypeName(module)); + exit(1); + } // end if !sim + mod = (AliITSmodule *)fITSmodules->At(module); + sim->SDigitiseModule(mod,module,evNumber); + // fills all branches - wasted disk space + fLoader->TreeS()->Fill(); + fDetTypeSim->ResetSDigits(); + } // end for module + + ClearModules(); + + fLoader->TreeS()->GetEntries(); + fLoader->TreeS()->AutoSave(); + fLoader->WriteSDigits("OVERWRITE"); + // reset tree + fLoader->TreeS()->Reset(); } -//_____________________________________________________________________________ -void AliITS::MakeBranch(Option_t* option){ - // - // Creates Tree branches for the ITS. - // +//_____________________________________________________________________ +void AliITS::HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size, + Option_t *opt0,Option_t *opt1, + const char *flnm){ + // keep galice.root for signal and name differently the file for + // background when add! otherwise the track info for signal will be lost ! + // the condition below will disappear when the geom class will be + // initialized for all versions - for the moment it is only for v5 ! + // Inputs: + // Int_t evnt Event to be processed. + // Int_t bgrev Background Hit tree number. + // Int_t size Size used by InitModules. See InitModules. + // Option_t *opt0 Option passed to FillModules. See FillModules. + // Option_t *opt1 String indicating if merging hits or not. To + // merge hits set equal to "Add". Otherwise no + // background hits are considered. + // Test_t *flnm File name containing the background hits.. + // Outputs: + // none. + // Return: + // none. + + + + if(!GetITSgeom()){ + Error("HitsToPreDigits","fGeom is null!"); + return; // need transformations to do digitization. + } + AliITSgeom *geom = GetITSgeom(); - Int_t buffersize = 4000; - char branchname[30]; - sprintf(branchname,"%s",GetName()); + AliITSLoader *pITSloader = (AliITSLoader*)fLoader; - AliDetector::MakeBranch(option); + const char *all = strstr(opt1,"All"); + const char *det[3] ={strstr(opt1,"SPD"),strstr(opt1,"SDD"), + strstr(opt1,"SSD")}; + Int_t nmodules; + InitModules(size,nmodules); + FillModules(evNumber,bgrev,nmodules,opt0,flnm); + AliITSsimulation *sim = 0; + AliITSmodule *mod = 0; + Int_t id,module; -// one branch for digits per type of detector + TTree *lTR = pITSloader->TreeR(); + if(!lTR) { + pITSloader->MakeTree("R"); + lTR = pITSloader->TreeR(); + } - char *det[3] = {"SPD","SDD","SSD"}; - - char *kDigclass; - char *kClclass; - - Int_t i; - for(i=0; iGetClassNames(kDigclass,kClclass); - //printf("i, digclass, recclass %d %s %s\n",i,kDigclass,kClclass); - // digits - (*fDtype)[i] = new TClonesArray(kDigclass,100); - fNdtype[i]=0; - // clusters - (*fCtype)[i] = new TClonesArray(kClclass,100); - fNctype[i]=0; - } - - - for(i=0; iTreeD()) { - gAlice->TreeD()->Branch(branchname,&((*fDtype)[i]), buffersize); - printf("Making Branch %s for digits of type %d\n",branchname,i+1); - } + TClonesArray* ptarray = new TClonesArray("AliITSRecPoint",1000); + TBranch* branch = (TBranch*)lTR->Branch("ITSRecPointsF",&ptarray); + branch->SetAddress(&ptarray); + //m.b. : this change is nothing but a nice way to make sure + //the CPU goes up ! + for(module=0;moduleGetIndexMax();module++){ + id = geom->GetModuleType(module); + if (!all && !det[id]) continue; + sim = (AliITSsimulation*)GetSimulationModel(id); + if (!sim) { + Error("HitsToFastPoints","The simulation class was not " + "instanciated for module %d type %x!",module, + geom->GetModuleTypeName(module)); + exit(1); + } // end if !sim + mod = (AliITSmodule *)fITSmodules->At(module); + sim->CreateFastRecPoints(mod,module,gRandom,ptarray); + lTR->Fill(); + } // end for module + + ClearModules(); + fLoader->WriteRecPoints("OVERWRITE"); + delete ptarray; +} +//_____________________________________________________________________ +Int_t AliITS::Hits2Clusters(TTree *hTree, TTree *cTree) { + //------------------------------------------------------------ + // This function creates ITS clusters + //------------------------------------------------------------ + if(!GetITSgeom()){ + Error("HitsToPreDigits","fGeom is null!"); + return 1; // need transformations to do digitization. + } + AliITSgeom *geom=GetITSgeom(); + Int_t mmax=geom->GetIndexMax(); + + InitModules(-1,mmax); + FillModules(hTree,0); + + TClonesArray *points = new TClonesArray("AliITSRecPoint",1000); + TBranch *branch=cTree->GetBranch("ITSRecPoints"); + if (!branch) cTree->Branch("ITSRecPoints",&points); + else branch->SetAddress(&points); + + AliITSsimulationFastPoints sim; + Int_t ncl=0; + for (Int_t m=0; mGetEntriesFast(); + cTree->Fill(); + points->Clear(); } - // only one branch for rec points for all detector types - sprintf(branchname,"%sRecPoints",GetName()); - - fRecPoints=new TClonesArray("AliITSRecPoint",1000); - - if (fRecPoints && gAlice->TreeR()) { - gAlice->TreeR()->Branch(branchname,&fRecPoints, buffersize); - printf("Making Branch %s for reconstructed space points\n",branchname); - } + Info("Hits2Clusters","Number of found fast clusters : %d",ncl); + //cTree->Write(); + delete points; + return 0; } -//___________________________________________ -void AliITS::SetTreeAddress() -{ - - // Set branch address for the Trees. - - char branchname[30]; - AliDetector::SetTreeAddress(); - - char *det[3] = {"SPD","SDD","SSD"}; - - TBranch *branch; - TTree *treeD = gAlice->TreeD(); - TTree *treeR = gAlice->TreeR(); - - Int_t i; - if (treeD) { - for(i=0; iGetBranch(branchname); - if (branch) branch->SetAddress(&((*fDtype)[i])); - } +//_____________________________________________________________________ +void AliITS::CheckLabels(Int_t lab[3]) const { + //------------------------------------------------------------ + // Tries to find mother's labels + //------------------------------------------------------------ + + if(lab[0]<0 && lab[1]<0 && lab[2]<0) return; // In case of no labels just exit + + Int_t ntracks = gAlice->GetMCApp()->GetNtrack(); + for (Int_t i=0;i<3;i++){ + Int_t label = lab[i]; + if (label>=0 && labelGetMCApp()->Particle(label); + if (part->P() < 0.005) { + Int_t m=part->GetFirstMother(); + if (m<0) { + continue; + } + if (part->GetStatusCode()>0) { + continue; + } + lab[i]=m; } - } - - - if (treeR) { - sprintf(branchname,"%sRecPoints",GetName()); - if (fRecPoints) { - branch = treeR->GetBranch(branchname); - if (branch) branch->SetAddress(&fRecPoints); - } + } } - } -//____________________________________________________________________________ -void AliITS::InitModules(Int_t size,Int_t &nmodules){ - - //initialize the modules array - - Int_t nl,indexMAX,index; - Int_t indSPD,indSDD; +//______________________________________________________________________ +void AliITS::SDigitsToDigits(Option_t *opt){ + // Standard Summable digits to Digits function. + // Inputs: + // none. + // Outputs: + // none. + if(!fDetTypeSim) { + Error("SDigitsToSDigits","fDetTypeSim is 0!"); + return; + } + + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); + SetDefaults(); + fDetTypeSim->SDigitsToDigits(opt,(Char_t*)GetName()); - if(size<=0){ // default to using data stored in AliITSgeom - if(fITSgeom==0) { - printf("Error in AliITS::InitModule fITSgeom not defined\n"); - return; - } // end if fITSgeom==0 - nl = fITSgeom->GetNlayers(); - indexMAX = fITSgeom->GetModuleIndex(nl,fITSgeom->GetNladders(nl), - fITSgeom->GetNdetectors(nl))+1; - nmodules = indexMAX; - fITSmodules = new TObjArray(indexMAX); - indSPD = fITSgeom->GetModuleIndex(2,fITSgeom->GetNladders(2), - fITSgeom->GetNdetectors(2)); - indSDD = fITSgeom->GetModuleIndex(4,fITSgeom->GetNladders(4), - fITSgeom->GetNdetectors(4)); - for(index=0;indexAddAt( new AliITSmodule(index),index); - else if(index<=indSDD) - fITSmodules->AddAt( new AliITSmodule(index),index); - else - fITSmodules->AddAt( new AliITSmodule(index),index); - } // end for index - }else{ - fITSmodules = new TObjArray(size); - nmodules = size; - } // end i size<=0 } -//____________________________________________________________________________ -void AliITS::FillModules(Int_t evnt,Int_t bgrev,Int_t lastev,Int_t nmodules,Option_t *option,Text_t *filename){ - - // fill the modules with the sorted by module hits; add hits from background - // if option=Add - - static TTree *trH1; //Tree with background hits - static TClonesArray *fHits2; //List of hits for one track only - - static Bool_t first=kTRUE; - static TFile *file; - char *add = strstr(option,"Add"); - - - if (add ) { - if(first) { - cout<<"filename "<cd(); - file->ls(); - // Get Hits Tree header from file - if(fHits2) fHits2->Clear(); - if(trH1) delete trH1; - trH1=0; - - char treeName[20]; - sprintf(treeName,"TreeH%d",bgrev); - trH1 = (TTree*)gDirectory->Get(treeName); - //printf("TrH1 %p of treename %s for event %d \n",trH1,treeName,bgrev); - - if (!trH1) { - printf("ERROR: cannot find Hits Tree for event:%d\n",bgrev); - } - // Set branch addresses - TBranch *branch; - char branchname[20]; - sprintf(branchname,"%s",GetName()); - if (trH1 && fHits2) { - branch = trH1->GetBranch(branchname); - if (branch) branch->SetAddress(&fHits2); - } - - // test - //Int_t ntracks1 =(Int_t)TrH1->GetEntries(); - //printf("background - ntracks1 - %d\n",ntracks1); - } +//______________________________________________________________________ +void AliITS::ResetDigits(){ + // Reset number of digits and the digits array for the ITS detector. + // Inputs: + // none. + // Outputs: + // none. + if(!fDetTypeSim) { + Error("ResetDigits","fDetTypeSim is 0!"); + return; + } + + fDetTypeSim->ResetDigits(); - // store temporarily coordinates of signal particles - see where delete - static Int_t *signal; - Int_t i; - if(!signal) signal=new Int_t[nmodules]; - memset(signal,0,sizeof(int)*nmodules); -// Float_t xhit[nmodules][4]; - Float_t **xhit = new Float_t*[nmodules]; - for(i=0;iGetEvent(evnt); - if(npart<=0) return; - TClonesArray *itsHits = this->Hits(); - Int_t lay,lad,det,index; - AliITShit *itsHit=0; - AliITSmodule *mod=0; - TTree *iTH = gAlice->TreeH(); - Int_t ntracks =(Int_t) iTH->GetEntries(); - - Int_t t,h; - for(t=0; tResetHits(); - iTH->GetEvent(t); - Int_t nhits = itsHits->GetEntriesFast(); - if (!nhits) continue; - // cout << nhits << " hits in track " << t << endl; - for(h=0; hUncheckedAt(h); - itsHit->GetDetectorID(lay,lad,det); - index = fITSgeom->GetModuleIndex(lay,lad,det); - mod = this->GetModule(index); - if (add) { - xhit[index][signal[index]]=itsHit->fX; - yhit[index][signal[index]]=itsHit->fY; - signal[index]++; - if (signal[index] >4) - printf("index,nsignal %d %d\n",index,signal[index]); - } - if(lay == 1 || lay == 2) - mod->AddHit((AliITShit *) itsHit,t,h); - else if(lay == 3 || lay == 4) - mod->AddHit((AliITShit *) itsHit,t,h); - else if(lay == 5 || lay ==6) - mod->AddHit((AliITShit *)itsHit,t,h); - else - mod->AddHit(itsHit,t,h); - - } // end loop over hits - } // end loop over tracks +} +//______________________________________________________________________ +void AliITS::ResetDigits(Int_t branch){ + // Reset number of digits and the digits array for this branch. + // Inputs: + // none. + // Outputs: + // none. + + if(!fDetTypeSim) { + Error("ResetDigits","fDetTypeSim is 0!"); + return; + } + + fDetTypeSim->ResetDigits(branch); - // open the file with background +} +//______________________________________________________________________ +void AliITS::AddSumDigit(AliITSpListItem &sdig){ + // Adds the a module full of summable digits to the summable digits tree. + // Inputs: + // AliITSpListItem &sdig SDigit to be added to SDigits tree. + // Outputs: + // none. + // Return: + // none. + + if(!fDetTypeSim) { + Error("AddSumDigit","fDetTypeSim is 0!"); + return; + } + fDetTypeSim->AddSumDigit(sdig); - if (add ) { - Int_t track,i,isig; - ntracks =(Int_t)trH1->GetEntries(); - //printf("background - ntracks1 %d\n",ntracks); - //printf("background - Start loop over tracks \n"); - // Loop over tracks - - for(track=0; trackClear(); - trH1->GetEvent(track); - // Loop over hits - for(i=0;iGetEntriesFast();++i) { - - itsHit=(AliITShit*) (*fHits2)[i]; - itsHit->GetDetectorID(lay,lad,det); - index = fITSgeom->GetModuleIndex(lay,lad,det); - mod = this->GetModule(index); - - Float_t xbgr=itsHit->fX; - Float_t ybgr=itsHit->fY; - Float_t ebgr=itsHit->GetIonization(); - Bool_t cond=kFALSE; - - for(isig =0; isig < signal[index]; isig++) { - Float_t dist= - (xbgr-xhit[index][isig])*(xbgr-xhit[index][isig]) - +(ybgr-yhit[index][isig])*(ybgr-yhit[index][isig]); - if (dist<0.2&& ebgr!=0) cond=kTRUE; // check this number for ITS! - } - if (!cond) continue; - if(lay == 1 || lay == 2) - mod->AddHit((AliITShit *) itsHit,track,i); - else if(lay == 3 || lay == 4) - mod->AddHit((AliITShit *) itsHit,track,i); - else if(lay == 5 || lay ==6) - mod->AddHit((AliITShit *)itsHit,track,i); - else - mod->AddHit(itsHit,track,i); - - - } // end loop over hits - } // end loop over tracks - - TTree *fAli=gAlice->TreeK(); - TFile *fileAli=0; - - if (fAli) fileAli =fAli->GetCurrentFile(); - //printf("fAli, file %p %p\n",fAli,file); - file->cd(); - - } // end if add - - - for(i=0;iPrint(); - } +//______________________________________________________________________ +void AliITS::AddSimDigit(Int_t branch, AliITSdigit *d){ + // Add a simulated digit. + // Inputs: + // Int_t id Detector type number. + // AliITSdigit *d Digit to be added to the Digits Tree. See + // AliITSdigit.h + // Outputs: + // none. + // Return: + // none. + + if(!fDetTypeSim) { + Error("AddSimDigit","fDetTypeSim is 0!"); + return; + } + fDetTypeSim->AddSimDigit(branch,d); - -//____________________________________________________________________________ -void AliITS::HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t lastev,Int_t size, -Option_t *option,Option_t *opt,Text_t *filename) -{ - // keep galice.root for signal and name differently the file for - // background when add! otherwise the track info for signal will be lost ! - - char *all = strstr(opt,"All"); - char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")}; - //printf("Det 1 2 3 %s %s %s \n",det[0],det[1],det[2]); - - Int_t nmodules; - InitModules(size,nmodules); - FillModules(evNumber,bgrev,lastev,nmodules,option,filename); - //printf("nmodules %d\n",nmodules); - - TBranch *branch; - AliITSsimulation* sim; - - TObjArray *branches=gAlice->TreeD()->GetListOfBranches(); - AliITSgeom *geom = GetITSgeom(); - - Int_t id,module; - for(id=0;id<3;id++) { - //printf("id %d All %s det[id] %s \n",id,all,det[id]); - if (!all && !det[id]) continue; - branch = (TBranch*)branches->UncheckedAt(id); - AliITSDetType *iDetType=DetType(id); - sim = (AliITSsimulation*)iDetType->GetSimulationModel(); - if (!sim) { - Error("HitsToDigits","The simulation class was not instantiated!"); - exit(1); - // or SetDefaultSimulation(id,iDetType*); - } - Int_t first = geom->GetStartDet(id); - Int_t last = geom->GetLastDet(id); - //printf("det type %d first, last %d %d \n",id,first,last); - for(module=first;module<=last;module++) { - AliITSmodule *mod = (AliITSmodule *)fITSmodules->At(module); - sim->DigitiseModule(mod,module,evNumber); - // fills all branches - wasted disk space - gAlice->TreeD()->Fill(); - ResetDigits(); - // try and fill only the branch - //branch->Fill(); - //ResetDigits(id); - } // loop over modules - } // loop over detector types - - - ClearModules(); - - //Int_t nentries=(Int_t)gAlice->TreeD()->GetEntries(); - - char hname[30]; - sprintf(hname,"TreeD%d",evNumber); - gAlice->TreeD()->Write(hname); - // reset tree - gAlice->TreeD()->Reset(); +} +//______________________________________________________________________ +void AliITS::AddSimDigit(Int_t branch,Float_t phys,Int_t *digits,Int_t *tracks, + Int_t *hits,Float_t *charges, Int_t sigexpanded){ + // Add a simulated digit to the list. + // Inputs: + // Int_t id Detector type number. + // Float_t phys Physics indicator. See AliITSdigits.h + // Int_t *digits Integer array containing the digits info. See + // AliITSdigit.h + // Int_t *tracks Integer array [AliITSdigitS?D::GetNTracks()] + // containing the track numbers that contributed to + // this digit. + // Int_t *hits Integer array [AliITSdigitS?D::GetNTracks()] + // containing the hit numbers, from AliITSmodule, that + // contributed to this digit. + // Float_t *charge Floating point array of the signals contributed + // to this digit by each track. + // Outputs: + // none. + // Return: + // none. + + if(!fDetTypeSim) { + Error("AddSimDigit","fDetTypeSim is 0!"); + return; + } + fDetTypeSim->AddSimDigit(branch,phys,digits,tracks,hits,charges,sigexpanded); } +//______________________________________________________________________ +void AliITS::Digits2Raw(){ + // convert digits of the current event to raw data + if(!fDetTypeSim) { + Error("Digits2Raw","fDetTypeSim is 0!"); + return; + } + fDetTypeSim->SetLoader((AliITSLoader*)fLoader); + SetDefaults(); + fDetTypeSim->GetLoader()->LoadDigits(); + TTree* digits = fDetTypeSim->GetLoader()->TreeD(); + if (!digits) { + Error("Digits2Raw", "no digits tree"); + return; + } + fDetTypeSim->SetTreeAddressD(digits,(Char_t*)GetName()); + AliITSDDLModuleMapSDD* ddlsdd=fDetTypeSim->GetDDLModuleMapSDD(); + + AliITSDDLRawData rawWriter; + //Verbose level + // 0: Silent + // 1: cout messages + // 2: txt files with digits + //BE CAREFUL, verbose level 2 MUST be used only for debugging and + //it is highly suggested to use this mode only for debugging digits files + //reasonably small, because otherwise the size of the txt files can reach + //quickly several MB wasting time and disk space. + rawWriter.SetVerbose(0); + + //SILICON PIXEL DETECTOR + Info("Digits2Raw", "Formatting raw data for SPD"); + rawWriter.RawDataSPD(digits->GetBranch("ITSDigitsSPD")); + + //SILICON DRIFT DETECTOR + Info("Digits2Raw", "Formatting raw data for SDD"); + rawWriter.RawDataSDD(digits->GetBranch("ITSDigitsSDD"),ddlsdd); + + //SILICON STRIP DETECTOR + Info("Digits2Raw", "Formatting raw data for SSD"); + rawWriter.RawDataSSD(digits->GetBranch("ITSDigitsSSD")); -//____________________________________________________________________________ -void AliITS::DigitsToRecPoints(Int_t evNumber,Int_t lastentry,Option_t *opt) -{ - - char *all = strstr(opt,"All"); - char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")}; - - static Bool_t first=kTRUE; - if (first) { - MakeTreeC("C"); - first=kFALSE; - } - TTree *iTC=TreeC(); - - TBranch *branch; - AliITSClusterFinder* rec; - - TObjArray *branches=gAlice->TreeR()->GetListOfBranches(); - AliITSgeom *geom = GetITSgeom(); - - Int_t id,module; - for(id=0;idUncheckedAt(id); - AliITSDetType *iDetType=DetType(id); - rec = (AliITSClusterFinder*)iDetType->GetReconstructionModel(); - if (!rec) { - Error("DigitsToRecPoints","The cluster finder class was not instantiated!"); - exit(1); - // or SetDefaultClusterFinder(id,iDetType*); - } - TClonesArray *itsDigits = this->DigitsAddress(id); - - Int_t first = geom->GetStartDet(id); - Int_t last = geom->GetLastDet(id); - //printf("det type %d first, last %d %d \n",id,first,last); - for(module=first;module<=last;module++) { - //printf("AliITS: module=%d\n",module); - this->ResetDigits(); - if (all) gAlice->TreeD()->GetEvent(lastentry+module); - else gAlice->TreeD()->GetEvent(lastentry+(module-first)); - Int_t ndigits = itsDigits->GetEntriesFast(); - if (ndigits) rec->FindRawClusters(); - gAlice->TreeR()->Fill(); - ResetRecPoints(); - iTC->Fill(); - ResetClusters(); - // try and fill only the branch - //branch->Fill(); - //ResetRecPoints(id); - } // loop over modules - } // loop over detector types - - - //Int_t nentries=(Int_t)gAlice->TreeR()->GetEntries(); - - //Int_t ncentries=(Int_t)TC->GetEntries(); - - char hname[30]; - sprintf(hname,"TreeR%d",evNumber); - gAlice->TreeR()->Write(hname); - // reset tree - gAlice->TreeR()->Reset(); - - sprintf(hname,"TreeC%d",evNumber); - iTC->Write(hname); - iTC->Reset(); + fLoader->UnloadDigits(); +} +//______________________________________________________________________ +AliLoader* AliITS::MakeLoader(const char* topfoldername){ + //builds ITSgetter (AliLoader type) + //if detector wants to use castomized getter, it must overload this method + + AliDebug(1,Form("Creating AliITSLoader. Top folder is %s.", + topfoldername)); + fLoader = new AliITSLoader(GetName(),topfoldername); + return fLoader; } - -//____________________________________________________________________________ -void AliITS::HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t lastev,Int_t size, -Option_t *option,Option_t *opt,Text_t *filename) +Bool_t AliITS::Raw2SDigits(AliRawReader* rawReader) { - // keep galice.root for signal and name differently the file for - // background when add! otherwise the track info for signal will be lost ! - - char *all = strstr(opt,"All"); - char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),strstr(opt,"SSD")}; - - Int_t nmodules; - InitModules(size,nmodules); - FillModules(evNumber,bgrev,lastev,nmodules,option,filename); - - static AliITSsimulationFastPoints* sim=0; - if (!sim) sim = new AliITSsimulationFastPoints(); + // + // Converts RAW data to SDigits + // + // Get TreeS + // + Int_t last = -1; + Int_t size = GetITSgeom()->GetIndexMax(); + TClonesArray** modA = new TClonesArray*[size]; + for (Int_t mod = 0; mod < size; mod++) modA[mod] = new TClonesArray("AliITSpListItem", 10000); + + AliLoader* loader = (gAlice->GetRunLoader())->GetLoader("ITSLoader"); + if (!loader) + { + Error("Open","Can not get ITS loader from Run Loader"); + return kFALSE; + } + TTree* tree = 0; + tree = loader->TreeS(); + if (!tree) + { + loader->MakeTree("S"); + tree = loader->TreeS(); + } + // + // Array for SDigits + // + TClonesArray aSDigits("AliITSpListItem",10000), *itsSDigits=&aSDigits; + Int_t bufsize = 32000; + tree->Branch("ITS", &itsSDigits, bufsize); + Int_t npx = 0; + // + // SPD + // + AliITSsegmentationSPD* segSPD = (AliITSsegmentationSPD*) fDetTypeSim->GetSegmentationModel(0); + if(!segSPD){ + AliWarning("Set AliITS defaults"); + SetDefaults(); + segSPD = (AliITSsegmentationSPD*) fDetTypeSim->GetSegmentationModel(0); + } + npx = segSPD->Npx(); + Double_t thr, sigma; + + AliITSRawStreamSPD inputSPD(rawReader); + while(1){ + Bool_t next = inputSPD.Next(); + if (!next) break; - AliITSgeom *geom = GetITSgeom(); + Int_t module = inputSPD.GetModuleID(); + Int_t column = inputSPD.GetColumn(); + Int_t row = inputSPD.GetRow(); + Int_t index = npx * column + row; - Int_t id,module; - for(id=0;id<3;id++) { - if (!all && !det[id]) continue; - Int_t first = geom->GetStartDet(id); - Int_t last = geom->GetLastDet(id); - for(module=first;module<=last;module++) { - AliITSmodule *mod = (AliITSmodule *)fITSmodules->At(module); - sim->CreateFastRecPoints(mod); - gAlice->TreeR()->Fill(); - ResetRecPoints(); - } // loop over modules - } // loop over detector types + if (module >= size) continue; + + last = (modA[module])->GetEntries(); + TClonesArray& dum = *modA[module]; + fDetTypeSim->GetCalibrationModel(module)->Thresholds(thr,sigma); + thr += 1.; + new (dum[last]) AliITSpListItem(-1, -1, module, index, thr); + } + rawReader->Reset(); + // + // SDD + // + AliITSsegmentationSDD* segSDD = (AliITSsegmentationSDD*) fDetTypeSim->GetSegmentationModel(1); + npx = segSDD->Npx(); + AliITSRawStreamSDD inputSDD(rawReader); + AliITSDDLModuleMapSDD* ddlmap=fDetTypeSim->GetDDLModuleMapSDD(); + inputSDD.SetDDLModuleMap(ddlmap); + while(1){ + Bool_t next = inputSDD.Next(); + if (!next) break; + + Int_t module = inputSDD.GetModuleID(); + Int_t anode = inputSDD.GetAnode(); + Int_t time = inputSDD.GetTime(); + Int_t signal10 = inputSDD.GetSignal(); + Int_t index = npx * anode + time; + + if (module >= size) continue; + last = modA[module]->GetEntries(); + TClonesArray& dum = *modA[module]; + new (dum[last]) AliITSpListItem(-1, -1, module, index, Double_t(signal10)); + ((AliITSpListItem*) dum.At(last))->AddSignalAfterElect(module, index, Double_t(signal10)); + + } + rawReader->Reset(); - ClearModules(); + // + // SSD + // + AliITSsegmentationSSD* segSSD = (AliITSsegmentationSSD*) fDetTypeSim->GetSegmentationModel(2); + npx = segSSD->Npx(); + AliITSRawStreamSSD inputSSD(rawReader); + while(1){ + Bool_t next = inputSSD.Next(); + if (!next) break; + + Int_t module = inputSSD.GetModuleID(); + Int_t side = inputSSD.GetSideFlag(); + Int_t strip = inputSSD.GetStrip(); + Int_t signal = inputSSD.GetSignal(); + Int_t index = npx * side + strip; + + if (module >= size) continue; + + last = modA[module]->GetEntries(); + TClonesArray& dum = *modA[module]; + new (dum[last]) AliITSpListItem(-1, -1, module, index, Double_t(signal)); + } + rawReader->Reset(); + AliITSpListItem* sdig = 0; + + for (Int_t mod = 0; mod < size; mod++) + { + Int_t nsdig = modA[mod]->GetEntries(); + for (Int_t ie = 0; ie < nsdig; ie++) { + sdig = (AliITSpListItem*) (modA[mod]->At(ie)); + new (aSDigits[ie]) AliITSpListItem(-1, -1, mod, sdig->GetIndex(), sdig->GetSignal()); + Float_t sig = sdig->GetSignalAfterElect(); + if (sig > 0.) { + sdig = (AliITSpListItem*)aSDigits[ie]; + sdig->AddSignalAfterElect(mod, sdig->GetIndex(), Double_t(sig)); + } + } + + tree->Fill(); + aSDigits.Clear(); + modA[mod]->Clear(); + } + loader->WriteSDigits("OVERWRITE"); + delete modA; + return kTRUE; +} - //Int_t nentries=(Int_t)gAlice->TreeR()->GetEntries(); - char hname[30]; - sprintf(hname,"TreeR%d",evNumber); - gAlice->TreeR()->Write(hname); - // reset tree - gAlice->TreeR()->Reset(); +//______________________________________________________________________ +void AliITS::UpdateInternalGeometry(){ -} + //reads new geometry from TGeo +// AliDebug(1,"Delete ITSgeom and create a new one reading TGeo"); -//______________________________________________________________________________ -void AliITS::Streamer(TBuffer &R__b) -{ - // Stream an object of class AliITS. - - Int_t i, j, l; - - if (R__b.IsReading()) { - Version_t R__v = R__b.ReadVersion(); - if (R__v) { - AliDetector::Streamer(R__b); - R__b >> fITSgeom; - R__b >> fITSmodules; - R__b >> fEuclidOut; - R__b >> fIdN; - delete []fIdSens; - fIdSens = new Int_t[fIdN]; - R__b.ReadFastArray(fIdSens,fIdN); - if(fIdName!=0) delete[] fIdName; - fIdName = new char*[fIdN]; - for(i=0;i> l; - fIdName[i] = new char[l+1]; // add room for null character. - for(j=0;j> fIdName[i][j]; - fIdName[i][l] = '\0'; // Null terminate this string. - } // end for i - //R__b.ReadArray(fIdName); - R__b >> fMajorVersion; - R__b >> fMinorVersion; - R__b >> fDetTypes; - R__b >> fNDetTypes; - R__b >> fDtype; - delete []fNdtype; - fNdtype = new Int_t[fNDetTypes]; - R__b.ReadFastArray(fNdtype,fNDetTypes); - R__b >> fCtype; - delete []fNctype; - fNctype = new Int_t[fNDetTypes]; - R__b.ReadFastArray(fNctype,fNDetTypes); - R__b >> fRecPoints; - R__b >> fNRecPoints; - R__b >> fTracks; - R__b >> fTreeC; - } // end if R__v - } else { // writing - R__b.WriteVersion(AliITS::IsA()); - AliDetector::Streamer(R__b); - R__b << fITSgeom; - R__b << fITSmodules; - R__b << fEuclidOut; - R__b << fIdN; - R__b.WriteFastArray(fIdSens,fIdN); - for(i=0;iStreamer(R__b); - R__b << fNRecPoints; - R__b << fTracks; - // fTracks->Streamer(R__b); - R__b << fTreeC; - } // end if + AliITSVersion_t version = (AliITSVersion_t)IsVersion(); + Int_t minor = 0; + if(version==kvPPRasymmFMD) minor=2; // default minor version for this geom. + AliITSInitGeometry initgeom; + AliITSgeom* geom = initgeom.CreateAliITSgeom(version,minor); + SetITSgeom(geom); } -ClassImp(AliITSRecPoint) -ClassImp(AliITSsegmentation) -ClassImp(AliITSresponse) -//ClassImp(AliITStrack) -