From 9adc82da294d332cd2ec5fd84bc835ac30124143 Mon Sep 17 00:00:00 2001
From: Maryam Hosseini <msh588@nyu.edu>
Date: Tue, 6 Aug 2024 12:59:31 -0400
Subject: [PATCH] This commit : 1- Creates a csv file of tile id and coordinate
 in both systems when generate is called. This is very handy for the user to
 refer back to if they need to know where certain tiles are. 2- Add an
 optional dest_path argument to save_csv and save_shapefile to be saved even
 if generate is not called and no source tile exists.

---
 src/tile2net/raster/grid.py   | 20 ++++++++++++--------
 src/tile2net/raster/raster.py |  3 ++-
 2 files changed, 14 insertions(+), 9 deletions(-)

diff --git a/src/tile2net/raster/grid.py b/src/tile2net/raster/grid.py
index 6397682..d85a46d 100644
--- a/src/tile2net/raster/grid.py
+++ b/src/tile2net/raster/grid.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import warnings
 import os
 import numpy as np
@@ -506,26 +508,28 @@ def create_grid_gdf(self):
         gdf_grid = gdf_grid.reset_index(drop=True)
         return gdf_grid
 
-    def save_shapefile(self):
+    def save_shapefile(self, dst_path: str = None) -> None:
         """
         Saves the grid data in shapefile format
         """
-        dst_path = self.project.tiles.path
-        if not os.path.exists(dst_path):
-            os.makedirs(dst_path)
+        if not dst_path is None:
+            dst_path = self.project.tiles.path
+            if not os.path.exists(dst_path):
+                os.makedirs(dst_path)
         poly = self._create_pseudo_tiles()
         tileinfo_df = self._create_info_dict(df=True)
         gdf_grid = gpd.GeoDataFrame(tileinfo_df, geometry=poly, crs=self.crs)
         gdf_grid.to_file(
             os.path.join(dst_path, f'{self.name}_{self.tile_size}_pseudo_tiles'))
 
-    def save_csv(self):
+    def save_csv(self, dst_path: str = None):
         """
         Saves the grid data in CSV format without geometry info
         """
-        dst_path = self.project.tiles.path
-        if not os.path.exists(dst_path):
-            os.makedirs(dst_path)
+        if not dst_path:
+            dst_path = self.project.tiles.path
+            if not os.path.exists(dst_path):
+                os.makedirs(dst_path)
         tileinfo_df = self._create_info_dict(df=True)
         tileinfo_df.to_csv(os.path.join(dst_path, f'{self.name}_{self.tile_size}_info.csv'))
 
diff --git a/src/tile2net/raster/raster.py b/src/tile2net/raster/raster.py
index e3995d9..e137b68 100644
--- a/src/tile2net/raster/raster.py
+++ b/src/tile2net/raster/raster.py
@@ -326,7 +326,7 @@ def __init__(
         )
 
     def __repr__(self):
-        if self.boundary_path != -1:
+        if self.boundary_path != "":
             tiles_within = f"{(self.num_inside / self.num_tiles) * 100:.1f}"
             return (
                 f"{self.name} Data Constructor. \nCoordinate reference system (CRS): {self.crs} \n"
@@ -813,6 +813,7 @@ def generate(self, step):
         """
         self.stitch(step)
         self.save_info_json(new_tstep=step)
+        self.save_csv()
         logger.info(f"Dumping to {self.project.tiles.info}")
         json.dump(
             dict(self.project.structure),