/* Copyright (C) 2009 The Android Open Source Project
**
** This software is licensed under the terms of the GNU General Public
** License version 2, as published by the Free Software Foundation, and
** may be copied, distributed, and modified under those terms.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
*/

#include "android/hw-sensors.h"
#include "android/utils/debug.h"
#include "android/utils/misc.h"
#include "android/hw-qemud.h"
#include "android/globals.h"
#include "qemu-char.h"
#include "qemu-timer.h"

#define  D(...)  VERBOSE_PRINT(sensors,__VA_ARGS__)

/* define T_ACTIVE to 1 to debug transport communications */
#define  T_ACTIVE  0

#if T_ACTIVE
#define  T(...)  VERBOSE_PRINT(sensors,__VA_ARGS__)
#else
#define  T(...)   ((void)0)
#endif

/* this code supports emulated sensor hardware
 *
 * Note that currently, only the accelerometer is really emulated, and only
 * for the purpose of allowing auto-rotating the screen in keyboard-less
 * configurations.
 *
 *
 */


static const struct {
    const char*  name;
    int          id;
} _sSensors[MAX_SENSORS] = {
#define SENSOR_(x,y)  { y, ANDROID_SENSOR_##x },
  SENSORS_LIST
#undef SENSOR_
};


static int
_sensorIdFromName( const char*  name )
{
    int  nn;
    for (nn = 0; nn < MAX_SENSORS; nn++)
        if (!strcmp(_sSensors[nn].name,name))
            return _sSensors[nn].id;
    return -1;
}


typedef struct {
    float   x, y, z;
} Acceleration;


typedef struct {
    float  x, y, z;
} MagneticField;


typedef struct {
    float  azimuth;
    float  pitch;
    float  roll;
} Orientation;


typedef struct {
    float  celsius;
} Temperature;


typedef struct {
    char       enabled;
    union {
        Acceleration   acceleration;
        MagneticField  magnetic;
        Orientation    orientation;
        Temperature    temperature;
    } u;
} Sensor;

/*
 * - when the qemu-specific sensors HAL module starts, it sends
 *   "list-sensors"
 *
 * - this code replies with a string containing an integer corresponding
 *   to a bitmap of available hardware sensors in the current AVD
 *   configuration (e.g. "1" a.k.a (1 << ANDROID_SENSOR_ACCELERATION))
 *
 * - the HAL module sends "set:<sensor>:<flag>" to enable or disable
 *   the report of a given sensor state. <sensor> must be the name of
 *   a given sensor (e.g. "accelerometer"), and <flag> must be either
 *   "1" (to enable) or "0" (to disable).
 *
 * - Once at least one sensor is "enabled", this code should periodically
 *   send information about the corresponding enabled sensors. The default
 *   period is 200ms.
 *
 * - the HAL module sends "set-delay:<delay>", where <delay> is an integer
 *   corresponding to a time delay in milli-seconds. This corresponds to
 *   a new interval between sensor events sent by this code to the HAL
 *   module.
 *
 * - the HAL module can also send a "wake" command. This code should simply
 *   send the "wake" back to the module. This is used internally to wake a
 *   blocking read that happens in a different thread. This ping-pong makes
 *   the code in the HAL module very simple.
 *
 * - each timer tick, this code sends sensor reports in the following
 *   format (each line corresponds to a different line sent to the module):
 *
 *      acceleration:<x>:<y>:<z>
 *      magnetic-field:<x>:<y>:<z>
 *      orientation:<azimuth>:<pitch>:<roll>
 *      temperature:<celsius>
 *      sync:<time_us>
 *
 *   Where each line before the sync:<time_us> is optional and will only
 *   appear if the corresponding sensor has been enabled by the HAL module.
 *
 *   Note that <time_us> is the VM time in micro-seconds when the report
 *   was "taken" by this code. This is adjusted by the HAL module to
 *   emulated system time (using the first sync: to compute an adjustment
 *   offset).
 */
#define  HEADER_SIZE  4
#define  BUFFER_SIZE  512

typedef struct {
    QemudService*   service;
    int32_t         delay_ms;
    uint32_t        enabledMask;
    QEMUTimer*      timer;
    Sensor          sensors[MAX_SENSORS];
} HwSensors;

/* forward */

static void  hw_sensors_receive( HwSensors*  h,
                                 uint8_t*    query,
                                 int         querylen );

static void  hw_sensors_timer_tick(void*  opaque);

/* Qemud service management */

static void
_hw_sensors_qemud_client_recv( void*  opaque, uint8_t*  msg, int  msglen,
                               QemudClient*  client )
{
    hw_sensors_receive(opaque, msg, msglen);
}

static QemudClient*
_hw_sensors_service_connect( void*  opaque, QemudService*  service, int  channel )
{
    HwSensors*    sensors = opaque;
    QemudClient*  client  = qemud_client_new(service, channel,
                                             sensors,
                                             _hw_sensors_qemud_client_recv,
                                             NULL);
    qemud_client_set_framing(client, 1);
    return client;
}

/* change the value of the emulated acceleration vector */
static void
hw_sensors_set_acceleration( HwSensors*  h, float x, float y, float z )
{
    Sensor*  s = &h->sensors[ANDROID_SENSOR_ACCELERATION];
    s->u.acceleration.x = x;
    s->u.acceleration.y = y;
    s->u.acceleration.z = z;
}

#if 0  /* not used yet */
/* change the value of the emulated magnetic vector */
static void
hw_sensors_set_magnetic_field( HwSensors*  h, float x, float y, float z )
{
    Sensor*  s = &h->sensors[ANDROID_SENSOR_MAGNETIC_FIELD];
    s->u.magnetic.x = x;
    s->u.magnetic.y = y;
    s->u.magnetic.z = z;
}

/* change the values of the emulated orientation */
static void
hw_sensors_set_orientation( HwSensors*  h, float azimuth, float pitch, float roll )
{
    Sensor*  s = &h->sensors[ANDROID_SENSOR_MAGNETIC_FIELD];
    s->u.orientation.azimuth = azimuth;
    s->u.orientation.pitch   = pitch;
    s->u.orientation.roll    = roll;
}

/* change the emulated temperature */
static void
hw_sensors_set_temperature( HwSensors*  h, float celsius )
{
    Sensor*  s = &h->sensors[ANDROID_SENSOR_MAGNETIC_FIELD];
    s->u.temperature.celsius = celsius;
}
#endif

/* change the coarse orientation (landscape/portrait) of the emulated device */
static void
hw_sensors_set_coarse_orientation( HwSensors*  h, AndroidCoarseOrientation  orient )
{
    /* The Android framework computes the orientation by looking at
     * the accelerometer sensor (*not* the orientation sensor !)
     *
     * That's because the gravity is a constant 9.81 vector that
     * can be determined quite easily.
     *
     * Also, for some reason, the framework code considers that the phone should
     * be inclined by 30 degrees along the phone's X axis to be considered
     * in its ideal "vertical" position
     *
     * If the phone is completely vertical, rotating it will not do anything !
     */
    const double  g      = 9.81;
    const double  cos_30 = 0.866025403784;
    const double  sin_30 = 0.5;

    switch (orient) {
    case ANDROID_COARSE_PORTRAIT:
        hw_sensors_set_acceleration( h, 0., g*cos_30, g*sin_30 );
        break;

    case ANDROID_COARSE_LANDSCAPE:
        hw_sensors_set_acceleration( h, g*cos_30, 0., g*sin_30 );
        break;
    default:
        ;
    }
}


/* initialize the sensors state */
static void
hw_sensors_init( HwSensors*  h )
{
    h->service = qemud_service_register("sensors", 1, h,
                                        _hw_sensors_service_connect );
    h->enabledMask = 0;
    h->delay_ms    = 1000;
    h->timer       = qemu_new_timer(vm_clock, hw_sensors_timer_tick, h);

    hw_sensors_set_coarse_orientation(h, ANDROID_COARSE_PORTRAIT);
}

/* send a one-line message to the HAL module through a qemud channel */
static void
hw_sensors_send( HwSensors*  hw, const uint8_t*  msg, int  msglen )
{
    D("%s: '%s'", __FUNCTION__, quote_bytes((const void*)msg, msglen));
    qemud_service_broadcast(hw->service, msg, msglen);
}

/* this function is called periodically to send sensor reports
 * to the HAL module, and re-arm the timer if necessary
 */
static void
hw_sensors_timer_tick( void*  opaque )
{
    HwSensors*  h = opaque;
    int64_t     delay = h->delay_ms;
    int64_t     now_ns;
    uint32_t    mask  = h->enabledMask;
    Sensor*     sensor;
    char        buffer[128];

    sensor = &h->sensors[ANDROID_SENSOR_ACCELERATION];
    if (sensor->enabled) {
        snprintf(buffer, sizeof buffer, "acceleration:%g:%g:%g",
                 sensor->u.acceleration.x,
                 sensor->u.acceleration.y,
                 sensor->u.acceleration.z);
        hw_sensors_send(h, (uint8_t*)buffer, strlen(buffer));
    }

    sensor = &h->sensors[ANDROID_SENSOR_MAGNETIC_FIELD];
    if (sensor->enabled) {
        snprintf(buffer, sizeof buffer, "magnetic-field:%g:%g:%g",
                 sensor->u.magnetic.x,
                 sensor->u.magnetic.y,
                 sensor->u.magnetic.z);
        hw_sensors_send(h, (uint8_t*)buffer, strlen(buffer));
    }

    sensor = &h->sensors[ANDROID_SENSOR_ORIENTATION];
    if (sensor->enabled) {
        snprintf(buffer, sizeof buffer, "orientation:%g:%g:%g",
                 sensor->u.orientation.azimuth,
                 sensor->u.orientation.pitch,
                 sensor->u.orientation.roll);
        hw_sensors_send(h, (uint8_t*)buffer, strlen(buffer));
    }

    sensor = &h->sensors[ANDROID_SENSOR_TEMPERATURE];
    if (sensor->enabled) {
        snprintf(buffer, sizeof buffer, "temperature:%g",
                 sensor->u.temperature.celsius);
        hw_sensors_send(h, (uint8_t*)buffer, strlen(buffer));
    }

    now_ns = qemu_get_clock(vm_clock);

    snprintf(buffer, sizeof buffer, "sync:%lld", now_ns/1000);
    hw_sensors_send(h, (uint8_t*)buffer, strlen(buffer));

    /* rearm timer, use a minimum delay of 20 ms, just to
     * be safe.
     */
    if (mask == 0)
        return;

    if (delay < 20)
        delay = 20;

    delay *= 1000000LL;  /* convert to nanoseconds */
    qemu_mod_timer(h->timer, now_ns + delay);
}

/* handle incoming messages from the HAL module */
static void
hw_sensors_receive( HwSensors*  hw, uint8_t*  msg, int  msglen )
{
    D("%s: '%.*s'", __FUNCTION__, msglen, msg);

    /* "list-sensors" is used to get an integer bit map of
     * available emulated sensors. We compute the mask from the
     * current hardware configuration.
     */
    if (msglen == 12 && !memcmp(msg, "list-sensors", 12)) {
        char  buff[12];
        int   mask = 0;

        if (android_hw->hw_accelerometer)
            mask |= (1 << ANDROID_SENSOR_ACCELERATION);

        /* XXX: TODO: Add other tests when we add the corresponding
         * properties to hardware-properties.ini et al. */

        snprintf(buff, sizeof buff, "%d", mask);
        hw_sensors_send(hw, (const uint8_t*)buff, strlen(buff));
        return;
    }

    /* "wake" is a special message that must be sent back through
     * the channel. It is used to exit a blocking read.
     */
    if (msglen == 4 && !memcmp(msg, "wake", 4)) {
        hw_sensors_send(hw, (const uint8_t*)"wake", 4);
        return;
    }

    /* "set-delay:<delay>" is used to set the delay in milliseconds
     * between sensor events
     */
    if (msglen > 10 && !memcmp(msg, "set-delay:", 10)) {
        hw->delay_ms = atoi((const char*)msg+10);
        if (hw->enabledMask != 0)
            hw_sensors_timer_tick(hw);

        return;
    }

    /* "set:<name>:<state>" is used to enable/disable a given
     * sensor. <state> must be 0 or 1
     */
    if (msglen > 4 && !memcmp(msg, "set:", 4)) {
        char*  q;
        int    id, enabled, oldEnabledMask = hw->enabledMask;
        msg += 4;
        q    = strchr((char*)msg, ':');
        if (q == NULL) {  /* should not happen */
            D("%s: ignore bad 'set' command", __FUNCTION__);
            return;
        }
        *q++ = 0;

        id = _sensorIdFromName((const char*)msg);
        if (id < 0) {
            D("%s: ignore unknown sensor name '%s'", __FUNCTION__, msg);
            return;
        }

        enabled = (q[0] == '1');

        hw->sensors[id].enabled = (char) enabled;
        if (enabled)
            hw->enabledMask |= (1 << id);
        else
            hw->enabledMask &= ~(1 << id);

        D("%s: %s %s sensor", __FUNCTION__,
          hw->sensors[id].enabled ? "enabling" : "disabling",  msg);

        if (oldEnabledMask == 0 && enabled) {
            /* we enabled our first sensor, start event reporting */
            D("%s: starting event reporting (mask=%04x)", __FUNCTION__,
              hw->enabledMask);
        }
        else if (hw->enabledMask == 0 && !enabled) {
            /* we disabled our last sensor, stop event reporting */
            D("%s: stopping event reporting", __FUNCTION__);
        }
        hw_sensors_timer_tick(hw);
        return;
    }

    D("%s: ignoring unknown query", __FUNCTION__);
}


static HwSensors    _sensorsState[1];

void
android_hw_sensors_init( void )
{
    HwSensors*  hw = _sensorsState;

    if (hw->service == NULL) {
        hw_sensors_init(hw);
        D("%s: sensors qemud service initialized", __FUNCTION__);
    }
}

/* change the coarse orientation value */
extern void
android_sensors_set_coarse_orientation( AndroidCoarseOrientation  orient )
{
    android_hw_sensors_init();
    hw_sensors_set_coarse_orientation(_sensorsState, orient);
}