From 62c19009fe70b94153f57044e7717e2e59980009 Mon Sep 17 00:00:00 2001
From: Thomas White <taw@physics.org>
Date: Wed, 13 May 2015 13:42:24 +0200
Subject: solve_svd(): Squash NaNs after rescaling

---
 libcrystfel/src/utils.c | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'libcrystfel/src')

diff --git a/libcrystfel/src/utils.c b/libcrystfel/src/utils.c
index 42108270..af0abedc 100644
--- a/libcrystfel/src/utils.c
+++ b/libcrystfel/src/utils.c
@@ -209,6 +209,16 @@ gsl_vector *solve_svd(gsl_vector *v, gsl_matrix *M, int *pn_filt, int verbose)
 	SAS_copy = gsl_matrix_alloc(SAS->size1, SAS->size2);
 	gsl_matrix_memcpy(SAS_copy, SAS);
 
+	for ( i=0; i<n; i++ ) {
+		int j;
+		if ( isnan(gsl_vector_get(SB, i)) ) gsl_vector_set(SB, i, 0.0);
+		for ( j=0; j<n; j++ ) {
+			if ( isnan(gsl_matrix_get(SAS, i, j)) ) {
+					gsl_matrix_set(SAS, i, j, 0.0);
+			}
+		}
+	}
+
 	/* Do the SVD */
 	s_val = gsl_vector_calloc(n);
 	s_vec = gsl_matrix_calloc(n, n);
-- 
cgit v1.2.3