#!/usr/bin/env python
# coding: utf-8

# References
# Useful docs / articles / etc
#   
#   1 - [A Python wrapper around AHK](https://pypi.org/project/ahk/)
# 
#   2 - [OpenCV on Wheels](https://pypi.org/project/opencv-python/)
# 
#   3 - [Autopilot for Elite Dangerous using OpenCV and thoughts on CV enabled bots in visual-to-keyboard loop]
#   (https://networkgeekstuff.com/projects/autopilot-for-elite-dangerous-using-opencv-and-thoughts-on-cv-enabled-bots-in-visual-to-keyboard-loop/)
#   
#   4 - [Using PyInstaller to Easily Distribute Python Applications](https://realpython.com/pyinstaller-python/)
#   
#   5 - [Direct Input to a Game - Python Plays GTA V]
#   (https://pythonprogramming.net/direct-input-game-python-plays-gta-v/)
#   
#   6 - [Cross-platform GUI automation for human beings](https://pyautogui.readthedocs.io/en/latest/index.html)

# Imports
import datetime
import math
import random as rand
import sys
from datetime import datetime
from json import loads
from os import environ, listdir
from os.path import join, isfile, getmtime, abspath
from time import sleep
from xml.etree.ElementTree import parse

import cv2  # see reference 2
import numpy as np
from PIL import ImageGrab
from pyautogui import size  # see reference 6

from logger import logger
from settings_api import getOption
from src.directinput import EDKeyCodes, PressKey, ReleaseKey  # see reference 5


def resource_path(relative_path):
    """ Get absolute path to resource, works for dev and for PyInstaller """
    try:
        # PyInstaller creates a temp folder and stores path in _MEIPASS
        base_path = sys._MEIPASS
    except AttributeError:
        base_path = abspath(".")

    return join(base_path, relative_path)


# logger.info('\n'+200*'-'+'\n'+'---- AUTOPILOT DATA '+180*'-'+'\n'+200*'-')
logger.info('---- AUTOPILOT DATA ' + 180 * '-')

# Constants
RELEASE = 'v19.05.15-binary-18'
PATH_LOG_FILES = None
PATH_KEYBINDINGS = None
KEY_MOD_DELAY = 0.010
KEY_DEFAULT_DELAY = 0.200
KEY_REPEAT_DELAY = 0.100
FUNCTION_DEFAULT_DELAY = 0.500
SCREEN_WIDTH, SCREEN_HEIGHT = size()

logger.info('RELEASE=' + str(RELEASE))
logger.info('PATH_LOG_FILES=' + str(PATH_LOG_FILES))
logger.info('PATH_KEYBINDINGS=' + str(PATH_KEYBINDINGS))
logger.info('KEY_MOD_DELAY=' + str(KEY_MOD_DELAY))
logger.info('KEY_DEFAULT_DELAY=' + str(KEY_DEFAULT_DELAY))
logger.info('KEY_REPEAT_DELAY=' + str(KEY_REPEAT_DELAY))
logger.info('FUNCTION_DEFAULT_DELAY=' + str(FUNCTION_DEFAULT_DELAY))
logger.info('SCREEN_WIDTH=' + str(SCREEN_WIDTH))
logger.info('SCREEN_HEIGHT=' + str(SCREEN_HEIGHT))


# Read ED logs

# Get latest log file
def get_latest_log(path_logs=None):
    """Returns the full path of the latest (most recent) elite log file (journal) from specified path"""
    path_logs = getOption('JournalPath')
    if not path_logs:
        user_path = environ.get('USERPROFILE')
        if user_path:
            path_logs = user_path + "/Saved Games/Frontier Developments/Elite Dangerous"
            from settings_api import setOption
            setOption('JournalPath', path_logs)
        else:
            raise FileNotFoundError("Journal path not found, define it in configs")
    list_of_logs = [join(path_logs, f) for f in listdir(path_logs) if
                    isfile(join(path_logs, f)) and f.startswith('Journal.')]
    if not list_of_logs:
        return None
    latest_log = max(list_of_logs, key=getmtime)
    return latest_log


logger.info('get_latest_log=' + str(get_latest_log(PATH_LOG_FILES)))


# Extract ship info from log
def ship():
    """Returns a 'status' dict containing relevant game status information (state, fuel, ...)"""
    latest_log = get_latest_log(PATH_LOG_FILES)
    ship_status = {
        'time': (datetime.now() - datetime.fromtimestamp(getmtime(latest_log))).seconds,
        'status': None,
        'type': None,
        'location': None,
        'star_class': None,
        'target': None,
        'fuel_capacity': None,
        'fuel_level': None,
        'fuel_percent': None,
        'is_scooping': False,
        'sys_fully_scanned': False
    }
    # Read log line by line and parse data
    with open(latest_log, encoding="utf-8") as f:
        for line in f:
            log = loads(line)

            # parse data
            try:
                # parse ship status
                log_event = log['event']

                if log_event == 'StartJump':
                    ship_status['status'] = str('starting_' + log['JumpType']).lower()
                    ship_status['sys_fully_scanned'] = False
                    if 'StarClass' in log:
                        ship_status['star_class'] = log['StarClass']

                elif log_event == 'SupercruiseEntry' or log_event == 'FSDJump':
                    ship_status['status'] = 'in_supercruise'

                elif log_event == 'SupercruiseExit' or log_event == 'DockingCancelled' or (
                        log_event == 'Music' and ship_status['status'] == 'in_undocking') or (
                        log_event == 'Location' and log['Docked'] is False):
                    ship_status['status'] = 'in_space'

                elif log_event == 'Undocked':
                    ship_status['status'] = 'in_space'

                elif log_event == 'DockingRequested':
                    ship_status['status'] = 'starting_docking'

                elif log_event == "Music" and log['MusicTrack'] == "DockingComputer":
                    if ship_status['status'] == 'starting_undocking':
                        ship_status['status'] = 'in_undocking'
                    elif ship_status['status'] == 'starting_docking':
                        ship_status['status'] = 'in_docking'

                elif log_event == 'Docked':
                    ship_status['status'] = 'in_station'

                elif log_event == 'FSSAllBodiesFound':
                    ship_status['sys_fully_scanned'] = True

                # parse ship type
                if log_event == 'LoadGame' or log_event == 'Loadout':
                    ship_status['type'] = log['Ship']

                # parse fuel
                if 'FuelLevel' in log and ship_status['type'] != 'TestBuggy':
                    ship_status['fuel_level'] = log['FuelLevel']
                if 'FuelCapacity' in log and ship_status['type'] != 'TestBuggy':
                    if type(log['FuelCapacity']) == float:
                        ship_status['fuel_capacity'] = log['FuelCapacity']
                    else:
                        ship_status['fuel_capacity'] = log['FuelCapacity']['Main']

                if log_event == 'FuelScoop' and 'Total' in log:
                    ship_status['fuel_level'] = log['Total']
                if ship_status['fuel_level'] and ship_status['fuel_capacity']:
                    ship_status['fuel_percent'] = round(
                        (ship_status['fuel_level'] / ship_status['fuel_capacity']) * 100)
                else:
                    ship_status['fuel_percent'] = 10

                # parse scoop
                if log_event == 'FuelScoop' and ship_status['time'] < 10 and ship_status['fuel_percent'] < 100:
                    ship_status['is_scooping'] = True
                else:
                    ship_status['is_scooping'] = False

                # parse location
                if (log_event == 'Location' or log_event == 'FSDJump') and 'StarSystem' in log:
                    ship_status['location'] = log['StarSystem']

                # parse target
                if log_event == 'FSDTarget':
                    if log['Name'] == ship_status['location']:
                        ship_status['target'] = None
                    else:
                        ship_status['target'] = log['Name']
                elif log_event == 'FSDJump':
                    if ship_status['location'] == ship_status['target']:
                        ship_status['target'] = None

            # exceptions
            except Exception as trace:
                logger.exception("Exception occurred: {}".format(trace))
    #     logger.debug('ship='+str(ship))
    return ship_status


logger.debug('ship=' + str(ship()))


# Control ED with direct input

# Get latest keybinds file
def get_latest_keybinds(path_bindings=None):
    path_bindings = getOption('BindingsPath')
    if not path_bindings:
        appdata = environ.get('LOCALAPPDATA')
        if appdata:
            path_bindings = appdata + "/Frontier Developments/Elite Dangerous/Options/Bindings"
            from settings_api import setOption
            setOption('BindingsPath', path_bindings)
        else:
            raise FileNotFoundError("Bindings path not found, define it in configs")
    list_of_bindings = [join(path_bindings, f) for f in listdir(path_bindings) if
                        (isfile(join(path_bindings, f)) and join(path_bindings, f).endswith("binds"))]
    if not list_of_bindings:
        return None
    latest_bindings = max(list_of_bindings, key=getmtime)
    return latest_bindings


logger.info("get_latest_keybinds=" + str(get_latest_keybinds()))

# Extract necessary keys
keys_to_obtain = [
    'YawLeftButton',
    'YawRightButton',
    'RollLeftButton',
    'RollRightButton',
    'PitchUpButton',
    'PitchDownButton',
    'SetSpeedZero',
    'SetSpeed100',
    'HyperSuperCombination',
    'UIFocus',
    'UI_Up',
    'UI_Down',
    'UI_Left',
    'UI_Right',
    'UI_Select',
    'UI_Back',
    'CycleNextPanel',
    'HeadLookReset',
    'MouseReset',
    'ExplorationFSSEnter',
    'ExplorationFSSQuit',
    'ExplorationFSSDiscoveryScan'
]


def get_bindings(keysToObtain=None):
    """Returns a dict struct with the direct input equivalent of the necessary elite keybindings"""
    if keysToObtain is None:
        keysToObtain = keys_to_obtain
    direct_input_keys = {}

    latest_bindings = get_latest_keybinds()
    bindings_tree = parse(latest_bindings)
    bindings_root = bindings_tree.getroot()

    for item in bindings_root:
        if item.tag in keysToObtain:
            new_key = None
            mod = None
            # Check primary
            if item[0].attrib['Device'].strip() == "Keyboard":
                new_key = item[0].attrib['Key']
                if len(item[0]) > 0:
                    mod = item[0][0].attrib['Key']
            # Check secondary (and prefer secondary)
            if item[1].attrib['Device'].strip() == "Keyboard":
                new_key = item[1].attrib['Key']
                if len(item[1]) > 0:
                    mod = item[1][0].attrib['Key']
            # Prepare final binding
            binding = None
            if new_key is not None:
                binding = {'pre_key': new_key}
                binding['key'] = EDKeyCodes[binding['pre_key']]
                if mod is not None:
                    binding['pre_mod'] = mod
                    binding['mod'] = EDKeyCodes[binding['pre_mod']]
            if binding is not None:
                direct_input_keys[item.tag] = binding
            # else:
            #    logger.warning("get_bindings: "+item.tag+" = does not have a valid keyboard keybind.")

    if len(list(direct_input_keys.keys())) < 1:
        return None
    else:
        return direct_input_keys


keys = get_bindings()
for key in keys_to_obtain:
    try:
        logger.info('get_bindings: ' + str(key) + ' = ' + str(keys[key]))
    except KeyError as e:
        logger.warning(str("get_bindings: " + key + " = does not have a valid keyboard keybind.").upper())


# Direct input function

# Send input
def send(key_to_send, hold=None, repeat=1, repeat_delay=None, state=None, cv_testing=False):
    if not cv_testing:
        global KEY_MOD_DELAY, KEY_DEFAULT_DELAY, KEY_REPEAT_DELAY

        if key_to_send is None:
            logger.warning('SEND=NONE !!!!!!!!')
            return

        logger.debug(
            'send=key:' + str(key_to_send) + ',hold:' + str(hold) + ',repeat:' + str(repeat) + ',repeat_delay:' + str(
                repeat_delay) + ',state:' + str(state))
        for i in range(repeat):

            if state is None or state == 1:
                if 'mod' in key_to_send:
                    PressKey(key_to_send['mod'])
                    sleep(KEY_MOD_DELAY)

                PressKey(key_to_send['key'])

            if state is None:
                if hold:
                    sleep(hold)
                else:
                    sleep(KEY_DEFAULT_DELAY)

            if state is None or state == 0:
                ReleaseKey(key_to_send['key'])

                if 'mod' in key_to_send:
                    sleep(KEY_MOD_DELAY)
                    ReleaseKey(key_to_send['mod'])

            if repeat_delay:
                sleep(repeat_delay)
            else:
                sleep(KEY_REPEAT_DELAY)


# Clear input
def clear_input(to_clear=None, cv_testing=False):
    if not cv_testing:
        logger.info('---- CLEAR INPUT ' + 183 * '-')
        send(to_clear['SetSpeedZero'])
        send(to_clear['MouseReset'])
        for key_to_clear in to_clear.keys():
            if key_to_clear in keys:
                send(to_clear[key_to_clear], state=0)
        logger.debug('clear_input')


# OpenCV

# Get screen
def get_screen(x_left, y_top, x_right, y_bot):
    screen = np.array(ImageGrab.grab(bbox=(round(x_left), round(y_top), round(x_right), round(y_bot))))
    screen = cv2.cvtColor(screen, cv2.COLOR_RGB2BGR)
    return screen


# HSV slider tool
def callback(x):
    return x


def hsv_slider(bandw=False):
    cv2.namedWindow('image')

    ilowH = 0
    ihighH = 179

    ilowS = 0
    ihighS = 255
    ilowV = 0
    ihighV = 255

    # create trackbars for color change
    cv2.createTrackbar('lowH', 'image', ilowH, 179, callback)
    cv2.createTrackbar('highH', 'image', ihighH, 179, callback)

    cv2.createTrackbar('lowS', 'image', ilowS, 255, callback)
    cv2.createTrackbar('highS', 'image', ihighS, 255, callback)

    cv2.createTrackbar('lowV', 'image', ilowV, 255, callback)
    cv2.createTrackbar('highV', 'image', ihighV, 255, callback)

    while True:
        # grab the frame
        frame = get_screen((5 / 16) * SCREEN_WIDTH, (5 / 8) * SCREEN_HEIGHT, (2 / 4) * SCREEN_WIDTH,
                           (15 / 16) * SCREEN_HEIGHT)
        if bandw:
            frame = equalize(frame)
            frame = cv2.cvtColor(frame, cv2.COLOR_GRAY2BGR)

        # get trackbar positions
        ilowH = cv2.getTrackbarPos('lowH', 'image')
        ihighH = cv2.getTrackbarPos('highH', 'image')
        ilowS = cv2.getTrackbarPos('lowS', 'image')
        ihighS = cv2.getTrackbarPos('highS', 'image')
        ilowV = cv2.getTrackbarPos('lowV', 'image')
        ihighV = cv2.getTrackbarPos('highV', 'image')

        hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
        lower_hsv = np.array([ilowH, ilowS, ilowV])
        higher_hsv = np.array([ihighH, ihighS, ihighV])
        mask = cv2.inRange(hsv, lower_hsv, higher_hsv)

        frame = cv2.bitwise_and(frame, frame, mask=mask)

        # show thresholded image
        cv2.imshow('image', frame)

        if cv2.waitKey(25) & 0xFF == ord('q'):
            cv2.destroyAllWindows()
            break


# Equalization
def equalize(image=None, testing=False):
    while True:
        if testing:
            img = get_screen((5 / 16) * SCREEN_WIDTH, (5 / 8) * SCREEN_HEIGHT, (2 / 4) * SCREEN_WIDTH,
                             (15 / 16) * SCREEN_HEIGHT)
        else:
            img = image.copy()
        # Load the image in greyscale
        img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        # create a CLAHE object (Arguments are optional).
        clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))
        img_out = clahe.apply(img_gray)
        if testing:
            cv2.imshow('Equalized', img_out)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return img_out


# Filter bright
def filter_bright(image=None, testing=False):
    while True:
        if testing:
            img = get_screen((5 / 16) * SCREEN_WIDTH, (5 / 8) * SCREEN_HEIGHT, (2 / 4) * SCREEN_WIDTH,
                             (15 / 16) * SCREEN_HEIGHT)
        else:
            img = image.copy()
        equalized = equalize(img)
        equalized = cv2.cvtColor(equalized, cv2.COLOR_GRAY2BGR)
        equalized = cv2.cvtColor(equalized, cv2.COLOR_BGR2HSV)
        filtered = cv2.inRange(equalized, np.array([0, 0, 215]), np.array([0, 0, 255]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Filter sun
def filter_sun(image=None, testing=False):
    while True:
        if testing:
            hsv = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                             (2 / 3) * SCREEN_HEIGHT)
        else:
            hsv = image.copy()
        # converting from BGR to HSV color space
        hsv = cv2.cvtColor(hsv, cv2.COLOR_BGR2HSV)
        # filter Elite UI orange
        filtered = cv2.inRange(hsv, np.array([0, 100, 240]), np.array([180, 255, 255]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Filter orange
def filter_orange(image=None, testing=False):
    while True:
        if testing:
            hsv = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                             (2 / 3) * SCREEN_HEIGHT)
        else:
            hsv = image.copy()
        # converting from BGR to HSV color space
        hsv = cv2.cvtColor(hsv, cv2.COLOR_BGR2HSV)
        # filter Elite UI orange
        filtered = cv2.inRange(hsv, np.array([0, 130, 123]), np.array([25, 235, 220]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Filter orange2
def filter_orange2(image=None, testing=False):
    while True:
        if testing:
            hsv = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                             (2 / 3) * SCREEN_HEIGHT)
        else:
            hsv = image.copy()
        # converting from BGR to HSV color space
        hsv = cv2.cvtColor(hsv, cv2.COLOR_BGR2HSV)
        # filter Elite UI orange
        filtered = cv2.inRange(hsv, np.array([15, 220, 220]), np.array([30, 255, 255]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Filter blue
def filter_blue(image=None, testing=False):
    while True:
        if testing:
            hsv = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                             (2 / 3) * SCREEN_HEIGHT)
        else:
            hsv = image.copy()
        # converting from BGR to HSV color space

        hsv = cv2.cvtColor(hsv, cv2.COLOR_BGR2HSV)
        # filter Elite UI orange
        filtered = cv2.inRange(hsv, np.array([0, 0, 200]), np.array([180, 100, 255]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Get sun
def sun_percent():
    screen = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT,
                        (2 / 3) * SCREEN_WIDTH, (2 / 3) * SCREEN_HEIGHT)
    filtered = filter_sun(screen)
    white = np.sum(filtered == 255)
    black = np.sum(filtered != 255)
    result = white / black
    return result * 100


# Get compass image
def get_compass_image(testing=False):
    if SCREEN_WIDTH == 3840:
        compass_template = cv2.imread(resource_path("templates/compass_3840.png"), cv2.IMREAD_GRAYSCALE)
    elif SCREEN_WIDTH == 2560:
        compass_template = cv2.imread(resource_path("templates/compass_2560.png"), cv2.IMREAD_GRAYSCALE)
    else:
        compass_template = cv2.imread(resource_path("templates/compass_1920.png"), cv2.IMREAD_GRAYSCALE)
    compass_width, compass_height = compass_template.shape[::-1]
    doubt = 10
    screen = get_screen((4 / 16) * SCREEN_WIDTH, (10 / 16) * SCREEN_HEIGHT, (8 / 16) * SCREEN_WIDTH,
                        (16 / 16) * SCREEN_HEIGHT)
    equalized = equalize(screen)
    match = cv2.matchTemplate(equalized, compass_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.2
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(match)
    pt = (doubt, doubt)
    if max_val >= threshold:
        pt = max_loc
    compass_image = screen[pt[1] - doubt: pt[1] + compass_height + doubt,
                    pt[0] - doubt: pt[0] + compass_width + doubt].copy()
    if testing:
        cv2.rectangle(screen, (pt[0] - doubt, pt[1] - doubt),
                      (pt[0] + (compass_width + doubt), pt[1] + (compass_height + doubt)), (0, 0, 255), 2)
        loc = np.where(match >= threshold)
        pts = tuple(zip(*loc[::-1]))
        match = cv2.cvtColor(match, cv2.COLOR_GRAY2RGB)
        for p in pts:
            cv2.circle(match, p, 1, (0, 0, 255), 1)
        cv2.circle(match, pt, 5, (0, 255, 0), 3)
        cv2.imshow('Compass Found', screen)  # This crashes linux machines
        # cv2.imshow('Compass Mask', equalized)
        cv2.imshow('Compass Match', match)
        cv2.waitKey(1)
    return compass_image, compass_width + (2 * doubt), compass_height + (2 * doubt)


# Get navpoint offset
same_last_count = 0
last_last = {'x': 1, 'y': 100}


def get_navpoint_offset(testing=False, last=None):
    global same_last_count, last_last
    if SCREEN_WIDTH == 3840:
        navpoint_template = cv2.imread(resource_path("templates/navpoint_3840.png"), cv2.IMREAD_GRAYSCALE)
    elif SCREEN_WIDTH == 2560:
        navpoint_template = cv2.imread(resource_path("templates/navpoint_2560.png"), cv2.IMREAD_GRAYSCALE)
    else:
        navpoint_template = cv2.imread(resource_path("templates/navpoint_1920.png"), cv2.IMREAD_GRAYSCALE)
    navpoint_width, navpoint_height = navpoint_template.shape[::-1]
    compass_image, compass_width, compass_height = get_compass_image()
    attempts = 0
    while compass_image.size == 0 and attempts < 10:
        compass_image, compass_width, compass_height = get_compass_image()
        attempts += 1
    if attempts > 1:
        logger.debug("Attempts at get_compass_image: " + str(attempts))
    filtered = filter_blue(compass_image)
    # filtered = filter_bright(compass_image)
    match = cv2.matchTemplate(filtered, navpoint_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.5
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(match)
    pt = (0, 0)
    if max_val >= threshold:
        pt = max_loc
    final_x = (pt[0] + ((1 / 2) * navpoint_width)) - ((1 / 2) * compass_width)
    final_y = ((1 / 2) * compass_height) - (pt[1] + ((1 / 2) * navpoint_height))
    if testing:
        cv2.rectangle(compass_image, pt, (pt[0] + navpoint_width, pt[1] + navpoint_height), (0, 0, 255), 2)
        loc = np.where(match >= threshold)
        pts = tuple(zip(*loc[::-1]))
        match = cv2.cvtColor(match, cv2.COLOR_GRAY2RGB)
        for p in pts:
            cv2.circle(match, p, 1, (0, 0, 255), 1)
        cv2.circle(match, pt, 5, (0, 255, 0), 3)
        cv2.imshow('Navpoint Found', compass_image)
        cv2.imshow('Navpoint Mask', filtered)
        cv2.imshow('Navpoint Match', match)
        cv2.waitKey(1)
    if pt == (0, 0):
        if last:
            if last == last_last:
                same_last_count = same_last_count + 1
            else:
                last_last = last
                same_last_count = 0
            if same_last_count > 5:
                same_last_count = 0
                if rand.random() < .9:
                    result = {'x': 1, 'y': 100}
                else:
                    result = {'x': 100, 'y': 1}
            else:
                result = last
        else:
            result = None
    else:
        result = {'x': final_x, 'y': final_y}
    logger.debug('get_navpoint_offset=' + str(result))
    return result


# Get destination offset
def get_destination_offset(testing=False):
    if SCREEN_WIDTH == 3840:
        destination_template = cv2.imread(resource_path("templates/destination_3840.png"), cv2.IMREAD_GRAYSCALE)
    elif SCREEN_WIDTH == 2560:
        destination_template = cv2.imread(resource_path("templates/destination_2560.png"), cv2.IMREAD_GRAYSCALE)
    else:
        destination_template = cv2.imread(resource_path("templates/destination_1920.png"), cv2.IMREAD_GRAYSCALE)
    destination_width, destination_height = destination_template.shape[::-1]
    width = (1 / 3) * SCREEN_WIDTH
    height = (1 / 3) * SCREEN_HEIGHT
    screen = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                        (2 / 3) * SCREEN_HEIGHT)
    filtered = filter_orange2(screen)
    match = cv2.matchTemplate(filtered, destination_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.2
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(match)
    pt = (0, 0)
    if max_val >= threshold:
        pt = max_loc
    final_x = (pt[0] + ((1 / 2) * destination_width)) - ((1 / 2) * width)
    final_y = ((1 / 2) * height) - (pt[1] + ((1 / 2) * destination_height))
    pt2 = (int(final_x), int(final_y))
    print(pt2)
    if testing:
        cv2.rectangle(screen, pt, (pt[0] + destination_width, pt[1] + destination_height), (0, 0, 255), 2)
        loc = np.where(match >= threshold)
        pts = tuple(zip(*loc[::-1]))
        match = cv2.cvtColor(match, cv2.COLOR_GRAY2RGB)
        for p in pts:
            cv2.circle(match, p, 1, (0, 0, 255), 1)
        cv2.circle(match, pt, 5, (0, 255, 0), 3)
        cv2.circle(screen, pt, 5, (0, 255, 0), 3)
        cv2.imshow('Destination Found', screen)
        cv2.imshow('Destination Mask', filtered)
        cv2.imshow('Destination Match', match)
        cv2.waitKey(1)
    if pt == (0, 0):
        result = None
    else:
        result = {'x': final_x, 'y': final_y}
    logger.debug('get_destination_offset=' + str(result))
    return result


# Autopilot routines

# Undock
def undock():
    logger.debug('undock')
    if ship()['status'] != "in_station":
        logger.error('undock=err1')
        raise Exception('undock error 1')
    send(keys['UI_Back'], repeat=10)
    send(keys['HeadLookReset'])
    send(keys['UI_Down'], hold=3)
    send(keys['UI_Select'])
    sleep(1)
    if not (ship()['status'] == "starting_undock" or ship()['status'] == "in_undock"):
        logger.error('undock=err2')
        raise Exception("undock error 2")
    send(keys['HeadLookReset'])
    send(keys['SetSpeedZero'], repeat=2)
    wait = 120
    for i in range(wait):
        sleep(1)
        if i > wait - 1:
            logger.error('undock=err3')
            raise Exception('undock error 3')
        if ship()['status'] == "in_space":
            break
    logger.debug('undock=complete')
    return True


# Dock
def dock():
    logger.debug('dock')
    if ship()['status'] != "in_space":
        logger.error('dock=err1')
        raise Exception('dock error 1')
    tries = 3
    for i in range(tries):
        send(keys['UI_Back'], repeat=10)
        send(keys['HeadLookReset'])
        send(keys['UIFocus'], state=1)
        send(keys['UI_Left'])
        send(keys['UIFocus'], state=0)
        send(keys['CycleNextPanel'], repeat=2)
        send(keys['UI_Up'], hold=3)
        send(keys['UI_Right'])
        send(keys['UI_Select'])
        sleep(1)
        if ship()['status'] == "starting_dock" or ship()['status'] == "in_dock":
            break
        if i > tries - 1:
            logger.error('dock=err2')
            raise Exception("dock error 2")
    send(keys['UI_Back'])
    send(keys['HeadLookReset'])
    send(keys['SetSpeedZero'], repeat=2)
    wait = 120
    for i in range(wait):
        sleep(1)
        if i > wait - 1:
            logger.error('dock=err3')
            raise Exception('dock error 3')
        if ship()['status'] == "in_station":
            break
    send(keys['UI_Up'], hold=3)
    send(keys['UI_Down'])
    send(keys['UI_Select'])
    logger.debug('dock=complete')
    return True


# Align
def x_angle(point=None):
    if not point or point['x'] == 0:
        return None
    result = math.degrees(math.atan(point['y'] / point['x']))
    if point['x'] > 0:
        return +90 - result
    else:
        return -90 - result


def align():
    logger.info('Starting Alignment')
    if not (ship()['status'] == 'in_supercruise' or ship()['status'] == 'in_space'):
        logger.error('align=error: ship not in supercruise')
        raise Exception('align error: ship not in supercruise')

    logger.info('Alignment Step: Avoid sun')
    while sun_percent() > 5:
        send(keys['PitchUpButton'], state=1)
    send(keys['PitchUpButton'], state=0)

    logger.debug('align=speed 100')
    send(keys['SetSpeed100'])

    logger.info('Alignment Step: Find navpoint')
    off = get_navpoint_offset()
    while not off:
        send(keys['PitchUpButton'], state=1)
        off = get_navpoint_offset()
    send(keys['PitchUpButton'], state=0)

    logger.info('Alignment Step: Crude align')
    close = 3
    close_a = 18
    hold_pitch = 0.350
    hold_roll = 0.170
    ang = x_angle(off)
    while (off['x'] > close and ang > close_a) or (off['x'] < -close and ang < -close_a) or \
            (off['y'] > close) or (off['y'] < -close):

        while (off['x'] > close and ang > close_a) or (off['x'] < -close and ang < -close_a):

            if off['x'] > close and ang > close:
                send(keys['RollRightButton'], hold=hold_roll)
            if off['x'] < -close and ang < -close:
                send(keys['RollLeftButton'], hold=hold_roll)

            if ship()['status'] == 'starting_hyperspace':
                return
            off = get_navpoint_offset(last=off)
            ang = x_angle(off)

        while (off['y'] > close) or (off['y'] < -close):

            if off['y'] > close:
                send(keys['PitchUpButton'], hold=hold_pitch)
            if off['y'] < -close:
                send(keys['PitchDownButton'], hold=hold_pitch)

            if ship()['status'] == 'starting_hyperspace':
                return
            off = get_navpoint_offset(last=off)

        off = get_navpoint_offset(last=off)
        ang = x_angle(off)

    logger.info('Alignment Step: Fine align')
    sleep(0.5)
    close = 50
    hold_pitch = 0.200
    hold_yaw = 0.400
    for i in range(5):
        new = get_destination_offset()
        if new:
            off = new
            break
        sleep(0.25)
    if not off:
        return
    while (off['x'] > close) or (off['x'] < -close) or (off['y'] > close) or (off['y'] < -close):

        if off['x'] > close:
            send(keys['YawRightButton'], hold=hold_yaw)
        if off['x'] < -close:
            send(keys['YawLeftButton'], hold=hold_yaw)
        if off['y'] > close:
            send(keys['PitchUpButton'], hold=hold_pitch)
        if off['y'] < -close:
            send(keys['PitchDownButton'], hold=hold_pitch)

        if ship()['status'] == 'starting_hyperspace':
            return

        for i in range(5):
            new = get_destination_offset()
            if new:
                off = new
                break
            sleep(0.25)
        if not off:
            return

    logger.info('Alignment Complete')


# Jump
def jump():
    logger.info('Starting Jump')
    tries = 5
    for i in range(tries):
        if ship()['status'] == 'starting_hyperspace':
            logger.info('Jump in Progress')
            while ship()['status'] != 'in_supercruise':
                sleep(1)
            logger.debug('jump=speed 0')
            send(keys['SetSpeedZero'])
            logger.info('Jump Complete')
            return True
        logger.debug('jump=try:' + str(i))
        if not (ship()['status'] == 'in_supercruise' or ship()['status'] == 'in_space'):
            logger.error('jump_err=status:' + ship()['status'])
            raise Exception('not ready to jump')
        sleep(0.5)
        send(keys['SetSpeed100'])
        logger.info('Charging FSD')
        send(keys['HyperSuperCombination'], hold=1)
        sleep(16)
        if ship()['status'] != 'starting_hyperspace':
            logger.info('Trajectory Misaligned. Jump Failed. Retrying Alignment')
            align()

    logger.error('jump=err2')
    raise Exception("jump failure")


# Refuel
def refuel(refuel_threshold=50):
    logger.debug('refuel')
    scoopable_stars = ['F', 'O', 'G', 'K', 'B', 'A', 'M']
    if ship()['status'] != 'in_supercruise':
        logger.error('refuel=error: Ship not in supercruise')
        return False

    if ship()['fuel_percent'] < refuel_threshold and ship()['star_class'] in scoopable_stars:
        logger.info('Starting Refuel')
        sleep(2)
        send(keys['SetSpeed100'])
        sleep(4)
        logger.info('Refuel in Progress')
        send(keys['SetSpeedZero'], repeat=3)
        while not ship()['fuel_percent'] == 100:
            sleep(1)
        logger.info('Refuel Complete')
        return True
    elif ship()['fuel_percent'] >= refuel_threshold:
        logger.info('Refuel not Needed')
        return False
    elif ship()['star_class'] not in scoopable_stars:
        logger.info('Refuel Needed, Unsuitable Star')
        return False
    else:
        return False


def scanFSS(aFFS):
    align()
    if aFFS and not ship()['sys_fully_scanned']:
        send(keys['SetSpeed100'])  # The farther away we are, the easier the system is to scan
        sleep(15)  # and there's far less chance of obstructed frequencies
    logger.info("Discovery scanning")
    send(keys['SetSpeedZero'])
    send(keys['ExplorationFSSEnter'])
    sleep(1)
    send(keys['ExplorationFSSDiscoveryScan'], 3.5)
    if aFFS:
        logger.info("Waiting for FSS to complete")
        while not ship()['sys_fully_scanned']:
            sleep(5)  # TODO: Actually scan around and such

    send(keys['ExplorationFSSQuit'])
    logger.info("Scan complete")


# Position
def position(refueled_multiplier=1):
    logger.debug('position')
    send(keys['PitchUpButton'], state=1)
    sleep(5)
    send(keys['PitchUpButton'], state=0)
    send(keys['SetSpeed100'])
    send(keys['PitchUpButton'], state=1)
    while sun_percent() > 3:
        sleep(1)
    sleep(5)
    send(keys['PitchUpButton'], state=0)
    sleep(5 * refueled_multiplier)
    logger.debug('position=complete')
    return True


# Autopilot main

# status reference
#
# 'in-station'
# 
# 'in-supercruise'
# 
# 'in-space'
# 
# 'starting-undocking'
# 
# 'in-undocking'
# 
# 'starting-docking'
# 
# 'in-docking'


def autopilot():
    # logger.info('\n'+200*'-'+'\n'+'---- AUTOPILOT START '+179*'-'+'\n'+200*'-')
    logger.info('---- AUTOPILOT START ' + 179 * '-')
    logger.debug('get_latest_log=' + str(get_latest_log(PATH_LOG_FILES)))
    logger.debug('ship=' + str(ship()))
    while ship()['target']:
        if ship()['status'] == 'in_space' or ship()['status'] == 'in_supercruise':
            logger.info('---- AUTOPILOT ALIGN ' + 179 * '-')
            align()

            if getOption('DiscoveryScan'):
                logger.info('---- AUTOPILOT SCAN ' + 180 * '-')
                scanFSS(getOption('AutoFSS'))

            logger.info('---- AUTOPILOT JUMP ' + 180 * '-')
            jump()

            logger.info('---- AUTOPILOT REFUEL ' + 178 * '-')
            refueled = refuel(int(getOption('RefuelThreshold')))
            if refueled:
                position(refueled_multiplier=4)
            else:
                position(refueled_multiplier=1)

    send(keys['SetSpeedZero'])
    logger.info('---- AUTOPILOT END ' + 181 * '-')
    from dev_tray import stop_action
    stop_action()

#!/usr/bin/env python
# coding: utf-8

# References
# Useful docs / articles / etc
#   
#   1 - [A Python wrapper around AHK](https://pypi.org/project/ahk/)
# 
#   2 - [OpenCV on Wheels](https://pypi.org/project/opencv-python/)
# 
#   3 - [Autopilot for Elite Dangerous using OpenCV and thoughts on CV enabled bots in visual-to-keyboard loop]
#   (https://networkgeekstuff.com/projects/autopilot-for-elite-dangerous-using-opencv-and-thoughts-on-cv-enabled-bots-in-visual-to-keyboard-loop/)
#   
#   4 - [Using PyInstaller to Easily Distribute Python Applications](https://realpython.com/pyinstaller-python/)
#   
#   5 - [Direct Input to a Game - Python Plays GTA V]
#   (https://pythonprogramming.net/direct-input-game-python-plays-gta-v/)
#   
#   6 - [Cross-platform GUI automation for human beings](https://pyautogui.readthedocs.io/en/latest/index.html)

# Imports
import datetime
import math
import random as rand
import sys
from datetime import datetime
from json import loads
from os import environ, listdir
from os.path import join, isfile, getmtime, abspath
from time import sleep
from xml.etree.ElementTree import parse

import cv2  # see reference 2
import numpy as np
from PIL import ImageGrab
from pyautogui import size  # see reference 6

from logger import logger
from settings_api import getOption
from src.directinput import EDKeyCodes, PressKey, ReleaseKey  # see reference 5


def resource_path(relative_path):
    """ Get absolute path to resource, works for dev and for PyInstaller """
    try:
        # PyInstaller creates a temp folder and stores path in _MEIPASS
        base_path = sys._MEIPASS
    except AttributeError:
        base_path = abspath(".")

    return join(base_path, relative_path)


# logger.info('\n'+200*'-'+'\n'+'---- AUTOPILOT DATA '+180*'-'+'\n'+200*'-')
logger.info('---- AUTOPILOT DATA ' + 180 * '-')

# Constants
RELEASE = 'v19.05.15-binary-18'
PATH_LOG_FILES = None
PATH_KEYBINDINGS = None
KEY_MOD_DELAY = 0.010
KEY_DEFAULT_DELAY = 0.200
KEY_REPEAT_DELAY = 0.100
FUNCTION_DEFAULT_DELAY = 0.500
SCREEN_WIDTH, SCREEN_HEIGHT = size()

logger.info('RELEASE=' + str(RELEASE))
logger.info('PATH_LOG_FILES=' + str(PATH_LOG_FILES))
logger.info('PATH_KEYBINDINGS=' + str(PATH_KEYBINDINGS))
logger.info('KEY_MOD_DELAY=' + str(KEY_MOD_DELAY))
logger.info('KEY_DEFAULT_DELAY=' + str(KEY_DEFAULT_DELAY))
logger.info('KEY_REPEAT_DELAY=' + str(KEY_REPEAT_DELAY))
logger.info('FUNCTION_DEFAULT_DELAY=' + str(FUNCTION_DEFAULT_DELAY))
logger.info('SCREEN_WIDTH=' + str(SCREEN_WIDTH))
logger.info('SCREEN_HEIGHT=' + str(SCREEN_HEIGHT))


# Read ED logs

# Get latest log file
def get_latest_log(path_logs=None):
    """Returns the full path of the latest (most recent) elite log file (journal) from specified path"""
    path_logs = getOption('JournalPath')
    if not path_logs:
        user_path = environ.get('USERPROFILE')
        if user_path:
            path_logs = user_path + "/Saved Games/Frontier Developments/Elite Dangerous"
            from settings_api import setOption
            setOption('JournalPath', path_logs)
        else:
            raise FileNotFoundError("Journal path not found, define it in configs")
    list_of_logs = [join(path_logs, f) for f in listdir(path_logs) if
                    isfile(join(path_logs, f)) and f.startswith('Journal.')]
    if not list_of_logs:
        return None
    latest_log = max(list_of_logs, key=getmtime)
    return latest_log


logger.info('get_latest_log=' + str(get_latest_log(PATH_LOG_FILES)))


# Extract ship info from log
def ship():
    """Returns a 'status' dict containing relevant game status information (state, fuel, ...)"""
    latest_log = get_latest_log(PATH_LOG_FILES)
    ship_status = {
        'time': (datetime.now() - datetime.fromtimestamp(getmtime(latest_log))).seconds,
        'status': None,
        'type': None,
        'location': None,
        'star_class': None,
        'target': None,
        'fuel_capacity': None,
        'fuel_level': None,
        'fuel_percent': None,
        'is_scooping': False,
        'sys_fully_scanned': False
    }
    # Read log line by line and parse data
    with open(latest_log, encoding="utf-8") as f:
        for line in f:
            log = loads(line)

            # parse data
            try:
                # parse ship status
                log_event = log['event']

                if log_event == 'StartJump':
                    ship_status['status'] = str('starting_' + log['JumpType']).lower()
                    ship_status['sys_fully_scanned'] = False
                    if 'StarClass' in log:
                        ship_status['star_class'] = log['StarClass']

                elif log_event == 'SupercruiseEntry' or log_event == 'FSDJump':
                    ship_status['status'] = 'in_supercruise'

                elif log_event == 'SupercruiseExit' or log_event == 'DockingCancelled' or (
                        log_event == 'Music' and ship_status['status'] == 'in_undocking') or (
                        log_event == 'Location' and log['Docked'] is False):
                    ship_status['status'] = 'in_space'

                elif log_event == 'Undocked':
                    ship_status['status'] = 'in_space'

                elif log_event == 'DockingRequested':
                    ship_status['status'] = 'starting_docking'

                elif log_event == "Music" and log['MusicTrack'] == "DockingComputer":
                    if ship_status['status'] == 'starting_undocking':
                        ship_status['status'] = 'in_undocking'
                    elif ship_status['status'] == 'starting_docking':
                        ship_status['status'] = 'in_docking'

                elif log_event == 'Docked':
                    ship_status['status'] = 'in_station'

                elif log_event == 'FSSAllBodiesFound':
                    ship_status['sys_fully_scanned'] = True

                # parse ship type
                if log_event == 'LoadGame' or log_event == 'Loadout':
                    ship_status['type'] = log['Ship']

                # parse fuel
                if 'FuelLevel' in log and ship_status['type'] != 'TestBuggy':
                    ship_status['fuel_level'] = log['FuelLevel']
                if 'FuelCapacity' in log and ship_status['type'] != 'TestBuggy':
                    if type(log['FuelCapacity']) == float:
                        ship_status['fuel_capacity'] = log['FuelCapacity']
                    else:
                        ship_status['fuel_capacity'] = log['FuelCapacity']['Main']

                if log_event == 'FuelScoop' and 'Total' in log:
                    ship_status['fuel_level'] = log['Total']
                if ship_status['fuel_level'] and ship_status['fuel_capacity']:
                    ship_status['fuel_percent'] = round(
                        (ship_status['fuel_level'] / ship_status['fuel_capacity']) * 100)
                else:
                    ship_status['fuel_percent'] = 10

                # parse scoop
                if log_event == 'FuelScoop' and ship_status['time'] < 10 and ship_status['fuel_percent'] < 100:
                    ship_status['is_scooping'] = True
                else:
                    ship_status['is_scooping'] = False

                # parse location
                if (log_event == 'Location' or log_event == 'FSDJump') and 'StarSystem' in log:
                    ship_status['location'] = log['StarSystem']

                # parse target
                if log_event == 'FSDTarget':
                    if log['Name'] == ship_status['location']:
                        ship_status['target'] = None
                    else:
                        ship_status['target'] = log['Name']
                elif log_event == 'FSDJump':
                    if ship_status['location'] == ship_status['target']:
                        ship_status['target'] = None

            # exceptions
            except Exception as trace:
                logger.exception("Exception occurred: {}".format(trace))
    #     logger.debug('ship='+str(ship))
    return ship_status


logger.debug('ship=' + str(ship()))


# Control ED with direct input

# Get latest keybinds file
def get_latest_keybinds(path_bindings=None):
    path_bindings = getOption('BindingsPath')
    if not path_bindings:
        appdata = environ.get('LOCALAPPDATA')
        if appdata:
            path_bindings = appdata + "/Frontier Developments/Elite Dangerous/Options/Bindings"
            from settings_api import setOption
            setOption('BindingsPath', path_bindings)
        else:
            raise FileNotFoundError("Bindings path not found, define it in configs")
    list_of_bindings = [join(path_bindings, f) for f in listdir(path_bindings) if
                        (isfile(join(path_bindings, f)) and join(path_bindings, f).endswith("binds"))]
    if not list_of_bindings:
        return None
    latest_bindings = max(list_of_bindings, key=getmtime)
    return latest_bindings


logger.info("get_latest_keybinds=" + str(get_latest_keybinds()))

# Extract necessary keys
keys_to_obtain = [
    'YawLeftButton',
    'YawRightButton',
    'RollLeftButton',
    'RollRightButton',
    'PitchUpButton',
    'PitchDownButton',
    'SetSpeedZero',
    'SetSpeed100',
    'HyperSuperCombination',
    'UIFocus',
    'UI_Up',
    'UI_Down',
    'UI_Left',
    'UI_Right',
    'UI_Select',
    'UI_Back',
    'CycleNextPanel',
    'HeadLookReset',
    'MouseReset',
    'ExplorationFSSEnter',
    'ExplorationFSSQuit',
    'ExplorationFSSDiscoveryScan'
]


def get_bindings(keysToObtain=None):
    """Returns a dict struct with the direct input equivalent of the necessary elite keybindings"""
    if keysToObtain is None:
        keysToObtain = keys_to_obtain
    direct_input_keys = {}

    latest_bindings = get_latest_keybinds()
    bindings_tree = parse(latest_bindings)
    bindings_root = bindings_tree.getroot()

    for item in bindings_root:
        if item.tag in keysToObtain:
            new_key = None
            mod = None
            # Check primary
            if item[0].attrib['Device'].strip() == "Keyboard":
                new_key = item[0].attrib['Key']
                if len(item[0]) > 0:
                    mod = item[0][0].attrib['Key']
            # Check secondary (and prefer secondary)
            if item[1].attrib['Device'].strip() == "Keyboard":
                new_key = item[1].attrib['Key']
                if len(item[1]) > 0:
                    mod = item[1][0].attrib['Key']
            # Prepare final binding
            binding = None
            if new_key is not None:
                binding = {'pre_key': new_key}
                binding['key'] = EDKeyCodes[binding['pre_key']]
                if mod is not None:
                    binding['pre_mod'] = mod
                    binding['mod'] = EDKeyCodes[binding['pre_mod']]
            if binding is not None:
                direct_input_keys[item.tag] = binding
            # else:
            #    logger.warning("get_bindings: "+item.tag+" = does not have a valid keyboard keybind.")

    if len(list(direct_input_keys.keys())) < 1:
        return None
    else:
        return direct_input_keys


keys = get_bindings()
for key in keys_to_obtain:
    try:
        logger.info('get_bindings: ' + str(key) + ' = ' + str(keys[key]))
    except KeyError as e:
        logger.warning(str("get_bindings: " + key + " = does not have a valid keyboard keybind.").upper())


# Direct input function

# Send input
def send(key_to_send, hold=None, repeat=1, repeat_delay=None, state=None, cv_testing=False):
    if not cv_testing:
        global KEY_MOD_DELAY, KEY_DEFAULT_DELAY, KEY_REPEAT_DELAY

        if key_to_send is None:
            logger.warning('SEND=NONE !!!!!!!!')
            return

        logger.debug(
            'send=key:' + str(key_to_send) + ',hold:' + str(hold) + ',repeat:' + str(repeat) + ',repeat_delay:' + str(
                repeat_delay) + ',state:' + str(state))
        for i in range(repeat):

            if state is None or state == 1:
                if 'mod' in key_to_send:
                    PressKey(key_to_send['mod'])
                    sleep(KEY_MOD_DELAY)

                PressKey(key_to_send['key'])

            if state is None:
                if hold:
                    sleep(hold)
                else:
                    sleep(KEY_DEFAULT_DELAY)

            if state is None or state == 0:
                ReleaseKey(key_to_send['key'])

                if 'mod' in key_to_send:
                    sleep(KEY_MOD_DELAY)
                    ReleaseKey(key_to_send['mod'])

            if repeat_delay:
                sleep(repeat_delay)
            else:
                sleep(KEY_REPEAT_DELAY)


# Clear input
def clear_input(to_clear=None, cv_testing=False):
    if not cv_testing:
        logger.info('---- CLEAR INPUT ' + 183 * '-')
        send(to_clear['SetSpeedZero'])
        send(to_clear['MouseReset'])
        for key_to_clear in to_clear.keys():
            if key_to_clear in keys:
                send(to_clear[key_to_clear], state=0)
        logger.debug('clear_input')


# OpenCV

# Get screen
def get_screen(x_left, y_top, x_right, y_bot):
    screen = np.array(ImageGrab.grab(bbox=(round(x_left), round(y_top), round(x_right), round(y_bot))))
    screen = cv2.cvtColor(screen, cv2.COLOR_RGB2BGR)
    return screen


# HSV slider tool
def callback(x):
    return x


def hsv_slider(bandw=False):
    cv2.namedWindow('image')

    ilowH = 0
    ihighH = 179

    ilowS = 0
    ihighS = 255
    ilowV = 0
    ihighV = 255

    # create trackbars for color change
    cv2.createTrackbar('lowH', 'image', ilowH, 179, callback)
    cv2.createTrackbar('highH', 'image', ihighH, 179, callback)

    cv2.createTrackbar('lowS', 'image', ilowS, 255, callback)
    cv2.createTrackbar('highS', 'image', ihighS, 255, callback)

    cv2.createTrackbar('lowV', 'image', ilowV, 255, callback)
    cv2.createTrackbar('highV', 'image', ihighV, 255, callback)

    while True:
        # grab the frame
        frame = get_screen((5 / 16) * SCREEN_WIDTH, (5 / 8) * SCREEN_HEIGHT, (2 / 4) * SCREEN_WIDTH,
                           (15 / 16) * SCREEN_HEIGHT)
        if bandw:
            frame = equalize(frame)
            frame = cv2.cvtColor(frame, cv2.COLOR_GRAY2BGR)

        # get trackbar positions
        ilowH = cv2.getTrackbarPos('lowH', 'image')
        ihighH = cv2.getTrackbarPos('highH', 'image')
        ilowS = cv2.getTrackbarPos('lowS', 'image')
        ihighS = cv2.getTrackbarPos('highS', 'image')
        ilowV = cv2.getTrackbarPos('lowV', 'image')
        ihighV = cv2.getTrackbarPos('highV', 'image')

        hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
        lower_hsv = np.array([ilowH, ilowS, ilowV])
        higher_hsv = np.array([ihighH, ihighS, ihighV])
        mask = cv2.inRange(hsv, lower_hsv, higher_hsv)

        frame = cv2.bitwise_and(frame, frame, mask=mask)

        # show thresholded image
        cv2.imshow('image', frame)

        if cv2.waitKey(25) & 0xFF == ord('q'):
            cv2.destroyAllWindows()
            break


# Equalization
def equalize(image=None, testing=False):
    while True:
        if testing:
            img = get_screen((5 / 16) * SCREEN_WIDTH, (5 / 8) * SCREEN_HEIGHT, (2 / 4) * SCREEN_WIDTH,
                             (15 / 16) * SCREEN_HEIGHT)
        else:
            img = image.copy()
        # Load the image in greyscale
        img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        # create a CLAHE object (Arguments are optional).
        clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))
        img_out = clahe.apply(img_gray)
        if testing:
            cv2.imshow('Equalized', img_out)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return img_out


# Filter bright
def filter_bright(image=None, testing=False):
    while True:
        if testing:
            img = get_screen((5 / 16) * SCREEN_WIDTH, (5 / 8) * SCREEN_HEIGHT, (2 / 4) * SCREEN_WIDTH,
                             (15 / 16) * SCREEN_HEIGHT)
        else:
            img = image.copy()
        equalized = equalize(img)
        equalized = cv2.cvtColor(equalized, cv2.COLOR_GRAY2BGR)
        equalized = cv2.cvtColor(equalized, cv2.COLOR_BGR2HSV)
        filtered = cv2.inRange(equalized, np.array([0, 0, 215]), np.array([0, 0, 255]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Filter sun
def filter_sun(image=None, testing=False):
    while True:
        if testing:
            hsv = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                             (2 / 3) * SCREEN_HEIGHT)
        else:
            hsv = image.copy()
        # converting from BGR to HSV color space
        hsv = cv2.cvtColor(hsv, cv2.COLOR_BGR2HSV)
        # filter Elite UI orange
        filtered = cv2.inRange(hsv, np.array([0, 100, 240]), np.array([180, 255, 255]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Filter orange
def filter_orange(image=None, testing=False):
    while True:
        if testing:
            hsv = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                             (2 / 3) * SCREEN_HEIGHT)
        else:
            hsv = image.copy()
        # converting from BGR to HSV color space
        hsv = cv2.cvtColor(hsv, cv2.COLOR_BGR2HSV)
        # filter Elite UI orange
        filtered = cv2.inRange(hsv, np.array([0, 130, 123]), np.array([25, 235, 220]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Filter orange2
def filter_orange2(image=None, testing=False):
    while True:
        if testing:
            hsv = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                             (2 / 3) * SCREEN_HEIGHT)
        else:
            hsv = image.copy()
        # converting from BGR to HSV color space
        hsv = cv2.cvtColor(hsv, cv2.COLOR_BGR2HSV)
        # filter Elite UI orange
        filtered = cv2.inRange(hsv, np.array([15, 220, 220]), np.array([30, 255, 255]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Filter blue
def filter_blue(image=None, testing=False):
    while True:
        if testing:
            hsv = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                             (2 / 3) * SCREEN_HEIGHT)
        else:
            hsv = image.copy()
        # converting from BGR to HSV color space

        hsv = cv2.cvtColor(hsv, cv2.COLOR_BGR2HSV)
        # filter Elite UI orange
        filtered = cv2.inRange(hsv, np.array([0, 0, 200]), np.array([180, 100, 255]))
        if testing:
            cv2.imshow('Filtered', filtered)
            if cv2.waitKey(25) & 0xFF == ord('q'):
                cv2.destroyAllWindows()
                break
        else:
            break
    return filtered


# Get sun
def sun_percent():
    screen = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT,
                        (2 / 3) * SCREEN_WIDTH, (2 / 3) * SCREEN_HEIGHT)
    filtered = filter_sun(screen)
    white = np.sum(filtered == 255)
    black = np.sum(filtered != 255)
    result = white / black
    return result * 100


# Get compass image
def get_compass_image(testing=False):
    if SCREEN_WIDTH == 3840:
        compass_template = cv2.imread(resource_path("templates/compass_3840.png"), cv2.IMREAD_GRAYSCALE)
    elif SCREEN_WIDTH == 2560:
        compass_template = cv2.imread(resource_path("templates/compass_2560.png"), cv2.IMREAD_GRAYSCALE)
    else:
        compass_template = cv2.imread(resource_path("templates/compass_1920.png"), cv2.IMREAD_GRAYSCALE)
    compass_width, compass_height = compass_template.shape[::-1]
    doubt = 10
    screen = get_screen((4 / 16) * SCREEN_WIDTH, (10 / 16) * SCREEN_HEIGHT, (8 / 16) * SCREEN_WIDTH,
                        (16 / 16) * SCREEN_HEIGHT)
    equalized = equalize(screen)
    match = cv2.matchTemplate(equalized, compass_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.2
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(match)
    pt = (doubt, doubt)
    if max_val >= threshold:
        pt = max_loc
    compass_image = screen[pt[1] - doubt: pt[1] + compass_height + doubt,
                    pt[0] - doubt: pt[0] + compass_width + doubt].copy()
    if testing:
        cv2.rectangle(screen, (pt[0] - doubt, pt[1] - doubt),
                      (pt[0] + (compass_width + doubt), pt[1] + (compass_height + doubt)), (0, 0, 255), 2)
        loc = np.where(match >= threshold)
        pts = tuple(zip(*loc[::-1]))
        match = cv2.cvtColor(match, cv2.COLOR_GRAY2RGB)
        for p in pts:
            cv2.circle(match, p, 1, (0, 0, 255), 1)
        cv2.circle(match, pt, 5, (0, 255, 0), 3)
        cv2.imshow('Compass Found', screen)  # This crashes linux machines
        # cv2.imshow('Compass Mask', equalized)
        cv2.imshow('Compass Match', match)
        cv2.waitKey(1)
    return compass_image, compass_width + (2 * doubt), compass_height + (2 * doubt)


# Get navpoint offset
same_last_count = 0
last_last = {'x': 1, 'y': 100}


def get_navpoint_offset(testing=False, last=None):
    global same_last_count, last_last
    if SCREEN_WIDTH == 3840:
        navpoint_template = cv2.imread(resource_path("templates/navpoint_3840.png"), cv2.IMREAD_GRAYSCALE)
    elif SCREEN_WIDTH == 2560:
        navpoint_template = cv2.imread(resource_path("templates/navpoint_2560.png"), cv2.IMREAD_GRAYSCALE)
    else:
        navpoint_template = cv2.imread(resource_path("templates/navpoint_1920.png"), cv2.IMREAD_GRAYSCALE)
    navpoint_width, navpoint_height = navpoint_template.shape[::-1]
    compass_image, compass_width, compass_height = get_compass_image()
    attempts = 0
    while compass_image.size == 0 and attempts < 10:
        compass_image, compass_width, compass_height = get_compass_image()
        attempts += 1
    if attempts > 1:
        logger.debug("Attempts at get_compass_image: " + str(attempts))
    filtered = filter_blue(compass_image)
    # filtered = filter_bright(compass_image)
    match = cv2.matchTemplate(filtered, navpoint_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.5
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(match)
    pt = (0, 0)
    if max_val >= threshold:
        pt = max_loc
    final_x = (pt[0] + ((1 / 2) * navpoint_width)) - ((1 / 2) * compass_width)
    final_y = ((1 / 2) * compass_height) - (pt[1] + ((1 / 2) * navpoint_height))
    if testing:
        cv2.rectangle(compass_image, pt, (pt[0] + navpoint_width, pt[1] + navpoint_height), (0, 0, 255), 2)
        loc = np.where(match >= threshold)
        pts = tuple(zip(*loc[::-1]))
        match = cv2.cvtColor(match, cv2.COLOR_GRAY2RGB)
        for p in pts:
            cv2.circle(match, p, 1, (0, 0, 255), 1)
        cv2.circle(match, pt, 5, (0, 255, 0), 3)
        cv2.imshow('Navpoint Found', compass_image)
        cv2.imshow('Navpoint Mask', filtered)
        cv2.imshow('Navpoint Match', match)
        cv2.waitKey(1)
    if pt == (0, 0):
        if last:
            if last == last_last:
                same_last_count = same_last_count + 1
            else:
                last_last = last
                same_last_count = 0
            if same_last_count > 5:
                same_last_count = 0
                if rand.random() < .9:
                    result = {'x': 1, 'y': 100}
                else:
                    result = {'x': 100, 'y': 1}
            else:
                result = last
        else:
            result = None
    else:
        result = {'x': final_x, 'y': final_y}
    logger.debug('get_navpoint_offset=' + str(result))
    return result


# Get destination offset
def get_destination_offset(testing=False):
    if SCREEN_WIDTH == 3840:
        destination_template = cv2.imread(resource_path("templates/destination_3840.png"), cv2.IMREAD_GRAYSCALE)
    elif SCREEN_WIDTH == 2560:
        destination_template = cv2.imread(resource_path("templates/destination_2560.png"), cv2.IMREAD_GRAYSCALE)
    else:
        destination_template = cv2.imread(resource_path("templates/destination_1920.png"), cv2.IMREAD_GRAYSCALE)
    destination_width, destination_height = destination_template.shape[::-1]
    width = (1 / 3) * SCREEN_WIDTH
    height = (1 / 3) * SCREEN_HEIGHT
    screen = get_screen((1 / 3) * SCREEN_WIDTH, (1 / 3) * SCREEN_HEIGHT, (2 / 3) * SCREEN_WIDTH,
                        (2 / 3) * SCREEN_HEIGHT)
    filtered = filter_orange2(screen)
    match = cv2.matchTemplate(filtered, destination_template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.2
    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(match)
    pt = (0, 0)
    if max_val >= threshold:
        pt = max_loc
    final_x = (pt[0] + ((1 / 2) * destination_width)) - ((1 / 2) * width)
    final_y = ((1 / 2) * height) - (pt[1] + ((1 / 2) * destination_height))
    pt2 = (int(final_x), int(final_y))
    print(pt2)
    if testing:
        cv2.rectangle(screen, pt, (pt[0] + destination_width, pt[1] + destination_height), (0, 0, 255), 2)
        loc = np.where(match >= threshold)
        pts = tuple(zip(*loc[::-1]))
        match = cv2.cvtColor(match, cv2.COLOR_GRAY2RGB)
        for p in pts:
            cv2.circle(match, p, 1, (0, 0, 255), 1)
        cv2.circle(match, pt, 5, (0, 255, 0), 3)
        cv2.circle(screen, pt, 5, (0, 255, 0), 3)
        cv2.imshow('Destination Found', screen)
        cv2.imshow('Destination Mask', filtered)
        cv2.imshow('Destination Match', match)
        cv2.waitKey(1)
    if pt == (0, 0):
        result = None
    else:
        result = {'x': final_x, 'y': final_y}
    logger.debug('get_destination_offset=' + str(result))
    return result


# Autopilot routines

# Undock
def undock():
    logger.debug('undock')
    if ship()['status'] != "in_station":
        logger.error('undock=err1')
        raise Exception('undock error 1')
    send(keys['UI_Back'], repeat=10)
    send(keys['HeadLookReset'])
    send(keys['UI_Down'], hold=3)
    send(keys['UI_Select'])
    sleep(1)
    if not (ship()['status'] == "starting_undock" or ship()['status'] == "in_undock"):
        logger.error('undock=err2')
        raise Exception("undock error 2")
    send(keys['HeadLookReset'])
    send(keys['SetSpeedZero'], repeat=2)
    wait = 120
    for i in range(wait):
        sleep(1)
        if i > wait - 1:
            logger.error('undock=err3')
            raise Exception('undock error 3')
        if ship()['status'] == "in_space":
            break
    logger.debug('undock=complete')
    return True


# Dock
def dock():
    logger.debug('dock')
    if ship()['status'] != "in_space":
        logger.error('dock=err1')
        raise Exception('dock error 1')
    tries = 3
    for i in range(tries):
        send(keys['UI_Back'], repeat=10)
        send(keys['HeadLookReset'])
        send(keys['UIFocus'], state=1)
        send(keys['UI_Left'])
        send(keys['UIFocus'], state=0)
        send(keys['CycleNextPanel'], repeat=2)
        send(keys['UI_Up'], hold=3)
        send(keys['UI_Right'])
        send(keys['UI_Select'])
        sleep(1)
        if ship()['status'] == "starting_dock" or ship()['status'] == "in_dock":
            break
        if i > tries - 1:
            logger.error('dock=err2')
            raise Exception("dock error 2")
    send(keys['UI_Back'])
    send(keys['HeadLookReset'])
    send(keys['SetSpeedZero'], repeat=2)
    wait = 120
    for i in range(wait):
        sleep(1)
        if i > wait - 1:
            logger.error('dock=err3')
            raise Exception('dock error 3')
        if ship()['status'] == "in_station":
            break
    send(keys['UI_Up'], hold=3)
    send(keys['UI_Down'])
    send(keys['UI_Select'])
    logger.debug('dock=complete')
    return True


# Align
def x_angle(point=None):
    if not point or point['x'] == 0:
        return None
    result = math.degrees(math.atan(point['y'] / point['x']))
    if point['x'] > 0:
        return +90 - result
    else:
        return -90 - result


def align():
    logger.info('Starting Alignment')
    if not (ship()['status'] == 'in_supercruise' or ship()['status'] == 'in_space'):
        logger.error('align=error: ship not in supercruise')
        raise Exception('align error: ship not in supercruise')

    logger.info('Alignment Step: Avoid sun')
    while sun_percent() > 5:
        send(keys['PitchUpButton'], state=1)
    send(keys['PitchUpButton'], state=0)

    logger.debug('align=speed 100')
    send(keys['SetSpeed100'])

    logger.info('Alignment Step: Find navpoint')
    off = get_navpoint_offset()
    while not off:
        send(keys['PitchUpButton'], state=1)
        off = get_navpoint_offset()
    send(keys['PitchUpButton'], state=0)

    logger.info('Alignment Step: Crude align')
    close = 3
    close_a = 18
    hold_pitch = 0.350
    hold_roll = 0.170
    ang = x_angle(off)
    while (off['x'] > close and ang > close_a) or (off['x'] < -close and ang < -close_a) or \
            (off['y'] > close) or (off['y'] < -close):

        while (off['x'] > close and ang > close_a) or (off['x'] < -close and ang < -close_a):

            if off['x'] > close and ang > close:
                send(keys['RollRightButton'], hold=hold_roll)
            if off['x'] < -close and ang < -close:
                send(keys['RollLeftButton'], hold=hold_roll)

            if ship()['status'] == 'starting_hyperspace':
                return
            off = get_navpoint_offset(last=off)
            ang = x_angle(off)

        while (off['y'] > close) or (off['y'] < -close):

            if off['y'] > close:
                send(keys['PitchUpButton'], hold=hold_pitch)
            if off['y'] < -close:
                send(keys['PitchDownButton'], hold=hold_pitch)

            if ship()['status'] == 'starting_hyperspace':
                return
            off = get_navpoint_offset(last=off)

        off = get_navpoint_offset(last=off)
        ang = x_angle(off)

    logger.info('Alignment Step: Fine align')
    sleep(0.5)
    close = 50
    hold_pitch = 0.200
    hold_yaw = 0.400
    for i in range(5):
        new = get_destination_offset()
        if new:
            off = new
            break
        sleep(0.25)
    if not off:
        return
    while (off['x'] > close) or (off['x'] < -close) or (off['y'] > close) or (off['y'] < -close):

        if off['x'] > close:
            send(keys['YawRightButton'], hold=hold_yaw)
        if off['x'] < -close:
            send(keys['YawLeftButton'], hold=hold_yaw)
        if off['y'] > close:
            send(keys['PitchUpButton'], hold=hold_pitch)
        if off['y'] < -close:
            send(keys['PitchDownButton'], hold=hold_pitch)

        if ship()['status'] == 'starting_hyperspace':
            return

        for i in range(5):
            new = get_destination_offset()
            if new:
                off = new
                break
            sleep(0.25)
        if not off:
            return

    logger.info('Alignment Complete')


# Jump
def jump():
    logger.info('Starting Jump')
    tries = 5
    for i in range(tries):
        if ship()['status'] == 'starting_hyperspace':
            logger.info('Jump in Progress')
            while ship()['status'] != 'in_supercruise':
                sleep(1)
            logger.debug('jump=speed 0')
            send(keys['SetSpeedZero'])
            logger.info('Jump Complete')
            return True
        logger.debug('jump=try:' + str(i))
        if not (ship()['status'] == 'in_supercruise' or ship()['status'] == 'in_space'):
            logger.error('jump_err=status:' + ship()['status'])
            raise Exception('not ready to jump')
        sleep(0.5)
        send(keys['SetSpeed100'])
        logger.info('Charging FSD')
        send(keys['HyperSuperCombination'], hold=1)
        sleep(16)
        if ship()['status'] != 'starting_hyperspace':
            logger.info('Trajectory Misaligned. Jump Failed. Retrying Alignment')
            align()

    logger.error('jump=err2')
    raise Exception("jump failure")


# Refuel
def refuel(refuel_threshold=50):
    logger.debug('refuel')
    scoopable_stars = ['F', 'O', 'G', 'K', 'B', 'A', 'M']
    if ship()['status'] != 'in_supercruise':
        logger.error('refuel=error: Ship not in supercruise')
        return False

    if ship()['fuel_percent'] < refuel_threshold and ship()['star_class'] in scoopable_stars:
        logger.info('Starting Refuel')
        sleep(2)
        send(keys['SetSpeed100'])
        sleep(4)
        logger.info('Refuel in Progress')
        send(keys['SetSpeedZero'], repeat=3)
        while not ship()['fuel_percent'] == 100:
            sleep(1)
        logger.info('Refuel Complete')
        return True
    elif ship()['fuel_percent'] >= refuel_threshold:
        logger.info('Refuel not Needed')
        return False
    elif ship()['star_class'] not in scoopable_stars:
        logger.info('Refuel Needed, Unsuitable Star')
        return False
    else:
        return False


def scanFSS(aFFS):
    align()
    if aFFS and not ship()['sys_fully_scanned']:
        send(keys['SetSpeed100'])  # The farther away we are, the easier the system is to scan
        sleep(15)  # and there's far less chance of obstructed frequencies
    logger.info("Discovery scanning")
    send(keys['SetSpeedZero'])
    send(keys['ExplorationFSSEnter'])
    sleep(1)
    send(keys['ExplorationFSSDiscoveryScan'], 3.5)
    if aFFS:
        logger.info("Waiting for FSS to complete")
        while not ship()['sys_fully_scanned']:
            sleep(5)  # TODO: Actually scan around and such

    send(keys['ExplorationFSSQuit'])
    logger.info("Scan complete")


# Position
def position(refueled_multiplier=1):
    logger.debug('position')
    send(keys['PitchUpButton'], state=1)
    sleep(5)
    send(keys['PitchUpButton'], state=0)
    send(keys['SetSpeed100'])
    send(keys['PitchUpButton'], state=1)
    while sun_percent() > 3:
        sleep(1)
    sleep(5)
    send(keys['PitchUpButton'], state=0)
    sleep(5 * refueled_multiplier)
    logger.debug('position=complete')
    return True


# Autopilot main

# status reference
#
# 'in-station'
# 
# 'in-supercruise'
# 
# 'in-space'
# 
# 'starting-undocking'
# 
# 'in-undocking'
# 
# 'starting-docking'
# 
# 'in-docking'


def autopilot():
    # logger.info('\n'+200*'-'+'\n'+'---- AUTOPILOT START '+179*'-'+'\n'+200*'-')
    logger.info('---- AUTOPILOT START ' + 179 * '-')
    logger.debug('get_latest_log=' + str(get_latest_log(PATH_LOG_FILES)))
    logger.debug('ship=' + str(ship()))
    while ship()['target']:
        if ship()['status'] == 'in_space' or ship()['status'] == 'in_supercruise':
            logger.info('---- AUTOPILOT ALIGN ' + 179 * '-')
            align()

            if getOption('DiscoveryScan'):
                logger.info('---- AUTOPILOT SCAN ' + 180 * '-')
                scanFSS(getOption('AutoFSS'))

            logger.info('---- AUTOPILOT JUMP ' + 180 * '-')
            jump()

            logger.info('---- AUTOPILOT REFUEL ' + 178 * '-')
            refueled = refuel(int(getOption('RefuelThreshold')))
            if refueled:
                position(refueled_multiplier=4)
            else:
                position(refueled_multiplier=1)

    send(keys['SetSpeedZero'])
    logger.info('---- AUTOPILOT END ' + 181 * '-')
    from dev_tray import stop_action
    stop_action()

#!/usr/bin/env python # coding: utf-8 # References # Useful docs / articles / etc # #   1 - [A Python wrapper around AHK](https://pypi.org/project/ahk/) # #   2 - [OpenCV on Wheels](https://pypi.org/project/opencv-python/) # #   3 - [Autopilot for Elite Dangerous using OpenCV and thoughts on CV enabled bots in visual-to-keyboard loop] #   (https://networkgeekstuff.com/projects/autopilot-for-elite-dangerous-using-opencv-and-thoughts-on-cv-enabled-bots-in-visual-to-keyboard-loop/) # #   4 - [Using PyInstaller to Easily Distribute Python Applications](https://realpython.com/pyinstaller-python/) # #   5 - [Direct Input to a Game - Python Plays GTA V] #   (https://pythonprogramming.net/direct-input-game-python-plays-gta-v/) # #   6 - [Cross-platform GUI automation for human beings](https://pyautogui.readthedocs.io/en/latest/index.html) # Imports # see reference 2 # see reference 6 # see reference 5 Get absolute path to resource, works for dev and for PyInstaller # PyInstaller creates a temp folder and stores path in _MEIPASS # logger.info('\n'+200*'-'+'\n'+'---- AUTOPILOT DATA '+180*'-'+'\n'+200*'-') # Constants # Read ED logs # Get latest log file Returns the full path of the latest (most recent) elite log file (journal) from specified path # Extract ship info from log Returns a 'status' dict containing relevant game status information (state, fuel, ...) # Read log line by line and parse data # parse data # parse ship status # parse ship type # parse fuel # parse scoop # parse location # parse target # exceptions #     logger.debug('ship='+str(ship)) # Control ED with direct input # Get latest keybinds file # Extract necessary keys Returns a dict struct with the direct input equivalent of the necessary elite keybindings # Check primary # Check secondary (and prefer secondary) # Prepare final binding # else: #    logger.warning("get_bindings: "+item.tag+" = does not have a valid keyboard keybind.") # Direct input function # Send input # Clear input # OpenCV # Get screen # HSV slider tool # create trackbars for color change # grab the frame # get trackbar positions # show thresholded image # Equalization # Load the image in greyscale # create a CLAHE object (Arguments are optional). # Filter bright # Filter sun # converting from BGR to HSV color space # filter Elite UI orange # Filter orange # converting from BGR to HSV color space # filter Elite UI orange # Filter orange2 # converting from BGR to HSV color space # filter Elite UI orange # Filter blue # converting from BGR to HSV color space # filter Elite UI orange # Get sun # Get compass image # This crashes linux machines # cv2.imshow('Compass Mask', equalized) # Get navpoint offset # filtered = filter_bright(compass_image) # Get destination offset # Autopilot routines # Undock # Dock # Align # Jump # Refuel # The farther away we are, the easier the system is to scan # and there's far less chance of obstructed frequencies # TODO: Actually scan around and such # Position # Autopilot main # status reference # # 'in-station' # # 'in-supercruise' # # 'in-space' # # 'starting-undocking' # # 'in-undocking' # # 'starting-docking' # # 'in-docking' # logger.info('\n'+200*'-'+'\n'+'---- AUTOPILOT START '+179*'-'+'\n'+200*'-')