check multiple ssw mirrors for remote psf files

HinodeXRT · Aug 31, 2023 · 099d704 · 099d704
1 parent e2a3e1f
commit 099d704
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/xrtpy/util/__init__.py b/xrtpy/util/__init__.py
@@ -5,3 +5,8 @@
     "epoch",
     "time",
 ]
+
+_SSW_MIRRORS = [
+    "https://sohoftp.nascom.nasa.gov/solarsoft/",
+    "https://hesperia.gsfc.nasa.gov/ssw/",
+]
diff --git a/xrtpy/util/xrt_deconvolve.py b/xrtpy/util/xrt_deconvolve.py
@@ -4,6 +4,8 @@
 __all__ = ["xrt_deconvolve"]
 
 import numpy as np
+from urllib.parse import urljoin
+
 
 from datetime import datetime
 from numpy.fft import fft2, fftshift, ifft2
@@ -12,19 +14,17 @@
 from sunpy.image.transform import affine_transform
 from sunpy.map import Map
 
+from xrtpy.util import _SSW_MIRRORS
+
 
 @manager.require(
     "PSF560.fits",
-    [
-        "https://sohoftp.nascom.nasa.gov/solarsoft/hinode/xrt/idl/util/XRT20170324_151721.0.PSF560.fits"
-    ],
+    [urljoin(mirror, "hinode/xrt/idl/util/XRT20170324_151721.0.PSF560.fits") for mirror in _SSW_MIRRORS],
     "0eaa5da6fb69661e7f46d1f0c463e4b3b1745426a399a4fbc53fc0c0ae87dd0d",
 )
 @manager.require(
     "PSF1000.fits",
-    [
-        "https://sohoftp.nascom.nasa.gov/solarsoft/hinode/xrt/idl/util/XRT20170324_161721.0.PSF1000.fits"
-    ],
+    [urljoin(mirror, "hinode/xrt/idl/util/XRT20170324_161721.0.PSF1000.fits") for mirror in _SSW_MIRRORS],
     "95590a7174692977a2f111b932811c9c7ae105a59b93bfe6c96fba862cefacf1",
 )
 def xrt_deconvolve(image_map, niter=5, verbose=False, psf1keV=False):