main.py

# main.py
# Sets up GUI and displays .csv files

# This script utilizes the functions in the other files to calculate
# ephemerides for Starlink, determine which are visible, and create
# an ACP observing plan for the Pomenis telescope
#
# Harry Krantz
# Steward Observatory
# University of Arizona
# Copyright May 2020
#
# This original script has been modified for developing a user-friendly
# interface for planning Starlink satellite observations.
# This is intended for students and amateur astronomers.

# @author modifications: Ian Snider
# Truman State University

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

import datetime as dt
import os
import pandas as pd

from starlinkPassPredictor import *
from locations import *
from writeAcpPlan import *
from printPlan import *

from skyfield import api
from skyfield import almanac
 
from tkinter import *
from tkinter import filedialog, messagebox, ttk

from datetime import datetime





# Start of GUI block
print("Starting GUI")


# create root window
root = Tk()


# root window title and dimensions
root.title("Starlink Pass Predictor")

root.geometry('600x900')

# adding menu bar in root window
# new item in menu bar labeled as 'New'
# adding more items in menu bar
# menu = Menu(root)
# item = Menu(menu)
# item.add_command(label = 'New')
# menu.add_cascade(label = 'File', menu = item)
# root.config(menu = menu)

# frame for treeview
display_frame = LabelFrame(root, text="File Data")
display_frame.place(height=320,width=900)

#frame for file dialog
file_frame = LabelFrame(root, text="Open File")
file_frame.place(height=100, width=500, rely=0.4, relx=0)


#frame for plan calculation
calc_frame = LabelFrame(root, text="Calculate Observation Plan")
calc_frame.place(height=400, width=600, rely=0.55, relx=0)

#frame inside calc_frame for setting a specific time
setTime_frame = LabelFrame(calc_frame, text="Specific Start Time")
setTime_frame.place(height=100, width=500, rely=0.5, relx=0.05)


#end of style GUI Block


##############################

#Browse observation plan files
selectFileBtn = Button(file_frame, text="Select File", fg = "purple", command=lambda: file_dialog())
selectFileBtn.place(rely=0.65, relx=0.5)

#Load observation plan files
loadFileBtn = Button(file_frame, text="Load File", fg = "purple", command=lambda: load_data())
loadFileBtn.place(rely=0.65, relx=0.25)

# label file
label_file = ttk.Label(file_frame, text="No File Selected")
label_file.place(rely=0, relx=0)


##############################

#PassPredictor plan GUI output 

# #plan header
# planX = Label(root, text = "Name                     ID       Rise Time             Rise Azimuth   Peak Time             Peak Alt   Peak Azimuth   Set Time              Set Azimuth   Duration")
# planX.place(x=5,y=65)

# #textbox output for copy/pasting
# planBox = ScrolledText(root, width=120, font=("lucida", 13))
# planBox.place(x=5,y=85)


##############################

# treeview widget
tv1 = ttk.Treeview(display_frame)
tv1.place(relheight=1, relwidth=1)

treescrolly = Scrollbar(display_frame, orient="vertical", command=tv1.yview)
treescrollx = Scrollbar(display_frame, orient="horizontal", command=tv1.xview)

tv1.configure(xscrollcommand=treescrollx.set, yscrollcommand=treescrolly.set)
treescrollx.pack(side="bottom", fill="x")
treescrolly.pack(side="right", fill="y")


# csv file display functions

def file_dialog():
    fileName = filedialog.askopenfilename(initialdir="/StarlinkPassPredictorUI-main", title="Select a file", filetypes=(("csv files", "*.csv"),("All Files","*.*")))
    label_file["text"] = fileName
    return None

def load_data():
    filePath = label_file["text"]
    try:
        csv_fileName = r"{}".format(filePath)
        df = pd.read_csv(csv_fileName)
    except ValueError:
        tk.messagebox.showerror("Information", "Invalid file")
        return None
    except FileNotFoundError:
        tk.messagebox.showerror("Information", "File not found")
        return None
    
    clear_data()
    tv1["column"] = list(df.columns)
    tv1["show"] = "headings"
    for column in tv1["columns"]:
        tv1.heading(column, text=column)
        
    df_rows = df.to_numpy().tolist()
    for row in df_rows:
        tv1.insert("", "end", values=row)
    return None
    

def clear_data():
    tv1.delete(*tv1.get_children())
    return None


##############################

# adding a label to the root window
locLbl = Label(calc_frame, text = "Input location (Latitude, Longitude, Elevation)")
locLbl.place(rely=0.05, relx=0.05)


# adding Entry field

#Latitude
latEntry = Entry(calc_frame, width=15)
latEntry.place(rely=0.15, relx=0.05)

#Logitude
lonEntry = Entry(calc_frame, width=15)
lonEntry.place(rely=0.15, relx=0.35)

#Elevation
elevationEntry = Entry(calc_frame, width=15)
elevationEntry.place(rely=0.15, relx=0.65)

#Specific time
startTimeEntry = Entry(setTime_frame, width=15)
startTimeEntry.place(rely=0.25, relx=0.05)

###############################

# run printPlan button
btn = Button(calc_frame, text = "Calculate Plan", fg = "purple", command=calculatePlan(latEntry, lonEntry, elevationEntry))
btn.place(rely=0.3, relx=0.05)

calcLbl = Label(calc_frame, text = "*Takes 1-2 mins*")
calcLbl.place(rely=0.4, relx=0.05)

###############################

# #Find .txt file for GUI
# #yyyy-mm-dd
# thisDay = datetime.utcnow().strftime("%Y-%m-%d")

# #find file
# searchFile = thisDay + "_Starlink" + "/" + filename

# f = open(searchFile, "r")

# #configure GUI plan details text

# satLabel = Label(root, text=f.readline())
# satLabel.place(x=1005,y=0)

###########################


#Execute tkinter
root.mainloop()

###########################

print("Done!")