# Copyright 2008 by Kate Scheppke and Wade Brainerd.  
# This file is part of Typing Turtle.
#
# Typing Turtle is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# 
# Typing Turtle is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with Typing Turtle.  If not, see <http://www.gnu.org/licenses/>.

import random, datetime
from gettext import gettext as _

import gobject, pygtk, gtk, pango, time

import medalscreen

KITE_SIZE = 300

class KiteGame(gtk.VBox):
    def __init__(self, lesson, activity):
        gtk.VBox.__init__(self)
        
        self.lesson = lesson
        self.activity = activity
        
        # Build title bar.
        title = gtk.Label()
        title.set_markup("<span size='20000'><b>" + lesson['name'] + "</b></span>")
        title.set_alignment(1.0, 0.0)
        
        stoplabel = gtk.Label(_('Go Back'))
        stopbtn =  gtk.Button()
        stopbtn.add(stoplabel)
        stopbtn.connect('clicked', self.stop_cb)
        
        hbox = gtk.HBox()
        hbox.pack_start(stopbtn, False, False, 10)
        hbox.pack_end(title, False, False, 10)
        
        # Build the game drawing area.
        self.area = gtk.DrawingArea()
        self.area.connect("expose-event", self.expose_cb)

        # Connect keyboard grabbing and releasing callbacks.        
        self.area.connect('realize', self.realize_cb)
        self.area.connect('unrealize', self.unrealize_cb)

        self.pack_start(hbox, False, False, 10)
        self.pack_start(self.area, True, True)
        
        self.show_all()
        
        # Initialize the game data.
        self.text = ''
        for i in range(0, self.lesson['length']):
            if i > 0: self.text += ' '
            self.text += random.choice(self.lesson['words'])

        self.key_hist = []

        self.kitex = 150 
        self.kitey = None 
        self.kitevx = 0

        self.wpm = 0
        self.correct_time = 0

        self.score = 0

        self.medal = None
        self.finished = False

        # Start the animation loop running.        
        self.update_timer = gobject.timeout_add(20, self.tick, priority=gobject.PRIORITY_HIGH_IDLE+30)
    
    def realize_cb(self, widget):
        self.activity.add_events(gtk.gdk.KEY_PRESS_MASK)
        self.key_press_cb_id = self.activity.connect('key-press-event', self.key_cb)
        
    def unrealize_cb(self, widget):
        self.activity.disconnect(self.key_press_cb_id)
    
    def stop_cb(self, widget):
        # Stop the animation loop.
        if self.update_timer:
            gobject.source_remove(self.update_timer)
        
        self.activity.pop_screen()

    def key_cb(self, widget, event):
        # Ignore hotkeys.
        if event.state & (gtk.gdk.CONTROL_MASK | gtk.gdk.MOD1_MASK):
            return False

        # Extract information about the key pressed.
        key = gtk.gdk.keyval_to_unicode(event.keyval)
        if key != 0: key = unichr(key)

        if self.finished:
            key_name = gtk.gdk.keyval_name(event.keyval)
            if key_name == 'Return':
                self.activity.pop_screen()

                # Show the new medal if there was one.
                if self.medal:
                    self.activity.push_screen(medalscreen.MedalScreen(self.medal, self.activity))

        else:
            # Automatically consume spaces when the next correct letter is typed.
            correct = False
            if self.text[0] == ' ' and len(self.text) > 1 and key == unicode(self.text[1]):
                self.text = self.text[1:]
                correct = True
            elif key == unicode(self.text[0]):
                correct = True

            if correct:
                self.text = self.text[1:]
                self.queue_draw_text()
                self.add_score(10)
                if len(self.text) == 0:
                    self.finish_game()
                self.correct_time = time.time()

            else:
                self.add_score(-10)

            self.key_hist.insert(0, (time.time(), correct))
            self.key_hist = self.key_hist[:5]

        return False
    
    def tick(self):
        if self.finished:
            return

        correct_keys = 0
        total_keys = len(self.key_hist) 

        t = time.time()
        avg_key_time = 0
        for h in self.key_hist[1:]:
            if h[1]:
                correct_keys += 1
            avg_key_time += t - h[0]
            t = h[0]

        if total_keys > 0:
            avg_key_time = avg_key_time / float(total_keys)
        if avg_key_time > 0:
            wpm = 12.0 / avg_key_time 
        else:
            wpm = 0

        wpm = self.wpm * 0.99 + wpm * 0.01
        if int(wpm) != int(self.wpm):
            self.queue_draw_score()
        self.wpm = wpm

        # Erase old kite.
        self.queue_draw_kite()

        progress = float(total_keys) / 20.0

        pct = wpm / 50.0

        oldkitey = self.kitey
        newkitey = (self.bounds.height - 50 - KITE_SIZE/2) * (1.0 - pct)
        if self.kitey is None:
            self.kitey = newkitey
        else:
            self.kitey += (newkitey - self.kitey) * 0.1
            self.kitey += (self.kitey - oldkitey) * 0.5

        if total_keys > 0:
            acc = float(correct_keys) / total_keys
        else:
            acc = 0
        acc = acc * progress

        oldkitex = self.kitex
        newkitex = 150 + acc * self.bounds.width*0.3
        if self.kitex is None:
            self.kitex = newkitex
        else:
            self.kitex += (newkitex - self.kitex) * 0.01
            self.kitex += (self.kitex - oldkitex) * 0.1

        # Draw new kite.
        self.queue_draw_kite()

        return True

    def draw_results(self, gc):
        # Draw background.
        w = self.bounds.width - 400
        h = self.bounds.height - 200
        x = self.bounds.width/2 - w/2
        y = self.bounds.height/2 - h/2

        gc.foreground = self.area.get_colormap().alloc_color(50000,50000,50000)
        self.area.window.draw_rectangle(gc, True, x, y, w, h)
        gc.foreground = self.area.get_colormap().alloc_color(0,0,0)
        self.area.window.draw_rectangle(gc, False, x, y, w, h)

        # Draw text
        gc.foreground = self.area.get_colormap().alloc_color(0,0,0)

        title = _('You finished!') + '\n'
        layout = self.area.create_pango_layout(title)
        layout.set_font_description(pango.FontDescription('Serif Bold 16'))    
        size = layout.get_size()
        tx = x+w/2-(size[0]/pango.SCALE)/2
        ty = y + 100
        self.area.window.draw_layout(gc, tx, ty, layout)

        report = ''
        report += _('Your score was %(score)d.') % { 'score': self.score } + '\n'
        if self.medal:
            report += _('You earned a %(type)s medal!') % self.medal + '\n'
        report += '\n'
        report += _('Press the ENTER key to continue.')
    
        layout = self.area.create_pango_layout(report)
        layout.set_font_description(pango.FontDescription('Times 12'))    
        size = layout.get_size()
        tx = x+w/2-(size[0]/pango.SCALE)/2
        ty = y + 200
        self.area.window.draw_layout(gc, tx, ty, layout)

    def finish_game(self):
        self.finished = True

        # Add to the lesson history.
        report = { 
            'lesson': self.lesson['name'],
            'score': self.score,
        }
        self.activity.add_history(report)

        # Show the medal screen, if one should be given.
        got_medal = None
        
        medals = self.lesson['medals']
        for medal in medals:
            if self.score >= medal['score']:
                got_medal = medal['name']
        
        if got_medal:
            # Award the medal.
            medal = {
                'lesson': self.lesson['name'],
                'type': got_medal,
                'date': datetime.date.today().strftime('%B %d, %Y'),
                'nick': self.activity.owner.props.nick,
                'score': self.score
            }
            self.medal = medal

            # Compare this medal with any existing medals for this lesson.
            # Only record the best one.
            add_medal = True
            if self.activity.data['medals'].has_key(self.lesson['name']):
                old_medal = self.activity.data['medals'][self.lesson['name']]

                order = ' '.join([m['name'] for m in medals])
                add_idx = order.index(medal['type'])
                old_idx = order.index(old_medal['type']) 

                if add_idx < old_idx:
                    add_medal = False
                elif add_idx == old_idx:
                    if medal['score'] < old_medal['score']:
                        add_medal = False
            
            if add_medal:
                self.activity.data['motd'] = 'newmedal'
                self.activity.data['medals'][self.lesson['name']] = medal
                
                # Refresh the main screen given the new medal.
                self.activity.mainscreen.show_lesson(self.activity.mainscreen.lesson_index)

        self.queue_draw()

    def queue_draw_kite(self):
        if self.kitey is None:
            return

        x = int(self.kitex)
        y = int(self.kitey) - KITE_SIZE/2
        if y < 0:
            y = 0

        self.queue_draw_area(
            x-KITE_SIZE/2, y,
            x+KITE_SIZE, y+KITE_SIZE*2)

    def draw_kite(self, gc):
        if self.kitey is None or self.kitex is None:
            return

        color = (65535, 0 ,0)
 
        p0 = (int(self.kitex), int(self.kitey-KITE_SIZE*0.3))
        p1 = (int(self.kitex+KITE_SIZE*0.3), int(self.kitey))
        p2 = (int(self.kitex), int(self.kitey+KITE_SIZE*0.45))
        p3 = (int(self.kitex-KITE_SIZE*0.3), int(self.kitey))
        pts = [ p0, p1, p2, p3 ]

        gc.foreground = self.area.get_colormap().alloc_color(color[0],color[1],color[2])
        self.area.window.draw_polygon(gc, True, pts)
    
    def add_score(self, num):
        self.score += num
        self.queue_draw_score()

    def get_score_text(self):
        return _("SCORE: %d") % self.score + "\n" + \
               _("WPM: %d") % int(self.wpm)

    def queue_draw_score(self):
        layout = self.area.create_pango_layout(self.get_score_text())
        layout.set_font_description(pango.FontDescription('Times 14'))    
        size = layout.get_size()
        x = self.bounds.width-20-size[0]/pango.SCALE
        y = 20
        self.queue_draw_area(x, y, x+size[0], y+size[1])

    def draw_score(self, gc):
        layout = self.area.create_pango_layout(self.get_score_text())
        layout.set_font_description(pango.FontDescription('Times 14'))    
        size = layout.get_size()
        x = self.bounds.width-20-size[0]/pango.SCALE
        y = 20
        self.area.window.draw_layout(gc, x, y, layout)

#    def draw_instructions(self, gc):
#        gc.foreground = self.area.get_colormap().alloc_color(0,0,0)
#
#        layout = self.area.create_pango_layout(_('Type the words to fly the kite!'))
#        layout.set_font_description(pango.FontDescription('Times 14'))    
#        size = layout.get_size()
#        x = (self.bounds.width - size[0]/pango.SCALE)/2
#        y = self.bounds.height-20 - size[1]/pango.SCALE 
#        self.area.window.draw_layout(gc, x, y, layout)

    def queue_draw_text(self):
        layout = self.area.create_pango_layout(self.text[:50])
        layout.set_font_description(pango.FontDescription('Times 14'))    
        size = layout.get_size()
        height = 20 + size[1] / pango.SCALE
        self.queue_draw_area(
            0, self.bounds.height - height, 
            self.bounds.width, self.bounds.height)

    def draw_text(self, gc):
        gc.foreground = self.area.get_colormap().alloc_color(0,0,0)

        layout = self.area.create_pango_layout(self.text[:50])
        layout.set_font_description(pango.FontDescription('Times 14'))    
        size = layout.get_size()
        x = 20
        y = self.bounds.height-20 - size[1]/pango.SCALE 
        self.area.window.draw_layout(gc, x, y, layout)

        self.area.window.draw_line(gc, x, y, x+self.bounds.width, y)

    def draw(self):
        self.bounds = self.area.get_allocation()

        gc = self.area.window.new_gc()
        
        gc.foreground = self.area.get_colormap().alloc_color(60000,60000,65535)
        self.area.window.draw_rectangle(gc, True, 0, 0, self.bounds.width, self.bounds.height)

        if self.finished:
            self.draw_results(gc)

        else:
#            self.draw_instructions(gc)
            self.draw_kite(gc)
            self.draw_text(gc)
            self.draw_score(gc)

    def expose_cb(self, area, event):
        self.draw()