From 5fc02387eeb54da82ec09b2a43cea41120c7733d Mon Sep 17 00:00:00 2001
From: nialov <nikolasovaskainen@gmail.com>
Date: Fri, 20 Dec 2024 16:18:59 +0200
Subject: [PATCH] docs(marimo): add validation app

---
 docs_src/marimo/validation.py | 179 ++++++++++++++++++++++++++++++++++
 1 file changed, 179 insertions(+)
 create mode 100644 docs_src/marimo/validation.py

diff --git a/docs_src/marimo/validation.py b/docs_src/marimo/validation.py
new file mode 100644
index 0000000..ecacce0
--- /dev/null
+++ b/docs_src/marimo/validation.py
@@ -0,0 +1,179 @@
+import marimo
+
+__generated_with = "0.10.6"
+app = marimo.App(width="medium")
+
+
+@app.cell
+def _():
+    return
+
+
+@app.cell
+def _():
+    import logging
+    from io import BytesIO
+    from pathlib import Path
+
+    import geopandas as gpd
+    import marimo as mo
+    import pyogrio
+
+    import fractopo.general
+    import fractopo.tval.trace_validation
+
+    return BytesIO, Path, fractopo, gpd, logging, mo, pyogrio
+
+
+@app.cell
+def _(mo):
+    input_traces_file = mo.ui.file(kind="area")
+    input_area_file = mo.ui.file(kind="area")
+    input_trace_layer_name = mo.ui.text()
+    input_area_layer_name = mo.ui.text()
+    input_snap_threshold = mo.ui.number(value=0.001)
+    input_button = mo.ui.run_button()
+
+    prompts = [
+        mo.md(f"## Upload trace data: {input_traces_file}"),
+        mo.md(f"Trace layer name, if applicable: {input_trace_layer_name}"),
+        mo.md(f"## Upload area data: {input_area_file}"),
+        mo.md(f"Area layer name, if applicable: {input_area_layer_name}"),
+        mo.md("Snap threshold: {}".format(input_snap_threshold)),
+        mo.md(f"Press to (re)start validation: {input_button}"),
+    ]
+
+    mo.vstack(prompts)
+    return (
+        input_area_file,
+        input_area_layer_name,
+        input_button,
+        input_snap_threshold,
+        input_trace_layer_name,
+        input_traces_file,
+        prompts,
+    )
+
+
+@app.cell
+def _(
+    Path,
+    fractopo,
+    gpd,
+    input_area_file,
+    input_area_layer_name,
+    input_button,
+    input_trace_layer_name,
+    input_traces_file,
+    mo,
+    pyogrio,
+):
+    def execute():
+        cli_args = mo.cli_args()
+        if len(cli_args) != 0:
+            cli_traces_path = Path(cli_args.get("traces-path"))
+            cli_area_path = Path(cli_args.get("area-path"))
+
+            name = cli_args.get("name") or cli_traces_path.stem
+
+            driver = pyogrio.detect_write_driver(cli_traces_path.name)
+
+            traces_gdf = gpd.read_file(cli_traces_path, driver=driver)
+            area_gdf = gpd.read_file(cli_area_path, driver=driver)
+        else:
+            mo.stop(not input_button.value)
+
+            trace_layer_name = (
+                input_trace_layer_name.value
+                if input_trace_layer_name.value != ""
+                else None
+            )
+            area_layer_name = (
+                input_area_layer_name.value
+                if input_area_layer_name.value != ""
+                else None
+            )
+
+            driver = pyogrio.detect_write_driver(input_traces_file.name())
+            print(f"Detected driver: {driver}")
+
+            print(
+                f"Trace layer name: {trace_layer_name}"
+                if trace_layer_name is not None
+                else "No layer specified"
+            )
+            traces_gdf = gpd.read_file(
+                input_traces_file.contents(),
+                layer=trace_layer_name,
+                # , driver=driver
+            )
+            print(
+                f"Area layer name: {area_layer_name}"
+                if area_layer_name is not None
+                else "No layer specified"
+            )
+            area_gdf = gpd.read_file(
+                input_area_file.contents(),
+                layer=area_layer_name,
+                # , driver=driver
+            )
+
+            name = (
+                Path(input_traces_file.name()).stem
+                if trace_layer_name is None
+                else trace_layer_name
+            )
+        validation = fractopo.tval.trace_validation.Validation(
+            traces=traces_gdf, area=area_gdf, name=name, allow_fix=True
+        )
+
+        validated = validation.run_validation()
+
+        validated_clean = fractopo.general.convert_list_columns(validated, allow=True)
+        validated_clean[validation.ERROR_COLUMN] = validated_clean[
+            validation.ERROR_COLUMN
+        ]
+
+        return validated_clean
+
+    return (execute,)
+
+
+@app.cell
+def _(execute, logging, mo):
+    with mo.redirect_stderr():
+        with mo.redirect_stdout():
+            try:
+                validated_clean = execute()
+            except Exception:
+                logging.error("Failed to validate input data.", exc_info=True)
+    return (validated_clean,)
+
+
+@app.cell
+def _(fractopo, mo, validated_clean):
+    if (
+        fractopo.tval.trace_validation.Validation.ERROR_COLUMN
+        in validated_clean.columns
+    ):
+        mo.output.replace(
+            validated_clean[
+                fractopo.tval.trace_validation.Validation.ERROR_COLUMN
+            ].value_counts()
+        )
+    return
+
+
+@app.cell
+def _(validated_clean):
+    validated_clean.drop(columns=["geometry"])
+    return
+
+
+@app.cell
+def _():
+    return
+
+
+if __name__ == "__main__":
+    app.run()