RALICE/AliSample.h

   1 #ifndef ALISAMPLE_H
   2 #define ALISAMPLE_H
   3 ///////////////////////////////////////////////////////////////////////////
   4 // Class AliSample
   5 // Perform statistics on various multi-dimensional data samples
   6 // A data sample can be filled using the "Enter" and/or "Remove" functions,
   7 // whereas the "Reset" function resets the complete sample to 'empty'.
   8 // The info which can be extracted from a certain data sample are the
   9 // sum, mean, variance, sigma, covariance and correlation.
  10 // The "Info" function provides all statistics data for a certain sample.
  11 // The variables for which these stat. parameters have to be calculated
  12 // are indicated by the index of the variable which is passed as an
  13 // argument to the various member functions.
  14 // The index convention for a data point (x,y) is : x=1  y=2
  15 //
  16 // Example :
  17 // ---------
  18 // For an AliSample s a data point (x,y) can be entered as s.Enter(x,y) and
  19 // the mean_x can be obtained as s.GetMean(1) whereas the mean_y is obtained
  20 // via s.GetMean(2).
  21 // The correlation between x and y is available via s.GetCor(1,2).
  22 // The x-statistics are obtained via s.Info(1), y-statistics via s.Info(2),
  23 // and the covariance and correlation between x and y via s.Info(1,2).
  24 // All statistics of a sample are obtained via s.Info().
  25 //
  26 //--- NvE 30-mar-1996 CERN Geneva
  27 ///////////////////////////////////////////////////////////////////////////
  28
  29 #include <math.h>
  30 #include <iostream.h>
  31
  32 #include "Rtypes.h"
  33
  34 class AliSample
  35 {
  36  public:
  37   AliSample();                                  // Default constructor
  38   virtual ~AliSample();                         // Default destructor
  39   void Reset();                                 // Reset complete statistics
  40   void Enter(Float_t x);                        // Enter value for 1-dim. sample
  41   void Remove(Float_t x);                       // Remove value from 1-dim. sample
  42   void Enter(Float_t x, Float_t y);             // Enter value for 2-dim. sample
  43   void Remove(Float_t x, Float_t y);            // Remove value from 2-dim. sample
  44   void Enter(Float_t x, Float_t y, Float_t z);  // Enter value for 3-dim. sample
  45   void Remove(Float_t x, Float_t y, Float_t z); // Remove value from 3-dim. sample
  46   Int_t GetDimension();                         // Provide dimension of the sample
  47   Int_t GetN();                                 // Provide the number of entries
  48   Float_t GetSum(Int_t i);                      // Provide sum for i-th variable
  49   Float_t GetMean(Int_t i);                     // Provide mean for i-th variable
  50   Float_t GetVar(Int_t i);                      // Provide variance for i-th variable
  51   Float_t GetSigma(Int_t i);                    // Standard deviation for i-th variable
  52   Float_t GetCov(Int_t i, Int_t j);             // Covariance for i-th and j-th variable
  53   Float_t GetCor(Int_t i, Int_t j);             // Correlation for i-th and j-th variable
  54   void Info();                                  // Stat. info for the complete sample
  55   void Info(Int_t i);                           // Stat. info for the i-th variable
  56   void Info(Int_t i, Int_t j);                  // Stat. info for i-th and j-th variable
  57
  58  private:
  59   Int_t fDim;                      // Dimension of the sample
  60   Int_t fN;                        // Number of entries of the sample
  61   enum  {fMaxdim=3};               // Maximum supported dimension
  62   char  fNames[fMaxdim];           // Variable names i.e. X,Y,Z
  63   Float_t fSum[fMaxdim];           // Total sum for each variable
  64   Float_t fSum2[fMaxdim][fMaxdim]; // Total sum**2 for each variable
  65   void  Compute();                 // Compute the various quantities
  66   Float_t fMean[fMaxdim];          // Mean for each variable
  67   Float_t fVar[fMaxdim];           // Variation for each variable
  68   Float_t fSigma[fMaxdim];         // Standard deviation for each variable
  69   Float_t fCov[fMaxdim][fMaxdim];  // Covariances of pairs of variables
  70   Float_t fCor[fMaxdim][fMaxdim];  // Correlations of pairs of variables
  71 };
  72 #endif