administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 02:35:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add zooming

Browse Source
This commit is contained in:
Nick Mowen 2022-12-14 14:53:26 -07:00
parent f744827737
commit 895788543b
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
frigate/ptz.py
View File

@ -21,6 +21,8 @@ class OnvifCommandEnum(str, Enum):
move_up = "move_up" move_up = "move_up"
preset = "preset" preset = "preset"
stop = "stop" stop = "stop"
zoom_in = "zoom_in"
zoom_out = "zoom_out"
class OnvifController: class OnvifController:
@ -98,7 +100,7 @@ class OnvifController:
"y": 0.5, "y": 0.5,
} }
} }
else: elif command == OnvifCommandEnum.move_down:
move_request.Velocity = { move_request.Velocity = {
"PanTilt": { "PanTilt": {
"x": 0, "x": 0,
@ -125,6 +127,23 @@ class OnvifController:
) )
self.cams[camera_name]["active"] = False self.cams[camera_name]["active"] = False
def _zoom(self, camera_name: str, command: OnvifCommandEnum) -> None:
if self.cams[camera_name]["active"]:
logger.warning(
f"{camera_name} is already performing an action, stopping..."
)
self._stop(camera_name)
self.cams[camera_name]["active"] = True
onvif: ONVIFCamera = self.cams[camera_name]["onvif"]
move_request = self.cams[camera_name]["move_request"]
if command == OnvifCommandEnum.zoom_in:
move_request.Velocity = {"Zoom": {"x": 0.5}}
elif command == OnvifCommandEnum.zoom_out:
move_request.Velocity = {"PanTilt": {"x": -0.5}}
onvif.get_service("ptz").ContinuousMove(move_request)
def handle_command( def handle_command(
self, camera_name: str, command: OnvifCommandEnum, param: str self, camera_name: str, command: OnvifCommandEnum, param: str
@ -143,5 +162,9 @@ class OnvifController:
self._stop(camera_name) self._stop(camera_name)
elif command == OnvifCommandEnum.preset: elif command == OnvifCommandEnum.preset:
self._move_to_preset(camera_name, param) self._move_to_preset(camera_name, param)
elif (
command == OnvifCommandEnum.zoom_in or command == OnvifCommandEnum.zoom_out
):
self._zoom(camera_name, command)
else: else:
self._move(camera_name, command) self._move(camera_name, command)