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!")