HLT/MUON/OnlineAnalysis/AliHLTMUONCalculations.h

   1 #ifndef ALIHLTMUONCALCULATIONS_H
   2 #define ALIHLTMUONCALCULATIONS_H
   3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 ////////////////////////////////////////////////////////////////////////////////
   9 //
  10 // Author: Artur Szostak
  11 // Email:  artur@alice.phy.uct.ac.za | artursz@iafrica.com
  12 //
  13 ////////////////////////////////////////////////////////////////////////////////
  14
  15 #include "AliHLTMUONDataTypes.h"
  16
  17 extern "C" struct AliHLTMUONTriggerRecordStruct;
  18
  19 /*
  20  * Note: this class uses static global variables so thread protection must be
  21  * explicit in any multi-threaded usage. Or the class should be rewritten.
  22  */
  23 class AliHLTMUONCalculations
  24 {
  25 public:
  26
  27         /// Calculates the momentum estimate given two track points behind the
  28         /// dipole magnet and assuming origin is the interaction point.
  29         static bool ComputeMomentum(
  30                         AliHLTFloat32_t x1,
  31                         AliHLTFloat32_t y1, AliHLTFloat32_t y2,
  32                         AliHLTFloat32_t z1, AliHLTFloat32_t z2
  33                 );
  34
  35         static AliHLTFloat32_t Zf() { return fgZf; }
  36         static void Zf(AliHLTFloat32_t value) { fgZf = value; }
  37         static AliHLTFloat32_t QBL();
  38         static void QBL(AliHLTFloat32_t value);
  39
  40         static AliHLTMUONParticleSign Sign() { return fgSign; }
  41         static AliHLTFloat32_t Px() { return fgPx; }
  42         static AliHLTFloat32_t Py() { return fgPy; }
  43         static AliHLTFloat32_t Pz() { return fgPz; }
  44
  45         /// Calculates the invariant mass for a pair of particles.
  46         static AliHLTFloat32_t ComputeMass(
  47                         AliHLTFloat32_t massA,
  48                         AliHLTFloat32_t pxA,
  49                         AliHLTFloat32_t pyA,
  50                         AliHLTFloat32_t pzA,
  51                         AliHLTFloat32_t massB,
  52                         AliHLTFloat32_t pxB,
  53                         AliHLTFloat32_t pyB,
  54                         AliHLTFloat32_t pzB
  55                 );
  56
  57         static bool FitLineToTriggerRecord(const AliHLTMUONTriggerRecordStruct& trigger);
  58
  59         static bool FitLineToTriggerRecord(
  60                         const AliHLTMUONTriggerRecordStruct& trigger,
  61                         const bool hitset[4]
  62                 );
  63
  64         static AliHLTFloat32_t IdealZ1() { return fgIdealZ1; }
  65         static void IdealZ1(AliHLTFloat32_t value) { fgIdealZ1 = value; }
  66         static AliHLTFloat32_t IdealZ2() { return fgIdealZ2; }
  67         static void IdealZ2(AliHLTFloat32_t value) { fgIdealZ2 = value; }
  68
  69         static AliHLTFloat32_t IdealX1() { return fgIdealX1; }
  70         static AliHLTFloat32_t IdealY1() { return fgIdealY1; }
  71         static AliHLTFloat32_t IdealX2() { return fgIdealX2; }
  72         static AliHLTFloat32_t IdealY2() { return fgIdealY2; }
  73
  74         static bool FitLineToData(
  75                         const AliHLTFloat32_t* x, const AliHLTFloat32_t* y,
  76                         const AliHLTFloat32_t* z, AliHLTUInt32_t n
  77                 );
  78
  79         static AliHLTFloat32_t Mzx() { return fgMzx; }
  80         static AliHLTFloat32_t Mzy() { return fgMzy; }
  81         static AliHLTFloat32_t Czx() { return fgCzx; }
  82         static AliHLTFloat32_t Czy() { return fgCzy; }
  83
  84         static AliHLTFloat32_t ComputeChi2(
  85                         const AliHLTFloat32_t* x, const AliHLTFloat32_t* y,
  86                         const AliHLTFloat32_t* z, AliHLTUInt32_t n
  87                 );
  88
  89         static AliHLTFloat32_t SigmaX2() { return fgSigmaX2; }
  90         static void SigmaX2(AliHLTFloat32_t value) { fgSigmaX2 = (value != 0 ? value : 1.); }
  91         static AliHLTFloat32_t SigmaY2() { return fgSigmaY2; }
  92         static void SigmaY2(AliHLTFloat32_t value) { fgSigmaY2 = (value != 0 ? value : 1.); }
  93
  94 private:
  95
  96         // Prevent destroying or creating of this object.
  97         AliHLTMUONCalculations();
  98         ~AliHLTMUONCalculations();
  99
 100         static AliHLTFloat32_t fgZf;  /// The Z coordinate of the middle of the dipole magnetic field.
 101         static AliHLTFloat32_t fgQBLScaled; /// The integrated field strength times units of charge (T.m.*c/1e9)
 102
 103         static AliHLTMUONParticleSign fgSign;  /// The calculated sign.
 104         static AliHLTFloat32_t fgPx;  /// The calculated X momentum (GeV/c).
 105         static AliHLTFloat32_t fgPy;  /// The calculated Y momentum (GeV/c).
 106         static AliHLTFloat32_t fgPz;  /// The calculated Z momentum (GeV/c).
 107
 108         static AliHLTFloat32_t fgSigmaX2;  /// The sigma squared value for the variance / uncertainty in X coordinates.
 109         static AliHLTFloat32_t fgSigmaY2;  /// The sigma squared value for the variance / uncertainty in Y coordinates.
 110
 111         static AliHLTFloat32_t fgMzx;  /// Calculated slope of the line fitted to the ZX plane. (x = fgMzx * z + fgCzx)
 112         static AliHLTFloat32_t fgMzy;  /// Calculated slope of the line fitted to the ZY plane. (y = fgMzy * z + fgCzy)
 113         static AliHLTFloat32_t fgCzx;  /// Calculated coefficient of the line fitted to the ZX plane. (x = fgMzx * z + fgCzx)
 114         static AliHLTFloat32_t fgCzy;  /// Calculated coefficient of the line fitted to the ZY plane. (y = fgMzy * z + fgCzy)
 115
 116         static AliHLTFloat32_t fgIdealX1;  /// Ideal X coordinate of the point on MT1
 117         static AliHLTFloat32_t fgIdealY1;  /// Ideal Y coordinate of the point on MT1
 118         static AliHLTFloat32_t fgIdealZ1;  /// Ideal Z coordinate of the point on MT1
 119         static AliHLTFloat32_t fgIdealX2;  /// Ideal X coordinate of the point on MT2
 120         static AliHLTFloat32_t fgIdealY2;  /// Ideal Y coordinate of the point on MT2
 121         static AliHLTFloat32_t fgIdealZ2;  /// Ideal Z coordinate of the point on MT2
 122 };
 123
 124 #endif // ALIHLTMUONCALCULATIONS_H