Rewrite constraint evaluation code.

Make it much more efficient and easier to add other seed functions.

1 files changed, 3 insertions, 1 deletions

diff --git a/nlsolve.c b/nlsolve.c
index 708cd08..2137767 100644
--- a/nlsolve.c
+++ b/nlsolve.c
@@ -235,7 +235,7 @@ finish:
     return ret;
 }
 
-void tdi_solver_print_stats(NLSolveContext *ctx)
+void tdi_nlsolve_print_stats(NLSolveContext *ctx)
 {
     NLSolvePriv *s = ctx->priv;
 
@@ -497,6 +497,8 @@ int tdi_nlsolve_context_init(NLSolveContext *ctx)
         return ret;
     }
 
+    ctx->colloc_grid = s->ps_ctx->colloc_grid;
+
     /* init the per-equation state */
     for (int i = 0; i < s->nb_equations; i++) {
         ret = eq_ctx_init(ctx, i);