prettier logs

[u/mrichter/AliRoot.git] / ANALYSIS / AliPIDResponseInputHandler.cxx

diff --git a/ANALYSIS/AliPIDResponseInputHandler.cxx b/ANALYSIS/AliPIDResponseInputHandler.cxx
index dd9e6de62d6c3b98009ecf8bbc90dde548d3ae2d..f4ad8e7acb7aaf486df5dddb5e226f336b2704fa 100644 (file)
--- a/ANALYSIS/AliPIDResponseInputHandler.cxx
+++ b/ANALYSIS/AliPIDResponseInputHandler.cxx
@@ -208,11 +208,13 @@ void AliPIDResponseInputHandler::SetRecoInfo()
   
   //find pass from file name (UGLY, but not stored in ESD... )
   TString fileName(file->GetName());
-  if (fileName.Contains("/pass1")) {
+  if (fileName.Contains("/pass1") || fileName.Contains(".pass1")) {
     fRecoPass=1;
-  } else if (fileName.Contains("/pass2")) {
+  } else if (fileName.Contains("/pass2") || fileName.Contains(".pass2")) {
     fRecoPass=2;
-  } else if (fileName.Contains("/pass3")) {
+  } else if (fileName.Contains("/pass3") || fileName.Contains(".pass3")) {
     fRecoPass=3;
   }
+
+  fPIDResponse->SetCurrentFile(fileName.Data());
 }