started re-implementing planet phi position by rotating results

src/wakeflow/grid.py

@@ -605,17 +605,6 @@ def _add_non_linear_perturbations(self, NonLinearPerts : _NonLinearPerts, rho_ba
         """
 
         nonlin = NonLinearPerts
-
-        # # Copy 2d non-linear results
-        # vr_2d   = copy(nonlin.vr)
-        # vphi_2d = copy(nonlin.vphi)
-        # rho_2d  = copy(nonlin.rho)
-
-        # # Rotate the grid as desired by the user
-        # if self.p.phi_planet != 0:
-        #     vr_2d   = rotate_planet(vr_2d)
-        #     vphi_2d = rotate_planet(vphi_2d)
-        #     rho_2d  = rotate_planet(rho_2d)
 
         # Add velocities, identical at all heights for now
         nl_vr   = nonlin.vr  [:, np.newaxis, :]
@@ -702,6 +691,9 @@ def _merge_grids(self, grid_to_merge : "_Grid") -> None:
         # update info
         self.info["Contains"] += " AND " + g.info["Contains"]
 
+    def rotate(self, rho_background : "_Grid.rho") -> None:
+        pass
+
     def _smooth_box_old(self, big_box_grid: "_Grid") -> None:
         """Under development. Smooths the solution between the linear and non-linear regimes. Currently
         only smooths in v_r (it would be easy to extend to the other components if you need it).

src/wakeflow/wakeflow.py

@@ -328,6 +328,10 @@ def _run_wakeflow(self, params: _Parameters) -> None:
             # merge grids for result
             if params.include_linear:
                 grid_nonlin_perts._merge_grids(grid_lin_perts)
+
+            # rotate results if desired
+            if params.rot_interp is True:
+                grid_nonlin_perts.rotate(grid_background.rho)
 
             # merge grids for results
             grid_background._merge_grids(grid_nonlin_perts)