#!/usr/bin/env python

# file: $(NEDC_NFC)/src/main.py
#
# revision history:
#  20210614 (JP): updated the location for the release
#  20160613 (MT): initial version
#
# This script calls the Main class to run a shell of the demo.
# ------------------------------------------------------------------------------

import pathlib as path
import os
import sys

os.environ["NEDC_NFC"] = os.path.abspath(os.path.join(os.path.dirname( __file__ ), os.pardir))
lib_path = os.path.abspath(os.path.join(os.path.dirname( __file__ ), os.pardir, 'lib'))
sys.path.insert(0, lib_path)

# import modules
#
from pyqtgraph.Qt import QtWidgets
from classes.demo_event_loop import DemoEventLoop

# add bin directory path for importing necessary modules
#
import nedc_file_tools as nft
import nedc_cmdl_parser as ncp

current_path = os.path.dirname(os.path.realpath(__file__))

NEDC_HELP_FILE = os.path.join(current_path, "resources", "nedc_demo_help.txt")
NEDC_USAGE_FILE = os.path.join(current_path, "resources", "nedc_demo_usage.txt")

# stylesheet for the application
stylesheet = """

QPushButton {
    padding: 8px 4px;
}

QMenu {
    background-color: #f9f9f9; /* sets background of the menu */
    border: 1px solid #D3D3D3;
    padding: 2px;
}

QMenu::item {
    /* sets background of menu item. set this to something non-transparent
        if you want menu color and menu item color to be different */
    background-color: transparent;
    padding: 8px 32px;
}

QMenu::item:selected { /* when user selects item using mouse or keyboard */
    background-color: #5692CD;
    color: white;
    border-radius: 5px;
}

QMenu::item:checked {
    padding: 8px 32px;
}

QMenu::item:unchecked {
}

QMenuBar {
    border-bottom: 1px solid #D3D3D3;
}

QMenuBar::item {
    padding: 10px;
    background: transparent;
}

QMenuBar::item:selected { /* when selected using mouse or keyboard */
    background: #5692CD;
    color: white;
}

QMenuBar::item:pressed {
    background: #5692CD;
    color: white;
}

"""

# ------------------------------------------------------------------------------
#
# the main program starts here
#
# ------------------------------------------------------------------------------

def main():
    """
    method: main

    arguments: none

    return: none

    description:
        This method calls Main and the necessary functions to display the demo
        window
    """
    # set local environment path
    #
    dirname, fname = os.path.split(os.path.realpath(__file__))
    os.environ["NEDC_EAS_BASE_SRC"] = dirname
    
    # declare the parser
    #
    parser = ncp.Cmdl(NEDC_USAGE_FILE, NEDC_HELP_FILE)
    parser.add_argument("file_arguments",  type=str, nargs='*')
    parser.add_argument("--montage", "-mtg", type=str)
    parser.add_argument("--map", "-m", type=str)

    # parse the user's arguments
    #
    args = parser.parse_args()
    montage_file_to_use = args.montage
    ann_map = args.map
    in_files = args.file_arguments
    
    # loop over files passed in accumulate edfs to open
    #
    files_to_open = []
    for file_argument in in_files:

        # if user passes in a line by line list of files, lets loop over
        # it and accumulate the files listed
        #
        if '.list' in file_argument or '.txt' in file_argument:
            with open(file_argument) as line_by_line_file_list:
                contents = line_by_line_file_list.readlines()
                contents = [line.strip() for line in contents]
                for line in contents:
                    files_to_open.append(line)
        else:
            files_to_open.append(file_argument)

    # Qt boilerplate - absolutely necessary
    #
    app = QtWidgets.QApplication([])
    app.setStyleSheet(stylesheet)
    # loop over files to open and open if file is an edf. If no
    # files_to_open have been accumulated, than launch a single event loop
    # without a file
    #
    if files_to_open:
        for file_to_open in files_to_open:
            if file_to_open.endswith(".edf"):
                file_to_read_on_init = os.path.abspath(file_to_open)
                initial_loop = DemoEventLoop(
                    montage_file_to_use = montage_file_to_use,
                    ann_map = ann_map)
                initial_loop.open_edf_file(file_to_read_on_init)

    else:
        initial_loop = DemoEventLoop()
    
    # more Qt boilerplate
    #
    QtWidgets.QApplication.instance().exec()
    
    # clean up and exit gracefully
    #
    sys.exit(0)
    

# begin gracefully
#
if __name__ == "__main__":
    main()
#
# end of file