Skip to content

Commit

Permalink
Merge branch 'main' into ogrgmt-int32
Browse files Browse the repository at this point in the history
  • Loading branch information
seisman committed Aug 20, 2023
2 parents 2097add + a245b6c commit a4ddcfe
Show file tree
Hide file tree
Showing 6 changed files with 50 additions and 46 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/dvc-diff.yml
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ jobs:

# workaround from https://github.com/iterative/cml/issues/1377
- name: Setup NodeJS
uses: actions/setup-node@v3.7.0
uses: actions/setup-node@v3.8.0
with:
node-version: '16'

Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/publish-to-pypi.yml
Original file line number Diff line number Diff line change
Expand Up @@ -68,10 +68,10 @@ jobs:
ls -lh dist/
- name: Publish to Test PyPI
uses: pypa/gh-action-pypi-publish@v1.8.8
uses: pypa/gh-action-pypi-publish@v1.8.10
with:
repository-url: https://test.pypi.org/legacy/

- name: Publish to PyPI
if: startsWith(github.ref, 'refs/tags')
uses: pypa/gh-action-pypi-publish@v1.8.8
uses: pypa/gh-action-pypi-publish@v1.8.10
1 change: 1 addition & 0 deletions MANIFEST.in
Original file line number Diff line number Diff line change
Expand Up @@ -11,3 +11,4 @@ exclude CONTRIBUTING.md
exclude Makefile
exclude environment.yml
exclude requirements.txt
exclude pygmt/tests/baseline/*.dvc
5 changes: 4 additions & 1 deletion pygmt/datasets/earth_mask.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,4 +89,7 @@ def load_earth_mask(resolution="01d", region=None, registration=None):
region=region,
registration=registration,
)
return grid.astype("int8")
# `return grid.astype("int8")` doesn't work because grid encoding is lost.
# See https://github.com/GenericMappingTools/pygmt/issues/2629.
grid.data = grid.data.astype("int8")
return grid
81 changes: 39 additions & 42 deletions pygmt/src/grdfilter.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,16 +25,17 @@
V="verbose",
f="coltypes",
r="registration",
x="cores",
)
@kwargs_to_strings(I="sequence", R="sequence")
def grdfilter(grid, **kwargs):
r"""
Filter a grid in the space (or time) domain.
Filter a grid file in the time domain using one of the selected convolution
or non-convolution isotropic or rectangular filters and compute distances
using Cartesian or Spherical geometries. The output grid file can
optionally be generated as a sub-region of the input (via ``region``)
Filter a grid file in the space (or time) domain using one of the selected
convolution or non-convolution isotropic or rectangular filters and compute
distances using Cartesian or Spherical geometries. The output grid file
can optionally be generated as a sub-region of the input (via ``region``)
and/or with new increment (via ``spacing``) or registration
(via ``toggle``). In this way, one may have "extra space" in the input
data so that the edges will not be used and the output can be within one
Expand All @@ -48,67 +49,63 @@ def grdfilter(grid, **kwargs):
Parameters
----------
grid : str or xarray.DataArray
The file name of the input grid or the grid loaded as a DataArray.
The file name of the input grid or the grid loaded as a
:class:`xarray.DataArray`.
outgrid : str or None
The name of the output netCDF file with extension .nc to store the grid
in.
filter : str
**b**\|\ **c**\|\ **g**\|\ **o**\|\ **m**\|\ **p**\|\ **h**\ *xwidth*\
**b**\|\ **c**\|\ **g**\|\ **o**\|\ **m**\|\ **p**\|\ **h**\ *width*\
[/*width2*\][*modifiers*].
Name of filter type you which to apply, followed by the width:
Name of the filter type you wish to apply, followed by the *width*:
b: Box Car
- **b** - Box Car
- **c** - Cosine Arch
- **g** - Gaussian
- **o** - Operator
- **m** - Median
- **p** - Maximum Likelihood probability
- **h** - Histogram
c: Cosine Arch
g: Gaussian
o: Operator
m: Median
p: Maximum Likelihood probability
h: histogram
distance : str
Distance *flag* tells how grid (x,y) relates to filter width as
follows:
State how the grid (x,y) relates to the filter *width*:
p: grid (px,py) with *width* an odd number of pixels; Cartesian
distances.
0: grid (x,y) same units as *width*, Cartesian distances.
1: grid (x,y) in degrees, *width* in kilometers, Cartesian distances.
2: grid (x,y) in degrees, *width* in km, dx scaled by cos(middle y),
Cartesian distances.
- ``"p"``: grid (px,py) with *width* an odd number of pixels,
Cartesian distances.
- ``"0"``: grid (x,y) same units as *width*, Cartesian distances.
- ``"1"``: grid (x,y) in degrees, *width* in kilometers, Cartesian
distances.
- ``"2"``: grid (x,y) in degrees, *width* in km, dx scaled by
cos(middle y), Cartesian distances.
The above options are fastest because they allow weight matrix to be
computed only once. The next three options are slower because they
recompute weights for each latitude.
3: grid (x,y) in degrees, *width* in km, dx scaled by cosine(y),
Cartesian distance calculation.
4: grid (x,y) in degrees, *width* in km, Spherical distance
calculation.
5: grid (x,y) in Mercator ``projection="m1"`` img units, *width* in km,
Spherical distance calculation.
- ``"3"``: grid (x,y) in degrees, *width* in km, dx scaled by cos(y),
Cartesian distance calculation.
- ``"4"``: grid (x,y) in degrees, *width* in km, Spherical distance
calculation.
- ``"5"``: grid (x,y) in Mercator ``projection="m1"`` img units,
*width* in km, Spherical distance calculation.
{spacing}
nans : str or float
**i**\|\ **p**\|\ **r**.
Determine how NaN-values in the input grid affects the filtered output.
Determine how NaN-values in the input grid affect the filtered output.
Use **i** to ignore all NaNs in the calculation of the filtered value
[Default]. **r** is same as **i** except if the input node was NaN then
the output node will be set to NaN (only applies if both grids are
co-registered). **p** will force the filtered value to be NaN if any
grid nodes with NaN-values are found inside the filter circle.
{region}
toggle : bool
Toggle the node registration for the output grid so as to become the
opposite of the input grid. [Default gives the same registration as the
input grid].
Toggle the node registration for the output grid to get the opposite of
the input grid [Default gives the same registration as the input grid].
{verbose}
{coltypes}
{registration}
{cores}
Returns
-------
Expand Down
3 changes: 3 additions & 0 deletions pygmt/tests/test_datasets_earth_mask.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ def test_earth_mask_01d():
assert data.attrs["horizontal_datum"] == "WGS84"
assert data.shape == (181, 361)
assert data.gmt.registration == 0
assert data.gmt.gtype == 1
assert data.dtype == "int8"
npt.assert_allclose(data.lat, np.arange(-90, 91, 1))
npt.assert_allclose(data.lon, np.arange(-180, 181, 1))
Expand All @@ -52,6 +53,8 @@ def test_earth_mask_01d_with_region():
data = load_earth_mask(resolution="01d", region=[-7, 4, 13, 19])
assert data.shape == (7, 12)
assert data.gmt.registration == 0
assert data.gmt.gtype == 1
assert data.dtype == "int8"
npt.assert_allclose(data.lat, np.arange(13, 20, 1))
npt.assert_allclose(data.lon, np.arange(-7, 5, 1))
npt.assert_allclose(data[1, 5], 1)

0 comments on commit a4ddcfe

Please sign in to comment.