connermajic/rods
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial commit: establish deterministic rod-string solver stack.

Browse Source 
Set up the C solver core, Node API orchestration, TS GUI workflow, and engineering documentation with cleaned repo hygiene for private Git hosting.

Made-with: Cursor
This commit is contained in:
Conner Majic
2026-04-16 21:59:42 -06:00
commit 725a72a773
83 changed files with 14687 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

252
solver-c/tests/test_solver.c Normal file
View File

@@ -0,0 +1,252 @@
#include "solver.h"
#include "solver_internal.h"
#include <math.h>
#include <stdio.h>
#include <string.h>
static void fill_base_inputs(SolverInputs *in) {
memset(in, 0, sizeof(SolverInputs));
in->schema_version = 2;
in->workflow = 0;
in->pumping_speed = 5.0;
/* SI equivalents of typical imperial base-case magnitudes */
in->pump_depth = 1727.0 * 0.3048;
in->tubing_anchor_location = 1361.3 * 0.3048;
in->rod_friction_coefficient = 0.2;
in->stuffing_box_friction = 100.0 * 4.4482216152605;
in->pump_friction = 200.0 * 4.4482216152605;
in->taper_factor = 1.0;
in->trajectory_friction_multiplier = 1.0;
in->fluid_density_kg_m3 = 1000.0;
in->gravity = 9.80665;
in->upstroke_damping = 0.05;
in->downstroke_damping = 0.15;
in->non_dim_damping = 1.5;
in->molded_guide_mu_scale = 1.5;
in->wheeled_guide_mu_scale = 0.1;
in->other_guide_mu_scale = 2.0;
}
static int test_deterministic_shape(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
SolverOutputs out1;
SolverOutputs out2;
if (solver_run(&inputs, &out1) != 0 || solver_run(&inputs, &out2) != 0) {
return 1;
}
if (out1.point_count != 200 || out2.point_count != 200) {
return 2;
}
for (int i = 0; i < out1.point_count; i++) {
if (fabs(out1.polished_load[i] - out2.polished_load[i]) > 1e-9) {
return 3;
}
}
return 0;
}
static int test_bounds(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
SolverOutputs outputs;
if (solver_run(&inputs, &outputs) != 0) {
return 1;
}
if (fabs(outputs.max_polished_load - outputs.min_polished_load) < 1e-3) {
return 2;
}
if (!isfinite(outputs.max_downhole_load) || !isfinite(outputs.min_downhole_load)) {
return 3;
}
return 0;
}
static int test_fea_deterministic_shape(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
SolverOutputs out1;
SolverOutputs out2;
if (solver_run_fea(&inputs, &out1) != 0 || solver_run_fea(&inputs, &out2) != 0) {
return 1;
}
if (out1.point_count != 200 || out2.point_count != 200) {
return 2;
}
for (int i = 0; i < out1.point_count; i++) {
if (fabs(out1.polished_load[i] - out2.polished_load[i]) > 1e-9) {
return 3;
}
}
return 0;
}
/* CFL diagnostic must be finite and physically plausible */
static int test_cfl_clamp(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
SolverOutputs outputs;
if (solver_run_fdm(&inputs, &outputs) != 0) {
return 1;
}
if (!(outputs.max_cfl > 0.0 && outputs.max_cfl < 1e6 && outputs.wave_speed_ref_m_s > 0.0)) {
return 2;
}
return 0;
}
/* Static equilibrium helper: mean polished load should be finite and within broad physical band */
static int test_static_load_band(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
SolverOutputs outputs;
if (solver_run_fdm(&inputs, &outputs) != 0) {
return 1;
}
double sum = 0.0;
for (int i = 0; i < outputs.point_count; i++) {
sum += outputs.polished_load[i];
}
const double mean = sum / (double)outputs.point_count;
if (!isfinite(mean) || mean <= 0.0) {
return 2;
}
return 0;
}
/* Card extrema must match reported peaks (sanity / regression) */
static int test_card_extrema_match(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
SolverOutputs outputs;
if (solver_run_fdm(&inputs, &outputs) != 0) {
return 1;
}
double maxp = outputs.polished_load[0];
double minp = outputs.polished_load[0];
for (int i = 1; i < outputs.point_count; i++) {
if (outputs.polished_load[i] > maxp) maxp = outputs.polished_load[i];
if (outputs.polished_load[i] < minp) minp = outputs.polished_load[i];
}
if (fabs(maxp - outputs.max_polished_load) > 1e-3 || fabs(minp - outputs.min_polished_load) > 1e-3) {
return 2;
}
return 0;
}
/* Undamped run remains finite (zero Rayleigh-like stroke factors) */
static int test_undamped_finite(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
inputs.upstroke_damping = 0.0;
inputs.downstroke_damping = 0.0;
inputs.non_dim_damping = 0.0;
SolverOutputs outputs;
if (solver_run_fdm(&inputs, &outputs) != 0) {
return 1;
}
for (int i = 0; i < outputs.point_count; i++) {
if (!isfinite(outputs.polished_load[i])) {
return 2;
}
}
return 0;
}
/* Zero harmonic drive: loads stay bounded (zero-input stability) */
static int test_zero_input_bounded(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
inputs.pumping_speed = 0.0;
SolverOutputs outputs;
if (solver_run_fdm(&inputs, &outputs) != 0) {
return 1;
}
for (int i = 0; i < outputs.point_count; i++) {
if (!isfinite(outputs.polished_load[i]) || fabs(outputs.polished_load[i]) > 1e12) {
return 2;
}
}
return 0;
}
/* FDM vs FEA peak polished load tolerance (regression gate) */
static int test_fdm_fea_peak_tolerance(void) {
SolverInputs inputs;
fill_base_inputs(&inputs);
SolverOutputs fdm;
SolverOutputs fea;
if (solver_run_fdm(&inputs, &fdm) != 0 || solver_run_fea(&inputs, &fea) != 0) {
return 1;
}
const double tol = 700000.0; /* explicit FDM vs Newmark FEA — tighten after unified BC */
if (fabs(fdm.max_polished_load - fea.max_polished_load) > tol) {
return 2;
}
if (fabs(fdm.min_polished_load - fea.min_polished_load) > tol) {
return 3;
}
return 0;
}
int main(void) {
int rc = test_deterministic_shape();
if (rc != 0) {
printf("test_deterministic_shape failed: %d\n", rc);
return 1;
}
rc = test_bounds();
if (rc != 0) {
printf("test_bounds failed: %d\n", rc);
return 1;
}
rc = test_fea_deterministic_shape();
if (rc != 0) {
printf("test_fea_deterministic_shape failed: %d\n", rc);
return 1;
}
rc = test_cfl_clamp();
if (rc != 0) {
printf("test_cfl_clamp failed: %d\n", rc);
return 1;
}
rc = test_static_load_band();
if (rc != 0) {
printf("test_static_load_band failed: %d\n", rc);
return 1;
}
rc = test_card_extrema_match();
if (rc != 0) {
printf("test_card_extrema_match failed: %d\n", rc);
return 1;
}
rc = test_undamped_finite();
if (rc != 0) {
printf("test_undamped_finite failed: %d\n", rc);
return 1;
}
rc = test_zero_input_bounded();
if (rc != 0) {
printf("test_zero_input_bounded failed: %d\n", rc);
return 1;
}
rc = test_fdm_fea_peak_tolerance();
if (rc != 0) {
printf("test_fdm_fea_peak_tolerance failed: %d\n", rc);
return 1;
}
printf("solver-c tests passed\n");
return 0;
}