/*
$Log$
-Revision 1.10 2000/01/19 17:16:51 fca
-Introducing a list of lists of hits -- more hits allowed for detector now
+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)<dx....
+
+Revision 1.9.2.1 2000/01/12 19:03:32 nilsen
+This is the version of the files after the merging done in December 1999.
+See the ReadMe110100.txt file for details
Revision 1.9 1999/11/14 14:33:25 fca
Correct problems with distructors and pointers, thanks to I.Hrivnacova
// Modified and documented by Bjorn S. Nilsen
// July 11 1999
//
+// Version: 2
+// Modified and documented by A. Bologna
+// October 18 1999
+//
// AliITS is the general base class for the ITS. Also see AliDetector for
// futher information.
//
#include <TMath.h>
#include <TRandom.h>
+#include <TBranch.h>
#include <TVector.h>
-#include <TGeometry.h>
-#include <TNode.h>
-#include <TTUBE.h>
+#include <TObjArray.h>
+#include <TROOT.h>
+#include <TObjectTable.h>
+#include <TFile.h>
+#include <TTree.h>
-#include "AliITSmodule.h"
-#include "AliDetector.h"
+
+
+#include "AliRun.h"
#include "AliITS.h"
-#include "TClonesArray.h"
-#include "TObjArray.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 "AliITShit.h"
+#include "AliITSgeom.h"
#include "AliITSdigit.h"
-#include "AliRun.h"
+#include "AliITSmodule.h"
+#include "AliITSRecPoint.h"
+#include "AliITSRawCluster.h"
-ClassImp(AliITS)
+const Int_t AliITS::fgkNTYPES=3;
-////////////////////////////////////////////////////////////////////////
-//
-// An overview of the basic philosophy of the ITS code development
-// and analysis is show in the figure below.
-//Begin_Html
-/*
-<img src="picts/ITS/ITS_Analysis_schema.gif">
-</pre>
-<br clear=left>
-<font size=+2 color=red>
-<p>Roberto Barbera is in charge of the ITS Offline code (1999).
-<a href="mailto:roberto.barbera@ct.infn.it">Roberto Barbera</a>.
-</font>
-<pre>
-*/
-//End_Html
-//
-// Version: 0
-// Written by Rene Brun, Federico Carminati, and Roberto Barbera
-//
-// Version: 1
-// Modified and documented by Bjorn S. Nilsen
-// July 11 1999
-//
-// AliITS is the general base class for the ITS. Also see AliDetector for
-// futher information.
-//
-// Data members:
-//
-// AliITSgeom *fITSgeom
-// All of the geometry information about the active volumes that
-// make up the ITS are described in the AliITSgeom class. This includes
-// the transformation functions between the local and global coordinate
-// systems and the like. See the full description found in the AliITSgeom
-// class. Here in the AliITS class is kept the pointer to the geometry
-// used in the simulations or that thought to be the correct one for the
-// data. Until a more general class is define and a more appropriate
-// place is found to keep this geometry information, it is kept here in
-// AliITS.
-//
-// TObjArray *fITSpoints
-// This is a pointer to the points, to be used by the tracking algorithms
-// for example, found in the detectors of the ITS. To allow for the most
-// general points structure it is defined to be a pointer to a TObjArray where
-// each array element would be one point found in the ITS detectors. An
-// Addpoints function is defined below. By default an array of 16 TObjects are
-// defined during the initialization of AliITS. This is automatically expanded
-// when necessary by the Addpoints function.
-//
-// Bool_t fEuclidOut
-// This is a flag used to indicate that an Euclid compatible CAD
-// file will be created upon the creation of the ITS Monte Carlo
-// geometry definition, in the function CreatGeometry. If fEuclidOut is
-// true, then a file called ITSgeometry.euc will be created.
-//
-// Int_t fIdN
-// This variable contains the number of layers defined in the ITS
-// geometry. It is primarily used as a size indicator for fIdSens and
-// fIdName described below. In general the number of layers, ladders, or
-// detectors should be gotten from the AliITSgeom functions. Upon
-// creating the AliITS object it is set to zero.
-//
-// Int_t *fIdSens
-// This is a pointer to an array containing the Monte Carlo volume
-// numbers for the different layers of the ITS. These numbers are needed
-// by the StepManager function to determine what layer a hit was on. It
-// is sized and initialized in the Init function and the AliITSv? Init
-// function, called after a call to CreateGeometry. Upon creating the
-// AliITS object it points to zero. This variable is made a pointer
-// in order to keep the maximum flexibility at this level of the code.
-//
-// char **fIdName
-// This is a pointer to an array of characters containing the names of
-// the different ITS layers as defined in the Monte Carlo geometry data-
-// base. It is sized and filled in the AliITSv? Init function, called
-// after a call to CreatGeometry. Upon creating the AliITS object it
-// points to zero. This variable is make a pointer in order to keep the
-// maximum flexibility at this level of the code.
-//
-// Member Functions:
-//
-// AliITS()
-// 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.
-//
-// AliITS(const char *name, const char *title)
-// 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.
-//
-// ~AliITS()
-// 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.
-//
-// AddHit(Int_t track, Int_t *vol, Float_t *hits)
-// 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.
-//
-// AddDigit(Int_t *track, Int_t *digits)
-// The function to add information to the AliITSdigits class. See the
-// AliITSdigits class for a full description. This function allocates the
-// necessary new space for the digits information and passes the pointers
-// *track and *digits to the AliITSdigits constructor function.
-//
-// BuildGeometry()
-// This function builds a simple ITS geometry used by the ROOT macro
-// display.C. In general the geometry as coded is wrong.
-//
-// CreateGeometry()
-// This function builds the detailed geometry used by the Geant
-// Monte Carlo. As defined here it is a dummy routine to be replaced
-// by the version coded up in AliITSv? where the specific geometry to
-// be used by the simulation is defined. See the definition of AliITSv5
-// or the other routines for a complete definition.
-//
-// CreateMaterials()
-// This function defines the default materials used in the Geant
-// Monte Carlo simulations. In general it is automatically replaced by
-// the CreatMaterials routine defined in AliITSv?. Should the function
-// CreateMaterials not exist for the geometry version you are using this
-// one is used. See the definition found in AliITSv5 or the other routine
-// for a complete definition.
-//
-// IsVersion()
-// Returns the version number of the AliITS class. At present it is
-// version 1.
-//
-// DistancetoPrimitive(Int_t x, Int_t y)
-// 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.
-//
-// Init()
-// 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.
-// Therefore it should be called after a call to CreateGeometry.
-//
-// MakeBranch(Option_t *Opt=" ")
-// Creates the TTree branch where the class AliITS is kept.
-//
-// SetEUCLID(bool_t euclid=1)
-// Sets the flag fEuclidOut to true (default) of false (euclid=0).
-// By setting or clearing the fEuclidOut flag you can controls whether
-// or not a euclid formatted output file of the ITS geometry is written.
-// If fEuclidOut is set true then a file called ITSgeometry.euc will be
-// written after the ITS geometry is defined in the Monte Carlo. If
-// fEuclidOut is set false then no file is created.
-//
-// StepManager()
-// Dummy routine which is replaced by the routine StepManager() defined
-// in AliITSv?. If no such routine exist then this routine returns zero.
-// See AliITSv? for a detailed description of the step manager routines.
-//
-// GetITSgeom()
-// Returns the value of the pointer fITSgeom. This is used to get
-// access to the ITS geometry stored in the file. See AliITSgeom for a
-// full description of the geometry package.
-//
-// GetITSpoints()
-// Returns the value of the pointer fITSpoints. This is used to get
-// access to the ITS cluster objects, if filled, stored in the file. See
-// AliITSCluster for a full description of the cluster data.
-////////////////////////////////////////////////////////////////////////
+ClassImp(AliITS)
+
//_____________________________________________________________________________
-AliITS::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.
+ // fITSpoints, fIdSens, and fIdName. The AliDetector default constructor
+ // is also called.
//
- fITSpoints = 0;
+
+
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;
}
+
//_____________________________________________________________________________
AliITS::AliITS(const char *name, const char *title):AliDetector(name,title){
//
// description of these parameters and its constructor functions.
//
+
fHits = new TClonesArray("AliITShit", 1560);
gAlice->AddHitList(fHits);
- fDigits = new TClonesArray("AliITSdigit",1000);
- fITSpoints = new TObjArray();
- fITSmodules = 0; //new AliITSmodules();
+
+ //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);
+
+ Int_t i;
+ for(i=0;i<fgkNTYPES;i++) {
+ (*fDetTypes)[i]=new AliITSDetType();
+ fNdtype[i]=0;
+ fNctype[i]=0;
+ }
+ //
SetMarkerColor(kRed);
+ fITSgeom=0;
}
+//___________________________________________________________________________
+AliITS::AliITS(AliITS &source){
+ // copy constructor
+ if(this==&source) return;
+ printf("Error: You are not allowed to make a copy of the AliITS\n");
+ exit(1);
+}
+//____________________________________________________________________________
+AliITS& AliITS::operator=(AliITS &source){
+ // assignment operator
+ if(this==&source) return *this;
+ printf("Error: You are not allowed to make a copy of the AliITS\n");
+ exit(1);
+ return *this; //fake return
+}
+//____________________________________________________________________________
+void AliITS::ClearModules(){
+ //clear the modules TObjArray
+
+ if(fITSmodules) fITSmodules->Delete();
+}
//_____________________________________________________________________________
AliITS::~AliITS(){
//
// 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;
- delete fITSmodules;
- if(fITSpoints!=0) delete fITSpoints;
-}
+ if(fITSmodules!=0) {
+ this->ClearModules();
+ delete fITSmodules;
+ }// end if fITSmodules!=0
-//_____________________________________________________________________________
-void AliITS::AddDigit(Int_t *tracks, Int_t *digits){
//
- // Add an ITS Digit
- // The function to add information to the AliITSdigits class. See the
- // AliITSdigits class for a full description. This function allocates the
- // necessary new space for the digits information and passes the pointers
- // *track and *digits to the AliITSdigits constructor function.
+ if(fDtype) {
+ fDtype->Delete();
+ delete fDtype;
+ }
+ delete [] fNdtype;
+ if (fCtype) {
+ fCtype->Delete();
+ delete fCtype;
+ }
+ delete [] fNctype;
//
- TClonesArray &ldigits = *fDigits;
- new(ldigits[fNdigits++]) AliITSdigit(tracks,digits);
+
+ if (fDetTypes) {
+ fDetTypes->Delete();
+ delete fDetTypes;
+ }
+
+ if (fTreeC) delete fTreeC;
+
+ if (fITSgeom) delete fITSgeom;
+
}
-Int_t AliITS::AddDigit(AliITSdigit* d) {
+//___________________________________________
+AliITSDetType* AliITS::DetType(Int_t id)
+{
+ //return pointer to id detector type
+ return ((AliITSDetType*) (*fDetTypes)[id]);
+
+}
+//___________________________________________
+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
+
+ ((AliITSDetType*) (*fDetTypes)[id])->ResponseModel(response);
+
+}
+
+//___________________________________________
+void AliITS::SetSegmentationModel(Int_t id, AliITSsegmentation *seg)
+{
+ //set the segmentation model for the id detector type
+
+ ((AliITSDetType*) (*fDetTypes)[id])->SegmentationModel(seg);
+
+}
+
+//___________________________________________
+void AliITS::SetSimulationModel(Int_t id, AliITSsimulation *sim)
+{
+ //set the simulation model for the id detector type
+
+ ((AliITSDetType*) (*fDetTypes)[id])->SimulationModel(sim);
+
+}
+//___________________________________________
+void AliITS::SetReconstructionModel(Int_t id, AliITSClusterFinder *reconst)
+{
+ //set the cluster finder model for the id detector type
+
+ ((AliITSDetType*) (*fDetTypes)[id])->ReconstructionModel(reconst);
- fDigits->Add(d);
- fNdigits = fDigits->GetEntriesFast();
- return fNdigits;
}
//_____________________________________________________________________________
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::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::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::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 (!fDtype) return;
+
+ Int_t i;
+ for (i=0;i<fgkNTYPES;i++ ) {
+ if ((*fDtype)[i]) ((TClonesArray*)(*fDtype)[i])->Clear();
+ if (fNdtype) fNdtype[i]=0;
+ }
+}
+
+//____________________________________________
+void AliITS::ResetDigits(Int_t i)
+{
+ //
+ // Reset number of digits and the digits array for this branch
+ //
+ if ((*fDtype)[i]) ((TClonesArray*)(*fDtype)[i])->Clear();
+ if (fNdtype) fNdtype[i]=0;
+}
+
+
+//____________________________________________
+void AliITS::ResetClusters()
+{
+ //
+ // Reset number of clusters and the clusters array for ITS
+ //
+
+ Int_t i;
+ for (i=0;i<fgkNTYPES;i++ ) {
+ if ((*fCtype)[i]) ((TClonesArray*)(*fCtype)[i])->Clear();
+ if (fNctype) fNctype[i]=0;
+ }
+
+}
+
+//____________________________________________
+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 ){
//
// 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.
+ // 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;
- //
printf("\n");
for(i=0;i<35;i++) printf("*");
printf(" ITS_INIT ");
printf("\n");
}
+//_____________________________________________________________________________
+void AliITS::SetDefaults()
+{
+ // sets the default segmentation, response, digit and raw cluster classes
+
+ printf("SetDefaults\n");
+
+ 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::SetDefaultSimulation()
+{
+ // to be written
+
+}
+//_____________________________________________________________________________
+void AliITS::SetDefaultClusterFinders()
+{
+ // to be written
+
+}
+//_____________________________________________________________________________
+
+void AliITS::MakeTreeC(Option_t *option)
+{
+ // create a separate tree to store the clusters
+
+ printf("MakeTreeC \n");
+
+ 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; i<fgkNTYPES ;i++) {
+ if (fgkNTYPES==3) sprintf(branchname,"%sClusters%s",GetName(),det[i]);
+ else sprintf(branchname,"%sClusters%d",GetName(),i+1);
+ if (fCtype && fTreeC) {
+ TreeC()->Branch(branchname,&((*fCtype)[i]), buffersize);
+ printf("Making Branch %s for Clusters of detector type %d\n",branchname,i+1);
+ }
+ }
+
+}
+
+//_____________________________________________________________________________
+void AliITS::GetTreeC(Int_t event)
+{
+
+ printf("GetTreeC \n");
+
+ // 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; i<fgkNTYPES; i++) {
+ if (fgkNTYPES==3) sprintf(branchname,"%sClusters%s",GetName(),det[i]);
+ else sprintf(branchname,"%sClusters%d",GetName(),i+1);
+ if (fCtype) {
+ branch = fTreeC->GetBranch(branchname);
+ if (branch) branch->SetAddress(&((*fCtype)[i]));
+ }
+ }
+ } else {
+ printf("ERROR: cannot find Clusters Tree for event:%d\n",event);
+ }
+
+}
//_____________________________________________________________________________
void AliITS::MakeBranch(Option_t* option){
//
- // Create Tree branches for the ITS.
- // Creates the TTree branch where the class AliITS is kept.
+ // Creates Tree branches for the ITS.
//
+
+
Int_t buffersize = 4000;
- char branchname[10];
+ char branchname[30];
sprintf(branchname,"%s",GetName());
AliDetector::MakeBranch(option);
- char *D = strstr(option,"D");
- if (fDigits && gAlice->TreeD() && D) {
- gAlice->TreeD()->Branch(branchname,&fDigits, buffersize);
- printf("Making Branch %s for digits\n",branchname);
- } // end if
+// 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; i<fgkNTYPES ;i++) {
+ AliITSDetType *iDetType=DetType(i);
+ iDetType->GetClassNames(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; i<fgkNTYPES ;i++) {
+ if (fgkNTYPES==3) sprintf(branchname,"%sDigits%s",GetName(),det[i]);
+ else sprintf(branchname,"%sDigits%d",GetName(),i+1);
+
+ if (fDtype && gAlice->TreeD()) {
+ gAlice->TreeD()->Branch(branchname,&((*fDtype)[i]), buffersize);
+ printf("Making Branch %s for digits of type %d\n",branchname,i+1);
+ }
+ }
+
+ // 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);
+ printf("Making Branch %s for reconstructed space points\n",branchname);
+ }
+
+
+}
+
+//___________________________________________
+void AliITS::SetTreeAddress()
+{
+
+ // Set branch address for the Trees.
+
+ char branchname[30];
+ AliDetector::SetTreeAddress();
+
+ char *det[3] = {"SPD","SDD","SSD"};
+
+ TBranch *branch;
+ TTree *treeD = gAlice->TreeD();
+ TTree *treeR = gAlice->TreeR();
+
+ Int_t i;
+ if (treeD) {
+ for (i=0; i<fgkNTYPES; i++) {
+ if (fgkNTYPES==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]));
+ }
+ }
+ }
+
+
+ if (treeR) {
+ sprintf(branchname,"%sRecPoints",GetName());
+ if (fRecPoints) {
+ branch = treeR->GetBranch(branchname);
+ if (branch) branch->SetAddress(&fRecPoints);
+ }
+ }
+
+
+}
+
+//____________________________________________________________________________
+void AliITS::InitModules(Int_t size,Int_t &nmodules){
+
+ //initialize the modules array
+
+ if(fITSmodules){
+ fITSmodules->Delete();
+ delete fITSmodules;
+ }
+
+ Int_t nl,indexMAX,index;
+
+ if(size<=0){ // default to using data stored in AliITSgeom
+ if(fITSgeom==0) {
+ printf("Error in AliITS::InitModule fITSgeom not defined\n");
+ return;
+ } // end if fITSgeom==0
+ nl = fITSgeom->GetNlayers();
+ indexMAX = fITSgeom->GetModuleIndex(nl,fITSgeom->GetNladders(nl),
+ fITSgeom->GetNdetectors(nl))+1;
+ nmodules = indexMAX;
+ fITSmodules = new TObjArray(indexMAX);
+ for(index=0;index<indexMAX;index++){
+ fITSmodules->AddAt( new AliITSmodule(index),index);
+ } // end for index
+ }else{
+ fITSmodules = new TObjArray(size);
+ for(index=0;index<size;index++) {
+ fITSmodules->AddAt( new AliITSmodule(index),index);
+ }
+
+ nmodules = size;
+ } // end i size<=0
+}
+
+//____________________________________________________________________________
+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 "<<filename<<endl;
+ file=new TFile(filename);
+ cout<<"I have opened "<<filename<<" file "<<endl;
+ fHits2 = new TClonesArray("AliITShit",1000 );
+ }
+ first=kFALSE;
+ file->cd();
+ file->ls();
+ // Get Hits Tree header from file
+ if(fHits2) fHits2->Clear();
+ if(trH1) delete trH1;
+ trH1=0;
+
+ char treeName[20];
+ sprintf(treeName,"TreeH%d",bgrev);
+ trH1 = (TTree*)gDirectory->Get(treeName);
+ //printf("TrH1 %p of treename %s for event %d \n",trH1,treeName,bgrev);
+
+ if (!trH1) {
+ printf("ERROR: cannot find Hits Tree for event:%d\n",bgrev);
+ }
+ // Set branch addresses
+ TBranch *branch;
+ char branchname[20];
+ sprintf(branchname,"%s",GetName());
+ if (trH1 && fHits2) {
+ branch = trH1->GetBranch(branchname);
+ if (branch) branch->SetAddress(&fHits2);
+ }
+
+ // test
+ //Int_t ntracks1 =(Int_t)TrH1->GetEntries();
+ //printf("background - ntracks1 - %d\n",ntracks1);
+ }
+
+ 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; t<ntracks; t++){
+ gAlice->ResetHits();
+ iTH->GetEvent(t);
+ Int_t nhits = itsHits->GetEntriesFast();
+ //printf("nhits %d\n",nhits);
+ if (!nhits) continue;
+ for(h=0; h<nhits; h++){
+ itsHit = (AliITShit *)itsHits->UncheckedAt(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
+
+ // 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; track<ntracks; track++) {
+
+ if (fHits2) fHits2->Clear();
+ trH1->GetEvent(track);
+ // Loop over hits
+ for(i=0;i<fHits2->GetEntriesFast();++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::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;id<fgkNTYPES;id++) {
+ if (!all && !det[id]) continue;
+ //branch = (TBranch*)branches->UncheckedAt(id);
+ AliITSDetType *iDetType=DetType(id);
+ sim = (AliITSsimulation*)iDetType->GetSimulationModel();
+ if (!sim) {
+ Error("HitsToDigits","The simulation class was not instantiated!");
+ exit(1);
+ // or SetDefaultSimulation();
+ }
+ if(geom) {
+ first = geom->GetStartDet(id);
+ last = geom->GetLastDet(id);
+ } else first=last=0;
+ printf("det type %d first, last %d %d \n",id,first,last);
+ for(module=first;module<=last;module++) {
+ AliITSmodule *mod = (AliITSmodule *)fITSmodules->At(module);
+ sim->DigitiseModule(mod,module,evNumber);
+ // fills all branches - wasted disk space
+ gAlice->TreeD()->Fill();
+ ResetDigits();
+ // try and fill only the branch
+ //branch->Fill();
+ //ResetDigits(id);
+ } // loop over modules
+ } // loop over detector types
+
+ ClearModules();
+
+ Int_t nentries=(Int_t)gAlice->TreeD()->GetEntries();
+ printf("nentries in TreeD %d\n",nentries);
+
+ char hname[30];
+ sprintf(hname,"TreeD%d",evNumber);
+ gAlice->TreeD()->Write(hname);
+ // reset tree
+ gAlice->TreeD()->Reset();
+
+}
+
+
+//____________________________________________________________________________
+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;id<fgkNTYPES;id++) {
+ if (!all && !det[id]) continue;
+ //branch = (TBranch*)branches->UncheckedAt(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();
+ printf(" nentries ncentries %d %d\n", nentries, ncentries);
+
+ 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::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 !
+
+
+ // 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;id<fgkNTYPES;id++) {
+ if (!all && !det[id]) continue;
+ AliITSDetType *iDetType=DetType(id);
+ sim = (AliITSsimulation*)iDetType->GetSimulationModel();
+ 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,j,l;
+
+ Int_t i;
if (R__b.IsReading()) {
- Version_t R__v = R__b.ReadVersion();
- if (R__v == 1) {
+ 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<fIdN;i++) fIdName[i].Streamer(R__b);
R__b >> fITSgeom;
-// R__b >> fITSmodules; //We do not write out modules so don't read them
- R__b >> fITSpoints;
+ R__b >> fITSmodules;
R__b >> fEuclidOut;
- R__b >> fIdN;
- if(fIdSens!=0) delete[] fIdSens;
- if(fIdName!=0) delete[] fIdName;
- fIdSens = new Int_t[fIdN];
- fIdName = new char*[fIdN];
- for(i=0;i<fIdN;i++) R__b >> fIdSens[i];
- for(i=0;i<fIdN;i++){
- R__b >> l;
- fIdName[i] = new char[l+1]; // add room for null character.
- for(j=0;j<l;j++) R__b >> fIdName[i][j];
- fIdName[i][l] = '\0'; // Null terminate this string.
- } // end for i
R__b >> fMajorVersion;
R__b >> fMinorVersion;
- } // end if (R__v)
- } else {
+ 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<fIdN;i++) fIdName[i].Streamer(R__b);
R__b << fITSgeom;
-// R__b << fITSmodules; //We don't want to write out the modules class.
- R__b << fITSpoints;
+ R__b << fITSmodules;
R__b << fEuclidOut;
- R__b << fIdN;
- for(i=0;i<fIdN;i++) R__b <<fIdSens[i];
- for(i=0;i<fIdN;i++){
- l = strlen(fIdName[i]);
- R__b << l;
- for(j=0;j<l;j++) R__b << fIdName[i][j];
- } // end for i
R__b << fMajorVersion;
R__b << fMinorVersion;
- }
+ R__b << fDetTypes;
+ R__b << fDtype;
+ R__b.WriteFastArray(fNdtype,fgkNTYPES);
+ R__b << fCtype;
+ R__b.WriteFastArray(fNctype,fgkNTYPES);
+ R__b << fRecPoints;
+ R__b << fNRecPoints;
+ R__b << fTreeC;
+ } // end if
+
}
+
+