def end_drag(self, event):
self.drag_info.update_cur_pos(event.GetPositionTuple())
rect = Objtrack.get_selection_rect(self.frozen_frame.shape,
self.drag_info.get_drag_data(),
self.bitmap_track_size)
try:
# self.objmatch = Objtrack.ObjectMatch(rect, self.frozen_frame, self.get_hist_channel())
self.objmatch = Objtrack.ObjectTrack(rect, self.frozen_frame, self.get_hist_channel())
self.frozen_frame = None
self.drag_info = None
self.toggle_drag_selection(False)
self.m_button_toggle_track.Enable(True)
except AssertionError,e:
wx.MessageBox(str(e), u"出现错误",wx.OK | wx.ICON_ERROR)
self.sbar.update(unicode(e))
self.frozen_frame = None
self.drag_info = None
self.toggle_drag_selection(False)
self.display_track_state = DISPLAY_TRACK_STATE_RAW
def get_adjusted_image(self, src):
return Objtrack.get_adjusted_image(src, self.track_image_adjust_value)
def open_track_video(self, comp):
self.track_image_adjust_value = Objtrack.get_image_adjust_value()
self.add_work(DISPLAY_TRACK_VIDEO)
self.display_track_state = DISPLAY_TRACK_STATE_RAW
util.toggle_button(comp, u'显示视频', u'关闭显示')
self.m_button_select_object.Enable(True)
def get_dragging_image(self, src, drag_data):
return Objtrack.get_dragging_image(src, drag_data, self.bitmap_track_size)
