def spoc_rot_fig(self):
flux_type = ['sap_flux', 'pdcsap_flux']
self.target_info_query()
if 'tess_pdc_rot_dict' in self.available_attributes:
for flux in flux_type:
if flux == 'sap_flux':
LS_res = self.tess_sap_rot_dict['LS_res']
LS_periodogram = self.tess_sap_rot_dict['LS_periodogram']
AC_res = self.tess_sap_rot_dict['AC_res']
AC_periodogram = self.tess_sap_rot_dict['AC_periodogram']
self.tess_sap_rot_fig = rot_tools.period_graph(
target_name='TIC ' + str(self.tic),
lc_df=self.spoc_lc,
flux_type=flux,
LS_res=LS_res,
LS_periodogram=LS_periodogram,
AC_res=AC_res,
AC_periodogram=AC_periodogram,
target_df=self.target_df)
if flux == 'pdcsap_flux':
LS_res = self.tess_pdc_rot_dict['LS_res']
LS_periodogram = self.tess_pdc_rot_dict['LS_periodogram']
AC_res = self.tess_pdc_rot_dict['AC_res']
AC_periodogram = self.tess_pdc_rot_dict['AC_periodogram']
self.tess_pdc_rot_fig = rot_tools.period_graph(
target_name='TIC ' + str(self.tic),
lc_df=self.spoc_lc,
flux_type=flux,
LS_res=LS_res,
LS_periodogram=LS_periodogram,
AC_res=AC_res,
AC_periodogram=AC_periodogram,
target_df=self.target_df)
else:
print("Need to run_spoc_rots first!")
def kepler_rot_fig(self):
flux_type = ['sap_flux', 'pdcsap_flux']
keep_cols = [
'time', 'sap_flux', 'sap_flux_err', 'pdcsap_flux',
'pdcsap_flux_err', 'sector'
]
kepler_lc = self.kepler_lc.rename(
columns={'quarter': 'sector'})[keep_cols]
self.target_info_query()
if 'kepler_pdc_rot_dict' in self.available_attributes:
for flux in flux_type:
if flux == 'sap_flux':
LS_res = self.kepler_sap_rot_dict['LS_res']
LS_periodogram = self.kepler_sap_rot_dict['LS_periodogram']
AC_res = self.kepler_sap_rot_dict['AC_res']
AC_periodogram = self.kepler_sap_rot_dict['AC_periodogram']
self.kepler_sap_rot_fig = rot_tools.period_graph(
target_name='KIC ' + str(self.kic),
lc_df=kepler_lc,
flux_type=flux,
LS_res=LS_res,
LS_periodogram=LS_periodogram,
AC_res=AC_res,
AC_periodogram=AC_periodogram,
target_df=self.target_df)
if flux == 'pdcsap_flux':
LS_res = self.kepler_pdc_rot_dict['LS_res']
LS_periodogram = self.kepler_pdc_rot_dict['LS_periodogram']
AC_res = self.kepler_pdc_rot_dict['AC_res']
AC_periodogram = self.kepler_pdc_rot_dict['AC_periodogram']
self.kepler_pdc_rot_fig = rot_tools.period_graph(
target_name='KIC ' + str(self.kic),
lc_df=kepler_lc,
flux_type=flux,
LS_res=LS_res,
LS_periodogram=LS_periodogram,
AC_res=AC_res,
AC_periodogram=AC_periodogram,
target_df=self.target_df)
else:
print("Need to run_kepler_rots first!")
def tpf_rot_fig(self):
if 'tpf_rot_dict' in self.available_attributes:
LS_res = self.tpf_rot_dict['LS_res']
LS_periodogram = self.tpf_rot_dict['LS_periodogram']
AC_res = self.tpf_rot_dict['AC_res']
AC_periodogram = self.tpf_rot_dict['AC_periodogram']
self.tpf_rot_fig = rot_tools.period_graph(
target_name='TIC ' + str(self.tic),
lc_df=self.tpf_lc,
flux_type='flux',
LS_res=LS_res,
LS_periodogram=LS_periodogram,
AC_res=AC_res,
AC_periodogram=AC_periodogram)
else:
print("Need to run_spoc_rots first!")
def run_rotation_tools(self,
flux_type=['SAP_flux', 'PDCSAP_flux'],
flux_err_avail=True,
min_freq=1 / 30):
try:
for flux in flux_type:
LS_res, LS_periodogram_df = rotation_tools.my_LS_multi_sector(
lc_df=self.spoc_lc,
flux_type=flux,
flux_err_avail=flux_err_avail,
min_freq=min_freq)
AC_res, AC_periodogram = rotation_tools.exo_acf_multi_sector(
lc_df=self.spoc_lc,
flux_type=flux,
flux_err_avail=flux_err_avail,
max_per=1 / min_freq)
amp_df = rotation_tools.amp_multi_sector(lc_df=self.spoc_lc,
flux_type=flux)
rot_fig = rotation_tools.period_graph(
target_name='TIC ' + str(self.tic),
lc_df=self.spoc_lc,
flux_type=flux,
LS_res=LS_res,
LS_periodogram=LS_periodogram_df,
AC_res=AC_res,
AC_periodogram=AC_periodogram)
if flux == 'SAP_flux':
self.LS_res_sap = LS_res
self.LS_periodogram_sap = LS_periodogram_df
self.AC_res_sap = AC_res
self.AC_periodogram_sap = AC_periodogram
self.amp_df_sap = amp_df
self.rot_fig_sap = rot_fig
if flux == 'PDCSAP_flux':
self.LS_res_pdc = LS_res
self.LS_periodogram_pdc = LS_periodogram_df
self.AC_res_pdc = AC_res
self.AC_periodogram_pdc = AC_periodogram
self.amp_df_pdc = amp_df
self.rot_fig_pdc = rot_fig
print("Rotations added!")
except:
print("Need to run 'download' function first!")
def cpm_rot_fig(self):
if 'cpm_rot_dict' in self.available_attributes:
self.target_info_query()
LS_res = self.cpm_rot_dict['LS_res']
LS_periodogram = self.cpm_rot_dict['LS_periodogram']
AC_res = self.cpm_rot_dict['AC_res']
AC_periodogram = self.cpm_rot_dict['AC_periodogram']
self.cpm_rot_fig = rot_tools.period_graph(
target_name='TIC ' + str(self.tic),
lc_df=self.cpm_lc,
flux_type='cpm',
LS_res=LS_res,
LS_periodogram=LS_periodogram,
AC_res=AC_res,
AC_periodogram=AC_periodogram,
target_df=self.target_df)
else:
print("Need to run_cpm_rots first!")
def k2sff_rot_fig(self):
if 'k2sff_rot_dict' in self.available_attributes:
keep_cols = ['time', 'flux', 'sector']
k2sff_lc = self.k2sff_lc.rename(
columns={'campaign': 'sector'})[keep_cols]
LS_res = self.k2sff_rot_dict['LS_res']
LS_periodogram = self.k2sff_rot_dict['LS_periodogram']
AC_res = self.k2sff_rot_dict['AC_res']
AC_periodogram = self.k2sff_rot_dict['AC_periodogram']
self.k2sff_rot_fig = rot_tools.period_graph(
target_name='EPIC ' + str(self.epic),
lc_df=k2sff_lc,
flux_type='flux',
LS_res=LS_res,
LS_periodogram=LS_periodogram,
AC_res=AC_res,
AC_periodogram=AC_periodogram)
else:
print("Need to run_k2sff_rots first!")