Add event listening function to openvr device

device/vr/openvr/openvr_device.cc

 // Copyright (c) 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #define _USE_MATH_DEFINES  // for M_PI
 #include "device/vr/openvr/openvr_device.h"
 #include <math.h>
 #include "third_party/openvr/src/headers/openvr.h"
 
 namespace {
@@ skipping 58 matching lines @@
   transform[13] = mat.m[1][3];
   transform[14] = mat.m[2][3];
   transform[15] = 1.0f;
   return transform;
 }
 
 }  // namespace
 
 namespace device {
 
-OpenVRDevice::OpenVRDevice() {}
+OpenVRDevice::OpenVRDevice() : weak_ptr_factory_(this) {}
 OpenVRDevice::~OpenVRDevice() {}
 
 void OpenVRDevice::CreateVRDisplayInfo(
     const base::Callback<void(mojom::VRDisplayInfoPtr)>& on_created) {
-  vr::EVRInitError init_error;
+  // Initialize init_error with success.
+  vr::EVRInitError init_error = vr::VRInitError_None;
+
+  // vr_system here could be used as openvr device initialization or update
+  // device info.
   auto vr_system =

[nhu, 2017/04/25 00:52:19]

In https://codereview.chromium.org/2825203004/, we consolidated the VR_Init to
OpenVRDeviceProvider to avoid multiple initializations. Will you follow that
approach? billorr@, any thoughts?

-      vr::VR_Init(&init_error, vr::EVRApplicationType::VRApplication_Scene);
+      render_loop_ == nullptr
+          ? vr::VR_Init(&init_error,
+                        vr::EVRApplicationType::VRApplication_Scene)
+          : render_loop_->GetVRSystem();
 
   if (init_error != vr::VRInitError_None) {
     LOG(ERROR) << vr::VR_GetVRInitErrorAsEnglishDescription(init_error);
     on_created.Run(nullptr);
     return;
   }
 
+  if (vr_system == nullptr) {
+    LOG(ERROR) << "OpenVRRenderLoop failed to return valid IVRSystem pointer";
+    on_created.Run(nullptr);
+    return;
+  }
+
   mojom::VRDisplayInfoPtr device = mojom::VRDisplayInfo::New();
   device->index = id();
   device->displayName =
       GetOpenVRString(vr_system, vr::Prop_ManufacturerName_String) + " " +
       GetOpenVRString(vr_system, vr::Prop_ModelNumber_String);
   device->capabilities = mojom::VRDisplayCapabilities::New();
   device->capabilities->hasPosition = true;
   device->capabilities->hasExternalDisplay = true;
   device->capabilities->canPresent = false;
 
@@ skipping 36 matching lines @@
 
   vr::IVRChaperone* chaperone = vr::VRChaperone();
   if (chaperone) {
     chaperone->GetPlayAreaSize(&device->stageParameters->sizeX,
                                &device->stageParameters->sizeZ);
   } else {
     device->stageParameters->sizeX = 0.0f;
     device->stageParameters->sizeZ = 0.0f;
   }
 
-  render_loop_ = std::make_unique<OpenVRRenderLoop>(vr_system);
+  // If it is first initialization, OpenVRRenderLoop instance need to be
+  // careated and callback need to be registed.
+  if (!render_loop_) {
+    render_loop_ = std::make_unique<OpenVRRenderLoop>(vr_system);
+
+    render_loop_->RegistPollingEventCallback(base::Bind(
+        &OpenVRDevice::OnPollingEvent, weak_ptr_factory_.GetWeakPtr()));
+  }
 
   on_created.Run(std::move(device));
 }
 
 void OpenVRDevice::RequestPresent(mojom::VRSubmitFrameClientPtr submit_client,
                                   const base::Callback<void(bool)>& callback) {
   callback.Run(false);
   // We don't support presentation currently.
 }
 
@@ skipping 11 matching lines @@
 }
 
 void OpenVRDevice::UpdateLayerBounds(int16_t frame_index,
                                      mojom::VRLayerBoundsPtr left_bounds,
                                      mojom::VRLayerBoundsPtr right_bounds,
                                      int16_t source_width,
                                      int16_t source_height) {
   // We don't support presentation currently, so don't do anything.
 }
 
+// TODO(shaobo): OnConnect and OnDisconnect need to be added to VrDisplayClient
+// in vr_service.mojom
+void OpenVRDevice::OnPollingEvent(OpenVRDevice::VREvent event) {
+  switch (event) {
+    case OpenVRDevice::VREvent::CONNECTED:
+    case OpenVRDevice::VREvent::DISCONNECTED:
+      break;
+
+    case OpenVRDevice::VREvent::CHANGED:
+      OnChanged();
+      break;
+
+    default:
+      break;
+  }
+}
+
 void OpenVRDevice::GetVRVSyncProvider(mojom::VRVSyncProviderRequest request) {
   render_loop_->Bind(std::move(request));
 }
 
 OpenVRDevice::OpenVRRenderLoop::OpenVRRenderLoop(vr::IVRSystem* vr_system)
     : vr_system_(vr_system),
       binding_(this),
       base::SimpleThread("OpenVRRenderLoop") {}
 
 void OpenVRDevice::OpenVRRenderLoop::Bind(
@@ skipping 39 matching lines @@
     pose->position.value()[1] = m[1][3];
     pose->position.value()[2] = m[2][3];
 
     pose->linearVelocity = HmdVector3ToWebVR(hmdPose.vVelocity);
     pose->angularVelocity = HmdVector3ToWebVR(hmdPose.vAngularVelocity);
   }
 
   return std::move(pose);
 }
 
+// TODO(shaobo) : Expose pointer directly will increase the risk that behaviour
+// out of OpenVRRenderLoop will affect OpenVRRenderLoop. Need to fix it.
+vr::IVRSystem* OpenVRDevice::OpenVRRenderLoop::GetVRSystem() {
+  return vr_system_;
+}
+
+// Only deal with event which will cause displayInfo changed now.
+void OpenVRDevice::OpenVRRenderLoop::PollEvents() {
+  if (!vr_system_ || on_polling_event_.is_null())
+    return;
+
+  vr::VREvent_t event;
+  bool is_changed = false;
+  while (vr_system_->PollNextEvent(&event, sizeof(event))) {
+    if (event.trackedDeviceIndex != vr::k_unTrackedDeviceIndex_Hmd &&
+        event.trackedDeviceIndex != vr::k_unTrackedDeviceIndexInvalid) {
+      continue;
+    }
+
+    switch (event.eventType) {
+      case vr::VREvent_TrackedDeviceActivated:
+        on_polling_event_.Run(OpenVRDevice::VREvent::CONNECTED);
+        break;
+
+      case vr::VREvent_TrackedDeviceDeactivated:
+        on_polling_event_.Run(OpenVRDevice::VREvent::DISCONNECTED);
+        break;
+
+      case vr::VREvent_TrackedDeviceUpdated:
+      case vr::VREvent_IpdChanged:
+      case vr::VREvent_ChaperoneDataHasChanged:
+      case vr::VREvent_ChaperoneSettingsHaveChanged:
+      case vr::VREvent_ChaperoneUniverseHasChanged:
+        is_changed = true;
+        break;
+
+      default:
+        break;
+    }
+  }
+
+  if (is_changed) {
+    on_polling_event_.Run(OpenVRDevice::VREvent::CHANGED);
+  }
+}
+
+// Regist a callback function to deal with system event.
+void OpenVRDevice::OpenVRRenderLoop::RegistPollingEventCallback(
+    const base::Callback<void(OpenVRDevice::VREvent)>& onPollingEvent) {
+  if (onPollingEvent.is_null())
+    return;
+
+  on_polling_event_ = onPollingEvent;
+}
+
+void OpenVRDevice::OpenVRRenderLoop::UnregistPollingEventCallback() {
+  on_polling_event_.Reset();
+}
+
 void OpenVRDevice::OpenVRRenderLoop::GetVSync(
     const mojom::VRVSyncProvider::GetVSyncCallback& callback) {
   static int16_t next_frame = 0;
   int16_t frame = next_frame++;
 
+  // VSync could be used as a signal to poll event.
+  PollEvents();
+
   // TODO(BillOrr): Give real values when VSync loop is hooked up.  This is the
   // presentation time for the frame. Just returning a default value for now
   // since we don't have VSync hooked up.
   base::TimeDelta time = base::TimeDelta::FromSecondsD(2.0);
 
   device::mojom::VRPosePtr pose = getPose();
   Sleep(11);  // TODO (billorr): Use real vsync timing instead of a sleep (this
               // sleep just throttles vsyncs so we don't fill message queues).
   callback.Run(std::move(pose), time, frame,
                device::mojom::VRVSyncProvider::Status::SUCCESS);
 }
 
 }  // namespace device