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

Jump to bottom

Customization of raster histogram bins #57

Draft
wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Customization of raster histogram bins #57

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
12 changes: 8 additions & 4 deletions rio_stac/stac.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -164,11 +164,11 @@ def get_eobands_info(
return eo_bands


def _get_stats(arr: numpy.ma.MaskedArray, **kwargs: Any) -> Dict:
def _get_stats(arr: numpy.ma.MaskedArray, bins: Union[int, List], **kwargs: Any) -> Dict:
"""Calculate array statistics."""
# Avoid non masked nan/inf values
numpy.ma.fix_invalid(arr, copy=False)
sample, edges = numpy.histogram(arr[~arr.mask])
sample, edges = numpy.histogram(arr[~arr.mask], bins=bins)
return {
"statistics": {
"mean": arr.mean().item(),
Expand All @@ -191,6 +191,7 @@ def _get_stats(arr: numpy.ma.MaskedArray, **kwargs: Any) -> Dict:
def get_raster_info( # noqa: C901
src_dst: Union[DatasetReader, DatasetWriter, WarpedVRT, MemoryFile],
max_size: int = 1024,
bins: Union[int, List] = 10
) -> List[Dict]:
"""Get raster metadata.

Expand Down Expand Up @@ -239,7 +240,8 @@ def get_raster_info( # noqa: C901

value.update(
_get_stats(
src_dst.read(indexes=band, out_shape=(height, width), masked=True)
src_dst.read(indexes=band, out_shape=(height, width), masked=True),
bins=bins
)
)
meta.append(value)
Expand Down Expand Up @@ -302,6 +304,7 @@ def create_stac_item(
with_raster: bool = False,
with_eo: bool = False,
raster_max_size: int = 1024,
raster_bins: Union[int, List] = 10,
geom_densify_pts: int = 0,
geom_precision: int = -1,
) -> pystac.Item:
Expand All @@ -324,6 +327,7 @@ def create_stac_item(
with_raster (bool): Add the `raster` extension and properties (default to False).
with_eo (bool): Add the `eo` extension and properties (default to False).
raster_max_size (int): Limit array size from which to get the raster statistics. Defaults to 1024.
raster_bins (int or list of scalars): Number of bins (or list of bin edges) in raster band histogram. Defaults to 10.
geom_densify_pts (int): Number of points to add to each edge to account for nonlinear edges transformation (Note: GDAL uses 21).
geom_precision (int): If >= 0, geometry coordinates will be rounded to this number of decimal.

Expand Down Expand Up @@ -392,7 +396,7 @@ def create_stac_item(
)

raster_info = {
"raster:bands": get_raster_info(dataset, max_size=raster_max_size)
"raster:bands": get_raster_info(dataset, max_size=raster_max_size, bins=raster_bins)
}

eo_info: Dict[str, List] = {}
Expand Down