def pause(self, widget, touch):
if not Widget.collide_point(widget, *
touch.pos) or touch.is_mouse_scrolling:
return False
elif self.playback_state == 'playing':
self.playback_state = 'paused'
return True
def stop(self, widget, touch):
if not Widget.collide_point(widget, *
touch.pos) or touch.is_mouse_scrolling:
return False
else:
self.playback_state = 'stopped'
self.seek_time = 0
return True
def handle_slider_touch(self, widget, touch, touch_state):
if not Widget.collide_point(widget, *
touch.pos) or touch.is_mouse_scrolling:
return False
else:
if touch_state == 'down' and self.playback_state == 'playing':
self.playback_state = 'paused'
elif touch_state == 'up' and self.playback_state == 'paused':
self.playback_state = 'playing'
self.play(widget, touch)
return True
def play(self, widget, touch):
if not Widget.collide_point(
widget, *touch.pos
) or self._audio_chunks is None or touch.is_mouse_scrolling:
return False
else:
seg = self._audio_chunks[0]
self.pa = pyaudio.PyAudio()
if self.stream is not None:
self.playback_state = 'playing'
return True
self.stream = self.pa.open(format=self.pa.get_format_from_width(
seg.sample_width),
channels=seg.channels,
rate=seg.frame_rate,
output=True)
self.playback_state = 'playing'
self.pool.apply_async(self.play_chunks)
return True
class DrawableVideo(Video):
bbs = ListProperty(None)
labelsColor = DictProperty(None)
video_loaded = ObjectProperty(None)
video_state_changed = ObjectProperty(None)
add_bb = ObjectProperty(None)
def __init__(self, **kwargs):
super(DrawableVideo, self).__init__(**kwargs)
self.activeLabel = None
self.textureWidget = Widget()
self.source = r'C:\bf1Movies\air.mp4'
def on_size(self, instance, value):
Clock.schedule_once(self.sched_on_bbs, 0)
def on_norm_image_size(self, instance, value):
rectPos = [(self.size[0] - self.norm_image_size[0]) / 2 + self.pos[0],
(self.size[1] - self.norm_image_size[1]) / 2 + self.pos[1]]
rectSize = self.norm_image_size
self.textureWidget.pos = rectPos
self.textureWidget.size = rectSize
# print('texture size changed pos:{} , size:{}'.format(str(self.textureWidget.pos), str(self.textureWidget.size) ))
def sched_on_bbs(self, dt):
self.on_bbs(None, None)
def on_loaded(self, inst, val):
self.state = 'stop'
self.video_loaded(val)
rectPos = [(self.size[0] - self.norm_image_size[0]) / 2 + self.pos[0],
(self.size[1] - self.norm_image_size[1]) / 2 + self.pos[1]]
rectSize = self.norm_image_size
self.add_widget(self.textureWidget)
def on_source(self, inst, val):
# when source changes the move is set to play
#on_loaded should stop it right it's loaded
self.state = 'play'
def on_bbs(self, instance, value):
# print(str(self.size))
self.textureWidget.clear_widgets()
for bb in self.bbs:
bbWidget = BoundingBoxWidget(labelName="")
self.textureWidget.add_widget(bbWidget)
bbWidget.update_bb(bb)
bbWidget.set_name(bb.label)
if bb.label in self.labelsColor:
bbWidget.bbColor = get_color_from_hex(
self.labelsColor[bb.label])
else:
bbWidget.bbColor = [1, 1, 1]
def on_labelsColor(self, instance, value):
self.textureWidget.clear_widgets()
for bb in self.bbs:
bbWidget = BoundingBoxWidget(labelName="")
self.textureWidget.add_widget(bbWidget)
bbWidget.update_bb(bb)
bbWidget.set_name(bb.label)
if bb.label in self.labelsColor:
bbWidget.bbColor = get_color_from_hex(
self.labelsColor[bb.label])
else:
bbWidget.bbColor = [1, 1, 1]
def stop_pressed(self):
if self.loaded is False:
return
self.state = 'stop'
def play_pause_pressed(self):
if self.loaded is False:
return
if self.state is 'play':
self.state = 'pause'
# self.ids.playPauseButton.text = 'play'
elif self.state is 'pause' or self.state is 'stop':
self.state = 'play'
# self.ids.playPauseButton.text = 'pause'
def on_touch_down(self, touch):
if 'pos' not in touch.profile:
return False
if self.loaded is False:
return False
if self.textureWidget.collide_point(touch.pos[0],
touch.pos[1]) is False:
return False
# else:
# print('collide: {}, {} -> {}, {}'.format(str(self.textureWidget.pos), str(self.textureWidget.size), touch.pos[0], touch.pos[1]))
if 'button' in touch.profile:
isLeftButton = (touch.button == 'left')
if isLeftButton:
self.currentLabelLeftTop = touch.pos
self.currentLabelPivot = touch.pos
self.activeLabel = BoundingBoxWidget()
self.textureWidget.add_widget(self.activeLabel)
self.activeLabel.bbColor = get_color_from_hex(colors.YELLOW)
def on_touch_move(self, touch):
if self.activeLabel is None:
return
if 'pos' in touch.profile:
if self.textureWidget.collide_point(touch.pos[0],
touch.pos[1]) is False:
touchX = touch.pos[0]
touchY = touch.pos[1]
if (touch.pos[0] < self.textureWidget.pos[0]):
touchX = self.textureWidget.pos[0]
elif (touch.pos[0] >
self.textureWidget.pos[0] + self.textureWidget.size[0]):
touchX = self.textureWidget.pos[
0] + self.textureWidget.size[0]
if (touch.pos[1] < self.textureWidget.pos[1]):
touchY = self.textureWidget.pos[1]
elif (touch.pos[1] >
self.textureWidget.pos[1] + self.textureWidget.size[1]):
touchY = self.textureWidget.pos[
1] + self.textureWidget.size[1]
touch.pos = (touchX, touchY)
if self.currentLabelPivot is not None:
xMin = min(self.currentLabelPivot[0], touch.pos[0])
xMax = max(self.currentLabelPivot[0], touch.pos[0])
yMin = max(self.currentLabelPivot[1], touch.pos[1])
yMax = min(self.currentLabelPivot[1], touch.pos[1])
self.currentLabelLeftTop = (xMin, yMin)
self.currentLabelSize = (xMax - xMin, yMin - yMax)
self.activeLabel.update(self.currentLabelLeftTop,
self.currentLabelSize)
def on_touch_up(self, touch):
if self.activeLabel is not None:
self.add_bb(self.activeLabel)
self.textureWidget.remove_widget(self.activeLabel)
self.currentLabelLeftTop = None
self.currentLabelPivot = None
self.activeLabel = None
