import bpy from bpy.app.handlers import persistent from .hmd import HMD from .preview import Preview from .lib import ( getDisplayBackend, ) import gpu # ############################################################ # Main Operator # ############################################################ class VirtualRealityDisplayOperator(bpy.types.Operator): """""" bl_idname = "view3d.virtual_reality_display" bl_label = "Toggle Virtual Reality Display" bl_description = "" _hmd = None _timer = None _handle = None _area_hash = -1 action = bpy.props.EnumProperty( description="", items=(("ENABLE", "Enable", "Enable"), ("DISABLE", "Disable", "Disable"), ("TOGGLE", "Toggle", "Toggle"), ), default="TOGGLE", options={'SKIP_SAVE'}, ) @classmethod def poll(cls, context): return context.area.type == 'VIEW_3D' def modal(self, context, event): wm = context.window_manager vr = wm.virtual_reality area = context.area if not area: self.quit(context) self._quit(context) return {'FINISHED'} if not vr.is_enabled: self._quit(context) area.tag_redraw() return {'FINISHED'} if event.type == 'TIMER': self.loop(context) if vr.preview_scale and context.area: area.tag_redraw() return {'PASS_THROUGH'} def invoke(self, context, event): wm = context.window_manager vr = wm.virtual_reality is_enabled = vr.is_enabled if self.action == 'TOGGLE': self.action = 'DISABLE' if is_enabled else 'ENABLE' if self.action == 'DISABLE': if vr.is_enabled: self.quit(context) return {'FINISHED'} else: self.report({'ERROR'}, "Virtual Reality Display is not enabled") return {'CANCELLED'} else: # ENABLE if vr.is_enabled: self.report({'ERROR'}, "Virtual Reality Display is already enabled") return {'CANCELLED'} if self.init(context): return {'RUNNING_MODAL'} else: # quit right away wm.virtual_reality.is_enabled = False self._quit(context) return {'CANCELLED'} def quit(self, context): """garbage collect""" # change it so the original modal operator will clean things up wm = context.window_manager vr = wm.virtual_reality vr.is_enabled = False vr.error_message = "" def _quit(self, context): """actual quit""" wm = context.window_manager if self._timer: wm.event_timer_remove(self._timer) del self._timer if self._handle: bpy.types.SpaceView3D.draw_handler_remove(self._handle, 'WINDOW') del self._handle self._preview.quit() if self._hmd: self._hmd.quit() # cleanup viewport if context.area: context.area.tag_redraw() def init(self, context): """ Initialize the callbacks and the external devices """ wm = context.window_manager vr = wm.virtual_reality vr.is_enabled = True vr.error_message = "" display_backend = getDisplayBackend(context) self._hmd = HMD(display_backend, self._error_callback) self._preview = Preview() if not self._hmd.init(context): self.report({'ERROR'}, "Error initializing device") return False # get the data from device color_object = [0, 0] for i in range(2): self._hmd.setEye(i) color_object[i] = self._hmd.color_object self._preview.init(color_object[0], color_object[1]) self._area_hash = hash(context.area) # setup modal self._timer = wm.event_timer_add(1.0 / 75.0, context.window) # 75 Hz self._handle = bpy.types.SpaceView3D.draw_handler_add(self._draw_callback_px, (context,), 'WINDOW', 'POST_PIXEL') wm.modal_handler_add(self) return True def loop(self, context): """ Get fresh tracking data and render into the FBO """ self._hmd.loop(context) for i in range(2): self._hmd.setEye(i) offscreen_object = self._hmd.offscreen_object projection_matrix = self._hmd.projection_matrix modelview_matrix = self._hmd.modelview_matrix # drawing gpu.offscreen_object_draw(offscreen_object, projection_matrix, modelview_matrix) self._hmd.frameReady() def _draw_callback_px(self, context): """callback function, run every time the viewport is refreshed""" if self._area_hash == hash(context.area): wm = context.window_manager vr = wm.virtual_reality self._preview.loop(vr.preview_scale) def _error_callback(self, message, is_fatal): """ Error handler, called from HMD class """ context = bpy.context wm = context.window_manager vr = wm.virtual_reality if is_fatal: self.report({'ERROR'}, message) self.quit(context) vr.error_message = message # ############################################################ # Global Properties # ############################################################ class VirtualRealityInfo(bpy.types.PropertyGroup): is_enabled = bpy.props.BoolProperty( name="Enabled", default=False, ) preview_scale = bpy.props.IntProperty( name="Preview Scale", min=0, max=100, default=100, subtype='PERCENTAGE', ) error_message = bpy.props.StringProperty( name="Error Message", ) # ############################################################ # Callbacks # ############################################################ @persistent def virtual_reality_load_pre(dummy): wm = bpy.context.window_manager wm.virtual_reality.is_enabled = False @persistent def virtual_reality_load_post(dummy): wm = bpy.context.window_manager vr = wm.virtual_reality vr.is_enabled = False vr.error_message = "" # ############################################################ # Un/Registration # ############################################################ def register(): bpy.app.handlers.load_pre.append(virtual_reality_load_pre) bpy.app.handlers.load_pre.append(virtual_reality_load_post) bpy.utils.register_class(VirtualRealityDisplayOperator) bpy.utils.register_class(VirtualRealityInfo) bpy.types.WindowManager.virtual_reality = bpy.props.PointerProperty( name="virtual_reality", type=VirtualRealityInfo, options={'HIDDEN'}, ) def unregister(): bpy.app.handlers.load_pre.remove(virtual_reality_load_pre) bpy.app.handlers.load_pre.remove(virtual_reality_load_post) bpy.utils.unregister_class(VirtualRealityDisplayOperator) del bpy.types.WindowManager.virtual_reality bpy.utils.unregister_class(VirtualRealityInfo)