edmt 1.0.1.dev0__py3-none-any.whl

edmt/mapping/mapping.py

@@ -0,0 +1,207 @@
+import geopandas as gpd
+import pandas as pd
+import matplotlib.pyplot as plt
+import contextily as cx
+from pyproj import CRS
+
+from edmt.contrib.utils import clean_vars
+
+class Mapping:
+    def __init__(self):
+        # Initialize any necessary attributes
+        pass
+
+#     @staticmethod
+#     def gplot(df, column=None, title=None, legend=True, fill=None, grids=None, **additional_args):
+#         df = df.copy()
+#         df = df.to_crs(epsg=4326)
+
+#         ax = plt.subplots(figsize=(10, 10))
+#         plot_args = {
+#             "ax": ax,
+#             "alpha": 0.6,
+#             "edgecolor": "black",
+#             "column": column,
+#             "legend": legend,
+#             "legend_kwds": {
+#                 "loc": "lower right",
+#                 "bbox_to_anchor": (1, 0),
+#                 "frameon": True,
+#                 "title": column,
+#             },
+#             "facecolor": fill,
+#         }
+#         plot_args = clean_vars(additional_args, **plot_args)
+#         df.plot(**plot_args)
+#         cx.add_basemap(ax, crs=df.crs, source=cx.providers.OpenStreetMap.Mapnik)
+#         return ax
+
+# class Mapping:
+
+#     def __init__(self):
+#         # Initialize any necessary attributes
+#         pass
+
+#     @staticmethod
+#     def gplot(df, column=None, title=None, ax=None, legend=True, fill=None,grids=None, **additional_args):
+#         # print(f"Plot started at: {datetime.now()}")
+#         # start_time = time.time()
+
+#         df = df.copy()
+#         df = df.to_crs(epsg=4326)  # Ensure WGS 84
+
+#         # Create plot
+#         if ax is None:
+#             _, ax = plt.subplots(figsize=(10, 10))
+
+#         # Default plot arguments
+#         plot_args = {
+#             "ax": ax,
+#             "alpha": 0.6,
+#             "edgecolor": "black",
+#             "column": column,
+#             "legend": legend,
+#             "legend_kwds": {
+#                 "loc": "lower right",
+#                 "bbox_to_anchor": (1, 0),
+#                 "frameon": True,
+#                 "title": column,
+#             },
+#             "facecolor" : fill
+#         }
+
+#         # Title
+#         if title:
+#             ax.set_title(title, fontsize=14)
+
+#         # Add grids
+#         if grids:
+#             ax.grid(visible=True, linestyle="--", linewidth=0.5, alpha=0.7)
+
+#         # Clean and merge additional arguments
+#         plot_args = clean_vars(additional_args, **plot_args)
+
+#         # Plot the GeoDataFrame
+#         df.plot(**plot_args)
+
+#         # Add a frame around the map
+#         for spine in ax.spines.values():
+#             spine.set_edgecolor("black")
+#             spine.set_linewidth(1.5)
+
+#         # Add basemap
+#         cx.add_basemap(ax, crs=df.crs, source=cx.providers.OpenStreetMap.Mapnik)
+
+#         # end_time = time.time()
+#         # execution_time = end_time - start_time
+#         # print(f"Execution time: {execution_time:.2f} seconds.")
+#         return ax
+
+#     # def gplot(df, column=None, title=None, legend=True, fill=None, grids=None):
+
+#     #     df = df.copy()
+#     #     if df.crs is None:
+#     #         raise ValueError("Input GeoDataFrame must have a CRS defined.")
+#     #     if df.crs != CRS.from_epsg(4326):
+#     #         df = df.to_crs(epsg=4326)  # Ensure WGS 84
+
+#     #     ax=ax
+
+#     #     # Default plot arguments
+#     #     plot_args = {
+#     #         "ax": ax,
+#     #         "alpha": 0.6,
+#     #         "edgecolor": "black",
+#     #     }
+
+#     #     # Add column-specific arguments if a column is provided
+#     #     if column:
+#     #         plot_args["column"] = column
+#     #         plot_args["legend"] = legend
+#     #         plot_args["legend_kwds"] = {
+#     #             "loc": "lower right",
+#     #             "bbox_to_anchor": (1, 0),
+#     #             "frameon": True,
+#     #             "title": column,
+#     #         }
+
+#     #     # Add fill color
+#     #     if fill:
+#     #         plot_args["color"] = fill
+
+#     #     # Set title
+#     #     if title:
+#     #         ax.set_title(title, fontsize=14)
+
+#     #     # Add grids if specified
+#     #     if grids:
+#     #         ax.grid(visible=True, linestyle="--", linewidth=0.5, alpha=0.7)
+
+#     #     # Plot the GeoDataFrame
+#     #     df.plot(**plot_args)
+
+#     #     # Add a frame around the map
+#     #     for spine in ax.spines.values():
+#     #         spine.set_edgecolor("black")
+#     #         spine.set_linewidth(1.5)
+
+#     #     # Add basemap
+#     #     cx.add_basemap(ax, crs=df.crs, source=cx.providers.OpenStreetMap.Mapnik)
+#     #     return ax
+    
+#     @staticmethod
+#     def TileLayer(self, df):
+
+#         "list of base layers to use"
+
+#         """
+#         Add names, opacity, 
+
+#         addl_args()
+
+#         Use clean var to
+#         """
+
+#     @staticmethod
+#     def title(df):
+#         """
+#         Plot a GeoDataFrame with optional dynamic column-based styling and a categorical legend.
+#         """
+#         df = df.copy()
+#         df = df.to_crs(epsg=4326)  # Ensure WGS 84 
+
+#         # Create plot
+#         if ax is None:
+#             _, ax = plt.subplots(figsize=(10, 10))
+
+#         return ax
+    
+#     @staticmethod
+#     def legend(df):
+
+#         return df
+    
+#     @staticmethod
+#     def scale_bar(df):
+
+#         return df
+    
+#     @staticmethod
+#     def add_table(df):
+
+#         return df
+    
+#     @staticmethod
+#     def html(df):
+
+#         return df
+    
+#     @staticmethod
+#     def png(df):
+
+#         return df
+    
+#     @staticmethod
+#     def legend(df):
+
+#         return df

edmt/mapping/maps.py

@@ -0,0 +1,77 @@
+import contextily as cx
+from edmt.contrib.utils import clean_vars
+# from geopandas import plotting as plot
+
+class Mapping:
+    def __init__(self,**kwargs):
+        self.default_crs = 4326
+        self.df_cache = None
+        self.ax = None
+
+    def process_df(self, df):
+        """
+        Process the GeoDataFrame for plotting.
+        Includes caching and CRS transformation.
+        """
+        self.df_cache = df.copy()
+        return self.df_cache.to_crs(epsg=self.default_crs)
+
+    def get_crs(self):
+        return self.default_crs
+
+    def set_crs(self, crs):
+        self.default_crs = crs
+
+    def gplot(self, df,column:str=None,**kwargs):
+        """
+        Plot the GeoDataFrame and store the axis object.
+        """
+        df = self.process_df(df)
+        self.ax = df.plot(alpha=0.7,column=column)
+        cx.add_basemap(self.ax, source=cx.providers.CartoDB.Positron)
+        self.ax.set_axis_off()
+        return self
+
+    def figure(self, width, height):
+        if self.ax:
+            self.ax.set_figwidth(width)
+            self.ax.set_figheight(height)
+        return self
+
+        # usage
+        # figure(12, 6)
+
+    def add_colorbar(self):
+        if self.ax:
+            self.ax.get_figure().colorbar
+        return self
+
+    def add_axis(self):
+        if self.ax:
+            self.ax.set_axis_on()
+        return self
+
+    def add_title(self, title):
+        if self.ax:
+            self.ax.set_title(title)
+        return self
+
+    def add_grids(self):
+        if self.ax:
+            self.ax.grid(visible=True, linestyle="--", linewidth=0.5, alpha=0.7)
+        return self
+
+    def add_labels(self):
+        if self.ax:
+            self.ax.tick_params(labeltop=False, labelright=False, labelsize=8, pad=-20)
+        return self
+
+    
+
+    def add_basemap(self, providers=None,tile=None):
+        if providers and tile:
+            source = f"cx.providers.{providers}.{tile}"
+            if self.ax:
+                cx.add_basemap(self.ax, source=source)
+                self.ax.set_axis_off()
+            return self

edmt/models/__init__.py

@@ -0,0 +1,15 @@
+from .drones import (
+    Airdata,
+    airPoint,
+    df_to_gdf,
+    airLine,
+    airSegment
+)
+
+__all__ = [
+    "Airdata",
+    "airPoint",
+    "df_to_gdf",
+    "airLine",
+    "airSegment"
+]