timewise 0.4.9__py3-none-any.whl → 0.4.12__py3-none-any.whl

timewise/__init__.py

@@ -2,4 +2,4 @@ from timewise.wise_data_by_visit import WiseDataByVisit
 from timewise.wise_bigdata_desy_cluster import WISEDataDESYCluster
 from timewise.parent_sample_base import ParentSampleBase
 
-__version__ = "0.4.9"
+__version__ = "0.4.12"

timewise/utils.py

@@ -12,6 +12,7 @@ from functools import cache
 from astropy.table import Table
 from PIL import Image
 from io import BytesIO
+import hashlib
 
 
 from timewise.general import cache_dir, backoff_hndlr
@@ -146,6 +147,36 @@ def plot_sdss_cutout(ra, dec, arcsec=20, arcsec_per_px=0.1, interactive=False, f
 #####################################################
 
 
+class PanSTARRSQueryError(Exception):
+    pass
+
+
+def load_cache_or_download(url):
+    logger.debug(f"loading or downloading {url}")
+    h = hashlib.md5(url.encode()).hexdigest()
+    cache_file = os.path.join(cache_dir, h + ".cache")
+    logger.debug(f"cache file is {cache_file}")
+    if not os.path.isfile(cache_file):
+        logger.debug(f"downloading {url}")
+        r = requests.get(url)
+        with open(cache_file, 'wb') as f:
+            f.write(r.content)
+        return r.content
+    else:
+        logger.debug(f"loading {cache_file}")
+        with open(cache_file, 'rb') as f:
+            return f.read()
+
+
+def annotate_not_available(ax):
+    xlim = ax.get_xlim()
+    ylim = ax.get_ylim()
+    x = sum(xlim) / 2
+    y = sum(ylim) / 2
+    logger.debug(f"annotate_not_available at {x}, {y}")
+    ax.annotate("Outside\nPanSTARRS\nFootprint", (x, y), color='red', ha='center', va='center', fontsize=10)
+
+
 def getimages(ra, dec, filters="grizy"):
     """Query ps1filenames.py service to get a list of images
 
@@ -157,7 +188,8 @@ def getimages(ra, dec, filters="grizy"):
 
     service = "https://ps1images.stsci.edu/cgi-bin/ps1filenames.py"
     url = f"{service}?ra={ra}&dec={dec}&filters={filters}"
-    table = Table.read(url, format='ascii')
+    content = load_cache_or_download(url)
+    table = Table.read(content.decode(), format='ascii')
     return table
 
 
@@ -180,6 +212,8 @@ def geturl(ra, dec, size=240, output_size=None, filters="grizy", format="jpg", c
     if format not in ("jpg", "png", "fits"):
         raise ValueError("format must be one of jpg, png, fits")
     table = getimages(ra, dec, filters=filters)
+    if len(table) == 0:
+        raise PanSTARRSQueryError("No images available")
     url = (f"https://ps1images.stsci.edu/cgi-bin/fitscut.cgi?"
            f"ra={ra}&dec={dec}&size={size}&format={format}")
     if output_size:
@@ -216,8 +250,8 @@ def getcolorim(ra, dec, size=240, output_size=None, filters="grizy", format="jpg
     if format not in ("jpg", "png"):
         raise ValueError("format must be jpg or png")
     url = geturl(ra, dec, size=size, filters=filters, output_size=output_size, format=format, color=True)
-    r = requests.get(url)
-    im = Image.open(BytesIO(r.content))
+    content = load_cache_or_download(url)
+    im = Image.open(BytesIO(content))
     return im
 
 
@@ -238,8 +272,8 @@ def getgrayim(ra, dec, size=240, output_size=None, filter="g", format="jpg"):
     if filter not in list("grizy"):
         raise ValueError("filter must be one of grizy")
     url = geturl(ra, dec, size=size, filters=filter, output_size=output_size, format=format)
-    r = requests.get(url[0])
-    im = Image.open(BytesIO(r.content))
+    content = load_cache_or_download(url[0])
+    im = Image.open(BytesIO(content))
     return im
 
 
@@ -276,9 +310,14 @@ def plot_panstarrs_cutout(
             axss = ax
 
         for j, fil in enumerate(list(filters)):
-            im = getgrayim(ra, dec, size=ang_px, filter=fil)
             axs = axss[1]
-            axs[j].imshow(im, cmap='gray', **imshow_kwargs)
+            try:
+                im = getgrayim(ra, dec, size=ang_px, filter=fil)
+                axs[j].imshow(im, cmap='gray', **imshow_kwargs)
+            except PanSTARRSQueryError:
+                axs[j].set_xlim(-arcsec / 2, arcsec / 2)
+                axs[j].set_ylim(-arcsec / 2, arcsec / 2)
+                annotate_not_available(axs[j])
 
             axs[j].scatter(*scatter_args, **scatter_kwargs)
             axs[j].set_title(fil)
@@ -292,8 +331,13 @@ def plot_panstarrs_cutout(
             fig = plt.gcf()
             axss = ax
 
-        im = getcolorim(ra, dec, size=ang_px)
-        axss.imshow(im, **imshow_kwargs)
+        try:
+            im = getcolorim(ra, dec, size=ang_px)
+            axss.imshow(im, **imshow_kwargs)
+        except PanSTARRSQueryError:
+            axss.set_xlim(-arcsec / 2, arcsec / 2)
+            axss.set_ylim(-arcsec / 2, arcsec / 2)
+            annotate_not_available(axss)
         axss.scatter(*scatter_args, **scatter_kwargs)
 
     _this_title = title if title else f"{ra}_{dec}"

timewise/wise_data_by_visit.py

@@ -5,6 +5,8 @@ import logging
 from scipy import stats
 import matplotlib.pyplot as plt
 from matplotlib.lines import Line2D
+from matplotlib.markers import MarkerStyle
+from matplotlib.transforms import Affine2D
 
 from timewise.wise_data_base import WISEDataBase
 from timewise.utils import get_excess_variance
@@ -538,7 +540,11 @@ class WiseDataByVisit(WISEDataBase):
         )
 
         # set markers for visits
-        markers = list(Line2D.filled_markers) + ["$1$", "$2$", "$3$", "$4$", "$5$", "$6$", "$7$", "$8$", "$9$"]
+        markers_strings = list(Line2D.filled_markers) + ["$1$", "$2$", "$3$", "$4$", "$5$", "$6$", "$7$", "$8$", "$9$"]
+        markers_straight = [MarkerStyle(im) for im in markers_strings]
+        rot = Affine2D().rotate_deg(180)
+        markers_rotated = [MarkerStyle(im, transform=rot) for im in markers_strings]
+        markers = markers_straight + markers_rotated
 
         # calculate ra and dec relative to center of cutout
         ra = (lightcurve.ra - pos[self.parent_sample.default_keymap["ra"]]) * 3600

{timewise-0.4.9.dist-info → timewise-0.4.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: timewise
-Version: 0.4.9
+Version: 0.4.12
 Summary: A small package to download infrared data from the WISE satellite
 Home-page: https://github.com/JannisNe/timewise
 License: MIT

{timewise-0.4.9.dist-info → timewise-0.4.12.dist-info}/RECORD

@@ -1,17 +1,17 @@
-timewise/__init__.py,sha256=DAKnirwteKpFkXhdHNc5M8MvdKNKGT8dtWvQIZXY_6o,203
+timewise/__init__.py,sha256=gnpyn_jjCr_5brbB3Xi6G1eDotPNrlP6PINA-UAHGnM,204
 timewise/big_parent_sample.py,sha256=B7w7TMJx2mohsVfY0RiBHQU9N6mn_MDqrB7Ica0aafU,3480
 timewise/cli.py,sha256=LRBR3cOd_qBEpGkyP7tbJBCKvO0XxRQA8BgoMZfw08k,533
 timewise/config_loader.py,sha256=gqu881fSoBIzK_-KSVSCDV1pLPuD_g0AHbWovKFtWEA,5675
 timewise/general.py,sha256=cadgPIACEWpFsa4yncFb-nboGgXmJ1TicjPQalB_SAU,1472
 timewise/parent_sample_base.py,sha256=fHRBS791w51JCEwkfrCMyzXSyyc94NNSu-J9AfRFf2E,3298
 timewise/point_source_utils.py,sha256=4dmxfujrrNxDLkh2rVziSR-NNaHzrKFa8xgx_Lj-ZNE,2171
-timewise/utils.py,sha256=OfIEkPmcKPheT3rQ6oSjRzYGosJWjh9k0FsdIsBpXf8,13627
+timewise/utils.py,sha256=-tS5jrRnMYa7tosWvs_Jitjv8qPfi9VZE-o0-DMHKEQ,15144
 timewise/wise_bigdata_desy_cluster.py,sha256=8Zd1jCgVpR8nefGrmHxFtyBasP2oaOZ04NXUuEikyhI,56173
 timewise/wise_data_base.py,sha256=PrOhI7RtgVgdAaZKzGcmiFAIPhwLjg2aT_y9opG8DhQ,82744
-timewise/wise_data_by_visit.py,sha256=e-tb_Zk_iEz0Jglzo-oKhQn244I6wKRramdtI-atGgY,28785
+timewise/wise_data_by_visit.py,sha256=5fR5qaDz_liWJaBwnDhsEx--yoyh3oxPKKpXGpEsXmk,29129
 timewise/wise_flux_conversion_correction.dat,sha256=XLnYqk0g1NVthVSNGsKlqinzKI7QUwJidO-qg4tHXKU,1095
-timewise-0.4.9.dist-info/LICENSE,sha256=sVoNJWiTlH-NarJx0wdsob468Pg3JE6vIIgll4lCa3E,1070
-timewise-0.4.9.dist-info/METADATA,sha256=WUvsLYuZBT_gRSehtyCTdILmj3M41LowpXAFHLsRonk,2495
-timewise-0.4.9.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-timewise-0.4.9.dist-info/entry_points.txt,sha256=yIWgzM0CZCJBrSR9-zbQW9d8MpFae0KlRVnZXF8rZow,54
-timewise-0.4.9.dist-info/RECORD,,
+timewise-0.4.12.dist-info/LICENSE,sha256=sVoNJWiTlH-NarJx0wdsob468Pg3JE6vIIgll4lCa3E,1070
+timewise-0.4.12.dist-info/METADATA,sha256=s4qqipfvBVCdeFyTy5ko5d7dsjXlXYPFWOcpHFEdFoc,2496
+timewise-0.4.12.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+timewise-0.4.12.dist-info/entry_points.txt,sha256=yIWgzM0CZCJBrSR9-zbQW9d8MpFae0KlRVnZXF8rZow,54
+timewise-0.4.12.dist-info/RECORD,,

{timewise-0.4.9.dist-info → timewise-0.4.12.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.8.1
+Generator: poetry-core 2.0.0
 Root-Is-Purelib: true
 Tag: py3-none-any