From ef1532ebd525eeb4ac1341f2aa13817708c7070d Mon Sep 17 00:00:00 2001
From: Michael Kraus <michael.kraus@ipp.mpg.de>
Date: Tue, 22 Sep 2020 13:34:34 +0200
Subject: [PATCH] Add convenience initialisers for Solov'ev equilibria.

---
 src/analytic/solovev.jl        | 12 ++++++++++++
 src/analytic/solovev_xpoint.jl | 12 ++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/src/analytic/solovev.jl b/src/analytic/solovev.jl
index f759f19..a438589 100644
--- a/src/analytic/solovev.jl
+++ b/src/analytic/solovev.jl
@@ -106,4 +106,16 @@ module Solovev
         return equilibrium
     end
 
+    function ITER(; perturbation=ZeroPerturbation())
+        equilibrium = SolovevEquilibriumITER()
+        load_equilibrium(equilibrium, perturbation; target_module=SolovevXpoint)
+        return equilibrium
+    end
+
+    function NSTX(; perturbation=ZeroPerturbation())
+        equilibrium = SolovevEquilibriumNSTX()
+        load_equilibrium(equilibrium, perturbation; target_module=SolovevXpoint)
+        return equilibrium
+    end
+
 end
diff --git a/src/analytic/solovev_xpoint.jl b/src/analytic/solovev_xpoint.jl
index 92c3f92..8b81890 100644
--- a/src/analytic/solovev_xpoint.jl
+++ b/src/analytic/solovev_xpoint.jl
@@ -127,4 +127,16 @@ module SolovevXpoint
         return equilibrium
     end
 
+    function ITER(; perturbation=ZeroPerturbation())
+        equilibrium = SolovevXpointEquilibriumITER()
+        load_equilibrium(equilibrium, perturbation; target_module=SolovevXpoint)
+        return equilibrium
+    end
+
+    function NSTX(; perturbation=ZeroPerturbation())
+        equilibrium = SolovevXpointEquilibriumNSTX()
+        load_equilibrium(equilibrium, perturbation; target_module=SolovevXpoint)
+        return equilibrium
+    end
+
 end