exovetter 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

exovetter/viz_transits.py

@@ -4,6 +4,7 @@ import matplotlib.pyplot as plt
 from astropy.convolution import convolve, Box1DKernel
 from exovetter import utils
 
+__all__ = ['plot_all_transits', 'plot_fold_transit']
 
 def plot_all_transits(time, flux, period, epoch, dur, depth, max_transits=20,
                       transit_only=False, plot=True, units="d"):
@@ -67,10 +68,10 @@ def plot_all_transits(time, flux, period, epoch, dur, depth, max_transits=20,
 
     #step_size = 6*np.std(flux[~intransit])
     step_size = depth * 1.25
-    if 3 * np.std(flux[~intransit]) > step_size:
-        step_size = 3 * np.std(flux[~intransit])
+    if 3 * np.std(flux[~intransit]) > step_size: 
+        step_size = 10 * np.std(flux[~intransit])
 
-    nsteps = len(np.unique(ntransit))  #np.ceil(np.max(ntransit))
+    nsteps = len(np.unique(ntransit))  # np.ceil(np.max(ntransit))
 
     if nsteps > max_transits:
         nsteps = max_transits
@@ -78,10 +79,10 @@ def plot_all_transits(time, flux, period, epoch, dur, depth, max_transits=20,
     if plot:
         plt.figure(figsize=(figwid, nsteps))
         transits  =  np.floor(np.unique(ntransit[intransit]))
-        print(transits[0:nsteps])
-        print(nsteps)
-        print(ntransit)
-        
+        # print(transits[0:nsteps])
+        # print(nsteps)
+        # print(ntransit)
+
         for i, nt in enumerate(transits[0:nsteps]): 
             
             ph = phases[ntransit == nt]
@@ -91,9 +92,14 @@ def plot_all_transits(time, flux, period, epoch, dur, depth, max_transits=20,
 
             plt.plot(ph, fl + step_size * i, '.--',
                      c=color, ms=5, lw=1)
+
             plt.annotate("Transit %i" % nt,
-                         (xmin, np.median(fl) + step_size * i),
-                         c=color)
+            (xmin, np.median(fl) + step_size * (i+0.15)), c=color)
+            # Added buffer between transit and annotation
+            
+            # plt.annotate("Transit %i" % nt,
+            #              (xmin, np.median(fl) + step_size * i),
+            #              c=color)
 
         plt.xlim(xmin, xmax)
         plt.axvspan(
@@ -102,6 +108,8 @@ def plot_all_transits(time, flux, period, epoch, dur, depth, max_transits=20,
             alpha=0.15)
         plt.xlabel("Phased Time [%s]" % units)
 
+    
+
     return n_has_data
 
 

{exovetter-0.0.2.dist-info → exovetter-0.0.4.dist-info}/METADATA

@@ -1,20 +1,21 @@
 Metadata-Version: 2.1
 Name: exovetter
-Version: 0.0.2
-Summary: Transiting Exoplanet Vetting Package for TESS and Kepler
+Version: 0.0.4
+Summary: Exoplanet vetting package
 Home-page: https://github.com/spacetelescope/exovetter
 Author: Susan Mullally et al.
 Author-email: smullally@stsci.edu
+License: BSD 3-Clause
 Requires-Python: >=3.7
 Description-Content-Type: text/x-rst
 License-File: LICENSE.rst
 Requires-Dist: numpy
-Requires-Dist: astropy (>=3)
+Requires-Dist: astropy >=3
 Requires-Dist: scipy
 Provides-Extra: all
 Requires-Dist: matplotlib ; extra == 'all'
 Requires-Dist: scikit-learn ; extra == 'all'
-Requires-Dist: lightkurve (>=2) ; extra == 'all'
+Requires-Dist: lightkurve >=2 ; extra == 'all'
 Requires-Dist: lpproj ; extra == 'all'
 Provides-Extra: docs
 Requires-Dist: sphinx-astropy ; extra == 'docs'

exovetter-0.0.4.dist-info/RECORD

@@ -0,0 +1,28 @@
+exovetter/__init__.py,sha256=79wPvKodLvzu1QiCX8tYpm7WcxST0mx-CWorfUz3D-c,278
+exovetter/const.py,sha256=aKBZmiZYlia2EuMDarzHpIRFYFKz-Vk81TJK2o2lUEE,996
+exovetter/leo.py,sha256=03kYnhwlbNU5pXmuIXwFRJ9CpSxrfQlWMOYtCJCY_mw,9345
+exovetter/lightkurve_utils.py,sha256=3cKkQpubFNT7nid2OKHrhezcJlZHYBRAUBZXPKKIL8g,1121
+exovetter/lpp.py,sha256=3U6PWGeY-TimeKmHhOHCJw1Pv7k1jU-NgTWesBKahaE,12978
+exovetter/model.py,sha256=ZRMoSLf_4V0Mgftpi_4A0iRgoEYHSdc96tWB8MjDH8k,2817
+exovetter/odd_even.py,sha256=g8TTd2JpwxLuTfYlonw2pbcVqnj2a7QvEAzL4qkxso4,6910
+exovetter/sweet.py,sha256=prbuteEaahFWhAoCtQCVP47SbA6h4pYJn1Wn9JkEalc,3066
+exovetter/tce.py,sha256=5JFKkbKoKCtdhm5EuW6FMAvp0LiaHLuqpFUbY0joyLI,7302
+exovetter/transit_coverage.py,sha256=Y4Muj7lKRsa0Sn6hD70O5On-2Aae1p5i-VTfJt5tsOw,2154
+exovetter/trapezoid_fit.py,sha256=Dok-H0zRsfigh4zfVO4R6Wi2TgK6hQXh4EIIYGouxiQ,31097
+exovetter/utils.py,sha256=EeP4_jRsPWGiFoOwmdGf7IgZhLz1Kve_nGqjlZqyRDc,17976
+exovetter/version.py,sha256=SLdro-VW9GFWjSUPvxHnmIQgPaSWyx6-CbUu_R_I9UA,337
+exovetter/vetters.py,sha256=70Douz_igYXslJ0HJcmaCIQ8ni8_JwIvH0N2QCNdRf4,30975
+exovetter/viz_transits.py,sha256=SGUP-LLKN8030xGGqqk47OgrO4Gwk_ladIUqH5wT8_8,5651
+exovetter/centroid/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+exovetter/centroid/centroid.py,sha256=LWqjn-Oerdw5j2wDCFYqB6omE4K9gckIwAA0-WZvgkY,8655
+exovetter/centroid/covar.py,sha256=suSgPjOVVEOclIi7rnhqc7wBbcSVsPOrdad91NXjX3Y,8973
+exovetter/centroid/disp.py,sha256=scVCRj16mNSqugHY28_uLt5bOkcdumIlJeFG3BXfldU,3674
+exovetter/centroid/fastpsffit.py,sha256=eRMxYUJ_4hsCxso4u1d1zQvFSJQlJ_gRzhNa2PXkKuE,4532
+exovetter/modshift/__init__.py,sha256=j5665q0RHVzbzdPBXv_jUozGB4sex8ANXtzB41EYzRQ,68
+exovetter/modshift/modshift.py,sha256=cAYT0zcWsJYcGRDyfVPUjHsk7l8zQdecP0LY0hTdjOw,14531
+exovetter/modshift/plotmodshift.py,sha256=U0XKxVDKlNDXknj1g1to5dkIr0bQ_FqGp__BHhIQ2FQ,2400
+exovetter-0.0.4.dist-info/LICENSE.rst,sha256=uhiFz7eEG8WszLtRtYIT89FNMI3ijy1aACnvl5BOz2Y,1492
+exovetter-0.0.4.dist-info/METADATA,sha256=kXfBbYmve3Tn_N_wZObJzLw6wmXi2ktrSS3gjrFBgUw,3706
+exovetter-0.0.4.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+exovetter-0.0.4.dist-info/top_level.txt,sha256=wh0_U5IPEspONX8zhGRaFUqfuovCGkgYjAmWuQe0w6Q,10
+exovetter-0.0.4.dist-info/RECORD,,

{exovetter-0.0.2.dist-info → exovetter-0.0.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.1)
+Generator: bdist_wheel (0.42.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

exovetter/__init__.py.mine

@@ -1,11 +0,0 @@
-# Licensed under a 3-clause BSD style license - see LICENSE.rst
-
-try:
-    from .version import version as __version__
-except ImportError:
-    __version__ = ''
-
-from . import lpp  # noqa
-from . import tce  # noqa
-from . import trapezoid_fit  # noqa
-#from . import vetters  # noqa

exovetter/centroid/untitled0.py

@@ -1,24 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-"""
-Created on Fri Jul 23 21:09:54 2021
-
-@author: smullally
-"""
-
-import numpy
-import matplotlib
-import exovetter as exo
-from exovetter import const
-from exovetter import utils
-import exovetter.vetters as vet
-import lightkurve as lk
-
-candidate = "TOI 565.01"
-tce = utils.get_mast_tce(candidate)
-lc = lk.search_lightcurve(candidate, exptime=120)[0].download()
-lc.plot()
-tpf = lk.search_targetpixelfile(candidate, exptime=120)[0].download()
-
-cent = vet.Centroid()
-cent.run(tce[0],tpf, plot=True)

exovetter/junk

@@ -1,89 +0,0 @@
-#Code to plot and evaluate individual transits.
-import numpy as np
-import matplotlib.pyplot as plt
-from astropy.convolution import convolve, Box1DKernel
-from exovetter import utils
-
-
-def all_transits(time, flux, period, epoch, dur, max_transits = 10):
-    """
-
-    Parameters
-    ----------
-    time : numpy array
-        times of measurements
-    flux : numpy array
-        brightness changes
-    period : float
-        period in same units as time
-    epoch : float
-        epoch of transits in same units and offset as time.
-    dur : float
-        duration of the transit in same units as the time.
-    max_transits : integer, optional
-        maximum number of transits to plot. The default is 10.
-
-    Returns
-    -------
-    
-
-    """
-
-    intransit = utils.mark_transit_cadences(time, period, epoch, dur,
-                          num_durations=1, flags=None)
-    
-
-def plot_fold_transit(time,flux,period,epoch,depth,dur,smooth = None):
-    """
-    Bins set to None will show not show the binned points. Otherwise
-    the binning is chosen
-
-    Parameters
-    ----------
-    time : TYPE
-        DESCRIPTION.
-    flux : TYPE
-        DESCRIPTION.
-    period : TYPE
-        DESCRIPTION.
-    epoch : TYPE
-        DESCRIPTION.
-    depth : TYPE
-        DESCRIPTION.
-    duration : TYPE
-        DESCRIPTION.
-    smooth : integer, optional
-        DESCRIPTION. The default is None.
-
-    Returns
-    -------
-    None.
-
-    """    
-    
-    
-    phases = utils.compute_phases(time, period, epoch, offset=0.25)
-    
-    
-    intransit = utils.mark_transit_cadences(time, period, epoch, dur,
-                          num_durations=1, flags=None)
-    
-    
-    if smooth is not None:
-        N = int(np.floor(len(phases[intransit])/smooth))
-        sort_index = np.argsort(phases)
-        smoothed_signal = convolve(flux[sort_index], Box1DKernel(N))
-
-    plt.figure(figsize=(10,6))
-    
-    plt.plot(phases*period, flux, 'k.', ms = 3, label="Folded")
-    
-    if smooth is not None:
-        sort_phases = phases[sort_index]
-        plt.plot(sort_phases[N:-N]*period, smoothed_signal[N:-N], 'r--',
-                 lw=1.5, label="Box1DSmooth")
-    
-    #med_flux = np.median(flux)
-    #plt.ylim(med_flux-1.7*depth, med_flux+1.2*depth)
-    plt.legend(loc="upper right")
-    plt.xlabel('Phased Times')

exovetter-0.0.2.dist-info/RECORD

@@ -1,29 +0,0 @@
-exovetter/__init__.py,sha256=79wPvKodLvzu1QiCX8tYpm7WcxST0mx-CWorfUz3D-c,278
-exovetter/__init__.py.mine,sha256=qkfZ03KtSVp6v4eJKNyhLIb7Ob8IBjLH9DusHpwg2qM,279
-exovetter/const.py,sha256=aKBZmiZYlia2EuMDarzHpIRFYFKz-Vk81TJK2o2lUEE,996
-exovetter/junk,sha256=zHyh5mWkuj14Gwz9fHlgLq6CS-KNavx0zTvJNdYPsik,2294
-exovetter/lightkurve_utils.py,sha256=2F8vWjpabIzI0-Vs7xq5c6T3JKmqe92OBOWDzb5DZdQ,1107
-exovetter/lpp.py,sha256=qZ-TeHjShNCeyX99JgPNuoOQQ4P2DPVqoRYeTJIW2Lk,12951
-exovetter/model.py,sha256=ZRMoSLf_4V0Mgftpi_4A0iRgoEYHSdc96tWB8MjDH8k,2817
-exovetter/odd_even.py,sha256=g8TTd2JpwxLuTfYlonw2pbcVqnj2a7QvEAzL4qkxso4,6910
-exovetter/sweet.py,sha256=9nuagC4sMG7-vHR__iYCzyc9gmdGsIAvgGTnixYWtDk,2883
-exovetter/tce.py,sha256=m1yfVNjjsPX36REXCWB_TPhklQ6xyEQ831O4g2rRFe4,7195
-exovetter/transit_coverage.py,sha256=Y4Muj7lKRsa0Sn6hD70O5On-2Aae1p5i-VTfJt5tsOw,2154
-exovetter/trapezoid_fit.py,sha256=eMw0Z7DvJoyCcTXEtVTLh356DI1gD8DqOrpsulOoC8k,30977
-exovetter/utils.py,sha256=kuxUMbvp69CnyVWqT30GItJk1U7zclrHEqTzyxc8sh4,16555
-exovetter/version.py,sha256=X3C4BtKCe_sMyuqyUb8QmenyhJrDeGFQ_wtLFIJbPsQ,337
-exovetter/vetters.py,sha256=lPjU2DXR0pK2-hIpgxVD_OHFSlty3wF-p-vschzkHwQ,16667
-exovetter/viz_transits.py,sha256=yUox9_2prKxPAv2OopkY36yqfHmI5Z_uHZlZN149t-w,5402
-exovetter/centroid/centroid.py,sha256=LxGRGqFHBFQSO7hNI4bB31dIyJxWyegx8NNcHSubeVA,8117
-exovetter/centroid/covar.py,sha256=2OwG4lBw9puV9hMNm6iGGukHW1Q1_P4LLk6WDhtUpIo,8923
-exovetter/centroid/disp.py,sha256=scVCRj16mNSqugHY28_uLt5bOkcdumIlJeFG3BXfldU,3674
-exovetter/centroid/fastpsffit.py,sha256=eRMxYUJ_4hsCxso4u1d1zQvFSJQlJ_gRzhNa2PXkKuE,4532
-exovetter/centroid/untitled0.py,sha256=a6t54TK1ADPfPIwXk6uoqBLZnvHiaKgSDJIF-GeCbww,539
-exovetter/modshift/__init__.py,sha256=j5665q0RHVzbzdPBXv_jUozGB4sex8ANXtzB41EYzRQ,68
-exovetter/modshift/modshift.py,sha256=cAYT0zcWsJYcGRDyfVPUjHsk7l8zQdecP0LY0hTdjOw,14531
-exovetter/modshift/plotmodshift.py,sha256=U0XKxVDKlNDXknj1g1to5dkIr0bQ_FqGp__BHhIQ2FQ,2400
-exovetter-0.0.2.dist-info/LICENSE.rst,sha256=uhiFz7eEG8WszLtRtYIT89FNMI3ijy1aACnvl5BOz2Y,1492
-exovetter-0.0.2.dist-info/METADATA,sha256=n0n-_dlrNy8H1zDZ2yhnxYzdXV7iEMT5RnM6BWQtMy0,3719
-exovetter-0.0.2.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-exovetter-0.0.2.dist-info/top_level.txt,sha256=wh0_U5IPEspONX8zhGRaFUqfuovCGkgYjAmWuQe0w6Q,10
-exovetter-0.0.2.dist-info/RECORD,,