From 228f4da5d30418f6176557d07d0b063f5ee6ac00 Mon Sep 17 00:00:00 2001
From: Thomas White <taw@physics.org>
Date: Fri, 15 Sep 2017 14:29:02 +0200
Subject: Add restraints to prediction refinement

---
 libcrystfel/src/predict-refine.c | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'libcrystfel/src')

diff --git a/libcrystfel/src/predict-refine.c b/libcrystfel/src/predict-refine.c
index 015fd94d..e702cf33 100644
--- a/libcrystfel/src/predict-refine.c
+++ b/libcrystfel/src/predict-refine.c
@@ -488,6 +488,18 @@ static int iterate(struct reflpeak *rps, int n, UnitCell *cell,
 
 	}
 
+	int k;
+	for ( k=0; k<num_params; k++ ) {
+		double M_curr;
+		M_curr = gsl_matrix_get(M, k, k);
+		if ( (rv[k] == GPARAM_DETX) || (rv[k] == GPARAM_DETY) ) {
+			M_curr += 10.0;
+		} else {
+			M_curr += 1e-18;
+		}
+		gsl_matrix_set(M, k, k, M_curr);
+	}
+
 	//show_matrix_eqn(M, v);
 	shifts = solve_svd(v, M, NULL, 0);
 	if ( shifts == NULL ) {
-- 
cgit v1.2.3