STEER/AliTracker.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 //-------------------------------------------------------------------------
  19 //               Implementation of the AliTracker class
  20 //  that is the base for AliTPCtracker, AliITStrackerV2 and AliTRDtracker
  21 //        Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
  22 //-------------------------------------------------------------------------
  23
  24 #include <TMath.h>
  25
  26 #include "AliTracker.h"
  27 #include "AliKalmanTrack.h"
  28 #include "AliCluster.h"
  29 #include "AliLog.h"
  30 #include "AliRun.h"
  31 #include "AliMagF.h"
  32
  33 const AliMagF *AliTracker::fgkFieldMap=0;
  34
  35 ClassImp(AliTracker)
  36
  37 AliTracker::AliTracker():
  38   fX(0),
  39   fY(0),
  40   fZ(0),
  41   fSigmaX(0.005),
  42   fSigmaY(0.005),
  43   fSigmaZ(0.010)
  44 {
  45   //--------------------------------------------------------------------
  46   // The default constructor.
  47   //--------------------------------------------------------------------
  48   if (!fgkFieldMap) AliWarning("Field map is not set. Call AliTracker::SetFieldMap before creating a tracker!");
  49 }
  50
  51 void AliTracker::SetFieldMap(const AliMagF* map, Bool_t uni) {
  52   //--------------------------------------------------------------------
  53   //This passes the field map to the reconstruction.
  54   //--------------------------------------------------------------------
  55   if (map==0) AliFatalClass("Can't access the field map !");
  56   AliKalmanTrack::SetFieldMap(map);
  57   if (uni) AliKalmanTrack::SetUniformFieldTracking();
  58   fgkFieldMap=map;
  59 }
  60
  61 //__________________________________________________________________________
  62 void AliTracker::CookLabel(AliKalmanTrack *t, Float_t wrong) const {
  63   //--------------------------------------------------------------------
  64   //This function "cooks" a track label. If label<0, this track is fake.
  65   //--------------------------------------------------------------------
  66   Int_t noc=t->GetNumberOfClusters();
  67   Int_t *lb=new Int_t[noc];
  68   Int_t *mx=new Int_t[noc];
  69   AliCluster **clusters=new AliCluster*[noc];
  70
  71   Int_t i;
  72   for (i=0; i<noc; i++) {
  73      lb[i]=mx[i]=0;
  74      Int_t index=t->GetClusterIndex(i);
  75      clusters[i]=GetCluster(index);
  76   }
  77
  78   Int_t lab=123456789;
  79   for (i=0; i<noc; i++) {
  80     AliCluster *c=clusters[i];
  81     lab=TMath::Abs(c->GetLabel(0));
  82     Int_t j;
  83     for (j=0; j<noc; j++) if (lb[j]==lab || mx[j]==0) break;
  84     lb[j]=lab;
  85     (mx[j])++;
  86   }
  87
  88   Int_t max=0;
  89   for (i=0; i<noc; i++) if (mx[i]>max) {max=mx[i]; lab=lb[i];}
  90
  91   for (i=0; i<noc; i++) {
  92     AliCluster *c=clusters[i];
  93     //if (TMath::Abs(c->GetLabel(1)) == lab ||
  94     //    TMath::Abs(c->GetLabel(2)) == lab ) max++;
  95     if (TMath::Abs(c->GetLabel(0)!=lab))
  96         if (TMath::Abs(c->GetLabel(1)) == lab ||
  97             TMath::Abs(c->GetLabel(2)) == lab ) max++;
  98   }
  99
 100   if ((1.- Float_t(max)/noc) > wrong) lab=-lab;
 101   t->SetFakeRatio((1.- Float_t(max)/noc));
 102   t->SetLabel(lab);
 103
 104   delete[] lb;
 105   delete[] mx;
 106   delete[] clusters;
 107 }
 108
 109 //____________________________________________________________________________
 110 void AliTracker::UseClusters(const AliKalmanTrack *t, Int_t from) const {
 111   //------------------------------------------------------------------
 112   //This function marks clusters associated with the track.
 113   //------------------------------------------------------------------
 114   Int_t noc=t->GetNumberOfClusters();
 115   for (Int_t i=from; i<noc; i++) {
 116      Int_t index=t->GetClusterIndex(i);
 117      AliCluster *c=GetCluster(index);
 118      c->Use();
 119   }
 120 }