# -*- coding: utf-8; -*-
-# gzv.py - an user interface to generate-zooming-video
+# gzv.py - an user interface to select people in a picture
#
# Copyright (C) 2008 Lincoln de Sousa <lincoln@minaslivre.org>
#
@staticmethod
def pythagorean(p1, p2):
- return math.sqrt((p2.x - p1.x)**2 + (p2.y - p2.y)**2)
+ return math.sqrt((p2.x - p1.x)**2 + (p2.y - p1.y)**2)
class Ball(object):
DEFAULT_WIDTH = 10
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.draw.set_from_pixbuf(pixbuf)
self.load_balls_to_treeview()
+ 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()
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
self.balls.save_to_file(self.project.focus_points_file)
+ def move_fp_up(self, *args):
+ selection = self.treeview.get_selection()
+ model, path = selection.get_selected()
+ if not path:
+ return
+
+ pos = model[path][0]
+ newpos = max(pos - 1, 0)
+ self.balls.insert(newpos, self.balls.pop(pos))
+
+ # normalizing the position of elements.
+ for index, item in enumerate(self.balls):
+ item.position = index
+
+ self.load_balls_to_treeview()
+ selection.select_path(str(newpos))
+
+ def move_fp_down(self, *args):
+ selection = self.treeview.get_selection()
+ model, path = selection.get_selected()
+ if not path:
+ return
+
+ pos = model[path][0]
+ newpos = min(pos + 1, len(self.balls))
+ self.balls.insert(newpos, self.balls.pop(pos))
+
+ # normalizing the position of elements.
+ for index, item in enumerate(self.balls):
+ item.position = index
+
+ self.load_balls_to_treeview()
+ selection.select_path(str(newpos))
+
def expose_draw(self, draw, event):
if not self.image:
return
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(ball.p.x, ball.p.y, ball.radius, 0, 64*math.pi)
+ 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(ball.p.x, ball.p.y, ball.radius+1, 0, 64*math.pi)
+ 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):
if event.button == 1:
for i in self.balls:
p1 = Point(event.x, event.y)
- p2 = Point(i.p.x, i.p.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 = event.x
- self.start_y = event.y
+ 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 not self.move_ball:
return
- self.move_ball.p.x += (event.x - self.move_ball.p.x)
- self.move_ball.p.y += (event.y - self.move_ball.p.y)
+ 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()
- self.last_x = event.x - self.move_ball.p.x
- self.last_y = event.y - self.move_ball.p.y
-
def finish_drawing(self):
if self.new_ball:
position = len(self.balls)