Corrections to the custom Streamer
[u/mrichter/AliRoot.git] / ITS / AliITS.h
index 67f157a822167ebcb81692948f170190cde2f30e..cfc4bd50b0fcbf1a065159b89a8937e18583adfb 100644 (file)
-#ifndef ITS_H
-#define ITS_H
-////////////////////////////////////////////////
-//  Manager and hits classes for set: ITS     //
-////////////////////////////////////////////////
+#ifndef ALIITS_H
+#define ALIITS_H
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+
+/* $Id$ */
+
+////////////////////////////////////////////////////////////////////////
+//           Manager class for set: ITS                               //
+////////////////////////////////////////////////////////////////////////
+
+#include <TObjArray.h>
+#include <TBranch.h>
+#include <TTree.h>
+#include <TFile.h>
+#include <TObjectTable.h>
 
 #include "AliDetector.h"
-#include "AliHit.h" 
-#include "AliDigit.h" 
+#include "AliITSgeom.h"
+#include "AliITSsegmentation.h"
+#include "AliITSresponse.h"
+#include "AliITSDetType.h"
+#include "AliITShit.h"
+#include "AliITSdigit.h"
+#include "AliITSmodule.h"
+#include "AliITSRecPoint.h"
+#include "AliITSRawCluster.h"
+
+
+class AliITSsimulation;
+class AliITSClusterFinder;
+//class AliITStrack;
 
 class AliITS : public AliDetector {
-  
-protected:
-  
-  Bool_t fEuclidOut;
-  
-  Int_t fIdSens1;    //First  layer identifier
-  Int_t fIdSens2;    //Second layer identifier
-  Int_t fIdSens3;    //Third  layer identifier
-  Int_t fIdSens4;    //Fourth layer identifier
-  Int_t fIdSens5;    //Fifth  layer identifier
-  Int_t fIdSens6;    //Sixth  layer identifier
-  
-public:
-  AliITS();
-  AliITS(const char *name, const char *title);
-  virtual       ~AliITS();
-  virtual void   AddHit(Int_t, Int_t*, Float_t*);
-  virtual void   AddDigit(Int_t*, Int_t*);
-  virtual void   BuildGeometry();
-  virtual void   CreateGeometry() {}
-  virtual void   CreateMaterials();
-  virtual Int_t  IsVersion() const =0;
-  Int_t          DistancetoPrimitive(Int_t px, Int_t py);
-  virtual void   Init();
-  virtual void   MakeBranch(Option_t *opt=" ");
-  virtual void   SetEUCLID(Bool_t euclid=1);
-  virtual void   StepManager()=0;
-  
-  ClassDef(AliITS,1)  //Hits manager for set:ITS
-};
 
-//___________________________________________
-class AliITSdigit: public AliDigit  {
-public:
-   Int_t       fEvent;      // Event number
-   Int_t       fLayer;      // Layer number
-   Int_t       fDet ;       // Detector number
-   Int_t       fNoverl;     // Number of overflow
-public:
-   AliITSdigit() {}
-   AliITSdigit(Int_t *tracks, Int_t *digits);
-   virtual ~AliITSdigit() {}
-   ClassDef(AliITSdigit,1)  //Digit (Header) object for set:ITS
-};
-//___________________________________________
-class AliITShit : public AliHit {
-public:
-   Int_t     fLayer;      // Layer number
-   Int_t     fLadder;     // Ladder number
-   Int_t     fDet;        // Detector number  
-   Float_t   fPx  ;       //PX
-   Float_t   fPy  ;       //PY
-   Float_t   fPz  ;       //PZ
-   Float_t   fDestep;     // Energy deposited in the current step
-public:
-   AliITShit() {}
-   AliITShit(Int_t shunt, Int_t track, Int_t *vol, Float_t *hits);
-   virtual ~AliITShit() {}
-   ClassDef(AliITShit,1)  //Hits object for set:ITS
+  public:
+                    AliITS();
+                    AliITS(const char *name, const char *title);
+    virtual        ~AliITS();
+    AliITS(AliITS &source);
+    AliITS & operator=(AliITS &source);
+
+    virtual void   AddHit(Int_t track, Int_t *vol, Float_t *hits);
+    virtual void   AddRealDigit(Int_t branch, Int_t *digits);
+    virtual void   AddDigit(Int_t branch, AliITSdigit *d);
+    virtual void   AddDigit(Int_t branch,Float_t phys,Int_t* digits, Int_t* tracks, Float_t* trkcharges); 
+    //
+    virtual void   AddCluster(Int_t branch, AliITSRawCluster *c);
+    virtual void   AddRecPoint(const AliITSRecPoint &p);
+    virtual void   AddTrack(Float_t* tracks) {}
+//    virtual void   AddTrack(const AliITStrack&) {} 
+
+    virtual void   ResetDigits();                   // depending on how the
+    virtual void   ResetDigits(Int_t branch);       // tree will be filled only
+    virtual void   ResetClusters();                 // one of the methods in 
+    virtual void   ResetClusters(Int_t branch);     // the pair will be kept
+    virtual void   ResetRecPoints();
+
+    virtual void   BuildGeometry() {}
+    virtual void   CreateGeometry() {}
+    virtual void   CreateMaterials() {}
+
+    void GetGeometryVersion(Int_t &a,Int_t &b) const 
+                  {a = fMajorVersion;b=fMinorVersion;return;}
+    virtual Int_t  IsVersion() const {return 1;}
+            Int_t  DistancetoPrimitive(Int_t px, Int_t py);
+    virtual void   Init();
+    virtual void   SetDefaults();
+    // create separate tree for clusters - declustering refining
+    virtual  void  MakeTreeC(Option_t *option="C");
+    void           GetTreeC(Int_t event);
+    virtual void   MakeBranch(Option_t *opt=" ");
+    void           SetTreeAddress();
+    virtual void   SetEUCLID(Bool_t euclid=1) {fEuclidOut = euclid;}
+    virtual void   StepManager() {}
+    // sort hits by module
+    virtual void   InitModules(Int_t size,Int_t &nmodules);  
+    virtual void   FillModules(Int_t evnt,Int_t bgrev,Int_t lastev,
+                       Int_t nmodules,Option_t *opt,Text_t *filename);
+    virtual void   ClearModules();
+    // Digitisation
+    void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t lastev,Int_t size,
+                 Option_t *add, Option_t *det, Text_t *filename);
+    // Reconstruct hits
+    void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det);
+    // Fast simulation of space points from hits
+    void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t lastev,Int_t size,
+                 Option_t *add, Option_t *det, Text_t *filename);
+
+
+
+    // Write digits into raw data format
+    virtual void Digits2RawData() {}
+
+    // Configuration Methods (per detector type )
+    // Set number of detector types
+    virtual void   SetNDetTypes(Int_t ndtypes=3) {fNDetTypes=ndtypes;}
+    // Set response 
+    virtual void   SetResponseModel(Int_t id, AliITSresponse *response);
+    // Set segmentation 
+    virtual void   SetSegmentationModel(Int_t id, AliITSsegmentation *seg);
+    // Set simulation - temporary 
+    virtual void   SetSimulationModel(Int_t id, AliITSsimulation *sim);
+    // Set reconstruction 
+    virtual void   SetReconstructionModel(Int_t id, AliITSClusterFinder *rec);
+    // Set class names for digit and rec point 
+    virtual void   SetClasses(Int_t id, char* digit, char* cluster);
+
+
+    // Getters
+    // ITS geometry functions
+    virtual AliITSgeom  *GetITSgeom() const {return fITSgeom;}
+    virtual TObjArray   *GetModules() const {return fITSmodules;}
+    AliITSmodule *GetModule(Int_t index) {return (AliITSmodule *)
+                                             (fITSmodules->At(index));}
+
+    // Return pointers to digits 
+    TObjArray            *Dtype() {return fDtype;}
+    Int_t                *Ndtype() {return fNdtype;}
+    virtual TClonesArray *DigitsAddress(Int_t id) 
+                   {return ((TClonesArray *) (*fDtype)[id]);}
+    // Return pointers to clusters 
+    TObjArray            *Ctype() {return fCtype;}
+    Int_t                *Nctype() {return fNctype;}
+    virtual TClonesArray *ClustersAddress(Int_t id) 
+                   {return ((TClonesArray *) (*fCtype)[id]);}
+
+    // Return pointer to rec points 
+    TClonesArray  *RecPoints()   {return fRecPoints;}
+    // Return pointer to tracks 
+    TClonesArray  *Tracks()   {return fTracks;}
+
+    // Return pointer to DetType #id
+    AliITSDetType  *DetType(Int_t id);
+    Int_t           NDetTypes() {return fNDetTypes;}
+
+    // Return pointer to the tree of clusters
+    TTree          *TreeC() {return fTreeC;}
+
+ protected:
+
+    AliITSgeom  *fITSgeom;                   // Pointer to ITS geometry
+    TObjArray   *fITSmodules;                // Pointer to ITS modules
+
+    Bool_t fEuclidOut;                       // Flag to write out geometry 
+                                             //in euclid format
+    Int_t  fIdN;                             // the number of layers
+    Int_t  *fIdSens;                         //[fIdN] layer identifier
+    char **fIdName;                          //layer identifier
+    // Geometry and Stepmanager version numbers used.
+    Int_t fMajorVersion,fMinorVersion;       //detailed and coarse(minor) versions
+    //
+    //
+    TObjArray            *fDetTypes;           // List of detector types
+    Int_t                 fNDetTypes;          // Number of Detector types
+
+    TObjArray            *fDtype;              // List of digits
+    Int_t                *fNdtype;             //[fNDetTypes] Number of digits per type of
+                                               // detector 
+    TObjArray            *fCtype;              // List of clusters
+    Int_t                *fNctype;             //[fNDetTypes] Number of clusters per type
+                                               // of detector
+
+
+    TClonesArray         *fRecPoints;          // List of reconstructed points
+    Int_t                 fNRecPoints;         // Number of rec points
+    TClonesArray         *fTracks;             // List of reconstructed tracks
+    TTree                *fTreeC;              // Tree for raw clusters
+
+
+    ClassDef(AliITS,1) // Base class for ITS
 };
+
 #endif