✨ Regrid eqdsk after read in

Author: je-cook

bluemira/equilibria/equilibrium.py

@@ -139,6 +139,7 @@ def _get_eqdsk(
         *,
         qpsi_positive: bool | None = None,
         full_coil: bool = False,
+        regrid_nx_nz: tuple[int, int] | None = None,
         **kwargs,
     ) -> tuple[EQDSKInterface, Grid]:
         """
@@ -190,7 +191,10 @@ def _get_eqdsk(
             e.dxc /= 2
             e.dzc /= 2
 
-        return e, Grid.from_eqdsk(e)
+        grid = Grid.from_eqdsk(e)
+        if regrid_nx_nz is not None:
+            grid = grid.regrid(*regrid_nx_nz)
+        return e, grid
 
     def to_eqdsk(
         self,
@@ -280,6 +284,7 @@ def from_eqdsk(
         *,
         qpsi_positive: bool | None = None,
         full_coil: bool = False,
+        regrid_nx_nz: tuple[int, int] | None = None,
         **kwargs,
     ):
         """
@@ -306,8 +311,10 @@ def from_eqdsk(
             from_cocos=from_cocos,
             full_coil=full_coil,
             qpsi_positive=qpsi_positive,
+            regrid_nx_nz=regrid_nx_nz,
             **kwargs,
         )
+
         psi_ax = e.psimag
         psi_b = e.psibdry
         lcfs = Coordinates({"x": e.xbdry, "z": e.zbdry})
@@ -472,6 +479,7 @@ def _get_eqdsk(
         user_coils: CoilSet | None = None,
         force_symmetry: bool = False,
         full_coil: bool = False,
+        regrid_nx_nz: tuple[int, int] | None = None,
         **kwargs,
     ) -> tuple[EQDSKInterface, Grid, CoilSet, Limiter | None]:
         """
@@ -515,6 +523,7 @@ def _get_eqdsk(
             from_cocos=from_cocos,
             qpsi_positive=qpsi_positive,
             full_coil=full_coil,
+            regrid_nx_nz=regrid_nx_nz,
             **kwargs,
         )
         coilset = user_coils if user_coils is not None else CoilSet.from_group_vecs(e)
@@ -659,6 +668,7 @@ def from_eqdsk(
         force_symmetry: bool,
         user_coils: CoilSet | None = None,
         full_coil: bool = False,
+        regrid_nx_nz: tuple[int, int] | None = None,
         **kwargs,
     ):
         """
@@ -692,8 +702,10 @@ def from_eqdsk(
             force_symmetry=force_symmetry,
             user_coils=user_coils,
             full_coil=full_coil,
+            regrid_nx_nz=regrid_nx_nz,
             **kwargs,
         )
+
         self = cls(coilset, grid, limiter=limiter, psi=eqdsk.psi, filename=filename)
         self._eqdsk = eqdsk
         return self
@@ -1025,6 +1037,7 @@ def from_eqdsk(
         force_symmetry: bool = False,
         user_coils: CoilSet | None = None,
         full_coil: bool = False,
+        regrid_nx_nz: tuple[int, int] | None = None,
         **kwargs,
     ):
         """
@@ -1062,6 +1075,7 @@ def from_eqdsk(
             force_symmetry=force_symmetry,
             user_coils=user_coils,
             full_coil=full_coil,
+            regrid_nx_nz=regrid_nx_nz,
             **kwargs,
         )
 

bluemira/equilibria/grid.py

@@ -132,6 +132,23 @@ def __init__(
             [(nx - 1, z) for z in range(nz)],
         ])
 
+    def regrid(self, nx: int, nz: int) -> Grid:
+        """Reinitialise the grid with different x and z discretisation
+
+        Parameters
+        ----------
+        nx:
+            number of x grid points
+        nz:
+            number of z grid points
+
+        Returns
+        -------
+        :
+            New grid object
+        """
+        return type(self)(self.x_min, self.x_max, self.z_min, self.z_max, nx, nz)
+
     @classmethod
     def from_eqdict(cls, e) -> Grid:
         """