graphicalUI.py

"""
Graphical User Interface for input and output values
@author Michael Schuetze
"""

import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk
from calculateTravelTime import *

"""
Main function for the program
Calls functions to collect user input, make calculations, and display them for the user
"""
class createWindow(Gtk.Window):

    """
    Initializes the GUI and adds all of the elements to it
    """
    def __init__(self):
        # Create a new window for the GUI
        Gtk.Window.__init__(self, title="Travel Times")
        self.set_border_width(10)
        self.set_size_request(200, 100)

        # Grid Layout
        self.grid = Gtk.Grid()
        self.grid.set_row_spacing(5)
        self.add(self.grid)


        ### Top of the Grid ###

        # Box to hold all user input labels and fields
        self.userInput = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=10)
        self.grid.attach(self.userInput, 0, 0, 2, 2)

        # Box to hold user input label and field for distance
        self.userInputDistance = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, spacing=10)
        self.userInput.pack_start(self.userInputDistance, True, True, 0)

        # Label for user input of distance
        self.distanceLabel = Gtk.Label()
        self.distanceLabel.set_label("Please Enter a Distance (mi):")
        self.userInputDistance.pack_start(self.distanceLabel, True, True, 0)

        # Entry for user input of distance
        self.distanceInput = Gtk.Entry()
        self.userInputDistance.pack_start(self.distanceInput, True, True, 0)

        # Box to hold user input label and field for price
        self.userInputPrice = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, spacing=10)
        self.userInput.pack_start(self.userInputPrice, True, True, 0)

        # Label for user input of price
        self.priceLabel = Gtk.Label()
        self.priceLabel.set_label("Please Enter a Price for Gas ($):")
        self.userInputPrice.pack_start(self.priceLabel, True, True, 0)

        # Entry for user input of price
        self.priceInput = Gtk.Entry()
        self.userInputPrice.pack_start(self.priceInput, True, True, 0)


        # Button for calculating data
        self.calculateButton = Gtk.Button(label="Calculate")
        self.calculateButton.connect("clicked", self.calculateButtonClicked)
        self.grid.attach(self.calculateButton, 0, 2, 2, 1)


        ### Bottom of the Grid ###

        # Data to be placed in the travelTimeChart
        self.speedTimeCostLoss_list_store = Gtk.ListStore(int, str, str, str)
        # Chart to display speeds and travel times
        self.travelTimeChart = Gtk.TreeView(self.speedTimeCostLoss_list_store)

        # Add columns to the chart
        for i, colTitle in enumerate(["Speed", "Travel Time", "Cost of Trip", "Money Lost"]):
            renderer = Gtk.CellRendererText()
            column = Gtk.TreeViewColumn(colTitle, renderer, text=i)

            self.travelTimeChart.append_column(column)

        self.grid.attach(self.travelTimeChart, 0, 3, 2, 1)


    """
    Collects the user input and sends it to be converted into speeds, times, and costs
    """
    def calculateButtonClicked(self, widget):
        travelData = runCalculations(self.distanceInput.get_text(), self.priceInput.get_text())
        self.changeTravelData(travelData)


    """
    Changes the chart on the GUI to the appropriate data
    @param travelData:tuple(int, str, float, float)- list of tuples containing speeds and times/costs associated with those speeds
    """
    def changeTravelData(self, travelData):
        # Remove all the data from the previous list
        self.speedTimeCostLoss_list_store.clear()

        # Add in the new data so that it may be displayed
        for speedTimeCostLoss in travelData:
            self.speedTimeCostLoss_list_store.append(list(speedTimeCostLoss))


"""
Main function for the program
Calls the function that creates the UI
"""
def main():
    window= createWindow()
    window.connect("delete-event", Gtk.main_quit)
    window.show_all()
    Gtk.main()

if __name__ == "__main__":
    main()