From 49d269dc1ab163ff6b1af250649158bab420f6ed Mon Sep 17 00:00:00 2001
From: Ignacio Rodríguez <ignacio@Ignacio-Magallanes.(none)>
Date: Fri, 26 Oct 2012 14:35:26 +0000
Subject:  Port finish, But I need Summary!

---
(limited to 'sprites.py')

diff --git a/sprites.py b/sprites.py
index baae94f..0ef8dca 100644
--- a/sprites.py
+++ b/sprites.py
@@ -73,9 +73,12 @@ def svg_str_to_pixbuf(svg_string):
     pl.close()
     pixbuf = pl.get_pixbuf()
     return pixbuf
+'''
+## <-----------GTK2---------------->##
 
 '''
 
+
 import pygtk
 pygtk.require('2.0')
 import gtk
@@ -84,35 +87,35 @@ import pangocairo
 import cairo
 
 class Sprites:
-    ''' A class for the list of sprites and everything they share in common '''
+    # A class for the list of sprites and everything they share in common #
 
     def __init__(self, widget):
-        ''' Initialize an empty array of sprites '''
+        # Initialize an empty array of sprites #
         self.widget = widget
         self.list = []
         self.cr = None
 
     def set_cairo_context(self, cr):
-        ''' Cairo context may be set or reset after __init__ '''
+        # Cairo context may be set or reset after __init__ #
         self.cr = cr
 
     def get_sprite(self, i):
-        ''' Return a sprint from the array '''
+        # Return a sprint from the array #
         if i < 0 or i > len(self.list) - 1:
             return(None)
         else:
             return(self.list[i])
 
     def length_of_list(self):
-        ''' How many sprites are there? '''
+        # How many sprites are there? #
         return(len(self.list))
 
     def append_to_list(self, spr):
-        ''' Append a new sprite to the end of the list. '''
+        # Append a new sprite to the end of the list. #
         self.list.append(spr)
 
     def insert_in_list(self, spr, i):
-        ''' Insert a sprite at position i. '''
+        # Insert a sprite at position i. #
         if i < 0:
             self.list.insert(0, spr)
         elif i > len(self.list) - 1:
@@ -121,21 +124,22 @@ class Sprites:
             self.list.insert(i, spr)
 
     def remove_from_list(self, spr):
-        ''' Remove a sprite from the list. '''
+        # Remove a sprite from the list. #
         if spr in self.list:
             self.list.remove(spr)
 
-    def find_sprite(self, pos):
-        ''' Search based on (x, y) position. Return the 'top/first' one. '''
+    def find_sprite(self, pos, inverse=False):
+        # Search based on (x, y) position. Return the 'top/first' one. #
         list = self.list[:]
-        list.reverse()
+        if not inverse:
+            list.reverse()
         for spr in list:
             if spr.hit(pos):
                 return spr
         return None
 
     def redraw_sprites(self, area=None, cr=None):
-        ''' Redraw the sprites that intersect area. '''
+        # Redraw the sprites that intersect area. #
         # I think I need to do this to save Cairo some work
         if cr is None:
             cr = self.cr
@@ -154,10 +158,10 @@ class Sprites:
 
 
 class Sprite:
-    ''' A class for the individual sprites '''
+    # A class for the individual sprites #
 
     def __init__(self, sprites, x, y, image):
-        ''' Initialize an individual sprite '''
+        # Initialize an individual sprite #
         self._sprites = sprites
         self.save_xy = (x, y)  # remember initial (x, y) position
         self.rect = gtk.gdk.Rectangle(int(x), int(y), 0, 0)
@@ -180,14 +184,15 @@ class Sprite:
         self._sprites.append_to_list(self)
 
     def set_image(self, image, i=0, dx=0, dy=0):
-        ''' Add an image to the sprite. '''
+        # Add an image to the sprite. #
         while len(self.cached_surfaces) < i + 1:
             self.cached_surfaces.append(None)
             self._dx.append(0)
             self._dy.append(0)
         self._dx[i] = dx
         self._dy[i] = dy
-        if isinstance(image, gtk.gdk.Pixbuf):
+        if isinstance(image, gtk.gdk.Pixbuf) or \
+           isinstance(image, cairo.ImageSurface):
             w = image.get_width()
             h = image.get_height()
         else:
@@ -200,48 +205,51 @@ class Sprite:
                 self.rect.width = w + dx
             if h + dy > self.rect.height:
                 self.rect.height = h + dy
-        surface = cairo.ImageSurface(
-            cairo.FORMAT_ARGB32, self.rect.width, self.rect.height)
-        context = cairo.Context(surface)
-        context = gtk.gdk.CairoContext(context)
-        context.set_source_pixbuf(image, 0, 0)
-        context.rectangle(0, 0, self.rect.width, self.rect.height)
-        context.fill()
-        self.cached_surfaces[i] = surface
+        if isinstance(image, cairo.ImageSurface):
+            self.cached_surfaces[i] = image
+        else:  # Convert to Cairo surface
+            surface = cairo.ImageSurface(
+                cairo.FORMAT_ARGB32, self.rect.width, self.rect.height)
+            context = cairo.Context(surface)
+            context = gtk.gdk.CairoContext(context)
+            context.set_source_pixbuf(image, 0, 0)
+            context.rectangle(0, 0, self.rect.width, self.rect.height)
+            context.fill()
+            self.cached_surfaces[i] = surface
 
     def move(self, pos):
-        ''' Move to new (x, y) position '''
+        # Move to new (x, y) position #
         self.inval()
         self.rect.x, self.rect.y = int(pos[0]), int(pos[1])
         self.inval()
 
     def move_relative(self, pos):
-        ''' Move to new (x+dx, y+dy) position '''
+        # Move to new (x+dx, y+dy) position #
         self.inval()
         self.rect.x += int(pos[0])
         self.rect.y += int(pos[1])
         self.inval()
 
     def get_xy(self):
-        ''' Return current (x, y) position '''
+        # Return current (x, y) position #
         return (self.rect.x, self.rect.y)
 
     def get_dimensions(self):
-        ''' Return current size '''
+        # Return current size #
         return (self.rect.width, self.rect.height)
 
     def get_layer(self):
-        ''' Return current layer '''
+        # Return current layer #
         return self.layer
 
     def set_shape(self, image, i=0):
-        ''' Set the current image associated with the sprite '''
+        # Set the current image associated with the sprite #
         self.inval()
         self.set_image(image, i)
         self.inval()
 
     def set_layer(self, layer=None):
-        ''' Set the layer for a sprite '''
+        # Set the layer for a sprite #
         self._sprites.remove_from_list(self)
         if layer is not None:
             self.layer = layer
@@ -254,7 +262,7 @@ class Sprite:
         self.inval()
 
     def set_label(self, new_label, i=0):
-        ''' Set the label drawn on the sprite '''
+        # Set the label drawn on the sprite #
         self._extend_labels_array(i)
         if type(new_label) is str or type(new_label) is unicode:
             # pango doesn't like nulls
@@ -264,11 +272,11 @@ class Sprite:
         self.inval()
 
     def set_margins(self, l=0, t=0, r=0, b=0):
-        ''' Set the margins for drawing the label '''
+        # Set the margins for drawing the label #
         self._margins = [l, t, r, b]
 
     def _extend_labels_array(self, i):
-        ''' Append to the labels attribute list '''
+        # Append to the labels attribute list #
         if self._fd is None:
             self.set_font('Sans')
         if self._color is None:
@@ -281,11 +289,11 @@ class Sprite:
             self._vert_align.append(self._vert_align[0])
 
     def set_font(self, font):
-        ''' Set the font for a label '''
+        # Set the font for a label #
         self._fd = pango.FontDescription(font)
 
     def set_label_color(self, rgb):
-        ''' Set the font color for a label '''
+        # Set the font color for a label #
         COLORTABLE = {'black': '#000000', 'white': '#FFFFFF',
                       'red': '#FF0000', 'yellow': '#FFFF00',
                       'green': '#00FF00', 'cyan': '#00FFFF',
@@ -301,7 +309,7 @@ class Sprite:
 
     def set_label_attributes(self, scale, rescale=True, horiz_align="center",
                              vert_align="middle", i=0):
-        ''' Set the various label attributes '''
+        # Set the various label attributes #
         self._extend_labels_array(i)
         self._scale[i] = scale
         self._rescale[i] = rescale
@@ -309,23 +317,23 @@ class Sprite:
         self._vert_align[i] = vert_align
 
     def hide(self):
-        ''' Hide a sprite '''
+        # Hide a sprite #
         self.inval()
         self._sprites.remove_from_list(self)
 
     def restore(self):
-        ''' Restore a hidden sprite '''
+        # Restore a hidden sprite #
         self.set_layer()
 
     def inval(self):
-        ''' Invalidate a region for gtk '''
+        # Invalidate a region for gtk #
         self._sprites.widget.queue_draw_area(self.rect.x,
                                              self.rect.y,
                                              self.rect.width,
                                              self.rect.height)
 
     def draw(self, cr=None):
-        ''' Draw the sprite (and label) '''
+        # Draw the sprite (and label) #
         if cr is None:
             print 'sprite.draw: no Cairo context.'
             return
@@ -342,7 +350,7 @@ class Sprite:
             self.draw_label(cr)
 
     def hit(self, pos):
-        ''' Is (x, y) on top of the sprite? '''
+        # Is (x, y) on top of the sprite? #
         x, y = pos
         if x < self.rect.x:
             return False
@@ -355,7 +363,7 @@ class Sprite:
         return True
 
     def draw_label(self, cr):
-        ''' Draw the label based on its attributes '''
+        # Draw the label based on its attributes #
         # Create a pangocairo context
         cr = pangocairo.CairoContext(cr)
         my_width = self.rect.width - self._margins[0] - self._margins[2]
@@ -404,7 +412,7 @@ class Sprite:
             cr.restore()
 
     def label_width(self):
-        ''' Calculate the width of a label '''
+        # Calculate the width of a label #
         cr = pangocairo.CairoContext(self._sprites.cr)
         if cr is not None:
             max = 0
@@ -421,19 +429,19 @@ class Sprite:
             return self.rect.width
 
     def label_safe_width(self):
-        ''' Return maximum width for a label '''
+        # Return maximum width for a label #
         return self.rect.width - self._margins[0] - self._margins[2]
 
     def label_safe_height(self):
-        ''' Return maximum height for a label '''
+        # Return maximum height for a label #
         return self.rect.height - self._margins[1] - self._margins[3]
 
     def label_left_top(self):
-        ''' Return the upper-left corner of the label safe zone '''
+        # Return the upper-left corner of the label safe zone #
         return(self._margins[0], self._margins[1])
 
     def get_pixel(self, pos, i=0):
-        ''' Return the pixel at (x, y) '''
+        # Return the pixel at (x, y) #
         x = int(pos[0] - self.rect.x)
         y = int(pos[1] - self.rect.y)
         if x < 0 or x > (self.rect.width - 1) or \
@@ -452,3 +460,393 @@ class Sprite:
         pixels = cs.get_data()
         return (ord(pixels[2]), ord(pixels[1]), ord(pixels[0]), 0)
 
+'''
+
+#<-----------------------GTK3----------->#
+import gi
+from gi.repository import Gtk, GdkPixbuf, Gdk
+from gi.repository import Pango, PangoCairo
+import cairo
+
+
+class Sprites:
+    ''' A class for the list of sprites and everything they share in common '''
+
+    def __init__(self, widget):
+        ''' Initialize an empty array of sprites '''
+        self.cr = None
+        self.widget = widget
+        self.list = []
+
+    def set_cairo_context(self, cr):
+        ''' Cairo context may be set or reset after __init__ '''
+        self.cr = cr
+
+    def get_sprite(self, i):
+        ''' Return a sprint from the array '''
+        if i < 0 or i > len(self.list)-1:
+            return(None)
+        else:
+            return(self.list[i])
+
+    def length_of_list(self):
+        ''' How many sprites are there? '''
+        return(len(self.list))
+
+    def append_to_list(self, spr):
+        ''' Append a new sprite to the end of the list. '''
+        self.list.append(spr)
+
+    def insert_in_list(self, spr, i):
+        ''' Insert a sprite at position i. '''
+        if i < 0:
+            self.list.insert(0, spr)
+        elif i > len(self.list) - 1:
+            self.list.append(spr)
+        else:
+            self.list.insert(i, spr)
+
+    def remove_from_list(self, spr):
+        ''' Remove a sprite from the list. '''
+        if spr in self.list:
+            self.list.remove(spr)
+
+    def find_sprite(self, pos):
+        ''' Search based on (x, y) position. Return the 'top/first' one. '''
+        list = self.list[:]
+        list.reverse()
+        for spr in list:
+            if spr.hit(pos):
+                return spr
+        return None
+
+    def redraw_sprites(self, area=None, cr=None):
+        ''' Redraw the sprites that intersect area. '''
+        # I think I need to do this to save Cairo some work
+        if cr is None:
+            cr = self.cr
+        else:
+            self.cr = cr
+        if cr is None:
+            print 'sprites.redraw_sprites: no Cairo context'
+            return
+        for spr in self.list:
+            if area == None:
+                spr.draw(cr=cr)
+            else:
+                intersection = spr.rect.intersect(area)
+                if intersection.width > 0 or intersection.height > 0:
+                    spr.draw(cr=cr)
+
+
+class Sprite:
+    ''' A class for the individual sprites '''
+
+    def __init__(self, sprites, x, y, image):
+        ''' Initialize an individual sprite '''
+        self._sprites = sprites
+        self.save_xy = (x, y)  # remember initial (x, y) position
+        self.rect = [int(x), int(y), 0, 0]
+        self._scale = [12]
+        self._rescale = [True]
+        self._horiz_align = ["center"]
+        self._vert_align = ["middle"]
+        self._x_pos = [None]
+        self._y_pos = [None]
+        self._fd = None
+        self._bold = False
+        self._italic = False
+        self._color = None
+        self._margins = [0, 0, 0, 0]
+        self.layer = 100
+        self.labels = []
+        self.images = []
+        self._dx = []  # image offsets
+        self._dy = []
+        self.type = None
+        self.set_image(image)
+        self._sprites.append_to_list(self)
+
+    def set_image(self, image, i=0, dx=0, dy=0):
+        ''' Add an image to the sprite. '''
+        while len(self.images) < i + 1:
+            self.images.append(None)
+            self._dx.append(0)
+            self._dy.append(0)
+        self.images[i] = image
+        self._dx[i] = dx
+        self._dy[i] = dy
+        if hasattr(self.images[i], 'get_width'):
+            w = self.images[i].get_width()
+            h = self.images[i].get_height()
+        else:
+            w, h = self.images[i].get_size()
+        if i == 0:  # Always reset width and height when base image changes.
+            self.rect[2] = w + dx
+            self.rect[3] = h + dy
+        else:
+            if w + dx > self.rect[2]:
+                self.rect[2] = w + dx
+            if h + dy > self.rect[3]:
+                self.rect[3] = h + dy
+
+    def move(self, pos):
+        ''' Move to new (x, y) position '''
+        self.inval()
+        self.rect[0], self.rect[1] = int(pos[0]), int(pos[1])
+        self.inval()
+
+    def move_relative(self, pos):
+        ''' Move to new (x+dx, y+dy) position '''
+        self.inval()
+        self.rect[0] += int(pos[0])
+        self.rect[1] += int(pos[1])
+        self.inval()
+
+    def get_xy(self):
+        ''' Return current (x, y) position '''
+        return (self.rect[0], self.rect[1])
+
+    def get_dimensions(self):
+        ''' Return current size '''
+        return (self.rect[2], self.rect[3])
+
+    def get_layer(self):
+        ''' Return current layer '''
+        return self.layer
+
+    def set_shape(self, image, i=0):
+        ''' Set the current image associated with the sprite '''
+        self.inval()
+        self.set_image(image, i)
+        self.inval()
+
+    def set_layer(self, layer=None):
+        ''' Set the layer for a sprite '''
+        self._sprites.remove_from_list(self)
+        if layer is not None:
+            self.layer = layer
+        for i in range(self._sprites.length_of_list()):
+            if self.layer < self._sprites.get_sprite(i).layer:
+                self._sprites.insert_in_list(self, i)
+                self.inval()
+                return
+        self._sprites.append_to_list(self)
+        self.inval()
+
+    def set_label(self, new_label, i=0):
+        ''' Set the label drawn on the sprite '''
+        self._extend_labels_array(i)
+        if type(new_label) is str or type(new_label) is unicode:
+            # pango doesn't like nulls
+            self.labels[i] = new_label.replace("\0", " ")
+        else:
+            self.labels[i] = str(new_label)
+        self.inval()
+
+    def set_margins(self, l=0, t=0, r=0, b=0):
+        ''' Set the margins for drawing the label '''
+        self._margins = [l, t, r, b]
+
+    def _extend_labels_array(self, i):
+        ''' Append to the labels attribute list '''
+        if self._fd is None:
+            self.set_font('Sans')
+        if self._color is None:
+            self._color = (0., 0., 0.)
+        while len(self.labels) < i + 1:
+            self.labels.append(" ")
+            self._scale.append(self._scale[0])
+            self._rescale.append(self._rescale[0])
+            self._horiz_align.append(self._horiz_align[0])
+            self._vert_align.append(self._vert_align[0])
+            self._x_pos.append(self._x_pos[0])
+            self._y_pos.append(self._y_pos[0])
+
+    def set_font(self, font):
+        ''' Set the font for a label '''
+        self._fd = Pango.FontDescription(font)
+
+    def set_label_color(self, rgb):
+        ''' Set the font color for a label '''
+        COLORTABLE = {'black': '#000000', 'white': '#FFFFFF',
+                      'red': '#FF0000', 'yellow': '#FFFF00',
+                      'green': '#00FF00', 'cyan': '#00FFFF',
+                      'blue': '#0000FF', 'purple': '#FF00FF',
+                      'gray': '#808080'}
+        if rgb.lower() in COLORTABLE:
+            rgb = COLORTABLE[rgb.lower()]
+        # Convert from '#RRGGBB' to floats
+        self._color = (int('0x' + rgb[1:3], 16) / 256.,
+                       int('0x' + rgb[3:5], 16) / 256.,
+                       int('0x' + rgb[5:7], 16) / 256.)
+        return
+
+    def set_label_attributes(self, scale, rescale=True, horiz_align="center",
+                             vert_align="middle", x_pos=None, y_pos=None, i=0):
+        ''' Set the various label attributes '''
+        self._extend_labels_array(i)
+        self._scale[i] = scale
+        self._rescale[i] = rescale
+        self._horiz_align[i] = horiz_align
+        self._vert_align[i] = vert_align
+        self._x_pos[i] = x_pos
+        self._y_pos[i] = y_pos
+
+    def hide(self):
+        ''' Hide a sprite '''
+        self.inval()
+        self._sprites.remove_from_list(self)
+
+    def restore(self):
+        ''' Restore a hidden sprite '''
+        self.set_layer()
+
+    def inval(self):
+        ''' Invalidate a region for gtk '''
+        # self._sprites.window.invalidate_rect(self.rect, False)
+        self._sprites.widget.queue_draw_area(self.rect[0],
+                                             self.rect[1],
+                                             self.rect[2],
+                                             self.rect[3])
+
+    def draw(self, cr=None):
+        ''' Draw the sprite (and label) '''
+        if cr is None:
+            cr = self._sprites.cr
+        if cr is None:
+            print 'sprite.draw: no Cairo context.'
+            return
+        for i, img in enumerate(self.images):
+            if isinstance(img, GdkPixbuf.Pixbuf):
+                Gdk.cairo_set_source_pixbuf(cr, img,
+                                            self.rect[0] + self._dx[i],
+                                            self.rect[1] + self._dy[i])
+                cr.rectangle(self.rect[0] + self._dx[i],
+                             self.rect[1] + self._dy[i],
+                             self.rect[2],
+                             self.rect[3])
+                cr.fill()
+            elif type(img) == cairo.ImageSurface:
+                cr.set_source_surface(img, self.rect[0] + self._dx[i],
+                                      self.rect[1] + self._dy[i])
+                cr.rectangle(self.rect[0] + self._dx[i],
+                             self.rect[1] + self._dy[i],
+                             self.rect[2],
+                             self.rect[3])
+                cr.fill()
+            else:
+                print 'sprite.draw: source not a pixbuf (%s)' % (type(img))
+        if len(self.labels) > 0:
+            self.draw_label(cr)
+
+    def hit(self, pos):
+        ''' Is (x, y) on top of the sprite? '''
+        x, y = pos
+        if x < self.rect[0]:
+            return False
+        if x > self.rect[0] + self.rect[2]:
+            return False
+        if y < self.rect[1]:
+            return False
+        if y > self.rect[1] + self.rect[3]:
+            return False
+        return True
+
+    def draw_label(self, cr):
+        ''' Draw the label based on its attributes '''
+        my_width = self.rect[2] - self._margins[0] - self._margins[2]
+        if my_width < 0:
+            my_width = 0
+        my_height = self.rect[3] - self._margins[1] - self._margins[3]
+        for i in range(len(self.labels)):
+            pl = PangoCairo.create_layout(cr)
+            pl.set_text(str(self.labels[i]), -1)
+            self._fd.set_size(int(self._scale[i] * Pango.SCALE))
+            pl.set_font_description(self._fd)
+            w = pl.get_size()[0] / Pango.SCALE
+            if w > my_width:
+                if self._rescale[i]:
+                    self._fd.set_size(
+                            int(self._scale[i] * Pango.SCALE * my_width / w))
+                    pl.set_font_description(self._fd)
+                    w = pl.get_size()[0] / Pango.SCALE
+                else:
+                    j = len(self.labels[i]) - 1
+                    while(w > my_width and j > 0):
+                        pl.set_text(
+                            "…" + self.labels[i][len(self.labels[i]) - j:], -1)
+                        self._fd.set_size(int(self._scale[i] * Pango.SCALE))
+                        pl.set_font_description(self._fd)
+                        w = pl.get_size()[0] / Pango.SCALE
+                        j -= 1
+            if self._x_pos[i] is not None:
+                x = int(self.rect[0] + self._x_pos[i])
+            elif self._horiz_align[i] == "center":
+                x = int(self.rect[0] + self._margins[0] + (my_width - w) / 2)
+            elif self._horiz_align[i] == 'left':
+                x = int(self.rect[0] + self._margins[0])
+            else: # right
+                x = int(self.rect[0] + self.rect[2] - w - self._margins[2])
+            h = pl.get_size()[1] / Pango.SCALE
+            if self._y_pos[i] is not None:
+                y = int(self.rect[1] + self._y_pos[i])
+            elif self._vert_align[i] == "middle":
+                y = int(self.rect[1] + self._margins[1] + (my_height - h) / 2)
+            elif self._vert_align[i] == "top":
+                y = int(self.rect[1] + self._margins[1])
+            else: # bottom
+                y = int(self.rect[1] + self.rect[3] - h - self._margins[3])
+            cr.save()
+            cr.translate(x, y)
+            cr.set_source_rgb(self._color[0], self._color[1], self._color[2])
+            PangoCairo.update_layout(cr, pl)
+            PangoCairo.show_layout(cr, pl)
+            cr.restore()
+
+    def label_width(self, cr=None):
+        ''' Calculate the width of a label '''
+        if cr is None:
+            cr = self._sprites.cr
+        max = 0
+        for i in range(len(self.labels)):
+            pl = PangoCairo.create_layout(cr)
+            pl.set_text(str(self.labels[i]), -1)
+            self._fd.set_size(int(self._scale[i] * Pango.SCALE))
+            pl.set_font_description(self._fd)
+            w = pl.get_size()[0] / Pango.SCALE
+            if w > max:
+                max = w
+        return max
+
+    def label_safe_width(self):
+        ''' Return maximum width for a label '''
+        return self.rect[2] - self._margins[0] - self._margins[2]
+
+    def label_safe_height(self):
+        ''' Return maximum height for a label '''
+        return self.rect[3] - self._margins[1] - self._margins[3]
+
+    def label_left_top(self):
+        ''' Return the upper-left corner of the label safe zone '''
+        return(self._margins[0], self._margins[1])
+
+    def get_pixel(self, pos, i=0):
+        ''' Return the pixl at (x, y) '''
+        x, y = pos
+        x = x - self.rect[0]
+        y = y - self.rect[1]
+        if y > self.images[i].get_height() - 1:
+            return(-1, -1, -1, -1)
+        try:
+            array = self.images[i].get_pixels()
+            if array is not None:
+                offset = (y * self.images[i].get_width() + x) * 4
+                r, g, b, a = ord(array[offset]), ord(array[offset + 1]),\
+                             ord(array[offset + 2]), ord(array[offset + 3])
+                return(r, g, b, a)
+            else:
+                return(-1, -1, -1, -1)
+        except IndexError:
+            print "Index Error: %d %d" % (len(array), offset)
+            return(-1, -1, -1, -1)
--
cgit v0.9.1