Skip to content

Commit

Permalink
add frame changed event
Browse files Browse the repository at this point in the history
  • Loading branch information
grokhi committed Sep 19, 2024
1 parent b062211 commit 8c6fe93
Showing 1 changed file with 29 additions and 7 deletions.
36 changes: 29 additions & 7 deletions src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -269,8 +269,30 @@ def _find_static_borders(m_kpts0, m_kpts1, image0_width, image1_width):
return matches[frame_id]


@app.event(sly.Event.Entity.FrameChanged)
def video_frame_changed(api: sly.Api, event: sly.Event.Entity.FrameChanged):
g.api = api
if event is None:
return
# g.curr_event[event.user_id] = event

global curr_event_tmp
curr_event_tmp = event

status_image_not_initialized.hide()
frame_id = f"{event.video_id}_{event.frame}"
sly.logger.debug("Event frame ID=%s", frame_id)
get_points_video(api, event, frame_id)

if not button_track.is_disabled():
return
# if g.previous_frame is not None:
# g.event_anns[event.user_id][g.previous_frame] = g.DONE # TODO previous frame
# g.previous_frame = frame_id


@app.event(sly.Event.ManualSelected.VideoChanged)
def video_changed(api: sly.Api, event: sly.Event.ManualSelected.VideoChanged):
def video_frame_changed(api: sly.Api, event: sly.Event.ManualSelected.VideoChanged):
g.api = api
if event is None:
return
Expand All @@ -281,14 +303,14 @@ def video_changed(api: sly.Api, event: sly.Event.ManualSelected.VideoChanged):

status_image_not_initialized.hide()
frame_id = f"{event.video_id}_{event.frame}"
sly.logger.debug("Event frame ID={}".format(frame_id))
sly.logger.debug("Event frame ID=%s", frame_id)
get_points_video(api, event, frame_id)

if not button_track.is_disabled():
return
if g.previous_frame is not None:
g.event_anns[event.user_id][g.previous_frame] = g.DONE # TODO previous frame
g.previous_frame = frame_id
# if g.previous_frame is not None:
# g.event_anns[event.user_id][g.previous_frame] = g.DONE # TODO previous frame
# g.previous_frame = frame_id


@app.event(sly.Event.ManualSelected.ImageChanged)
Expand Down Expand Up @@ -382,8 +404,8 @@ def rectangle_changed(api: sly.Api, event: sly.Event.Tools.Rectangle.FigureChang
if not button_track.is_disabled():
return

if event.frame is not None:
process_video_rectangle_changed(api, event)
if getattr(event, "frame", None) is not None:
# process_video_rectangle_changed(api, event)
return

t = datetime.now().timestamp()
Expand Down

0 comments on commit 8c6fe93

Please sign in to comment.