Taking into account the new signature of AddHitKartez

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

diff --git a/TRD/AliTRDtrackHits.cxx b/TRD/AliTRDtrackHits.cxx
index 5986b800828c2ebf7b64c5dfd8c8b099566c63c6..a585a192624aa0f2d59f7f4f8c272c2ddac51b7b 100644 (file)
--- a/TRD/AliTRDtrackHits.cxx
+++ b/TRD/AliTRDtrackHits.cxx
@@ -23,7 +23,7 @@
 #include "TClonesArray.h"    
 #include "AliTRDhit.h"    
 
-#include <iostream.h>
+#include <Riostream.h>
 
 
 ClassImp(AliTRDtrackHits)
@@ -37,7 +37,7 @@ void  AliTRDtrackHits::AddHitTRD(Int_t volumeID, Int_t trackID, Double_t x,
 
   if (inDrift) q=2*q+1;
   else q=2*q;
-  AddHitKartez(volumeID, trackID,x,y,z,q);
+  AddHitKartez(volumeID, trackID,x,y,z,q,0);
 }
 
 Bool_t AliTRDtrackHits::First()
@@ -45,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;
@@ -80,9 +86,9 @@ Bool_t AliTRDtrackHits::Next()
   fCurrentHit->fStackIndex++;
 
   AliTrackHitsParamV2 *param =  (AliTrackHitsParamV2 *)fArray->At(fCurrentHit->fParamIndex);
-  if (fCurrentHit->fStackIndex>=((UInt_t) param->fNHits)){
+  if (fCurrentHit->fStackIndex>= param->fNHits){
     fCurrentHit->fParamIndex++;
-    if (fCurrentHit->fParamIndex>=((UInt_t) fArray->GetEntriesFast())){
+    if (fCurrentHit->fParamIndex>= fArray->GetEntriesFast()){
       fCurrentHit->fStatus=kFALSE;
       return kFALSE;
     }