X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITS.cxx;h=9f1bf6f7634a985393d256282aee1c761995d1fa;hb=16a4aa1dce38f8842b311563f8feaca579fd9d37;hp=195439c5da44d87ce5eabe5cc1d6e73191feb8dc;hpb=114da3a41ac4da8bf3d738e8f86564484709602c;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITS.cxx b/ITS/AliITS.cxx index 195439c5da4..9f1bf6f7634 100644 --- a/ITS/AliITS.cxx +++ b/ITS/AliITS.cxx @@ -13,50 +13,15 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* -$Log$ -Revision 1.21 2000/10/02 21:28:08 fca -Removal of useless dependecies via forward declarations - -Revision 1.20 2000/10/02 16:31:39 barbera -General code clean-up - -Revision 1.9.2.14 2000/10/02 15:43:51 barbera -General code clean-up (e.g., printf -> cout) - -Revision 1.19 2000/09/22 12:13:25 nilsen -Patches and updates for fixes to this and other routines. - -Revision 1.18 2000/07/12 05:32:20 fca -Correcting several syntax problem with static members - -Revision 1.17 2000/07/10 16:07:18 fca -Release version of ITS code - -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,1113 +64,1140 @@ the AliITS class. // futher information. // /////////////////////////////////////////////////////////////////////////////// - -#include -#include -#include -#include -#include -#include -#include + +#include +#include #include -#include +#include #include - - - -#include "AliRun.h" +#include +#include +#include "AliDetector.h" #include "AliITS.h" -#include "AliITSMap.h" -#include "AliITSDetType.h" -#include "AliITSClusterFinder.h" -#include "AliITSsimulation.h" -#include "AliITSsegmentationSPD.h" -#include "AliITSresponseSPD.h" -#include "AliITSsegmentationSDD.h" -#include "AliITSresponseSDD.h" -#include "AliITSsegmentationSSD.h" -#include "AliITSresponseSSD.h" +#include "AliITSDetTypeSim.h" +#include "AliITSDDLRawData.h" +#include "AliITSLoader.h" #include "AliITShit.h" -#include "AliITSgeom.h" -#include "AliITSdigit.h" #include "AliITSmodule.h" -#include "AliITSRecPoint.h" -#include "AliITSRawCluster.h" +#include "AliITSpListItem.h" +#include "AliITSsimulation.h" +#include "AliITSsimulationFastPoints.h" #include "AliMC.h" -#include "stdlib.h" - -const Int_t AliITS::fgkNTYPES=3; +#include "AliITSDigitizer.h" +#include "AliITSRecPoint.h" +#include "AliITSclusterV2.h" +#include "AliRun.h" +#include "AliLog.h" ClassImp(AliITS) - -//_____________________________________________________________________________ -AliITS::AliITS() : AliDetector() { - // - // Default initialiser for ITS - // The default constructor of the AliITS class. In addition to + +//______________________________________________________________________ +AliITS::AliITS() : AliDetector(){ + // 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; - - //fNDetTypes = fgkNTYPES; - fIdN = 0; - fIdName = 0; - fIdSens = 0; - fITSmodules = 0; - // - fDetTypes = 0; - // - fDtype = 0; - fNdtype = 0; - fCtype = 0; - fNctype = 0; - fRecPoints = 0; - fNRecPoints = 0; - fTreeC = 0; - // - fITSgeom=0; + + fEuclidOut=0; + fOpt="All"; + fIdSens=0; + fIdName=0; + fDetTypeSim=0; + fIshunt = 0; // not zeroed in AliDetector. + fHits =0; + fNhits=0; + fITSmodules=0; + +// SetDetectors(); // default to fOpt="All". This variable not written out. + SetMarkerColor(kRed); } - -//_____________________________________________________________________________ +//______________________________________________________________________ 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); - - //fNDetTypes = fgkNTYPES; - - fNdtype = new Int_t[fgkNTYPES]; - fDtype = new TObjArray(fgkNTYPES); - - fNctype = new Int_t[fgkNTYPES]; - fCtype = new TObjArray(fgkNTYPES); - - - fRecPoints = 0; - fNRecPoints = 0; - - fTreeC = 0; + // 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. + + fEuclidOut=0; + fOpt="All"; + fIdSens=0; + fIdName=0; + fDetTypeSim=0; + fHits = new TClonesArray("AliITShit",1560); + if(gAlice->GetMCApp()) gAlice->GetMCApp()->AddHitList(fHits); + fNhits=0; + fITSmodules = 0; - fITSmodules = 0; + fIshunt = 0; // not zeroed in AliDetector + // Not done in AliDetector. - fIshunt = 0; fEuclidOut = 0; - fIdN = 0; + SetDetectors(); // default to fOpt="All". This variable not written out. + fIdName = 0; fIdSens = 0; - - fDetTypes = new TObjArray(fgkNTYPES); - - Int_t i; - for(i=0;iDelete(); + 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){ - // assignment operator - if(this==&source) return *this; - Error("AliITS::operator=", - "You are not allowed to make a copy of the AliITS"); - exit(1); - return *this; //fake return +//______________________________________________________________________ +AliITS::AliITS(const AliITS &source) : AliDetector(source){ + // Copy constructor. This is a function which is not allowed to be + // done to the ITS. It exits with an error. + // Inputs: + // AliITS &source An AliITS class. + // Outputs: + // none. + // Return: + // none. + + if(this==&source) return; + Error("Copy constructor", + "You are not allowed to make a copy of the AliITS"); + exit(1); } -//____________________________________________________________________________ -void AliITS::ClearModules(){ - //clear the modules TObjArray - - if(fITSmodules) fITSmodules->Delete(); - +//______________________________________________________________________ +AliITS& AliITS::operator=(const AliITS &source){ + // Assignment operator. This is a function which is not allowed to be + // done to the ITS. It exits with an error. + // Inputs: + // AliITS &source An AliITS class. + // Outputs: + // none. + // Return: + // none. + + if(this==&source) return *this; + Error("operator=","You are not allowed to make a copy of the AliITS"); + exit(1); + return *this; //fake return } -//_____________________________________________________________________________ -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 - - // - if(fDtype) { - fDtype->Delete(); - delete fDtype; - } - delete [] fNdtype; - if (fCtype) { - fCtype->Delete(); - delete fCtype; - } - delete [] fNctype; - // - - if (fDetTypes) { - fDetTypes->Delete(); - delete fDetTypes; - } - - if (fTreeC) delete fTreeC; - - if (fITSgeom) delete fITSgeom; - +//______________________________________________________________________ +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::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; -//___________________________________________ -AliITSDetType* AliITS::DetType(Int_t id) -{ - //return pointer to id detector type - return ((AliITSDetType*) (*fDetTypes)[id]); - + SetDefaults(); + // Array of TStrings + if(gMC) for(i=0;iVolId(fIdName[i]); + + WriteGeometry(); } -//___________________________________________ -void AliITS::SetClasses(Int_t id, const char *digit, const char *cluster) -{ - //set the digit and cluster classes to be used for the id detector type - ((AliITSDetType*) (*fDetTypes)[id])->ClassNames(digit,cluster); -} -//___________________________________________ -void AliITS::SetResponseModel(Int_t id, AliITSresponse *response) -{ - //set the response model for the id detector type +//______________________________________________________________________ +void AliITS::WriteGeometry(){ + + //Writes ITS geomtry on gAlice - ((AliITSDetType*) (*fDetTypes)[id])->ResponseModel(response); + if(!fLoader) MakeLoader(AliConfig::GetDefaultEventFolderName()); + AliRunLoader* rl = fLoader->GetRunLoader(); + rl->CdGAFile(); + AliITSgeom* geom = GetITSgeom(); + geom->Write(); } -//___________________________________________ -void AliITS::SetSegmentationModel(Int_t id, AliITSsegmentation *seg) -{ - //set the segmentation model for the id detector type - - ((AliITSDetType*) (*fDetTypes)[id])->SegmentationModel(seg); - -} +//______________________________________________________________________ +void AliITS::SetDefaults(){ + // sets the default segmentation, response, digit and raw cluster classes. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + AliInfoClass("Setting Defaults"); + + if(!fDetTypeSim) { + Error("SetDefaults()","fDetTypeSim is 0!"); + return; + } -//___________________________________________ -void AliITS::SetSimulationModel(Int_t id, AliITSsimulation *sim) -{ - //set the simulation model for the id detector type + fDetTypeSim->SetDefaults(); - ((AliITSDetType*) (*fDetTypes)[id])->SimulationModel(sim); } -//___________________________________________ -void AliITS::SetReconstructionModel(Int_t id, AliITSClusterFinder *reconst) -{ - //set the cluster finder model for the id detector type - - ((AliITSDetType*) (*fDetTypes)[id])->ReconstructionModel(reconst); +//______________________________________________________________________ +void AliITS::SetDefaultSimulation(){ + // sets the default simulation. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + + if(!fDetTypeSim) { + Error("SetDefaultSimulation()","fDetTypeSim is 0!"); + return; + } -} + fDetTypeSim->SetDefaultSimulation(); -//_____________________________________________________________________________ -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); } -//_____________________________________________________________________________ -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::AddSimDigit(Int_t id, AliITSdigit *d) -{ - - // add a simulated digit - - TClonesArray &ldigits = *((TClonesArray*)(*fDtype)[id]); - - 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; +//______________________________________________________________________ +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::AddSimDigit(Int_t id,Float_t phys,Int_t *digits,Int_t *tracks,Int_t *hits,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,hits); - break; - case 1: - new(ldigits[fNdtype[id]++]) AliITSdigitSDD(phys,digits,tracks,hits,charges); - break; - case 2: - new(ldigits[fNdtype[id]++]) AliITSdigitSSD(digits,tracks,hits); - break; - } - -} + 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); -//_____________________________________________________________________________ -void AliITS::AddCluster(Int_t id, AliITSRawCluster *c) -{ - - // add a cluster to the list - - TClonesArray &lcl = *((TClonesArray*)(*fCtype)[id]); - - 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; - } } +//___________________________________________________________________ +void AliITS::MakeBranchS(const char* fl){ + // 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. -//_____________________________________________________________________________ -void AliITS::AddRecPoint(const AliITSRecPoint &r) -{ - // - // Add a reconstructed space point to the list - // - TClonesArray &lrecp = *fRecPoints; - new(lrecp[fNRecPoints++]) AliITSRecPoint(r); -} - - -//____________________________________________ -void AliITS::ResetDigits() -{ - // - // Reset number of digits and the digits array for the ITS detector - // + + if(!fDetTypeSim){ + Error("MakeBranchS","fDetTypeSim is 0!"); + } + Int_t buffersize = 4000; + char branchname[30]; - if (!fDtype) return; + // only one branch for SDigits. + sprintf(branchname,"%s",GetName()); - Int_t i; - for (i=0;iClear(); - if (fNdtype) fNdtype[i]=0; - } + if(fLoader->TreeS()){ + if(fDetTypeSim->GetSDigits()==0x0) fDetTypeSim->SetSDigits(new TClonesArray("AliITSpListItem",1000)); + TClonesArray* sdig = (TClonesArray*)fDetTypeSim->GetSDigits(); + MakeBranchInTree(fLoader->TreeS(),branchname,&sdig,buffersize,fl); + } } +//______________________________________________________________________ +void AliITS::MakeBranchD(const char* file){ -//____________________________________________ -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; + //Make branch for digits + if(!fDetTypeSim) { + Warning("MakeBranchD","fDetTypeSim is 0!"); + return; + } + fDetTypeSim->SetLoader(fLoader); + MakeBranchInTreeD(fLoader->TreeD(),file); } +//___________________________________________________________________ +void AliITS:: MakeBranchInTreeD(TTree* treeD, const char* file){ + // Creates Tree branches for the ITS. -//____________________________________________ -void AliITS::ResetClusters() -{ - // - // Reset number of clusters and the clusters array for ITS - // + if(!fDetTypeSim){ + Error("MakeBranchS","fDetTypeSim is 0!"); + } + fDetTypeSim->SetLoader(fLoader); - Int_t i; - for (i=0;iClear(); - if (fNctype) fNctype[i]=0; + 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::SetTreeAddress(){ + // Set branch address for the Trees. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + + if(!fDetTypeSim) { + Error("SetTreeAddress","fDetTypeSim is 0!"); + return; + } -//____________________________________________ -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; - -} - - -//____________________________________________ -void AliITS::ResetRecPoints() -{ - // - // Reset number of rec points and the rec points array - // - if (fRecPoints) fRecPoints->Clear(); - fNRecPoints = 0; - -} - -//_____________________________________________________________________________ -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; -} + fDetTypeSim->SetLoader(fLoader); -//_____________________________________________________________________________ -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. - // + 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()); - SetDefaults(); - Int_t i; - cout << endl; - for(i=0;i<30;i++) cout << "*";cout << " ITS_INIT "; - for(i=0;i<30;i++) cout << "*";cout << endl; - for(i=0;iVolId(fIdName[i]); - // - for(i=0;i<70;i++) cout << "*"; - cout << endl; } - -//_____________________________________________________________________________ -void AliITS::SetDefaults() -{ - // sets the default segmentation, response, digit and raw cluster classes - - cout << "AliITS::SetDefaults" << endl; - - 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 *kData0=resp0->DataType(); - if (strstr(kData0,"real")) { - iDetType->ClassNames("AliITSdigit","AliITSRawClusterSPD"); - } else iDetType->ClassNames("AliITSdigitSPD","AliITSRawClusterSPD"); - - // SDD // - AliITSresponseSDD *resp1=new AliITSresponseSDD(); - AliITSsegmentationSDD *seg1=new AliITSsegmentationSDD(fITSgeom,resp1); - iDetType=DetType(1); - //printf("SetDefaults: iDetType %p\n",iDetType); - if (!iDetType->GetSegmentationModel()) SetSegmentationModel(1,seg1); - //printf("SetDefaults: segm %p\n",iDetType->GetSegmentationModel()); - if (!iDetType->GetResponseModel()) SetResponseModel(1,resp1); - //printf("SetDefaults: resp %p\n",iDetType->GetResponseModel()); - const char *kData1=resp1->DataType(); - const char *kopt=resp1->ZeroSuppOption(); - if ((!strstr(kopt,"2D")) && (!strstr(kopt,"1D")) || strstr(kData1,"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); - const char *kData2=resp2->DataType(); - if (strstr(kData2,"real")) { - iDetType->ClassNames("AliITSdigit","AliITSRawClusterSSD"); - } else iDetType->ClassNames("AliITSdigitSSD","AliITSRawClusterSSD"); - - if (fgkNTYPES>3) { - Warning("SetDefaults","Only the three basic detector types are initialised!"); - } +//______________________________________________________________________ +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::SetDefaultSimulation() -{ - // to be written -} -//_____________________________________________________________________________ -void AliITS::SetDefaultClusterFinders() -{ - // to be written +//______________________________________________________________________ +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. + 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::MakeTreeC(Option_t *option) -{ - // create a separate tree to store the clusters - - cout << "AliITS::MakeTreeC" << endl; - - char *optC = strstr(option,"C"); - if (optC && !fTreeC) fTreeC = new TTree("TC","Clusters in ITS"); - else return; - - 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); - cout << "Making Branch " << branchname; - cout << " for Clusters of detector type " << i+1 << endl; - } +//______________________________________________________________________ +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::GetTreeC(Int_t event) -{ - - cout << "AliITS::GetTreeC" << endl; - - // get the clusters tree for this event and set the branch address - char treeName[20]; - char branchname[30]; - - char *det[3] = {"SPD","SDD","SSD"}; - - ResetClusters(); - if (fTreeC) { - delete fTreeC; +//______________________________________________________________________ +void 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; } - sprintf(treeName,"TreeC%d",event); - fTreeC = (TTree*)gDirectory->Get(treeName); - + Int_t nl,indexMAX,index; - TBranch *branch; - if (fTreeC) { - Int_t i; - for (i=0; iGetBranch(branchname); - if (branch) branch->SetAddress(&((*fCtype)[i])); - } - } - } else { - Error("AliITS::GetTreeC", - "cannot find Clusters Tree for event:%d\n",event); - } + if(size<=0){ // default to using data stored in AliITSgeom + if(fDetTypeSim->GetITSgeom()==0) { + Error("InitModules","fITSgeom not defined"); + return; + } // 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 } -//_____________________________________________________________________________ -void AliITS::MakeBranch(Option_t* option){ - // - // Creates Tree branches for the ITS. - // - - - Int_t buffersize = 4000; - char branchname[30]; - sprintf(branchname,"%s",GetName()); - - AliDetector::MakeBranch(option); - - -// one branch for digits per type of detector +//______________________________________________________________________ +void AliITS::Hits2SDigits(){ + // Standard Hits to summable Digits function. + // Inputs: + // none. + // Outputs: + // none. - char *det[3] = {"SPD","SDD","SSD"}; - - char digclass[40]; - char clclass[40]; - - Int_t i; - for (i=0; iGetClassNames(digclass,clclass); - //printf("i, digclass, recclass %d %s %s\n",i,digclass,clclass); - // digits - (*fDtype)[i] = new TClonesArray(digclass,10000); - // clusters - (*fCtype)[i] = new TClonesArray(clclass,10000); - } - - - for (i=0; iTreeD()) { - gAlice->TreeD()->Branch(branchname,&((*fDtype)[i]), buffersize); - cout << "Making Branch " << branchname; - cout << " for digits of type "<< i+1 << endl; - } - } - // only one branch for rec points for all detector types - sprintf(branchname,"%sRecPoints",GetName()); - - fRecPoints=new TClonesArray("AliITSRecPoint",10000); - - if (fRecPoints && gAlice->TreeR()) { - gAlice->TreeR()->Branch(branchname,&fRecPoints, buffersize); - cout << "Making Branch " << branchname; - cout << " for reconstructed space points" << endl; - } + if(!fDetTypeSim) { + Error("Hits2SDigits","fDetTypeSim is null!"); + return; + + } + + SetDefaults(); + fLoader->LoadHits("read"); + fLoader->LoadSDigits("recreate"); + AliRunLoader* rl = fLoader->GetRunLoader(); + fDetTypeSim->SetLoader(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(){ + //Conversion from hits to digits + if(!fDetTypeSim) { + Error("Hits2SDigits","fDetTypeSim is 0!"); + return; + } + + fDetTypeSim->SetLoader(fLoader); + SetDefaults(); + 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::SetTreeAddress() -{ +//______________________________________________________________________ +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(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 - // Set branch address for the Trees. + if(!fDetTypeSim) { + Error("Hits2SDigits","fDetTypeSim is 0!"); + return; + } + + fDetTypeSim->SetLoader(fLoader); + SetDefaults(); + + HitsToPreDigits(fLoader->GetRunLoader()->GetEventNumber(), + 0,-1," ",fOpt," "); +} - char branchname[30]; - AliDetector::SetTreeAddress(); +//______________________________________________________________________ +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. - 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])); - } - } + + if(!fDetTypeSim) { + Error("HitsToPreDigits","fDetTypeSim is null!"); + return; } + fDetTypeSim->SetLoader(fLoader); - - if (treeR) { - sprintf(branchname,"%sRecPoints",GetName()); - if (fRecPoints) { - branch = treeR->GetBranch(branchname); - if (branch) branch->SetAddress(&fRecPoints); - } + 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); -} + 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(); -//____________________________________________________________________________ -void AliITS::InitModules(Int_t size,Int_t &nmodules){ + + fLoader->TreeS()->GetEntries(); + fLoader->TreeS()->AutoSave(); + fLoader->WriteSDigits("OVERWRITE"); + // reset tree + fLoader->TreeS()->Reset(); +} - //initialize the modules array +//_____________________________________________________________________ +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(); + + AliITSLoader *pITSloader = (AliITSLoader*)fLoader; + + 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; + + TTree *lTR = pITSloader->TreeR(); + TBranch* branch = lTR->GetBranch("ITSRecPointsF"); + TClonesArray* ptarray = new TClonesArray("AliITSRecPoint",1000); + if(branch) branch->SetAddress(ptarray); + else lTR->Branch("ITSRecPointsF",&ptarray); + - if(fITSmodules){ - fITSmodules->Delete(); - delete fITSmodules; + //m.b. : this change is nothing but a nice way to make sure + //the CPU goes up ! + AliDebug(1,Form("N mod = %d",geom->GetIndexMax())); + 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 *clusters=new TClonesArray("AliITSclusterV2",1000); + TBranch *branch=cTree->GetBranch("Clusters"); + if (!branch) cTree->Branch("Clusters",&clusters); + else branch->SetAddress(&clusters); + + TClonesArray *points = new TClonesArray("AliITSRecPoint",1000); + + AliITSsimulationFastPoints sim; + Int_t ncl=0; + for (Int_t m=0; mGetEntriesFast(); + cTree->Fill(); + clusters->Clear(); + points->Clear(); } - Int_t nl,indexMAX,index; + Info("Hits2Clusters","Number of found fast clusters : %d",ncl); - if(size<=0){ // default to using data stored in AliITSgeom - if(fITSgeom==0) { - Error("AliITS::InitModules", - "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); - for(index=0;indexAddAt( new AliITSmodule(index),index); - } // end for index - }else{ - fITSmodules = new TObjArray(size); - for(index=0;indexAddAt( new AliITSmodule(index),index); - } + //cTree->Write(); - nmodules = size; - } // end i size<=0 + delete clusters; + delete points; + return 0; } +//_____________________________________________________________________ +void AliITS::RecPoints2Clusters +(const TClonesArray *points, Int_t idx, TClonesArray *clusters) { + //------------------------------------------------------------ + // Conversion AliITSRecPoint -> AliITSclusterV2 for the ITS + // subdetector indexed by idx + //------------------------------------------------------------ + + if(!GetITSgeom()){ + Error("HitsToPreDigits","fGeom is null!"); + return; // need transformations to do digitization. + } + AliITSgeom *g=GetITSgeom(); + Int_t mmax = g->GetIndexMax(); + Float_t yshift[2200]; + Float_t zshift[2200]; + Int_t ndet[2200]; + for (Int_t m=0; mGetModuleId(m,lay,lad,det); + Float_t x,y,z; g->GetTrans(lay,lad,det,x,y,z); + Double_t rot[9]; g->GetRotMatrix(lay,lad,det,rot); + Double_t alpha=TMath::ATan2(rot[1],rot[0])+TMath::Pi(); + Double_t ca=TMath::Cos(alpha), sa=TMath::Sin(alpha); + yshift[m] = x*ca + y*sa; + zshift[m] = (Double_t)z; + ndet[m] = (lad-1)*g->GetNdetectors(lay) + (det-1); + } -//____________________________________________________________________________ -void AliITS::FillModules(Int_t evnt,Int_t bgrev,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 *addBgr = strstr(option,"Add"); - - - if (addBgr ) { - 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) { - Error("AliITS::FillModules", - "cannot find Hits Tree for event:%d\n",bgrev); + //SPD geometry + Float_t lastSPD1=g->GetModuleIndex(2,1,1)-1; + + TClonesArray &cl=*clusters; + Int_t ncl=points->GetEntriesFast(); + for (Int_t i=0; iUncheckedAt(i); + Float_t lp[5]; + lp[0]=-(-p->GetX()+yshift[idx]); if (idx<=lastSPD1) lp[0]*=-1; //SPD1 + lp[1]= -p->GetZ()+zshift[idx]; + lp[2]=p->GetSigmaX2(); + lp[3]=p->GetSigmaZ2(); + lp[4]=p->GetQ()*36./23333.; //electrons -> ADC + Int_t lab[4]; + lab[0]=p->GetLabel(0); lab[1]=p->GetLabel(1); lab[2]=p->GetLabel(2); + lab[3]=ndet[idx]; + CheckLabels(lab); + Int_t dummy[3]={0,0,0}; + new (cl[i]) AliITSclusterV2(lab,lp, dummy); + } +} +//_____________________________________________________________________ +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; } - // Set branch addresses - TBranch *branch; - char branchname[20]; - sprintf(branchname,"%s",GetName()); - if (trH1 && fHits2) { - branch = trH1->GetBranch(branchname); - if (branch) branch->SetAddress(&fHits2); + if (part->GetStatusCode()>0) { + continue; } + lab[i]=m; + } + } + } + +} - // test - //Int_t ntracks1 =(Int_t)TrH1->GetEntries(); - //printf("background - ntracks1 - %d\n",ntracks1); - } +//______________________________________________________________________ +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(fLoader); + SetDefaults(); + fDetTypeSim->SDigitsToDigits(opt,(Char_t*)GetName()); - Int_t npart = gAlice->GetEvent(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(); - //printf("nhits %d\n",nhits); - if (!nhits) continue; - for(h=0; hUncheckedAt(h); - itsHit->GetDetectorID(lay,lad,det); - // temporarily index=det-1 !!! - if(fITSgeom) index = fITSgeom->GetModuleIndex(lay,lad,det); - else index=det-1; - // - mod = this->GetModule(index); - mod->AddHit(itsHit,t,h); - } // end loop over hits - } // end loop over tracks +//______________________________________________________________________ +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(); - // open the file with background - - if (addBgr ) { - Int_t track,i; - 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); - // temporarily index=det-1 !!! - if(fITSgeom) index = fITSgeom->GetModuleIndex(lay,lad,det); - else index=det-1; - // - mod = this->GetModule(index); - 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(); - fileAli->cd(); - - } // end if add - - //gObjectTable->Print(); } +//______________________________________________________________________ +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); - -//____________________________________________________________________________ -void AliITS::HitsToDigits(Int_t evNumber,Int_t bgrev,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 ! - - // the condition below will disappear when the geom class will be - // initialised for all versions - for the moment it is only for v5 ! - // 7 is the SDD beam test version - Int_t ver = this->IsVersion(); - if(ver!=5 && ver!=7) return; - - 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,nmodules,option,filename); - - //TBranch *branch; - AliITSsimulation* sim; - //TObjArray *branches=gAlice->TreeD()->GetListOfBranches(); - AliITSgeom *geom = GetITSgeom(); - - Int_t id,module; - Int_t first,last; - for (id=0;idUncheckedAt(id); - AliITSDetType *iDetType=DetType(id); - sim = (AliITSsimulation*)iDetType->GetSimulationModel(); - if (!sim) { - Error("HitsToDigits","The simulation class was not instantiated!"); - exit(1); - // or SetDefaultSimulation(); - } - if(geom) { - first = geom->GetStartDet(id); - last = geom->GetLastDet(id); - } else first=last=0; - cout << "det type " << id << " first, last "<< first << last << endl; - 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(); - cout << "nentries in TreeD" << nentries << endl; - - char hname[30]; - sprintf(hname,"TreeD%d",evNumber); - gAlice->TreeD()->Write(hname); - // reset tree - gAlice->TreeD()->Reset(); +} +//______________________________________________________________________ +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); + +} +//______________________________________________________________________ +void AliITS::AddRealDigit(Int_t branch, Int_t *digits){ + // Add a real digit - as coming from data. + // Inputs: + // Int_t id Detector type number. + // Int_t *digits Integer array containing the digits info. See + // AliITSdigit.h + // Outputs: + // none. + // Return: + // none. + + if(!fDetTypeSim) { + Error("AddRealDigit","fDetTypeSim is 0!"); + return; + } + fDetTypeSim->AddRealDigit(branch,digits); } +//______________________________________________________________________ +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::AddSimDigit(Int_t branch,Float_t phys,Int_t *digits,Int_t *tracks, + Int_t *hits,Float_t *charges){ + // 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); -//____________________________________________________________________________ -void AliITS::DigitsToRecPoints(Int_t evNumber,Int_t lastentry,Option_t *opt) -{ - // cluster finding and reconstruction of space points - - // the condition below will disappear when the geom class will be - // initialised for all versions - for the moment it is only for v5 ! - // 7 is the SDD beam test version - Int_t ver = this->IsVersion(); - if(ver!=5 && ver!=7) return; - - 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 SetDefaultClusterFinders(); - } - TClonesArray *itsDigits = this->DigitsAddress(id); - - Int_t first,last; - if(geom) { - first = geom->GetStartDet(id); - last = geom->GetLastDet(id); - } else first=last=0; - //printf("first last %d %d\n",first,last); - for(module=first;module<=last;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)iTC->GetEntries(); - cout << " nentries ncentries " << nentries << ncentries << endl; - - 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(); } +//______________________________________________________________________ +void AliITS::Digits2Raw(){ + // convert digits of the current event to raw data + if(!fDetTypeSim) { + Error("Digits2Raw","fDetTypeSim is 0!"); + return; + } + fDetTypeSim->SetLoader(fLoader); + SetDefaults(); + fDetTypeSim->GetLoader()->LoadDigits(); + TTree* digits = fDetTypeSim->GetLoader()->TreeD(); + if (!digits) { + Error("Digits2Raw", "no digits tree"); + return; + } + fDetTypeSim->SetTreeAddressD(digits,(Char_t*)GetName()); + + 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")); + + //SILICON STRIP DETECTOR + Info("Digits2Raw", "Formatting raw data for SSD"); + rawWriter.RawDataSSD(digits->GetBranch("ITSDigitsSSD")); -//____________________________________________________________________________ -void AliITS::HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,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 ! + fLoader->UnloadDigits(); - // the condition below will disappear when the geom class will be - // initialised for all versions - for the moment it is only for v5 ! - Int_t ver = this->IsVersion(); - if(ver!=5) return; - - 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,nmodules,option,filename); - - - AliITSsimulation* sim; - AliITSgeom *geom = GetITSgeom(); - - TRandom *random=new TRandom[9]; - random[0].SetSeed(111); - random[1].SetSeed(222); - random[2].SetSeed(333); - random[3].SetSeed(444); - random[4].SetSeed(555); - random[5].SetSeed(666); - random[6].SetSeed(777); - random[7].SetSeed(888); - random[8].SetSeed(999); - - - Int_t id,module; - for (id=0;idGetSimulationModel(); - if (!sim) { - Error("HitsToFastPoints","The simulation class was not instantiated!"); - exit(1); - // or SetDefaultSimulation(); - } - 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,module,random); - gAlice->TreeR()->Fill(); - ResetRecPoints(); - } // loop over modules - } // loop over detector types - - - ClearModules(); - - //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(); - - delete [] random; + } -//____________________________________________________________________________ -void AliITS::Streamer(TBuffer &R__b){ - // Stream an object of class AliITS. - - Int_t i; - - if (R__b.IsReading()) { - Version_t R__v = R__b.ReadVersion(); - if (R__v) { - AliDetector::Streamer(R__b); - R__b >> fIdN; - R__b.ReadArray(fIdSens); - for(i=0;i> fITSgeom; - R__b >> fITSmodules; - R__b >> fEuclidOut; - R__b >> fMajorVersion; - R__b >> fMinorVersion; - R__b >> fDetTypes; - R__b >> fDtype; - delete []fNdtype; - fNdtype = new Int_t[fgkNTYPES]; - R__b.ReadFastArray(fNdtype,fgkNTYPES); - R__b >> fCtype; - delete []fNctype; - fNctype = new Int_t[fgkNTYPES]; - R__b.ReadFastArray(fNctype,fgkNTYPES); - R__b >> fRecPoints; - R__b >> fNRecPoints; - R__b >> fTreeC; - } // end if R__v - } else { // writing - R__b.WriteVersion(AliITS::IsA()); - AliDetector::Streamer(R__b); - R__b << fIdN; - R__b.WriteArray(fIdSens,fIdN); - for(i=0;i