forked from jdibenes/hl2ss
-
Notifications
You must be signed in to change notification settings - Fork 0
/
client_stream_pv.py
116 lines (94 loc) · 3.78 KB
/
client_stream_pv.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
#------------------------------------------------------------------------------
# This script receives video from the HoloLens front RGB camera and plays it.
# The camera supports various resolutions and framerates. See
# https://github.com/jdibenes/hl2ss/blob/main/etc/pv_configurations.txt
# for a list of supported formats. The default configuration is 1080p 30 FPS.
# The stream supports three operating modes: 0) video, 1) video + camera pose,
# 2) query calibration (single transfer).
# Press esc to stop.
#------------------------------------------------------------------------------
from pynput import keyboard
import cv2
import hl2ss_imshow
import hl2ss
import hl2ss_lnm
# Settings --------------------------------------------------------------------
# HoloLens address
host = "192.168.1.7"
# Operating mode
# 0: video
# 1: video + camera pose
# 2: query calibration (single transfer)
mode = hl2ss.StreamMode.MODE_1
# Enable Mixed Reality Capture (Holograms)
enable_mrc = False
# Enable Shared Capture
# If another program is already using the PV camera, you can still stream it by
# enabling shared mode, however you cannot change the resolution and framerate
shared = False
# Camera parameters
# Ignored in shared mode
width = 1920
height = 1080
framerate = 30
# Framerate denominator (must be > 0)
# Effective FPS is framerate / divisor
divisor = 1
# Video encoding profile and bitrate (None = default)
profile = hl2ss.VideoProfile.H265_MAIN
bitrate = None
# Decoded format
# Options include:
# 'bgr24'
# 'rgb24'
# 'bgra'
# 'rgba'
# 'gray8'
decoded_format = 'bgr24'
#------------------------------------------------------------------------------
hl2ss_lnm.start_subsystem_pv(host, hl2ss.StreamPort.PERSONAL_VIDEO, enable_mrc=enable_mrc, shared=shared)
if (mode == hl2ss.StreamMode.MODE_2):
data = hl2ss_lnm.download_calibration_pv(host, hl2ss.StreamPort.PERSONAL_VIDEO, width, height, framerate)
print('Calibration')
print(f'Focal length: {data.focal_length}')
print(f'Principal point: {data.principal_point}')
print(f'Radial distortion: {data.radial_distortion}')
print(f'Tangential distortion: {data.tangential_distortion}')
print('Projection')
print(data.projection)
print('Intrinsics')
print(data.intrinsics)
print('RigNode Extrinsics')
print(data.extrinsics)
print(f'Intrinsics MF: {data.intrinsics_mf}')
print(f'Extrinsics MF: {data.extrinsics_mf}')
else:
enable = True
def on_press(key):
global enable
enable = key != keyboard.Key.esc
return enable
listener = keyboard.Listener(on_press=on_press)
listener.start()
client = hl2ss_lnm.rx_pv(host, hl2ss.StreamPort.PERSONAL_VIDEO, mode=mode, width=width, height=height, framerate=framerate, divisor=divisor, profile=profile, bitrate=bitrate, decoded_format=decoded_format)
client.open()
while (enable):
data = client.get_next_packet()
print(f'Frame captured at {data.timestamp}')
print(f'Focal length: {data.payload.focal_length}')
print(f'Principal point: {data.payload.principal_point}')
print(f'Exposure Time: {data.payload.exposure_time}')
print(f'Exposure Compensation: {data.payload.exposure_compensation}')
print(f'Lens Position (Focus): {data.payload.lens_position}')
print(f'Focus State: {data.payload.focus_state}')
print(f'ISO Speed: {data.payload.iso_speed}')
print(f'White Balance: {data.payload.white_balance}')
print(f'ISO Gains: {data.payload.iso_gains}')
print(f'White Balance Gains: {data.payload.white_balance_gains}')
print(f'Pose')
print(data.pose)
cv2.imshow('Video', data.payload.image)
cv2.waitKey(1)
client.close()
listener.join()
hl2ss_lnm.stop_subsystem_pv(host, hl2ss.StreamPort.PERSONAL_VIDEO)