# -*- coding: utf-8 -*-


"""
Munter Time Calculation
Alexander Vasarab
Wylark Mountaineering LLC

A rudimentary program which implements the Munter time calculation.
"""

import sys
import argparse

class InvalidUnitsException(Exception):
    pass

rates = {
    'uphill': { 'rate': 4, 'direction': '↑' },
    'flat': { 'rate': 6, 'direction': '→' }, # or downhill on foot
    'downhill': { 'rate': 10, 'direction': '↓' },
    'bushwhacking': { 'rate': 2, 'direction': '↹' },
}

fitnesses = {
    'slow': 1.2,
    'average': 1,
    'fast': .7,
}

unit_choices = ['metric', 'imperial']
travel_mode_choices = rates.keys()
fitness_choices = fitnesses.keys()

def time_calc(distance, elevation, fitness='average', rate='uphill',
    units='imperial'):
    retval = {}

    if units not in unit_choices:
        raise InvalidUnitsException

    unit_count = 0

    if 'imperial' == units:
        # convert to metric
        distance = (distance * 1.609) # mi to km
        elevation = (elevation * .305) # ft to m

    unit_count = distance + (elevation / 100.0)

    retval['time'] = (distance + (elevation / 100.0)) / rates[rate]['rate']
    retval['time'] = retval['time'] * fitnesses[fitness]

    retval['unit_count'] = unit_count
    retval['direction'] = rates[rate]['direction']
    retval['pace'] = rates[rate]['rate']

    return retval

def print_ugly_estimate(est):
    hours = int(est['time'])
    minutes = int((est['time'] - hours) * 60)
    print("{human_time}".format(
            human_time="{hours} hours {minutes} minutes".format(
                hours=hours,
                minutes=minutes)))

def print_pretty_estimate(est):
    hours = int(est['time'])
    minutes = int((est['time'] - hours) * 60)

    # NOTE: Below, the line with the unicode up arrow uses an alignment
    #       value of 31. In the future, consider using e.g. wcwidth
    #       library so that this is more elegant.
    print("\n\t╒═══════════════════════════════╕")
    print("\t╎▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒╎╮")
    print("\t╎▒{:^29}▒╎│".format(''))
    print("\t╎▒{pace_readable:^31}▒╎│".format(
            pace_readable="{units} {direction} @ {pace}".format(
                units=round(est['unit_count']),
                direction=est['direction'],
                pace=est['pace'])))
    print("\t╎▒{human_time:^29}▒╎│".format(
            human_time="{hours} hours {minutes} minutes".format(
                hours=hours,
                minutes=minutes)))
    print("\t╎▒{:^29}▒╎│".format(''))
    print("\t╎▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒╎│")
    print("\t╘═══════════════════════════════╛│")
    print("\t └───────────────────────────────┘\n")

def get_parser():
    parser = argparse.ArgumentParser(description='Implementation of '
        'the Munter time calculation')

    # No required args anymore, since -g overrides any requirement
    parser.add_argument('--distance',
        '-d',
        default=0.0,
        type=float,
        required=False,
        help='Distance (in km, by default)')

    parser.add_argument('--elevation',
        '-e',
        default=0.0,
        type=float,
        required=False,
        help='Elevation change (in m, by default)')

    parser.add_argument('--travel-mode',
        '-t',
        type=str,
        default='uphill',
        choices=travel_mode_choices, required=False,
        help='Travel mode (uphill, by default)')

    parser.add_argument('--fitness',
        '-f',
        type=str,
        default='average',
        choices=fitness_choices, required=False,
        help='Fitness modifier (average, by default)')

    parser.add_argument('--units',
        '-u',
        type=str,
        default='imperial',
        required=False,
        choices=unit_choices,
        help='Units of input values')

    parser.add_argument('--pretty',
        '-p',
        action='store_true',
        default=False,
        required=False,
        help="Make output pretty");

    parser.add_argument('--gui',
        '-g',
        action='store_true',
        default=False,
        required=False,
        help='Launch GUI mode (overrides --pretty)')

    return parser

def main():
    parser = get_parser()
    opts = parser.parse_args()

    distance = opts.distance
    elevation = opts.elevation
    fitness = opts.fitness
    units = opts.units
    travel_mode = opts.travel_mode
    pretty = opts.pretty
    gui = opts.gui

    time_estimate = time_calc(distance=distance, elevation=elevation,
        fitness=fitness, rate=travel_mode, units=units)

    if gui:
        from . import gui
        gui.startup()
    else:
        if pretty:
            print_pretty_estimate(time_estimate)
        else:
            print_ugly_estimate(time_estimate)

    return 0

if __name__ == "__main__":
    sys.exit(main())