From 14447e9e2bd8c9069f0d0f67468f0fa5751ee66a Mon Sep 17 00:00:00 2001
From: Mladen Gibanica <11275336+mgcth@users.noreply.github.com>
Date: Mon, 6 May 2024 19:05:33 +0200
Subject: [PATCH] Match scb to detailed regions and municipalities, untested

---
 src/lantmateriet/choropleth.py | 46 +++++++++++++++++++++-------------
 1 file changed, 28 insertions(+), 18 deletions(-)

diff --git a/src/lantmateriet/choropleth.py b/src/lantmateriet/choropleth.py
index 90f36f4..6bd50c8 100644
--- a/src/lantmateriet/choropleth.py
+++ b/src/lantmateriet/choropleth.py
@@ -6,24 +6,35 @@
 from shapely import geometry, ops, polygonize
 
 
-def extract_choropleth(all_files: list[Path]):
+def read_data(path: str) -> gpd.GeoDataFrame:
+    """Read SCB defined regions."""
+    return gpd.read_file(path, use_arrow=True, engine="pyogrio")
+
+
+def extract_choropleth_polygons(all_files: list[Path]):
     """Extract Choropleth polygons from list of LineStrings."""
-    all_data = [
-        geo
-        for file in all_files
-        for geo in gpd.read_file(
-            file, use_arrow=True, engine="pyogrio"
-        ).geometry.to_list()
-    ]
+    all_data = [geo for file in all_files for geo in read_data(file).geometry.to_list()]
 
     return polygonize(ops.linemerge(geometry.MultiLineString(all_data)).geoms)
 
 
-def get_municipalities_choropleth(administrative_path: str, scb_path: str):
+def get_choropleth(scb_path: str, all_files: list[str]):
     """Get municipalities choropleth."""
-    admin = Path(administrative_path)
-    _ = Path(scb_path)
+    all_polygons = extract_choropleth_polygons(all_files)
+    scb_df = read_data(scb_path)
+
+    for row in scb_df.iterrows():
+        for polygon in all_polygons:
+            if polygon.contains(row.centroid):
+                row.geometry = polygon
 
+    return scb_df
+
+
+def get_municipality(admin_folder: str, scb_folder: str):
+    """Get municipalities."""
+    admin = Path(admin_folder)
+    scb = Path(scb_folder) / "Kommun_Sweref99TM.geojson"
     all_files = [
         admin / "04_riksgrans.geojson",
         admin / "05_sjoterritoriets_grans_i_havet.geojson",
@@ -31,18 +42,17 @@ def get_municipalities_choropleth(administrative_path: str, scb_path: str):
         admin / "02_lansgrans.geojson",
     ]
 
-    return extract_choropleth(all_files)
-
+    return get_choropleth(scb, all_files)
 
-def get_regions_choropleth(administrative_path: str, scb_path: str):
-    """Get regions choropleth."""
-    admin = Path(administrative_path)
-    _ = Path(scb_path)
 
+def get_regions(admin_folder: str, scb_folder: str):
+    """Get regions."""
+    admin = Path(admin_folder)
+    scb = Path(scb_folder) / "Lan_Sweref99TM_region.geojson"
     all_files = [
         admin / "04_riksgrans.geojson",
         admin / "05_sjoterritoriets_grans_i_havet.geojson",
         admin / "02_lansgrans.geojson",
     ]
 
-    return extract_choropleth(all_files)
+    return get_choropleth(scb, all_files)