/*
   cx231xx-i2c.c - driver for Conexant Cx23100/101/102 USB video capture devices

   Copyright (C) 2008 <srinivasa.deevi at conexant dot com>
                Based on em28xx driver
                Based on Cx23885 driver

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.

   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.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

#include "cx231xx.h"
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/i2c.h>
#include <linux/i2c-mux.h>
#include <media/v4l2-common.h>
#include <media/tuner.h>


/* ----------------------------------------------------------- */

static unsigned int i2c_scan;
module_param(i2c_scan, int, 0444);
MODULE_PARM_DESC(i2c_scan, "scan i2c bus at insmod time");

static unsigned int i2c_debug;
module_param(i2c_debug, int, 0644);
MODULE_PARM_DESC(i2c_debug, "enable debug messages [i2c]");

#define dprintk1(lvl, fmt, args...)                     \
do {                                                    \
        if (i2c_debug >= lvl) {                         \
                printk(fmt, ##args);                    \
                }                                       \
} while (0)

#define dprintk2(lvl, fmt, args...)                     \
do {                                                    \
        if (i2c_debug >= lvl) {                         \
                printk(KERN_DEBUG "%s at %s: " fmt,     \
                       dev->name, __func__ , ##args);   \
      }                                                 \
} while (0)

static inline int get_real_i2c_port(struct cx231xx *dev, int bus_nr)
{
        if (bus_nr == 1)
                return dev->port_3_switch_enabled ? I2C_1_MUX_3 : I2C_1_MUX_1;
        return bus_nr;
}

static inline bool is_tuner(struct cx231xx *dev, struct cx231xx_i2c *bus,
                        const struct i2c_msg *msg, int tuner_type)
{
        int i2c_port = get_real_i2c_port(dev, bus->nr);

        if (i2c_port != dev->board.tuner_i2c_master)
                return false;

        if (msg->addr != dev->board.tuner_addr)
                return false;

        if (dev->tuner_type != tuner_type)
                return false;

        return true;
}

/*
 * cx231xx_i2c_send_bytes()
 */
static int cx231xx_i2c_send_bytes(struct i2c_adapter *i2c_adap,
                                  const struct i2c_msg *msg)
{
        struct cx231xx_i2c *bus = i2c_adap->algo_data;
        struct cx231xx *dev = bus->dev;
        struct cx231xx_i2c_xfer_data req_data;
        int status = 0;
        u16 size = 0;
        u8 loop = 0;
        u8 saddr_len = 1;
        u8 *buf_ptr = NULL;
        u16 saddr = 0;
        u8 need_gpio = 0;

        if (is_tuner(dev, bus, msg, TUNER_XC5000)) {
                size = msg->len;

                if (size == 2) {        /* register write sub addr */
                        /* Just writing sub address will cause problem
                        * to XC5000. So ignore the request */
                        return 0;
                } else if (size == 4) { /* register write with sub addr */
                        if (msg->len >= 2)
                                saddr = msg->buf[0] << 8 | msg->buf[1];
                        else if (msg->len == 1)
                                saddr = msg->buf[0];

                        switch (saddr) {
                        case 0x0000:    /* start tuner calibration mode */
                                need_gpio = 1;
                                /* FW Loading is done */
                                dev->xc_fw_load_done = 1;
                                break;
                        case 0x000D:    /* Set signal source */
                        case 0x0001:    /* Set TV standard - Video */
                        case 0x0002:    /* Set TV standard - Audio */
                        case 0x0003:    /* Set RF Frequency */
                                need_gpio = 1;
                                break;
                        default:
                                if (dev->xc_fw_load_done)
                                        need_gpio = 1;
                                break;
                        }

                        if (need_gpio) {
                                dprintk1(1,
                                "GPIO WRITE: addr 0x%x, len %d, saddr 0x%x\n",
                                msg->addr, msg->len, saddr);

                                return dev->cx231xx_gpio_i2c_write(dev,
                                                                   msg->addr,
                                                                   msg->buf,
                                                                   msg->len);
                        }
                }

                /* special case for Xc5000 tuner case */
                saddr_len = 1;

                /* adjust the length to correct length */
                size -= saddr_len;
                buf_ptr = (u8 *) (msg->buf + 1);

                do {
                        /* prepare xfer_data struct */
                        req_data.dev_addr = msg->addr;
                        req_data.direction = msg->flags;
                        req_data.saddr_len = saddr_len;
                        req_data.saddr_dat = msg->buf[0];
                        req_data.buf_size = size > 16 ? 16 : size;
                        req_data.p_buffer = (u8 *) (buf_ptr + loop * 16);

                        bus->i2c_nostop = (size > 16) ? 1 : 0;
                        bus->i2c_reserve = (loop == 0) ? 0 : 1;

                        /* usb send command */
                        status = dev->cx231xx_send_usb_command(bus, &req_data);
                        loop++;

                        if (size >= 16)
                                size -= 16;
                        else
                                size = 0;

                } while (size > 0);

                bus->i2c_nostop = 0;
                bus->i2c_reserve = 0;

        } else {                /* regular case */

                /* prepare xfer_data struct */
                req_data.dev_addr = msg->addr;
                req_data.direction = msg->flags;
                req_data.saddr_len = 0;
                req_data.saddr_dat = 0;
                req_data.buf_size = msg->len;
                req_data.p_buffer = msg->buf;

                /* usb send command */
                status = dev->cx231xx_send_usb_command(bus, &req_data);
        }

        return status < 0 ? status : 0;
}

/*
 * cx231xx_i2c_recv_bytes()
 * read a byte from the i2c device
 */
static int cx231xx_i2c_recv_bytes(struct i2c_adapter *i2c_adap,
                                  const struct i2c_msg *msg)
{
        struct cx231xx_i2c *bus = i2c_adap->algo_data;
        struct cx231xx *dev = bus->dev;
        struct cx231xx_i2c_xfer_data req_data;
        int status = 0;
        u16 saddr = 0;
        u8 need_gpio = 0;

        if (is_tuner(dev, bus, msg, TUNER_XC5000)) {
                if (msg->len == 2)
                        saddr = msg->buf[0] << 8 | msg->buf[1];
                else if (msg->len == 1)
                        saddr = msg->buf[0];

                if (dev->xc_fw_load_done) {

                        switch (saddr) {
                        case 0x0009:    /* BUSY check */
                                dprintk1(1,
                                "GPIO R E A D: Special case BUSY check \n");
                                /*Try read BUSY register, just set it to zero*/
                                msg->buf[0] = 0;
                                if (msg->len == 2)
                                        msg->buf[1] = 0;
                                return 0;
                        case 0x0004:    /* read Lock status */
                                need_gpio = 1;
                                break;

                        }

                        if (need_gpio) {
                                /* this is a special case to handle Xceive tuner
                                clock stretch issue with gpio based I2C */

                                dprintk1(1,
                                "GPIO R E A D: addr 0x%x, len %d, saddr 0x%x\n",
                                msg->addr, msg->len,
                                msg->buf[0] << 8 | msg->buf[1]);

                                status =
                                    dev->cx231xx_gpio_i2c_write(dev, msg->addr,
                                                                msg->buf,
                                                                msg->len);
                                status =
                                    dev->cx231xx_gpio_i2c_read(dev, msg->addr,
                                                               msg->buf,
                                                               msg->len);
                                return status;
                        }
                }

                /* prepare xfer_data struct */
                req_data.dev_addr = msg->addr;
                req_data.direction = msg->flags;
                req_data.saddr_len = msg->len;
                req_data.saddr_dat = msg->buf[0] << 8 | msg->buf[1];
                req_data.buf_size = msg->len;
                req_data.p_buffer = msg->buf;

                /* usb send command */
                status = dev->cx231xx_send_usb_command(bus, &req_data);

        } else {

                /* prepare xfer_data struct */
                req_data.dev_addr = msg->addr;
                req_data.direction = msg->flags;
                req_data.saddr_len = 0;
                req_data.saddr_dat = 0;
                req_data.buf_size = msg->len;
                req_data.p_buffer = msg->buf;

                /* usb send command */
                status = dev->cx231xx_send_usb_command(bus, &req_data);
        }

        return status < 0 ? status : 0;
}

/*
 * cx231xx_i2c_recv_bytes_with_saddr()
 * read a byte from the i2c device
 */
static int cx231xx_i2c_recv_bytes_with_saddr(struct i2c_adapter *i2c_adap,
                                             const struct i2c_msg *msg1,
                                             const struct i2c_msg *msg2)
{
        struct cx231xx_i2c *bus = i2c_adap->algo_data;
        struct cx231xx *dev = bus->dev;
        struct cx231xx_i2c_xfer_data req_data;
        int status = 0;
        u16 saddr = 0;
        u8 need_gpio = 0;

        if (msg1->len == 2)
                saddr = msg1->buf[0] << 8 | msg1->buf[1];
        else if (msg1->len == 1)
                saddr = msg1->buf[0];

        if (is_tuner(dev, bus, msg2, TUNER_XC5000)) {
                if ((msg2->len < 16)) {

                        dprintk1(1,
                        "i2c_read: addr 0x%x, len %d, saddr 0x%x, len %d\n",
                        msg2->addr, msg2->len, saddr, msg1->len);

                        switch (saddr) {
                        case 0x0008:    /* read FW load status */
                                need_gpio = 1;
                                break;
                        case 0x0004:    /* read Lock status */
                                need_gpio = 1;
                                break;
                        }

                        if (need_gpio) {
                                status =
                                    dev->cx231xx_gpio_i2c_write(dev, msg1->addr,
                                                                msg1->buf,
                                                                msg1->len);
                                status =
                                    dev->cx231xx_gpio_i2c_read(dev, msg2->addr,
                                                               msg2->buf,
                                                               msg2->len);
                                return status;
                        }
                }
        }

        /* prepare xfer_data struct */
        req_data.dev_addr = msg2->addr;
        req_data.direction = msg2->flags;
        req_data.saddr_len = msg1->len;
        req_data.saddr_dat = saddr;
        req_data.buf_size = msg2->len;
        req_data.p_buffer = msg2->buf;

        /* usb send command */
        status = dev->cx231xx_send_usb_command(bus, &req_data);

        return status < 0 ? status : 0;
}

/*
 * cx231xx_i2c_check_for_device()
 * check if there is a i2c_device at the supplied address
 */
static int cx231xx_i2c_check_for_device(struct i2c_adapter *i2c_adap,
                                        const struct i2c_msg *msg)
{
        struct cx231xx_i2c *bus = i2c_adap->algo_data;
        struct cx231xx *dev = bus->dev;
        struct cx231xx_i2c_xfer_data req_data;
        int status = 0;
        u8 buf[1];

        /* prepare xfer_data struct */
        req_data.dev_addr = msg->addr;
        req_data.direction = I2C_M_RD;
        req_data.saddr_len = 0;
        req_data.saddr_dat = 0;
        req_data.buf_size = 1;
        req_data.p_buffer = buf;

        /* usb send command */
        status = dev->cx231xx_send_usb_command(bus, &req_data);

        return status < 0 ? status : 0;
}

/*
 * cx231xx_i2c_xfer()
 * the main i2c transfer function
 */
static int cx231xx_i2c_xfer(struct i2c_adapter *i2c_adap,
                            struct i2c_msg msgs[], int num)
{
        struct cx231xx_i2c *bus = i2c_adap->algo_data;
        struct cx231xx *dev = bus->dev;
        int addr, rc, i, byte;

        if (num <= 0)
                return 0;
        mutex_lock(&dev->i2c_lock);
        for (i = 0; i < num; i++) {

                addr = msgs[i].addr;

                dprintk2(2, "%s %s addr=0x%x len=%d:",
                         (msgs[i].flags & I2C_M_RD) ? "read" : "write",
                         i == num - 1 ? "stop" : "nonstop", addr, msgs[i].len);
                if (!msgs[i].len) {
                        /* no len: check only for device presence */
                        rc = cx231xx_i2c_check_for_device(i2c_adap, &msgs[i]);
                        if (rc < 0) {
                                dprintk2(2, " no device\n");
                                mutex_unlock(&dev->i2c_lock);
                                return rc;
                        }

                } else if (msgs[i].flags & I2C_M_RD) {
                        /* read bytes */
                        rc = cx231xx_i2c_recv_bytes(i2c_adap, &msgs[i]);
                        if (i2c_debug >= 2) {
                                for (byte = 0; byte < msgs[i].len; byte++)
                                        printk(KERN_CONT " %02x", msgs[i].buf[byte]);
                        }
                } else if (i + 1 < num && (msgs[i + 1].flags & I2C_M_RD) &&
                           msgs[i].addr == msgs[i + 1].addr
                           && (msgs[i].len <= 2) && (bus->nr < 3)) {
                        /* write bytes */
                        if (i2c_debug >= 2) {
                                for (byte = 0; byte < msgs[i].len; byte++)
                                        printk(KERN_CONT " %02x", msgs[i].buf[byte]);
                                printk(KERN_CONT "\n");
                        }
                        /* read bytes */
                        dprintk2(2, "plus %s %s addr=0x%x len=%d:",
                                (msgs[i+1].flags & I2C_M_RD) ? "read" : "write",
                                i+1 == num - 1 ? "stop" : "nonstop", addr, msgs[i+1].len);
                        rc = cx231xx_i2c_recv_bytes_with_saddr(i2c_adap,
                                                               &msgs[i],
                                                               &msgs[i + 1]);
                        if (i2c_debug >= 2) {
                                for (byte = 0; byte < msgs[i+1].len; byte++)
                                        printk(KERN_CONT " %02x", msgs[i+1].buf[byte]);
                        }
                        i++;
                } else {
                        /* write bytes */
                        if (i2c_debug >= 2) {
                                for (byte = 0; byte < msgs[i].len; byte++)
                                        printk(KERN_CONT " %02x", msgs[i].buf[byte]);
                        }
                        rc = cx231xx_i2c_send_bytes(i2c_adap, &msgs[i]);
                }
                if (rc < 0)
                        goto err;
                if (i2c_debug >= 2)
                        printk(KERN_CONT "\n");
        }
        mutex_unlock(&dev->i2c_lock);
        return num;
err:
        dprintk2(2, " ERROR: %i\n", rc);
        mutex_unlock(&dev->i2c_lock);
        return rc;
}

/* ----------------------------------------------------------- */

/*
 * functionality()
 */
static u32 functionality(struct i2c_adapter *adap)
{
        return I2C_FUNC_SMBUS_EMUL | I2C_FUNC_I2C;
}

static struct i2c_algorithm cx231xx_algo = {
        .master_xfer = cx231xx_i2c_xfer,
        .functionality = functionality,
};

static struct i2c_adapter cx231xx_adap_template = {
        .owner = THIS_MODULE,
        .name = "cx231xx",
        .algo = &cx231xx_algo,
};

/* ----------------------------------------------------------- */

/*
 * i2c_devs
 * incomplete list of known devices
 */
static const char *i2c_devs[128] = {
        [0x20 >> 1] = "demod",
        [0x60 >> 1] = "colibri",
        [0x88 >> 1] = "hammerhead",
        [0x8e >> 1] = "CIR",
        [0x32 >> 1] = "GeminiIII",
        [0x02 >> 1] = "Aquarius",
        [0xa0 >> 1] = "eeprom",
        [0xc0 >> 1] = "tuner",
        [0xc2 >> 1] = "tuner",
};

/*
 * cx231xx_do_i2c_scan()
 * check i2c address range for devices
 */
void cx231xx_do_i2c_scan(struct cx231xx *dev, int i2c_port)
{
        unsigned char buf;
        int i, rc;
        struct i2c_client client;

        if (!i2c_scan)
                return;

        /* Don't generate I2C errors during scan */
        dev->i2c_scan_running = true;

        memset(&client, 0, sizeof(client));
        client.adapter = cx231xx_get_i2c_adap(dev, i2c_port);

        for (i = 0; i < 128; i++) {
                client.addr = i;
                rc = i2c_master_recv(&client, &buf, 0);
                if (rc < 0)
                        continue;
                dev_info(dev->dev,
                         "i2c scan: found device @ port %d addr 0x%x  [%s]\n",
                         i2c_port,
                         i << 1,
                         i2c_devs[i] ? i2c_devs[i] : "???");
        }

        dev->i2c_scan_running = false;
}

/*
 * cx231xx_i2c_register()
 * register i2c bus
 */
int cx231xx_i2c_register(struct cx231xx_i2c *bus)
{
        struct cx231xx *dev = bus->dev;

        BUG_ON(!dev->cx231xx_send_usb_command);

        bus->i2c_adap = cx231xx_adap_template;
        bus->i2c_adap.dev.parent = dev->dev;

        snprintf(bus->i2c_adap.name, sizeof(bus->i2c_adap.name), "%s-%d", bus->dev->name, bus->nr);

        bus->i2c_adap.algo_data = bus;
        i2c_set_adapdata(&bus->i2c_adap, &dev->v4l2_dev);
        i2c_add_adapter(&bus->i2c_adap);

        if (0 != bus->i2c_rc)
                dev_warn(dev->dev,
                         "i2c bus %d register FAILED\n", bus->nr);

        return bus->i2c_rc;
}

/*
 * cx231xx_i2c_unregister()
 * unregister i2c_bus
 */
int cx231xx_i2c_unregister(struct cx231xx_i2c *bus)
{
        i2c_del_adapter(&bus->i2c_adap);
        return 0;
}

/*
 * cx231xx_i2c_mux_select()
 * switch i2c master number 1 between port1 and port3
 */
static int cx231xx_i2c_mux_select(struct i2c_adapter *adap,
                        void *mux_priv, u32 chan_id)
{
        struct cx231xx *dev = mux_priv;

        return cx231xx_enable_i2c_port_3(dev, chan_id);
}

int cx231xx_i2c_mux_register(struct cx231xx *dev, int mux_no)
{
        struct i2c_adapter *i2c_parent = &dev->i2c_bus[1].i2c_adap;
        /* what is the correct mux_dev? */
        struct device *mux_dev = dev->dev;

        dev->i2c_mux_adap[mux_no] = i2c_add_mux_adapter(i2c_parent,
                                mux_dev,
                                dev /* mux_priv */,
                                0,
                                mux_no /* chan_id */,
                                0 /* class */,
                                &cx231xx_i2c_mux_select,
                                NULL);

        if (!dev->i2c_mux_adap[mux_no])
                dev_warn(dev->dev,
                         "i2c mux %d register FAILED\n", mux_no);

        return 0;
}

void cx231xx_i2c_mux_unregister(struct cx231xx *dev, int mux_no)
{
        i2c_del_mux_adapter(dev->i2c_mux_adap[mux_no]);
        dev->i2c_mux_adap[mux_no] = NULL;
}

struct i2c_adapter *cx231xx_get_i2c_adap(struct cx231xx *dev, int i2c_port)
{
        switch (i2c_port) {
        case I2C_0:
                return &dev->i2c_bus[0].i2c_adap;
        case I2C_1:
                return &dev->i2c_bus[1].i2c_adap;
        case I2C_2:
                return &dev->i2c_bus[2].i2c_adap;
        case I2C_1_MUX_1:
                return dev->i2c_mux_adap[0];
        case I2C_1_MUX_3:
                return dev->i2c_mux_adap[1];
        default:
                return NULL;
        }
}
EXPORT_SYMBOL_GPL(cx231xx_get_i2c_adap);