From fedacced5387aed3d4941aea33ab84dbbd0a7835 Mon Sep 17 00:00:00 2001
From: sunwen <sunwen@realcan.cn>
Date: Tue, 24 Oct 2023 13:09:17 +0800
Subject: [PATCH] Fix imageExtrapolation.

---
 .../preprocessData/imageExtrapolation.cpp     | 24 +++++++------------
 1 file changed, 9 insertions(+), 15 deletions(-)

diff --git a/src/reflectionReconstruction/preprocessData/imageExtrapolation.cpp b/src/reflectionReconstruction/preprocessData/imageExtrapolation.cpp
index 4b2455b..a4f8db8 100644
--- a/src/reflectionReconstruction/preprocessData/imageExtrapolation.cpp
+++ b/src/reflectionReconstruction/preprocessData/imageExtrapolation.cpp
@@ -30,24 +30,18 @@ Matrix Recon::imageExtrapolation(const Matrix& aImg, int aDim)
 
         if(!indexe.empty())
         {
-            Matrix result = aImg;
-            std::vector<int> lowerZIndexe;
-            for(int i=0; i<=indexe[0] - 1; ++i)
+            Matrix result = aImg;       
+            if(indexe[0] - 1 > 0)
             {
-                lowerZIndexe.push_back(i);
-            }
-            for(int i=0; i<lowerZIndexe.size(); ++i)
-            {
-                Matrix insert = repmat(aImg($,$,indexe[0]).toMatrix(), 1, 1, sum(zIndexZero.block(0, 0, *lowerZIndexe.end()), FunctionDirection::All)[0]);
+                Matrix insert = repmat(aImg($,$,indexe[0]).toMatrix(), 1, 1, indexe[0]);    
+                result.setBlock(2, 0, indexe[0] - 1, insert);
             }
 
-            // lowerZIndexe.clear();
-            // for(size_t i=0; i<=*indexe.end(); ++i)
-            // {
-            //     lowerZIndexe.push_back(i);
-            // }
-            Matrix insert = repmat(aImg($,$,*(indexe.end() - 1)).toMatrix(), 1, 1, sum(zIndexZero, FunctionDirection::All)[0]);
-            result.setBlock(2, *(indexe.end()-1) + 1, zSize-1, insert);
+            if(*(indexe.end()-1) + 1 < zSize-1)
+            {
+                Matrix insert = repmat(aImg($,$,*(indexe.end() - 1)).toMatrix(), 1, 1, zSize - *(indexe.end() - 1) - 1);
+                result.setBlock(2, *(indexe.end()-1) + 1, zSize-1, insert);
+            }
             return result;
         }
     }