add Cake widget (copied from branch escalable)

3 files changed, 144 insertions, 0 deletions

diff --git a/fracciones.activity/bg.png b/fracciones.activity/bg.png
new file mode 100644
index 0000000..d99b493
--- /dev/null
+++ b/fracciones.activity/bg.png
diff --git a/fracciones.activity/fg.png b/fracciones.activity/fg.png
new file mode 100644
index 0000000..67fbb65
--- /dev/null
+++ b/fracciones.activity/fg.png
diff --git a/fracciones.activity/gtkcake.py b/fracciones.activity/gtkcake.py
new file mode 100755
index 0000000..f2e14a3
--- /dev/null
+++ b/fracciones.activity/gtkcake.py
@@ -0,0 +1,144 @@
+#!/usr/bin/env python
+# -*- encoding: utf8 -*-
+"""
+La torta es una imagen redonda en un archivo de 500x500 pixeles.
+Tiene un borde de 30 pixeles.
+
+Pasado a coordenadas mundiales el radio es: (500-2*30)/(2*500) = 0.44
+
+"""
+import math
+import gtk
+import cairo
+
+
+WIDTH = 500
+HEIGHT = 500
+RADIUS = 220
+WRADIUS = 0.44
+
+
+class Cake(gtk.DrawingArea):
+    """Widget que dibuja una torta y permite seleccionar trozos de ella"""
+
+    def __init__(self, N):
+        gtk.DrawingArea.__init__(self)
+        self.connect("expose_event", self.expose)
+        self.connect("button_press_event", self.button_press)
+        # Los eventos del raton no estan activados para el DrawingArea
+        self.add_events(gtk.gdk.BUTTON_PRESS_MASK)
+        self.N = N
+        self.selected = N * [0]
+
+
+    def expose(self, widget, event):
+        """Manejador del evento expose_event"""
+        context = widget.window.cairo_create()
+        # Elegimos solo la region expuesta al evento
+        context.rectangle(event.area.x, event.area.y, event.area.width, event.area.height)
+        context.clip()
+        self.draw(context)
+
+
+    def button_press(self, widget, event):
+        """Manejador del evento button_press_event"""
+        x, y = event.get_coords()
+        if self.select(x, y):
+            self.draw(widget.window.cairo_create())
+        
+
+    def draw(self, context):
+        """Dibuja el contenido del widget"""
+
+        def draw_grid(context, N):
+            """Dibuja la rejilla de la torta y sus subdivisiones"""
+            context.set_source_rgb(0, 0, 0)
+            context.arc(WIDTH/2, HEIGHT/2, RADIUS, 0, 2 * math.pi)
+            context.stroke()
+            for i in xrange(N):
+                angle = 2 * math.pi * i / N
+                context.move_to(WIDTH/2, HEIGHT/2)
+                context.line_to(
+                    WIDTH/2 + RADIUS*math.cos(angle),
+                    HEIGHT/2 + RADIUS*math.sin(angle)
+                    )
+                context.stroke()
+
+
+        def mask_image(context, selected, image):
+            """Enmascara la imagen de la torta y dibuja solo los trozos que no
+            fueron seleccionados
+            """
+            N = len(selected)
+            image_ctx = cairo.Context(image)
+            image_ctx.set_operator(cairo.OPERATOR_CLEAR)
+            for sector in xrange(N):
+                if selected[sector]:
+                    angle_start = 2 * math.pi * sector / N
+                    angle_end = 2 * math.pi * (sector + 1) / N
+                    image_ctx.move_to(WIDTH/2, HEIGHT/2)
+                    image_ctx.line_to(
+                        WIDTH/2 + RADIUS*math.cos(angle_start),
+                        HEIGHT/2 + RADIUS*math.sin(angle_start)
+                        )
+                    image_ctx.arc(WIDTH/2, HEIGHT/2, RADIUS, angle_start,
+                        angle_end)
+                    image_ctx.close_path()
+                    image_ctx.fill()
+            context.set_source_surface(image)
+            context.paint()
+
+
+        # Carga imagenes
+        image_bg = cairo.ImageSurface.create_from_png("bg.png")
+        image_fg = cairo.ImageSurface.create_from_png("fg.png")
+
+        # Escala la imagen al tamaño de la superficie
+        rect = self.get_allocation()
+        context.save()
+        context.scale(
+            float(rect.width) / WIDTH,
+            float(rect.height) / HEIGHT
+            )
+
+        # Dibuja el fondo
+        context.set_source_surface(image_bg, 0, 0)
+        context.paint()
+
+        # Dibuja el frente 
+        mask_image(context, self.selected, image_fg)
+
+        # Dibuja la rejilla
+        draw_grid(context, self.N)
+        context.restore()
+
+
+    def select(self, ux, uy):
+        """Selecciona un trozo de la torta. Devuelve True si un objeto fue
+        seleccionado, False en caso contrario."""
+        # Transformamos las coordenadas del usuario a coordenadas reales (0-1)
+        rect = self.get_allocation()
+        wx = float(ux - rect.x) / float(rect.width)
+        wy = float(uy - rect.y) / float(rect.height)
+        # Centramos
+        wx -= 0.5
+        wy -= 0.5
+        if math.pow(wx, 2) + math.pow(wy, 2) > math.pow(WRADIUS, 2):
+            return False
+        angle = math.atan2(wy, wx)
+        if angle < 0:
+            angle += 2 * math.pi
+        sector = angle * self.N / (2 * math.pi)
+        index = int(math.floor(sector))
+        self.selected[index] = 1 - self.selected[index]
+        return True
+
+
+if __name__ == "__main__":
+    # Probamos el nuevo widget Cake
+    window = gtk.Window()
+    cake = Cake(6)
+    window.add(cake)
+    window.connect("destroy", gtk.main_quit)
+    window.show_all()
+    gtk.main()