Added shapefile to geojson conversion, updated copyrights

SEED-platform · Jun 4, 2024 · 8236768 · 8236768
1 parent d3a721b
commit 8236768
Show file tree

Hide file tree

Showing 11 changed files with 31 additions and 12 deletions.
diff --git a/main.py b/main.py
@@ -1,7 +1,7 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 import gzip

diff --git a/utils/__init__.py b/utils/__init__.py
@@ -1,5 +1,5 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
diff --git a/utils/chunk.py b/utils/chunk.py
@@ -1,7 +1,7 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 

diff --git a/utils/common.py b/utils/common.py
@@ -1,7 +1,7 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 from typing import TypedDict

diff --git a/utils/geocode_addresses.py b/utils/geocode_addresses.py
@@ -1,7 +1,7 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 import requests

diff --git a/utils/geojson_helpers.py b/utils/geojson_helpers.py
@@ -1,6 +1,6 @@
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 

diff --git a/utils/normalize_address.py b/utils/normalize_address.py
@@ -1,7 +1,7 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 import re

diff --git a/utils/shp_to_geojson.py b/utils/shp_to_geojson.py
@@ -0,0 +1,19 @@
+# !/usr/bin/env python
+"""
+SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
+"""
+
+import os
+
+import geopandas as gpd
+from pyproj import CRS
+
+from utils.ubid import add_ubid_to_geodataframe
+
+
+def shp_to_geojson(shapefile: str):
+    file_name, _ext = os.path.splitext(shapefile)
+    gdf = gpd.read_file(shapefile).to_crs(CRS.from_epsg(4326))
+    (add_ubid_to_geodataframe(gdf, additional_ubid_columns_to_create=[])
+     .to_file(f"{file_name}.geojson", driver='GeoJSON'))
diff --git a/utils/ubid.py b/utils/ubid.py
@@ -1,7 +1,7 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 from buildingid.code import decode, encode
@@ -10,10 +10,10 @@
 from shapely.geometry import Point, Polygon
 
 
-def encode_ubid(geometry: Polygon) -> str:
+def encode_ubid(geometry: Polygon, code_length: int = PAIR_CODE_LENGTH_) -> str:
     min_longitude, min_latitude, max_longitude, max_latitude = geometry.bounds
     centroid = geometry.centroid
-    ubid = encode(min_latitude, min_longitude, max_latitude, max_longitude, centroid.y, centroid.x, codeLength=PAIR_CODE_LENGTH_)
+    ubid = encode(min_latitude, min_longitude, max_latitude, max_longitude, centroid.y, centroid.x, codeLength=code_length)
     return ubid
 
 

diff --git a/utils/update_dataset_links.py b/utils/update_dataset_links.py
@@ -1,7 +1,7 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 import base64

diff --git a/utils/update_quadkeys.py b/utils/update_quadkeys.py
@@ -1,7 +1,7 @@
 # !/usr/bin/env python
 """
 SEED Platform (TM), Copyright (c) Alliance for Sustainable Energy, LLC, and other contributors.
-See also https://github.com/seed-platform/seed/main/LICENSE.md
+See also https://github.com/SEED-platform/seed/blob/main/LICENSE.md
 """
 
 from pathlib import Path