X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITS.cxx;h=368ffa985dae8cd4ef6bd3c7ccac459743a73b57;hb=4bb132778bf14da06034f4bfb00d0cef054c5989;hp=9169e5b128f3a46258c11ca543b95d073fc65d41;hpb=d5da1ecf57110ceb80a56b59f11311e579e6b494;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITS.cxx b/ITS/AliITS.cxx index 9169e5b128f..368ffa985da 100644 --- a/ITS/AliITS.cxx +++ b/ITS/AliITS.cxx @@ -13,37 +13,8 @@ * provided "as is" without express or implied warranty. * **************************************************************************/ -/* -$Log$ -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) -#include + +#include +#include + #include -#include -#include -#include -#include +#include #include -#include +#include +#include +#include +#include #include +#include +#include +#include - - -#include "AliRun.h" +#include "AliConfig.h" +#include "AliHeader.h" #include "AliITS.h" -#include "AliITSMap.h" +#include "AliITSClusterFinderSDD.h" +#include "AliITSClusterFinderSPD.h" +#include "AliITSClusterFinderSSD.h" #include "AliITSDetType.h" -#include "AliITSClusterFinder.h" -#include "AliITSsimulation.h" -#include "AliITSsegmentationSPD.h" +#include "AliITSLoader.h" +#include "AliITSRawClusterSPD.h" +#include "AliITSRawClusterSDD.h" +#include "AliITSRawClusterSSD.h" +#include "AliITSRecPoint.h" +#include "AliITSdigitSPD.h" +#include "AliITSdigitSDD.h" +#include "AliITSdigitSSD.h" +#include "AliITSgeom.h" +#include "AliITShit.h" +#include "AliITSmodule.h" +#include "AliITSpList.h" +#include "AliITSresponseSDD.h" #include "AliITSresponseSPD.h" +#include "AliITSresponseSSD.h" #include "AliITSsegmentationSDD.h" -#include "AliITSresponseSDD.h" +#include "AliITSsegmentationSPD.h" #include "AliITSsegmentationSSD.h" -#include "AliITSresponseSSD.h" -#include "AliITShit.h" -#include "AliITSgeom.h" -#include "AliITSdigit.h" -#include "AliITSmodule.h" -#include "AliITSRecPoint.h" -#include "AliITSRawCluster.h" - -const Int_t AliITS::fgkNTYPES=3; +#include "AliITSsimulationSDD.h" +#include "AliITSsimulationSPD.h" +#include "AliITSsimulationSSD.h" +#include "AliMC.h" +#include "AliITSDigitizer.h" +#include "AliITSclustererV2.h" +#include "AliITStrackerV2.h" +#include "AliITSpidESD.h" +#include "AliV0vertexer.h" +#include "AliCascadeVertexer.h" +#include "AliESD.h" ClassImp(AliITS) - -//_____________________________________________________________________________ + +//______________________________________________________________________ AliITS::AliITS() : AliDetector() { - // - // Default initialiser 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 - // 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; -} + // 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 + // is also called. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // Blank ITS class. + + fIshunt = 0; // not zeroed in AliDetector. + + // AliITS variables. + fEuclidOut = 0; + fITSgeom = 0; + fITSmodules = 0; + fOpt = "All"; +// SetDetectors(); // default to fOpt="All". This variable not written out. + + fIdN = 0; + fIdName = 0; + fIdSens = 0; + + fNDetTypes = kNTYPES; + fDetTypes = 0; + + fSDigits = 0; + fNSDigits = 0; + + fNdtype = 0; + fDtype = 0; + + fCtype = 0; + fNctype = 0; + + fRecPoints = 0; + fNRecPoints = 0; -//_____________________________________________________________________________ + 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; - - fITSmodules = 0; - - fIshunt = 0; - fEuclidOut = 0; - fIdN = 0; - fIdName = 0; - fIdSens = 0; - - fDetTypes = new TObjArray(fgkNTYPES); + // The standard Constructor for the ITS class. In addition to + // creating the AliITS class, it allocates memory for the TClonesArrays + // fHits, fSDigits, fDigits, fITSpoints, and the TObjArray of fCtype + // (clusters). 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. + // Inputs: + // const char *name Detector name. Should always be "ITS" + // const char *title Detector title. + // Outputs: + // none. + // Return: + // An ITS class. + + fIshunt = 0; // not zeroed in AliDetector + fHits = new TClonesArray("AliITShit", 1560);//not done in AliDetector + gAlice->GetMCApp()->AddHitList(fHits); // Not done in AliDetector. + + fEuclidOut = 0; + fITSgeom = 0; + fITSmodules = 0; + SetDetectors(); // default to fOpt="All". This variable not written out. + + fIdN = 0; + fIdName = 0; + fIdSens = 0; + + fNDetTypes = kNTYPES; + fDetTypes = new TObjArray(fNDetTypes); + + fSDigits = new TClonesArray("AliITSpListItem",1000); + fNSDigits = 0; + + fNdtype = new Int_t[fNDetTypes]; + fDtype = new TObjArray(fNDetTypes); + + fCtype = new TObjArray(fNDetTypes); + fNctype = new Int_t[fNDetTypes]; + + fRecPoints = new TClonesArray("AliITSRecPoint",1000); + fNRecPoints = 0; - Int_t i; - for(i=0;iAddAt(new AliITSDetType(),i); + fNdtype[i] = 0; + fNctype[i] = 0; + } // end for i - SetMarkerColor(kRed); + SetMarkerColor(kRed); - fITSgeom=0; } -//___________________________________________________________________________ -AliITS::AliITS(AliITS &source){ - // copy constructor - if(this==&source) return; - Error("AliITS::Copy constructor", - "You are not allowed to make a copy of the AliITS"); - 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 the fHits, fDigits, + // fSDigits, fCtype, fITSmodules, fITSgeom, fRecPoints, fIdSens, fIdName, + // fITSpoints, fDetType and it's contents. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + + if (fHits) { + fHits->Delete(); + delete fHits; + fHits=0; + } + if (fSDigits) { + fSDigits->Delete(); + delete fSDigits; + fSDigits=0; + } + if (fDigits) { + fDigits->Delete(); + delete fDigits; + fDigits=0; + } + if (fRecPoints) { + fRecPoints->Delete(); + delete fRecPoints; + fRecPoints=0; + } + delete[] fIdName; // Array of TStrings + delete[] fIdSens; + if(fITSmodules) { + this->ClearModules(); + delete fITSmodules; + fITSmodules = 0; + }// end if fITSmodules!=0 + + if(fDtype) { + fDtype->Delete(); + delete fDtype; + } // end if fDtype + delete [] fNdtype; + if (fCtype) { + fCtype->Delete(); + delete fCtype; + fCtype = 0; + } // end if fCtype + delete [] fNctype; + + if (fDetTypes) { + fDetTypes->Delete(); + delete fDetTypes; + fDetTypes = 0; + } // end if fDetTypes + + + if (fITSgeom) delete fITSgeom; +} +//______________________________________________________________________ +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); } -//____________________________________________________________________________ +//______________________________________________________________________ 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 + // 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 } -//____________________________________________________________________________ -void AliITS::ClearModules(){ - //clear the modules TObjArray - - if(fITSmodules) fITSmodules->Delete(); - +//______________________________________________________________________ +Int_t AliITS::DistancetoPrimitive(Int_t,Int_t) const{ + // 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. + // Inputs: + // Int_t Dummy screen coordinate. + // Int_t Dummy screen coordinate. + // Outputs: + // none. + // Return: + // Int_t Dummy = 9999 distance to ITS. + + return 9999; } -//_____________________________________________________________________________ -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; +//______________________________________________________________________ +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; + SetDefaults(); + // Array of TStrings + for(i=0;iVolId(fIdName[i]); } - -//___________________________________________ -AliITSDetType* AliITS::DetType(Int_t id) -{ - //return pointer to id detector type - return ((AliITSDetType*) (*fDetTypes)[id]); - +//______________________________________________________________________ +void AliITS::SetDefaults(){ + // sets the default segmentation, response, digit and raw cluster classes. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + + if(fDebug) printf("%s: SetDefaults\n",ClassName()); + + AliITSDetType *iDetType; + + //SPD + iDetType=DetType(0); + if (!iDetType->GetSegmentationModel()) { + AliITSsegmentationSPD *seg0=new AliITSsegmentationSPD(fITSgeom); + SetSegmentationModel(0,seg0); + } // end if + if (!iDetType->GetResponseModel()) { + SetResponseModel(0,new AliITSresponseSPD()); + } // end if + // set digit and raw cluster classes to be used + + const char *kData0=(iDetType->GetResponseModel())->DataType(); + if (strstr(kData0,"real")) { + iDetType->ClassNames("AliITSdigit","AliITSRawClusterSPD"); + } else iDetType->ClassNames("AliITSdigitSPD","AliITSRawClusterSPD"); + + // SDD + iDetType=DetType(1); + if (!iDetType->GetResponseModel()) { + SetResponseModel(1,new AliITSresponseSDD("simulated")); + } // end if + AliITSresponse *resp1=iDetType->GetResponseModel(); + if (!iDetType->GetSegmentationModel()) { + AliITSsegmentationSDD *seg1=new AliITSsegmentationSDD(fITSgeom,resp1); + SetSegmentationModel(1,seg1); + } // end if + const char *kData1=(iDetType->GetResponseModel())->DataType(); + const char *kopt=iDetType->GetResponseModel()->ZeroSuppOption(); + if((!strstr(kopt,"2D"))&&(!strstr(kopt,"1D")) || strstr(kData1,"real") ){ + iDetType->ClassNames("AliITSdigit","AliITSRawClusterSDD"); + } else iDetType->ClassNames("AliITSdigitSDD","AliITSRawClusterSDD"); + + // SSD + iDetType=DetType(2); + if (!iDetType->GetSegmentationModel()) { + AliITSsegmentationSSD *seg2=new AliITSsegmentationSSD(fITSgeom); + SetSegmentationModel(2,seg2); + } // end if + if (!iDetType->GetResponseModel()) { + SetResponseModel(2,new AliITSresponseSSD("simulated")); + } // end if + const char *kData2=(iDetType->GetResponseModel())->DataType(); + if (strstr(kData2,"real")) { + iDetType->ClassNames("AliITSdigit","AliITSRawClusterSSD"); + } else iDetType->ClassNames("AliITSdigitSSD","AliITSRawClusterSSD"); + + if (kNTYPES>3) { + Warning("SetDefaults", + "Only the three basic detector types are initialized!"); + } // end if } -//___________________________________________ -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::SetDefaultSimulation(){ + // sets the default simulation. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + + AliITSDetType *iDetType; + AliITSsimulation *sim; + iDetType=DetType(0); + sim = iDetType->GetSimulationModel(); + if (!sim) { + AliITSsegmentation *seg0= + (AliITSsegmentation*)iDetType->GetSegmentationModel(); + AliITSresponse *res0 = (AliITSresponse*)iDetType->GetResponseModel(); + AliITSsimulationSPD *sim0=new AliITSsimulationSPD(seg0,res0); + SetSimulationModel(0,sim0); + }else{ // simulation exists, make sure it is set up properly. + ((AliITSsimulationSPD*)sim)->Init( + (AliITSsegmentationSPD*) iDetType->GetSegmentationModel(), + (AliITSresponseSPD*) iDetType->GetResponseModel()); +// if(sim->GetResponseModel()==0) sim->SetResponseModel( +// (AliITSresponse*)iDetType->GetResponseModel()); +// if(sim->GetSegmentationModel()==0) sim->SetSegmentationModel( +// (AliITSsegmentation*)iDetType->GetSegmentationModel()); + } // end if + iDetType=DetType(1); + sim = iDetType->GetSimulationModel(); + if (!sim) { + AliITSsegmentation *seg1= + (AliITSsegmentation*)iDetType->GetSegmentationModel(); + AliITSresponse *res1 = (AliITSresponse*)iDetType->GetResponseModel(); + AliITSsimulationSDD *sim1=new AliITSsimulationSDD(seg1,res1); + SetSimulationModel(1,sim1); + }else{ // simulation exists, make sure it is set up properly. + ((AliITSsimulationSDD*)sim)->Init( + (AliITSsegmentationSDD*) iDetType->GetSegmentationModel(), + (AliITSresponseSDD*) iDetType->GetResponseModel()); +// if(sim->GetResponseModel()==0) sim->SetResponseModel( +// (AliITSresponse*)iDetType->GetResponseModel()); +// if(sim->GetSegmentationModel()==0) sim->SetSegmentationModel( +// (AliITSsegmentation*)iDetType->GetSegmentationModel()); + } //end if + iDetType=DetType(2); + sim = iDetType->GetSimulationModel(); + if (!sim) { + AliITSsegmentation *seg2= + (AliITSsegmentation*)iDetType->GetSegmentationModel(); + AliITSresponse *res2 = (AliITSresponse*)iDetType->GetResponseModel(); + AliITSsimulationSSD *sim2=new AliITSsimulationSSD(seg2,res2); + SetSimulationModel(2,sim2); + }else{ // simulation exists, make sure it is set up properly. + ((AliITSsimulationSSD*)sim)->Init( + (AliITSsegmentationSSD*) iDetType->GetSegmentationModel(), + (AliITSresponseSSD*) iDetType->GetResponseModel()); +// if(sim->GetResponseModel()==0) sim->SetResponseModel( +// (AliITSresponse*)iDetType->GetResponseModel()); +// if(sim->GetSegmentationModel()==0) sim->SetSegmentationModel( +// (AliITSsegmentation*)iDetType->GetSegmentationModel()); + } // end if } -//___________________________________________ -void AliITS::SetResponseModel(Int_t id, AliITSresponse *response) -{ - //set the response model for the id detector type - - ((AliITSDetType*) (*fDetTypes)[id])->ResponseModel(response); - +//______________________________________________________________________ +void AliITS::SetDefaultClusterFinders(){ + // Sets the default cluster finders. Used in finding RecPoints. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + + MakeTreeC(); + AliITSDetType *iDetType; + + // SPD + iDetType=DetType(0); + if (!iDetType->GetReconstructionModel()) { + AliITSsegmentation *seg0 = + (AliITSsegmentation*)iDetType->GetSegmentationModel(); + TClonesArray *dig0=DigitsAddress(0); + TClonesArray *recp0=ClustersAddress(0); + AliITSClusterFinderSPD *rec0 = new AliITSClusterFinderSPD(seg0,dig0, + recp0); + SetReconstructionModel(0,rec0); + } // end if + + // SDD + iDetType=DetType(1); + if (!iDetType->GetReconstructionModel()) { + AliITSsegmentation *seg1 = + (AliITSsegmentation*)iDetType->GetSegmentationModel(); + AliITSresponse *res1 = (AliITSresponse*)iDetType->GetResponseModel(); + TClonesArray *dig1=DigitsAddress(1); + TClonesArray *recp1=ClustersAddress(1); + AliITSClusterFinderSDD *rec1 = + new AliITSClusterFinderSDD(seg1,res1,dig1,recp1); + SetReconstructionModel(1,rec1); + } // end if + + // SSD + iDetType=DetType(2); + if (!iDetType->GetReconstructionModel()) { + AliITSsegmentation *seg2= + (AliITSsegmentation*)iDetType->GetSegmentationModel(); + TClonesArray *dig2=DigitsAddress(2); + AliITSClusterFinderSSD *rec2= new AliITSClusterFinderSSD(seg2,dig2); + SetReconstructionModel(2,rec2); + } // end if } +//______________________________________________________________________ +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. + Bool_t cH = (strstr(option,"H")!=0); + Bool_t cS = (strstr(option,"S")!=0); + Bool_t cD = (strstr(option,"D")!=0); + Bool_t cR = (strstr(option,"R")!=0); + Bool_t cRF = (strstr(option,"RF")!=0); + + if(cRF)cR = kFALSE; + if(cH && (fHits == 0x0)) fHits = new TClonesArray("AliITShit", 1560); + + AliDetector::MakeBranch(option); -//___________________________________________ -void AliITS::SetSegmentationModel(Int_t id, AliITSsegmentation *seg) -{ - //set the segmentation model for the id detector type - - ((AliITSDetType*) (*fDetTypes)[id])->SegmentationModel(seg); - + if(cS) MakeBranchS(0); + if(cD) MakeBranchD(0); + if(cR) MakeBranchR(0); + if(cRF) MakeBranchRF(0); } +//______________________________________________________________________ +void AliITS::SetTreeAddress(){ + // Set branch address for the Trees. + // Inputs: + // none. + // Outputs: + // none. + // Return: + // none. + TTree *treeS = fLoader->TreeS(); + TTree *treeD = fLoader->TreeD(); + TTree *treeR = fLoader->TreeR(); + if (fLoader->TreeH() && (fHits == 0x0)) fHits = new TClonesArray("AliITShit", 1560); + + AliDetector::SetTreeAddress(); -//___________________________________________ -void AliITS::SetSimulationModel(Int_t id, AliITSsimulation *sim) -{ - //set the simulation model for the id detector type - - ((AliITSDetType*) (*fDetTypes)[id])->SimulationModel(sim); - + SetTreeAddressS(treeS); + SetTreeAddressD(treeD); + SetTreeAddressR(treeR); } -//___________________________________________ -void AliITS::SetReconstructionModel(Int_t id, AliITSClusterFinder *reconst) -{ - //set the cluster finder model for the id detector type - - ((AliITSDetType*) (*fDetTypes)[id])->ReconstructionModel(reconst); - +//______________________________________________________________________ +AliITSDetType* AliITS::DetType(Int_t id){ + // Return pointer to id detector type. + // Inputs: + // Int_t id detector id number. + // Outputs: + // none. + // Return: + // returned, a pointer to a AliITSDetType. + + return ((AliITSDetType*) fDetTypes->At(id)); } - -//_____________________________________________________________________________ -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::SetResponseModel(Int_t id, AliITSresponse *response){ + // Set the response model for the id detector type. + // Inputs: + // Int_t id detector id number. + // AliITSresponse* a pointer containing an instance of AliITSresponse + // to be stored/owned b y AliITSDetType. + // Outputs: + // none. + // Return: + // none. + + ((AliITSDetType*) fDetTypes->At(id))->ResponseModel(response); } -//_____________________________________________________________________________ -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::SetSegmentationModel(Int_t id, AliITSsegmentation *seg){ + // Set the segmentation model for the id detector type. + // Inputs: + // Int_t id detector id number. + // AliITSsegmentation* a pointer containing an instance of + // AliITSsegmentation to be stored/owned b y + // AliITSDetType. + // Outputs: + // none. + // Return: + // none. + + ((AliITSDetType*) fDetTypes->At(id))->SegmentationModel(seg); } -//_____________________________________________________________________________ -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::SetSimulationModel(Int_t id, AliITSsimulation *sim){ + // Set the simulation model for the id detector type. + // Inputs: + // Int_t id detector id number. + // AliITSresponse* a pointer containing an instance of AliITSresponse + // to be stored/owned b y AliITSDetType. + // Outputs: + // none. + // Return: + // none. + + ((AliITSDetType*) fDetTypes->At(id))->SimulationModel(sim); } - -//_____________________________________________________________________________ -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; - } - +//______________________________________________________________________ +void AliITS::SetReconstructionModel(Int_t id, AliITSClusterFinder *reconst){ + // Set the cluster finder model for the id detector type. + // Inputs: + // Int_t id detector id number. + // AliITSClusterFinder* a pointer containing an instance of + // AliITSClusterFinder to be stored/owned b y + // AliITSDetType. + // Outputs: + // none. + // Return: + // none. + + ((AliITSDetType*) fDetTypes->At(id))->ReconstructionModel(reconst); +} +//______________________________________________________________________ +void AliITS::SetClasses(Int_t id, const char *digit, const char *cluster){ + // Set the digit and cluster classes name to be used for the id detector + // type. + // Inputs: + // Int_t id detector id number. + // const char *digit Digit class name for detector id. + // const char *cluster Cluster class name for detector id. + // Outputs: + // none. + // Return: + // none. + + ((AliITSDetType*) fDetTypes->At(id))->ClassNames(digit,cluster); +} +//______________________________________________________________________ +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::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 -//_____________________________________________________________________________ -void AliITS::AddCluster(Int_t id, AliITSRawCluster *c) -{ + Int_t nl,indexMAX,index; - // add a cluster to the list + if(size<=0){ // default to using data stored in AliITSgeom + if(fITSgeom==0) { + Error("InitModules","fITSgeom not defined"); + 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); + } // end for index - TClonesArray &lcl = *((TClonesArray*)(*fCtype)[id]); + nmodules = size; + } // end i size<=0 +} +//______________________________________________________________________ +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. + // 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. + static TTree *trH1; //Tree with background hits + static Bool_t first=kTRUE; + static TFile *file; + const char *addBgr = strstr(option,"Add"); - 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; - } + 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::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 (fITSgeom) { + index = fITSgeom->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::ClearModules(){ + // Clear the modules TObjArray. + // Inputs: + // none. + // Outputs: + // none. + if(fITSmodules) fITSmodules->Delete(); } +//______________________________________________________________________ +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. + // Outputs: + // none. + // Return: + // none. + Int_t buffersize = 4000; + char branchname[30]; + // only one branch for SDigits. + sprintf(branchname,"%s",GetName()); + -//_____________________________________________________________________________ -void AliITS::AddRecPoint(const AliITSRecPoint &r) -{ - // - // Add a reconstructed space point to the list - // - TClonesArray &lrecp = *fRecPoints; - new(lrecp[fNRecPoints++]) AliITSRecPoint(r); + if(fLoader->TreeS()){ + if (fSDigits == 0x0) fSDigits = new TClonesArray("AliITSpListItem",1000); + MakeBranchInTree(fLoader->TreeS(),branchname,&fSDigits,buffersize,fl); + } // end if } - - -//____________________________________________ -void AliITS::ResetDigits() -{ - // - // Reset number of digits and the digits array for the ITS detector - // +//______________________________________________________________________ +void AliITS::SetTreeAddressS(TTree *treeS){ + // Set branch address for the ITS summable digits Trees. + // Inputs: + // TTree *treeS Tree containing the SDigits. + // Outputs: + // none. + // Return: + // none. + char branchname[30]; - if (!fDtype) return; + if(!treeS) return; + if (fSDigits == 0x0) fSDigits = new TClonesArray("AliITSpListItem",1000); + TBranch *branch; + sprintf(branchname,"%s",GetName()); + branch = treeS->GetBranch(branchname); + if (branch) branch->SetAddress(&fSDigits); +} +//______________________________________________________________________ +void AliITS::MakeBranchInTreeD(TTree *treeD,const char *file){ + // Creates Tree branches for the ITS. + // Inputs: + // TTree *treeD Pointer to the Digits Tree. + // cont char *file File name where Digits branch is to be written + // to. If blank it write the SDigits to the same + // file in which the Hits were found. + // Outputs: + // none. + // Return: + // none. + Int_t buffersize = 4000; + char branchname[30]; + sprintf(branchname,"%s",GetName()); + // one branch for digits per type of detector + const char *det[3] = {"SPD","SDD","SSD"}; + char digclass[40]; + char clclass[40]; Int_t i; - for (i=0;iClear(); - if (fNdtype) fNdtype[i]=0; - } + for (i=0; iGetClassNames(digclass,clclass); + // digits + if (fDtype == 0x0) fDtype = new TObjArray(fNDetTypes); + if(!(fDtype->At(i))) fDtype->AddAt(new TClonesArray(digclass,1000),i); + else ResetDigits(i); + } // end for i + for (i=0; iClear(); - if (fNdtype) fNdtype[i]=0; + if(!treeD) return; + for (i=0; iGetClassNames(digclass,clclass); + // digits + if (fDtype == 0x0) fDtype = new TObjArray(fNDetTypes); + if(!(fDtype->At(i))) fDtype->AddAt(new TClonesArray(digclass,1000),i); + else ResetDigits(i); + if (kNTYPES==3) sprintf(branchname,"%sDigits%s",GetName(),det[i]); + else sprintf(branchname,"%sDigits%d",GetName(),i+1); + if (fDtype) { + branch = treeD->GetBranch(branchname); + if (branch) branch->SetAddress(&((*fDtype)[i])); + } // end if fDtype + } // end for i } - - -//____________________________________________ -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; +//______________________________________________________________________ +void AliITS::Hits2SDigits(){ + // Standard Hits to summable Digits function. + // Inputs: + // none. + // Outputs: + // none. + +// return; // Using Hits in place of the larger sDigits. + fLoader->LoadHits("read"); + fLoader->LoadSDigits("recreate"); + AliRunLoader* rl = fLoader->GetRunLoader(); + + 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(); + HitsToSDigits(iEvent,0,-1," ",fOpt," "); } - + + fLoader->UnloadHits(); + fLoader->UnloadSDigits(); } - -//____________________________________________ -void AliITS::ResetClusters(Int_t i) +//______________________________________________________________________ +void AliITS::Hits2PreDigits(){ + // Standard Hits to summable Digits function. + // Inputs: + // none. + // Outputs: + // none. + + AliHeader *header=fLoader->GetRunLoader()->GetHeader(); // Get event number from this file. + // 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. + HitsToPreDigits(header->GetEvent(),0,-1," ",fOpt," "); +} +//______________________________________________________________________ +AliDigitizer* AliITS::CreateDigitizer(AliRunDigitizer* manager) const { - // - // Reset number of clusters and the clusters array for this branch - // - if ((*fCtype)[i]) ((TClonesArray*)(*fCtype)[i])->Clear(); - if (fNctype) fNctype[i]=0; - + return new AliITSDigitizer(manager); } +//______________________________________________________________________ +void AliITS::SDigitsToDigits(Option_t *opt){ + // Standard Summable digits to Digits function. + // Inputs: + // none. + // Outputs: + // none. + char name[20] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0}; + + 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")}; + if( !det[0] && !det[1] && !det[2] ) all = "All"; + else all = 0; + static Bool_t setDef=kTRUE; + if (setDef) SetDefaultSimulation(); + setDef=kFALSE; + + AliITSsimulation *sim = 0; + AliITSDetType *iDetType = 0; + TTree *trees = fLoader->TreeS(); + if( !(trees && this->GetSDigits()) ){ + Error("SDigits2Digits","Error: No trees or SDigits. Returning."); + return; + } // end if + sprintf( name, "%s", this->GetName() ); + TBranch *brchSDigits = trees->GetBranch( name ); + + Int_t id,module; + for(module=0;moduleGetIndexMax();module++){ + id = geom->GetModuleType(module); + if (!all && !det[id]) continue; + iDetType = DetType(id); + sim = (AliITSsimulation*)iDetType->GetSimulationModel(); + if (!sim) { + Error("SDigit2Digits", + "The simulation class was not instanciated!"); + exit(1); + } // end if !sim + sim->InitSimulationModule(module,gAlice->GetEvNumber()); +// + // add summable digits to module + this->GetSDigits()->Clear(); + brchSDigits->GetEvent(module); + sim->AddSDigitsToModule(GetSDigits(),0); +// + // Digitise current module sum(SDigits)->Digits + sim->FinishSDigitiseModule(); + // fills all branches - wasted disk space + fLoader->TreeD()->Fill(); + this->ResetDigits(); + } // end for module -//____________________________________________ -void AliITS::ResetRecPoints() -{ - // - // Reset number of rec points and the rec points array - // - if (fRecPoints) fRecPoints->Clear(); - fNRecPoints = 0; + fLoader->TreeD()->GetEntries(); + fLoader->TreeD()->AutoSave(); + // reset tree + fLoader->TreeD()->Reset(); + } +//______________________________________________________________________ +void AliITS::Hits2Digits(){ + // Standard Hits to Digits function. + // Inputs: + // none. + // Outputs: + // none. + + fLoader->LoadHits("read"); + fLoader->LoadDigits("recreate"); + AliRunLoader* rl = fLoader->GetRunLoader(); + + 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->TreeD()) fLoader->MakeTree("D"); + MakeBranch("D"); + SetTreeAddress(); + HitsToDigits(iEvent,0,-1," ",fOpt," "); + } -//_____________________________________________________________________________ -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; + fLoader->UnloadHits(); + fLoader->UnloadDigits(); } - -//_____________________________________________________________________________ -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. - // - - - 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::HitsToSDigits(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. Dummy routine. Hits are ITS's Summable + // Digits. + // 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. +// return; // using Hits instead of the larger sdigits. + + HitsToPreDigits(evNumber,bgrev,size,option,opt,filename); } - -//_____________________________________________________________________________ -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::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. + + 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; + AliITSDetType *iDetType = 0; + AliITSmodule *mod = 0; + Int_t id,module; + for(module=0;moduleGetIndexMax();module++){ + id = geom->GetModuleType(module); + if (!all && !det[id]) continue; + iDetType = DetType(id); + sim = (AliITSsimulation*)iDetType->GetSimulationModel(); + if (!sim) { + Error("HitsToSDigits", + "The simulation class was not instanciated!"); + exit(1); + } // end if !sim + mod = (AliITSmodule *)fITSmodules->At(module); + sim->SDigitiseModule(mod,module,evNumber); + // fills all branches - wasted disk space + fLoader->TreeS()->Fill(); + ResetSDigits(); + } // end for module + + ClearModules(); + + fLoader->TreeS()->GetEntries(); + fLoader->TreeS()->AutoSave(); + fLoader->WriteSDigits("OVERWRITE"); + // reset tree + fLoader->TreeS()->Reset(); } -//_____________________________________________________________________________ -void AliITS::SetDefaultSimulation() -{ - // to be written - +//______________________________________________________________________ +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(!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; + AliITSDetType *iDetType = 0; + AliITSmodule *mod = 0; + Int_t id,module; + for(module=0;moduleGetIndexMax();module++){ + id = geom->GetModuleType(module); + if (!all && !det[id]) continue; + iDetType = DetType(id); + sim = (AliITSsimulation*)iDetType->GetSimulationModel(); + if (!sim) { + Error("HitsToDigits", + "The simulation class was not instanciated!"); + 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::SetDefaultClusterFinders() -{ - // to be written - +//______________________________________________________________________ +void AliITS::ResetSDigits(){ + // Reset the Summable Digits array. + // Inputs: + // none. + // Outputs: + // none. + + if (fSDigits) fSDigits->Clear(); + fNSDigits = 0; } -//_____________________________________________________________________________ - -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]; +//______________________________________________________________________ +void AliITS::ResetDigits(){ + // Reset number of digits and the digits array for the ITS detector. + // Inputs: + // none. + // Outputs: + // none. - 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; - } - } + if (!fDtype) return; + Int_t i; + for (i=0;iAt(i)) ((TClonesArray*)fDtype->At(i))->Clear(); + if (fNdtype) fNdtype[i]=0; + } // end for i } - -//_____________________________________________________________________________ -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; - } - - sprintf(treeName,"TreeC%d",event); - fTreeC = (TTree*)gDirectory->Get(treeName); - - - 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); - } - +//______________________________________________________________________ +void AliITS::ResetDigits(Int_t i){ + // Reset number of digits and the digits array for this branch. + // Inputs: + // none. + // Outputs: + // none. + + if (fDtype->At(i)) ((TClonesArray*)fDtype->At(i))->Clear(); + if (fNdtype) fNdtype[i]=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 - - 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; - } +//______________________________________________________________________ +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. + + TClonesArray &lsdig = *fSDigits; + new(lsdig[fNSDigits++]) AliITSpListItem(sdig); +} +//______________________________________________________________________ +void AliITS::AddRealDigit(Int_t id, 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. + + TClonesArray &ldigits = *((TClonesArray*)fDtype->At(id)); + new(ldigits[fNdtype[id]++]) AliITSdigit(digits); +} +//______________________________________________________________________ +void AliITS::AddSimDigit(Int_t id, 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. + + TClonesArray &ldigits = *((TClonesArray*)fDtype->At(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; + } // end switch id +} +//______________________________________________________________________ +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. + // 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. + + TClonesArray &ldigits = *((TClonesArray*)fDtype->At(id)); + AliITSresponseSDD *resp = 0; + switch(id){ + case 0: + new(ldigits[fNdtype[id]++]) AliITSdigitSPD(digits,tracks,hits); + break; + case 1: + resp = (AliITSresponseSDD*)DetType(1)->GetResponseModel(); + new(ldigits[fNdtype[id]++]) AliITSdigitSDD(phys,digits,tracks, + hits,charges,resp); + break; + case 2: + new(ldigits[fNdtype[id]++]) AliITSdigitSSD(digits,tracks,hits); + break; + } // end switch id +} +//______________________________________________________________________ +void AliITS::MakeTreeC(Option_t *option){ + // Create a separate tree to store the clusters. + // Inputs: + // Option_t *option string which must contain "C" otherwise + // no Cluster Tree is created. + // Outputs: + // none. + // Return: + // none. + + AliITSLoader *pITSLoader = (AliITSLoader*)fLoader; + + if (pITSLoader == 0x0) { + Error("MakeTreeC","fLoader == 0x0 option=%s",option); + return; } - - // 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 (pITSLoader->TreeC() == 0x0) pITSLoader->MakeTree("C"); + MakeBranchC(); } -//___________________________________________ -void AliITS::SetTreeAddress() +void AliITS::MakeBranchC() { +//Makes barnches in treeC + AliITSLoader *pITSLoader = (AliITSLoader*)fLoader; + if (pITSLoader == 0x0) + { + Error("MakeTreeC","fLoader == 0x0"); + return; + } + TTree * lTC = pITSLoader->TreeC(); + if (lTC == 0x0) + { + Error("MakeTreeC","Can not get TreeC from Loader"); + return; + } - // Set branch address for the Trees. - + Int_t buffersize = 4000; 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])); - } - } - } - - - if (treeR) { - sprintf(branchname,"%sRecPoints",GetName()); - if (fRecPoints) { - branch = treeR->GetBranch(branchname); - if (branch) branch->SetAddress(&fRecPoints); - } - } - - + const char *det[3] = {"SPD","SDD","SSD"}; + char digclass[40]; + char clclass[40]; + + // one branch for Clusters per type of detector + Int_t i; + for (i=0; iGetClassNames(digclass,clclass); + // clusters + if (fCtype == 0x0) fCtype = new TObjArray(fNDetTypes); + if(!ClustersAddress(i)) + { + fCtype->AddAt(new TClonesArray(clclass,1000),i); + } + if (kNTYPES==3) sprintf(branchname,"%sClusters%s",GetName(),det[i]); + else sprintf(branchname,"%sClusters%d",GetName(),i+1); + if (fCtype && lTC) + { + if (lTC->GetBranch(branchname)) + { + Warning("MakeBranchC","Branch %s alread exists in TreeC",branchname); + } + else + { + Info("MakeBranchC","Creating branch %s in TreeC",branchname); + lTC->Branch(branchname,&((*fCtype)[i]), buffersize); + } + } // end if fCtype && lTC + } // end for i } -//____________________________________________________________________________ -void AliITS::InitModules(Int_t size,Int_t &nmodules){ +//______________________________________________________________________ +void AliITS::GetTreeC(Int_t event){ + // Get the clusters tree for this event and set the branch address. + // Inputs: + // Int_t event Event number for the cluster tree. + // Outputs: + // none. + // Return: + // none. + char branchname[30]; + const char *det[3] = {"SPD","SDD","SSD"}; - //initialize the modules array + AliITSLoader *pITSLoader = (AliITSLoader*)fLoader; + TTree * lTC = pITSLoader->TreeC(); - if(fITSmodules){ - fITSmodules->Delete(); - delete fITSmodules; - } + ResetClusters(); + if (lTC) { + pITSLoader->CleanRawClusters(); + } // end if TreeC() - Int_t nl,indexMAX,index; - 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); - } + TBranch *branch; - nmodules = size; - } // end i size<=0 + if (lTC) { + Int_t i; + char digclass[40]; + char clclass[40]; + for (i=0; iGetClassNames(digclass,clclass); + // clusters + if (fCtype == 0x0) fCtype = new TObjArray(fNDetTypes); + if(!fCtype->At(i)) fCtype->AddAt(new TClonesArray(clclass,1000),i); + if(kNTYPES==3) sprintf(branchname,"%sClusters%s",GetName(),det[i]); + else sprintf(branchname,"%sClusters%d",GetName(),i+1); + if (fCtype) { + branch = lTC->GetBranch(branchname); + if (branch) branch->SetAddress(&((*fCtype)[i])); + } // end if fCtype + } // end for i + } else { + Error("GetTreeC","cannot find Clusters Tree for event:%d",event); + } // end if lTC } +//______________________________________________________________________ +void AliITS::AddCluster(Int_t id, AliITSRawCluster *c){ + // Add a cluster to the list. + // Inputs: + // Int_t id Detector type number. + // AliITSRawCluster *c Cluster class to be added to the tree of + // clusters. + // Outputs: + // none. + // Return: + // none. + + TClonesArray &lc = *((TClonesArray*)fCtype->At(id)); + + switch(id){ + case 0: + new(lc[fNctype[id]++]) AliITSRawClusterSPD(*((AliITSRawClusterSPD*)c)); + break; + case 1: + new(lc[fNctype[id]++]) AliITSRawClusterSDD(*((AliITSRawClusterSDD*)c)); + break; + case 2: + new(lc[fNctype[id]++]) AliITSRawClusterSSD(*((AliITSRawClusterSSD*)c)); + break; + } // end switch id +} +//______________________________________________________________________ +void AliITS::ResetClusters(){ + // Reset number of clusters and the clusters array for ITS. + // Inputs: + // none. + // Outputs: + // none. -//____________________________________________________________________________ -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); - } - // 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); - } - - 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 + Int_t i; + for (i=0;iAt(i)) ((TClonesArray*)fCtype->At(i))->Clear(); + if (fNctype) fNctype[i]=0; +} +//______________________________________________________________________ +void AliITS::MakeBranchR(const char *file, Option_t *opt){ + // Creates Tree branches for the ITS Reconstructed points. + // Inputs: + // cont char *file File name where RecPoints branch is to be written + // to. If blank it write the SDigits to the same + // file in which the Hits were found. + // Outputs: + // none. + // Return: + // none. + Int_t buffsz = 4000; + char branchname[30]; - // open the file with background + // only one branch for rec points for all detector types + Bool_t oFast= (strstr(opt,"Fast")!=0); + if(oFast){ + sprintf(branchname,"%sRecPointsF",GetName()); + } else { + sprintf(branchname,"%sRecPoints",GetName()); + } - 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(); - + + if(!fRecPoints)fRecPoints = new TClonesArray("AliITSRecPoint",1000); + if (fLoader->TreeR()) { + if (fRecPoints == 0x0) fRecPoints = new TClonesArray("AliITSRecPoint",1000); + MakeBranchInTree(fLoader->TreeR(),branchname,&fRecPoints,buffsz,file); + } // end if } +//______________________________________________________________________ +void AliITS::SetTreeAddressR(TTree *treeR){ + // Set branch address for the Reconstructed points Trees. + // Inputs: + // TTree *treeR Tree containing the RecPoints. + // Outputs: + // none. + // Return: + // none. + char branchname[30]; - -//____________________________________________________________________________ -void AliITS::HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size, Option_t *option, Option_t *opt,Text_t *filename) -{ + if(!treeR) return; + if (fRecPoints == 0x0) fRecPoints = new TClonesArray("AliITSRecPoint",1000); + TBranch *branch; + sprintf(branchname,"%sRecPoints",GetName()); + branch = treeR->GetBranch(branchname); + if (branch) { + branch->SetAddress(&fRecPoints); + } + else { + sprintf(branchname,"%sRecPointsF",GetName()); + branch = treeR->GetBranch(branchname); + if (branch) { + branch->SetAddress(&fRecPoints); + } + } +} +//______________________________________________________________________ +void AliITS::AddRecPoint(const AliITSRecPoint &r){ + // Add a reconstructed space point to the list + // Inputs: + // const AliITSRecPoint &r RecPoint class to be added to the tree + // of reconstructed points TreeR. + // Outputs: + // none. + // Return: + // none. + + TClonesArray &lrecp = *fRecPoints; + new(lrecp[fNRecPoints++]) AliITSRecPoint(r); +} +//______________________________________________________________________ +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 - // 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;idGetSimulationModel(); + if (!sim) + { + Error("HitsToFastPoints","The simulation class was not instanciated!"); 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(); - + } // end if !sim + mod = (AliITSmodule *)fITSmodules->At(module); + sim->CreateFastRecPoints(mod,module,gRandom); + cout<TreeR()->Fill(); + TTree *lTR = pITSloader->TreeR(); + TBranch *br=lTR->GetBranch("ITSRecPointsF"); + br->Fill(); + ResetRecPoints(); + } // end for module + + ClearModules(); + + fLoader->WriteRecPoints("OVERWRITE"); } - - -//____________________________________________________________________________ -void AliITS::DigitsToRecPoints(Int_t evNumber,Int_t lastentry,Option_t *opt) -{ +//______________________________________________________________________ +void AliITS::Digits2Reco(){ + // Find clusters and reconstruct space points. + // Inputs: + // none. + // Outputs: + // none. + + AliHeader *header=fLoader->GetRunLoader()->GetHeader(); + // to Digits to RecPoints for event in file, all digits in file, and + // all ITS detectors. + DigitsToRecPoints(header->GetEvent(),0,fOpt); +} +//______________________________________________________________________ +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(); + // 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 evNumber Event number to be processed. + // Int_t lastentry Offset for module when not all of the modules + // are processed. + // Option_t *opt String indicating which ITS sub-detectors should + // be processed. If ="All" then all of the ITS + // sub detectors are processed. + // Outputs: + // none. + // Return: + // none. + + if(!GetITSgeom()) return; + 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 setRec=kTRUE; + if (setRec) SetDefaultClusterFinders(); + setRec=kFALSE; + + AliITSLoader *pITSloader = (AliITSLoader*)fLoader; + TTree *treeC=pITSloader->TreeC(); + AliITSClusterFinder *rec = 0; + AliITSDetType *iDetType = 0; + Int_t id,module,first=0; + for(module=0;moduleGetIndexMax();module++){ + id = geom->GetModuleType(module); + if (!all && !det[id]) continue; + if(det[id]) first = geom->GetStartDet(id); + iDetType = DetType(id); + rec = (AliITSClusterFinder*)iDetType->GetReconstructionModel(); + TClonesArray *itsDigits = this->DigitsAddress(id); + if (!rec) { + Error("DigitsToRecPoints", + "The reconstruction class was not instanciated! event=%d", + evNumber); + exit(1); + } // end if !rec + this->ResetDigits(); + TTree *lTD = pITSloader->TreeD(); + if (all) { + lTD->GetEvent(lastentry+module); + }else { + lTD->GetEvent(lastentry+(module-first)); + } + Int_t ndigits = itsDigits->GetEntriesFast(); + if (ndigits) rec->FindRawClusters(module); + pITSloader->TreeR()->Fill(); + ResetRecPoints(); + treeC->Fill(); + ResetClusters(); + } // end for module - //TBranch *branch; - AliITSClusterFinder* rec; - //TObjArray *branches=gAlice->TreeR()->GetListOfBranches(); - AliITSgeom *geom = GetITSgeom(); + pITSloader->WriteRecPoints("OVERWRITE"); + pITSloader->WriteRawClusters("OVERWRITE"); +} +//______________________________________________________________________ +void AliITS::ResetRecPoints(){ + // Reset number of rec points and the rec points array. + // Inputs: + // none. + // Outputs: + // none. - 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(); + if (fRecPoints) fRecPoints->Clear(); + fNRecPoints = 0; +} +//______________________________________________________________________ +AliLoader* AliITS::MakeLoader(const char* topfoldername) +{ + //builds ITSgetter (AliLoader type) + //if detector wants to use castomized getter, it must overload this method + + Info("MakeLoader","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 size, -Option_t *option,Option_t *opt,Text_t *filename) +//_____________________________________________________________________________ +void AliITS::Reconstruct() const { - // 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 ! - 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")}; +// reconstruct clusters - Int_t nmodules; - InitModules(size,nmodules); - FillModules(evNumber,bgrev,nmodules,option,filename); + AliLoader* loader = GetLoader(); + loader->LoadRecPoints("recreate"); + loader->LoadDigits("read"); + AliITSclustererV2 clusterer(GetITSgeom()); + AliRunLoader* runLoader = loader->GetRunLoader(); + Int_t nEvents = runLoader->GetNumberOfEvents(); - 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 - + for (Int_t iEvent = 0; iEvent < nEvents; iEvent++) { + runLoader->GetEvent(iEvent); - 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(); + TTree* treeClusters = loader->TreeR(); + if (!treeClusters) { + loader->MakeTree("R"); + treeClusters = loader->TreeR(); + } + TTree* treeDigits = loader->TreeD(); + if (!treeDigits) { + Error("Reconstruct", "Can't get digits tree !"); + return; + } - delete [] random; + clusterer.Digits2Clusters(treeDigits, treeClusters); + + loader->WriteRecPoints("OVERWRITE"); + } + loader->UnloadRecPoints(); + loader->UnloadDigits(); } +//_____________________________________________________________________________ +AliTracker* AliITS::CreateTracker() const +{ +// create an ITS tracker -//____________________________________________________________________________ -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;iGetVertex(vtx, cvtx); + vtxer.SetVertex(vtx); + vtxer.Tracks2V0vertices(esd); + + // cascade finding + Double_t cts[]={33., // max. allowed chi2 + 0.05, // min. allowed V0 impact parameter + 0.008, // window around the Lambda mass + 0.035, // min. allowed bachelor's impact parameter + 0.10, // max. allowed DCA between a V0 and a track + 0.9985, //max. allowed cosine of the cascade pointing angle + 0.9, // min. radius of the fiducial volume + 2.9 // max. radius of the fiducial volume + }; + AliCascadeVertexer cvtxer=AliCascadeVertexer(cts); + cvtxer.V0sTracks2CascadeVertices(esd); +}