import wx
import sys

if sys.platform.count("win32") > 0:
    FONT_VARIANT = 0
    PNL_WIDTH = 460
else:
    FONT_VARIANT = 1
    PNL_WIDTH = 500
FAMILY = ['serif', 'sas-serif', 'fantasy', 'monospace']
SIZE = [8, 9, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 36, 48, 72]
STYLE = ['normal', 'italic']
WEIGHT = ['light', 'normal', 'bold']
COLOR = ['black', 'blue', 'green', 'red', 'cyan', 'magenta', 'yellow']


class TextDialog(wx.Dialog):
    def __init__(self, parent, id, title, label='', unit=None):
        """
        Dialog window pops- up when selecting 'Add Text' on the toolbar
        """
        wx.Dialog.__init__(self, parent, id, title, size=(PNL_WIDTH, 280))
        self.parent = parent
        # Font
        self.SetWindowVariant(variant=FONT_VARIANT)
        # default
        self.family = FAMILY[1]
        self.size = SIZE[3]
        self.style = STYLE[0]
        self.weight = WEIGHT[1]
        self.color = COLOR[0]
        self.tick_label = False
        # Dialog interface
        vbox = wx.BoxSizer(wx.VERTICAL)
        text_box = wx.BoxSizer(wx.HORIZONTAL)
        sizer = wx.GridBagSizer(1, 3)
        _BOX_WIDTH = 60
        font_description = wx.StaticBox(self, -1, 'Font', size=(PNL_WIDTH - 20, 70))
        font_box = wx.StaticBoxSizer(font_description, wx.VERTICAL)
        family_box = wx.BoxSizer(wx.HORIZONTAL)
        style_box = wx.BoxSizer(wx.HORIZONTAL)
        # tcA
        if unit != None:
            styles = wx.TAB_TRAVERSAL
            height = -1
            unit_text = wx.StaticText(self, -1, 'Unit :')
            unit_text.SetToolTipString("Unit of the axis.")
            self.unit_ctrl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, -1))
            self.unit_ctrl.SetValue(str(unit))
            unit_box = wx.BoxSizer(wx.HORIZONTAL)
            tick_label_text = wx.StaticText(self, -1, 'Tick label')
            tick_label_text.SetToolTipString("Apply to tick label too.")
            self.tick_label_check = wx.CheckBox(self, -1, '', (10, 10))
            self.tick_label_check.SetValue(False)
            self.tick_label_check.SetToolTipString("Apply to tick label too.")
            wx.EVT_CHECKBOX(self, self.tick_label_check.GetId(),
                            self.on_tick_label)
            enter_text = 'Enter text:'
        else:
            styles = wx.TAB_TRAVERSAL | wx.TE_MULTILINE | wx.TE_LINEWRAP | \
                              wx.TE_PROCESS_ENTER | wx.SUNKEN_BORDER | wx.HSCROLL
            height = 60
            unit_text = None
            self.unit_ctrl = None
            unit_box = None
            tick_label_text = None
            self.tick_label_check = None
            enter_text = 'Enter text'
            if len(label) > 0:
                enter_text += " (this text won't be auto-updated if modified.):"
            else:
                enter_text += ":"
        self.text_string = wx.TextCtrl(self, -1, size=(PNL_WIDTH - 30, height), style=styles)
        self.text_string.SetValue(str(label))
        self.text_string.SetToolTipString("The text that will be displayed.")
        # font family
        self.font_family = wx.ComboBox(self, -1, style=wx.CB_READONLY)
        wx.EVT_COMBOBOX(self.font_family, -1, self.on_family)
        self.font_family.SetMinSize((_BOX_WIDTH, -1))
        self._set_family_list()
        self.font_family.SetSelection(1)
        self.font_family.SetToolTipString("Font family of the text.")
        # font weight
        self.font_weight = wx.ComboBox(self, -1, style=wx.CB_READONLY)
        wx.EVT_COMBOBOX(self.font_weight, -1, self.on_weight)
        self.font_weight.SetMinSize((_BOX_WIDTH, -1))
        self._set_weight_list()
        self.font_weight.SetSelection(1)
        self.font_weight.SetToolTipString("Font weight of the text.")
        # font family
        self.font_size = wx.ComboBox(self, -1, style=wx.CB_READONLY)
        wx.EVT_COMBOBOX(self.font_size, -1, self.on_size)
        self.font_size.SetMinSize((_BOX_WIDTH, -1))
        self._set_size_list()
        self.font_size.SetSelection(5)
        self.font_size.SetToolTipString("Font size of the text.")
        # font style
        self.font_style = wx.ComboBox(self, -1, style=wx.CB_READONLY)
        wx.EVT_COMBOBOX(self.font_style, -1, self.on_style)
        self.font_style.SetMinSize((_BOX_WIDTH, -1))
        self._set_style_list()
        self.font_style.SetSelection(0)
        self.font_style.SetToolTipString("Font style of the text.")
        # font color
        self.font_color = wx.ComboBox(self, -1, style=wx.CB_READONLY)
        wx.EVT_COMBOBOX(self.font_color, -1, self.on_color)
        self.font_color.SetMinSize((_BOX_WIDTH, -1))
        self._set_color_list()
        self.font_color.SetSelection(0)
        self.font_color.SetToolTipString("Font color of the text.")
        # Buttons on the bottom
        self.static_line_1 = wx.StaticLine(self, -1)
        self.ok_button = wx.Button(self, wx.ID_OK, 'OK', size=(_BOX_WIDTH, 25))
        self.close_button = wx.Button(self, wx.ID_CANCEL, 'Cancel', size=(_BOX_WIDTH, 25))

        # Intro
        explanation = "Select font properties :"
        vbox.Add(sizer)
        ix = 0
        iy = 1
        sizer.Add(wx.StaticText(self, -1, explanation), (iy, ix),
                  (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
        family_box.Add(wx.StaticText(self, -1, 'Family :'), -1, 0)
        family_box.Add(self.font_family, -1, 0)
        family_box.Add((_BOX_WIDTH / 2, -1))
        family_box.Add(wx.StaticText(self, -1, 'Size :'), -1, 0)
        family_box.Add(self.font_size, -1, 0)
        if unit_box != None:
            family_box.Add((_BOX_WIDTH / 2, -1))
            family_box.Add(tick_label_text, -1, 0)
            family_box.Add(self.tick_label_check, -1, 0)
        style_box.Add(wx.StaticText(self, -1, 'Style :'), -1, 0)
        style_box.Add(self.font_style, -1, 0)
        style_box.Add((_BOX_WIDTH / 2, -1))
        style_box.Add(wx.StaticText(self, -1, 'Weight :'), -1, 0)
        style_box.Add(self.font_weight, -1, 0)
        style_box.Add((_BOX_WIDTH / 2, -1))
        style_box.Add(wx.StaticText(self, -1, 'Color :'), -1, 0)
        style_box.Add(self.font_color, -1, 0)
        font_box.Add(family_box, -1, 10)
        font_box.Add(style_box, -1, 10)
        iy += 1
        ix = 0
        sizer.Add(font_box, (iy, ix),
                  (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
        iy += 2
        ix = 0
        sizer.Add(wx.StaticText(self, -1, enter_text), (iy, ix),
                  (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
        text_box.Add((15, 10))
        text_box.Add(self.text_string)
        vbox.Add(text_box, 0, wx.EXPAND, 15)
        if unit_box != None:
            unit_box.Add(unit_text, -1, 0)
            unit_box.Add(self.unit_ctrl, -1, 0)
            vbox.Add((5, 5))
            vbox.Add(unit_box, 0, wx.LEFT, 15)

        vbox.Add((10, 10))
        vbox.Add(self.static_line_1, 0, wx.EXPAND, 10)
        sizer_button = wx.BoxSizer(wx.HORIZONTAL)
        sizer_button.Add((20, 20), 1, wx.EXPAND | wx.ADJUST_MINSIZE, 0)
        sizer_button.Add(self.ok_button, 0,
                         wx.LEFT | wx.RIGHT | wx.ADJUST_MINSIZE, 10)
        sizer_button.Add(self.close_button, 0,
                         wx.LEFT | wx.RIGHT | wx.ADJUST_MINSIZE, 10)
        vbox.Add(sizer_button, 0, wx.EXPAND | wx.BOTTOM | wx.TOP, 10)
        self.SetSizer(vbox)
        self.Centre()

    def _set_family_list(self):
        """
        Set the list of font family
        """
        # list of family choices
        for idx in range(len(FAMILY)):
            self.font_family.Append(FAMILY[idx], idx)

    def _set_size_list(self):
        """
        Set the list of font size
        """
        # list of size choices
        for idx in range(len(SIZE)):
            self.font_size.Append(str(SIZE[idx]), idx)

    def _set_weight_list(self):
        """
        Set the list of font weight
        """
        # list of weight choices
        for idx in range(len(WEIGHT)):
            self.font_weight.Append(WEIGHT[idx], idx)

    def _set_style_list(self):
        """
        Set the list of font style
        """
        # list of style choices
        for idx in range(len(STYLE)):
            self.font_style.Append(STYLE[idx], idx)

    def _set_color_list(self):
        """
        Set the list of font color
        """
        # list of tyle choices
        for idx in range(len(COLOR)):
            self.font_color.Append(COLOR[idx], idx)

    def on_tick_label(self, event):
        """
        Set the font for tick label
        """
        event.Skip()
        self.tick_label = self.tick_label_check.GetValue()

    def on_family(self, event):
        """
        Set the family
        """
        event.Skip()
        self.family = self.font_family.GetValue()

    def on_style(self, event):
        """
        Set the style
        """
        event.Skip()
        self.style = self.font_style.GetValue()

    def on_weight(self, event):
        """
        Set the weight
        """
        event.Skip()
        self.weight = self.font_weight.GetValue()

    def on_size(self, event):
        """
        Set the size
        """
        event.Skip()
        self.size = self.font_size.GetValue()

    def on_color(self, event):
        """
        Set the color
        """
        event.Skip()
        self.color = self.font_color.GetValue()

    def getText(self):
        """
        Returns text string as input by user.
        """
        return self.text_string.GetValue()

    def getUnit(self):
        """
        Returns unit string as input by user.
        """
        return self.unit_ctrl.GetValue()

    def getFamily(self):
        """
        Returns font family for the text box
        """
        return str(self.family)

    def getStyle(self):
        """
        Returns font tyle for the text box
        """
        return str(self.style)

    def getWeight(self):
        """
        Returns font weight for the text box
        """
        return str(self.weight)

    def getSize(self):
        """
        Returns font size for the text box
        """
        return int(self.size)

    def getColor(self):
        """
        Returns font size for the text box
        """
        return str(self.color)

    def getTickLabel(self):
        """
        Bool for use on tick label
        """
        return self.tick_label