tglc 0.6.4__tar.gz → 0.6.6__tar.gz

{tglc-0.6.4/tglc.egg-info → tglc-0.6.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tglc
-Version: 0.6.4
+Version: 0.6.6
 Summary: TESS-Gaia Light Curve
 Home-page: https://github.com/TeHanHunter/TESS_Gaia_Light_Curve
 Author: Te Han
@@ -8,9 +8,21 @@ Author-email: tehanhunter@gmail.com
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8
+Requires-Python: >=3.8, <3.13
 Description-Content-Type: text/x-rst
 License-File: LICENSE
+Requires-Dist: astropy>=5.1
+Requires-Dist: astroquery==0.4.7
+Requires-Dist: matplotlib
+Requires-Dist: numpy
+Requires-Dist: oauthlib
+Requires-Dist: requests
+Requires-Dist: scipy
+Requires-Dist: threadpoolctl
+Requires-Dist: tqdm
+Requires-Dist: wheel
+Requires-Dist: wotan
+Requires-Dist: seaborn
 
 ==================================
 Introduction
@@ -18,6 +30,7 @@ Introduction
 
 TESS-Gaia Light Curve (`TGLC <https://archive.stsci.edu/hlsp/tglc>`_) is a dataset of TESS full-frame image light curves publicly available via the MAST portal. It is fitted with effective PSF and decontaminated with Gaia DR3 and achieved percent-level photometric precision down to 16th TESS magnitude! It unlocks astrophysics to a vast number of dim stars below 12th TESS magnitude. A package called tglc is pip-installable for customized light curve fits.
 
+
 ==================================
 Usage
 ==================================

{tglc-0.6.4 → tglc-0.6.6}/README.rst

@@ -4,6 +4,7 @@ Introduction
 
 TESS-Gaia Light Curve (`TGLC <https://archive.stsci.edu/hlsp/tglc>`_) is a dataset of TESS full-frame image light curves publicly available via the MAST portal. It is fitted with effective PSF and decontaminated with Gaia DR3 and achieved percent-level photometric precision down to 16th TESS magnitude! It unlocks astrophysics to a vast number of dim stars below 12th TESS magnitude. A package called tglc is pip-installable for customized light curve fits.
 
+
 ==================================
 Usage
 ==================================

{tglc-0.6.4 → tglc-0.6.6}/setup.py

@@ -6,7 +6,7 @@ with open("README.rst", "r", encoding="utf-8") as fh:
     long_description = fh.read()
 setuptools.setup(
     name="tglc",
-    version='0.6.4',
+    version='0.6.6',
     author="Te Han",
     author_email="tehanhunter@gmail.com",
     description="TESS-Gaia Light Curve",
@@ -18,9 +18,9 @@ setuptools.setup(
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
-    install_requires=['astropy>=5.1', 'astroquery', 'matplotlib', 'numpy', 'oauthlib', 'requests', 'scipy',
-                      'threadpoolctl', 'tqdm', 'wheel', 'wotan'],
+    install_requires=['astropy>=5.1', 'astroquery==0.4.7', 'matplotlib', 'numpy', 'oauthlib', 'requests', 'scipy',
+                      'threadpoolctl', 'tqdm', 'wheel', 'wotan', 'seaborn'],
     packages=setuptools.find_packages(include=['tglc', 'tglc.*']),
-    python_requires=">=3.8",
+    python_requires=">=3.8, <3.13",
     include_package_data=True
 )

tglc-0.6.6/tglc/__init__.py

@@ -0,0 +1,3 @@
+__version__ = "0.6.6"
+__author__ = 'Te Han, Timothy Brandt'
+__credits__ = 'University of California, Santa Barbara'

{tglc-0.6.4 → tglc-0.6.6}/tglc/effective_psf.py

@@ -191,16 +191,19 @@ def fit_lc(A, source, star_info=None, x=0., y=0., star_num=0, factor=2, psf_size
     coord = np.arange(size ** 2).reshape(size, size)
     index = np.array(coord[down:up, left:right]).flatten()
     A_cut = np.zeros((len(index), np.shape(A)[1]))
+    A_target = np.zeros((len(index), np.shape(A)[1]))
     for i in range(len(index)):
         A_ = np.zeros(np.shape(A)[-1])
         star_pos = np.where(star_info_num[0] == index[i])[0]
         A_[star_info_num[1][star_pos]] = star_info_num[2][star_pos]
+        A_target[i] = A_
         A_cut[i] = A[index[i], :] - A_
     aperture = np.zeros((len(source.time), len(index)))
     for j in range(len(source.time)):
         aperture[j] = np.array(source.flux[j][down:up, left:right]).flatten() - np.dot(A_cut, e_psf[j])
     aperture = aperture.reshape((len(source.time), up - down, right - left))
-    # np.save(f'_residual_{source.sector}.npy', aperture)
+    target_5x5 = (np.dot(A_target, np.nanmedian(e_psf, axis=0)).reshape(cut_size, cut_size))
+    field_stars_5x5 = (np.dot(A_cut, np.nanmedian(e_psf, axis=0)).reshape(cut_size, cut_size))
 
     # psf_lc
     over_size = psf_size * factor + 1
@@ -262,7 +265,7 @@ def fit_lc(A, source, star_info=None, x=0., y=0., star_num=0, factor=2, psf_size
     portion = np.nansum(psf_shape[:, 4:7, 4:7]) / np.nansum(psf_shape)
     # print(np.nansum(psf_shape[:, 5, 5]) / np.nansum(psf_shape))
     # np.save(f'toi-5344_psf_{source.sector}.npy', psf_shape)
-    return aperture, psf_lc, y - down, x - left, portion
+    return aperture, psf_lc, y - down, x - left, portion, target_5x5, field_stars_5x5
 
 
 def fit_lc_float_field(A, source, star_info=None, x=np.array([]), y=np.array([]), star_num=0, factor=2, psf_size=11,

{tglc-0.6.4 → tglc-0.6.6}/tglc/ffi.py

@@ -86,7 +86,7 @@ def convert_gaia_id(catalogdata_tic):
             FROM gaiadr3.dr2_neighbourhood
             WHERE dr2_source_id IN {gaia_ids}
             """
-    gaia_array = np.array(catalogdata_tic['GAIA'])
+    gaia_array = np.array([str(item) for item in catalogdata_tic['GAIA']], dtype=object)
     gaia_array = gaia_array[gaia_array != 'None']
     # np.save('gaia_array.npy', gaia_array)
     segment = (len(gaia_array) - 1) // 10000

{tglc-0.6.4 → tglc-0.6.6}/tglc/quick_lc.py

@@ -14,12 +14,14 @@ from astroquery.mast import Catalogs
 import astropy.units as u
 from astropy.coordinates import SkyCoord
 from astroquery.mast import Tesscut
+import sys
+import warnings
 # Tesscut._service_api_connection.TIMEOUT = 6000
-
 # warnings.simplefilter('ignore', UserWarning)
 from threadpoolctl import ThreadpoolController, threadpool_limits
 import numpy as np
-
+import seaborn as sns
+import itertools
 controller = ThreadpoolController()
 
 
@@ -40,10 +42,6 @@ def tglc_lc(target='TIC 264468702', local_directory='', size=90, save_aper=True,
     os.makedirs(local_directory + f'lc/', exist_ok=True)
     os.makedirs(local_directory + f'epsf/', exist_ok=True)
     os.makedirs(local_directory + f'source/', exist_ok=True)
-    if first_sector_only:
-        sector = 'first'
-    elif last_sector_only:
-        sector = 'last'
     print(f'Target: {target}')
     target_ = Catalogs.query_object(target, radius=42 * 0.707 / 3600, catalog="Gaia", version=2)
     if len(target_) == 0:
@@ -77,7 +75,7 @@ def tglc_lc(target='TIC 264468702', local_directory='', size=90, save_aper=True,
     elif first_sector_only:
         print(f'Only processing the first sector the target is observed in: Sector {sector_table["sector"][0]}.')
         print('Downloading Data from MAST and Gaia ...')
-
+        sector = sector_table["sector"][0]
         source = ffi_cut(target=target, size=size, local_directory=local_directory, sector=sector,
                          limit_mag=limit_mag, transient=transient)  # sector
         source.select_sector(sector=source.sector_table['sector'][0])
@@ -86,6 +84,7 @@ def tglc_lc(target='TIC 264468702', local_directory='', size=90, save_aper=True,
     elif last_sector_only:
         print(f'Only processing the last sector the target is observed in: Sector {sector_table["sector"][-1]}.')
         print('Downloading Data from MAST and Gaia ...')
+        sector = sector_table["sector"][-1]
         source = ffi_cut(target=target, size=size, local_directory=local_directory, sector=sector,
                          limit_mag=limit_mag, transient=transient)  # sector
         source.select_sector(sector=source.sector_table['sector'][-1])
@@ -296,47 +295,58 @@ def plot_pf_lc(local_directory=None, period=None, mid_transit_tbjd=None, kind='c
     plt.close(fig)
 
 
-def plot_contamination(local_directory=None, gaia_dr3=None):
-    files = glob(f'{local_directory}lc/*.fits')
+def plot_contamination(local_directory=None, gaia_dr3=None, ymin=None, ymax=None, pm_years=3000):
+    sns.set(rc={'font.family': 'serif', 'font.serif': 'DejaVu Serif', 'font.size': 12,
+                'axes.edgecolor': '0.2', 'axes.labelcolor': '0.', 'xtick.color': '0.', 'ytick.color': '0.',
+                'axes.facecolor': '0.95', 'grid.color': '0.9'})
+    files = glob(f'{local_directory}lc/*{gaia_dr3}*.fits')
     os.makedirs(f'{local_directory}plots/', exist_ok=True)
     for i in range(len(files)):
         with fits.open(files[i], mode='denywrite') as hdul:
             sector = hdul[0].header['SECTOR']
+            q = [a and b for a, b in
+                 zip(list(hdul[1].data['TESS_flags'] == 0), list(hdul[1].data['TGLC_flags'] == 0))]
+            if ymin is None and ymax is None:
+                ymin = np.nanmin(hdul[1].data['cal_aper_flux'][q]) - 0.01
+                ymax = np.nanmax(hdul[1].data['cal_aper_flux'][q]) + 0.01
             with open(glob(f'{local_directory}source/*_{sector}.pkl')[0], 'rb') as input_:
                 source = pickle.load(input_)
                 source.select_sector(sector=sector)
                 star_num = np.where(source.gaia['DESIGNATION'] == f'Gaia DR3 {gaia_dr3}')
-                plt.imshow(source.flux[0], origin='lower')
-                plt.show()
-                plt.close()
-                # print(source.gaia[891])
-                # print(source.gaia[star_num])
-                nearby_stars = np.argsort(
+
+                distances = np.sqrt(
                     (source.gaia[f'sector_{sector}_x'][:500] - source.gaia[star_num][f'sector_{sector}_x']) ** 2 +
-                    (source.gaia[f'sector_{sector}_y'][:500] - source.gaia[star_num][f'sector_{sector}_y']) ** 2)[0:5]
-                # print(f'sector = {source.sector}')
+                    (source.gaia[f'sector_{sector}_y'][:500] - source.gaia[star_num][f'sector_{sector}_y']) ** 2)
+
+                # Find closest 5 stars (6-self) or those within 5 pixels
+                nearby_stars = np.argsort(distances)[:6]
+                nearby_stars = nearby_stars[distances[nearby_stars] <= 5]
                 star_x = source.gaia[star_num][f'sector_{sector}_x'][0]
                 star_y = source.gaia[star_num][f'sector_{sector}_y'][0]
                 max_flux = np.nanmax(
-                    np.nanmedian(source.flux[:, round(star_y) - 2:round(star_y) + 3, round(star_x) - 2:round(star_x) + 3],
-                              axis=0))
-                fig = plt.figure(constrained_layout=False, figsize=(15, 5))
-                gs = fig.add_gridspec(5, 16)
-                gs.update(wspace=0.1, hspace=0.1)
-                ax0 = fig.add_subplot(gs[:5, :5])
+                    np.nanmedian(
+                        source.flux[:, round(star_y) - 2:round(star_y) + 3, round(star_x) - 2:round(star_x) + 3],
+                        axis=0))
+                fig = plt.figure(constrained_layout=False, figsize=(20, 12))
+                gs = fig.add_gridspec(21, 10)
+                gs.update(wspace=0.03, hspace=0.1)
+                ax0 = fig.add_subplot(gs[:10, :3])
                 ax0.imshow(np.median(source.flux, axis=0), cmap='RdBu', vmin=-max_flux, vmax=max_flux, origin='lower')
-
-                ax0.scatter(source.gaia[f'sector_{sector}_x'][:500], source.gaia[f'sector_{sector}_y'][:500], s=50,
-                            c='r', label='background stars')
-                ax0.scatter(source.gaia[f'sector_{sector}_x'][nearby_stars],
-                            source.gaia[f'sector_{sector}_y'][nearby_stars], s=50,
+                ax0.set_xlabel('x pixel')
+                ax0.set_ylabel('y pixel')
+                ax0.scatter(star_x, star_y, s=300, c='r', marker='*', label='target star')
+                ax0.scatter(source.gaia[f'sector_{sector}_x'][:500], source.gaia[f'sector_{sector}_y'][:500], s=30,
                             c='r', label='background stars')
+                ax0.scatter(source.gaia[f'sector_{sector}_x'][nearby_stars[nearby_stars != star_num[0][0]]],
+                            source.gaia[f'sector_{sector}_y'][nearby_stars[nearby_stars != star_num[0][0]]],
+                            s=30, c='r', edgecolor='black', linewidth=1, label='background stars')
+                ax0.grid(False)
                 for l in range(len(nearby_stars)):
                     index = np.where(
                         source.tic['dr3_source_id'] == int(source.gaia['DESIGNATION'][nearby_stars[l]].split(' ')[-1]))
                     gaia_targets = source.gaia
                     median_time = np.median(source.time)
-                    interval = (median_time - 388.5) / 365.25 + 3000
+                    interval = (median_time - 388.5) / 365.25 + pm_years
                     ra = gaia_targets['ra'][nearby_stars[l]]
                     dec = gaia_targets['dec'][nearby_stars[l]]
                     if not np.isnan(gaia_targets['pmra'][nearby_stars[l]]):
@@ -352,43 +362,37 @@ def plot_contamination(local_directory=None, gaia_dr3=None):
                               y_gaia - source.gaia[f'sector_{sector}_y'][nearby_stars[l]],
                               width=0.02, color='r', edgecolor=None, head_width=0.1)
                     try:
-                        ax0.text(source.gaia[f'sector_{sector}_x'][nearby_stars[l]] - 0.1,
-                                 source.gaia[f'sector_{sector}_y'][nearby_stars[l]] + 0.3,
-                                 f'TIC {int(source.tic["TIC"][index])}', rotation=90)
-                    except kindError:
-                        ax0.text(source.gaia[f'sector_{sector}_x'][nearby_stars[l]] - 0.1,
-                                 source.gaia[f'sector_{sector}_y'][nearby_stars[l]] + 0.2,
-                                 f'{source.gaia[f"DESIGNATION"][nearby_stars[l]]}', rotation=90)
-                ax0.scatter(star_x, star_y, s=300, c='r', marker='*', label='target star')
+                        txt = ax0.text(source.gaia[f'sector_{sector}_x'][nearby_stars[l]] + 0.5,
+                                       source.gaia[f'sector_{sector}_y'][nearby_stars[l]] - 0.05,
+                                       f'TIC {int(source.tic["TIC"][index])}', size=7)
+
+                    except TypeError:
+                        designation = source.gaia[f"DESIGNATION"][nearby_stars[l]]
+                        formatted_text = '\n'.join([designation[i:i + 15] for i in range(0, len(designation), 15)])
 
-                # ax0.legend()
+                        txt = ax0.text(source.gaia[f'sector_{sector}_x'][nearby_stars[l]] + 0.5,
+                                       source.gaia[f'sector_{sector}_y'][nearby_stars[l]] - 0.05,
+                                       formatted_text, size=7)
                 ax0.set_xlim(round(star_x) - 5.5, round(star_x) + 5.5)
                 ax0.set_ylim(round(star_y) - 5.5, round(star_y) + 5.5)
-                ax0.set_title(f'TIC_{hdul[0].header["TICID"]}_sector_{hdul[0].header["SECTOR"]:04d}')
-                ax0.vlines(round(star_x) - 2.5, round(star_y) - 2.5, round(star_y) + 2.5, colors='k')
-                ax0.vlines(round(star_x) + 2.5, round(star_y) - 2.5, round(star_y) + 2.5, colors='k')
-                ax0.hlines(round(star_y) - 2.5, round(star_x) - 2.5, round(star_x) + 2.5, colors='k')
-                ax0.hlines(round(star_y) + 2.5, round(star_x) - 2.5, round(star_x) + 2.5, colors='k')
-                # for j in range(5):
-                #     for k in range(5):
-                #         ax_ = fig.add_subplot(gs[(4 - j), (5 + k)])
-                #         ax_.patch.set_facecolor('C0')
-                #         ax_.patch.set_alpha(max(0, np.median(source.flux[:, round(star_y) - 2 + j, round(star_x) - 2 + k]) / max_flux))
-                #         cal_lc, trend = flatten(hdul[1].data['time'],
-                #                                      source.flux[:, round(star_y) - 2 + j, round(star_x) - 2 + k],
-                #                                      window_length=1, method='biweight', return_trend=True)
-                #         ax_.plot(hdul[1].data['time'], cal_lc, '.k', ms=1, label='center pixel')
-
-                t_, y_, x_ = np.shape(hdul[0].data)
+                ax0.set_title(f'TIC_{hdul[0].header["TICID"]}_Sector_{hdul[0].header["SECTOR"]:04d}')
+                ax0.vlines(round(star_x) - 2.5, round(star_y) - 2.5, round(star_y) + 2.5, colors='k', lw=1.2)
+                ax0.vlines(round(star_x) + 2.5, round(star_y) - 2.5, round(star_y) + 2.5, colors='k', lw=1.2)
+                ax0.hlines(round(star_y) - 2.5, round(star_x) - 2.5, round(star_x) + 2.5, colors='k', lw=1.2)
+                ax0.hlines(round(star_y) + 2.5, round(star_x) - 2.5, round(star_x) + 2.5, colors='k', lw=1.2)
+                try:
+                    t_, y_, x_ = np.shape(hdul[0].data)
+                except ValueError:
+                    warnings.warn('Light curves need to have the primary hdu. Set save_aperture=True when producing the light curve to enable this plot.')
+                    sys.exit()
                 max_flux = np.max(
                     np.median(source.flux[:, int(star_y) - 2:int(star_y) + 3, int(star_x) - 2:int(star_x) + 3], axis=0))
+                arrays = []
                 for j in range(y_):
                     for k in range(x_):
-                        ax_ = fig.add_subplot(gs[(4 - j), (6 + 2 * k):(8 + 2 * k)])
-                        ax_.patch.set_facecolor('C0')
+                        ax_ = fig.add_subplot(gs[(19 - 2 * j):(21 - 2 * j), (2 * k):(2 + 2 * k)])
+                        ax_.patch.set_facecolor('#4682B4')
                         ax_.patch.set_alpha(min(1, max(0, 5 * np.nanmedian(hdul[0].data[:, j, k]) / max_flux)))
-                        q = [a and b for a, b in
-                             zip(list(hdul[1].data['TESS_flags'] == 0), list(hdul[1].data['TGLC_flags'] == 0))]
 
                         _, trend = flatten(hdul[1].data['time'][q],
                                            hdul[0].data[:, j, k][q] - np.nanmin(hdul[0].data[:, j, k][q]) + 1000,
@@ -396,18 +400,68 @@ def plot_contamination(local_directory=None, gaia_dr3=None):
                         cal_aper = (hdul[0].data[:, j, k][q] - np.nanmin(
                             hdul[0].data[:, j, k][q]) + 1000 - trend) / np.nanmedian(
                             hdul[0].data[:, j, k][q]) + 1
-                        # ax_.plot(hdul[1].data['time'][q], cal_aper, '.k', ms=0.5)
-                        ax_.plot(hdul[1].data['time'][q], hdul[0].data[:, j, k][q], '.k', ms=0.5)
-                        # ax_.set_ylim(0.7, 1.3)
+                        if 1 <= j <= 3 and 1 <= k <= 3:
+                            arrays.append(cal_aper)
+                        ax_.plot(hdul[1].data['time'][q], cal_aper, '.k', ms=0.5)
+                        # ax_.plot(hdul[1].data['time'][q], hdul[0].data[:, j, k][q], '.k', ms=0.5)
+                        ax_.set_ylim(ymin, ymax)
+                        ax_.set_xlabel('TBJD')
+                        ax_.set_ylabel('')
                         if j != 0:
                             ax_.set_xticklabels([])
+                            ax_.set_xlabel('')
                         if k != 0:
                             ax_.set_yticklabels([])
-
-                plt.savefig(f'{local_directory}plots/contamination_sector_{hdul[0].header["SECTOR"]:04d}.pdf',
-                            dpi=300)
-                plt.close()
-
+                        if j == 2 and k == 0:
+                            ax_.set_ylabel('Normalized and detrended Flux of each pixel')
+
+                combinations = itertools.combinations(arrays, 2)
+                median_abs_diffs = []
+                for arr_a, arr_b in combinations:
+                    abs_diff = np.abs(arr_a - arr_b)
+                    median_diff = np.median(abs_diff)
+                    median_abs_diffs.append(median_diff)
+                median_abs_diffs = np.array(median_abs_diffs)
+                iqr = np.percentile(median_abs_diffs, 75) - np.percentile(median_abs_diffs, 25)
+                print(f"Interquartile Range (IQR): {iqr}")
+                std_dev = np.std(median_abs_diffs)
+                print(f"Standard Deviation: {std_dev}")
+                ax1 = fig.add_subplot(gs[:10, 4:7])
+                ax1.hist(median_abs_diffs, color='k', edgecolor='k', facecolor='none', rwidth=0.8, linewidth=2)
+                ax1.set_box_aspect(1)
+                ax1.set_title(f'Distribution of the MADs among combinations of the center 3*3 pixels')
+                ax1.set_xlabel('MAD between combinations of center 3*3 pixel fluxes')
+                ax1.set_ylabel('Counts')
+                text_ax = fig.add_axes([0.71, 0.9, 0.3, 0.3])  # [left, bottom, width, height] in figure coordinates
+                text_ax.axis('off')  # Turn off axis lines, ticks, etc.
+                text_ax.text(0., 0., f"Gaia DR3 {gaia_dr3} \n"
+                                     f" ←← TESS SPOC FFI and TIC/Gaia stars with proper motions. \n"
+                                     f"        Arrows show Gaia proper motion after {pm_years} years. \n"
+                                     f" ←  Histogram of the MADs between 3*3 pixel fluxes. \n"
+                                     f" ↓  Fluxes of each pixels after contaminations are removed. \n"
+                                     f"      The fluxes are normalized and detrended. The background \n"
+                                     f"      color shows the pixel brightness after the decontamination. \n"
+                                     f"\n"
+                                     f"How to interpret these plots: \n"
+                                     f"     If the signals you are interested in (i.e. transits, \n"
+                                     f"     eclipses, variable stars) show similar amplitudes in \n"
+                                     f"     all (especially the center 3*3) pixels, then the star \n"
+                                     f"     is likely to be the source. The median absolute \n"
+                                     f"     differences (MADs) taken between all combinations \n"
+                                     f"     of the center pixel fluxes are shown in the histogram \n"
+                                     f"     for a quantititive comparison to other possible sources. \n"
+                                     f"     The star with smaller distribution width (IQR or \n"
+                                     f"     STD) is more likely to be the source of the signal. \n"
+                                     f"\n"
+                                     f"Interquartile Range (IQR): {iqr:05f} \n"
+                                     f"Standard Deviation: {std_dev:05f}", transform=text_ax.transAxes, ha='left',
+                             va='top')
+                plt.subplots_adjust(top=.98, bottom=0.05, left=0.05, right=0.95)
+                plt.savefig(
+                    f'{local_directory}plots/contamination_sector_{hdul[0].header["SECTOR"]:04d}_Gaia_DR3_{gaia_dr3}.pdf',
+                    dpi=300)
+                # plt.savefig(f'{local_directory}plots/contamination_sector_{hdul[0].header["SECTOR"]:04d}_Gaia_DR3_{gaia_dr3}.png',
+                #             dpi=600)
 
 def plot_epsf(local_directory=None):
     files = glob(f'{local_directory}epsf/*.npy')
@@ -458,7 +512,7 @@ def get_tglc_lc(tics=None, method='query', server=1, directory=None, prior=None)
 
 if __name__ == '__main__':
     tics = [16005254]
-    directory = f'/home/tehan/data/'
+    directory = f'/Users/tehan/Downloads/'
     os.makedirs(directory, exist_ok=True)
     get_tglc_lc(tics=tics, method='query', server=1, directory=directory)
     # plot_lc(local_directory=f'{directory}TIC {tics[0]}/', kind='cal_aper_flux')

{tglc-0.6.4 → tglc-0.6.6}/tglc/target_lightcurve.py

@@ -20,7 +20,7 @@ warnings.simplefilter('always', UserWarning)
 def lc_output(source, local_directory='', index=0, time=None, psf_lc=None, cal_psf_lc=None, aper_lc=None,
               cal_aper_lc=None, bg=None, tess_flag=None, tglc_flag=None, cadence=None, aperture=None,
               cut_x=None, cut_y=None, star_x=2, star_y=2, x_aperture=None, y_aperture=None, near_edge=False,
-              local_bg=None, save_aper=False, portion=1, prior=None, transient=None):
+              local_bg=None, save_aper=False, portion=1, prior=None, transient=None, target_5x5=None, field_stars_5x5=None):
     """
     lc output to .FITS file in MAST HLSP standards
     :param tglc_flag: np.array(), required
@@ -82,6 +82,13 @@ def lc_output(source, local_directory='', index=0, time=None, psf_lc=None, cal_p
         primary_hdu = fits.PrimaryHDU(aperture)
     else:
         primary_hdu = fits.PrimaryHDU()
+    # Simulated star images based on ePSF, used to estimate contamination ratio and others
+    image_data = np.zeros((3, 5, 5))
+    image_data[0] = target_5x5
+    image_data[1] = field_stars_5x5
+    # This is the pixel-wise contamination ratio
+    image_data[2] = field_stars_5x5/target_5x5
+    image_hdu = fits.ImageHDU(data=image_data)
 
     primary_hdu.header = fits.Header(cards=[
         fits.Card('SIMPLE', True, 'conforms to FITS standard'),
@@ -116,6 +123,7 @@ def lc_output(source, local_directory='', index=0, time=None, psf_lc=None, cal_p
         fits.Card('GAIA_bp', gaia_bp, 'Gaia DR3 bp band magnitude'),
         fits.Card('GAIA_rp', gaia_rp, 'Gaia DR3 rp band magnitude'),
         fits.Card('RAWFLUX', raw_flux, 'median flux of raw FFI'),
+        fits.Card('CONTAMRT', round(np.nansum(field_stars_5x5[1:4,1:4])/np.nansum(target_5x5[1:4,1:4]), 9), 'contamination ratio of default 3*3 aperture'),
         fits.Card('CALIB', 'TGLC', 'pipeline used for image calibration')])
     if save_aper:
         primary_hdu.header.comments['NAXIS1'] = "Time (hdul[1].data['time'])"
@@ -190,13 +198,14 @@ def lc_output(source, local_directory='', index=0, time=None, psf_lc=None, cal_p
     if type(prior) == float:
         table_hdu.header.append(('PRIOR', prior, 'prior of field stars'), end=True)
 
-    hdul = fits.HDUList([primary_hdu, table_hdu])
+    hdul = fits.HDUList([primary_hdu, table_hdu, image_hdu])
     hdul.writeto(
-        f'{local_directory}hlsp_tglc_tess_ffi_gaiaid-{objid}-s{source.sector:04d}-cam{source.camera}-ccd{source.ccd}_tess_v1_llc.fits',
+        f'{local_directory}hlsp_tglc_tess_ffi_gaiaid-{objid}-s{source.sector:04d}-cam{source.camera}-ccd{source.ccd}_tess_v2_llc.fits',
         overwrite=True)
     return
 
 
+
 def epsf(source, psf_size=11, factor=2, local_directory='', target=None, cut_x=0, cut_y=0, sector=0,
          limit_mag=16, edge_compression=1e-4, power=1.4, name=None, save_aper=False, no_progress_bar=False, prior=None):
     """
@@ -310,7 +319,7 @@ def epsf(source, psf_size=11, factor=2, local_directory='', target=None, cut_x=0
                     fit_lc_float_field(A, source, star_info=star_info, x=x_round, y=y_round, star_num=i, e_psf=e_psf,
                                        near_edge=near_edge, prior=prior)
             else:
-                aperture, psf_lc, star_y, star_x, portion = \
+                aperture, psf_lc, star_y, star_x, portion, target_5x5, field_stars_5x5 = \
                     fit_lc(A, source, star_info=star_info, x=x_round[i], y=y_round[i], star_num=i, e_psf=e_psf,
                            near_edge=near_edge)
             aper_lc = np.sum(
@@ -359,4 +368,4 @@ def epsf(source, psf_size=11, factor=2, local_directory='', target=None, cut_x=0
                               bg=background_, time=source.time, psf_lc=psf_lc, cal_psf_lc=cal_psf_lc, aper_lc=aper_lc,
                               cal_aper_lc=cal_aper_lc, local_bg=local_bg, x_aperture=x_aperture[i],
                               y_aperture=y_aperture[i], near_edge=near_edge, save_aper=save_aper, portion=portion,
-                              prior=prior, transient=source.transient)
+                              prior=prior, transient=source.transient, target_5x5=target_5x5, field_stars_5x5=field_stars_5x5)

{tglc-0.6.4 → tglc-0.6.6/tglc.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tglc
-Version: 0.6.4
+Version: 0.6.6
 Summary: TESS-Gaia Light Curve
 Home-page: https://github.com/TeHanHunter/TESS_Gaia_Light_Curve
 Author: Te Han
@@ -8,9 +8,21 @@ Author-email: tehanhunter@gmail.com
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.8
+Requires-Python: >=3.8, <3.13
 Description-Content-Type: text/x-rst
 License-File: LICENSE
+Requires-Dist: astropy>=5.1
+Requires-Dist: astroquery==0.4.7
+Requires-Dist: matplotlib
+Requires-Dist: numpy
+Requires-Dist: oauthlib
+Requires-Dist: requests
+Requires-Dist: scipy
+Requires-Dist: threadpoolctl
+Requires-Dist: tqdm
+Requires-Dist: wheel
+Requires-Dist: wotan
+Requires-Dist: seaborn
 
 ==================================
 Introduction
@@ -18,6 +30,7 @@ Introduction
 
 TESS-Gaia Light Curve (`TGLC <https://archive.stsci.edu/hlsp/tglc>`_) is a dataset of TESS full-frame image light curves publicly available via the MAST portal. It is fitted with effective PSF and decontaminated with Gaia DR3 and achieved percent-level photometric precision down to 16th TESS magnitude! It unlocks astrophysics to a vast number of dim stars below 12th TESS magnitude. A package called tglc is pip-installable for customized light curve fits.
 
+
 ==================================
 Usage
 ==================================

{tglc-0.6.4 → tglc-0.6.6}/tglc.egg-info/requires.txt

@@ -1,5 +1,5 @@
 astropy>=5.1
-astroquery
+astroquery==0.4.7
 matplotlib
 numpy
 oauthlib
@@ -9,3 +9,4 @@ threadpoolctl
 tqdm
 wheel
 wotan
+seaborn

tglc-0.6.4/tglc/__init__.py

@@ -1,3 +0,0 @@
-__version__ = "0.6.4"
-__author__ = 'Te Han, Timothy Brandt'
-__credits__ = 'University of California, Santa Barbara'