Remove content::BrowserThread knowledge from Device Sensors

content/browser/device_sensors/sensor_manager_android.cc

 // Copyright 2014 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.
 
 #include "content/browser/device_sensors/sensor_manager_android.h"
 
 #include <string.h>
 
 #include "base/android/context_utils.h"
 #include "base/android/jni_android.h"
 #include "base/bind.h"
 #include "base/memory/singleton.h"
 #include "base/metrics/histogram_macros.h"
-#include "content/public/browser/browser_thread.h"
 #include "jni/DeviceSensors_jni.h"
 
 using base::android::AttachCurrentThread;
 using base::android::JavaParamRef;
 
 namespace {
 
 void UpdateDeviceOrientationHistogram(
     content::SensorManagerAndroid::OrientationSensorType type) {
   UMA_HISTOGRAM_ENUMERATION("InertialSensor.DeviceOrientationSensorAndroid",
@@ skipping 25 matching lines @@
 
 namespace content {
 
 SensorManagerAndroid::SensorManagerAndroid()
     : number_active_device_motion_sensors_(0),
       device_light_buffer_(nullptr),
       device_motion_buffer_(nullptr),
       device_orientation_buffer_(nullptr),
       motion_buffer_initialized_(false),
       orientation_buffer_initialized_(false),
+      ui_task_runner_(nullptr),
       is_shutdown_(false) {
   memset(received_motion_data_, 0, sizeof(received_motion_data_));
   device_sensors_.Reset(Java_DeviceSensors_getInstance(
       AttachCurrentThread(), base::android::GetApplicationContext()));
 }
 
 SensorManagerAndroid::~SensorManagerAndroid() {
 }
 
 bool SensorManagerAndroid::Register(JNIEnv* env) {
@@ skipping 165 matching lines @@
       Java_DeviceSensors_getOrientationSensorTypeUsed(AttachCurrentThread(),
                                                       device_sensors_));
 }
 
 // ----- Shared memory API methods
 
 // --- Device Light
 
 void SensorManagerAndroid::StartFetchingDeviceLightData(
     DeviceLightHardwareBuffer* buffer) {
-  if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+  if (ui_task_runner_->RunsTasksOnCurrentThread()) {
     StartFetchingLightDataOnUI(buffer);
   } else {
-    BrowserThread::PostTask(
-        BrowserThread::UI, FROM_HERE,
-        base::Bind(&SensorManagerAndroid::StartFetchingLightDataOnUI,
-                   base::Unretained(this),
-                   buffer));
+    ui_task_runner_->PostTask(
+        FROM_HERE, base::Bind(&SensorManagerAndroid::StartFetchingLightDataOnUI,
+                              base::Unretained(this), buffer));
   }
 }
 
 void SensorManagerAndroid::StartFetchingLightDataOnUI(
     DeviceLightHardwareBuffer* buffer) {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   DCHECK(buffer);
   if (is_shutdown_)
     return;
 
   {
     base::AutoLock autolock(light_buffer_lock_);
     device_light_buffer_ = buffer;
     SetLightBufferValue(-1);
   }
   bool success = Start(CONSUMER_TYPE_LIGHT);
   if (!success) {
     base::AutoLock autolock(light_buffer_lock_);
     SetLightBufferValue(std::numeric_limits<double>::infinity());
   }
 }
 
 void SensorManagerAndroid::StopFetchingDeviceLightData() {
-  if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+  if (ui_task_runner_->RunsTasksOnCurrentThread()) {
     StopFetchingLightDataOnUI();
     return;
   }
 
-  BrowserThread::PostTask(
-      BrowserThread::UI, FROM_HERE,
-      base::Bind(&SensorManagerAndroid::StopFetchingLightDataOnUI,
-                 base::Unretained(this)));
+  ui_task_runner_->PostTask(
+      FROM_HERE, base::Bind(&SensorManagerAndroid::StopFetchingLightDataOnUI,
+                            base::Unretained(this)));
 }
 
 void SensorManagerAndroid::StopFetchingLightDataOnUI() {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   if (is_shutdown_)
     return;
 
   Stop(CONSUMER_TYPE_LIGHT);
   {
     base::AutoLock autolock(light_buffer_lock_);
     if (device_light_buffer_) {
       SetLightBufferValue(-1);
       device_light_buffer_ = nullptr;
     }
   }
 }
 
 void SensorManagerAndroid::SetLightBufferValue(double lux) {
   device_light_buffer_->seqlock.WriteBegin();
   device_light_buffer_->data.value = lux;
   device_light_buffer_->seqlock.WriteEnd();
 }
 
 // --- Device Motion
 
 void SensorManagerAndroid::StartFetchingDeviceMotionData(
     DeviceMotionHardwareBuffer* buffer) {
-  if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+  if (ui_task_runner_->RunsTasksOnCurrentThread()) {
     StartFetchingMotionDataOnUI(buffer);
   } else {
-    BrowserThread::PostTask(
-        BrowserThread::UI, FROM_HERE,
+    ui_task_runner_->PostTask(
+        FROM_HERE,
         base::Bind(&SensorManagerAndroid::StartFetchingMotionDataOnUI,
-                   base::Unretained(this),
-                   buffer));
+                   base::Unretained(this), buffer));
   }
 }
 
 void SensorManagerAndroid::StartFetchingMotionDataOnUI(
     DeviceMotionHardwareBuffer* buffer) {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   DCHECK(buffer);
   if (is_shutdown_)
     return;
 
   {
     base::AutoLock autolock(motion_buffer_lock_);
     device_motion_buffer_ = buffer;
     ClearInternalMotionBuffers();
   }
   Start(CONSUMER_TYPE_MOTION);
 
   // If no motion data can ever be provided, the number of active device motion
   // sensors will be zero. In that case flag the shared memory buffer
   // as ready to read, as it will not change anyway.
   number_active_device_motion_sensors_ = GetNumberActiveDeviceMotionSensors();
   {
     base::AutoLock autolock(motion_buffer_lock_);
     CheckMotionBufferReadyToRead();
   }
 }
 
 void SensorManagerAndroid::StopFetchingDeviceMotionData() {
-  if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+  if (ui_task_runner_->RunsTasksOnCurrentThread()) {
     StopFetchingMotionDataOnUI();
     return;
   }
 
-  BrowserThread::PostTask(
-      BrowserThread::UI, FROM_HERE,
-      base::Bind(&SensorManagerAndroid::StopFetchingMotionDataOnUI,
-                 base::Unretained(this)));
+  ui_task_runner_->PostTask(
+      FROM_HERE, base::Bind(&SensorManagerAndroid::StopFetchingMotionDataOnUI,
+                            base::Unretained(this)));
 }
 
 void SensorManagerAndroid::StopFetchingMotionDataOnUI() {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   if (is_shutdown_)
     return;
 
   Stop(CONSUMER_TYPE_MOTION);
   {
     base::AutoLock autolock(motion_buffer_lock_);
     if (device_motion_buffer_) {
       ClearInternalMotionBuffers();
       device_motion_buffer_ = nullptr;
     }
@@ skipping 32 matching lines @@
 void SensorManagerAndroid::ClearInternalMotionBuffers() {
   memset(received_motion_data_, 0, sizeof(received_motion_data_));
   number_active_device_motion_sensors_ = 0;
   SetMotionBufferReadyStatus(false);
 }
 
 // --- Device Orientation
 
 void SensorManagerAndroid::StartFetchingDeviceOrientationData(
     DeviceOrientationHardwareBuffer* buffer) {
-  if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+  if (ui_task_runner_->RunsTasksOnCurrentThread()) {
     StartFetchingOrientationDataOnUI(buffer);
   } else {
-    BrowserThread::PostTask(
-        BrowserThread::UI, FROM_HERE,
+    ui_task_runner_->PostTask(
+        FROM_HERE,
         base::Bind(&SensorManagerAndroid::StartFetchingOrientationDataOnUI,
-                   base::Unretained(this),
-                   buffer));
+                   base::Unretained(this), buffer));
   }
 }
 
 void SensorManagerAndroid::StartFetchingOrientationDataOnUI(
     DeviceOrientationHardwareBuffer* buffer) {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   DCHECK(buffer);
   if (is_shutdown_)
     return;
 
   {
     base::AutoLock autolock(orientation_buffer_lock_);
     device_orientation_buffer_ = buffer;
   }
   bool success = Start(CONSUMER_TYPE_ORIENTATION);
 
   {
     base::AutoLock autolock(orientation_buffer_lock_);
     // If Start() was unsuccessful then set the buffer ready flag to true
     // to start firing all-null events.
     SetOrientationBufferStatus(buffer, !success /* ready */,
         false /* absolute */);
     orientation_buffer_initialized_ = !success;
   }
 
   if (!success)
     UpdateDeviceOrientationHistogram(NOT_AVAILABLE);
 }
 
 void SensorManagerAndroid::StopFetchingDeviceOrientationData() {
-  if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+  if (ui_task_runner_->RunsTasksOnCurrentThread()) {
     StopFetchingOrientationDataOnUI();
     return;
   }
 
-  BrowserThread::PostTask(
-      BrowserThread::UI, FROM_HERE,
+  ui_task_runner_->PostTask(
+      FROM_HERE,
       base::Bind(&SensorManagerAndroid::StopFetchingOrientationDataOnUI,
                  base::Unretained(this)));
 }
 
 void SensorManagerAndroid::StopFetchingOrientationDataOnUI() {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   if (is_shutdown_)
     return;
 
   Stop(CONSUMER_TYPE_ORIENTATION);
   {
     base::AutoLock autolock(orientation_buffer_lock_);
     if (device_orientation_buffer_) {
       SetOrientationBufferStatus(device_orientation_buffer_, false, false);
       orientation_buffer_initialized_ = false;
       device_orientation_buffer_ = nullptr;
     }
   }
 }
 
 void SensorManagerAndroid::StartFetchingDeviceOrientationAbsoluteData(
     DeviceOrientationHardwareBuffer* buffer) {
-  if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+  if (ui_task_runner_->RunsTasksOnCurrentThread()) {
     StartFetchingOrientationAbsoluteDataOnUI(buffer);
   } else {
-    BrowserThread::PostTask(
-        BrowserThread::UI, FROM_HERE,
+    ui_task_runner_->PostTask(
+        FROM_HERE,
         base::Bind(
             &SensorManagerAndroid::StartFetchingOrientationAbsoluteDataOnUI,
-            base::Unretained(this),
-            buffer));
+            base::Unretained(this), buffer));
   }
 }
 
 void SensorManagerAndroid::StopFetchingDeviceOrientationAbsoluteData() {
-  if (BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+  if (ui_task_runner_->RunsTasksOnCurrentThread()) {
     StopFetchingOrientationAbsoluteDataOnUI();
     return;
   }
 
-  BrowserThread::PostTask(
-      BrowserThread::UI, FROM_HERE,
+  ui_task_runner_->PostTask(
+      FROM_HERE,
       base::Bind(&SensorManagerAndroid::StopFetchingOrientationAbsoluteDataOnUI,
                  base::Unretained(this)));
 }
 
 void SensorManagerAndroid::StartFetchingOrientationAbsoluteDataOnUI(
     DeviceOrientationHardwareBuffer* buffer) {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   DCHECK(buffer);
   if (is_shutdown_)
     return;
 
   {
     base::AutoLock autolock(orientation_absolute_buffer_lock_);
     device_orientation_absolute_buffer_ = buffer;
   }
   bool success = Start(CONSUMER_TYPE_ORIENTATION_ABSOLUTE);
 
   {
     base::AutoLock autolock(orientation_absolute_buffer_lock_);
     // If Start() was unsuccessful then set the buffer ready flag to true
     // to start firing all-null events.
     SetOrientationBufferStatus(buffer, !success /* ready */,
         false /* absolute */);
     orientation_absolute_buffer_initialized_ = !success;
   }
 }
 
 void SensorManagerAndroid::StopFetchingOrientationAbsoluteDataOnUI() {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   if (is_shutdown_)
     return;
 
   Stop(CONSUMER_TYPE_ORIENTATION_ABSOLUTE);
   {
     base::AutoLock autolock(orientation_absolute_buffer_lock_);
     if (device_orientation_absolute_buffer_) {
       SetOrientationBufferStatus(device_orientation_absolute_buffer_, false,
           false);
       orientation_absolute_buffer_initialized_ = false;
       device_orientation_absolute_buffer_ = nullptr;
     }
   }
 }
 
-void SensorManagerAndroid::Shutdown() {
-  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+void SensorManagerAndroid::SetUITaskRunner(
+    scoped_refptr<base::SequencedTaskRunner> ui_task_runner) {
+  ui_task_runner_ = ui_task_runner;
+  DCHECK(ui_task_runner_);
+}
+
+void SensorManagerAndroid::ShutDownOnUIThread() {
+  DCHECK(ui_task_runner_->RunsTasksOnCurrentThread());
   is_shutdown_ = true;
 }
 
 }  // namespace content