Update of tracking code provided by Sergei

[u/mrichter/AliRoot.git] / TRD / AliTRDmcTrack.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  TRD MC track                                                             //
//  Used for efficiency estimates and matching of reconstructed tracks       //
//  to MC particles                                                          //                    
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include "AliTRDmcTrack.h"
#include "AliTRDgeometry.h"

ClassImp(AliTRDmcTrack)

//_____________________________________________________________________________
AliTRDmcTrack::AliTRDmcTrack() 
{ 

  //
  // Default constructor 
  //

  fLab = -1; 
  fPrimary = kFALSE; 
  fMass = 0; 
  fCharge = 0;
  fN = 0; 
  
  for(Int_t i=0; i<200; i++) fIndex[i]=-1;
  
  for(Int_t i=0; i<6; i++) {
    for(Int_t j=0; j<3; j++) { 
      fPin[i][j]=0.; 
      fPout[i][j] = 0.;
    }
  }

}

//_____________________________________________________________________________
AliTRDmcTrack::AliTRDmcTrack(Int_t label, Bool_t primary, Float_t mass
			    ,Int_t charge, Int_t pdg) 
{ 

  //
  // Main constructor 
  //

  fLab = label; 
  fPrimary = primary; 
  fMass = mass; 
  fCharge = charge;
  fPDG = pdg;
  fN = 0; 
  
  for(Int_t i=0; i<200; i++) fIndex[i]=-1;
  
  for(Int_t i=0; i<6; i++) {
    for(Int_t j=0; j<3; j++) { 
      fPin[i][j]=0.; 
      fPout[i][j] = 0.;
    }
  }

}

//_____________________________________________________________________________
void AliTRDmcTrack::GetPxPyPz(Double_t& px, Double_t& py, Double_t& pz, 
                              Int_t opt) const 
{
  //
  // Returns momentum components at the entrance (opt >= 0), or
  // exit (opt < 0) of TRD. 
  //

  Int_t i;

  if(opt >= 0) {
    for(i = 0; i < AliTRDgeometry::Nplan(); i++) {
      if(  fPin[i][0] * fPin[i][0]
         + fPin[i][1] * fPin[i][1]
         + fPin[i][2] * fPin[i][2] > 0.0005) break;
    }
    px = fPin[0][0]; py = fPin[0][1]; pz = fPin[0][2];
  }
  else {
    for(i = AliTRDgeometry::Nplan() - 1; i >= 0; i--) {
      if(  fPout[i][0] * fPout[i][0]
         + fPout[i][1] * fPout[i][1]
         + fPout[i][2] * fPout[i][2] > 0.0005) break;
    }
    px = fPout[i][0]; py = fPout[i][1]; pz = fPout[i][2];
  }

  return;

}

//_____________________________________________________________________________
void AliTRDmcTrack::GetPlanePxPyPz(Double_t& px, Double_t& py, Double_t& pz, 
                                   Int_t plane, Int_t opt) const 
{
  //
  // Returns momentum components at the entrance (opt >= 0), or
  // exit (opt < 0) of TRD plane <plane>. 
  //

  if(opt >= 0) {
    px = fPin[plane][0]; py = fPin[plane][1]; pz = fPin[plane][2];
  }
  else {
    px = fPout[plane][0]; py = fPout[plane][1]; pz = fPout[plane][2];
  }

  return;

}