From 2a0fe0efc2b59c0605bd6704e81df772cdd24cc6 Mon Sep 17 00:00:00 2001
From: "Noah D. Brenowitz" <nbrenowitz@nvidia.com>
Date: Thu, 19 Sep 2024 10:46:58 -0700
Subject: [PATCH 1/2] Add example for regridding to arbitrary lat lon

---
 examples/regrid_to_given_lat_lon.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 examples/regrid_to_given_lat_lon.py

diff --git a/examples/regrid_to_given_lat_lon.py b/examples/regrid_to_given_lat_lon.py
new file mode 100644
index 0000000..843af78
--- /dev/null
+++ b/examples/regrid_to_given_lat_lon.py
@@ -0,0 +1,29 @@
+import numpy as np
+import torch
+
+import earth2grid
+
+device = "cuda"
+
+
+# the source grid (from 90N to 90S and 0E to 360E)
+ll = earth2grid.latlon.equiangular_lat_lon_grid(721, 1440)
+
+# a 2d grid of target lat lons
+target_lat = np.linspace(30, 50, 32)
+target_lon = np.linspace(100, 120, 64)
+target_lat, target_lon = np.meshgrid(target_lat, target_lon)
+
+# Some source data on the original grid
+data = torch.ones([721, 1440]).to(device)
+
+# Create a bilinear regridding object earth2grid
+regrid = ll.get_bilinear_regridder_to(target_lat, target_lon)
+
+# need to move the weights to same device and dtype as data
+regrid.to(data)
+
+# perform the regridding
+out = regrid(data)
+assert out.shape == target_lat.shape  # noqa
+print("data shape", out.shape)

From c45702af8bd80c70e5e5e7bff2497dabb40bc7d8 Mon Sep 17 00:00:00 2001
From: "Noah D. Brenowitz" <nbrenowitz@nvidia.com>
Date: Thu, 19 Sep 2024 11:20:06 -0700
Subject: [PATCH 2/2] add license header

---
 examples/regrid_to_given_lat_lon.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/examples/regrid_to_given_lat_lon.py b/examples/regrid_to_given_lat_lon.py
index 843af78..5a41148 100644
--- a/examples/regrid_to_given_lat_lon.py
+++ b/examples/regrid_to_given_lat_lon.py
@@ -1,3 +1,17 @@
+# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 import numpy as np
 import torch