Skip to content

Commit

Permalink
Fixed interpolation methods for tides
Browse files Browse the repository at this point in the history
  • Loading branch information
jreniel committed Apr 5, 2021
1 parent 417e25d commit 60a4ce1
Show file tree
Hide file tree
Showing 3 changed files with 22 additions and 7 deletions.
2 changes: 0 additions & 2 deletions adcircpy/cmd/best_track_run.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,4 @@ def main():
)
logging.Formatter.converter = lambda *args: datetime.now(
tz=timezone('UTC')).timetuple()
# if len(args.constituents) == 0:
# args.constituents = ['all']
BestTrackRunCommand(args).run()
14 changes: 11 additions & 3 deletions adcircpy/forcing/tides/hamtide.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,13 +134,21 @@ def _get_interpolation(
yi = yi.flatten()
dataset = self._get_dataset(variable, constituent)
zi = dataset[netcdf_variable][yidx, xidx].flatten()

return griddata(
values = griddata(
(xi[~zi.mask], yi[~zi.mask]),
zi[~zi.mask],
(xq, yq),
method='linear'
method='linear',
fill_value=np.nan
)
nan_idxs = np.where(np.isnan(values))
values[nan_idxs] = griddata(
(xi[~zi.mask], yi[~zi.mask]),
zi[~zi.mask],
(xq[nan_idxs], yq[nan_idxs]),
method='nearest',
)
return values

def _prepend_path(self, filename: str) -> str:
if self.path is None:
Expand Down
13 changes: 11 additions & 2 deletions adcircpy/forcing/tides/tpxo.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,9 +118,18 @@ def _get_interpolation(self, tpxo_array: np.ndarray, constituent: str,
)

# "method" can be 'spline' or any string accepted by griddata()'s method kwarg.
return griddata(
values = griddata(
(x[_idx], y[_idx]),
array[_idx],
(_x, _y),
method='linear'
method='linear',
fill_value=np.nan,
)
nan_idxs = np.where(np.isnan(values))
values[nan_idxs] = griddata(
(x[_idx], y[_idx]),
array[_idx],
(_x[nan_idxs], _y[nan_idxs]),
method='nearest',
)
return values

0 comments on commit 60a4ce1

Please sign in to comment.