timewise 0.5.4__py3-none-any.whl → 1.0.0a2__py3-none-any.whl

timewise/parent_sample_base.py

@@ -1,89 +0,0 @@
-import abc, os
-import pandas as pd
-import numpy as np
-import logging
-
-from timewise.general import get_directories
-from timewise.utils import plot_sdss_cutout, plot_panstarrs_cutout
-
-
-logger = logging.getLogger(__name__)
-
-
-class ParentSampleBase(abc.ABC):
-    """
-    Base class for parent sample.
-    Any subclass must implement
-
-    - `ParentSample.df`: A `pandas.DataFrame` consisting of minimum three columns: two columns holding the sky positions of each object in the form of right ascension and declination and one row with a unique identifier.
-    - `ParentSample.default_keymap`: a dictionary, mapping the column in `ParentSample.df` to 'ra', 'dec' and 'id'
-
-    :param base_name: determining the location of any data in the `timewise` data directory.
-    """
-
-    df = pd.DataFrame()
-    default_keymap = dict()
-
-    def __init__(self, base_name):
-        # set up directories
-        d = get_directories()
-        self.cache_dir = d["cache_dir"] / base_name
-        self.plots_dir = d["plots_dir"] / base_name
-
-        for d in [self.cache_dir, self.plots_dir]:
-            d.parent.mkdir(parents=True, exist_ok=True)
-
-        self.local_sample_copy = self.cache_dir / 'sample.csv'
-
-    def plot_cutout(self, ind, arcsec=20, interactive=False, **kwargs):
-        """
-        Plot the coutout images in all filters around the position of object with index i
-
-        :param ind: the index in the sample
-        :type ind: int or list-like
-        :param arcsec: the radius of the cutout
-        :type arcsec: float
-        :param interactive: interactive mode
-        :type interactive: bool
-        :param kwargs: any additional kwargs will be passed to `matplotlib.pyplot.subplots()`
-        :return: figure and axes if `interactive=True`
-        """
-        sel = self.df.loc[np.atleast_1d(ind)]
-        ra, dec = sel[self.default_keymap["ra"]], sel[self.default_keymap["dec"]]
-        title = [r[self.default_keymap["id"]] for i, r in sel.iterrows()]
-
-        fn = kwargs.pop(
-            "fn",
-            [self.plots_dir / f"{i}_{r[self.default_keymap['id']]}.pdf"
-             for i, r in sel.iterrows()]
-        )
-        self.plots_dir.mkdir(parents=True, exist_ok=True)
-
-        logger.debug(f"\nRA: {ra}\nDEC: {dec}\nTITLE: {title}\nFN: {fn}")
-        ou = list()
-
-        ras = np.atleast_1d(ra)
-        decs = np.atleast_1d(dec)
-        title = np.atleast_1d(title) if title else [None] * len(ras)
-        fn = np.atleast_1d(fn) if fn else [None] * len(ras)
-
-        for _ra, _dec, _title, _fn in zip(ras, decs, title, fn):
-            ou.append(self._plot_cutout(_ra, _dec, arcsec, interactive, title=_title, fn=_fn, **kwargs))
-
-        if len(ou) == 1:
-            ou = ou[0]
-        return ou
-
-    @staticmethod
-    def _plot_cutout(ra, dec, arcsec, interactive, which="sdss", **kwargs):
-        if which == "sdss":
-            return plot_sdss_cutout(ra, dec, arcsec=arcsec, interactive=interactive, **kwargs)
-        elif which == "panstarrs":
-            return plot_panstarrs_cutout(ra, dec, arcsec=arcsec, interactive=interactive, **kwargs)
-        else:
-            raise ValueError(f"{which} not an implemented survey! Choose one of 'sdss' or 'panstarrs'.")
-
-    def save_local(self):
-        logger.debug(f"saving under {self.local_sample_copy}")
-        self.local_sample_copy.parent.mkdir(parents=True, exist_ok=True)
-        self.df.to_csv(self.local_sample_copy)

timewise/point_source_utils.py

@@ -1,68 +0,0 @@
-import logging
-import pandas as pd
-
-from timewise.parent_sample_base import ParentSampleBase
-from timewise.wise_data_by_visit import WiseDataByVisit
-
-
-logger = logging.getLogger(__name__)
-
-
-###########################################################################################################
-#            START POINT SOURCE  UTILS              #
-#####################################################
-
-
-def get_point_source_parent_sample(base_name, ra, dec):
-
-    class PointSourceParentSample(ParentSampleBase):
-        default_keymap = {
-            'ra': 'ra',
-            'dec': 'dec',
-            'id': 'id'
-        }
-
-        def __init__(self):
-
-            super().__init__(base_name=base_name)
-
-            self.base_name = base_name
-            self.df = pd.DataFrame({'ra': [ra], 'dec': [dec], 'id': [base_name]})
-
-        def save_local(self):
-            logger.debug(f"not saving")
-
-    return PointSourceParentSample
-
-
-def get_point_source_wise_data(base_name, ra, dec, min_sep_arcsec=10, match=False, **kwargs):
-    """
-    Get a WISEData instance for a point source
-
-    :param base_name: base name for storage in the data directory
-    :type base_name: str
-    :param ra: right ascencion
-    :type ra: float
-    :param dec: declination
-    :type dec: float
-    :param min_sep_arcsec: search radius in arcsec
-    :type min_sep_arcsec: float
-    :param match: match to AllWISE Source Catalogue
-    :type match: bool
-    :param kwargs: keyword arguments passed to WISEData.get_photometric_data()
-    :type kwargs: dict
-    :return: WISEData
-    """
-    ps = get_point_source_parent_sample(base_name, ra, dec)
-    wd = WiseDataByVisit(n_chunks=1, base_name=base_name, parent_sample_class=ps, min_sep_arcsec=min_sep_arcsec)
-    if match:
-        wd.match_all_chunks()
-    service = kwargs.pop('service', 'gator')
-    wd.get_photometric_data(service=service, **kwargs)
-    wd.plot_lc(parent_sample_idx=0, service=service)
-    return wd
-
-
-#####################################################
-#            END POINT SOURCE  UTILS                #
-###########################################################################################################