Adding a write bathymetry feature that is used later to generate MD.out with the seafloor file in options

Author: Yuksel-Rudy

famodel/project.py

@@ -4147,7 +4147,7 @@ def extractFarmInfo(self, cmax=5, fmax=10/6, Cmeander=1.9):
 
         return wts, yaw_init  
 
-    def FFarmCompatibleMDOutput(self, filename, unrotateTurbines=True, renameBody=True, removeBody=True, MDoptionsDict={}):
+    def FFarmCompatibleMDOutput(self, filename, unrotateTurbines=True, renameBody=True, removeBody=True, MDoptionsDict={}, bathymetryFile=None):
         '''
         Function to create FFarm-compatible MoorDyn input file:
 
@@ -4218,7 +4218,22 @@ def FFarmCompatibleMDOutput(self, filename, unrotateTurbines=True, renameBody=Tr
 
             with open(filename, 'w') as f:
                 f.writelines(newLines)
-           
+        
+        if bathymetryFile:
+            with open(filename, 'r') as f:
+                lines = f.readlines()
+
+            newLines = []
+
+            for i, line in enumerate(lines):
+                newLines.append(line)
+                if '---' in line and 'OPTIONS' in line.upper():
+                    newLines.append(f" {bathymetryFile}             Seafloor File\n")
+                
+            
+            with open(filename, 'w') as f:
+                f.writelines(newLines)
+                
     def resetArrayCenter(self, FOWTOnly=True):
         '''
         Function to reset array center such that the farm origin is the mid-point between all FOWT platforms:

famodel/seabed/seabed_tools.py

@@ -42,7 +42,32 @@ def readBathymetryFile(filename, dtype=float):
 
     return bathGrid_Xs, bathGrid_Ys, bathGrid
 
-
+def writeBathymetryFile(filename, grid_x, grid_y, grid_depth):
+    """
+    Writes bathymetry data to a file in the format expected by readBathymetryFile.
+
+    Parameters:
+        filename (str): The name of the file to write to.
+        grid_x (list or np.ndarray): The X coordinates of the grid.
+        grid_y (list or np.ndarray): The Y coordinates of the grid.
+        grid_depth (np.ndarray): The bathymetry grid (depth values).
+    """
+    with open(filename, 'w') as f:
+        # Write a placeholder header
+        f.write("Bathymetry Data File\n")
+        
+        # Write the number of grid values in the x and y directions
+        f.write(f"nGridX {len(grid_x)}\n")
+        f.write(f"nGridY {len(grid_y)}\n")
+        
+        # Write the X coordinates
+        f.write(" ".join(map(str, grid_x)) + "\n")
+        
+        # Write the Y coordinates and the bathymetry grid
+        for i, y in enumerate(grid_y):
+            row = [y] + list(grid_depth[i, :])
+            f.write(" ".join(map(str, row)) + "\n")
+            
 def getSoilTypes(filename):
     '''function to read in a preliminary input text file format of soil type information'''