Minor fix for the pid efficiency calculation + cosmetics in the drawProtonQAResults...

[u/mrichter/AliRoot.git] / HLT / TPCLib / tracking-ca / AliHLTTPCCATrackParam.h

//-*- Mode: C++ -*-
// $Id$
// ************************************************************************
// This file is property of and copyright by the ALICE HLT Project        * 
// ALICE Experiment at CERN, All rights reserved.                         *
// See cxx source for full Copyright notice                               *
//                                                                        *
//*************************************************************************


#ifndef ALIHLTTPCCATRACKPARAM_H
#define ALIHLTTPCCATRACKPARAM_H

#include "AliHLTTPCCADef.h"


/**
 * @class AliHLTTPCCATrackParam
 *
 * AliHLTTPCCATrackParam class describes the track parametrisation
 * which is used by the AliHLTTPCCATracker slice tracker.
 *
 */
class AliHLTTPCCATrackParam
{
 public:

  struct AliHLTTPCCATrackFitParam
  {
    Float_t fBethe, fE,fTheta2, fEP2, fSigmadE2, fK22,fK33,fK43,fK44;// parameters
  };

  GPUd() Float_t X()     const { return fX;    }
  GPUd() Float_t Y()     const { return fP[0]; }
  GPUd() Float_t Z()     const { return fP[1]; }
  GPUd() Float_t SinPhi()const { return fP[2]; }
  GPUd() Float_t DzDs()  const { return fP[3]; }
  GPUd() Float_t Kappa() const { return fP[4]; }
  GPUd() Float_t CosPhi()const { return fCosPhi; }
  GPUd() Float_t Chi2()  const { return fChi2; }
  GPUd() Int_t   NDF()   const { return fNDF; }

  Float_t Err2Y()      const { return fC[0]; }
  Float_t Err2Z()      const { return fC[2]; }
  Float_t Err2SinPhi() const { return fC[5]; }
  Float_t Err2DzDs()   const { return fC[9]; }
  Float_t Err2Kappa()  const { return fC[14]; }

  GPUd() Float_t GetX()      const { return fX; }
  GPUd() Float_t GetY()      const { return fP[0]; }
  GPUd() Float_t GetZ()      const { return fP[1]; }
  GPUd() Float_t GetSinPhi() const { return fP[2]; }
  GPUd() Float_t GetDzDs()   const { return fP[3]; }
  GPUd() Float_t GetKappa()  const { return fP[4]; }
  GPUd() Float_t GetCosPhi() const { return fCosPhi; }
  GPUd() Float_t GetChi2()   const { return fChi2; }
  GPUd() Int_t   GetNDF()    const { return fNDF; }

  GPUd() Float_t GetErr2Y()      const { return fC[0]; }
  GPUd() Float_t GetErr2Z()      const { return fC[2]; }
  GPUd() Float_t GetErr2SinPhi() const { return fC[5]; }
  GPUd() Float_t GetErr2DzDs()   const { return fC[9]; }
  GPUd() Float_t GetErr2Kappa()  const { return fC[14]; }

  GPUhd() const Float_t *Par() const { return fP; }
  GPUhd() const Float_t *Cov() const { return fC; }

  const Float_t *GetPar() const { return fP; }
  const Float_t *GetCov() const { return fC; }

  GPUhd() void SetPar( Int_t i, Float_t v ){ fP[i] = v; }
  GPUhd() void SetCov( Int_t i, Float_t v ){ fC[i] = v; }

  GPUd() void SetX( Float_t v )     {  fX = v;    }
  GPUd() void SetY( Float_t v )     {  fP[0] = v; }
  GPUd() void SetZ( Float_t v )     {  fP[1] = v; }
  GPUd() void SetSinPhi( Float_t v ){  fP[2] = v; }
  GPUd() void SetDzDs( Float_t v )  {  fP[3] = v; }
  GPUd() void SetKappa( Float_t v ) {  fP[4] = v; }
  GPUd() void SetCosPhi( Float_t v ){  fCosPhi = v; }
  GPUd() void SetChi2( Float_t v )  {  fChi2 = v; }
  GPUd() void SetNDF( Int_t v )   { fNDF = v; }  


  GPUd() Float_t GetDist2( const AliHLTTPCCATrackParam &t ) const;
  GPUd() Float_t GetDistXZ2( const AliHLTTPCCATrackParam &t ) const;

  GPUd() void ConstructXY3( const Float_t x[3], const Float_t y[3], const Float_t sigmaY2[3], Float_t CosPhi0 );

  GPUd() void ConstructXYZ3( const Float_t p0[5], const Float_t p1[5], const Float_t p2[5], 
			      Float_t CosPhi0, Float_t t0[]=0 );

  GPUd() Float_t GetS( Float_t x, Float_t y ) const;

  GPUd() void GetDCAPoint( Float_t x, Float_t y, Float_t z,
			   Float_t &px, Float_t &py, Float_t &pz ) const;
  
  GPUd() Int_t TransportToX( Float_t X, Float_t maxSinPhi );
  GPUd() Int_t TransportToX( Float_t X, AliHLTTPCCATrackParam &t0, Float_t maxSinPhi );

  GPUd() Bool_t TransportToXWithMaterial( Float_t X, AliHLTTPCCATrackFitParam &par );
  GPUd() Bool_t TransportToXWithMaterial( Float_t X, AliHLTTPCCATrackParam &t0, AliHLTTPCCATrackFitParam &par );
  GPUd() Bool_t TransportToXWithMaterial( Float_t X, Float_t Bz );
  GPUd() Bool_t Rotate( Float_t alpha, Float_t maxSinPhi=.99 );
  GPUd() Bool_t RotateNoCos( Float_t alpha, AliHLTTPCCATrackParam &t0, Float_t maxSinPhi=.99 );

  GPUd() Bool_t Filter2( Float_t y, Float_t z, Float_t err2Y, Float_t err2Z, Float_t maxSinPhi=.99 );
  GPUd() Bool_t Filter2NoCos( Float_t y, Float_t z, Float_t err2Y, Float_t err2Z );

  GPUd() Int_t TransportToX0( Float_t X, Float_t sinPhi, Float_t cosPhi );
  GPUd() Int_t TransportToX0( Float_t X, Float_t maxSinPhi );
  GPUd() Bool_t Filter20( Float_t y, Float_t z, Float_t err2Y, Float_t err2Z, Float_t maxSinPhi=.99 );
  GPUd() Bool_t Filter200( Float_t y, Float_t z, Float_t err2Y, Float_t err2Z );

  GPUd() Bool_t Filter2v1( Float_t y, Float_t z, Float_t err2Y, Float_t err2Z, Float_t maxSinPhi=.99 );
  GPUd() void FilterY( Float_t y, Float_t erry );
  GPUd() void FilterZ( Float_t z, Float_t errz );

  GPUd() static Float_t ApproximateBetheBloch( Float_t beta2 );
  GPUd() void CalculateFitParameters( AliHLTTPCCATrackFitParam &par, Float_t Bz, Float_t mass = 0.13957 );
  GPUd() Bool_t CorrectForMeanMaterial( Float_t xOverX0,  Float_t xTimesRho, const AliHLTTPCCATrackFitParam &par );
  GPUd() void Print() const;

private:

  Float_t fX;      // x position
  Float_t fCosPhi; // cosPhi
  Float_t fP[5];   // 'active' track parameters: Y, Z, SinPhi, DzDs, Kappa
  Float_t fC[15];  // the covariance matrix for Y,Z,SinPhi,..
  Float_t fChi2;   // the chi^2 value
  Int_t   fNDF;    // the Number of Degrees of Freedom
};


#endif