X-Git-Url: http://git.cascardo.eti.br/?a=blobdiff_plain;f=gzv.py;h=664b7f9366347be9184ed91bf19870e1a9dc73a9;hb=715cd1b218caf92fc360ec9b75da60f81b881138;hp=42dce80e0554c489bba7b4221cd01fead3ce3c5e;hpb=4a4b78c9de4daf237cc4215ed1cca56c37b16fa1;p=cascardo%2Fmovie.git diff --git a/gzv.py b/gzv.py index 42dce80..664b7f9 100644 --- a/gzv.py +++ b/gzv.py @@ -1,3 +1,4 @@ +# -*- coding: utf-8; -*- # gzv.py - an user interface to generate-zooming-video # # Copyright (C) 2008 Lincoln de Sousa @@ -22,14 +23,22 @@ from ConfigParser import ConfigParser _ = lambda x:x +class Point(object): + def __init__(self, x, y): + self.x = x + self.y = y + + @staticmethod + def pythagorean(p1, p2): + return math.sqrt((p2.x - p1.x)**2 + (p2.y - p1.y)**2) + class Ball(object): DEFAULT_WIDTH = 10 def __init__(self, x, y, r, name='', position=0, selected=False): self.position = position self.selected = selected - self.x = int(x) - self.y = int(y) + self.p = Point(int(x), int(y)) self.radius = int(r) self.name = name @@ -40,7 +49,7 @@ class BallManager(list): def save_to_file(self, path): target = open(path, 'w') for i in self: - target.write('%d,%d %d %s\n' % (i.x, i.y, i.radius, i.name)) + target.write('%d,%d %d %s\n' % (i.p.x, i.p.y, i.radius, i.name)) target.close() class GladeLoader(object): @@ -138,6 +147,11 @@ class Gzv(GladeLoader): self.evtbox.connect('button-press-event', self.button_press) self.evtbox.connect('button-release-event', self.button_release) self.evtbox.connect('motion-notify-event', self.motion_notify) + self.evtbox.connect('motion-notify-event', self.ball_motion) + + # making it possible to grab motion events when the mouse is + # over the widget. + self.evtbox.set_events(gtk.gdk.POINTER_MOTION_MASK) self.model = gtk.ListStore(int, str) self.treeview = self.wid('treeview') @@ -157,11 +171,19 @@ class Gzv(GladeLoader): self.load_balls_to_treeview() self.setup_treeview() + self.new_ball = False + self.move_ball = None + # drawing stuff self.start_x = -1 self.start_y = -1 + self.last_x = -1 + self.last_y = -1 self.radius = Ball.DEFAULT_WIDTH + def show(self): + self.window.show_all() + def setup_treeview(self): self.model.connect('rows-reordered', self.on_rows_reordered) @@ -285,6 +307,17 @@ class Gzv(GladeLoader): self.draw.queue_draw() + def select_fp_from_image(self, ball): + selection = self.treeview.get_selection() + selection.select_path(str(ball.position)) + + # making sure that only one ball is selected + for i in self.balls: + i.selected = False + ball.selected = True + + self.draw.queue_draw() + def save_fp_list(self, *args): assert self.project is not None @@ -316,34 +349,78 @@ class Gzv(GladeLoader): if self.start_x < 0: return False - ball = Ball(self.start_x, self.start_y, self.radius) - self.draw_ball(ball) + if self.new_ball: + ball = Ball(self.start_x, self.start_y, self.radius) + self.draw_ball(ball) return False + def ball_width_border(self, ball): + iw, ih = self.draw.size_request() + w = self.draw.get_allocation().width + h = self.draw.get_allocation().height + + x = ((w / 2) - (iw / 2)) + ball.p.x + y = ((h / 2) - (ih / 2)) + ball.p.y + return Point(x, y) + + def point_without_border(self, point): + iw, ih = self.draw.size_request() + w = self.draw.get_allocation().width + h = self.draw.get_allocation().height + + x = point.x - ((w / 2) - (iw / 2)) + y = point.y - ((h / 2) - (ih / 2)) + return Point(x, y) + def draw_ball(self, ball): ctx = self.draw.window.cairo_create() - ctx.arc(ball.x, ball.y, ball.radius, 0, 64*math.pi) + ctx.arc(self.ball_width_border(ball).x, + self.ball_width_border(ball).y, + ball.radius, 0, 64*math.pi) ctx.set_source_rgba(0.0, 0.0, 0.5, 0.4) ctx.fill() if ball.selected: ctx.set_source_rgba(0.0, 0.5, 0.0, 0.4) ctx.set_line_width(5) - ctx.arc(ball.x, ball.y, ball.radius+1, 0, 64*math.pi) + ctx.arc(self.ball_width_border(ball).x, + self.ball_width_border(ball).y, + ball.radius+1, 0, 64*math.pi) ctx.stroke() def button_press(self, widget, event): + self.new_ball = True + + self.last_x = event.x + self.last_y = event.y + if event.button == 1: - self.start_x = event.x - self.start_y = event.y - self.last_x = event.x + for i in self.balls: + p1 = Point(event.x, event.y) + p2 = self.ball_width_border(i) + if Point.pythagorean(p1, p2) < i.radius: + self.last_x = event.x - i.p.x + self.last_y = event.y - i.p.y + self.select_fp_from_image(i) + + self.new_ball = False + self.move_ball = i + break + + self.start_x = self.point_without_border(event).x + self.start_y = self.point_without_border(event).y def button_release(self, widget, event): + self.move_ball = None + if event.button == 1: self.finish_drawing() def motion_notify(self, widget, event): + if not self.new_ball: + return + self.draw.queue_draw() if event.x > self.last_x: @@ -353,12 +430,23 @@ class Gzv(GladeLoader): self.last_x = event.x + def ball_motion(self, widget, event): + if not self.move_ball: + return + + self.move_ball.p.x = self.point_without_border(event).x + self.move_ball.p.y = self.point_without_border(event).y + + self.draw.queue_draw() + def finish_drawing(self): - position = len(self.balls) - ball = Ball(self.start_x, self.start_y, self.radius, '', position) - self.balls.append(ball) - self.model.append([position, '']) - self.treeview.set_cursor(str(position), self.fpcolumn, True) + if self.new_ball: + position = len(self.balls) + ball = Ball(self.start_x, self.start_y, self.radius, '', position) + self.balls.append(ball) + self.model.append([position, '']) + self.treeview.set_cursor(str(position), self.fpcolumn, True) + self.new_ball = False # reseting to the default coordenades self.start_x = -1 @@ -366,5 +454,5 @@ class Gzv(GladeLoader): self.radius = Ball.DEFAULT_WIDTH if __name__ == '__main__': - Gzv().window.show_all() + Gzv().show() gtk.main()