Find lid sensor by motionsense commands

Useful in case the lid sensor is not the first one in the memmap.
On Framework 12 it currently is (BIOS 3.03) but it might not be in the
future.

Signed-off-by: Daniel Schaefer <dhs@frame.work>

Author: JohnAZoidberg

FrameworkSensors/AccelerometerClient.cpp

@@ -61,6 +61,63 @@ UINT8 CrosEcGetMotionSensorCount(HANDLE Handle)
     return res.SensorCount;
 }
 
+// Returns STATUS_NOT_FOUND if either base or lid accelerometer sensors are not found.
+NTSTATUS
+CrosEcGetAccelIndeces(HANDLE Handle, UINT8 *BaseSensor, UINT8 *LidSensor)
+{
+    EC_REQUEST_MOTION_SENSE_INFO req{};
+    EC_RESPONSE_MOTION_SENSE_INFO res{};
+    UINT8 SensorCount = CrosEcGetMotionSensorCount(Handle);
+    BOOLEAN FoundBase = FALSE;
+    BOOLEAN FoundLid = FALSE;
+
+    if (BaseSensor == nullptr || LidSensor == nullptr)
+    {
+        TraceError("%!FUNC! Invalid BaseSensor or LidSensor pointer");
+        return STATUS_INVALID_PARAMETER;
+    }
+
+    for (UINT8 i = 0; i < SensorCount; i++)
+    {
+        req.Cmd = 1;
+        req.SensorNum = i;
+        if (0 == CrosEcSendCommand(
+            Handle,
+            EC_CMD_MOTION_SENSE,
+            0,
+            &req,
+            sizeof(req),
+            &res,
+            sizeof(res)
+        )) {
+            TraceError("%!FUNC! EC_CMD_MOTION_SENSE_INFO failed for sensor %d", i);
+            continue;
+        }
+        if (res.SensorType != MOTION_SENSE_TYPE_ACCEL) {
+            continue;
+        }
+
+        switch (res.Location) {
+        case MOTION_SENSE_LOCATION_BASE:
+            FoundBase = TRUE;
+            *BaseSensor = i;
+            break;
+        case MOTION_SENSE_LOCATION_LID:
+            FoundLid = TRUE;
+            *LidSensor = i;
+            break;
+        }
+    }
+
+    if (!FoundBase || !FoundLid)
+    {
+        TraceError("%!FUNC! Base or Lid accelerometer sensor not found");
+        return STATUS_NOT_FOUND;
+    }
+
+    return STATUS_SUCCESS;
+}
+
 
 //------------------------------------------------------------------------------
 // Function: Initialize
@@ -102,6 +159,13 @@ AccelerometerDevice::Initialize(
         goto Exit;
     }
 
+    Status = CrosEcGetAccelIndeces(Context->m_CrosEcHandle, &m_LidSensorIndex, &m_LidBaseSensor);
+    if (!NT_SUCCESS(Status))
+    {
+        TraceError("%!FUNC! Failed to get accelerometer indeces: %!STATUS!", Status);
+        goto Exit;
+    }
+
     //
     // Create Lock
     //
@@ -446,13 +510,17 @@ AccelerometerDevice::GetData(
     UINT16 lid_angle = lid_angle_bytes[0] + (lid_angle_bytes[1] << 8);
     TraceInformation("Lid Angle Status: %dDeg%s", lid_angle, lid_angle == 500 ?  "(Unreliable)" : "");
 
+    // Lid accelerometer is relevant for screen rotation
+    // Base accelerometer is not used in this driver
+    // It's only used for lid angle in the EC firmware
+    UINT SensorOffset = 6 * m_LidSensorIndex + EC_MEMMAP_ACC_DATA + 2;
     UINT16 Sensor1[6] = {0};
-    CrosEcReadMemU8(Handle, EC_MEMMAP_ACC_DATA + 2, (UINT8*)&Sensor1[0]);
-    CrosEcReadMemU8(Handle, EC_MEMMAP_ACC_DATA + 3, (UINT8*)&Sensor1[1]);
-    CrosEcReadMemU8(Handle, EC_MEMMAP_ACC_DATA + 4, (UINT8*)&Sensor1[2]);
-    CrosEcReadMemU8(Handle, EC_MEMMAP_ACC_DATA + 5, (UINT8*)&Sensor1[3]);
-    CrosEcReadMemU8(Handle, EC_MEMMAP_ACC_DATA + 6, (UINT8*)&Sensor1[4]);
-    CrosEcReadMemU8(Handle, EC_MEMMAP_ACC_DATA + 7, (UINT8*)&Sensor1[5]);
+    CrosEcReadMemU8(Handle, SensorOffset, (UINT8*)&Sensor1[0]);
+    CrosEcReadMemU8(Handle, SensorOffset + 1, (UINT8*)&Sensor1[1]);
+    CrosEcReadMemU8(Handle, SensorOffset + 2, (UINT8*)&Sensor1[2]);
+    CrosEcReadMemU8(Handle, SensorOffset + 3, (UINT8*)&Sensor1[3]);
+    CrosEcReadMemU8(Handle, SensorOffset + 4, (UINT8*)&Sensor1[4]);
+    CrosEcReadMemU8(Handle, SensorOffset + 5, (UINT8*)&Sensor1[5]);
     m_CachedData.Axis.X = (float) (Sensor1[0] + (Sensor1[1] << 8));
     m_CachedData.Axis.Y = (float) (Sensor1[2] + (Sensor1[3] << 8));
     m_CachedData.Axis.Z = (float) (Sensor1[4] + (Sensor1[5] << 8));

FrameworkSensors/Clients.h

@@ -221,6 +221,8 @@ typedef class _AccelerometerDevice : public _ComboDevice
     AccelerometerSample                       m_CachedThresholds;
     AccelerometerSample                       m_CachedData;
     AccelerometerSample                       m_LastSample;
+    UINT8                                     m_LidSensorIndex;
+    UINT8                                     m_LidBaseSensor;
 
 public:
 

FrameworkSensors/EcCommunication.h

@@ -109,6 +109,28 @@ typedef struct {
 	// UINT8 Sensors[];
 } EC_RESPONSE_MOTION_SENSE_DUMP;
 
+typedef struct {
+	// Info = 1
+	UINT8 Cmd;
+	UINT8 SensorNum;
+} EC_REQUEST_MOTION_SENSE_INFO;
+
+#define MOTION_SENSE_TYPE_ACCEL   0x00
+#define MOTION_SENSE_TYPE_GYRO    0x01
+#define MOTION_SENSE_TYPE_MAG     0x02
+#define MOTION_SENSE_TYPE_PROX    0x03
+#define MOTION_SENSE_TYPE_LIGHT   0x04
+
+#define MOTION_SENSE_LOCATION_BASE   0x00
+#define MOTION_SENSE_LOCATION_LID    0x01
+#define MOTION_SENSE_LOCATION_CAMERA 0x02
+
+typedef struct {
+	UINT8 SensorType;
+	UINT8 Location;
+	UINT8 Chip;
+} EC_RESPONSE_MOTION_SENSE_INFO;
+
 #include <poppack.h>
 
 int CrosEcSendCommand(