# This file is part of pyunicorn.
# Copyright (C) 2008--2024 Jonathan F. Donges and pyunicorn authors
# URL: <https://www.pik-potsdam.de/members/donges/software-2/software>
# License: BSD (3-clause)
#
# Please acknowledge and cite the use of this software and its authors
# when results are used in publications or published elsewhere.
#
# You can use the following reference:
# J.F. Donges, J. Heitzig, B. Beronov, M. Wiedermann, J. Runge, Q.-Y. Feng,
# L. Tupikina, V. Stolbova, R.V. Donner, N. Marwan, H.A. Dijkstra,
# and J. Kurths, "Unified functional network and nonlinear time series analysis
# for complex systems science: The pyunicorn package"
"""
Provides class for horizontal two-dimensional spatio-temporal grid.
"""
from typing import Tuple
from collections.abc import Hashable
import numpy as np
# Import package to calculate points inside a polygon
try:
from matplotlib import path
except ImportError:
print("An error occurred when importing matplotlib.path! "
"Some functionality in GeoGrid class might not be available.")
from ._ext.types import to_cy, FIELD
from ._ext.numerics import _calculate_angular_distance
from .cache import Cached
from .grid import Grid
[docs]
class GeoGrid(Grid):
"""
Encapsulates a horizontal two-dimensional spatio-temporal grid on the
sphere.
The spatial grid points can be arbitrarily distributed, which is useful
for representing station data or geodesic grids.
"""
#
# Definitions of internal methods
#
[docs]
def __init__(self, time_seq: np.ndarray,
lat_seq: np.ndarray, lon_seq: np.ndarray,
silence_level: int = 0):
"""
Initialize an instance of GeoGrid.
:type time_seq: 1D Numpy array [time]
:arg time_seq: The increasing sequence of temporal sampling points.
:type lat_seq: 1D Numpy array [index]
:arg lat_seq: The sequence of latitudinal sampling points.
:type lon_seq: 1D Numpy array [index]
:arg lon_seq: The sequence of longitudinal sampling points.
:type silence_level: number (int)
:arg silence_level: The inverse level of verbosity of the object.
"""
Grid.__init__(self, time_seq, np.vstack((lat_seq, lon_seq)),
silence_level)
[docs]
def __cache_state__(self) -> Tuple[Hashable, ...]:
return Grid.__cache_state__(self)
[docs]
def __str__(self):
"""
Return a string representation of the GeoGrid object.
"""
return (f"GeoGrid: {self._grid_size['space']} grid points, "
f"{self._grid_size['time']} timesteps.")
#
# Functions for loading and saving the Grid object
#
[docs]
def save_txt(self, filename):
"""
Save the GeoGrid object to text files.
The latitude, longitude and time sequences are stored in three separate
text files.
:arg str filename: The name of the files where Grid object is stored
(excluding ending).
"""
# Gather sequences
lat_seq = self.lat_sequence()
lon_seq = self.lon_sequence()
time_seq = self.grid()["time"]
# Store as text files
try:
np.savetxt(filename + "_lat.txt", lat_seq)
np.savetxt(filename + "_lon.txt", lon_seq)
np.savetxt(filename + "_time.txt", time_seq)
except IOError:
print("An error occurred while saving Grid instance to "
f"text files {filename}")
[docs]
@staticmethod
def LoadTXT(filename):
"""
Return a GeoGrid object stored in text files.
The latitude, longitude and time sequences are loaded from three
separate text files.
:arg str filename: The name of the files where the GeoGrid object is
stored (excluding endings).
:rtype: Grid object
:return: :class:`GeoGrid` instance.
"""
try:
lat_seq = np.loadtxt(filename + "_lat.txt")
lon_seq = np.loadtxt(filename + "_lon.txt")
time_seq = np.loadtxt(filename + "_time.txt")
except IOError:
print("An error occurred while loading Grid instance from "
f"text files {filename}")
return GeoGrid(time_seq, lat_seq, lon_seq)
#
# Alternative constructors and Grid generation methods
#
[docs]
@staticmethod
def SmallTestGrid():
"""
Return test grid of 6 spatial grid points with 10 temporal sampling
points each.
:rtype: GeoGrid instance
:return: a GeoGrid instance for testing purposes.
"""
return GeoGrid(time_seq=np.arange(10),
lat_seq=np.array([0, 5, 10, 15, 20, 25]),
lon_seq=np.array([2.5, 5., 7.5, 10., 12.5, 15.]),
silence_level=2)
[docs]
@staticmethod
def RegularGrid(time_seq, space_grid, silence_level=0):
"""
Initialize an instance of a regular grid.
**Examples:**
>>> GeoGrid.RegularGrid(
... time_seq=np.arange(2),
... space_grid=(np.array([0.,5.]),
... np.array([1.,2.])),
... silence_level=2).lat_sequence()
array([ 0., 0., 5., 5.], dtype=float32)
>>> GeoGrid.RegularGrid(
... time_seq=np.arange(2),
... space_grid=(np.array([0.,5.]),
... np.array([1.,2.])),
... silence_level=2).lon_sequence()
array([ 1., 2., 1., 2.], dtype=float32)
:type time_seq: 1D Numpy array [time]
:arg time_seq: The increasing sequence of temporal sampling points.
:type space_grid: tuple or list of two 1D Numpy arrays
([n_lat], [n_lon])
:arg space_grid: The spatial grid, consisting of the latitudinal
and the longitudinal grid.
:type silence_level: number (int)
:arg silence_level: The inverse level of verbosity of the object.
:rtype: GeoGrid object
:return: :class:`GeoGrid` instance.
"""
try:
(lat_grid, lon_grid) = space_grid
except ValueError as e:
raise ValueError("'space_grid' must be a tuple or list of two "
"items: lat_grid, lon_grid") from e
# Generate sequence of latitudes and longitudes for all nodes
lat_seq, lon_seq = GeoGrid.coord_sequence_from_rect_grid(lat_grid,
lon_grid)
# Return instance of Grid
return GeoGrid(time_seq, lat_seq, lon_seq, silence_level)
#
# Definitions of grid related functions
#
[docs]
@staticmethod
def coord_sequence_from_rect_grid(lat_grid, lon_grid):
"""
Return the sequences of latitude and longitude for a regular and
rectangular grid.
**Example:**
>>> GeoGrid.coord_sequence_from_rect_grid(
... lat_grid=np.array([0.,5.]), lon_grid=np.array([1.,2.]))
(array([ 0., 0., 5., 5.]), array([ 1., 2., 1., 2.]))
:type lat_grid: 1D Numpy array [lat]
:arg lat_grid: The grid's latitudinal sampling points.
:type lon_grid: 1D Numpy array [lon]
:arg lon_grid: The grid's longitudinal sampling points.
:rtype: tuple of two 1D Numpy arrays [index]
:return: the coordinates of all nodes in the grid.
"""
space_seq = Grid.coord_sequence_from_rect_grid([lat_grid, lon_grid])
# Return results as a tuple
return (space_seq[0], space_seq[1])
[docs]
def lat_sequence(self):
"""
Return the sequence of latitudes for all nodes.
**Example:**
>>> GeoGrid.SmallTestGrid().lat_sequence()
array([ 0., 5., 10., 15., 20., 25.], dtype=float32)
:rtype: 1D Numpy array [index]
:return: the sequence of latitudes for all nodes.
"""
return self.sequence(0)
[docs]
def lon_sequence(self):
"""
Return the sequence of longitudes for all nodes.
**Example:**
>>> GeoGrid.SmallTestGrid().lon_sequence()
array([ 2.5, 5. , 7.5, 10. , 12.5, 15. ], dtype=float32)
:rtype: 1D Numpy array [index]
:return: the sequence of longitudes for all nodes.
"""
return self.sequence(1)
[docs]
def convert_lon_coordinates(self, lon_seq):
"""
Return longitude coordinates in the system
-180 deg W <= lon <= +180 deg O for all nodes.
Accepts longitude coordinates in the system 0 deg <= lon <= 360 deg.
0 deg corresponds to Greenwich, England.
**Example:**
>>> GeoGrid.SmallTestGrid().convert_lon_coordinates(
... np.array([10.,350.,20.,340.,170.,190.]))
array([ 10., -10., 20., -20., 170., -170.])
:type lon_seq: 1D Numpy array [index]
:arg lon_seq: Sequence of longitude coordinates.
:rtype: 1D Numpy array [index]
:return: the converted longitude coordinates for all nodes.
"""
new_lon_grid = np.empty(self.N)
for i in range(self.N):
if lon_seq[i] > 180.:
new_lon_grid[i] = lon_seq[i] - 360.
else:
new_lon_grid[i] = lon_seq[i]
return new_lon_grid
[docs]
def node_number(self, lat_node, lon_node):
"""
Return the index of the closest node given geographical coordinates.
**Example:**
>>> GeoGrid.SmallTestGrid().node_number(lat_node=14., lon_node=9.)
3
:type lat_node: number (float)
:arg lat_node: The latitude coordinate.
:type lon_node: number (float)
:arg lon_node: The longitude coordinate.
:rtype: number (int)
:return: the closest node's index.
"""
# Get sequences of cosLat, sinLat, cosLon and sinLon for all nodes
cos_lat = self.cos_lat()
sin_lat = self.sin_lat()
cos_lon = self.cos_lon()
sin_lon = self.sin_lon()
sin_lat_v = np.sin(lat_node * np.pi / 180)
cos_lat_v = np.cos(lat_node * np.pi / 180)
sin_lon_v = np.sin(lon_node * np.pi / 180)
cos_lon_v = np.cos(lon_node * np.pi / 180)
# Calculate angular distance from the given coordinate to all
# other nodes
expr = sin_lat*sin_lat_v + cos_lat*cos_lat_v * (sin_lon*sin_lon_v
+ cos_lon*cos_lon_v)
# Correct for rounding errors
expr[expr < -1.] = -1.
expr[expr > 1.] = 1.
angdist = np.arccos(expr)
# Get index of closest node
n_node = angdist.argmin()
return n_node
[docs]
def cos_lat(self):
"""
Return the sequence of cosines of latitude for all nodes.
**Example:**
>>> r(GeoGrid.SmallTestGrid().cos_lat()[:2])
array([ 1. , 0.9962])
:rtype: 1D Numpy array [index]
:return: the cosine of latitudes for all nodes.
"""
return np.cos(self.lat_sequence() * np.pi / 180)
[docs]
def sin_lat(self):
"""
Return the sequence of sines of latitude for all nodes.
**Example:**
>>> r(GeoGrid.SmallTestGrid().sin_lat()[:2])
array([ 0. , 0.0872])
:rtype: 1D Numpy array [index]
:return: the sine of latitudes for all nodes.
"""
return np.sin(self.lat_sequence() * np.pi / 180)
[docs]
def cos_lon(self):
"""
Return the sequence of cosines of longitude for all nodes.
**Example:**
>>> r(GeoGrid.SmallTestGrid().cos_lon()[:2])
array([ 0.999 , 0.9962])
:rtype: 1D Numpy array [index]
:return: the cosine of longitudes for all nodes.
"""
return np.cos(self.lon_sequence() * np.pi / 180)
[docs]
def sin_lon(self):
"""
Return the sequence of sines of longitude for all nodes.
**Example:**
>>> r(GeoGrid.SmallTestGrid().sin_lon()[:2])
array([ 0.0436, 0.0872])
:rtype: 1D Numpy array [index]
:return: the sine of longitudes for all nodes.
"""
return np.sin(self.lon_sequence() * np.pi / 180)
[docs]
def distance(self):
"""
Calculate and return the standard distance matrix of the corresponding
grid type
:rtype: 2D Numpy array [index, index]
:return: the distance matrix.
"""
return self.angular_distance()
[docs]
@Cached.method(name="angular great circle distance")
def angular_distance(self):
"""
Calculate the angular great circle distance matrix.
**No normalization applied anymore!** Return values are in the range
0 to Pi.
**Example:**
>>> rr(GeoGrid.SmallTestGrid().angular_distance(), 2)
[['0' '0.1' '0.19' '0.29' '0.39' '0.48']
['0.1' '0' '0.1' '0.19' '0.29' '0.39']
['0.19' '0.1' '0' '0.1' '0.19' '0.29']
['0.29' '0.19' '0.1' '0' '0.1' '0.19']
['0.39' '0.29' '0.19' '0.1' '0' '0.1']
['0.48' '0.39' '0.29' '0.19' '0.1' '0']]
:rtype: 2D Numpy array [index, index]
:return: the angular great circle distance matrix.
"""
# Get number of nodes
N = self.N
# Initialize cython cof of angular distance matrix
cosangdist = np.zeros((N, N), dtype=FIELD)
_calculate_angular_distance(
# Get sequences of cosLat, sinLat, cosLon and sinLon for all nodes
to_cy(self.cos_lat(), FIELD),
to_cy(self.sin_lat(), FIELD),
to_cy(self.cos_lon(), FIELD),
to_cy(self.sin_lon(), FIELD),
cosangdist, N)
return np.arccos(cosangdist)
[docs]
def boundaries(self):
"""
Return the spatio-temporal grid boundaries.
Structure of the returned dictionary:
- boundaries = {"time_min": self._boundaries["time_min"],
"time_max": self._boundaries["time_max"],
"lat_min": self._boundaries["space_min"][0],
"lat_max": self._boundaries["space_max"][1],
"lon_min": self._boundaries["space_min"][0],
"lon_max": self._boundaries["space_max"][1]}
:rtype: dictionary
:return: the spatio-temporal grid boundaries.
"""
boundaries = {"time_min": self._boundaries["time_min"],
"time_max": self._boundaries["time_max"],
"lat_min": self._boundaries["space_min"][0],
"lat_max": self._boundaries["space_max"][0],
"lon_min": self._boundaries["space_min"][1],
"lon_max": self._boundaries["space_max"][1]}
return boundaries
[docs]
def print_boundaries(self):
"""
Pretty print the spatio-temporal grid boundaries.
**Example:**
>>> print(GeoGrid.SmallTestGrid().print_boundaries())
time lat lon
min 0.0 0.00 2.50
max 9.0 25.00 15.00
:rtype: string
:return: printable string for the spatio-temporal grid boundaries
"""
return (
" time lat lon"
"\n min {time_min:6.1f} {lat_min: 7.2f} {lon_min: 7.2f}"
"\n max {time_max:6.1f} {lat_max: 7.2f} {lon_max: 7.2f}"
).format(**self.boundaries())
[docs]
def grid(self):
"""
Return the grid's spatio-temporal sampling points.
Structure of the returned dictionary:
- grid = {"time": self._grid["time"],
"lat": self._grid["space"][0],
"lon": self._grid["space"][1]}
**Examples:**
>>> Grid.SmallTestGrid().grid()["space"][0]
array([ 0., 5., 10., 15., 20., 25.], dtype=float32)
>>> Grid.SmallTestGrid().grid()["space"][0][5]
15.0
:rtype: dictionary
:return: the grid's spatio-temporal sampling points.
"""
grid = {"time": self._grid["time"],
"lat": self._grid["space"][0],
"lon": self._grid["space"][1]}
return grid
#
# Methods for selecting regions
#
[docs]
def region_indices(self, region):
"""
Returns a boolean array of nodes with True values when the node
is inside the region.
**Example:**
>>> GeoGrid.SmallTestGrid().region_indices(
... np.array([0.,0.,0.,11.,11.,11.,11.,0.])).astype(int)
array([0, 1, 1, 0, 0, 0])
:type region: 1D Numpy array [n_polygon_nodes]
:arg region: array of lon, lat, lon, lat, ...
[-80.2, 5., -82.4, 5.3, ...] as copied from Google Earth
Polygon file
:rtype: 1D bool array [index]
:return: bool array with True for nodes inside region
"""
# Reshape Google Earth array into (n,2) array
remapped_region = region.reshape(len(region)//2, 2)
# Remap from East-West to 360 degree map if the longitudes are [0, 360]
if self._grid["space"][1].min() >= 0:
remapped_region[remapped_region[:, 0] < 0, 0] = \
360 + remapped_region[remapped_region[:, 0] < 0, 0]
lat_lon_map = np.column_stack((self._grid["space"][1],
self._grid["space"][0]))
return path.Path(remapped_region).contains_points(lat_lon_map)
[docs]
@staticmethod
def region(name):
"""Return some standard regions."""
if name == 'ENSO':
return np.array([-79.28273150749884, -10.49311965331937,
-79.29849791038734, 10.12527300655218,
-174.9221853596061, 10.07293121423917,
-174.8362810586096, -10.46407198776264,
-80.13229308153623, -10.36724072894785,
-79.28273150749884, -10.49311965331937])
elif name == 'NINO34':
return np.array([-118.6402427933005, 7.019906838300821,
-171.0067408177714, 6.215022481004243,
-171.0364908514962, -5.768616252424354,
-119.245702264066, -5.836385150138187,
-118.6402427933005, 7.019906838300821])
else:
return None