#
#   Albow - Fields
#

from pygame import draw
import pygame
from pygame.locals import K_LEFT, K_RIGHT, K_TAB, K_c, K_v, SCRAP_TEXT
from widget import Widget, overridable_property
from controls import Control

#---------------------------------------------------------------------------

class TextEditor(Widget):

    upper = False
    tab_stop = True
    
    _text = u""

    def __init__(self, width, upper = None, **kwds):
        Widget.__init__(self, **kwds)
        self.set_size_for_text(width)
        if upper is not None:
            self.upper = upper
        self.insertion_point = None
    
    def get_text(self):
        return self._text
    
    def set_text(self, text):
        self._text = text
    
    text = overridable_property('text')
    
    def draw(self, surface):
        frame = self.get_margin_rect()
        fg = self.fg_color
        font = self.font
        focused = self.has_focus()
        text, i = self.get_text_and_insertion_point()
        if focused and i is None:
            surface.fill(self.sel_color, frame)
        image = font.render(text, True, fg)
        surface.blit(image, frame)
        if focused and i is not None:
            x, h = font.size(text[:i])
            x += frame.left
            y = frame.top
            draw.line(surface, fg, (x, y), (x, y + h - 1))
    
    def key_down(self, event):
        if not (event.cmd or event.alt):
            k = event.key
            if k == K_LEFT:
                self.move_insertion_point(-1)
                return
            if k == K_RIGHT:
                self.move_insertion_point(1)
                return
            if k == K_TAB:
                self.attention_lost()
                self.tab_to_next()
                return
            try:
                c = event.unicode
            except ValueError:
                c = ""
            if self.insert_char(c) <> 'pass':
                return
        if event.cmd and event.unicode:
            if event.key == K_c:
                try:
                    pygame.scrap.put(SCRAP_TEXT, self.text)
                except:
                    print "scrap not available"
                    
            elif event.key == K_v:
                try:
                    t = pygame.scrap.get(SCRAP_TEXT).replace('\0', '')
                    self.text = t
                except:
                    print "scrap not available"
                #print repr(t)
            else:
                self.attention_lost()
                
        self.call_parent_handler('key_down', event)
    
    def get_text_and_insertion_point(self):
        text = self.get_text()
        i = self.insertion_point
        if i is not None:
            i = max(0, min(i, len(text)))
        return text, i
    
    def move_insertion_point(self, d):
        text, i = self.get_text_and_insertion_point()
        if i is None:
            if d > 0:
                i = len(text)
            else:
                i = 0
        else:
            i = max(0, min(i + d, len(text)))
        self.insertion_point = i
    
    def insert_char(self, c):
        if self.upper:
            c = c.upper()
        if c <= "\x7f":
            if c == "\x08" or c == "\x7f":
                text, i = self.get_text_and_insertion_point()
                if i is None:
                    text = ""
                    i = 0
                else:
                    text = text[:i-1] + text[i:]
                    i -= 1
                self.change_text(text)
                self.insertion_point = i
                return
            elif c == "\r" or c == "\x03":
                return self.call_handler('enter_action')
            elif c == "\x1b":
                return self.call_handler('escape_action')
            elif c >= "\x20":
                if self.allow_char(c):
                    text, i = self.get_text_and_insertion_point()
                    if i is None:
                        text = c
                        i = 1
                    else:
                        text = text[:i] + c + text[i:]
                        i += 1
                    self.change_text(text)
                    self.insertion_point = i
                    return
        return 'pass'
    
    def allow_char(self, c):
        return True

    def mouse_down(self, e):
        self.focus()
        if e.num_clicks == 2:
            self.insertion_point = None
            return
            
        x, y = e.local
        i = self.pos_to_index(x)
        self.insertion_point = i

    def pos_to_index(self, x):
        text = self.get_text()
        font = self.font
        def width(i):
            return font.size(text[:i])[0]
        i1 = 0
        i2 = len(text)
        x1 = 0
        x2 = width(i2)
        while i2 - i1 > 1:
            i3 = (i1 + i2) // 2
            x3 = width(i3)
            if x > x3:
                i1, x1 = i3, x3
            else:
                i2, x2 = i3, x3
        if x - x1 > (x2 - x1) // 2:
            i = i2
        else:
            i = i1
        
        return i
        
    def change_text(self, text):
        self.set_text(text)
        self.call_handler('change_action')

#---------------------------------------------------------------------------

class Field(Control, TextEditor):
    #  type      func(string) -> value
    #  editing   boolean

    empty = NotImplemented
    format = u"%s"
    min = None
    max = None
    enter_passes = False;
    
    def __init__(self, width = None, **kwds):
        min = self.predict_attr(kwds, 'min')
        max = self.predict_attr(kwds, 'max')
        if 'format' in kwds:
            self.format = kwds.pop('format')
        if 'empty' in kwds:
            self.empty = kwds.pop('empty')
        self.editing = False
        if width is None:
            w1 = w2 = ""
            if min is not None:
                w1 = self.format_value(min)
            if max is not None:
                w2 = self.format_value(max)
            if w2:
                if len(w1) > len(w2):
                    width = w1
                else:
                    width = w2
        if width is None:
            width = 100
        TextEditor.__init__(self, width, **kwds)

    def format_value(self, x):
        if x == self.empty:
            return ""
        else:
            return self.format % x

    def get_text(self):
        if self.editing:
            return self._text
        else:
            return self.format_value(self.value)
    
    def set_text(self, text):
        self.editing = True
        self._text = text
        if self.should_commit_immediately(text):
            self.commit();
    
    def should_commit_immediately(self, text):
        return False;
        
    def enter_action(self):
        if self.editing:
            self.commit()
        elif self.enter_passes:
            return 'pass'
    
    def escape_action(self):
        if self.editing:
            self.editing = False
            self.insertion_point = None
        else:
            return 'pass'
    
    def attention_lost(self):
        self.commit(notify=True)
    
    def clamp_value(self, value):
        if self.max is not None: value = min(value, self.max)
        if self.min is not None: value = max(value, self.min)
        return value
        
    def commit(self, notify = False):
        if self.editing:
            text = self._text
            if text:
                try:
                    value = self.type(text)
                except ValueError:
                    return
                value = self.clamp_value(value)
            else:
                value = self.empty
                if value is NotImplemented:
                    return
            self.value = value
            self.insertion_point = None
            if(notify):
                self.change_text(unicode(value))
            else:
                self._text = unicode(value)
            self.editing = False
            
        else:
            self.insertion_point = None

#    def get_value(self):
#        self.commit()
#        return Control.get_value(self)
#    
#    def set_value(self, x):
#        Control.set_value(self, x)
#        self.editing = False

#---------------------------------------------------------------------------

class TextField(Field):
    type = unicode
    _value = u""

class IntField(Field):
    tooltipText = "Point here and use mousewheel to adjust"
    def type(self, i):
        try:
            return eval(i);
        except:
            try:
                return int(i);
            except:
                return 0
    
    _shift_increment = 16
    _increment = 1
    @property
    def increment(self):
        if key.get_mods() & KMOD_SHIFT:
            return self._shift_increment
        else:
            return self._increment
        return self._increment
        
    @increment.setter
    def increment(self, val):
        self._increment = val
        
    def decrease_value(self):
        self.value = self.clamp_value(self.value - self.increment)
 
    def increase_value(self):
        self.value = self.clamp_value(self.value + self.increment)
                 
    def mouse_down(self, evt):
        if evt.button == 5:
            self.decrease_value()
            
            self.change_text(str(self.value))
            
        elif evt.button == 4:
            self.increase_value()
            self.change_text(str(self.value))
            
        else: Field.mouse_down(self, evt)
        
    allowed_chars = '-+*/<>()0123456789';
    def allow_char(self, c):
        return c in self.allowed_chars
        
    def should_commit_immediately(self, text):
        try:
            return str(eval(text)) == text
        except:
            return False;
            
class TimeField(Field):
    allowed_chars = ':0123456789 APMapm';
    def format_value(self, hm):
        format = "%d:%02d"
        h,m = hm
        if h >= 12:
            h -= 12
            return format % (h or 12,m) + " PM"
        else:
            return format % (h or 12,m) + " AM"
            
    def allow_char(self, c):
        return c in self.allowed_chars
    
    def type(self, i):
        h,m = 0,0
        i = i.upper()
        
        pm = "PM" in i
        for a in "APM":
            i = i.replace(a, "")
            
        parts = i.split(":")
        
        if len(parts):
            h = int(parts[0])
        if len(parts)>1:
            m = int(parts[1])

        if pm and h<12: h += 12
        h %= 24
        m %= 60
        return (h,m)
        
    def mouse_down(self, evt):
        if evt.button == 5:
            delta = -1
        elif evt.button == 4:
            delta = 1
        else: 
            return Field.mouse_down(self, evt)
            
        (h,m) = self.value
        pos = self.pos_to_index(evt.local[0])
        if pos < 2:
            h += delta
        elif pos < 5:
            m += delta
        else:
            h = (h + 12) % 24
            
        self.value = (h,m)
            
    def set_value(self, v):
        h,m = v
        super(TimeField, self).set_value((h%24, m%60))
        
from pygame import key
from pygame.locals import KMOD_SHIFT
    
class FloatField(Field):
    type = float
    _increment = 1.0;
    _shift_increment = 16.0;
    tooltipText = "Point here and use mousewheel to adjust"
    
    allowed_chars = '-+.0123456789f';
    def allow_char(self, c):
        return c in self.allowed_chars
    
    @property
    def increment(self):
        if key.get_mods() & KMOD_SHIFT:
            return self._shift_increment
        return self._increment
    
    @increment.setter
    def increment(self, val):
        self._increment = self.clamp_value(val)
           
    def decrease_value(self):
        self.value = self.clamp_value(self.value - self.increment)
     
    def increase_value(self):
        self.value = self.clamp_value(self.value + self.increment)
    
    def mouse_down(self, evt):
        if evt.button == 5:
            self.decrease_value()
            
            self.change_text(str(self.value))
            
        elif evt.button == 4:
            self.increase_value()
            self.change_text(str(self.value))
            
        else: Field.mouse_down(self, evt)
    
#---------------------------------------------------------------------------
