deal with boundary outside loop (2D)

1 files changed, 21 insertions, 6 deletions

diff --git a/src/Core/regularisers_CPU/FGP_TV_core.c b/src/Core/regularisers_CPU/FGP_TV_core.c
index 69c92bc..eddefd6 100644
--- a/src/Core/regularisers_CPU/FGP_TV_core.c
+++ b/src/Core/regularisers_CPU/FGP_TV_core.c
@@ -191,15 +191,30 @@ float Grad_func2D(float *P1, float *P2, float *D, float *R1, float *R2, float la
     long i,j,index;
     multip = (1.0f/(8.0f*lambda));
 #pragma omp parallel for shared(P1,P2,D,R1,R2,multip) private(index,i,j,val1,val2)
-    for(j=0; j<dimY; j++) {
-        for(i=0; i<dimX; i++) {
-			index = j*dimX+i;
+    for(j=0; j<dimY-1; j++) {
+        for(i=0; i<dimX-1; i++) {
+            index = j*dimX+i;
             /* boundary conditions */
-            if (i == dimX-1) val1 = 0.0f; else val1 = D[index] - D[j*dimX + (i+1)];
-            if (j == dimY-1) val2 = 0.0f; else val2 = D[index] - D[(j+1)*dimX + i];
+            if (i == dimX-1) val1 = 0.0f; else val1 = D[index] - D[index+1];
+            if (j == dimY-1) val2 = 0.0f; else val2 = D[index] - D[index + dimX];
             P1[index] = R1[index] + multip*val1;
             P2[index] = R2[index] + multip*val2;
-        }}
+        }
+    }
+
+    /* boundary conditions */
+    i = dimX - 1 ;
+    for(j=0; j<dimY; j++) {
+        index = j*dimX+i;
+        P1[index] = R1[index];
+    }
+    j = dimY - 1;
+    for (i=0;i<dimX; i++)
+    {
+        index = j*dimX+i;
+        P2[index] = R2[index];
+
+    }
     return 1;
 }
 float Proj_func2D(float *P1, float *P2, int methTV, long DimTotal)