added a protection
[u/mrichter/AliRoot.git] / STEER / AliAODTracklets.cxx
CommitLineData
21b22f32 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
26ClassImp(AliAODTracklets)
27
28AliAODTracklets::AliAODTracklets() : TNamed(), fNTracks(0), fTheta(0), fPhi(0), fDeltaPhi(0), fLabels(0)
29{
30 // default constructor
31}
32
33AliAODTracklets::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
5c1dc41f 38AliAODTracklets::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
59AliAODTracklets& AliAODTracklets::operator=(const AliAODTracklets& tracklet)
60{
61// Assignment operator
62 if(&tracklet == this) return *this;
63 TNamed::operator=(tracklet);
64 fNTracks = tracklet.fNTracks;
786172af 65 for (Int_t i = 0; i < fNTracks; i++) {
5c1dc41f 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 }
786172af 71 return *this;
5c1dc41f 72}
73
21b22f32 74void AliAODTracklets::CreateContainer(Int_t nTracks)
75{
76 // function that creates container to store tracklets
77
78 DeleteContainer();
79
80 fNTracks = nTracks;
81
f51b5257 82 if (fNTracks <= 0) {
83 fNTracks = 0;
21b22f32 84 return;
f51b5257 85 }
21b22f32 86
d59deed5 87 fTheta = new Double32_t[fNTracks];
88 fPhi = new Double32_t[fNTracks];
89 fDeltaPhi = new Double32_t[fNTracks];
21b22f32 90 fLabels = new Int_t[fNTracks];
91}
92
5c1dc41f 93
21b22f32 94AliAODTracklets::~AliAODTracklets()
95{
96 // destructor
97
98 DeleteContainer();
99}
100
101void 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
d59deed5 132Bool_t AliAODTracklets::SetTracklet(Int_t pos, Double32_t theta, Double32_t phi, Double32_t deltaPhi, Int_t label)
21b22f32 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}