From d451e140926584ae59b0a0552d5463b9d6114ac2 Mon Sep 17 00:00:00 2001
From: Anton Khirnov <anton@khirnov.net>
Date: Sun, 28 Aug 2016 11:20:16 +0200
Subject: Switch to polar coordinates in the solver.

---
 init.c | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'init.c')

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;
 
-- 
cgit v1.2.3