Move pgm write to utils

src/fpm/cli.py

@@ -117,7 +117,8 @@ def tasks(g, base_path, config, **kwargs):
 
 
 def get_occ_grid(g, base_path, **kwargs):
-    generate_occ_grid(g, base_path, **kwargs)
+    output_path = get_output_path(base_path, "maps")
+    generate_occ_grid(g, output_path, **kwargs)
 
 
 def get_output_path(base_path, subfolder, model_name=None):

src/fpm/generators/occ_grid.py

@@ -102,9 +102,10 @@ def generate_occ_grid(g, output_path, **custom_args):
     )
     # draw_floorplan_opening(g, "Window", draw, west, south, resolution, border, free, coords_m)
 
-    name_image = "{}.pgm".format(map_name)
     im = ImageOps.flip(im)
-    im.save(os.path.join(output_path, name_image), quality=95)
+
+    name_image = "{}.pgm".format(map_name)
+    save_file(output_path, name_image, im)
 
 
 def draw_floorplan_obstacle(

src/fpm/utils.py

@@ -33,6 +33,8 @@ def save_file(output_path, file_name, contents):
     if ext in [".yaml"]:
         with open(output_file, "w") as f:
             yaml.dump(contents, f, default_flow_style=None)
+    elif ext in [".pgm"]:
+        contents.save(output_file, quality=95)
     else:
         with open(output_file, "w") as f:
             f.write(contents)