pyAvMap.py

#!/usr/bin/env python3

#  Copyright (c) 2019 Garrett Herschleb
#
#  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 2 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, write to the Free Software
#  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

import sys, os, time

import logging
import logging.config
import argparse
try:
    from PyQt5.QtGui import *
    from PyQt5.QtWidgets import *
    from PyQt5.QtCore import *
    PYQT = 5
except:
    from PyQt4.QtGui import *
    from PyQt4.QtCore import *
    PYQT = 4
import yaml
import gui
import hmi
from hmi import Menu

if "pyAvTools" not in ''.join(sys.path):
    neighbor_tools = os.path.join ('..', 'pyAvTools')
    if os.path.isdir (neighbor_tools):
        sys.path.append (neighbor_tools)
    elif 'TOOLS_PATH' in os.environ:
        sys.path.append (os.environ['TOOLS_PATH'])

try:
    import pyavui
    import pyavtools.fix as fix
except:
    print ("You need to have pyAvTools installed, or in an adjacent directory to pyAvMap.")
    print ("Or set the environment variable 'TOOLS_PATH' to point to the location of pyAvTools.")
    sys.exit(-1)

import pyavmap

class Main(QMainWindow):
    keyPress = pyqtSignal(QEvent)
    keyRelease = pyqtSignal(QEvent)
    def keyPressEvent(self, event):
        self.keyPress.emit(event)
    def keyReleaseEvent(self, event):
        self.keyRelease.emit(event)


if __name__ == "__main__":
    app = QApplication(sys.argv)
    parser = argparse.ArgumentParser(description='pyAvMap')
    parser.add_argument('--debug', action='store_true',
                        help='Run in debug mode')
    parser.add_argument('--verbose', '-v', action='store_true',
                        help='Run in verbose mode')
    parser.add_argument('--config-file', default='config/main.yaml', type=argparse.FileType('r'),
                        help='Alternate configuration file')

    args = parser.parse_args()

    # if we passed in a configuration file on the command line...
    config = yaml.load(args.config_file)

    if 'logging' in config:
        logging.config.dictConfig(config['logging'])
    else:
        logging.basicConfig()

    log = logging.getLogger()
    if args.verbose:
        log.setLevel(logging.INFO)
    if args.debug:
        log.setLevel(logging.DEBUG)
    log.info("Starting pyAvMap")

    log.debug("PyQT Version = %d" % PYQT)

    fix.initialize(config)

    pyavmap.configure_charts (config['charts_dir'])

    main_window = Main()
    screenWidth = int(config["main"]["screenWidth"])
    screenHeight = int(config["main"]["screenHeight"])
    main_window.resize(screenWidth, screenHeight)
    avmap = pyavmap.AvMap (config, main_window)
    avmap.resize(screenWidth, screenHeight)

    hmi.initialize(config)
    hmi.keys.initialize(main_window, config["keybindings"])
    if 'menu' in config:
        menu = Menu(main_window, config["menu"])
        menu.start()
        menu.register_map(avmap)
        enc = "ENC1" if "rotary_encoder" not in config else config["rotary_encoder"]
        enc_sel = "BTN6" if "rotary_select" not in config else config["rotary_select"]
        ctsel_widget = gui.ChartTypeSel(enc, enc_sel, pyavmap.chart_types(),
                        avmap.set_chart_type, main_window)
        ctsel_widget.resize (100, 65)
        ctsel_widget.move (30, 32)
        menu.register_target ("Chart Type", ctsel_widget)


    if 'displays' in config:
        for title,contents in config['displays'].items():
            position = [0,0]
            kwargs = {'parent': main_window}
            dbitems = list()
            for k,v in contents.items():
                if k == 'keys':
                    for key in v:
                        item = fix.db.get_item(key)
                        dbitems.append(item)
                elif k == 'position':
                    position = v
                else:
                    kwargs[k] = v
            d = pyavui.FIXDisplay(title, dbitems, **kwargs)
            if position[0] < 0:
                position[0] = screenWidth+position[0]-d.width()
            if position[1] < 0:
                position[1] = screenHeight+position[1]-d.height()
            d.move (*tuple(position))

    main_window.show()
    track = fix.db.get_item("TRACK")
    avmap.setTrack(track.value)
    lat = fix.db.get_item("LAT")
    avmap.setLat(lat.value)
    lon = fix.db.get_item("LONG")
    avmap.setLon (lon.value)
    lat.valueChanged[float].connect(avmap.setLat)
    lon.valueChanged[float].connect(avmap.setLon)
    track.valueChanged[float].connect(avmap.setTrack)

    # Main program loop
    result = app.exec_()

    # Clean up and get out
    fix.stop()
    log.info("PyAvMap Exiting Normally")
    sys.exit(result)