X-Git-Url: http://git.cascardo.eti.br/?a=blobdiff_plain;f=gzv.py;h=ac4dce3a9d94ee3563ba8c87219eaecd97517152;hb=6a65c73d806b38167701cee3d75f2751822f8f6d;hp=081e06a181527f2dd877884ffa52be3a067635db;hpb=bbfc3969346db1b11309f60cc72c979258c26bf3;p=cascardo%2Fmovie.git diff --git a/gzv.py b/gzv.py index 081e06a..ac4dce3 100644 --- a/gzv.py +++ b/gzv.py @@ -1,4 +1,5 @@ -# gzv.py - an user interface to generate-zooming-video +# -*- coding: utf-8; -*- +# gzv.py - an user interface to select people in a picture # # Copyright (C) 2008 Lincoln de Sousa # @@ -22,13 +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): + def __init__(self, x, y, r, name='', position=0, selected=False): self.position = position - self.x = int(x) - self.y = int(y) + self.selected = selected + self.p = Point(int(x), int(y)) self.radius = int(r) self.name = name @@ -39,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): @@ -115,6 +125,10 @@ class NewProject(GladeLoader): self.dialog.set_transient_for(parent) def get_project(self): + # This '1' was defined in the glade file + if not self.dialog.run() == 1: + return None + fname = self.wid('image').get_filename() width = self.wid('width').get_text() height = self.wid('height').get_text() @@ -133,13 +147,19 @@ 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') self.treeview.set_model(self.model) + self.treeview.connect('button-press-event', self.select_fp) self.draw = self.wid('draw') - self.draw.connect('expose-event', self.expose_draw) + self.draw.connect_after('expose-event', self.expose_draw) # Starting with an empty project with no image loaded self.project = None @@ -151,12 +171,19 @@ class Gzv(GladeLoader): self.load_balls_to_treeview() self.setup_treeview() + self.new_ball = False + self.move_ball = None + # drawing stuff - self.selecting = False 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) @@ -177,16 +204,15 @@ class Gzv(GladeLoader): def on_cell_edited(self, renderer, path, value): self.balls[int(path)].name = value self.load_balls_to_treeview() - self.draw.queue_draw() def new_project(self, button): proj = NewProject(self.window) - - # This '1' was defined in the glade file - if proj.dialog.run() == 1: - self.load_project(proj.get_project()) + project = proj.get_project() proj.destroy() + if project: + self.load_project(project) + def open_project(self, *args): fc = gtk.FileChooserDialog(_('Choose a gzv project'), self.window, buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, @@ -208,19 +234,41 @@ class Gzv(GladeLoader): self.balls.save_to_file(self.project.focus_points_file) fc.destroy() - def load_project(self, project): self.project = project self.balls = self.load_balls_from_file(project.focus_points_file) self.image = project.image + + # I'm loading a pixbuf first because I need to get its + # dimensions this with a pixbuf is easier than with an image. + try: + pixbuf = gtk.gdk.pixbuf_new_from_file(project.image) + except gobject.GError: + msg = _("Couldn't recognize the image file format.") + dialog = gtk.MessageDialog(self.window, + gtk.DIALOG_MODAL, + gtk.MESSAGE_ERROR, + gtk.BUTTONS_CLOSE) + dialog.set_markup(msg) + dialog.run() + dialog.destroy() + return self.unload_project() + + self.draw.set_from_pixbuf(pixbuf) self.load_balls_to_treeview() - self.draw.queue_draw() + self.set_widgets_sensitivity(True) def unload_project(self): self.project = None self.image = None self.balls = BallManager() self.draw.queue_draw() + self.set_widgets_sensitivity(False) + + def set_widgets_sensitivity(self, sensitive): + for i in 'toolbutton1', 'toolbutton5', 'scrolledwindow1', \ + 'hbox2', 'imagemenuitem3': + self.wid(i).set_sensitive(sensitive) def load_balls_to_treeview(self): self.model.clear() @@ -251,6 +299,41 @@ class Gzv(GladeLoader): del model[path] self.draw.queue_draw() + def select_fp(self, treeview, event): + path, column, x, y = \ + self.treeview.get_path_at_pos(int(event.x), int(event.y)) + if path: + model = self.treeview.get_model() + ball = self.balls[model[path][0]] + + # making sure that only one ball is selected + for i in self.balls: + i.selected = False + ball.selected = True + + # available space to the image + w = self.evtbox.get_allocation().width + h = self.evtbox.get_allocation().height + + # point begining from the left image border + wib = self.point_with_border(ball) + + #self.wid('viewport').get_vadjustment().value = wib.x # + (w / 2) + #self.wid('viewport').get_hadjustment().value = wib.y # + (h / 2) + + 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 @@ -272,78 +355,122 @@ class Gzv(GladeLoader): self.balls.save_to_file(self.project.focus_points_file) - def expose_draw(self, draw, event): - if not self.image: + def move_fp_up(self, *args): + selection = self.treeview.get_selection() + model, path = selection.get_selected() + if not path: return - # loading the picture image and getting some useful - # information to draw it in the widget's background - try: - img = gtk.gdk.pixbuf_new_from_file(self.image) - except gobject.GError: - msg = _("Couldn't recognize the image file format.") - dialog = gtk.MessageDialog(self.window, - gtk.DIALOG_MODAL, - gtk.MESSAGE_ERROR, - gtk.BUTTONS_CLOSE) - dialog.set_markup(msg) - dialog.run() - dialog.destroy() + pos = model[path][0] + newpos = max(pos - 1, 0) + self.balls.insert(newpos, self.balls.pop(pos)) - self.draw.stop_emission('expose-event') - return self.unload_project() + # normalizing the position of elements. + for index, item in enumerate(self.balls): + item.position = index - pixels = img.get_pixels() - rowstride = img.get_rowstride() - width = img.get_width() - height = img.get_height() - gc = draw.style.black_gc + self.load_balls_to_treeview() + selection.select_path(str(newpos)) - # sets the correct size of the eventbox, to show the scrollbar - # when needed. - self.evtbox.set_size_request(width, height) + def move_fp_down(self, *args): + selection = self.treeview.get_selection() + model, path = selection.get_selected() + if not path: + return - # drawing the picture in the background of the drawing area, - # this is really important. - draw.window.draw_rgb_image(gc, 0, 0, width, height, - 'normal', pixels, rowstride, - 0, 0) + pos = model[path][0] + newpos = min(pos + 1, len(self.balls)) + self.balls.insert(newpos, self.balls.pop(pos)) - # this call makes the ball being drown be shown correctly. - self.draw_current_ball() + # normalizing the position of elements. + for index, item in enumerate(self.balls): + item.position = index - # drawing other balls stored in the self.balls list. - ctx = draw.window.cairo_create() - ctx.fill() + self.load_balls_to_treeview() + selection.select_path(str(newpos)) - ctx.set_line_width(10.0) - ctx.set_source_rgba(0.5, 0.0, 0.0, 0.4) + def expose_draw(self, draw, event): + if not self.image: + return for i in self.balls: - ctx.arc(i.x, i.y, i.radius, 0, 64*math.pi) - ctx.fill() + self.draw_ball(i) - def draw_current_ball(self): if self.start_x < 0: - return + return False + + if self.new_ball: + ball = Ball(self.start_x, self.start_y, self.radius) + self.draw_ball(ball) + + return False + + def point_with_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(self.start_x, self.start_y, self.radius, 0, 64*math.pi) - ctx.set_source_rgba(0.5, 0.0, 0.0, 0.4) + ctx.arc(self.point_with_border(ball).x, + self.point_with_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(self.point_with_border(ball).x, + self.point_with_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.selecting = True - 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.point_with_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.selecting = False 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,16 +480,29 @@ class Gzv(GladeLoader): self.last_x = event.x - def finish_drawing(self): - self.draw_current_ball() - self.ball_width = Ball.DEFAULT_WIDTH + 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 - 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.draw.queue_draw() + + def finish_drawing(self): + 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 + self.start_y = -1 + self.radius = Ball.DEFAULT_WIDTH if __name__ == '__main__': - Gzv().window.show_all() + Gzv().show() gtk.main()