04-apr-2004 NvE SetMass() invoked from AliTrack::Set3Momentum to get also the energy...
[u/mrichter/AliRoot.git] / RALICE / AliCalorimeter.h
1 #ifndef ALICALORIMETER_H
2 #define ALICALORIMETER_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 // $Id$
7
8 #include <math.h>
9  
10 #include "TNamed.h"
11 #include "TObjArray.h"
12 #include "TH2.h"
13 #include "TString.h"
14
15 #include "AliObjMatrix.h"
16 #include "AliCalmodule.h"
17 #include "AliCalcluster.h"
18 #include "AliSignal.h"
19 #include "AliPositionObj.h"
20 #include "AliAttribObj.h"
21  
22 class AliCalorimeter : public TNamed
23 {
24  public:
25   AliCalorimeter();                                      // Default constructor
26   AliCalorimeter(Int_t nrow,Int_t ncol);                 // Create a calorimeter matrix
27   virtual ~AliCalorimeter();                             // Destructor
28   AliCalorimeter(const AliCalorimeter& c);               // Copy constructor
29   virtual TObject* Clone(const char* name="") const;     // Make a deep copy and provide pointer of the copy
30   Int_t GetNrows() const;                                // Return number of rows of the matrix
31   Int_t GetNcolumns() const;                             // Return number of columns of the matrix
32   void SetSignal(Int_t row,Int_t col,Float_t s);         // Set signal for a certain module
33   void AddSignal(Int_t row,Int_t col,Float_t s);         // Add signal to a certain module
34   void AddSignal(AliCalmodule* m);                       // Add module signal to current calorimeter
35   void Reset(Int_t row,Int_t col);                       // Reset signal for a certain module
36   void Reset(Int_t mode=0);                              // Reset the complete calorimeter
37   Float_t GetSignal(Int_t row,Int_t col,Int_t mode=0) const; // Provide signal of a certain module
38   Int_t GetNsignals() const;                             // Return number of modules with a signal
39   void Group(Int_t n=1,Int_t mode=1);                    // Group modules into clusters (n rings)
40   Int_t GetNclusters() const;                            // Return number of clusters
41   Float_t GetClusteredSignal(Int_t row,Int_t col) const; // Provide module signal after clustering
42   AliCalcluster* GetCluster(Int_t j) const;              // Access to cluster number j
43   AliCalmodule* GetModule(Int_t j) const;                // Access to 'fired' module number j
44   AliCalmodule* GetModule(Int_t row,Int_t col) const;    // Access to module at (row,col)
45   void SetEdgeOn(Int_t row,Int_t col);                   // Indicate module as 'edge module'
46   void SetEdgeOff(Int_t row,Int_t col);                  // Indicate module as 'non-edge module'
47   Int_t GetEdgeValue(Int_t row,Int_t col) const;         // Provide the edge flag of a module
48   void SetDead(Int_t row,Int_t col);                     // Indicate module as 'dead module'
49   void SetAlive(Int_t row,Int_t col);                    // Indicate module as 'active module'
50   Int_t GetDeadValue(Int_t row,Int_t col) const;         // Provide the dead flag of a module
51   void SetGain(Int_t row,Int_t col,Float_t g);           // Set the gain value for a module
52   void SetOffset(Int_t row,Int_t col,Float_t o);         // Set the offset value for a module
53   Float_t GetGain(Int_t row,Int_t col) const;            // Provide the gain value of a module
54   Int_t GetGainFlag(Int_t row,Int_t col) const;          // Provide the gain flag value of a module
55   Float_t GetOffset(Int_t row,Int_t col) const;          // Provide the offset value of a module
56   Int_t GetOffsetFlag(Int_t row,Int_t col) const;        // Provide the offset flag value of a module
57   void SetPosition(Int_t row,Int_t col,Float_t* r,TString f); // Set module position
58   void SetPosition(Int_t row,Int_t col,Ali3Vector& r);   // Set module position
59   void GetPosition(Int_t row,Int_t col,Float_t* r,TString f) const; // Return module position
60   AliPosition* GetPosition(Int_t row,Int_t col) const;   // Access to module position
61   TH2F* DrawModules(Float_t thresh=0.,Int_t mode=0);     // Lego plot of module (corr.) signals above threshold
62   TH2F* DrawClusters(Float_t thresh=0.);                 // Lego plot of cluster signals above threshold
63   void AddVetoSignal(AliSignal& s);                      // Associate (extrapolated) signal
64   void AddVetoSignal(AliSignal* s) { AddVetoSignal(*s); }
65   AliSignal* GetVetoSignal(Int_t j) const;               // Access to veto signal number j
66   Int_t GetNvetos() const;                               // Provide the number of veto signals
67   void SetSwapMode(Int_t swap=1);                        // Set the swapmode for the storage of the matrices
68   Int_t GetSwapMode() const;                             // Provide the swapmode for the storage of the matrices
69  
70  protected:
71   Int_t fNrows;                              // The number of rows
72   Int_t fNcolumns;                           // The number of columns
73   AliObjMatrix* fMatrix;                     // The matrix of module pointers
74   Int_t fSwap;                               // The swapmode for the module and position matrices
75   void SortM(AliCalmodule** a,Int_t& n);     // Order the modules with decreasing signal (matrix search)
76   void SortA(AliCalmodule** a,Int_t& n);     // Order the modules with decreasing signal (fired array search)
77   TObjArray* fClusters;                      // The array of clusters
78   void AddRing(Int_t row,Int_t col,Int_t n); // add signals of n rings around cluster center
79   void Ungroup();                            // Restore module matrix as before clustering
80   TH2F* fHmodules;                           //! The module 2-D histogram for event display
81   TH2F* fHclusters;                          //! The cluster 2-D histogram for event display
82   TObjArray* fVetos;                         // The array of associated (extrapolated) veto signals
83   TObjArray* fAttributes;                    //! Matrix with module attributes (e.g. gain, offset etc...)
84   AliObjMatrix* fPositions;                  //! The matrix of module position pointers
85  
86  ClassDef(AliCalorimeter,8) // Description of a modular calorimeter system.
87 };
88 #endif