Index

_

__init__() (pyshqg.backend.abstract.Backend method) (pyshqg.core.dissipation.QGEkmanDissipation method) (pyshqg.core.dissipation.QGSelectiveDissipation method) (pyshqg.core.dissipation.QGThermalDissipation method) (pyshqg.core.integration.RungeKuttaModelIntegrator method) (pyshqg.core.model.QGModelTrajectory method) (pyshqg.core.poisson.QGPoissonSolver method) (pyshqg.core.source.QGForcing method) (pyshqg.core.spectral_transformations.SpectralTransformations method) (pyshqg.preprocessing.vertical_parametrisation.VerticalParametrisation method)

_add_padding() (in module pyshqg.preprocessing.reference_data) _data_file() (in module pyshqg.preprocessing.reference_data) _num_batch_dimensions() (pyshqg.core.model.QGModelTrajectory method) _open_zarr_data() (in module pyshqg.preprocessing.reference_data)

A

append() (pyshqg.core.model.QGModelTrajectory method) apply_euler_step() (pyshqg.core.model.QGModel method) apply_fft() (pyshqg.backend.abstract.Backend static method) (pyshqg.backend.numpy_backend.NumpyBackend static method) (pyshqg.backend.tensorflow_backend.TensorflowBackend static method)

apply_ifft() (pyshqg.backend.abstract.Backend static method) (pyshqg.backend.numpy_backend.NumpyBackend static method) (pyshqg.backend.tensorflow_backend.TensorflowBackend static method)

B

Backend (class in pyshqg.backend.abstract) backend (pyshqg.core.dissipation.QGEkmanDissipation attribute) (pyshqg.core.dissipation.QGSelectiveDissipation attribute) (pyshqg.core.dissipation.QGThermalDissipation attribute) (pyshqg.core.model.QGModel attribute) (pyshqg.core.poisson.QGPoissonSolver attribute) (pyshqg.core.spectral_transformations.SpectralTransformations attribute)

C

compute_dependent_variable() (pyshqg.core.model.QGModel method) compute_dependent_variables() (pyshqg.core.model.QGModel method) compute_ekman_dissipation() (pyshqg.core.dissipation.QGEkmanDissipation method) compute_forcing() (pyshqg.core.source.QGForcing method) compute_model_tendencies() (pyshqg.core.model.QGModel method) compute_selective_dissipation() (pyshqg.core.dissipation.QGSelectiveDissipation method)

compute_thermal_dissipation() (pyshqg.core.dissipation.QGThermalDissipation method) concatenate() (pyshqg.backend.abstract.Backend static method) (pyshqg.backend.numpy_backend.NumpyBackend static method) (pyshqg.backend.tensorflow_backend.TensorflowBackend static method) construct_integrator() (in module pyshqg.core.constructors) construct_model() (in module pyshqg.core.constructors) core_dimensions (pyshqg.core.model.QGModelTrajectory attribute)

D

dissipation (pyshqg.core.model.QGModel attribute) dmu_dphi (pyshqg.core.dissipation.QGEkmanDissipation attribute)

dmu_dtheta (pyshqg.core.dissipation.QGEkmanDissipation attribute) dt (pyshqg.core.integration.RungeKuttaModelIntegrator attribute)

E

einsum() (pyshqg.backend.abstract.Backend static method) (pyshqg.backend.numpy_backend.NumpyBackend static method) (pyshqg.backend.tensorflow_backend.TensorflowBackend static method)

ekman (pyshqg.core.dissipation.QGDissipation attribute) expand_dims() (pyshqg.backend.abstract.Backend static method) (pyshqg.backend.numpy_backend.NumpyBackend static method) (pyshqg.backend.tensorflow_backend.TensorflowBackend static method)

F

floatx (pyshqg.backend.abstract.Backend attribute) forcing (pyshqg.core.model.QGModel attribute) (pyshqg.core.source.QGForcing attribute)

forward() (pyshqg.core.integration.RungeKuttaModelIntegrator method) from_numpy() (pyshqg.backend.abstract.Backend method) (pyshqg.backend.numpy_backend.NumpyBackend method) (pyshqg.backend.tensorflow_backend.TensorflowBackend method)

G

grid_to_spec() (pyshqg.core.spectral_transformations.SpectralTransformations method)

I

interpolate_data() (in module pyshqg.preprocessing.reference_data)

L

land_sea_mask (pyshqg.preprocessing.orography.Orography attribute) laplacian_spectrum (pyshqg.core.spectral_transformations.SpectralTransformations attribute) lats (pyshqg.core.spectral_transformations.SpectralTransformations attribute)

legendre (pyshqg.core.spectral_transformations.SpectralTransformations attribute) load_reference_data() (in module pyshqg.preprocessing.reference_data) load_test_data() (in module pyshqg.preprocessing.reference_data) lons (pyshqg.core.spectral_transformations.SpectralTransformations attribute)

M

model (pyshqg.core.integration.RungeKuttaModelIntegrator attribute) (pyshqg.core.model.QGModelTrajectory attribute) model_state() (pyshqg.core.model.QGModel static method) model_trajectory() (pyshqg.core.model.QGModel method) module pyshqg.backend.abstract pyshqg.backend.numpy_backend pyshqg.backend.tensorflow_backend pyshqg.core.constructors pyshqg.core.dissipation pyshqg.core.integration pyshqg.core.model pyshqg.core.poisson pyshqg.core.source pyshqg.core.spectral_transformations pyshqg.preprocessing.orography pyshqg.preprocessing.reference_data pyshqg.preprocessing.vertical_parametrisation

mu (pyshqg.core.dissipation.QGEkmanDissipation attribute)

N

num_levels (pyshqg.preprocessing.vertical_parametrisation.VerticalParametrisation attribute)

NumpyBackend (class in pyshqg.backend.numpy_backend)

O

Orography (class in pyshqg.preprocessing.orography)

orography (pyshqg.preprocessing.orography.Orography attribute)

P

pad() (pyshqg.backend.abstract.Backend static method) (pyshqg.backend.numpy_backend.NumpyBackend static method) (pyshqg.backend.tensorflow_backend.TensorflowBackend static method) planet_omega (pyshqg.core.spectral_transformations.SpectralTransformations attribute) planet_radius (pyshqg.core.spectral_transformations.SpectralTransformations attribute) poisson_solver (pyshqg.core.model.QGModel attribute) precompute_coupling_matrix() (pyshqg.preprocessing.vertical_parametrisation.VerticalParametrisation method) precompute_ekman_friction() (pyshqg.preprocessing.orography.Orography method) precompute_grid_nodes() (pyshqg.core.spectral_transformations.SpectralTransformations method) precompute_laplacian_spectrum() (pyshqg.core.spectral_transformations.SpectralTransformations method) precompute_legendre_coefficients() (pyshqg.core.spectral_transformations.SpectralTransformations method) precompute_planet_vorticity() (pyshqg.core.poisson.QGPoissonSolver method) (pyshqg.core.spectral_transformations.SpectralTransformations method) precompute_q_to_psi_coupling() (pyshqg.core.poisson.QGPoissonSolver method) precorrect_planet_vorticity() (pyshqg.preprocessing.orography.Orography method) psi_to_zeta() (pyshqg.core.poisson.QGPoissonSolver method) pyshqg.backend.abstract module pyshqg.backend.numpy_backend module pyshqg.backend.tensorflow_backend module

pyshqg.core.constructors module pyshqg.core.dissipation module pyshqg.core.integration module pyshqg.core.model module pyshqg.core.poisson module pyshqg.core.source module pyshqg.core.spectral_transformations module pyshqg.preprocessing.orography module pyshqg.preprocessing.reference_data module pyshqg.preprocessing.vertical_parametrisation module

Q

q_to_psi_coupling (pyshqg.core.poisson.QGPoissonSolver attribute) q_to_total_q() (pyshqg.core.poisson.QGPoissonSolver method) QGDissipation (class in pyshqg.core.dissipation) QGEkmanDissipation (class in pyshqg.core.dissipation) QGForcing (class in pyshqg.core.source)

QGModel (class in pyshqg.core.model) QGModelTrajectory (class in pyshqg.core.model) QGPoissonSolver (class in pyshqg.core.poisson) QGSelectiveDissipation (class in pyshqg.core.dissipation) QGThermalDissipation (class in pyshqg.core.dissipation) qp (pyshqg.core.spectral_transformations.SpectralTransformations attribute)

R

range() (pyshqg.backend.abstract.Backend method) (pyshqg.backend.numpy_backend.NumpyBackend method) (pyshqg.backend.tensorflow_backend.TensorflowBackend method) repeat() (pyshqg.backend.abstract.Backend static method) (pyshqg.backend.numpy_backend.NumpyBackend static method) (pyshqg.backend.tensorflow_backend.TensorflowBackend static method)

rossby_radius_list (pyshqg.preprocessing.vertical_parametrisation.VerticalParametrisation attribute) run() (pyshqg.core.integration.RungeKuttaModelIntegrator method) RungeKuttaModelIntegrator (class in pyshqg.core.integration)

S

selective (pyshqg.core.dissipation.QGDissipation attribute) spec_qp (pyshqg.core.poisson.QGPoissonSolver attribute) spec_to_grid() (pyshqg.core.spectral_transformations.SpectralTransformations method) spec_to_grid_generic() (pyshqg.core.spectral_transformations.SpectralTransformations method) spec_to_grid_grad_phi() (pyshqg.core.spectral_transformations.SpectralTransformations method)

spec_to_grid_grad_theta() (pyshqg.core.spectral_transformations.SpectralTransformations method) spectral_transformations (pyshqg.core.model.QGModel attribute) (pyshqg.core.poisson.QGPoissonSolver attribute) SpectralTransformations (class in pyshqg.core.spectral_transformations) spectrum (pyshqg.core.dissipation.QGSelectiveDissipation attribute) steps (pyshqg.core.integration.RungeKuttaModelIntegrator attribute)

T

T (pyshqg.core.spectral_transformations.SpectralTransformations attribute) T_grid (pyshqg.core.spectral_transformations.SpectralTransformations attribute) TensorflowBackend (class in pyshqg.backend.tensorflow_backend) thermal (pyshqg.core.dissipation.QGDissipation attribute) thermal_coupling (pyshqg.core.dissipation.QGThermalDissipation attribute) time (pyshqg.core.model.QGModelTrajectory attribute)

to_numpy() (pyshqg.backend.abstract.Backend static method) (pyshqg.backend.numpy_backend.NumpyBackend static method) (pyshqg.backend.tensorflow_backend.TensorflowBackend static method) (pyshqg.core.model.QGModelTrajectory method) to_xarray() (pyshqg.core.model.QGModelTrajectory method) total_q_to_psi() (pyshqg.core.poisson.QGPoissonSolver method)

V

variables (pyshqg.core.model.QGModelTrajectory attribute)

VerticalParametrisation (class in pyshqg.preprocessing.vertical_parametrisation)

W

weights (pyshqg.core.integration.RungeKuttaModelIntegrator attribute)