STEER/AliAODCluster.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2007, 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 //     AOD cluster base class
  20 //     Author: Markus Oldenburg, CERN
  21 //-------------------------------------------------------------------------
  22
  23 #include "AliAODCluster.h"
  24
  25 ClassImp(AliAODCluster)
  26
  27 //______________________________________________________________________________
  28 AliAODCluster::AliAODCluster() :
  29   AliVCluster(),
  30   fEnergy(0),
  31   fChi2(-999.),
  32   fID(-999),
  33   fNLabel(0),
  34   fLabel(0x0),
  35   fFilterMap(0),
  36   fType(kUndef)
  37 {
  38   // default constructor
  39
  40   SetPosition(NULL);
  41   SetPID(NULL);
  42 }
  43
  44 //______________________________________________________________________________
  45 AliAODCluster::AliAODCluster(Int_t id,
  46                              UInt_t nLabel,
  47                              Int_t *label,
  48                              Double_t energy,
  49                              Double_t x[3],
  50                              Double_t pid[13],
  51                              Char_t ttype,
  52                              UInt_t selectInfo) :
  53   AliVCluster(),
  54   fEnergy(energy),
  55   fChi2(-999.),
  56   fID(id),
  57   fNLabel(0),
  58   fLabel(0x0),
  59   fFilterMap(selectInfo),
  60   fType(ttype)
  61 {
  62   // constructor
  63   for (Int_t i = 0; i <  3; i++) fPosition[i] = 0.;
  64   for (Int_t i = 0; i < 13; i++) fPID[i]      = 0;
  65
  66   if(x)   {for (Int_t i = 0; i < 3  ; i++) SetPositionAt(x[i]  ,i);}
  67   if(pid) {for (Int_t i = 0; i < 13 ; i++) SetPIDAt     (pid[i],i);}
  68   SetLabel(label, nLabel);
  69 }
  70
  71 //______________________________________________________________________________
  72 AliAODCluster::AliAODCluster(Int_t id,
  73                              UInt_t nLabel,
  74                              Int_t *label,
  75                              Float_t energy,
  76                              Float_t x[3],
  77                              Float_t pid[13],
  78                              Char_t ttype,
  79                              UInt_t selectInfo) :
  80   AliVCluster(),
  81   fEnergy(energy),
  82   fChi2(-999.),
  83   fID(id),
  84   fNLabel(0),
  85   fLabel(0x0),
  86   fFilterMap(selectInfo),
  87   fType(ttype)
  88 {
  89   // constructor
  90   for (Int_t i = 0; i <  3; i++) fPosition[i] = 0.;
  91   for (Int_t i = 0; i < 13; i++) fPID[i]      = 0;
  92
  93   if(x)   {for (Int_t i = 0; i < 3  ; i++) SetPositionAt(x[i]  ,i);}
  94   if(pid) {for (Int_t i = 0; i < 13 ; i++) SetPIDAt     (pid[i],i);}
  95   SetLabel(label, nLabel);
  96 }
  97
  98
  99 //______________________________________________________________________________
 100 AliAODCluster::~AliAODCluster()
 101 {
 102   // destructor
 103
 104   RemoveLabel();
 105 }
 106
 107 //______________________________________________________________________________
 108 void AliAODCluster::Clear(const Option_t*)
 109 {
 110   // Clear
 111
 112   RemoveLabel();
 113 }
 114
 115
 116 //______________________________________________________________________________
 117 AliAODCluster::AliAODCluster(const AliAODCluster& clus) :
 118   AliVCluster(clus),
 119   fEnergy(clus.fEnergy),
 120   fChi2(clus.fChi2),
 121   fID(clus.fID),
 122   fNLabel(0),
 123   fLabel(0x0),
 124   fFilterMap(clus.fFilterMap),
 125   fType(clus.fType)
 126 {
 127   // Copy constructor
 128
 129   for(Int_t i = 0; i < 3  ; i++) fPosition[i]  = clus.fPosition[i];
 130   for(Int_t i = 0; i < 13 ; i++)  fPID[i]      = clus.fPID[i];
 131
 132   SetLabel(clus.fLabel, clus.fNLabel);
 133 }
 134
 135 //______________________________________________________________________________
 136 AliAODCluster& AliAODCluster::operator=(const AliAODCluster& clus)
 137 {
 138   // Assignment operator
 139   if(this!=&clus) {
 140
 141         for(Int_t i = 0; i < 3 ;  i++) fPosition[i] = clus.fPosition[i];
 142         for(Int_t i = 0; i < 13 ; i++) fPID[i]      = clus.fPID[i];
 143
 144     fEnergy = clus.fEnergy;
 145     fChi2 = clus.fChi2;
 146
 147     fID = clus.fID;
 148     SetLabel(clus.fLabel, clus.fNLabel);
 149     fFilterMap = clus.fFilterMap;
 150
 151     fType = clus.fType;
 152   }
 153
 154   return *this;
 155 }
 156
 157 //______________________________________________________________________________
 158 void AliAODCluster::SetPosition(Float_t *x)
 159 {
 160   // set the position
 161
 162   if (x) {
 163     fPosition[0] = x[0];
 164     fPosition[1] = x[1];
 165     fPosition[2] = x[2];
 166   } else {
 167     fPosition[0] = -999.;
 168     fPosition[1] = -999.;
 169     fPosition[2] = -999.;
 170   }
 171 }
 172
 173 //______________________________________________________________________________
 174 UShort_t AliAODCluster::GetMostProbablePID() const
 175 {
 176   // Returns the most probable PID array element.
 177
 178   Int_t nPID = 13;
 179   UShort_t unknown = AliVCluster::kUnknown;
 180
 181   UShort_t loc = unknown;
 182   Double_t max = 0.;
 183   Bool_t allTheSame = kTRUE;
 184
 185   for (Int_t iPID = 0; iPID < nPID; iPID++) {
 186     if (fPID[iPID] >= max) {
 187       if (fPID[iPID] > max) {
 188         allTheSame = kFALSE;
 189         max = fPID[iPID];
 190         loc = (UShort_t)iPID;
 191       } else {
 192         allTheSame = kTRUE;
 193       }
 194     }
 195   }
 196   return allTheSame ? unknown : loc;
 197 }
 198
 199 //______________________________________________________________________________
 200 void AliAODCluster::SetLabel(Int_t *label, UInt_t size)
 201 {
 202   if (label && size>0) {
 203     if (size != (UInt_t)fNLabel) {
 204       RemoveLabel();
 205       fNLabel = size;
 206       fLabel = new Int_t[fNLabel];
 207     }
 208
 209     for (Int_t i = 0; i < fNLabel; i++) {
 210       fLabel[i] = label[i];
 211     }
 212   } else {
 213     RemoveLabel();
 214   }
 215
 216   return;
 217 }
 218
 219 //______________________________________________________________________________
 220 Int_t AliAODCluster::GetLabelAt(UInt_t i) const
 221 {
 222   if (fLabel && i < (UInt_t)fNLabel) {
 223     return fLabel[i];
 224   } else {
 225     return -999;
 226   }
 227 }
 228
 229 //______________________________________________________________________________
 230 void AliAODCluster::RemoveLabel()
 231 {
 232   delete[] fLabel;
 233   fLabel = 0x0;
 234   fNLabel = 0;
 235
 236   return;
 237 }
 238
 239 //______________________________________________________________________________
 240 void AliAODCluster::Print(Option_t* /* option */) const
 241 {
 242   // prints information about AliAODCluster
 243
 244   printf("Cluster type: %d\n", GetType());
 245   printf("     energy = %f\n", E());
 246   printf("       chi2 = %f\n", Chi2());
 247   const Double_t *pid = GetPID();
 248   printf("PID weights: photon %0.2f, pi0 %0.2f, electron %0.2f, conversion electron %0.2f\n, hadrons: pion %0.2f, kaon %0.2f, proton %0.2f , neutron %0.2f, kaon %0.2f \n",
 249          pid[AliVCluster::kPhoton],   pid[AliVCluster::kPi0],
 250          pid[AliVCluster::kElectron], pid[AliVCluster::kEleCon],
 251          pid[AliVCluster::kPion],     pid[AliVCluster::kKaon],   pid[AliVCluster::kProton],
 252          pid[AliVCluster::kNeutron],  pid[AliVCluster::kKaon0]);
 253 }