Bugfix for calculation of cluster widths.

[u/mrichter/AliRoot.git] / TRD / AliTRDtrackHits.cxx

diff --git a/TRD/AliTRDtrackHits.cxx b/TRD/AliTRDtrackHits.cxx
index 63b582e16f0f0da3189cb872a4ed285a73a0d16b..696e687ba173964d5375e41c14341a4ecb3d41cc 100644 (file)
--- a/TRD/AliTRDtrackHits.cxx
+++ b/TRD/AliTRDtrackHits.cxx
@@ -13,11 +13,17 @@
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
+////////////////////////////////////////////////
+//                                            //
+//  Manager class for TRD   hits              //
+//                                            //
+////////////////////////////////////////////////
+
 #include "AliTRDtrackHits.h"
 #include "TClonesArray.h"    
 #include "AliTRDhit.h"    
 
-#include <iostream.h>
+#include <Riostream.h>
 
 
 ClassImp(AliTRDtrackHits)
@@ -25,6 +31,10 @@ ClassImp(AliTRDtrackHits)
 void  AliTRDtrackHits::AddHitTRD(Int_t volumeID, Int_t trackID, Double_t x, 
                    Double_t y, Double_t z,Int_t q, Bool_t inDrift)
 {
+  //
+  // Add one TRD hit
+  //
+
   if (inDrift) q=2*q+1;
   else q=2*q;
   AddHitKartez(volumeID, trackID,x,y,z,q);
@@ -35,6 +45,12 @@ Bool_t AliTRDtrackHits::First()
   //
   //set Current hit for the first hit
   //
+
+  if (fArray->GetSize()<=0) {
+    fCurrentHit->fStatus = kFALSE;
+    return kFALSE;
+  }
+
   AliTrackHitsParamV2 *param = (AliTrackHitsParamV2 *)fArray->At(0);
   if (!fHit) 
     fHit = new AliTRDhit;
@@ -70,9 +86,9 @@ Bool_t AliTRDtrackHits::Next()
   fCurrentHit->fStackIndex++;
 
   AliTrackHitsParamV2 *param =  (AliTrackHitsParamV2 *)fArray->At(fCurrentHit->fParamIndex);
-  if (fCurrentHit->fStackIndex>=param->fNHits){
+  if (fCurrentHit->fStackIndex>= param->fNHits){
     fCurrentHit->fParamIndex++;
-    if (fCurrentHit->fParamIndex>=fArray->GetEntriesFast()){
+    if (fCurrentHit->fParamIndex>= fArray->GetEntriesFast()){
       fCurrentHit->fStatus=kFALSE;
       return kFALSE;
     }