def __init__(self, port):
""" Constructor
"""
super(CaptureService, self).__init__(port, True)
self._update_state(CaptureService.StateValue.OK)
self.__hyperion_connector = None
self.__delay_timer = DelayTimer()
# Register settings.
self._register_settings_unit([Setting.HYPERION_IP_ADDRESS,
Setting.HYPERION_PROTO_PORT],
self.__update_hyperion_connector)
self._register_settings_unit([Setting.CAPTURE_FRAME_RATE],
self.__update_timer)
self._register_settings_unit([Setting.CAPTURE_SCALE_WIDTH,
Setting.CAPTURE_SCALE_HEIGHT,
Setting.CAPTURE_PRIORITY])
class CaptureService(BaseService):
""" Capture Service class.
"""
class StateValue(object):
""" State Value class.
"""
OK = 1
ERROR = 2
__ERROR_DELAY = 5
__IMAGE_DURATION = 500
def __init__(self, port):
""" Constructor
"""
super(CaptureService, self).__init__(port, True)
self._update_state(CaptureService.StateValue.OK)
self.__hyperion_connector = None
self.__delay_timer = DelayTimer()
# Register settings.
self._register_settings_unit([Setting.HYPERION_IP_ADDRESS,
Setting.HYPERION_PROTO_PORT],
self.__update_hyperion_connector)
self._register_settings_unit([Setting.CAPTURE_FRAME_RATE],
self.__update_timer)
self._register_settings_unit([Setting.CAPTURE_SCALE_WIDTH,
Setting.CAPTURE_SCALE_HEIGHT,
Setting.CAPTURE_PRIORITY])
def _setup(self):
self.__update_hyperion_connector()
def _enable(self, enable):
if enable:
try:
self.__hyperion_connector.connect()
except HyperionError:
pass
else:
self.__hyperion_connector.disconnect()
def __update_hyperion_connector(self):
if self.__hyperion_connector is not None:
self.__hyperion_connector.disconnect()
self.__hyperion_connector = HyperionProto(
self._get_setting(Setting.HYPERION_IP_ADDRESS),
self._get_setting(Setting.HYPERION_PROTO_PORT))
def __update_timer(self):
self.__delay_timer.set_delay(
1.0 / self._get_setting(Setting.CAPTURE_FRAME_RATE))
def __update_pixel_buffer(self):
self.__data = self.scale_pixel_buffer(
self.get_pixel_buffer(),
self._get_setting(Setting.CAPTURE_SCALE_WIDTH),
self._get_setting(Setting.CAPTURE_SCALE_HEIGHT))
def _run_service(self):
self.__delay_timer.start()
try:
# Check that an hyperion connection is available.
if not self.__hyperion_connector.is_connected():
self.__hyperion_connector.connect()
self._update_state(CaptureService.StateValue.OK)
# Capture frame.
self.__update_pixel_buffer()
# Send to hyperion server.
self.__hyperion_connector.send_image(
self._get_setting(Setting.CAPTURE_SCALE_WIDTH),
self._get_setting(Setting.CAPTURE_SCALE_HEIGHT),
self.__data.get_pixels(),
self._get_setting(Setting.CAPTURE_PRIORITY),
CaptureService.__IMAGE_DURATION)
except HyperionError as err:
self._update_state(CaptureService.StateValue.ERROR, err.msg)
self._safe_delay(CaptureService.__ERROR_DELAY)
# Wait until next run.
self.__delay_timer.delay()
# TODO Maybe use Gst?
@staticmethod
def get_pixel_buffer():
win = Gdk.get_default_root_window()
h = win.get_height()
w = win.get_width()
return Gdk.pixbuf_get_from_window(win, 0, 0, w, h)
# @staticmethod
# def get_pixel_buffer():
# from PyQt5.QtWidgets import QApplication
# app = QApplication([])
# return QApplication.screens()[0].grabWindow(
# QApplication.desktop().winId()).toImage()
@staticmethod
def scale_pixel_buffer(pixel_buffer, width, height):
return pixel_buffer.scale_simple(width, height,
GdkPixbuf.InterpType.BILINEAR)