STEER/AliAODTracklets.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 class to store tracklets
  20 //     Author: Jan Fiete Grosse-Oetringhaus, CERN
  21 //     Class created from AliMultiplicity
  22 //-------------------------------------------------------------------------
  23
  24 #include "AliAODTracklets.h"
  25
  26 ClassImp(AliAODTracklets)
  27
  28 AliAODTracklets::AliAODTracklets() : TNamed(), fNTracks(0), fTheta(0), fPhi(0), fDeltaPhi(0), fLabels(0)
  29 {
  30   // default constructor
  31 }
  32
  33 AliAODTracklets::AliAODTracklets(const char* name, const char* title) : TNamed(name, title), fNTracks(0), fTheta(0), fPhi(0), fDeltaPhi(0), fLabels(0)
  34 {
  35   // TNamed constructor
  36 }
  37
  38 AliAODTracklets::AliAODTracklets(const AliAODTracklets& tracklet) :
  39     TNamed(tracklet),
  40     fNTracks(tracklet.fNTracks),
  41     fTheta(0),
  42     fPhi(0),
  43     fDeltaPhi(0),
  44     fLabels(0)
  45 {
  46 // Copy constructor
  47     fTheta = new Double32_t[fNTracks];
  48     fPhi = new Double32_t[fNTracks];
  49     fDeltaPhi = new Double32_t[fNTracks];
  50     fLabels = new Int_t[fNTracks];
  51     for (Int_t i = 0; i < fNTracks; i++) {
  52         fTheta[i]    = tracklet.fTheta[i];
  53         fPhi[i]      = tracklet.fPhi[i];
  54         fDeltaPhi[i] = tracklet.fDeltaPhi[i];
  55         fLabels[i]   = tracklet.fLabels[i];
  56     }
  57 }
  58
  59 AliAODTracklets& AliAODTracklets::operator=(const AliAODTracklets& tracklet)
  60 {
  61 // Assignment operator
  62     if(&tracklet == this) return *this;
  63     TNamed::operator=(tracklet);
  64     fNTracks = tracklet.fNTracks;
  65     for (Int_t i = 0; i < fNTracks; i++) {
  66         fTheta[i]    = tracklet.fTheta[i];
  67         fPhi[i]      = tracklet.fPhi[i];
  68         fDeltaPhi[i] = tracklet.fDeltaPhi[i];
  69         fLabels[i]   = tracklet.fLabels[i];
  70     }
  71     return *this;
  72 }
  73
  74 void AliAODTracklets::CreateContainer(Int_t nTracks)
  75 {
  76   // function that creates container to store tracklets
  77
  78   DeleteContainer();
  79
  80   fNTracks = nTracks;
  81
  82   if (fNTracks <= 0) {
  83     fNTracks = 0;
  84     return;
  85   }
  86
  87   fTheta = new Double32_t[fNTracks];
  88   fPhi = new Double32_t[fNTracks];
  89   fDeltaPhi = new Double32_t[fNTracks];
  90   fLabels = new Int_t[fNTracks];
  91 }
  92
  93
  94 AliAODTracklets::~AliAODTracklets()
  95 {
  96   // destructor
  97
  98   DeleteContainer();
  99 }
 100
 101 void AliAODTracklets::DeleteContainer()
 102 {
 103   // deletes allocated memory
 104
 105   if (fTheta)
 106   {
 107     delete[] fTheta;
 108     fTheta = 0;
 109   }
 110
 111   if (fPhi)
 112   {
 113     delete[] fPhi;
 114     fPhi = 0;
 115   }
 116
 117   if (fDeltaPhi)
 118   {
 119     delete[] fDeltaPhi;
 120     fDeltaPhi = 0;
 121   }
 122
 123   if (fLabels)
 124   {
 125     delete[] fLabels;
 126     fLabels = 0;
 127   }
 128
 129   fNTracks = 0;
 130 }
 131
 132 Bool_t AliAODTracklets::SetTracklet(Int_t pos, Double32_t theta, Double32_t phi, Double32_t deltaPhi, Int_t label)
 133 {
 134   // Sets a tracklet at the given position
 135
 136   if (pos < 0 || pos >= fNTracks)
 137     return kFALSE;
 138
 139   fTheta[pos] = theta;
 140   fPhi[pos] = phi;
 141   fDeltaPhi[pos] = deltaPhi;
 142   fLabels[pos] = label;
 143
 144   return kTRUE;
 145 }