| /* |
| * Copyright (C) 2016 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <errno.h> |
| #include <float.h> |
| #include <stdlib.h> |
| #include <string.h> |
| |
| #include <eventnums.h> |
| #include <gpio.h> |
| #include <heap.h> |
| #include <hostIntf.h> |
| #include <isr.h> |
| #include <i2c.h> |
| #include <nanohubPacket.h> |
| #include <sensors.h> |
| #include <seos.h> |
| #include <timer.h> |
| #include <util.h> |
| |
| #include <cpu/inc/cpuMath.h> |
| |
| #include <plat/inc/exti.h> |
| #include <plat/inc/gpio.h> |
| #include <plat/inc/syscfg.h> |
| |
| #define S3708_APP_ID APP_ID_MAKE(APP_ID_VENDOR_GOOGLE, 13) |
| #define S3708_APP_VERSION 2 |
| |
| #define I2C_BUS_ID 0 |
| #define I2C_SPEED 400000 |
| #define I2C_ADDR 0x20 |
| |
| #define S3708_REG_PAGE_SELECT 0xFF |
| |
| #define S3708_REG_F01_DATA_BASE 0x06 |
| #define S3708_INT_STATUS_LPWG 0x04 |
| |
| #define S3708_REG_DATA_BASE 0x08 |
| #define S3708_REG_DATA_4_OFFSET 0x02 |
| #define S3708_INT_STATUS_DOUBLE_TAP 0x03 |
| |
| #define S3708_REG_F01_CTRL_BASE 0x14 |
| #define S3708_NORMAL_MODE 0x00 |
| #define S3708_SLEEP_MODE 0x01 |
| |
| #define S3708_REG_CTRL_BASE 0x1b |
| #define S3708_REG_CTRL_20_OFFSET 0x07 |
| #define S3708_REPORT_MODE_CONT 0x00 |
| #define S3708_REPORT_MODE_LPWG 0x02 |
| |
| #define MAX_PENDING_I2C_REQUESTS 4 |
| #define MAX_I2C_TRANSFER_SIZE 8 |
| #define MAX_I2C_RETRY_DELAY 250000000ull // 250 milliseconds |
| #define MAX_I2C_RETRY_COUNT (15000000000ull / MAX_I2C_RETRY_DELAY) // 15 seconds |
| #define HACK_RETRY_SKIP_COUNT 1 |
| |
| #define DEFAULT_PROX_RATE_HZ SENSOR_HZ(5.0f) |
| #define DEFAULT_PROX_LATENCY 0.0 |
| #define PROXIMITY_THRESH_NEAR 5.0f // distance in cm |
| |
| #define EVT_SENSOR_PROX sensorGetMyEventType(SENS_TYPE_PROX) |
| |
| #define ENABLE_DEBUG 0 |
| |
| #define INFO_PRINT(fmt, ...) osLog(LOG_INFO, "[DoubleTouch] " fmt, ##__VA_ARGS__) |
| #define ERROR_PRINT(fmt, ...) osLog(LOG_ERROR, "[DoubleTouch] " fmt, ##__VA_ARGS__) |
| #if ENABLE_DEBUG |
| #define DEBUG_PRINT(fmt, ...) INFO_PRINT(fmt, ##__VA_ARGS__) |
| #else |
| #define DEBUG_PRINT(fmt, ...) ((void)0) |
| #endif |
| |
| |
| #ifndef TOUCH_PIN |
| #error "TOUCH_PIN is not defined; please define in variant.h" |
| #endif |
| |
| #ifndef TOUCH_IRQ |
| #error "TOUCH_IRQ is not defined; please define in variant.h" |
| #endif |
| |
| enum SensorEvents |
| { |
| EVT_SENSOR_I2C = EVT_APP_START + 1, |
| EVT_SENSOR_TOUCH_INTERRUPT, |
| EVT_SENSOR_RETRY_TIMER, |
| }; |
| |
| enum TaskState |
| { |
| STATE_ENABLE_0, |
| STATE_ENABLE_1, |
| STATE_DISABLE_0, |
| STATE_INT_HANDLE_0, |
| STATE_INT_HANDLE_1, |
| STATE_IDLE, |
| STATE_CANCELLED, |
| }; |
| |
| struct I2cTransfer |
| { |
| size_t tx; |
| size_t rx; |
| int err; |
| uint8_t txrxBuf[MAX_I2C_TRANSFER_SIZE]; |
| uint8_t state; |
| bool inUse; |
| }; |
| |
| struct TaskStatistics { |
| uint64_t enabledTimestamp; |
| uint64_t proxEnabledTimestamp; |
| uint64_t lastProxFarTimestamp; |
| uint64_t totalEnabledTime; |
| uint64_t totalProxEnabledTime; |
| uint64_t totalProxFarTime; |
| }; |
| |
| enum ProxState { |
| PROX_STATE_UNKNOWN, |
| PROX_STATE_NEAR, |
| PROX_STATE_FAR |
| }; |
| |
| static struct TaskStruct |
| { |
| struct Gpio *pin; |
| struct ChainedIsr isr; |
| struct TaskStatistics stats; |
| struct I2cTransfer transfers[MAX_PENDING_I2C_REQUESTS]; |
| uint32_t id; |
| uint32_t handle; |
| uint32_t retryTimerHandle; |
| uint32_t retryCnt; |
| uint32_t proxHandle; |
| enum ProxState proxState; |
| bool on; |
| bool gestureEnabled; |
| bool isrEnabled; |
| } mTask; |
| |
| static inline void enableInterrupt(bool enable) |
| { |
| if (!mTask.isrEnabled && enable) { |
| extiEnableIntGpio(mTask.pin, EXTI_TRIGGER_FALLING); |
| extiChainIsr(TOUCH_IRQ, &mTask.isr); |
| } else if (mTask.isrEnabled && !enable) { |
| extiUnchainIsr(TOUCH_IRQ, &mTask.isr); |
| extiDisableIntGpio(mTask.pin); |
| } |
| mTask.isrEnabled = enable; |
| } |
| |
| static bool touchIsr(struct ChainedIsr *localIsr) |
| { |
| struct TaskStruct *data = container_of(localIsr, struct TaskStruct, isr); |
| |
| if (!extiIsPendingGpio(data->pin)) { |
| return false; |
| } |
| |
| osEnqueuePrivateEvt(EVT_SENSOR_TOUCH_INTERRUPT, NULL, NULL, data->id); |
| |
| extiClearPendingGpio(data->pin); |
| |
| return true; |
| } |
| |
| static void i2cCallback(void *cookie, size_t tx, size_t rx, int err) |
| { |
| struct I2cTransfer *xfer = cookie; |
| |
| xfer->tx = tx; |
| xfer->rx = rx; |
| xfer->err = err; |
| |
| osEnqueuePrivateEvt(EVT_SENSOR_I2C, cookie, NULL, mTask.id); |
| // Do not print error for ENXIO since we expect there to be times where we |
| // cannot talk to the touch controller. |
| if (err == -ENXIO) { |
| DEBUG_PRINT("i2c error (tx: %d, rx: %d, err: %d)\n", tx, rx, err); |
| } else if (err != 0) { |
| ERROR_PRINT("i2c error (tx: %d, rx: %d, err: %d)\n", tx, rx, err); |
| } |
| } |
| |
| static void retryTimerCallback(uint32_t timerId, void *cookie) |
| { |
| osEnqueuePrivateEvt(EVT_SENSOR_RETRY_TIMER, cookie, NULL, mTask.id); |
| } |
| |
| // Allocate a buffer and mark it as in use with the given state, or return NULL |
| // if no buffers available. Must *not* be called from interrupt context. |
| static struct I2cTransfer *allocXfer(uint8_t state) |
| { |
| size_t i; |
| |
| for (i = 0; i < ARRAY_SIZE(mTask.transfers); i++) { |
| if (!mTask.transfers[i].inUse) { |
| mTask.transfers[i].inUse = true; |
| mTask.transfers[i].state = state; |
| memset(mTask.transfers[i].txrxBuf, 0x00, sizeof(mTask.transfers[i].txrxBuf)); |
| return &mTask.transfers[i]; |
| } |
| } |
| |
| ERROR_PRINT("Ran out of I2C buffers!"); |
| return NULL; |
| } |
| |
| // Helper function to initiate the I2C transfer. Returns true is the transaction |
| // was successfully register by I2C driver. Otherwise, returns false. |
| static bool performXfer(struct I2cTransfer *xfer, size_t txBytes, size_t rxBytes) |
| { |
| int ret; |
| |
| if ((txBytes > MAX_I2C_TRANSFER_SIZE) || (rxBytes > MAX_I2C_TRANSFER_SIZE)) { |
| ERROR_PRINT("txBytes and rxBytes must be less than %d", MAX_I2C_TRANSFER_SIZE); |
| return false; |
| } |
| |
| if (rxBytes) { |
| ret = i2cMasterTxRx(I2C_BUS_ID, I2C_ADDR, xfer->txrxBuf, txBytes, xfer->txrxBuf, rxBytes, i2cCallback, xfer); |
| } else { |
| ret = i2cMasterTx(I2C_BUS_ID, I2C_ADDR, xfer->txrxBuf, txBytes, i2cCallback, xfer); |
| } |
| |
| if (ret != 0) { |
| ERROR_PRINT("I2C transfer was not successful (error %d)!", ret); |
| } |
| |
| return (ret == 0); |
| } |
| |
| // Helper function to write a one byte register. Returns true if we got a |
| // successful return value from i2cMasterTx(). |
| static bool writeRegister(uint8_t reg, uint8_t value, uint8_t state) |
| { |
| struct I2cTransfer *xfer = allocXfer(state); |
| |
| if (xfer != NULL) { |
| xfer->txrxBuf[0] = reg; |
| xfer->txrxBuf[1] = value; |
| return performXfer(xfer, 2, 0); |
| } |
| |
| return false; |
| } |
| |
| static bool setSleepEnable(bool enable, uint8_t state) |
| { |
| return writeRegister(S3708_REG_F01_CTRL_BASE, enable ? S3708_SLEEP_MODE : S3708_NORMAL_MODE, state); |
| } |
| |
| static bool setReportingMode(uint8_t mode, uint8_t state) |
| { |
| struct I2cTransfer *xfer; |
| |
| xfer = allocXfer(state); |
| if (xfer != NULL) { |
| xfer->txrxBuf[0] = S3708_REG_CTRL_BASE + S3708_REG_CTRL_20_OFFSET; |
| xfer->txrxBuf[1] = 0x00; |
| xfer->txrxBuf[2] = 0x00; |
| xfer->txrxBuf[3] = mode; |
| return performXfer(xfer, 4, 0); |
| } |
| |
| return false; |
| } |
| |
| static void setRetryTimer() |
| { |
| mTask.retryCnt++; |
| if (mTask.retryCnt < MAX_I2C_RETRY_COUNT) { |
| mTask.retryTimerHandle = timTimerSet(MAX_I2C_RETRY_DELAY, 0, 50, retryTimerCallback, NULL, true); |
| if (!mTask.retryTimerHandle) { |
| ERROR_PRINT("failed to allocate timer"); |
| } |
| } else { |
| ERROR_PRINT("could not communicate with touch controller"); |
| } |
| } |
| |
| static void setGesturePower(bool enable, bool skipI2c) |
| { |
| bool ret; |
| size_t i; |
| |
| INFO_PRINT("gesture: %d", enable); |
| |
| // Cancel any pending I2C transactions by changing the callback state |
| for (i = 0; i < ARRAY_SIZE(mTask.transfers); i++) { |
| if (mTask.transfers[i].inUse) { |
| mTask.transfers[i].state = STATE_CANCELLED; |
| } |
| } |
| |
| if (enable) { |
| mTask.retryCnt = 0; |
| |
| // Set page number to 0x00 |
| ret = writeRegister(S3708_REG_PAGE_SELECT, 0x00, STATE_ENABLE_0); |
| } else { |
| // Cancel any pending retries |
| if (mTask.retryTimerHandle) { |
| timTimerCancel(mTask.retryTimerHandle); |
| mTask.retryTimerHandle = 0; |
| } |
| |
| if (skipI2c) { |
| ret = true; |
| } else { |
| // Reset to continuous reporting mode |
| ret = setReportingMode(S3708_REPORT_MODE_CONT, STATE_DISABLE_0); |
| } |
| } |
| |
| if (ret) { |
| mTask.gestureEnabled = enable; |
| enableInterrupt(enable); |
| } |
| } |
| |
| static void configProx(bool on) { |
| if (on) { |
| mTask.stats.proxEnabledTimestamp = sensorGetTime(); |
| sensorRequest(mTask.id, mTask.proxHandle, DEFAULT_PROX_RATE_HZ, |
| DEFAULT_PROX_LATENCY); |
| osEventSubscribe(mTask.id, EVT_SENSOR_PROX); |
| } else { |
| sensorRelease(mTask.id, mTask.proxHandle); |
| osEventUnsubscribe(mTask.id, EVT_SENSOR_PROX); |
| |
| mTask.stats.totalProxEnabledTime += sensorGetTime() - mTask.stats.proxEnabledTimestamp; |
| if (mTask.proxState == PROX_STATE_FAR) { |
| mTask.stats.totalProxFarTime += sensorGetTime() - mTask.stats.lastProxFarTimestamp; |
| } |
| } |
| mTask.proxState = PROX_STATE_UNKNOWN; |
| } |
| |
| static bool callbackPower(bool on, void *cookie) |
| { |
| uint32_t enabledSeconds, proxEnabledSeconds, proxFarSeconds; |
| |
| INFO_PRINT("power: %d", on); |
| |
| if (on) { |
| mTask.stats.enabledTimestamp = sensorGetTime(); |
| } else { |
| mTask.stats.totalEnabledTime += sensorGetTime() - mTask.stats.enabledTimestamp; |
| } |
| |
| enabledSeconds = U64_DIV_BY_U64_CONSTANT(mTask.stats.totalEnabledTime, 1000000000); |
| proxEnabledSeconds = U64_DIV_BY_U64_CONSTANT(mTask.stats.totalProxEnabledTime, 1000000000); |
| proxFarSeconds = U64_DIV_BY_U64_CONSTANT(mTask.stats.totalProxFarTime, 1000000000); |
| INFO_PRINT("STATS: enabled %02" PRIu32 ":%02" PRIu32 ":%02" PRIu32 |
| ", prox enabled %02" PRIu32 ":%02" PRIu32 ":%02" PRIu32 |
| ", prox far %02" PRIu32 ":%02" PRIu32 ":%02" PRIu32, |
| enabledSeconds / 3600, (enabledSeconds % 3600) / 60, enabledSeconds % 60, |
| proxEnabledSeconds / 3600, (proxEnabledSeconds % 3600) / 60, proxEnabledSeconds % 60, |
| proxFarSeconds / 3600, (proxFarSeconds % 3600) / 60, proxFarSeconds % 60); |
| |
| // If the task is disabled, that means the AP is on and has switched the I2C |
| // mux. Therefore, no I2C transactions will succeed so skip them. |
| if (mTask.gestureEnabled) { |
| setGesturePower(false, true /* skipI2c */); |
| } |
| |
| mTask.on = on; |
| configProx(on); |
| |
| return sensorSignalInternalEvt(mTask.handle, SENSOR_INTERNAL_EVT_POWER_STATE_CHG, mTask.on, 0); |
| } |
| |
| static bool callbackFirmwareUpload(void *cookie) |
| { |
| return sensorSignalInternalEvt(mTask.handle, SENSOR_INTERNAL_EVT_FW_STATE_CHG, 1, 0); |
| } |
| |
| static bool callbackSetRate(uint32_t rate, uint64_t latency, void *cookie) |
| { |
| return sensorSignalInternalEvt(mTask.handle, SENSOR_INTERNAL_EVT_RATE_CHG, rate, latency); |
| } |
| |
| static bool callbackFlush(void *cookie) |
| { |
| return osEnqueueEvt(sensorGetMyEventType(SENS_TYPE_DOUBLE_TOUCH), SENSOR_DATA_EVENT_FLUSH, NULL); |
| } |
| |
| static const struct SensorInfo mSensorInfo = { |
| .sensorName = "Double Touch", |
| .sensorType = SENS_TYPE_DOUBLE_TOUCH, |
| .numAxis = NUM_AXIS_EMBEDDED, |
| .interrupt = NANOHUB_INT_WAKEUP, |
| .minSamples = 20 |
| }; |
| |
| static const struct SensorOps mSensorOps = |
| { |
| .sensorPower = callbackPower, |
| .sensorFirmwareUpload = callbackFirmwareUpload, |
| .sensorSetRate = callbackSetRate, |
| .sensorFlush = callbackFlush, |
| }; |
| |
| static void processI2cResponse(struct I2cTransfer *xfer) |
| { |
| struct I2cTransfer *nextXfer; |
| union EmbeddedDataPoint sample; |
| |
| switch (xfer->state) { |
| case STATE_ENABLE_0: |
| setSleepEnable(false, STATE_ENABLE_1); |
| break; |
| |
| case STATE_ENABLE_1: |
| // HACK: DozeService reactivates pickup gesture before the screen |
| // comes on, so we need to wait for some time after enabling before |
| // trying to talk to touch controller. We may see the touch |
| // controller on the first few samples and then have communication |
| // switched off. So, wait HACK_RETRY_SKIP_COUNT samples before we |
| // consider the transaction. |
| if (mTask.retryCnt < HACK_RETRY_SKIP_COUNT) { |
| setRetryTimer(); |
| } else { |
| setReportingMode(S3708_REPORT_MODE_LPWG, STATE_IDLE); |
| } |
| break; |
| |
| case STATE_DISABLE_0: |
| setSleepEnable(true, STATE_IDLE); |
| break; |
| |
| case STATE_INT_HANDLE_0: |
| // If the interrupt was from the LPWG function, read the function interrupt status register |
| if (xfer->txrxBuf[1] & S3708_INT_STATUS_LPWG) { |
| nextXfer = allocXfer(STATE_INT_HANDLE_1); |
| if (nextXfer != NULL) { |
| nextXfer->txrxBuf[0] = S3708_REG_DATA_BASE + S3708_REG_DATA_4_OFFSET; |
| performXfer(nextXfer, 1, 5); |
| } |
| } |
| break; |
| |
| case STATE_INT_HANDLE_1: |
| // Verify the LPWG interrupt status |
| if (xfer->txrxBuf[0] & S3708_INT_STATUS_DOUBLE_TAP) { |
| DEBUG_PRINT("Sending event"); |
| sample.idata = 1; |
| osEnqueueEvt(sensorGetMyEventType(SENS_TYPE_DOUBLE_TOUCH), sample.vptr, NULL); |
| } |
| break; |
| |
| default: |
| break; |
| } |
| } |
| |
| static void handleI2cEvent(struct I2cTransfer *xfer) |
| { |
| if (xfer->err == 0) { |
| processI2cResponse(xfer); |
| } else if (xfer->state == STATE_ENABLE_0 || xfer->state == STATE_ENABLE_1) { |
| setRetryTimer(); |
| } |
| |
| xfer->inUse = false; |
| } |
| |
| static void handleEvent(uint32_t evtType, const void* evtData) |
| { |
| struct I2cTransfer *xfer; |
| union EmbeddedDataPoint embeddedSample; |
| enum ProxState lastProxState; |
| int ret; |
| |
| switch (evtType) { |
| case EVT_APP_START: |
| osEventUnsubscribe(mTask.id, EVT_APP_START); |
| ret = i2cMasterRequest(I2C_BUS_ID, I2C_SPEED); |
| // Since the i2c bus can be shared with other drivers, it is |
| // possible that one of the other drivers requested the bus first. |
| // Therefore, either 0 or -EBUSY is an acceptable return. |
| if ((ret < 0) && (ret != -EBUSY)) { |
| ERROR_PRINT("i2cMasterRequest() failed!"); |
| } |
| |
| sensorFind(SENS_TYPE_PROX, 0, &mTask.proxHandle); |
| |
| sensorRegisterInitComplete(mTask.handle); |
| break; |
| |
| case EVT_SENSOR_I2C: |
| handleI2cEvent((struct I2cTransfer *)evtData); |
| break; |
| |
| case EVT_SENSOR_TOUCH_INTERRUPT: |
| if (mTask.on) { |
| // Read the interrupt status register |
| xfer = allocXfer(STATE_INT_HANDLE_0); |
| if (xfer != NULL) { |
| xfer->txrxBuf[0] = S3708_REG_F01_DATA_BASE; |
| performXfer(xfer, 1, 2); |
| } |
| } |
| break; |
| |
| case EVT_SENSOR_PROX: |
| if (mTask.on) { |
| // cast off the const, and cast to union |
| embeddedSample = (union EmbeddedDataPoint)((void*)evtData); |
| lastProxState = mTask.proxState; |
| mTask.proxState = (embeddedSample.fdata < PROXIMITY_THRESH_NEAR) ? PROX_STATE_NEAR : PROX_STATE_FAR; |
| |
| if ((lastProxState != PROX_STATE_FAR) && (mTask.proxState == PROX_STATE_FAR)) { |
| mTask.stats.lastProxFarTimestamp = sensorGetTime(); |
| setGesturePower(true, false); |
| } else if ((lastProxState == PROX_STATE_FAR) && (mTask.proxState == PROX_STATE_NEAR)) { |
| mTask.stats.totalProxFarTime += sensorGetTime() - mTask.stats.lastProxFarTimestamp; |
| setGesturePower(false, false); |
| } |
| } |
| break; |
| |
| case EVT_SENSOR_RETRY_TIMER: |
| if (mTask.on) { |
| // Set page number to 0x00 |
| writeRegister(S3708_REG_PAGE_SELECT, 0x00, STATE_ENABLE_0); |
| } |
| break; |
| } |
| } |
| |
| static bool startTask(uint32_t taskId) |
| { |
| mTask.id = taskId; |
| mTask.handle = sensorRegister(&mSensorInfo, &mSensorOps, NULL, false); |
| |
| mTask.pin = gpioRequest(TOUCH_PIN); |
| gpioConfigInput(mTask.pin, GPIO_SPEED_LOW, GPIO_PULL_NONE); |
| syscfgSetExtiPort(mTask.pin); |
| mTask.isr.func = touchIsr; |
| |
| osEventSubscribe(taskId, EVT_APP_START); |
| return true; |
| } |
| |
| static void endTask(void) |
| { |
| enableInterrupt(false); |
| extiUnchainIsr(TOUCH_IRQ, &mTask.isr); |
| extiClearPendingGpio(mTask.pin); |
| gpioRelease(mTask.pin); |
| |
| i2cMasterRelease(I2C_BUS_ID); |
| |
| sensorUnregister(mTask.handle); |
| } |
| |
| INTERNAL_APP_INIT(S3708_APP_ID, S3708_APP_VERSION, startTask, endTask, handleEvent); |
