geocif 0.0.1__tar.gz → 0.1.2__tar.gz

geocif-0.1.2/PKG-INFO

@@ -0,0 +1,80 @@
+Metadata-Version: 2.1
+Name: geocif
+Version: 0.1.2
+Summary: Models to visualize and forecast crop conditions and yields
+Home-page: https://ritviksahajpal.github.io/yield_forecasting/
+Author: Ritvik Sahajpal
+Author-email: ritvik@umd.edu
+License: MIT license
+Keywords: geocif
+Classifier: Development Status :: 2 - Pre-Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Natural Language :: English
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: arfs==2.2.5
+Requires-Dist: arrow==1.3.0
+Requires-Dist: Boruta==0.3
+Requires-Dist: BorutaShap==1.0.17
+Requires-Dist: BorutaShap==1.0.17
+Requires-Dist: Bottleneck==1.3.7
+Requires-Dist: catboost==1.2.5
+Requires-Dist: cycler==0.12.1
+Requires-Dist: GDAL==3.8.1
+Requires-Dist: geopandas==0.14.1
+Requires-Dist: geoprepare==0.4.8
+Requires-Dist: icclim==6.4.0
+Requires-Dist: kneed==0.8.5
+Requires-Dist: logzero==1.7.0
+Requires-Dist: MAPIE==0.8.3
+Requires-Dist: matplotlib==3.8.2
+Requires-Dist: neptune==1.10.3
+Requires-Dist: numpy==1.25.2
+Requires-Dist: optuna==3.5.0
+Requires-Dist: palettable==3.3.3
+Requires-Dist: pandas==2.2.2
+Requires-Dist: pangres==4.2.1
+Requires-Dist: pooch==1.8.1
+Requires-Dist: powershap==0.0.11
+Requires-Dist: pycallgraph2==1.1.3
+Requires-Dist: pygam==0.9.1
+Requires-Dist: pygeoutil==0.1.dev0
+Requires-Dist: pymannkendall==1.4.3
+Requires-Dist: python_dateutil==2.8.2
+Requires-Dist: Rbeast==0.1.19
+Requires-Dist: Rbeast==0.1.19
+Requires-Dist: scikit_learn==1.4.2
+Requires-Dist: scipy==1.13.0
+Requires-Dist: seaborn==0.13.2
+Requires-Dist: setuptools==68.2.2
+Requires-Dist: shap==0.45.0
+Requires-Dist: sklearn_genetic_opt==0.10.1
+Requires-Dist: SQLAlchemy==2.0.25
+Requires-Dist: statsmodels==0.14.0
+Requires-Dist: tqdm==4.66.1
+
+# geocif
+
+
+[![image](https://img.shields.io/pypi/v/geocif.svg)](https://pypi.python.org/pypi/geocif)
+[![image](https://img.shields.io/conda/vn/conda-forge/geocif.svg)](https://anaconda.org/conda-forge/geocif)
+
+
+**Models to visualize and forecast crop conditions and yields**
+
+
+-   Free software: MIT license
+-   Documentation: https://ritviksahajpal.github.io/geocif
+    
+
+## Features
+
+-   TODO
+
+## Credits
+
+This package was created with [Cookiecutter](https://github.com/cookiecutter/cookiecutter) and the [giswqs/pypackage](https://github.com/giswqs/pypackage) project template.

geocif-0.1.2/geocif/__init__.py

@@ -0,0 +1,7 @@
+"""Top-level package for geocif."""
+
+__author__ = """Ritvik Sahajpal"""
+__email__ = "ritvik@umd.edu"
+__version__ = "0.0.1"
+
+__all__ = ["ml", "cei"]

geocif-0.1.2/geocif/agmet/geoagmet.py

@@ -0,0 +1,170 @@
+import pandas as pd
+from tqdm import tqdm
+
+from geocif.backup import geo
+from geocif.agmet import plot, utils
+
+
+def create_title_for_plot(obj):
+    """
+
+    Args:
+        obj:
+
+    Returns:
+
+    """
+    region_name = obj.region.replace("_", " ").title()
+    calendar_region_name = obj.calendar_region.replace("_", " ").title()
+    country_name = obj.country.replace("_", " ").title()
+    long_crop_name = utils.get_crop_name(obj.crop)
+
+    # Get name of crop based on metadata/crop_per_season.csv file
+    df_crop_per_season = pd.read_csv(obj.dir_metadata / "crop_per_season.csv")
+
+    crop_name = df_crop_per_season[
+        (df_crop_per_season["country"] == obj.country)
+        & (df_crop_per_season["crop"] == obj.crop)
+        & (df_crop_per_season["season"] == int(obj.growing_season))
+    ]["name"].values
+
+    crop_name = crop_name[0] if crop_name else long_crop_name.replace("_", " ").title()
+
+    title_line_1 = f"{region_name} ({calendar_region_name}, {country_name})"
+    title_line_2 = f"{crop_name} {obj.plot_season}"
+
+    sup_title = f"{title_line_1}\n{title_line_2}"
+
+    return sup_title
+
+
+def loop_agmet(path_config_file=None):
+    """
+    Args:
+    """
+    # Create geo object
+    obj = geo.geo(path_config_file)
+
+    # Read in data on crop names in each country
+    obj.read_statistics(read_countries=True)
+
+    # Create combinations of run parameters
+    all_combinations = obj.create_run_combinations()
+
+    pbar = tqdm(all_combinations, total=len(all_combinations))
+    for country, scale, crop, growing_season in pbar:
+        # Setup country information
+        obj.setup_country(country, scale, crop, growing_season)
+
+        # Loop through seasons and plot for each admin_1 region and calendar region
+        for plot_season in obj.plot_seasons:
+            # Get list of the years that are closest to the year which we want to plot
+            obj.get_closest_season(plot_season)
+
+            for region in obj.list_regions:
+                # Setup the region information
+                obj.setup_region(region, plot_season, "region")
+
+                # If available, add CHIRPS-GEFS data to the dataframe
+                if obj.precip_var == "chirps":
+                    obj.add_precip_forecast(plot_season)
+
+                # Get crop calendar dates for region
+                dates_calendar = [
+                    obj.date_planting,
+                    obj.date_greenup,
+                    obj.date_senescence,
+                    obj.date_harvesting,
+                ]
+
+                # TODO: Only plot if dates_calendar has valid dates
+                # Create title for plot
+                sup_title = create_title_for_plot(obj)
+
+                ###############################################################
+                # Agmet plots for regions
+                ###############################################################
+                # D:\Dropbox\Projects\GEOGLAM\crop_condition\AMIS\argentina
+                plot.plots_ts_cur_yr(
+                    obj.df_region,
+                    obj.eo_plot,
+                    closest=obj.closest,
+                    dates_cal=dates_calendar,
+                    frcast_yr=obj.plot_season,
+                    logos=[obj.logo_harvest, obj.logo_geoglam],
+                    dir_out=obj.dir_agmet / obj.scale,
+                    sup_title=sup_title,
+                    fname=f"{obj.region}.png",
+                )
+
+                ###############################################################
+                # Agmet plots for calendar regions
+                ###############################################################
+                columns = obj.eo_model + ["month", "day", "yield"]
+                if "chirps" in obj.df_region.columns:
+                    bool_year_check, bool_date_check = obj.check_date(
+                        obj.df_region, obj.plot_season
+                    )
+
+                    if bool_date_check and bool_year_check:
+                        columns = obj.eo_model + [
+                            "month",
+                            "day",
+                            "chirps_gefs",
+                            "yield",
+                        ]
+
+                df_calendar_region = (
+                    obj.df_region.groupby(
+                        [
+                            "country",
+                            "calendar_region",
+                            "harvest_season",
+                            "doy",
+                            "datetime",
+                        ]
+                    )[columns]
+                    .mean()
+                    .reset_index()
+                )
+                df_calendar_region.loc[:, "average_temperature"] = (
+                    df_calendar_region["cpc_tmax"] + df_calendar_region["cpc_tmin"]
+                ) / 2.0
+                df_calendar_region.set_index(
+                    pd.DatetimeIndex(df_calendar_region["datetime"]),
+                    inplace=True,
+                    drop=True,
+                )
+                df_calendar_region.index.name = None
+                df_calendar_region.sort_values(by="datetime", inplace=True)
+
+                # TODO: Only plot if dates_calendar has valid dates
+                # Create title for plot
+                sup_title = create_title_for_plot(obj)
+
+                if not df_calendar_region.empty:
+                    plot.plots_ts_cur_yr(
+                        df_calendar_region,
+                        obj.eo_plot,
+                        closest=obj.closest,
+                        dates_cal=dates_calendar,
+                        frcast_yr=obj.plot_season,
+                        logos=[obj.logo_harvest, obj.logo_geoglam],
+                        dir_out=obj.dir_agmet / "district",
+                        sup_title=sup_title,
+                        fname=f"{obj.calendar_region}.png",
+                    )
+
+
+def run():
+    loop_agmet(
+        [
+            "D:/Users/ritvik/projects/geoprepare/geoprepare/geoprepare.txt",
+            "D:/Users/ritvik/projects/geoprepare/geoprepare/geoextract.txt",
+            "D:/Users/ritvik/projects/geocif/geocif/config/geocif.txt",
+        ]
+    )
+
+
+if __name__ == "__main__":
+    run()