Disable isotach perturbation is GAHM RMW perturbation is not selected

noaa-ocs-modeling · Oct 1, 2024 · 587a57c · 587a57c
1 parent 33cb296
commit 587a57c
Showing 1 changed file with 20 additions and 18 deletions.
diff --git a/ensembleperturbation/perturbation/atcf.py b/ensembleperturbation/perturbation/atcf.py
@@ -1927,25 +1927,27 @@ def write_perturbed_track(
             # Compute potential changes in the central pressure in accordance with Holland B relationship
             dataframe[CentralPressure.name] = self.compute_pc_from_Vmax(dataframe)
 
-            # Compute potential changes to r34/50,64 radii at all quadrants in accordance with the GAHM profile
-            quadrants = [
-                IsotachRadiusNEQ(),
-                IsotachRadiusSEQ(),
-                IsotachRadiusSWQ(),
-                IsotachRadiusNWQ(),
-            ]
-            for quadrant in quadrants:
-                dataframe = quadrant.perturb(
-                    vortex_dataframe=dataframe,
-                    original_dataframe=self.track.data,
-                    inplace=True,
+            if RadiusOfMaximumWinds in variables:
+                # If perturbing for GAHM, use isotach perturbation
+                # Compute potential changes to r34/50,64 radii at all quadrants in accordance with the GAHM profile
+                quadrants = [
+                    IsotachRadiusNEQ(),
+                    IsotachRadiusSEQ(),
+                    IsotachRadiusSWQ(),
+                    IsotachRadiusNWQ(),
+                ]
+                for quadrant in quadrants:
+                    dataframe = quadrant.perturb(
+                        vortex_dataframe=dataframe,
+                        original_dataframe=self.track.data,
+                        inplace=True,
+                    )
+                # remove any rows that now have all 0 isotach radii for the 50-kt or 64-kt isotach
+                quadrant_names = [q.name for q in quadrants]
+                drop_row = (dataframe[quadrant_names] == 0).all(axis=1) & (
+                    dataframe['isotach_radius'].isin([50, 64])
                 )
-            # remove any rows that now have all 0 isotach radii for the 50-kt or 64-kt isotach
-            quadrant_names = [q.name for q in quadrants]
-            drop_row = (dataframe[quadrant_names] == 0).all(axis=1) & (
-                dataframe['isotach_radius'].isin([50, 64])
-            )
-            dataframe.drop(index=dataframe.index[drop_row], inplace=True)
+                dataframe.drop(index=dataframe.index[drop_row], inplace=True)
 
         # write out the modified `fort.22`
         VortexTrack(storm=dataframe).to_file(filename, overwrite=True)