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 "AliCluster.h"
  28 #include "AliKalmanTrack.h"
  29 #include "AliRun.h"
  30 #include "AliMagF.h"
  31
  32 #include "TFile.h"
  33 #include "TError.h"
  34
  35
  36 extern AliRun* gAlice;
  37
  38 ClassImp(AliTracker)
  39
  40 //__________________________________________________________________________
  41 void AliTracker::CookLabel(AliKalmanTrack *t, Float_t wrong) const {
  42   //--------------------------------------------------------------------
  43   //This function "cooks" a track label. If label<0, this track is fake.
  44   //--------------------------------------------------------------------
  45   Int_t noc=t->GetNumberOfClusters();
  46   Int_t *lb=new Int_t[noc];
  47   Int_t *mx=new Int_t[noc];
  48   AliCluster **clusters=new AliCluster*[noc];
  49
  50   Int_t i;
  51   for (i=0; i<noc; i++) {
  52      lb[i]=mx[i]=0;
  53      Int_t index=t->GetClusterIndex(i);
  54      clusters[i]=GetCluster(index);
  55   }
  56
  57   Int_t lab=123456789;
  58   for (i=0; i<noc; i++) {
  59     AliCluster *c=clusters[i];
  60     lab=TMath::Abs(c->GetLabel(0));
  61     Int_t j;
  62     for (j=0; j<noc; j++) if (lb[j]==lab || mx[j]==0) break;
  63     lb[j]=lab;
  64     (mx[j])++;
  65   }
  66
  67   Int_t max=0;
  68   for (i=0; i<noc; i++) if (mx[i]>max) {max=mx[i]; lab=lb[i];}
  69
  70   max=0;
  71   for (i=0; i<noc; i++) {
  72     AliCluster *c=clusters[i];
  73     if (TMath::Abs(c->GetLabel(0)) == lab ||
  74         TMath::Abs(c->GetLabel(1)) == lab ||
  75         TMath::Abs(c->GetLabel(2)) == lab ) max++;
  76   }
  77
  78   if ((1.- Float_t(max)/noc) > wrong) lab=-lab;
  79   t->SetLabel(lab);
  80
  81   delete[] lb;
  82   delete[] mx;
  83   delete[] clusters;
  84 }
  85
  86 //____________________________________________________________________________
  87 void AliTracker::UseClusters(const AliKalmanTrack *t, Int_t from) const {
  88   //------------------------------------------------------------------
  89   //This function marks clusters associated with the track.
  90   //------------------------------------------------------------------
  91   Int_t noc=t->GetNumberOfClusters();
  92   for (Int_t i=from; i<noc; i++) {
  93      Int_t index=t->GetClusterIndex(i);
  94      AliCluster *c=GetCluster(index);
  95      c->Use();
  96   }
  97 }
  98
  99 ////////////////////////////////////////////////////////////////////////
 100 Int_t AliTracker::SetFieldFactor() {
 101 //
 102 // Utility class to set the value of the magnetic field in the barrel
 103 // It supposes that the correct object gAlice is in the memory
 104 //
 105    AliKalmanTrack::
 106       SetConvConst(1000/0.299792458/gAlice->Field()->SolenoidField());
 107    Double_t field=gAlice->Field()->SolenoidField();
 108    ::Info("SetFieldFactor","Magnetic field in kGauss: %f\n",field);
 109    return 0;
 110 }
 111 ////////////////////////////////////////////////////////////////////////
 112 Int_t AliTracker::SetFieldFactor(TFile *file, Bool_t deletegAlice) {
 113 //
 114 // Utility class to set the value of the magnetic field in the barrel
 115 // gAlice object is read from the file, and optionally deleted
 116 //
 117   if (!(gAlice=(AliRun*)file->Get("gAlice"))) {
 118    ::Warning
 119    ("SetFieldFactor","gAlice has not been found in file %s\n",file->GetName());
 120     return 1;
 121   }
 122   Int_t rc = SetFieldFactor();
 123   if (deletegAlice) {
 124     delete gAlice;
 125     gAlice = 0;
 126   }
 127   return rc;
 128 }
 129 ////////////////////////////////////////////////////////////////////////
 130 Int_t AliTracker::SetFieldFactor(const char* fileName, Bool_t closeFile) {
 131 //
 132 // Utility class to set the value of the magnetic field in the barrel
 133 // gAlice object is read from the file, the file is optionally closed
 134 //
 135    TFile *file=TFile::Open(fileName);
 136    if (!file->IsOpen()) {
 137       ::Warning("AliTracker::SetFieldFactor","Cannnot open %s !\n",fileName);
 138       return 1;
 139    }
 140    Int_t rc = SetFieldFactor(file, closeFile) ;
 141    if (closeFile) file->Close();
 142    return rc;
 143 }
 144 ////////////////////////////////////////////////////////////////////////
 145