#Python script to read the ligth curve data files from Campos Estrada et al. 2024
import matplotlib.pyplot as plt
import statsmodels.api as sm
import numpy as np
import matplotlib
import math
import pandas as pd
from numba import jit
from scipy.signal import savgol_filter

#data details to read the file
dt = np.dtype([('time', np.float64), ('extinction', np.float64), ('scattering', np.float64), ('transit_depth', np.float64)])

#specify the correct path to the data folder
folder = 'YOUR PATH HERE'
lc_file = 'light_curve.bin'

comp = 'Mg08Fe12SiO4'
dust_label = r'$\mathrm{Mg_{0.8}Fe_{1.2}SiO_4}$'
#NOTE t0 is related to the "t" factor in the folder the file is in.
#t0 = value after the t + 0.5
t0 = 5.5 
time = 5.0 #this is the value after the t in the folder name
size=1.5 #initial dust grain size
mdot = 1.0 #dust mass-loss rate
g = 'sph' #geometry

file = folder+lc_file
data = np.fromfile(file,dt)
df_plot = pd.DataFrame(data)
transit_depth = -1.0*(df_plot['transit_depth']-1.0)*100 #transit depth in percentage
#this function smooths out the data over the Kepler long cadence
td_smooth = savgol_filter(transit_depth, 6,1)