Switch to polar coordinates in the solver.

1 files changed, 8 insertions, 1 deletions

diff --git a/init.c b/init.c
index fa15325..72dabed 100644
--- a/init.c
+++ b/init.c
@@ -36,8 +36,15 @@ static int brill_init_check_options(BDContext *bd)
     if (!s->qfunc)
         return ret;
 
+    s->coord_system = COORD_SYSTEM_RADIAL;
+
     s->basis[0] = bdi_basis_init(BASIS_FAMILY_SB_EVEN, bd->basis_scale_factor[0]);
-    s->basis[1] = bdi_basis_init(BASIS_FAMILY_SB_EVEN, bd->basis_scale_factor[0]);
+
+    if (s->coord_system == COORD_SYSTEM_RADIAL)
+        s->basis[1] = bdi_basis_init(BASIS_FAMILY_COS_EVEN, 0);
+    else
+        s->basis[1] = bdi_basis_init(BASIS_FAMILY_SB_EVEN, bd->basis_scale_factor[0]);
+
     if (!s->basis[0] || !s->basis[1])
         return -ENOMEM;