Atone for B0 name change made in pull request #1168

PlasmaControl · Sep 4, 2024 · 2363303 · 2363303
1 parent d2bfe12
commit 2363303
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/desc/compute/_field.py b/desc/compute/_field.py
@@ -116,10 +116,10 @@ def _B_sup_zeta(params, transforms, profiles, data, **kwargs):
     transforms={},
     profiles=[],
     coordinates="rtz",
-    data=["B0", "phi_z", "theta_PEST_t", "phi_t", "theta_PEST_z"],
+    data=["psi_r/sqrt(g)", "phi_z", "theta_PEST_t", "phi_t", "theta_PEST_z"],
 )
 def _B_sup_phi(params, transforms, profiles, data, **kwargs):
-    data["B^phi"] = data["B0"] * (
+    data["B^phi"] = data["psi_r/sqrt(g)"] * (
         data["phi_z"] * data["theta_PEST_t"] - data["phi_t"] * data["theta_PEST_z"]
     )
     return data
@@ -138,22 +138,22 @@ def _B_sup_phi(params, transforms, profiles, data, **kwargs):
     profiles=[],
     coordinates="rtz",
     data=[
-        "B0",
+        "psi_r/sqrt(g)",
         "phi_z",
         "theta_PEST_t",
         "phi_t",
         "theta_PEST_z",
-        "B0_r",
+        "(psi_r/sqrt(g))_r",
         "phi_rz",
         "theta_PEST_rt",
         "phi_rt",
         "theta_PEST_rz",
     ],
 )
 def _B_sup_phi_r(params, transforms, profiles, data, **kwargs):
-    data["B^phi_r"] = data["B0_r"] * (
+    data["B^phi_r"] = data["(psi_r/sqrt(g))_r"] * (
         data["phi_z"] * data["theta_PEST_t"] - data["phi_t"] * data["theta_PEST_z"]
-    ) + data["B0"] * (
+    ) + data["psi_r/sqrt(g)"] * (
         data["phi_rz"] * data["theta_PEST_t"]
         + data["phi_z"] * data["theta_PEST_rt"]
         - data["phi_rt"] * data["theta_PEST_z"]
@@ -175,22 +175,22 @@ def _B_sup_phi_r(params, transforms, profiles, data, **kwargs):
     profiles=[],
     coordinates="rtz",
     data=[
-        "B0",
+        "psi_r/sqrt(g)",
         "phi_z",
         "theta_PEST_t",
         "phi_t",
         "theta_PEST_z",
-        "B0_t",
+        "(psi_r/sqrt(g))_t",
         "phi_tz",
         "theta_PEST_tt",
         "phi_tt",
         "theta_PEST_tz",
     ],
 )
 def _B_sup_phi_t(params, transforms, profiles, data, **kwargs):
-    data["B^phi_t"] = data["B0_t"] * (
+    data["B^phi_t"] = data["(psi_r/sqrt(g))_t"] * (
         data["phi_z"] * data["theta_PEST_t"] - data["phi_t"] * data["theta_PEST_z"]
-    ) + data["B0"] * (
+    ) + data["psi_r/sqrt(g)"] * (
         data["phi_tz"] * data["theta_PEST_t"]
         + data["phi_z"] * data["theta_PEST_tt"]
         - data["phi_tt"] * data["theta_PEST_z"]
@@ -212,22 +212,22 @@ def _B_sup_phi_t(params, transforms, profiles, data, **kwargs):
     profiles=[],
     coordinates="rtz",
     data=[
-        "B0",
+        "psi_r/sqrt(g)",
         "phi_z",
         "theta_PEST_t",
         "phi_t",
         "theta_PEST_z",
-        "B0_z",
+        "(psi_r/sqrt(g))_z",
         "phi_zz",
         "theta_PEST_tz",
         "phi_tz",
         "theta_PEST_zz",
     ],
 )
 def _B_sup_phi_z(params, transforms, profiles, data, **kwargs):
-    data["B^phi_z"] = data["B0_z"] * (
+    data["B^phi_z"] = data["(psi_r/sqrt(g))_z"] * (
         data["phi_z"] * data["theta_PEST_t"] - data["phi_t"] * data["theta_PEST_z"]
-    ) + data["B0"] * (
+    ) + data["psi_r/sqrt(g)"] * (
         data["phi_zz"] * data["theta_PEST_t"]
         + data["phi_z"] * data["theta_PEST_tz"]
         - data["phi_tz"] * data["theta_PEST_z"]