Quelle  mma9553.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Freescale MMA9553L Intelligent Pedometer driver
 * Copyright (c) 2014, Intel Corporation.
 */

#include <linux/i2c.h>
#include <linux/interrupt.h>
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/iio/iio.h>
#include <linux/iio/sysfs.h>
#include <linux/iio/events.h>
#include <linux/pm_runtime.h>
#include "mma9551_core.h"

/* Pedometer configuration registers (R/W) */
#define MMA9553_REG_CONF_SLEEPMIN  0x00
#define MMA9553_REG_CONF_SLEEPMAX  0x02
#define MMA9553_REG_CONF_SLEEPTHD  0x04
#define MMA9553_MASK_CONF_WORD   GENMASK(15, 0)

#define MMA9553_REG_CONF_CONF_STEPLEN  0x06
#define MMA9553_MASK_CONF_CONFIG  BIT(15)
#define MMA9553_MASK_CONF_ACT_DBCNTM  BIT(14)
#define MMA9553_MASK_CONF_SLP_DBCNTM  BIT(13)
#define MMA9553_MASK_CONF_STEPLEN  GENMASK(7, 0)

#define MMA9553_REG_CONF_HEIGHT_WEIGHT  0x08
#define MMA9553_MASK_CONF_HEIGHT  GENMASK(15, 8)
#define MMA9553_MASK_CONF_WEIGHT  GENMASK(7, 0)

#define MMA9553_REG_CONF_FILTER   0x0A
#define MMA9553_MASK_CONF_FILTSTEP  GENMASK(15, 8)
#define MMA9553_MASK_CONF_MALE   BIT(7)
#define MMA9553_MASK_CONF_FILTTIME  GENMASK(6, 0)

#define MMA9553_REG_CONF_SPEED_STEP  0x0C
#define MMA9553_MASK_CONF_SPDPRD  GENMASK(15, 8)
#define MMA9553_MASK_CONF_STEPCOALESCE  GENMASK(7, 0)

#define MMA9553_REG_CONF_ACTTHD   0x0E
#define MMA9553_MAX_ACTTHD   GENMASK(15, 0)

/* Pedometer status registers (R-only) */
#define MMA9553_REG_STATUS   0x00
#define MMA9553_MASK_STATUS_MRGFL  BIT(15)
#define MMA9553_MASK_STATUS_SUSPCHG  BIT(14)
#define MMA9553_MASK_STATUS_STEPCHG  BIT(13)
#define MMA9553_MASK_STATUS_ACTCHG  BIT(12)
#define MMA9553_MASK_STATUS_SUSP  BIT(11)
#define MMA9553_MASK_STATUS_ACTIVITY  GENMASK(10, 8)
#define MMA9553_MASK_STATUS_VERSION  GENMASK(7, 0)

#define MMA9553_REG_STEPCNT   0x02
#define MMA9553_REG_DISTANCE   0x04
#define MMA9553_REG_SPEED   0x06
#define MMA9553_REG_CALORIES   0x08
#define MMA9553_REG_SLEEPCNT   0x0A

/* Pedometer events are always mapped to this pin. */
#define MMA9553_DEFAULT_GPIO_PIN mma9551_gpio6
#define MMA9553_DEFAULT_GPIO_POLARITY 0

/* Bitnum used for GPIO configuration = bit number in high status byte */
#define MMA9553_STATUS_TO_BITNUM(bit) (ffs(bit) - 9)
#define MMA9553_MAX_BITNUM  MMA9553_STATUS_TO_BITNUM(BIT(16))

#define MMA9553_DEFAULT_SAMPLE_RATE 30 /* Hz */

/*
 * The internal activity level must be stable for ACTTHD samples before
 * ACTIVITY is updated. The ACTIVITY variable contains the current activity
 * level and is updated every time a step is detected or once a second
 * if there are no steps.
 */
#define MMA9553_ACTIVITY_THD_TO_SEC(thd) ((thd) / MMA9553_DEFAULT_SAMPLE_RATE)
#define MMA9553_ACTIVITY_SEC_TO_THD(sec) ((sec) * MMA9553_DEFAULT_SAMPLE_RATE)

/*
 * Autonomously suspend pedometer if acceleration vector magnitude
 * is near 1g (4096 at 0.244 mg/LSB resolution) for 30 seconds.
 */
#define MMA9553_DEFAULT_SLEEPMIN 3688 /* 0,9 g */
#define MMA9553_DEFAULT_SLEEPMAX 4508 /* 1,1 g */
#define MMA9553_DEFAULT_SLEEPTHD (MMA9553_DEFAULT_SAMPLE_RATE * 30)

#define MMA9553_CONFIG_RETRIES  2

/* Status register - activity field  */
enum activity_level {
 ACTIVITY_UNKNOWN,
 ACTIVITY_REST,
 ACTIVITY_WALKING,
 ACTIVITY_JOGGING,
 ACTIVITY_RUNNING,
};

static const struct mma9553_event_info {
 enum iio_chan_type type;
 enum iio_modifier mod;
 enum iio_event_direction dir;
} mma9553_events_info[] = {
 {
  .type = IIO_STEPS,
  .mod = IIO_NO_MOD,
  .dir = IIO_EV_DIR_NONE,
 },
 {
  .type = IIO_ACTIVITY,
  .mod = IIO_MOD_STILL,
  .dir = IIO_EV_DIR_RISING,
 },
 {
  .type = IIO_ACTIVITY,
  .mod = IIO_MOD_STILL,
  .dir = IIO_EV_DIR_FALLING,
 },
 {
  .type = IIO_ACTIVITY,
  .mod = IIO_MOD_WALKING,
  .dir = IIO_EV_DIR_RISING,
 },
 {
  .type = IIO_ACTIVITY,
  .mod = IIO_MOD_WALKING,
  .dir = IIO_EV_DIR_FALLING,
 },
 {
  .type = IIO_ACTIVITY,
  .mod = IIO_MOD_JOGGING,
  .dir = IIO_EV_DIR_RISING,
 },
 {
  .type = IIO_ACTIVITY,
  .mod = IIO_MOD_JOGGING,
  .dir = IIO_EV_DIR_FALLING,
 },
 {
  .type = IIO_ACTIVITY,
  .mod = IIO_MOD_RUNNING,
  .dir = IIO_EV_DIR_RISING,
 },
 {
  .type = IIO_ACTIVITY,
  .mod = IIO_MOD_RUNNING,
  .dir = IIO_EV_DIR_FALLING,
 },
};

#define MMA9553_EVENTS_INFO_SIZE ARRAY_SIZE(mma9553_events_info)

struct mma9553_event {
 const struct mma9553_event_info *info;
 bool enabled;
};

struct mma9553_conf_regs {
 u16 sleepmin;
 u16 sleepmax;
 u16 sleepthd;
 u16 config;
 u16 height_weight;
 u16 filter;
 u16 speed_step;
 u16 actthd;
} __packed;

struct mma9553_data {
 struct i2c_client *client;
 /*
 * 1. Serialize access to HW (requested by mma9551_core API).
 * 2. Serialize sequences that power on/off the device and access HW.
 */
 struct mutex mutex;
 struct mma9553_conf_regs conf;
 struct mma9553_event events[MMA9553_EVENTS_INFO_SIZE];
 int num_events;
 u8 gpio_bitnum;
 /*
 * This is used for all features that depend on step count:
 * step count, distance, speed, calories.
 */
 bool stepcnt_enabled;
 u16 stepcnt;
 u8 activity;
 s64 timestamp;
};

static u8 mma9553_get_bits(u16 val, u16 mask)
{
 return (val & mask) >> (ffs(mask) - 1);
}

static u16 mma9553_set_bits(u16 current_val, u16 val, u16 mask)
{
 return (current_val & ~mask) | (val << (ffs(mask) - 1));
}

static enum iio_modifier mma9553_activity_to_mod(enum activity_level activity)
{
 switch (activity) {
 case ACTIVITY_RUNNING:
  return IIO_MOD_RUNNING;
 case ACTIVITY_JOGGING:
  return IIO_MOD_JOGGING;
 case ACTIVITY_WALKING:
  return IIO_MOD_WALKING;
 case ACTIVITY_REST:
  return IIO_MOD_STILL;
 case ACTIVITY_UNKNOWN:
 default:
  return IIO_NO_MOD;
 }
}

static void mma9553_init_events(struct mma9553_data *data)
{
 int i;

 data->num_events = MMA9553_EVENTS_INFO_SIZE;
 for (i = 0; i < data->num_events; i++) {
  data->events[i].info = &mma9553_events_info[i];
  data->events[i].enabled = false;
 }
}

static struct mma9553_event *mma9553_get_event(struct mma9553_data *data,
            enum iio_chan_type type,
            enum iio_modifier mod,
            enum iio_event_direction dir)
{
 int i;

 for (i = 0; i < data->num_events; i++)
  if (data->events[i].info->type == type &&
      data->events[i].info->mod == mod &&
      data->events[i].info->dir == dir)
   return &data->events[i];

 return NULL;
}

static bool mma9553_is_any_event_enabled(struct mma9553_data *data,
      bool check_type,
      enum iio_chan_type type)
{
 int i;

 for (i = 0; i < data->num_events; i++)
  if ((check_type && data->events[i].info->type == type &&
       data->events[i].enabled) ||
       (!check_type && data->events[i].enabled))
   return true;

 return false;
}

static int mma9553_set_config(struct mma9553_data *data, u16 reg,
         u16 *p_reg_val, u16 val, u16 mask)
{
 int ret, retries;
 u16 reg_val, config;

 reg_val = *p_reg_val;
 if (val == mma9553_get_bits(reg_val, mask))
  return 0;

 reg_val = mma9553_set_bits(reg_val, val, mask);
 ret = mma9551_write_config_word(data->client, MMA9551_APPID_PEDOMETER,
     reg, reg_val);
 if (ret < 0) {
  dev_err(&data->client->dev,
   "error writing config register 0x%x\n", reg);
  return ret;
 }

 *p_reg_val = reg_val;

 /* Reinitializes the pedometer with current configuration values */
 config = mma9553_set_bits(data->conf.config, 1,
      MMA9553_MASK_CONF_CONFIG);

 ret = mma9551_write_config_word(data->client, MMA9551_APPID_PEDOMETER,
     MMA9553_REG_CONF_CONF_STEPLEN, config);
 if (ret < 0) {
  dev_err(&data->client->dev,
   "error writing config register 0x%x\n",
   MMA9553_REG_CONF_CONF_STEPLEN);
  return ret;
 }

 retries = MMA9553_CONFIG_RETRIES;
 do {
  mma9551_sleep(MMA9553_DEFAULT_SAMPLE_RATE);
  ret = mma9551_read_config_word(data->client,
            MMA9551_APPID_PEDOMETER,
            MMA9553_REG_CONF_CONF_STEPLEN,
            &config);
  if (ret < 0)
   return ret;
 } while (mma9553_get_bits(config, MMA9553_MASK_CONF_CONFIG) &&
   --retries > 0);

 return 0;
}

static int mma9553_read_activity_stepcnt(struct mma9553_data *data,
      u8 *activity, u16 *stepcnt)
{
 u16 buf[2];
 int ret;

 ret = mma9551_read_status_words(data->client, MMA9551_APPID_PEDOMETER,
     MMA9553_REG_STATUS, ARRAY_SIZE(buf),
     buf);
 if (ret < 0) {
  dev_err(&data->client->dev,
   "error reading status and stepcnt\n");
  return ret;
 }

 *activity = mma9553_get_bits(buf[0], MMA9553_MASK_STATUS_ACTIVITY);
 *stepcnt = buf[1];

 return 0;
}

static int mma9553_conf_gpio(struct mma9553_data *data)
{
 u8 bitnum = 0, appid = MMA9551_APPID_PEDOMETER;
 int ret;
 struct mma9553_event *ev_step_detect;
 bool activity_enabled;

 activity_enabled = mma9553_is_any_event_enabled(data, true,
       IIO_ACTIVITY);
 ev_step_detect = mma9553_get_event(data, IIO_STEPS, IIO_NO_MOD,
        IIO_EV_DIR_NONE);

 /*
 * If both step detector and activity are enabled, use the MRGFL bit.
 * This bit is the logical OR of the SUSPCHG, STEPCHG, and ACTCHG flags.
 */
 if (activity_enabled && ev_step_detect->enabled)
  bitnum = MMA9553_STATUS_TO_BITNUM(MMA9553_MASK_STATUS_MRGFL);
 else if (ev_step_detect->enabled)
  bitnum = MMA9553_STATUS_TO_BITNUM(MMA9553_MASK_STATUS_STEPCHG);
 else if (activity_enabled)
  bitnum = MMA9553_STATUS_TO_BITNUM(MMA9553_MASK_STATUS_ACTCHG);
 else   /* Reset */
  appid = MMA9551_APPID_NONE;

 if (data->gpio_bitnum == bitnum)
  return 0;

 /* Save initial values for activity and stepcnt */
 if (activity_enabled || ev_step_detect->enabled) {
  ret = mma9553_read_activity_stepcnt(data, &data->activity,
          &data->stepcnt);
  if (ret < 0)
   return ret;
 }

 ret = mma9551_gpio_config(data->client, MMA9553_DEFAULT_GPIO_PIN, appid,
      bitnum, MMA9553_DEFAULT_GPIO_POLARITY);
 if (ret < 0)
  return ret;
 data->gpio_bitnum = bitnum;

 return 0;
}

static int mma9553_init(struct mma9553_data *data)
{
 int ret;

 ret = mma9551_read_version(data->client);
 if (ret)
  return ret;

 /*
 * Read all the pedometer configuration registers. This is used as
 * a device identification command to differentiate the MMA9553L
 * from the MMA9550L.
 */
 ret = mma9551_read_config_words(data->client, MMA9551_APPID_PEDOMETER,
     MMA9553_REG_CONF_SLEEPMIN,
     sizeof(data->conf) / sizeof(u16),
     (u16 *)&data->conf);
 if (ret < 0) {
  dev_err(&data->client->dev,
   "failed to read configuration registers\n");
  return ret;
 }

 /* Reset GPIO */
 data->gpio_bitnum = MMA9553_MAX_BITNUM;
 ret = mma9553_conf_gpio(data);
 if (ret < 0)
  return ret;

 ret = mma9551_app_reset(data->client, MMA9551_RSC_PED);
 if (ret < 0)
  return ret;

 /* Init config registers */
 data->conf.sleepmin = MMA9553_DEFAULT_SLEEPMIN;
 data->conf.sleepmax = MMA9553_DEFAULT_SLEEPMAX;
 data->conf.sleepthd = MMA9553_DEFAULT_SLEEPTHD;
 data->conf.config = mma9553_set_bits(data->conf.config, 1,
          MMA9553_MASK_CONF_CONFIG);
 /*
 * Clear the activity debounce counter when the activity level changes,
 * so that the confidence level applies for any activity level.
 */
 data->conf.config = mma9553_set_bits(data->conf.config, 1,
          MMA9553_MASK_CONF_ACT_DBCNTM);
 ret = mma9551_write_config_words(data->client, MMA9551_APPID_PEDOMETER,
      MMA9553_REG_CONF_SLEEPMIN,
      sizeof(data->conf) / sizeof(u16),
      (u16 *)&data->conf);
 if (ret < 0) {
  dev_err(&data->client->dev,
   "failed to write configuration registers\n");
  return ret;
 }

 return mma9551_set_device_state(data->client, true);
}

static int mma9553_read_status_word(struct mma9553_data *data, u16 reg,
        u16 *tmp)
{
 bool powered_on;
 int ret;

 /*
 * The HW only counts steps and other dependent
 * parameters (speed, distance, calories, activity)
 * if power is on (from enabling an event or the
 * step counter).
 */
 powered_on = mma9553_is_any_event_enabled(data, false, 0) ||
       data->stepcnt_enabled;
 if (!powered_on) {
  dev_err(&data->client->dev, "No channels enabled\n");
  return -EINVAL;
 }

 mutex_lock(&data->mutex);
 ret = mma9551_read_status_word(data->client, MMA9551_APPID_PEDOMETER,
           reg, tmp);
 mutex_unlock(&data->mutex);
 return ret;
}

static int mma9553_read_raw(struct iio_dev *indio_dev,
       struct iio_chan_spec const *chan,
       int *val, int *val2, long mask)
{
 struct mma9553_data *data = iio_priv(indio_dev);
 int ret;
 u16 tmp;
 u8 activity;

 switch (mask) {
 case IIO_CHAN_INFO_PROCESSED:
  switch (chan->type) {
  case IIO_STEPS:
   ret = mma9553_read_status_word(data,
             MMA9553_REG_STEPCNT,
             &tmp);
   if (ret < 0)
    return ret;
   *val = tmp;
   return IIO_VAL_INT;
  case IIO_DISTANCE:
   ret = mma9553_read_status_word(data,
             MMA9553_REG_DISTANCE,
             &tmp);
   if (ret < 0)
    return ret;
   *val = tmp;
   return IIO_VAL_INT;
  case IIO_ACTIVITY:
   ret = mma9553_read_status_word(data,
             MMA9553_REG_STATUS,
             &tmp);
   if (ret < 0)
    return ret;

   activity =
       mma9553_get_bits(tmp, MMA9553_MASK_STATUS_ACTIVITY);

   /*
 * The device does not support confidence value levels,
 * so we will always have 100% for current activity and
 * 0% for the others.
 */
   if (chan->channel2 == mma9553_activity_to_mod(activity))
    *val = 100;
   else
    *val = 0;
   return IIO_VAL_INT;
  default:
   return -EINVAL;
  }
 case IIO_CHAN_INFO_RAW:
  switch (chan->type) {
  case IIO_VELOCITY: /* m/h */
   if (chan->channel2 != IIO_MOD_ROOT_SUM_SQUARED_X_Y_Z)
    return -EINVAL;
   ret = mma9553_read_status_word(data,
             MMA9553_REG_SPEED,
             &tmp);
   if (ret < 0)
    return ret;
   *val = tmp;
   return IIO_VAL_INT;
  case IIO_ENERGY: /* Cal or kcal */
   ret = mma9553_read_status_word(data,
             MMA9553_REG_CALORIES,
             &tmp);
   if (ret < 0)
    return ret;
   *val = tmp;
   return IIO_VAL_INT;
  case IIO_ACCEL:
   mutex_lock(&data->mutex);
   ret = mma9551_read_accel_chan(data->client,
            chan, val, val2);
   mutex_unlock(&data->mutex);
   return ret;
  default:
   return -EINVAL;
  }
 case IIO_CHAN_INFO_SCALE:
  switch (chan->type) {
  case IIO_VELOCITY: /* m/h to m/s */
   if (chan->channel2 != IIO_MOD_ROOT_SUM_SQUARED_X_Y_Z)
    return -EINVAL;
   *val = 0;
   *val2 = 277; /* 0.000277 */
   return IIO_VAL_INT_PLUS_MICRO;
  case IIO_ENERGY: /* Cal or kcal to J */
   *val = 4184;
   return IIO_VAL_INT;
  case IIO_ACCEL:
   return mma9551_read_accel_scale(val, val2);
  default:
   return -EINVAL;
  }
 case IIO_CHAN_INFO_ENABLE:
  *val = data->stepcnt_enabled;
  return IIO_VAL_INT;
 case IIO_CHAN_INFO_CALIBHEIGHT:
  tmp = mma9553_get_bits(data->conf.height_weight,
           MMA9553_MASK_CONF_HEIGHT);
  *val = tmp / 100; /* cm to m */
  *val2 = (tmp % 100) * 10000;
  return IIO_VAL_INT_PLUS_MICRO;
 case IIO_CHAN_INFO_CALIBWEIGHT:
  *val = mma9553_get_bits(data->conf.height_weight,
     MMA9553_MASK_CONF_WEIGHT);
  return IIO_VAL_INT;
 case IIO_CHAN_INFO_DEBOUNCE_COUNT:
  switch (chan->type) {
  case IIO_STEPS:
   *val = mma9553_get_bits(data->conf.filter,
      MMA9553_MASK_CONF_FILTSTEP);
   return IIO_VAL_INT;
  default:
   return -EINVAL;
  }
 case IIO_CHAN_INFO_DEBOUNCE_TIME:
  switch (chan->type) {
  case IIO_STEPS:
   *val = mma9553_get_bits(data->conf.filter,
      MMA9553_MASK_CONF_FILTTIME);
   return IIO_VAL_INT;
  default:
   return -EINVAL;
  }
 case IIO_CHAN_INFO_INT_TIME:
  switch (chan->type) {
  case IIO_VELOCITY:
   if (chan->channel2 != IIO_MOD_ROOT_SUM_SQUARED_X_Y_Z)
    return -EINVAL;
   *val = mma9553_get_bits(data->conf.speed_step,
      MMA9553_MASK_CONF_SPDPRD);
   return IIO_VAL_INT;
  default:
   return -EINVAL;
  }
 default:
  return -EINVAL;
 }
}

static int mma9553_write_raw(struct iio_dev *indio_dev,
        struct iio_chan_spec const *chan,
        int val, int val2, long mask)
{
 struct mma9553_data *data = iio_priv(indio_dev);
 int ret, tmp;

 switch (mask) {
 case IIO_CHAN_INFO_ENABLE:
  if (data->stepcnt_enabled == !!val)
   return 0;
  mutex_lock(&data->mutex);
  ret = mma9551_set_power_state(data->client, val);
  if (ret < 0) {
   mutex_unlock(&data->mutex);
   return ret;
  }
  data->stepcnt_enabled = val;
  mutex_unlock(&data->mutex);
  return 0;
 case IIO_CHAN_INFO_CALIBHEIGHT:
  /* m to cm */
  tmp = val * 100 + val2 / 10000;
  if (tmp < 0 || tmp > 255)
   return -EINVAL;
  mutex_lock(&data->mutex);
  ret = mma9553_set_config(data,
      MMA9553_REG_CONF_HEIGHT_WEIGHT,
      &data->conf.height_weight,
      tmp, MMA9553_MASK_CONF_HEIGHT);
  mutex_unlock(&data->mutex);
  return ret;
 case IIO_CHAN_INFO_CALIBWEIGHT:
  if (val < 0 || val > 255)
   return -EINVAL;
  mutex_lock(&data->mutex);
  ret = mma9553_set_config(data,
      MMA9553_REG_CONF_HEIGHT_WEIGHT,
      &data->conf.height_weight,
      val, MMA9553_MASK_CONF_WEIGHT);
  mutex_unlock(&data->mutex);
  return ret;
 case IIO_CHAN_INFO_DEBOUNCE_COUNT:
  switch (chan->type) {
  case IIO_STEPS:
   /*
 * Set to 0 to disable step filtering. If the value
 * specified is greater than 6, then 6 will be used.
 */
   if (val < 0)
    return -EINVAL;
   if (val > 6)
    val = 6;
   mutex_lock(&data->mutex);
   ret = mma9553_set_config(data, MMA9553_REG_CONF_FILTER,
       &data->conf.filter, val,
       MMA9553_MASK_CONF_FILTSTEP);
   mutex_unlock(&data->mutex);
   return ret;
  default:
   return -EINVAL;
  }
 case IIO_CHAN_INFO_DEBOUNCE_TIME:
  switch (chan->type) {
  case IIO_STEPS:
   if (val < 0 || val > 127)
    return -EINVAL;
   mutex_lock(&data->mutex);
   ret = mma9553_set_config(data, MMA9553_REG_CONF_FILTER,
       &data->conf.filter, val,
       MMA9553_MASK_CONF_FILTTIME);
   mutex_unlock(&data->mutex);
   return ret;
  default:
   return -EINVAL;
  }
 case IIO_CHAN_INFO_INT_TIME:
  switch (chan->type) {
  case IIO_VELOCITY:
   if (chan->channel2 != IIO_MOD_ROOT_SUM_SQUARED_X_Y_Z)
    return -EINVAL;
   /*
 * If set to a value greater than 5, then 5 will be
 * used. Warning: Do not set SPDPRD to 0 or 1 as
 * this may cause undesirable behavior.
 */
   if (val < 2)
    return -EINVAL;
   if (val > 5)
    val = 5;
   mutex_lock(&data->mutex);
   ret = mma9553_set_config(data,
       MMA9553_REG_CONF_SPEED_STEP,
       &data->conf.speed_step, val,
       MMA9553_MASK_CONF_SPDPRD);
   mutex_unlock(&data->mutex);
   return ret;
  default:
   return -EINVAL;
  }
 default:
  return -EINVAL;
 }
}

static int mma9553_read_event_config(struct iio_dev *indio_dev,
         const struct iio_chan_spec *chan,
         enum iio_event_type type,
         enum iio_event_direction dir)
{
 struct mma9553_data *data = iio_priv(indio_dev);
 struct mma9553_event *event;

 event = mma9553_get_event(data, chan->type, chan->channel2, dir);
 if (!event)
  return -EINVAL;

 return event->enabled;
}

static int mma9553_write_event_config(struct iio_dev *indio_dev,
          const struct iio_chan_spec *chan,
          enum iio_event_type type,
          enum iio_event_direction dir,
          bool state)
{
 struct mma9553_data *data = iio_priv(indio_dev);
 struct mma9553_event *event;
 int ret;

 event = mma9553_get_event(data, chan->type, chan->channel2, dir);
 if (!event)
  return -EINVAL;

 if (event->enabled == state)
  return 0;

 mutex_lock(&data->mutex);

 ret = mma9551_set_power_state(data->client, state);
 if (ret < 0)
  goto err_out;
 event->enabled = state;

 ret = mma9553_conf_gpio(data);
 if (ret < 0)
  goto err_conf_gpio;

 mutex_unlock(&data->mutex);

 return 0;

err_conf_gpio:
 if (state) {
  event->enabled = false;
  mma9551_set_power_state(data->client, false);
 }
err_out:
 mutex_unlock(&data->mutex);
 return ret;
}

static int mma9553_read_event_value(struct iio_dev *indio_dev,
        const struct iio_chan_spec *chan,
        enum iio_event_type type,
        enum iio_event_direction dir,
        enum iio_event_info info,
        int *val, int *val2)
{
 struct mma9553_data *data = iio_priv(indio_dev);

 *val2 = 0;
 switch (info) {
 case IIO_EV_INFO_VALUE:
  switch (chan->type) {
  case IIO_STEPS:
   *val = mma9553_get_bits(data->conf.speed_step,
      MMA9553_MASK_CONF_STEPCOALESCE);
   return IIO_VAL_INT;
  case IIO_ACTIVITY:
   /*
 * The device does not support confidence value levels.
 * We set an average of 50%.
 */
   *val = 50;
   return IIO_VAL_INT;
  default:
   return -EINVAL;
  }
 case IIO_EV_INFO_PERIOD:
  switch (chan->type) {
  case IIO_ACTIVITY:
   *val = MMA9553_ACTIVITY_THD_TO_SEC(data->conf.actthd);
   return IIO_VAL_INT;
  default:
   return -EINVAL;
  }
 default:
  return -EINVAL;
 }
}

static int mma9553_write_event_value(struct iio_dev *indio_dev,
         const struct iio_chan_spec *chan,
         enum iio_event_type type,
         enum iio_event_direction dir,
         enum iio_event_info info,
         int val, int val2)
{
 struct mma9553_data *data = iio_priv(indio_dev);
 int ret;

 switch (info) {
 case IIO_EV_INFO_VALUE:
  switch (chan->type) {
  case IIO_STEPS:
   if (val < 0 || val > 255)
    return -EINVAL;
   mutex_lock(&data->mutex);
   ret = mma9553_set_config(data,
      MMA9553_REG_CONF_SPEED_STEP,
      &data->conf.speed_step, val,
      MMA9553_MASK_CONF_STEPCOALESCE);
   mutex_unlock(&data->mutex);
   return ret;
  default:
   return -EINVAL;
  }
 case IIO_EV_INFO_PERIOD:
  switch (chan->type) {
  case IIO_ACTIVITY:
   if (val < 0 || val > MMA9553_ACTIVITY_THD_TO_SEC(
       MMA9553_MAX_ACTTHD))
    return -EINVAL;
   mutex_lock(&data->mutex);
   ret = mma9553_set_config(data, MMA9553_REG_CONF_ACTTHD,
       &data->conf.actthd,
       MMA9553_ACTIVITY_SEC_TO_THD
       (val), MMA9553_MASK_CONF_WORD);
   mutex_unlock(&data->mutex);
   return ret;
  default:
   return -EINVAL;
  }
 default:
  return -EINVAL;
 }
}

static int mma9553_get_calibgender_mode(struct iio_dev *indio_dev,
     const struct iio_chan_spec *chan)
{
 struct mma9553_data *data = iio_priv(indio_dev);
 u8 gender;

 gender = mma9553_get_bits(data->conf.filter, MMA9553_MASK_CONF_MALE);
 /*
 * HW expects 0 for female and 1 for male,
 * while iio index is 0 for male and 1 for female.
 */
 return !gender;
}

static int mma9553_set_calibgender_mode(struct iio_dev *indio_dev,
     const struct iio_chan_spec *chan,
     unsigned int mode)
{
 struct mma9553_data *data = iio_priv(indio_dev);
 u8 gender = !mode;
 int ret;

 if ((mode != 0) && (mode != 1))
  return -EINVAL;
 mutex_lock(&data->mutex);
 ret = mma9553_set_config(data, MMA9553_REG_CONF_FILTER,
     &data->conf.filter, gender,
     MMA9553_MASK_CONF_MALE);
 mutex_unlock(&data->mutex);

 return ret;
}

static const struct iio_event_spec mma9553_step_event = {
 .type = IIO_EV_TYPE_CHANGE,
 .dir = IIO_EV_DIR_NONE,
 .mask_separate = BIT(IIO_EV_INFO_ENABLE) | BIT(IIO_EV_INFO_VALUE),
};

static const struct iio_event_spec mma9553_activity_events[] = {
 {
  .type = IIO_EV_TYPE_THRESH,
  .dir = IIO_EV_DIR_RISING,
  .mask_separate = BIT(IIO_EV_INFO_ENABLE) |
     BIT(IIO_EV_INFO_VALUE) |
     BIT(IIO_EV_INFO_PERIOD),
  },
 {
  .type = IIO_EV_TYPE_THRESH,
  .dir = IIO_EV_DIR_FALLING,
  .mask_separate = BIT(IIO_EV_INFO_ENABLE) |
     BIT(IIO_EV_INFO_VALUE) |
     BIT(IIO_EV_INFO_PERIOD),
 },
};

static const char * const mma9553_calibgender_modes[] = { "male", "female" };

static const struct iio_enum mma9553_calibgender_enum = {
 .items = mma9553_calibgender_modes,
 .num_items = ARRAY_SIZE(mma9553_calibgender_modes),
 .get = mma9553_get_calibgender_mode,
 .set = mma9553_set_calibgender_mode,
};

static const struct iio_chan_spec_ext_info mma9553_ext_info[] = {
 IIO_ENUM("calibgender", IIO_SHARED_BY_TYPE, &mma9553_calibgender_enum),
 IIO_ENUM_AVAILABLE("calibgender", IIO_SHARED_BY_TYPE, &mma9553_calibgender_enum),
 { }
};

#define MMA9553_PEDOMETER_CHANNEL(_type, _mask) {  \
 .type = _type,      \
 .info_mask_separate = BIT(IIO_CHAN_INFO_ENABLE)      | \
         BIT(IIO_CHAN_INFO_CALIBHEIGHT) | \
         _mask,    \
 .ext_info = mma9553_ext_info,    \
}

#define MMA9553_ACTIVITY_CHANNEL(_chan2) {    \
 .type = IIO_ACTIVITY,      \
 .modified = 1,       \
 .channel2 = _chan2,      \
 .info_mask_separate = BIT(IIO_CHAN_INFO_PROCESSED),  \
 .info_mask_shared_by_type = BIT(IIO_CHAN_INFO_CALIBHEIGHT) | \
        BIT(IIO_CHAN_INFO_ENABLE),  \
 .event_spec = mma9553_activity_events,    \
 .num_event_specs = ARRAY_SIZE(mma9553_activity_events),  \
 .ext_info = mma9553_ext_info,     \
}

static const struct iio_chan_spec mma9553_channels[] = {
 MMA9551_ACCEL_CHANNEL(IIO_MOD_X),
 MMA9551_ACCEL_CHANNEL(IIO_MOD_Y),
 MMA9551_ACCEL_CHANNEL(IIO_MOD_Z),

 {
  .type = IIO_STEPS,
  .info_mask_separate = BIT(IIO_CHAN_INFO_PROCESSED) |
         BIT(IIO_CHAN_INFO_ENABLE) |
         BIT(IIO_CHAN_INFO_DEBOUNCE_COUNT) |
         BIT(IIO_CHAN_INFO_DEBOUNCE_TIME),
  .event_spec = &mma9553_step_event,
  .num_event_specs = 1,
 },

 MMA9553_PEDOMETER_CHANNEL(IIO_DISTANCE, BIT(IIO_CHAN_INFO_PROCESSED)),
 {
  .type = IIO_VELOCITY,
  .modified = 1,
  .channel2 = IIO_MOD_ROOT_SUM_SQUARED_X_Y_Z,
  .info_mask_separate = BIT(IIO_CHAN_INFO_RAW) |
          BIT(IIO_CHAN_INFO_SCALE) |
          BIT(IIO_CHAN_INFO_INT_TIME) |
          BIT(IIO_CHAN_INFO_ENABLE),
  .info_mask_shared_by_type = BIT(IIO_CHAN_INFO_CALIBHEIGHT),
  .ext_info = mma9553_ext_info,
 },
 MMA9553_PEDOMETER_CHANNEL(IIO_ENERGY, BIT(IIO_CHAN_INFO_RAW) |
      BIT(IIO_CHAN_INFO_SCALE) |
      BIT(IIO_CHAN_INFO_CALIBWEIGHT)),

 MMA9553_ACTIVITY_CHANNEL(IIO_MOD_RUNNING),
 MMA9553_ACTIVITY_CHANNEL(IIO_MOD_JOGGING),
 MMA9553_ACTIVITY_CHANNEL(IIO_MOD_WALKING),
 MMA9553_ACTIVITY_CHANNEL(IIO_MOD_STILL),
};

static const struct iio_info mma9553_info = {
 .read_raw = mma9553_read_raw,
 .write_raw = mma9553_write_raw,
 .read_event_config = mma9553_read_event_config,
 .write_event_config = mma9553_write_event_config,
 .read_event_value = mma9553_read_event_value,
 .write_event_value = mma9553_write_event_value,
};

static irqreturn_t mma9553_irq_handler(int irq, void *private)
{
 struct iio_dev *indio_dev = private;
 struct mma9553_data *data = iio_priv(indio_dev);

 data->timestamp = iio_get_time_ns(indio_dev);
 /*
 * Since we only configure the interrupt pin when an
 * event is enabled, we are sure we have at least
 * one event enabled at this point.
 */
 return IRQ_WAKE_THREAD;
}

static irqreturn_t mma9553_event_handler(int irq, void *private)
{
 struct iio_dev *indio_dev = private;
 struct mma9553_data *data = iio_priv(indio_dev);
 u16 stepcnt;
 u8 activity;
 struct mma9553_event *ev_activity, *ev_prev_activity, *ev_step_detect;
 int ret;

 mutex_lock(&data->mutex);
 ret = mma9553_read_activity_stepcnt(data, &activity, &stepcnt);
 if (ret < 0) {
  mutex_unlock(&data->mutex);
  return IRQ_HANDLED;
 }

 ev_prev_activity = mma9553_get_event(data, IIO_ACTIVITY,
          mma9553_activity_to_mod(
          data->activity),
          IIO_EV_DIR_FALLING);
 ev_activity = mma9553_get_event(data, IIO_ACTIVITY,
     mma9553_activity_to_mod(activity),
     IIO_EV_DIR_RISING);
 ev_step_detect = mma9553_get_event(data, IIO_STEPS, IIO_NO_MOD,
        IIO_EV_DIR_NONE);

 if (ev_step_detect->enabled && (stepcnt != data->stepcnt)) {
  data->stepcnt = stepcnt;
  iio_push_event(indio_dev,
          IIO_UNMOD_EVENT_CODE(IIO_STEPS, 0,
          IIO_EV_TYPE_CHANGE,
          IIO_EV_DIR_NONE),
          data->timestamp);
 }

 if (activity != data->activity) {
  data->activity = activity;
  /* ev_activity can be NULL if activity == ACTIVITY_UNKNOWN */
  if (ev_prev_activity && ev_prev_activity->enabled)
   iio_push_event(indio_dev,
           IIO_MOD_EVENT_CODE(IIO_ACTIVITY, 0,
      ev_prev_activity->info->mod,
      IIO_EV_TYPE_THRESH,
      IIO_EV_DIR_FALLING),
           data->timestamp);

  if (ev_activity && ev_activity->enabled)
   iio_push_event(indio_dev,
           IIO_MOD_EVENT_CODE(IIO_ACTIVITY, 0,
         ev_activity->info->mod,
         IIO_EV_TYPE_THRESH,
         IIO_EV_DIR_RISING),
           data->timestamp);
 }
 mutex_unlock(&data->mutex);

 return IRQ_HANDLED;
}

static int mma9553_probe(struct i2c_client *client)
{
 const struct i2c_device_id *id = i2c_client_get_device_id(client);
 struct mma9553_data *data;
 struct iio_dev *indio_dev;
 const char *name = NULL;
 int ret;

 indio_dev = devm_iio_device_alloc(&client->dev, sizeof(*data));
 if (!indio_dev)
  return -ENOMEM;

 data = iio_priv(indio_dev);
 i2c_set_clientdata(client, indio_dev);
 data->client = client;

 if (id)
  name = id->name;
 else
  name = iio_get_acpi_device_name(&client->dev);
 if (!name)
  return -ENOSYS;

 mutex_init(&data->mutex);
 mma9553_init_events(data);

 ret = mma9553_init(data);
 if (ret < 0)
  return ret;

 indio_dev->channels = mma9553_channels;
 indio_dev->num_channels = ARRAY_SIZE(mma9553_channels);
 indio_dev->name = name;
 indio_dev->modes = INDIO_DIRECT_MODE;
 indio_dev->info = &mma9553_info;

 if (client->irq > 0) {
  ret = devm_request_threaded_irq(&client->dev, client->irq,
      mma9553_irq_handler,
      mma9553_event_handler,
      IRQF_TRIGGER_RISING,
      "mma9553_event", indio_dev);
  if (ret < 0) {
   dev_err(&client->dev, "request irq %d failed\n",
    client->irq);
   goto out_poweroff;
  }
 }

 ret = pm_runtime_set_active(&client->dev);
 if (ret < 0)
  goto out_poweroff;

 pm_runtime_enable(&client->dev);
 pm_runtime_set_autosuspend_delay(&client->dev,
      MMA9551_AUTO_SUSPEND_DELAY_MS);
 pm_runtime_use_autosuspend(&client->dev);

 ret = iio_device_register(indio_dev);
 if (ret < 0) {
  dev_err(&client->dev, "unable to register iio device\n");
  goto err_pm_cleanup;
 }

 dev_dbg(&indio_dev->dev, "Registered device %s\n", name);
 return 0;

err_pm_cleanup:
 pm_runtime_dont_use_autosuspend(&client->dev);
 pm_runtime_disable(&client->dev);
out_poweroff:
 mma9551_set_device_state(client, false);
 return ret;
}

static void mma9553_remove(struct i2c_client *client)
{
 struct iio_dev *indio_dev = i2c_get_clientdata(client);
 struct mma9553_data *data = iio_priv(indio_dev);

 iio_device_unregister(indio_dev);

 pm_runtime_disable(&client->dev);
 pm_runtime_set_suspended(&client->dev);

 mutex_lock(&data->mutex);
 mma9551_set_device_state(data->client, false);
 mutex_unlock(&data->mutex);
}

static int mma9553_runtime_suspend(struct device *dev)
{
 struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
 struct mma9553_data *data = iio_priv(indio_dev);
 int ret;

 mutex_lock(&data->mutex);
 ret = mma9551_set_device_state(data->client, false);
 mutex_unlock(&data->mutex);
 if (ret < 0) {
  dev_err(&data->client->dev, "powering off device failed\n");
  return -EAGAIN;
 }

 return 0;
}

static int mma9553_runtime_resume(struct device *dev)
{
 struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
 struct mma9553_data *data = iio_priv(indio_dev);
 int ret;

 ret = mma9551_set_device_state(data->client, true);
 if (ret < 0)
  return ret;

 mma9551_sleep(MMA9553_DEFAULT_SAMPLE_RATE);

 return 0;
}

static int mma9553_suspend(struct device *dev)
{
 struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
 struct mma9553_data *data = iio_priv(indio_dev);
 int ret;

 mutex_lock(&data->mutex);
 ret = mma9551_set_device_state(data->client, false);
 mutex_unlock(&data->mutex);

 return ret;
}

static int mma9553_resume(struct device *dev)
{
 struct iio_dev *indio_dev = i2c_get_clientdata(to_i2c_client(dev));
 struct mma9553_data *data = iio_priv(indio_dev);
 int ret;

 mutex_lock(&data->mutex);
 ret = mma9551_set_device_state(data->client, true);
 mutex_unlock(&data->mutex);

 return ret;
}

static const struct dev_pm_ops mma9553_pm_ops = {
 SYSTEM_SLEEP_PM_OPS(mma9553_suspend, mma9553_resume)
 RUNTIME_PM_OPS(mma9553_runtime_suspend, mma9553_runtime_resume, NULL)
};

static const struct acpi_device_id mma9553_acpi_match[] = {
 {"MMA9553", 0},
 { }
};

MODULE_DEVICE_TABLE(acpi, mma9553_acpi_match);

static const struct i2c_device_id mma9553_id[] = {
 { "mma9553" },
 { }
};

MODULE_DEVICE_TABLE(i2c, mma9553_id);

static struct i2c_driver mma9553_driver = {
 .driver = {
     .name = "mma9553",
     .acpi_match_table = mma9553_acpi_match,
     .pm = pm_ptr(&mma9553_pm_ops),
 },
 .probe = mma9553_probe,
 .remove = mma9553_remove,
 .id_table = mma9553_id,
};

module_i2c_driver(mma9553_driver);

MODULE_AUTHOR("Irina Tirdea ");
MODULE_LICENSE("GPL v2");
MODULE_DESCRIPTION("MMA9553L pedometer platform driver");
MODULE_IMPORT_NS("IIO_MMA9551");