From fc0874544eb77c518c206a581562dfe3897a4d0c Mon Sep 17 00:00:00 2001
From: Cosmin Tanislav <demonsingur@gmail.com>
Date: Tue, 21 Sep 2021 02:43:52 +0300
Subject: [PATCH] rosemary: sensors: Standardize xiaomi pickup sensor

 * Use standard sensor type
 * Ignore non-wakeup sensor variant
 * Ignore events that do not properly match a pickup

[LinkBoi00] Adapt sensor name for our device

Change-Id: I2bab89362aa225344a69c4969096ea9a61ef3033
Signed-off-by: LinkBoi00 <linkdevel@protonmail.com>
---
 sensors/HalProxy.cpp         | 24 ++++++++++++++++++++++++
 sensors/HalProxyCallback.cpp |  8 +++++++-
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/sensors/HalProxy.cpp b/sensors/HalProxy.cpp
index ecc3740..f95f3d2 100644
--- a/sensors/HalProxy.cpp
+++ b/sensors/HalProxy.cpp
@@ -81,6 +81,25 @@ int64_t msFromNs(int64_t nanos) {
     return nanos / nanosecondsInAMillsecond;
 }
 
+bool patchXiaomiPickupSensor(V2_1::SensorInfo& sensor) {
+    if (sensor.typeAsString != "xiaomi pick up sensor") {
+        return true;
+    }
+
+    /*
+     * Implement only the wake-up version of this sensor.
+     */
+    if (!(sensor.flags & V1_0::SensorFlagBits::WAKE_UP)) {
+        return false;
+    }
+
+    sensor.type = V2_1::SensorType::PICK_UP_GESTURE;
+    sensor.typeAsString = SENSOR_STRING_TYPE_PICK_UP_GESTURE;
+    sensor.maxRange = 1;
+
+    return true;
+}
+
 HalProxy::HalProxy() {
     const char* kMultiHalConfigFile = "/vendor/etc/sensors/hals.conf";
     initializeSubHalListFromConfigFile(kMultiHalConfigFile);
@@ -491,6 +510,11 @@ void HalProxy::initializeSensorList() {
                     ALOGV("Loaded sensor: %s", sensor.name.c_str());
                     sensor.sensorHandle = setSubHalIndex(sensor.sensorHandle, subHalIndex);
                     setDirectChannelFlags(&sensor, mSubHalList[subHalIndex]);
+                    bool keep = patchXiaomiPickupSensor(sensor);
+                    if (!keep) {
+                        continue;
+                    }
+
                     mSensors[sensor.sensorHandle] = sensor;
                 }
             }
diff --git a/sensors/HalProxyCallback.cpp b/sensors/HalProxyCallback.cpp
index 3c1b17c..122c382 100644
--- a/sensors/HalProxyCallback.cpp
+++ b/sensors/HalProxyCallback.cpp
@@ -68,11 +68,17 @@ std::vector<V2_1::Event> HalProxyCallbackBase::processEvents(const std::vector<V
     std::vector<V2_1::Event> eventsOut;
     for (V2_1::Event event : events) {
         event.sensorHandle = setSubHalIndex(event.sensorHandle, mSubHalIndex);
-        eventsOut.push_back(event);
         const V2_1::SensorInfo& sensor = mCallback->getSensorInfo(event.sensorHandle);
+
+        if (sensor.type == V2_1::SensorType::PICK_UP_GESTURE
+            && event.u.scalar != 1) {
+            continue;
+        }
+
         if ((sensor.flags & V1_0::SensorFlagBits::WAKE_UP) != 0) {
             (*numWakeupEvents)++;
         }
+        eventsOut.push_back(event);
     }
     return eventsOut;
 }