From f1e6ec0852315b902986a7cb9784947aa64f112d Mon Sep 17 00:00:00 2001
From: Ben Purinton <ben.purinton@gmail.com>
Date: Mon, 25 Nov 2024 14:01:21 -0800
Subject: [PATCH] select processing levels

---
 asp_plot/altimetry.py         | 7 +++++++
 asp_plot/cli/asp_plot.py      | 1 +
 notebooks/icesat2_plots.ipynb | 1 +
 3 files changed, 9 insertions(+)

diff --git a/asp_plot/altimetry.py b/asp_plot/altimetry.py
index 0d6725e..df972d5 100644
--- a/asp_plot/altimetry.py
+++ b/asp_plot/altimetry.py
@@ -99,6 +99,7 @@ def __init__(
 
     def request_atl06sr_multi_processing(
         self,
+        processing_levels=["high_confidence", "ground", "canopy", "top_of_canopy"],
         res=20,
         len=40,
         ats=20,
@@ -132,6 +133,12 @@ def request_atl06sr_multi_processing(
             },
         }
 
+        parms_dict = {
+            key: parms for key, parms in parms_dict.items() if key in processing_levels
+        }
+
+        print("\n\nparms_dict: ", parms_dict)
+
         for key, parms in parms_dict.items():
             parms["poly"] = region
             parms["res"] = res
diff --git a/asp_plot/cli/asp_plot.py b/asp_plot/cli/asp_plot.py
index e8fa4a4..727421c 100644
--- a/asp_plot/cli/asp_plot.py
+++ b/asp_plot/cli/asp_plot.py
@@ -162,6 +162,7 @@ def main(
         icesat = Altimetry(directory=directory, dem_fn=asp_dem)
 
         icesat.request_atl06sr_multi_processing(
+            processing_levels=["ground"],
             save_to_parquet=False,
         )
 
diff --git a/notebooks/icesat2_plots.ipynb b/notebooks/icesat2_plots.ipynb
index 1e0f00b..49b5720 100644
--- a/notebooks/icesat2_plots.ipynb
+++ b/notebooks/icesat2_plots.ipynb
@@ -96,6 +96,7 @@
     "#     cnt=5,\n",
     "#     maxi=5,\n",
     "#     save_to_parquet=True,\n",
+    "#     processing_levels=[\"ground\"],\n",
     "# )"
    ]
   },