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(AliCalorimeter& c);               // Copy constructor
  29   virtual TObject* Clone(char* name="");           // Make a deep copy and provide pointer of the copy
  30   Int_t GetNrows();                                // Return number of rows of the matrix
  31   Int_t GetNcolumns();                             // 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); // Provide signal of a certain module
  38   Int_t GetNsignals();                             // 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();                            // Return number of clusters
  41   Float_t GetClusteredSignal(Int_t row,Int_t col); // Provide module signal after clustering
  42   AliCalcluster* GetCluster(Int_t j);              // Access to cluster number j
  43   AliCalmodule* GetModule(Int_t j);                // Access to 'fired' module number j
  44   AliCalmodule* GetModule(Int_t row,Int_t col);    // 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);         // 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);         // 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);            // Provide the gain value of a module
  54   Int_t GetGainFlag(Int_t row,Int_t col);          // Provide the gain flag value of a module
  55   Float_t GetOffset(Int_t row,Int_t col);          // Provide the offset value of a module
  56   Int_t GetOffsetFlag(Int_t row,Int_t col);        // 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); // Return module position
  60   AliPosition* GetPosition(Int_t row,Int_t col);   // Access to module position
  61   TH2F* DrawModules(Float_t thresh=0.,Int_t mode=0); // Lego plot of module (corrected) 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);               // Access to veto signal number j
  66   Int_t GetNvetos();                               // 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();                             // 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,6) // Description of a modular calorimeter system.
  87 };
  88 #endif