from tkinter import Text, END
import math
import time
import display_centre.menu as menu

class Display(object):
    MENU_HEIGHT = 10
    SELECTOR_WIDTH = 0.47
    ROW_OFFSET = 6.0
    TITLE = '================== r_e_c_u_r =================='

    def __init__(self, tk, video_driver, shaders, message_handler, data):
        self.tk = tk
        self.video_driver = video_driver
        self.capture = None
        self.shaders = shaders
        self.message_handler = message_handler
        self.data = data
        self.browser_menu = menu.BrowserMenu(self.data, self.message_handler, self.MENU_HEIGHT)      
        self.settings_menu = menu.SettingsMenu(self.data, self.message_handler, self.MENU_HEIGHT)
        self.shaders_menu = self.shaders.shaders_menu

        #self.top_menu_index = 0
        #self.selected_list_index = self.top_menu_index
        
        self.display_text = self._create_display_text(self.tk)
        self._add_tags()
        self._update_screen_every_second()

    @staticmethod
    def _create_display_text(tk):
        return Text(tk, bg="black", fg="white", font=('Liberation Mono', 13), undo=False)

    def _add_tags(self):
        self.display_text.tag_configure("SELECT", background="white", foreground="black")
        self.display_text.tag_configure("TITLE", background="black", foreground="red")
        self.display_text.tag_configure("DISPLAY_MODE", background="black", foreground="magenta")
        self.display_text.tag_configure("ERROR_MESSAGE", background="red", foreground="white")
        self.display_text.tag_configure("INFO_MESSAGE", background="blue", foreground="white")
        self.display_text.tag_configure("NOW_PLAYER_INFO", background="black", foreground="yellow")
        self.display_text.tag_configure("NEXT_PLAYER_INFO", background="black", foreground="cyan")
        self.display_text.tag_configure("COLUMN_NAME", background="black", foreground="VioletRed1")
        self.display_text.tag_configure("SHADER_PARAM", background="VioletRed1", foreground="black")
        self.display_text.tag_configure("FUNCTION", background="yellow", foreground="black")
        self.display_text.tag_configure("BROKEN_PATH", background="black", foreground="gray")
        
    def _load_display(self):
        self._load_title()
        self._load_player()
        self._load_display_body()
        self._load_message()
        #print('the number of tags are {}'.format(len(self.display_text.tag_names())))
        self.display_text.pack()

    def _load_title(self):
        self.display_text.insert(END, self.TITLE + ' \n')
        self.display_text.tag_add("TITLE", 1.19, 1.28)

    def _load_player(self):
        if self.data.player_mode == 'now':
            now_banner = self._get_banner_for_player('now')
            self.display_text.insert(END, now_banner + '\n')
            self.display_text.tag_add("NOW_PLAYER_INFO", 2.0, 2.0 + self.SELECTOR_WIDTH)
        elif self.data.player_mode == 'next':
            next_banner = self._get_banner_for_player('next')
            self.display_text.insert(END, next_banner + '\n')
            self.display_text.tag_add("NEXT_PLAYER_INFO", 2.0, 2.0 + self.SELECTOR_WIDTH)

        status = self._get_status_for_player()
        self.display_text.insert(END, status + '\n')
        self.display_text.tag_add("NOW_ALPHA", 3.0, 3.17)
        self.display_text.tag_add("CAPTURE_ALPHA", 3.18, 3.29)
        self.display_text.tag_add("NEXT_ALPHA", 3.29, 3.47)      

    def _load_display_body(self):
        if self.data.display_mode == 'BROWSER':
            self._load_browser()
        elif self.data.display_mode == 'SETTINGS':
            self._load_settings()
        elif self.data.display_mode == 'SAMPLER':
            self._load_sampler()
        elif self.data.display_mode == 'SHADERS':
            self._load_shaders()
        self.display_text.tag_add("COLUMN_NAME", 5.0, 6.0)
        

    def _load_sampler(self):
        bank_data = self.data.bank_data[self.data.bank_number]
        self.display_text.insert(END, '------------------ <SAMPLER> ------------------ \n')
        self.display_text.tag_add("DISPLAY_MODE", 4.19, 4.29)
        self.display_text.insert(END, '{:>6} {:<17} {:>5} {:<5} {:<5} {:<4}\n'.format(
            '{}-slot'.format(self.data.bank_number), 'name', 'length', 'start', 'end', 'spd'))
        for index, slot in enumerate(bank_data):
            name_without_extension =  slot['name'].rsplit('.',1)[0]
            self.display_text.insert(END, '{:^6} {:<17} {:^5} {:>5} {:<5} {:>4}\n'.format(
                index, name_without_extension[0:17], self.format_time_value(slot['length']),
                self.format_time_value(slot['start']), self.format_time_value(slot['end']),self.format_speed_value(slot['rate'])))
            if self.data.is_this_path_broken(slot['location']):
                self.display_text.tag_add("BROKEN_PATH", self.ROW_OFFSET + index,
                                  self.ROW_OFFSET + self.SELECTOR_WIDTH + index)
        current_bank , current_slot = self.data.split_bankslot_number(self.video_driver.current_player.bankslot_number)
        if current_bank is self.data.bank_number:
            self._highlight_this_row(current_slot)

    def _load_browser(self):
        browser_list = self.browser_menu.menu_list
        number_of_lines_displayed = 0
        self.display_text.insert(END, '------------------ <BROWSER> ------------------ \n')
        self.display_text.tag_add("DISPLAY_MODE", 4.19, 4.29)
        self.display_text.insert(END, '{:40} {:5} \n'.format('path', 'slot'))

        number_of_browser_items = len(browser_list)
        for index in range(number_of_browser_items):
            if number_of_lines_displayed >= self.MENU_HEIGHT:
                break
            if index >= self.browser_menu.top_menu_index:
                path = browser_list[index]
                self.display_text.insert(END, '{:40} {:5} \n'.format(path['name'][0:38], path['slot']))
                number_of_lines_displayed = number_of_lines_displayed + 1

        for index in range(self.MENU_HEIGHT - number_of_browser_items):
            self.display_text.insert(END, '\n')

        self._highlight_this_row(self.browser_menu.selected_list_index - self.browser_menu.top_menu_index)

    def _load_settings(self):
        line_count = 0
        settings_list = self.settings_menu.menu_list
        self.display_text.insert(END, '------------------ <SETTINGS> ----------------- \n')
        self.display_text.tag_add("DISPLAY_MODE", 4.19, 4.29)
        self.display_text.insert(END, '{:^23} {:^22} \n'.format('SETTING', 'VALUE'))
        number_of_settings_items = len(settings_list)
        for index in range(number_of_settings_items):
            if line_count >= self.MENU_HEIGHT:
                break
            if index >= self.settings_menu.top_menu_index:
                setting = settings_list[index]
                self.display_text.insert(END, '{:<23} {:<22} \n'.format(setting['name'][0:22], setting['value']))
                line_count = line_count + 1

        for index in range(self.MENU_HEIGHT - number_of_settings_items):
            self.display_text.insert(END, '\n')

        self._highlight_this_row(self.settings_menu.selected_list_index - self.settings_menu.top_menu_index)

    def _load_shaders(self):
        line_count = 0
        self.display_text.insert(END, '------------------ <SHADERS> ------------------ \n')
        self.display_text.tag_add("DISPLAY_MODE", 4.19, 4.28)
        ## showing current shader info:
        shader = self.shaders.selected_shader
        self.display_text.insert(END, '{:<1}:{:<1}{:<2} {:<17} '.format \
            (self.shaders.selected_status,shader['shad_type'][0], \
            format(shader['shad_index'],'02d'), shader['name'][0:17] ))
        for i in range(min(4,shader['param_number'])):
            display_param = self.format_param_value(self.shaders.selected_param_values[i])
            if display_param == 100:
                display_param == 99
            self.display_text.insert(END, 'x{}:{:02d}'.format(i, display_param))
        self.display_text.insert(END,'\n')
        self.display_text.tag_add("COLUMN_NAME", 5.0, 6.0)
        ## showing list of other shaders:
        shaders_list = self.shaders.shaders_menu_list
        number_of_shader_items = len(shaders_list)
        for index in range(number_of_shader_items):
            if line_count >= self.MENU_HEIGHT :
                break
            if index >= self.shaders.shaders_menu.top_menu_index:
                shader_line = shaders_list[index]
                self.display_text.insert(END, '{:<40} {:<5} \n'.format(shader_line['name'][0:30], shader_line['shad_type']))
                line_count = line_count + 1
        for index in range(self.MENU_HEIGHT - number_of_shader_items):
            self.display_text.insert(END, '\n')        
        self._highlight_this_row(self.shaders.shaders_menu.selected_list_index - self.shaders.shaders_menu.top_menu_index)
        if self.data.control_mode == "SHADER_PARAM":
            self._highlight_this_param(self.shaders.focused_param)


    def _load_message(self):
        if self.message_handler.current_message[1]:
            self.display_text.insert(END, '{:5} {:42} \n'.format(
                self.message_handler.current_message[0], self.message_handler.current_message[1][0:38]))
            self.display_text.tag_add('{}_MESSAGE'.format(
                self.message_handler.current_message[0]), 16.0,16.0 + self.SELECTOR_WIDTH)
            if self.message_handler.current_message[2]:
                self.message_handler.current_message[2] = False
                message_length = 4000
                self.tk.after(message_length, self.message_handler.clear_message)
        elif self.data.function_on:
            self.display_text.insert(END, '{:^47} \n'.format('< FUNCTION KEY ON >'))
            self.display_text.tag_add('FUNCTION', 16.0,16.0 + self.SELECTOR_WIDTH)
        else:
            self.display_text.insert(END, '{:8} {:<10} \n'.format('CONTROL:', self.data.control_mode))
            self.display_text.tag_add('TITLE', 16.0,16.0 + self.SELECTOR_WIDTH)

    def _highlight_this_row(self, row):
        self.display_text.tag_add("SELECT", self.ROW_OFFSET + row,
                                  self.ROW_OFFSET + self.SELECTOR_WIDTH + row)

    def _unhighlight_this_row(self, row):
        self.display_text.tag_remove("SELECT", self.ROW_OFFSET + row,
                                     self.ROW_OFFSET + self.SELECTOR_WIDTH + row)

    def _highlight_this_param(self, param_num):
        param_row = self.ROW_OFFSET - 1
        column_offset = 0.24
        param_length = 0.05
        self.display_text.tag_add("SHADER_PARAM", round(param_row + column_offset + param_num*param_length,2),
round(param_row + column_offset + (param_num+1)*param_length, 2))


    def _get_status_for_player(self):
        now_slot, now_status, now_alpha, next_slot, next_status, next_alpha = self.video_driver.get_player_info_for_status()

        if self.capture is not None:
            capture_status = self._generate_capture_status()
            preview_alpha = self.capture.get_preview_alpha()
        else:
            capture_status = ''
            preview_alpha = 0
            

        if preview_alpha == None:
            preview_alpha = 0
        #print('capture alpha is {}'.format(preview_alpha))

        self._set_colour_from_alpha(now_alpha, preview_alpha, next_alpha)        

        now_info = 'NOW [{}] {}'.format(now_slot, now_status)
        next_info = 'NEXT [{}] {}'.format(next_slot, next_status)
        capture_info = '{}'.format(capture_status)
        return  '{:17} {:10} {:17}'.format(now_info[:17], capture_info[:10], next_info[:18])

    def _get_banner_for_player(self,player):
        start, end, position = self.video_driver.get_player_info_for_banner(player)
        banner = self.create_video_display_banner(start, end, position)
        time_been = self.format_time_value(position - start)
        time_left = self.format_time_value(end - position)
        return ' {:5} {} {:5}'.format(time_been, banner, time_left)

    def _generate_capture_status(self):
        is_previewing = self.capture.is_previewing 
        is_recording = self.capture.is_recording
        rec_time = -1
        if is_recording == True:
            rec_time = self.capture.get_recording_time()
        capture_status = ''
        if is_previewing and is_recording == True:
            capture_status = '<{}>'.format('REC'+ self.format_time_value(rec_time))
        elif is_previewing and is_recording == 'saving':
            capture_status = '<{}>'.format('_saving_')
        elif is_previewing:
            capture_status = '<{}>'.format('_preview')
        elif is_recording == True:
            capture_status = '[{}]'.format('REC'+ self.format_time_value(rec_time))
        elif is_recording == 'saving':
            capture_status =  '[{}]'.format('_saving_')
        else:
            capture_status = ''

        return capture_status

    @staticmethod
    def create_video_display_banner(start, end, position):
        banner_list = ['[', '-', '-', '-', '-', '-', '-', '-', '-',
                       '-', '-', '-', '-', '-', '-', '-', '-', '-',
                       '-', '-', '-', '-', '-', '-', '-', '-', '-',
                       '-', '-', '-', '-', '-',
                       ']']
        max = len(banner_list) - 1
        if position < start:
            banner_list[0] = '<'
        elif position > end:
            banner_list[max] = '>'
        elif end - start != 0 and not math.isnan(position) :
            #print('start value is {}, end value is {}, position is {}'.format(start, end, position))
            marker = int(math.floor(float(position - start) /
                                    float(end - start) * (max - 1)) + 1)
            banner_list[marker] = '*'

        return ''.join(banner_list)

    def _set_colour_from_alpha(self, now_alpha, preview_alpha, next_alpha):
        upper_bound = 150
        if self.capture is not None:
            is_recording = self.capture.is_recording == True
        else:
            is_recording = False
        ### scale values
        scaled_now = int(( now_alpha / 255 ) * (255 - upper_bound) + upper_bound)
        scaled_preview = int(( preview_alpha / 255 ) * (255 - upper_bound) + upper_bound)
        scaled_next = int(( next_alpha / 255 ) * (255 - upper_bound) + upper_bound)
        
        ### convert to hex
        now_colour = self.hex_from_rgb(scaled_now, scaled_now, 0)
        capture_colour = self.hex_from_rgb(255 * is_recording, 100, scaled_preview )
        next_colour = self.hex_from_rgb(0, scaled_next, scaled_next)
        ### update the colours
        self.display_text.tag_configure("NOW_ALPHA", background="black", foreground=now_colour)
        self.display_text.tag_configure("CAPTURE_ALPHA", background="black", foreground=capture_colour)
        self.display_text.tag_configure("NEXT_ALPHA", background="black", foreground=next_colour) 

    @staticmethod
    def hex_from_rgb(r, g, b):
        return '#%02x%02x%02x' % (r, g, b)

    def _update_screen_every_second(self):
        self.refresh_display()
        self.tk.after(500, self._update_screen_every_second)

    def refresh_display(self):
        if self.data.update_screen:
            self.display_text.configure(state='normal')
            self.display_text.delete(1.0, END)
            self._load_display()
            self.display_text.configure(state='disable')
            self.display_text.focus_set()

    @staticmethod
    def format_time_value(time_in_seconds):
        if time_in_seconds < 0:
            return ''
        elif time_in_seconds >= 6000:
            return '99:99'
        else:
            return time.strftime("%M:%S", time.gmtime(time_in_seconds))

    @staticmethod
    def format_speed_value(value):
        if value == 1:
            return ''
        else:
            return value

    @staticmethod
    def format_param_value(value):
        display_param = int(100 * value)
        if display_param == 100:
            display_param = 99
        return display_param