from Map import Map
from CarController import CarController

from PySide6.QtWidgets import QFileDialog
from PySide6.QtCore import QElapsedTimer

class mainLoop():
    def __init__(self, parent):
        self.parent = parent
        self.painter = parent.ui.mainSurf

        self.map = Map()

        self.controller = CarController(self.map)

        self.painter.addMap(self.map)
        self.painter.addCarController(self.controller)

    def update(self):
        self.controller.update()

    def openNetwork(self):
        fileName = QFileDialog.getOpenFileName(self.parent,"Open Network", "./", "Network File (*.net.xml)")
        if fileName[0] == '':
            return
        self.map.fromPath(fileName[0])
        self.painter.generateTransform()

        self.controller.prepareRoute()

    def openVehicles(self):
        fileName = QFileDialog.getOpenFileName(self.parent,"Open Vehicle trip description", "./", "Route File (*.rou.xml)")
        if fileName[0] == '':
            return
        self.controller.fromPath(fileName[0])
        if self.map.isLoaded():
            self.controller.prepareRoute()

    def quickLoad(self):
        self.map.fromPath("test2.net.xml")
        self.painter.generateTransform()
        self.controller.fromPath("test5.rou.xml")
        self.controller.prepareRoute()

    def addInfosDisplay(self, widget):
        self.controller.setInfoWidget(widget)