for #222

rspatial · Dec 7, 2021 · d49f975 · d49f975
1 parent de8faaf
commit d49f975
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 15 deletions.
diff --git a/src/focal_fun.cpp b/src/focal_fun.cpp
@@ -46,7 +46,9 @@ std::vector<double> do_focal_fun(std::vector<double> d, Rcpp::NumericMatrix w, s
 					size_t q = 0;
 					for (int j = -wr; j <= wr; j++) {
 						for (int k = -wc; k <= wc; k++) {
-							x[q] = d[j * ncol + k + i] * w[q];
+							if (!std::isnan(w[q])) {
+								x[q] = d[j * ncol + k + i] * w[q];
+							}
 							q++;
 						}
 					}
@@ -80,7 +82,9 @@ std::vector<double> do_focal_fun(std::vector<double> d, Rcpp::NumericMatrix w, s
 				size_t q = 0;
 				for (int j = -wr; j <= wr; j++) {
 					for (int k = -wc; k <= wc; k++) {
-						x[q] = d[j * ncol + k + i] * w[q];
+						if (!std::isnan(w[q])) {
+							x[q] = d[j * ncol + k + i] * w[q];
+						}
 						q++;
 					}
 				}

diff --git a/src/focal_sum.cpp b/src/focal_sum.cpp
@@ -46,12 +46,14 @@ std::vector<double> do_focal_sum(std::vector<double> d, Rcpp::NumericMatrix w, s
 						size_t p = 0;
 						for (int j = -wr; j <= wr; j++) {
 							for (int k = -wc; k <= wc; k++) {
-								double a = d[j * ncol + k + i];
-								if ( !std::isnan(a) ) {
-									val[i] += a * w[q];
-									p++;
+								if (!std::isnan(w[q])) {
+									double a = d[j * ncol + k + i];
+									if ( !std::isnan(a) ) {
+										val[i] += a * w[q];
+										p++;
+									}
+									q++;
 								}
-								q++;
 							}
 						}
 						if (p==0) {
@@ -85,10 +87,12 @@ std::vector<double> do_focal_sum(std::vector<double> d, Rcpp::NumericMatrix w, s
 					val[i] = 0;
 					for (int j = -wr; j <= wr; j++) {
 						for (int k = -wc; k <= wc; k++) {
-							double a = d[j * ncol + k + i];
-							if ( !std::isnan(a) ) {
-								val[i] += a * w[q];
-								p++;
+							if (!std::isnan(w[q])) {				
+								double a = d[j * ncol + k + i];
+								if ( !std::isnan(a) ) {
+									val[i] += a * w[q];
+									p++;
+								}
 							}
 							q++;
 						}
@@ -130,16 +134,20 @@ std::vector<double> do_focal_sum(std::vector<double> d, Rcpp::NumericMatrix w, s
 							for (int j = -wr; j <= wr; j++) {
 								bool jnot0 = j != 0;
 								for (int k = -wc; k <= wc; k++) {
-									if (jnot0 && (k != 0)) {
-										val[i] += d[j * ncol + k + i]  * w[q];
+									if (!std::isnan(w[q])) {	
+										if (jnot0 && (k != 0)) {
+											val[i] += d[j * ncol + k + i]  * w[q];
+										}
 										q++;
 									}
 								}
 							}
 						} else {
 							for (int j = -wr; j <= wr; j++) {
 								for (int k = -wc; k <= wc; k++) {
-									val[i] += d[j * ncol + k + i]  * w[q];
+									if (!std::isnan(w[q])) {
+										val[i] += d[j * ncol + k + i]  * w[q];
+									}
 									q++;
 								}
 							}
@@ -170,7 +178,9 @@ std::vector<double> do_focal_sum(std::vector<double> d, Rcpp::NumericMatrix w, s
 					size_t q = 0;
 					for (int j = -wr; j <= wr; j++) {
 						for (int k = -wc; k <= wc; k++) {
-							val[i] += d[j * ncol + k + i]  * w[q];
+							if (!std::isnan(w[q])) {				
+								val[i] += d[j * ncol + k + i]  * w[q];
+							}
 							q++;
 						}
 					}