Make code compliant to coding conventions
[u/mrichter/AliRoot.git] / TRD / AliTRDsegmentArray.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 /*
17 $Log$
18 Revision 1.2  2000/05/08 16:17:27  cblume
19 Merge TRD-develop
20
21 Revision 1.1.4.1  2000/05/08 14:55:03  cblume
22 Bug fixes
23
24 Revision 1.1  2000/02/28 19:02:32  cblume
25 Add new TRD classes
26
27 */
28
29 ///////////////////////////////////////////////////////////////////////////////
30 //                                                                           //
31 //  Alice segment manager class                                              //
32 //                                                                           //
33 ///////////////////////////////////////////////////////////////////////////////
34
35 #include "AliTRD.h"
36 #include "AliTRDgeometry.h"
37 #include "AliTRDsegmentArray.h"
38
39 ClassImp(AliTRDsegmentArray)
40
41 //_____________________________________________________________________________
42 AliTRDsegmentArray::AliTRDsegmentArray():AliTRDsegmentArrayBase()
43 {
44   //
45   // Default constructor
46   //
47
48 }
49
50 //_____________________________________________________________________________
51 AliTRDsegmentArray::AliTRDsegmentArray(Text_t *classname, Int_t n)
52                    :AliTRDsegmentArrayBase(classname,n)
53 {
54   //
55   // Constructor creating an array of AliTRDdataArray of size <n>
56   //
57
58   AliTRDdataArray *dataArray;  
59
60   for (Int_t i = 0; i < n; i++) {
61     dataArray = (AliTRDdataArray *) AddSegment(i);
62   }
63
64 }
65
66 //_____________________________________________________________________________
67 AliTRDsegmentArray::AliTRDsegmentArray(AliTRDsegmentArray &a)
68 {
69   //
70   // AliTRDsegmentArray copy constructor
71   //
72
73   a.Copy(*this);
74
75 }
76
77 //_____________________________________________________________________________
78 AliTRDsegmentArray::~AliTRDsegmentArray()
79 {
80   //
81   // AliTRDsegmentArray destructor
82   //
83 }
84
85 //_____________________________________________________________________________
86 void AliTRDsegmentArray::Copy(AliTRDsegmentArray &a)
87 {
88   //
89   // Copy function
90   //
91
92   AliTRDsegmentArrayBase::Copy(a);
93
94 }
95
96 //_____________________________________________________________________________
97 void AliTRDsegmentArray::Delete()
98 {
99   //
100   // Deletes all detector segments from the array
101   //
102
103   for (Int_t iDet = 0; iDet < fNSegment; iDet++) {
104     ClearSegment(iDet);
105   }
106
107 }
108
109 //_____________________________________________________________________________
110 Bool_t AliTRDsegmentArray::LoadArray(const Char_t *branchname)
111 {
112   //
113   // Loads all segments of the array from the branch <branchname> of
114   // the digits tree
115   //
116
117   // Connect the digits tree
118   fTree = gAlice->TreeD();
119   if (!fTree) return kFALSE;
120
121   // Get the branch
122   fBranch = fTree->GetBranch(branchname);
123   if (!fBranch) return kFALSE;
124
125   // Loop through all segments and read them from the tree
126   Bool_t status = kTRUE;
127   for (Int_t iSegment = 0; iSegment < fNSegment; iSegment++) {
128     AliTRDdataArray *dataArray = (AliTRDdataArray *) fSegment->At(iSegment);
129     if (!dataArray) {
130       status = kFALSE;
131       break;    
132     }
133     fBranch->SetAddress(&dataArray);
134     fBranch->GetEntry(iSegment);
135   }
136
137   return status;
138
139 }
140
141 //_____________________________________________________________________________
142 Bool_t AliTRDsegmentArray::StoreArray(const Char_t *branchname)
143 {
144   //
145   // Stores all segments of the array in the branch <branchname> of 
146   // the digits tree
147   //
148
149   // Connect the digits tree
150   fTree = gAlice->TreeD();
151   if (!fTree) return kFALSE;
152
153   // Get the branch
154   fBranch = fTree->GetBranch(branchname);
155   if (!fBranch) return kFALSE;
156
157   // Loop through all segments and fill them into the tree
158   Bool_t status = kTRUE;
159   for (Int_t iSegment = 0; iSegment < fNSegment; iSegment++) {
160     const AliTRDdataArray *kDataArray = 
161          (AliTRDdataArray *) AliTRDsegmentArrayBase::At(iSegment);
162     if (!kDataArray) {
163       status = kFALSE;
164       break;
165     }
166     fBranch->SetAddress(&kDataArray);
167     fBranch->Fill();
168   }
169
170   return status;
171
172 }
173
174 //_____________________________________________________________________________
175 AliTRDdataArray *AliTRDsegmentArray::GetDataArray(Int_t det)
176 {
177   //
178   // Returns the data array for a given detector
179   //
180
181   return ((AliTRDdataArray *) AliTRDsegmentArrayBase::At(det));
182
183 }
184
185 //_____________________________________________________________________________
186 AliTRDdataArray *AliTRDsegmentArray::GetDataArray(Int_t pla, Int_t cha, Int_t sec)
187 {
188   //
189   // Returns the data array for a given detector
190   //
191
192   if (gAlice) {
193
194     AliTRDgeometry *geo = ((AliTRD*) gAlice->GetDetector("TRD"))->GetGeometry();  
195     Int_t det = geo->GetDetector(pla,cha,sec);
196     return GetDataArray(det);
197
198   }
199   else {
200
201     printf("AliTRDsegmentArray::GetDigits -- ");
202     printf("gAlice is not defined\n");
203     return NULL;
204
205   }
206
207 }