Refs #5416 compare iReal to getNumberHistograms

modified: FFT.cpp
mantidproject · May 24, 2012 · 6b03919 · 6b03919
1 parent a03380e
commit 6b03919
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Code/Mantid/Framework/Algorithms/src/FFT.cpp b/Code/Mantid/Framework/Algorithms/src/FFT.cpp
@@ -181,12 +181,14 @@ void FFT::exec()
   const int ySize = static_cast<int>(inWS->blocksize());
   const int xSize = static_cast<int>(X.size());
 
-  if(iReal >= ySize) throw std::invalid_argument("Property Real is out of range");
+  int nHist = static_cast<int>(inWS->getNumberHistograms());
+  if( iReal >= nHist ) throw std::invalid_argument("Property Real is out of range");
   if( isComplex )
   {
     const int yImagSize = static_cast<int>(inImagWS->blocksize());
     if( ySize != yImagSize ) throw std::length_error("Real and Imaginary sizes do not match");
-    if( iImag >= ySize ) throw std::invalid_argument("Property Imaginary is out of range");
+    nHist = static_cast<int>(inImagWS->getNumberHistograms());
+    if( iImag >= nHist ) throw std::invalid_argument("Property Imaginary is out of range");
   }
 
   //Check that the x values are evenly spaced