From 663ad11e5539a764d0679f88ca1e5fb39c1f4511 Mon Sep 17 00:00:00 2001
From: sunwen <sunwen@realcan.com>
Date: Thu, 25 May 2023 16:23:16 +0800
Subject: [PATCH] Update removeDataFromArrays to pass unitTest.

---
 .../removeDataFromArrays.cpp                  | 56 +++++++++----------
 1 file changed, 28 insertions(+), 28 deletions(-)

diff --git a/src/common/dataBlockCreation/removeDataFromArrays.cpp b/src/common/dataBlockCreation/removeDataFromArrays.cpp
index 776ede3..9cc3d95 100644
--- a/src/common/dataBlockCreation/removeDataFromArrays.cpp
+++ b/src/common/dataBlockCreation/removeDataFromArrays.cpp
@@ -11,36 +11,36 @@ using namespace Aurora;
 
 Aurora::Matrix Recon::removeDataFromArrays(const Aurora::Matrix& aOutPutMatrix, const Aurora::Matrix& aRemoveList)
 {
-    // Matrix result = aOutPutMatrix;
-    // int minusNum = 0;
-    // for(int i=0; i<aRemoveList.getDataSize(); ++i)
-    // {
-    //     if(aRemoveList[i] == 0)
-    //     {
-    //         result = deleteColumn(result, i - minusNum);
-    //         ++minusNum;
-    //     }       
-    // }
-
-    // return result;
-
-    size_t columnSize = sum(aRemoveList, FunctionDirection::All)[0];
-    double* resultData = Aurora::malloc(columnSize * aOutPutMatrix.getDimSize(0));
-    Matrix result = Matrix::New(resultData, aOutPutMatrix.getDimSize(0), columnSize);
-    std::vector<std::pair<int,int>> copyList;
-    size_t length = 0;
-    for(size_t i=0; i<aRemoveList.getDataSize(); ++i)
+    Matrix result = aOutPutMatrix;
+    int minusNum = 0;
+    for(int i=0; i<aRemoveList.getDataSize(); ++i)
     {
-        if(aRemoveList[i] == 1)
+        if(aRemoveList[i] == 0)
         {
-            ++length;
-        }
-
-        if(aRemoveList[i] == 0 && length != 0)
-        {
-            copyList.push_back({});
-            length = 0;
-        }
+            result = deleteColumn(result, i - minusNum);
+            ++minusNum;
+        }       
     }
 
+    return result;
+
+    // size_t columnSize = sum(aRemoveList, FunctionDirection::All)[0];
+    // double* resultData = Aurora::malloc(columnSize * aOutPutMatrix.getDimSize(0));
+    // Matrix result = Matrix::New(resultData, aOutPutMatrix.getDimSize(0), columnSize);
+    // std::vector<std::pair<int,int>> copyList;
+    // size_t length = 0;
+    // for(size_t i=0; i<aRemoveList.getDataSize(); ++i)
+    // {
+    //     if(aRemoveList[i] == 1)
+    //     {
+    //         ++length;
+    //     }
+
+    //     if(aRemoveList[i] == 0 && length != 0)
+    //     {
+    //         copyList.push_back({});
+    //         length = 0;
+    //     }
+    // }
+
 }
\ No newline at end of file