# Implementation of the status bar.

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

from common import *

import mlx.const as const

import math

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

class Statusbar(gtk.Frame, FlightStatusHandler):
    """A status bar for the logger."""
    def __init__(self):
        """Construct the status bar."""
        gtk.Frame.__init__(self)
        FlightStatusHandler.__init__(self)

        self._connecting = False
        self._connected = False
        
        self.set_shadow_type(gtk.ShadowType.NONE if pygobject
                             else gtk.SHADOW_NONE)

        frameAlignment = gtk.Alignment(xscale = 1.0, yscale = 1.0)

        frameAlignment.set_padding(padding_top = 2, padding_bottom = 2,
                                   padding_left = 16, padding_right = 16)
        self.add(frameAlignment)

        statusBox = gtk.HBox()
        frameAlignment.add(statusBox)

        self._connStateArea = gtk.DrawingArea()
        self._connStateArea.set_size_request(16, 16)
        self._connStateArea.set_tooltip_markup('The state of the connection.\n'
                                               '<span foreground="grey">Grey</span> means idle.\n'
                                               '<span foreground="red">Red</span> means trying to connect.\n'
                                               '<span foreground="green">Green</span> means connected.')

        if pygobject:
            self._connStateArea.connect("draw", self._drawConnState)
        else:
            self._connStateArea.connect("expose_event", self._drawConnState)

        alignment = gtk.Alignment(xalign = 0.5, yalign = 0.5)
        alignment.add(self._connStateArea)        

        statusBox.pack_start(alignment, False, False, 8)

        statusBox.pack_start(gtk.VSeparator(), False, False, 8)

        self._stageLabel = gtk.Label()
        self._stageLabel.set_width_chars(20)
        self._stageLabel.set_tooltip_text("The flight stage")
        self._stageLabel.set_alignment(0.0, 0.5)
        
        statusBox.pack_start(self._stageLabel, False, False, 8)

        statusBox.pack_start(gtk.VSeparator(), False, False, 8)

        self._ratingLabel = gtk.Label()
        self._ratingLabel.set_width_chars(12)
        self._ratingLabel.set_tooltip_text("The flight rating")
        self._ratingLabel.set_alignment(0.0, 0.5)
        
        statusBox.pack_start(self._ratingLabel, False, False, 8)

        self._busyLabel = gtk.Label()
        self._busyLabel.set_width_chars(30)
        self._busyLabel.set_tooltip_text("The status of the background tasks.")
        self._busyLabel.set_alignment(1.0, 0.5)
        statusBox.pack_start(self._busyLabel, True, True, 8)
        
        self._updateFlightStatus()

    def updateConnection(self, connecting, connected):
        """Update the connection status."""
        self._connecting = connecting
        self._connected = connected
        self._connStateArea.queue_draw()

    def updateBusyState(self, message):
        """Update the busy state."""
        self._busyLabel.set_text("" if message is None else message)

    def _drawConnState(self, connStateArea, eventOrContext):
        """Draw the connection state."""        
        context = eventOrContext if pygobject else connStateArea.window.cairo_create()

        if self._connecting:
            if self._connected:
                context.set_source_rgb(0.0, 1.0, 0.0)
            else:
                context.set_source_rgb(1.0, 0.0, 0.0)
        else:
            context.set_source_rgb(0.75, 0.75, 0.75)

        width = connStateArea.get_allocated_width() if pygobject \
                else connStateArea.allocation.width
        height = connStateArea.get_allocated_height() if pygobject \
                 else connStateArea.allocation.height
        context.arc(width/2, height/2, width/2, 0, 2*math.pi)

        context.fill()

    def _updateFlightStatus(self):
        """Update the flight status information."""
        self._stageLabel.set_text("-" if self._stage is None
                                  else const.stage2string(self._stage).upper())
    
        if self._noGoReason is None:
            rating = "%.0f%%" % (self._rating,)
        else:
            rating = '<span foreground="red">' + self._noGoReason + '</span>'
        self._ratingLabel.set_markup(rating)