from PyQt5.QtWidgets import *
from PyQt5.QtGui import QPixmap
import sys
from PyQt5 import QtCore
from functions.get_img_map import get_img_map


class Window(QMainWindow):
    def __init__(self):
        super().__init__()
        self.acceptDrops()
        self.setWindowTitle("Image")
        self.setGeometry(0, 0, 600, 450)
        self.label = QLabel(self)
        self.delta = 0.02
        self.lon = 37.530887
        self.lat = 55.703118
        get_img_map(self.delta, (self.lon, self.lat))
        self.pixmap = QPixmap('image.png')
        self.label.setPixmap(self.pixmap)
        self.label.resize(self.pixmap.width(), self.pixmap.height())
        self.show()

    def keyPressEvent(self, event):
        try:
            if event.key() == QtCore.Qt.Key_PageUp:
                self.delta += 0.01
                self.update()
            if event.key() == QtCore.Qt.Key_PageDown:
                self.delta -= 0.01
                self.update()
            if event.key() == QtCore.Qt.Key_W:
                self.lat += 0.01
                self.update()
            if event.key() == QtCore.Qt.Key_S:
                self.lat -= 0.01
                self.update()
            if event.key() == QtCore.Qt.Key_A:
                self.lon -= 0.01
                self.update()
            if event.key() == QtCore.Qt.Key_D:
                self.lon += 0.01
                self.update()
            event.accept()
        except Exception as e:
            print(e)

    def update(self):
        get_img_map(self.delta, (self.lon, self.lat))
        self.pixmap = QPixmap('image.png')
        self.label.setPixmap(self.pixmap)
        self.label.resize(self.pixmap.width(), self.pixmap.height())
        self.show()


App = QApplication(sys.argv)
window = Window()
sys.exit(App.exec())