openforis-whisp 2.0.0b3__py3-none-any.whl → 3.0.0a1__py3-none-any.whl

openforis_whisp/__init__.py

@@ -2,8 +2,11 @@ import ee
 from google.oauth2 import service_account
 
 
-def initialize_ee(credentials_path=None):
-    """Initializes Google Earth Engine using the provided path or defaults to normal if no path is given."""
+def initialize_ee(credentials_path=None, use_high_vol_endpoint=False):
+    """Initializes Google Earth Engine using the provided path or defaults to normal if no path is given.
+    Args:
+    use_high_vol_endpoint: True/False to use high-volume endpoint via opt_url parameter (defaults to False).
+    """
     try:
         if not ee.data._initialized:
             print(credentials_path)
@@ -12,10 +15,21 @@ def initialize_ee(credentials_path=None):
                     credentials_path,
                     scopes=["https://www.googleapis.com/auth/earthengine"],
                 )
-                ee.Initialize(credentials)
+                if use_high_vol_endpoint == False:
+                    ee.Initialize(credentials)
+                else:
+                    ee.Initialize(
+                        credentials,
+                        opt_url="https://earthengine-highvolume.googleapis.com",
+                    )
                 print("EE initialized with credentials from:", credentials_path)
             else:
-                ee.Initialize()
+                if use_high_vol_endpoint == False:
+                    ee.Initialize()
+                else:
+                    ee.Initialize(
+                        opt_url="https://earthengine-highvolume.googleapis.com"
+                    )
                 print("EE initialized with default credentials.")
     except Exception as e:
         print("Error initializing EE:", e)
@@ -44,9 +58,18 @@ from openforis_whisp.stats import (
     whisp_formatted_stats_geojson_to_df,
     whisp_formatted_stats_geojson_to_geojson,
     set_point_geometry_area_to_zero,
+    reformat_geometry_type,
     convert_iso3_to_iso2,
 )
 
+from openforis_whisp.advanced_stats import (
+    whisp_stats_geojson_to_df_concurrent,
+    whisp_formatted_stats_geojson_to_df_concurrent,
+    whisp_stats_geojson_to_df_sequential,
+    whisp_formatted_stats_geojson_to_df_sequential,
+    whisp_formatted_stats_geojson_to_df_fast,
+)
+
 # temporary parameters to be removed once isio3 to iso2 conversion server side is implemented
 from openforis_whisp.parameters.config_runtime import (
     iso3_country_column,
@@ -59,6 +82,7 @@ from openforis_whisp.reformat import (
     validate_dataframe,
     create_schema_from_dataframe,
     load_schema_if_any_file_changed,
+    format_stats_dataframe,
     # log_missing_columns,
 )
 
@@ -73,3 +97,10 @@ from openforis_whisp.data_conversion import (
 from openforis_whisp.risk import whisp_risk, detect_unit_type
 
 from openforis_whisp.utils import get_example_data_path, generate_test_polygons
+
+from openforis_whisp.data_checks import (
+    analyze_geojson,
+    validate_geojson_constraints,
+    _check_metric_constraints,
+    suggest_method,
+)