Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix nodata handling in wofe CLI functions #478

Merged
merged 1 commit into from
Jan 9, 2025
Merged

fix nodata handling in wofe CLI functions #478

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions eis_toolkit/cli.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,8 @@
from beartype.typing import List, Optional, Tuple, Union
from typing_extensions import Annotated

from eis_toolkit.utilities.nodata import nan_to_nodata, nodata_to_nan

app = typer.Typer()


Expand Down Expand Up @@ -3135,6 +3137,7 @@ def weights_of_evidence_calculate_weights_cli(
out_rasters_dict = {}
file_name = input_raster.name.split(".")[0]
for key, array in arrays.items():
array = nan_to_nodata(array, raster_meta["nodata"])
output_raster_name = file_name + "_weights_" + weights_type + "_" + key
output_raster_path = output_dir.joinpath(output_raster_name + ".tif")
with rasterio.open(output_raster_path, "w", **raster_meta) as dst:
Expand Down Expand Up @@ -3183,13 +3186,15 @@ def weights_of_evidence_calculate_responses_cli(
if raster_weights is not None:
with rasterio.open(raster_weights) as src:
array_W = src.read(1)
array_W = nodata_to_nan(array_W, src.nodata)

if raster_profile is None:
raster_profile = src.profile

if raster_std is not None:
with rasterio.open(raster_std) as src:
array_S_W = src.read(1)
array_S_W = nodata_to_nan(array_S_W, src.nodata)

dict_array.append({"W+": array_W, "S_W+": array_S_W})

Expand All @@ -3200,12 +3205,15 @@ def weights_of_evidence_calculate_responses_cli(
)
typer.echo("Progress: 75%")

posterior_probabilities = nan_to_nodata(posterior_probabilities, raster_profile["nodata"])
with rasterio.open(output_probabilities, "w", **raster_profile) as dst:
dst.write(posterior_probabilities, 1)

posterior_probabilies_std = nan_to_nodata(posterior_probabilies_std, raster_profile["nodata"])
with rasterio.open(output_probabilities_std, "w", **raster_profile) as dst:
dst.write(posterior_probabilies_std, 1)

confidence_array = nan_to_nodata(confidence_array, raster_profile["nodata"])
with rasterio.open(output_confidence_array, "w", **raster_profile) as dst:
dst.write(confidence_array, 1)

Expand All @@ -3230,9 +3238,11 @@ def agterberg_cheng_CI_test_cli(

with rasterio.open(input_posterior_probabilities) as src:
posterior_probabilities = src.read(1)
posterior_probabilities = nodata_to_nan(posterior_probabilities, src.nodata)

with rasterio.open(input_posterior_probabilities_std) as src:
posterior_probabilities_std = src.read(1)
posterior_probabilities_std = nodata_to_nan(posterior_probabilities_std, src.nodata)

typer.echo("Progress: 25%")

Expand Down
Loading