feat(android): introduce new rotation sensor (#80)

Author: gladiuscode

android/src/main/java/com/orientationdirector/implementation/OrientationSensorsEventListener.kt

@@ -10,12 +10,19 @@ import com.facebook.react.bridge.ReactApplicationContext
 class OrientationSensorsEventListener(
   context: ReactApplicationContext,
 ) : SensorEventListener {
-  private var mSensorManager: SensorManager = context.getSystemService(SENSOR_SERVICE) as SensorManager
+  private var mSensorManager: SensorManager =
+    context.getSystemService(SENSOR_SERVICE) as SensorManager
 
-  private var mAccelerometerSensor: Sensor? = mSensorManager.getDefaultSensor(Sensor.TYPE_ACCELEROMETER)
-  private var mMagneticFieldSensor: Sensor? = mSensorManager.getDefaultSensor(Sensor.TYPE_MAGNETIC_FIELD)
+  private var mRotationSensor: Sensor? =
+    mSensorManager.getDefaultSensor(Sensor.TYPE_ROTATION_VECTOR)
+  private var mAccelerometerSensor: Sensor? =
+    mSensorManager.getDefaultSensor(Sensor.TYPE_ACCELEROMETER)
+  private var mMagneticFieldSensor: Sensor? =
+    mSensorManager.getDefaultSensor(Sensor.TYPE_MAGNETIC_FIELD)
 
-  private var hasRequiredSensors: Boolean = mAccelerometerSensor != null && mMagneticFieldSensor != null
+  private var hasRotationSensor: Boolean = mRotationSensor != null
+  private var hasAccelerometerAndMagneticFieldSensors: Boolean =
+    mAccelerometerSensor != null && mMagneticFieldSensor != null
 
   private val accelerometerReading = FloatArray(3)
   private val magnetometerReading = FloatArray(3)
@@ -32,6 +39,59 @@ class OrientationSensorsEventListener(
       return
     }
 
+    if (event.sensor.type == Sensor.TYPE_ROTATION_VECTOR) {
+      computeOrientationFromRotationSensor(event.values);
+      return
+    }
+
+    computeOrientationFromOtherSensors(event)
+  }
+
+  override fun onAccuracyChanged(sensor: Sensor?, accuracy: Int) {}
+
+  fun enable() {
+    if (hasRotationSensor) {
+      mSensorManager.registerListener(
+        this,
+        mRotationSensor,
+        SensorManager.SENSOR_DELAY_NORMAL,
+        SensorManager.SENSOR_DELAY_UI
+      )
+      return
+    }
+
+    if (hasAccelerometerAndMagneticFieldSensors) {
+      mSensorManager.registerListener(
+        this,
+        mAccelerometerSensor,
+        SensorManager.SENSOR_DELAY_NORMAL,
+        SensorManager.SENSOR_DELAY_UI
+      )
+      mSensorManager.registerListener(
+        this,
+        mMagneticFieldSensor,
+        SensorManager.SENSOR_DELAY_NORMAL,
+        SensorManager.SENSOR_DELAY_UI
+      )
+      return
+    }
+  }
+
+  fun disable() {
+    mSensorManager.unregisterListener(this)
+  }
+
+  private fun computeOrientationFromRotationSensor(values: FloatArray) {
+    val rotationMatrix = FloatArray(9)
+    SensorManager.getRotationMatrixFromVector(rotationMatrix, values)
+
+    val orientationAngles = FloatArray(3)
+    SensorManager.getOrientation(rotationMatrix, orientationAngles)
+
+    notifyOrientationAnglesChanged(orientationAngles)
+  }
+
+  private fun computeOrientationFromOtherSensors(event: SensorEvent) {
     if (event.sensor.type == Sensor.TYPE_ACCELEROMETER) {
       System.arraycopy(event.values, 0, accelerometerReading, 0, accelerometerReading.size)
     }
@@ -54,26 +114,15 @@ class OrientationSensorsEventListener(
     val orientationAngles = FloatArray(3)
     SensorManager.getOrientation(rotationMatrix, orientationAngles)
 
+    notifyOrientationAnglesChanged(orientationAngles)
+  }
+
+  private fun notifyOrientationAnglesChanged(orientationAngles: FloatArray) {
     if (lastComputedOrientationAngles.contentEquals(orientationAngles)) {
       return
     }
 
     onOrientationAnglesChangedCallback?.invoke(orientationAngles)
     lastComputedOrientationAngles = orientationAngles
   }
-
-  override fun onAccuracyChanged(sensor: Sensor?, accuracy: Int) {}
-
-  fun enable() {
-    if (!hasRequiredSensors) {
-      return
-    }
-
-    mSensorManager.registerListener(this, mAccelerometerSensor, SensorManager.SENSOR_DELAY_NORMAL)
-    mSensorManager.registerListener(this, mMagneticFieldSensor, SensorManager.SENSOR_DELAY_NORMAL)
-  }
-
-  fun disable() {
-    mSensorManager.unregisterListener(this)
-  }
 }