diff options
| -rw-r--r-- | src/Core/regularisers_CPU/FGP_TV_core.c | 27 |
1 files changed, 6 insertions, 21 deletions
diff --git a/src/Core/regularisers_CPU/FGP_TV_core.c b/src/Core/regularisers_CPU/FGP_TV_core.c index eddefd6..69c92bc 100644 --- a/src/Core/regularisers_CPU/FGP_TV_core.c +++ b/src/Core/regularisers_CPU/FGP_TV_core.c @@ -191,30 +191,15 @@ 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-1; j++) { - for(i=0; i<dimX-1; i++) { - index = j*dimX+i; + for(j=0; j<dimY; j++) { + for(i=0; i<dimX; i++) { + index = j*dimX+i; /* boundary conditions */ - 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]; + 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]; 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) |