#Copyright (c) 2007-8, Playful Invention Company. #Copyright (c) 2008-10, Walter Bender #Permission is hereby granted, free of charge, to any person obtaining a copy #of this software and associated documentation files (the "Software"), to deal #in the Software without restriction, including without limitation the rights #to use, copy, modify, merge, publish, distribute, sublicense, and/or sell #copies of the Software, and to permit persons to whom the Software is #furnished to do so, subject to the following conditions: #The above copyright notice and this permission notice shall be included in #all copies or substantial portions of the Software. #THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR #IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, #FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE #AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER #LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, #OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN #THE SOFTWARE. import gtk from math import sin, cos, pi import pango import cairo from sprites import Sprite from tasprite_factory import SVG from tautils import image_to_base64, data_to_string, round_int from taconstants import CANVAS_LAYER, BLACK, WHITE import logging _logger = logging.getLogger('turtleart-activity') def wrap100(n): """ A variant on mod... 101 -> 99; 199 -> 1 """ n = int(n) n %= 200 if n > 99: n = 199 - n return n def calc_shade(c, s, invert=False): """ Convert a color to the current shade (lightness/darkness). """ # Assumes 16 bit input values if invert: if s < 0: return int(c / (1 + s * 0.8)) return int((c - 65536 * s * 0.9) / (1 - (s * 0.9))) else: if s < 0: return int(c * (1 + s * 0.8)) return int(c + (65536 - c) * s * 0.9) def calc_gray(c, g, invert=False): """ Gray is a psuedo saturation calculation. """ # Assumes 16 bit input values if g == 100: return c if invert: return int(((c * 100) - (32768 * (100 - g))) / g) else: return int(((c * g) + (32768 * (100 - g))) / 100) colors = {} DEGTOR = 2 * pi / 360 color_table = ( 0xFF0000, 0xFF0D00, 0xFF1A00, 0xFF2600, 0xFF3300, 0xFF4000, 0xFF4D00, 0xFF5900, 0xFF6600, 0xFF7300, 0xFF8000, 0xFF8C00, 0xFF9900, 0xFFA600, 0xFFB300, 0xFFBF00, 0xFFCC00, 0xFFD900, 0xFFE600, 0xFFF200, 0xFFFF00, 0xE6FF00, 0xCCFF00, 0xB3FF00, 0x99FF00, 0x80FF00, 0x66FF00, 0x4DFF00, 0x33FF00, 0x1AFF00, 0x00FF00, 0x00FF0D, 0x00FF1A, 0x00FF26, 0x00FF33, 0x00FF40, 0x00FF4D, 0x00FF59, 0x00FF66, 0x00FF73, 0x00FF80, 0x00FF8C, 0x00FF99, 0x00FFA6, 0x00FFB3, 0x00FFBF, 0x00FFCC, 0x00FFD9, 0x00FFE6, 0x00FFF2, 0x00FFFF, 0x00F2FF, 0x00E6FF, 0x00D9FF, 0x00CCFF, 0x00BFFF, 0x00B3FF, 0x00A6FF, 0x0099FF, 0x008CFF, 0x0080FF, 0x0073FF, 0x0066FF, 0x0059FF, 0x004DFF, 0x0040FF, 0x0033FF, 0x0026FF, 0x001AFF, 0x000DFF, 0x0000FF, 0x0D00FF, 0x1A00FF, 0x2600FF, 0x3300FF, 0x4000FF, 0x4D00FF, 0x5900FF, 0x6600FF, 0x7300FF, 0x8000FF, 0x8C00FF, 0x9900FF, 0xA600FF, 0xB300FF, 0xBF00FF, 0xCC00FF, 0xD900FF, 0xE600FF, 0xF200FF, 0xFF00FF, 0xFF00E6, 0xFF00CC, 0xFF00B3, 0xFF0099, 0xFF0080, 0xFF0066, 0xFF004D, 0xFF0033, 0xFF001A) class TurtleGraphics: """ A class for the Turtle graphics canvas """ def __init__(self, tw, width, height): """ Create a sprite to hold the canvas. """ self.tw = tw self.width = width self.height = height if self.tw.interactive_mode: self.canvas = Sprite(tw.sprite_list, 0, 0, gtk.gdk.Pixmap(self.tw.area, self.width * 2, self.height * 2, -1)) else: self.canvas = Sprite(None, 0, 0, self.tw.window) self.canvas.set_layer(CANVAS_LAYER) (self.cx, self.cy) = self.canvas.get_xy() self.canvas.type = 'canvas' self.gc = self.canvas.images[0].new_gc() self.cm = self.gc.get_colormap() self.fgrgb = [255, 0, 0] self.fgcolor = self.cm.alloc_color('red') self.bgrgb = [255, 248, 222] self.bgcolor = self.cm.alloc_color('#fff8de') self.textsize = 48 # depreciated self.textcolor = self.cm.alloc_color('blue') self.tw.active_turtle.show() self.shade = 0 self.pendown = False self.xcor = 0 self.ycor = 0 self.heading = 0 self.pensize = 5 self.tcolor = 0 self.color = 0 self.gray = 100 self.fill = False self.poly_points = [] self.svg = SVG() self.svg.set_fill_color('none') self.tw.svg_string = '' self.clearscreen(False) def start_fill(self): """ Start accumulating points of a polygon to fill. """ self.fill = True self.poly_points = [] def stop_fill(self): """ Fill the polygon. """ self.fill = False if len(self.poly_points) == 0: return minx = self.poly_points[0][0] miny = self.poly_points[0][1] maxx = minx maxy = miny for p in self.poly_points: if p[0] < minx: minx = p[0] elif p[0] > maxx: maxx = p[0] if p[1] < miny: miny = p[1] elif p[1] > maxy: maxy = p[1] w = maxx - minx h = maxy - miny self.canvas.images[0].draw_polygon(self.gc, True, self.poly_points) self.invalt(minx - self.pensize * self.tw.coord_scale / 2 - 3, miny - self.pensize * self.tw.coord_scale / 2 - 3, w + self.pensize * self.tw.coord_scale + 6, h + self.pensize * self.tw.coord_scale + 6) self.poly_points = [] def clearscreen(self, share=True): """Clear the canvas and reset most graphics attributes to defaults.""" rect = gtk.gdk.Rectangle(0, 0, self.width, self.height) self.gc.set_foreground(self.bgcolor) self.canvas.images[0].draw_rectangle(self.gc, True, *rect) self.invalt(0, 0, self.width, self.height) self.setpensize(5, share) self.setgray(100, share) self.setcolor(0, share) self.settextcolor(70) self.setshade(50, share) for turtle_key in iter(self.tw.turtles.dict): self.set_turtle(turtle_key) self.tw.active_turtle.set_color(0) self.tw.active_turtle.set_shade(50) self.tw.active_turtle.set_gray(100) self.tw.active_turtle.set_pen_size(5) self.tw.active_turtle.reset_shapes() self.seth(0, share) self.setpen(False, share) self.setxy(0, 0, share) self.setpen(True, share) self.tw.active_turtle.hide() self.set_turtle(self.tw.default_turtle_name) self.tw.svg_string = '' self.svg.reset_min_max() self.fill = False self.poly_points = [] def forward(self, n, share=True): """ Move the turtle forward.""" nn = n * self.tw.coord_scale self.gc.set_foreground(self.fgcolor) oldx, oldy = self.xcor, self.ycor try: self.xcor += nn * sin(self.heading * DEGTOR) self.ycor += nn * cos(self.heading * DEGTOR) except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return if self.pendown: self.draw_line(oldx, oldy, self.xcor, self.ycor) self.move_turtle() if self.tw.saving_svg and self.pendown: self.tw.svg_string += self.svg.new_path(oldx, self.height / 2 - oldy) self.tw.svg_string += self.svg.line_to(self.xcor, self.height / 2 - self.ycor) self.tw.svg_string += "\"\n" self.tw.svg_string += self.svg.style() if self.tw.sharing() and share: self.tw.activity.send_event("f|%s" % \ (data_to_string([self.tw.nick, int(n)]))) def seth(self, n, share=True): """ Set the turtle heading. """ try: self.heading = n except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return self.heading %= 360 self.turn_turtle() if self.tw.sharing() and share: self.tw.activity.send_event("r|%s" % \ (data_to_string([self.tw.nick, round_int(self.heading)]))) def right(self, n, share=True): """ Rotate turtle clockwise """ try: self.heading += n except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return self.heading %= 360 self.turn_turtle() if self.tw.sharing() and share: self.tw.activity.send_event("r|%s" % \ (data_to_string([self.tw.nick, round_int(self.heading)]))) def arc(self, a, r, share=True): """ Draw an arc """ self.gc.set_foreground(self.fgcolor) rr = r * self.tw.coord_scale try: if a < 0: self.larc(-a, rr) else: self.rarc(a, rr) except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return self.move_turtle() if self.tw.sharing() and share: self.tw.activity.send_event("a|%s" % \ (data_to_string([self.tw.nick, [round_int(a), round_int(r)]]))) def rarc(self, a, r): """ draw a clockwise arc """ if r < 0: r = -r a = -a s = 0 else: s = 1 oldx, oldy = self.xcor, self.ycor cx = self.xcor + r * cos(self.heading * DEGTOR) cy = self.ycor - r * sin(self.heading * DEGTOR) x = self.width / 2 + int(cx - r) y = self.height / 2 - int(cy + r) w = int(2 * r) h = w if self.pendown: self.canvas.images[0].draw_arc(self.gc, False, x, y, w, h, int(180 - self.heading - a) * 64, int(a) * 64) self.invalt(x - self.pensize * self.tw.coord_scale / 2 - 3, y - self.pensize * self.tw.coord_scale / 2 - 3, w + self.pensize * self.tw.coord_scale + 6, h + self.pensize * self.tw.coord_scale + 6) self.right(a, False) self.xcor = cx - r * cos(self.heading * DEGTOR) self.ycor = cy + r * sin(self.heading * DEGTOR) if self.tw.saving_svg and self.pendown: self.tw.svg_string += self.svg.new_path(oldx, self.height / 2 - oldy) self.tw.svg_string += self.svg.arc_to(self.xcor, self.height / 2 - self.ycor, r, a, 0, s) self.tw.svg_string += "\"\n" self.tw.svg_string += self.svg.style() def larc(self, a, r): """ draw a counter-clockwise arc """ if r < 0: r = -r a = -a s = 1 else: s = 0 oldx, oldy = self.xcor, self.ycor cx = self.xcor - r * cos(self.heading * DEGTOR) cy = self.ycor + r * sin(self.heading * DEGTOR) x = self.width / 2 + int(cx - r) y = self.height / 2 - int(cy + r) w = int(2 * r) h = w if self.pendown: self.canvas.images[0].draw_arc(self.gc, False, x, y, w, h, int(360 - self.heading) * 64, int(a) * 64) self.invalt(x - self.pensize * self.tw.coord_scale / 2 - 3, y - self.pensize * self.tw.coord_scale / 2 - 3, w + self.pensize * self.tw.coord_scale + 6, h + self.pensize * self.tw.coord_scale + 6) self.right(-a, False) self.xcor = cx + r * cos(self.heading * DEGTOR) self.ycor = cy - r * sin(self.heading * DEGTOR) if self.tw.saving_svg and self.pendown: self.tw.svg_string += self.svg.new_path(oldx, self.height / 2 - oldy) self.tw.svg_string += self.svg.arc_to(self.xcor, self.height / 2 - self.ycor, r, a, 0, s) self.tw.svg_string += "\"\n" self.tw.svg_string += self.svg.style() def setxy(self, x, y, share=True, pendown=True): """ Move turtle to position x,y """ oldx, oldy = self.xcor, self.ycor x *= self.tw.coord_scale y *= self.tw.coord_scale try: self.xcor, self.ycor = x, y except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return if self.pendown and pendown: self.gc.set_foreground(self.fgcolor) self.draw_line(oldx, oldy, self.xcor, self.ycor) self.move_turtle() if self.tw.sharing() and share: self.tw.activity.send_event("x|%s" % \ (data_to_string([self.tw.nick, [round_int(x), round_int(y)]]))) def setpensize(self, ps, share=True): """ Set the pen size """ try: if ps < 0: ps = 0 self.pensize = ps except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return self.tw.active_turtle.set_pen_size(ps) self.gc.set_line_attributes(int(self.pensize * self.tw.coord_scale), gtk.gdk.LINE_SOLID, gtk.gdk.CAP_ROUND, gtk.gdk.JOIN_MITER) self.svg.set_stroke_width(self.pensize) if self.tw.sharing() and share: self.tw.activity.send_event("w|%s" % \ (data_to_string([self.tw.nick, round_int(ps)]))) def setcolor(self, c, share=True): """ Set the pen color """ try: self.color = c self.tcolor = c except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return self.tw.active_turtle.set_color(c) self.set_fgcolor() self.set_textcolor() if self.tw.sharing() and share: self.tw.activity.send_event("c|%s" % \ (data_to_string([self.tw.nick, round_int(c)]))) def setgray(self, g, share=True): """ Set the gray level """ try: self.gray = g except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return if self.gray < 0: self.gray = 0 if self.gray > 100: self.gray = 100 self.set_fgcolor() self.set_textcolor() self.tw.active_turtle.set_gray(self.gray) if self.tw.sharing() and share: self.tw.activity.send_event("g|%s" % \ (data_to_string([self.tw.nick, round_int(self.gray)]))) def settextcolor(self, c): """ Set the text color """ try: self.tcolor = c except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return self.set_textcolor() def settextsize(self, c): # depreciated """ Set the text size """ try: self.tw.textsize = c except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) def setshade(self, s, share=True): """ Set the color shade """ try: self.shade = s except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return self.tw.active_turtle.set_shade(s) self.set_fgcolor() self.set_textcolor() if self.tw.sharing() and share: self.tw.activity.send_event("s|%s" % \ (data_to_string([self.tw.nick, round_int(s)]))) def fillscreen(self, c, s): """ Fill screen with color/shade and reset to defaults """ oldc, olds = self.color, self.shade self.setcolor(c, False) self.setshade(s, False) rect = gtk.gdk.Rectangle(0, 0, self.width, self.height) self.gc.set_foreground(self.fgcolor) self.bgrgb = self.fgrgb[:] self.canvas.images[0].draw_rectangle(self.gc, True, *rect) self.invalt(0, 0, self.width, self.height) self.setcolor(oldc, False) self.setshade(olds, False) self.tw.svg_string = '' self.svg.reset_min_max() self.fill = False self.poly_points = [] def set_fgcolor(self): """ Set the foreground color """ if self.color == WHITE or self.shade == WHITE: r = 0xFF00 g = 0xFF00 b = 0xFF00 elif self.color == BLACK or self.shade == BLACK: r = 0x0000 g = 0x0000 b = 0x0000 else: sh = (wrap100(self.shade) - 50) / 50.0 rgb = color_table[wrap100(self.color)] r = (rgb >> 8) & 0xff00 r = calc_gray(r, self.gray) r = calc_shade(r, sh) g = rgb & 0xff00 g = calc_gray(g, self.gray) g = calc_shade(g, sh) b = (rgb << 8) & 0xff00 b = calc_gray(b, self.gray) b = calc_shade(b, sh) self.fgrgb = [r >> 8, g >> 8, b >> 8] self.fgcolor = self.cm.alloc_color(r, g, b) self.svg.set_stroke_color("#%02x%02x%02x" % (self.fgrgb[0], self.fgrgb[1], self.fgrgb[2])) def set_textcolor(self): """ Set the text color to foreground color. """ self.tw.textcolor = self.fgcolor def setpen(self, bool, share=True): """ Lower or raise the pen """ self.pendown = bool self.tw.active_turtle.set_pen_state(bool) if self.tw.sharing() and share: self.tw.activity.send_event("p|%s" % \ (data_to_string([self.tw.nick, bool]))) def draw_pixbuf(self, pixbuf, a, b, x, y, w, h, path): """ Draw a pixbuf """ w *= self.tw.coord_scale h *= self.tw.coord_scale self.canvas.images[0].draw_pixbuf(self.gc, pixbuf, a, b, x, y) self.invalt(x, y, w, h) if self.tw.saving_svg: if self.tw.running_sugar: # In Sugar, we need to embed the images inside the SVG self.tw.svg_string += self.svg.image(x - self.width / 2, y, w, h, path, image_to_base64(pixbuf, self.tw.activity)) else: self.tw.svg_string += self.svg.image(x - self.width / 2, y, w, h, path) def draw_text(self, label, x, y, size, w): """ Draw text """ w *= self.tw.coord_scale self.gc.set_foreground(self.tw.textcolor) fd = pango.FontDescription('Sans') try: fd.set_size(int(size * self.tw.coord_scale) * pango.SCALE) except TypeError, ValueError: _logger.debug("bad value sent to %s" % (__name__)) return if self.tw.interactive_mode: if type(label) == str or type(label) == unicode: pl = self.tw.window.create_pango_layout( label.replace('\0', ' ')) elif type(label) == float or type(label) == int: pl = self.tw.window.create_pango_layout(str(label)) else: pl = self.tw.window.create_pango_layout(str(label)) pl.set_font_description(fd) pl.set_width(int(w) * pango.SCALE) self.canvas.images[0].draw_layout(self.gc, int(x), int(y), pl) w, h = pl.get_pixel_size() self.invalt(x, y, w, h) else: # pixmap doesn't support pango message = str(label).replace('\0', ' ') context = self.canvas.images[0].cairo_create() context.set_font_size(size) q, k, w, h = context.text_extents(message)[:4] context.set_source_rgb(0, 0, 0) context.move_to(x, y + h) context.show_text(message) if self.tw.saving_svg and self.pendown: self.tw.svg_string += self.svg.text(x - self.width / 2, y + size, size, w, label) def draw_line(self, x1, y1, x2, y2): """ Draw a line """ x1, y1 = self.width / 2 + int(x1), self.height / 2 - int(y1) x2, y2 = self.width / 2 + int(x2), self.height / 2 - int(y2) if x1 < x2: minx, maxx = x1, x2 else: minx, maxx = x2, x1 if y1 < y2: miny, maxy = y1, y2 else: miny, maxy = y2, y1 w, h = maxx - minx, maxy - miny self.canvas.images[0].draw_line(self.gc, x1, y1, x2, y2) if self.fill and self.poly_points == []: self.poly_points.append((x1, y1)) if self.fill: self.poly_points.append((x2, y2)) self.invalt(minx - self.pensize * self.tw.coord_scale / 2 - 3, miny - self.pensize * self.tw.coord_scale / 2 - 3, w + self.pensize * self.tw.coord_scale + 6, h + self.pensize * self.tw.coord_scale + 6) def turn_turtle(self): """ Change the orientation of the turtle """ self.tw.active_turtle.set_heading(self.heading) def move_turtle(self): """ Move the turtle """ x, y = self.width / 2 + int(self.xcor), \ self.height / 2 - int(self.ycor) self.tw.active_turtle.move( (int(self.cx + x - self.tw.active_turtle.spr.rect.width / 2), int(self.cy + y - self.tw.active_turtle.spr.rect.height / 2))) def invalt(self, x, y, w, h): """ Mark a region for refresh """ if self.tw.interactive_mode: self.tw.area.invalidate_rect( gtk.gdk.Rectangle(int(x + self.cx), int(y + self.cy), int(w), int(h)), False) def get_color_index(self, r, g, b, a=0): """ Find the closest palette entry to the rgb triplet """ if self.shade != 50 or self.gray != 100: r <<= 8 g <<= 8 b <<= 8 if self.shade != 50: sh = (wrap100(self.shade) - 50) / 50.0 r = calc_shade(r, sh, True) g = calc_shade(g, sh, True) b = calc_shade(b, sh, True) if self.gray != 100: r = calc_gray(r, self.gray, True) g = calc_gray(g, self.gray, True) b = calc_gray(b, self.gray, True) r >>= 8 g >>= 8 b >>= 8 min_distance = 1000000 closest_color = -1 for i, c in enumerate(color_table): cr = int((c & 0xff0000) >> 16) cg = int((c & 0x00ff00) >> 8) cb = int((c & 0x0000ff)) distance_squared = \ ((cr - r) ** 2) + ((cg - g) ** 2) + ((cb - b) ** 2) if distance_squared == 0: return i if distance_squared < min_distance: min_distance = distance_squared closest_color = i return closest_color def get_pixel(self): """ Read the pixel at x, y """ if self.tw.interactive_mode: return self.canvas.get_pixel( (self.width / 2 + int(self.xcor), self.height / 2 - int(self.ycor)), 0, self.tw.color_mode) else: return(-1, -1, -1, -1) def set_turtle(self, k, colors=None): """ Select the current turtle and associated pen status """ if k not in self.tw.turtles.dict: # if it is a new turtle, start it in the center of the screen self.tw.active_turtle = self.tw.turtles.get_turtle(k, True, colors) self.seth(0, False) self.setxy(0, 0, False, pendown=False) self.tw.active_turtle.set_pen_state(True) self.tw.active_turtle = self.tw.turtles.get_turtle(k, False) self.tw.active_turtle.show() tx, ty = self.tw.active_turtle.get_xy() self.xcor = -self.width / 2 + tx + \ self.tw.active_turtle.spr.rect.width / 2 self.ycor = self.height / 2 - ty - \ self.tw.active_turtle.spr.rect.height / 2 self.heading = self.tw.active_turtle.get_heading() self.setcolor(self.tw.active_turtle.get_color(), False) self.setgray(self.tw.active_turtle.get_gray(), False) self.setshade(self.tw.active_turtle.get_shade(), False) self.setpensize(self.tw.active_turtle.get_pen_size(), False) self.setpen(self.tw.active_turtle.get_pen_state(), False) def svg_close(self): """ Close current SVG graphic """ if self.tw.svg_string == '': return self.svg.calc_w_h(False) self.tw.svg_string = "%s%s%s%s" % (self.svg.header(True), self.svg.background("#%02x%02x%02x" % \ (self.bgrgb[0], self.bgrgb[1], self.bgrgb[2])), self.tw.svg_string, self.svg.footer())