/*
 * adv7393 - ADV7393 Video Encoder Driver
 *
 * The encoder hardware does not support SECAM.
 *
 * Copyright (C) 2010-2012 ADVANSEE - http://www.advansee.com/
 * Benoît Thébaudeau <benoit.thebaudeau@advansee.com>
 *
 * Based on ADV7343 driver,
 *
 * Copyright (C) 2009 Texas Instruments Incorporated - http://www.ti.com/
 *
 * 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 version 2.
 *
 * This program is distributed .as is. WITHOUT ANY WARRANTY of any
 * kind, whether express or implied; without even the implied warranty
 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */

#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/ctype.h>
#include <linux/slab.h>
#include <linux/i2c.h>
#include <linux/device.h>
#include <linux/delay.h>
#include <linux/module.h>
#include <linux/videodev2.h>
#include <linux/uaccess.h>

#include <media/i2c/adv7393.h>
#include <media/v4l2-device.h>
#include <media/v4l2-ctrls.h>

#include "adv7393_regs.h"

MODULE_DESCRIPTION("ADV7393 video encoder driver");
MODULE_LICENSE("GPL");

static bool debug;
module_param(debug, bool, 0644);
MODULE_PARM_DESC(debug, "Debug level 0-1");

struct adv7393_state {
 struct v4l2_subdev sd;
 struct v4l2_ctrl_handler hdl;
 u8 reg00;
 u8 reg01;
 u8 reg02;
 u8 reg35;
 u8 reg80;
 u8 reg82;
 u32 output;
 v4l2_std_id std;
};

static inline struct adv7393_state *to_state(struct v4l2_subdev *sd)
{
 return container_of(sd, struct adv7393_state, sd);
}

static inline struct v4l2_subdev *to_sd(struct v4l2_ctrl *ctrl)
{
 return &container_of(ctrl->handler, struct adv7393_state, hdl)->sd;
}

static inline int adv7393_write(struct v4l2_subdev *sd, u8 reg, u8 value)
{
 struct i2c_client *client = v4l2_get_subdevdata(sd);

 return i2c_smbus_write_byte_data(client, reg, value);
}

static const u8 adv7393_init_reg_val[] = {
 ADV7393_SOFT_RESET, ADV7393_SOFT_RESET_DEFAULT,
 ADV7393_POWER_MODE_REG, ADV7393_POWER_MODE_REG_DEFAULT,

 ADV7393_HD_MODE_REG1, ADV7393_HD_MODE_REG1_DEFAULT,
 ADV7393_HD_MODE_REG2, ADV7393_HD_MODE_REG2_DEFAULT,
 ADV7393_HD_MODE_REG3, ADV7393_HD_MODE_REG3_DEFAULT,
 ADV7393_HD_MODE_REG4, ADV7393_HD_MODE_REG4_DEFAULT,
 ADV7393_HD_MODE_REG5, ADV7393_HD_MODE_REG5_DEFAULT,
 ADV7393_HD_MODE_REG6, ADV7393_HD_MODE_REG6_DEFAULT,
 ADV7393_HD_MODE_REG7, ADV7393_HD_MODE_REG7_DEFAULT,

 ADV7393_SD_MODE_REG1, ADV7393_SD_MODE_REG1_DEFAULT,
 ADV7393_SD_MODE_REG2, ADV7393_SD_MODE_REG2_DEFAULT,
 ADV7393_SD_MODE_REG3, ADV7393_SD_MODE_REG3_DEFAULT,
 ADV7393_SD_MODE_REG4, ADV7393_SD_MODE_REG4_DEFAULT,
 ADV7393_SD_MODE_REG5, ADV7393_SD_MODE_REG5_DEFAULT,
 ADV7393_SD_MODE_REG6, ADV7393_SD_MODE_REG6_DEFAULT,
 ADV7393_SD_MODE_REG7, ADV7393_SD_MODE_REG7_DEFAULT,
 ADV7393_SD_MODE_REG8, ADV7393_SD_MODE_REG8_DEFAULT,

 ADV7393_SD_TIMING_REG0, ADV7393_SD_TIMING_REG0_DEFAULT,

 ADV7393_SD_HUE_ADJUST, ADV7393_SD_HUE_ADJUST_DEFAULT,
 ADV7393_SD_CGMS_WSS0, ADV7393_SD_CGMS_WSS0_DEFAULT,
 ADV7393_SD_BRIGHTNESS_WSS, ADV7393_SD_BRIGHTNESS_WSS_DEFAULT,
};

/*
 *       2^32
 * FSC(reg) =  FSC (HZ) * --------
 *     27000000
 */
static const struct adv7393_std_info stdinfo[] = {
 {
  /* FSC(Hz) = 4,433,618.75 Hz */
  SD_STD_NTSC, 705268427, V4L2_STD_NTSC_443,
 }, {
  /* FSC(Hz) = 3,579,545.45 Hz */
  SD_STD_NTSC, 569408542, V4L2_STD_NTSC,
 }, {
  /* FSC(Hz) = 3,575,611.00 Hz */
  SD_STD_PAL_M, 568782678, V4L2_STD_PAL_M,
 }, {
  /* FSC(Hz) = 3,582,056.00 Hz */
  SD_STD_PAL_N, 569807903, V4L2_STD_PAL_Nc,
 }, {
  /* FSC(Hz) = 4,433,618.75 Hz */
  SD_STD_PAL_N, 705268427, V4L2_STD_PAL_N,
 }, {
  /* FSC(Hz) = 4,433,618.75 Hz */
  SD_STD_PAL_M, 705268427, V4L2_STD_PAL_60,
 }, {
  /* FSC(Hz) = 4,433,618.75 Hz */
  SD_STD_PAL_BDGHI, 705268427, V4L2_STD_PAL,
 },
};

static int adv7393_setstd(struct v4l2_subdev *sd, v4l2_std_id std)
{
 struct adv7393_state *state = to_state(sd);
 const struct adv7393_std_info *std_info;
 int num_std;
 u8 reg;
 u32 val;
 int err = 0;
 int i;

 num_std = ARRAY_SIZE(stdinfo);

 for (i = 0; i < num_std; i++) {
  if (stdinfo[i].stdid & std)
   break;
 }

 if (i == num_std) {
  v4l2_dbg(1, debug, sd,
    "Invalid std or std is not supported: %llx\n",
      (unsigned long long)std);
  return -EINVAL;
 }

 std_info = &stdinfo[i];

 /* Set the standard */
 val = state->reg80 & ~SD_STD_MASK;
 val |= std_info->standard_val3;
 err = adv7393_write(sd, ADV7393_SD_MODE_REG1, val);
 if (err < 0)
  goto setstd_exit;

 state->reg80 = val;

 /* Configure the input mode register */
 val = state->reg01 & ~INPUT_MODE_MASK;
 val |= SD_INPUT_MODE;
 err = adv7393_write(sd, ADV7393_MODE_SELECT_REG, val);
 if (err < 0)
  goto setstd_exit;

 state->reg01 = val;

 /* Program the sub carrier frequency registers */
 val = std_info->fsc_val;
 for (reg = ADV7393_FSC_REG0; reg <= ADV7393_FSC_REG3; reg++) {
  err = adv7393_write(sd, reg, val);
  if (err < 0)
   goto setstd_exit;
  val >>= 8;
 }

 val = state->reg82;

 /* Pedestal settings */
 if (std & (V4L2_STD_NTSC | V4L2_STD_NTSC_443))
  val |= SD_PEDESTAL_EN;
 else
  val &= SD_PEDESTAL_DI;

 err = adv7393_write(sd, ADV7393_SD_MODE_REG2, val);
 if (err < 0)
  goto setstd_exit;

 state->reg82 = val;

setstd_exit:
 if (err != 0)
  v4l2_err(sd, "Error setting std, write failed\n");

 return err;
}

static int adv7393_setoutput(struct v4l2_subdev *sd, u32 output_type)
{
 struct adv7393_state *state = to_state(sd);
 u8 val;
 int err = 0;

 if (output_type > ADV7393_SVIDEO_ID) {
  v4l2_dbg(1, debug, sd,
   "Invalid output type or output type not supported:%d\n",
        output_type);
  return -EINVAL;
 }

 /* Enable Appropriate DAC */
 val = state->reg00 & 0x03;

 if (output_type == ADV7393_COMPOSITE_ID)
  val |= ADV7393_COMPOSITE_POWER_VALUE;
 else if (output_type == ADV7393_COMPONENT_ID)
  val |= ADV7393_COMPONENT_POWER_VALUE;
 else
  val |= ADV7393_SVIDEO_POWER_VALUE;

 err = adv7393_write(sd, ADV7393_POWER_MODE_REG, val);
 if (err < 0)
  goto setoutput_exit;

 state->reg00 = val;

 /* Enable YUV output */
 val = state->reg02 | YUV_OUTPUT_SELECT;
 err = adv7393_write(sd, ADV7393_MODE_REG0, val);
 if (err < 0)
  goto setoutput_exit;

 state->reg02 = val;

 /* configure SD DAC Output 1 bit */
 val = state->reg82;
 if (output_type == ADV7393_COMPONENT_ID)
  val &= SD_DAC_OUT1_DI;
 else
  val |= SD_DAC_OUT1_EN;
 err = adv7393_write(sd, ADV7393_SD_MODE_REG2, val);
 if (err < 0)
  goto setoutput_exit;

 state->reg82 = val;

 /* configure ED/HD Color DAC Swap bit to zero */
 val = state->reg35 & HD_DAC_SWAP_DI;
 err = adv7393_write(sd, ADV7393_HD_MODE_REG6, val);
 if (err < 0)
  goto setoutput_exit;

 state->reg35 = val;

setoutput_exit:
 if (err != 0)
  v4l2_err(sd, "Error setting output, write failed\n");

 return err;
}

static int adv7393_log_status(struct v4l2_subdev *sd)
{
 struct adv7393_state *state = to_state(sd);

 v4l2_info(sd, "Standard: %llx\n", (unsigned long long)state->std);
 v4l2_info(sd, "Output: %s\n", (state->output == 0) ? "Composite" :
   ((state->output == 1) ? "Component" : "S-Video"));
 return 0;
}

static int adv7393_s_ctrl(struct v4l2_ctrl *ctrl)
{
 struct v4l2_subdev *sd = to_sd(ctrl);

 switch (ctrl->id) {
 case V4L2_CID_BRIGHTNESS:
  return adv7393_write(sd, ADV7393_SD_BRIGHTNESS_WSS,
     ctrl->val & SD_BRIGHTNESS_VALUE_MASK);

 case V4L2_CID_HUE:
  return adv7393_write(sd, ADV7393_SD_HUE_ADJUST,
     ctrl->val - ADV7393_HUE_MIN);

 case V4L2_CID_GAIN:
  return adv7393_write(sd, ADV7393_DAC123_OUTPUT_LEVEL,
     ctrl->val);
 }
 return -EINVAL;
}

static const struct v4l2_ctrl_ops adv7393_ctrl_ops = {
 .s_ctrl = adv7393_s_ctrl,
};

static const struct v4l2_subdev_core_ops adv7393_core_ops = {
 .log_status = adv7393_log_status,
};

static int adv7393_s_std_output(struct v4l2_subdev *sd, v4l2_std_id std)
{
 struct adv7393_state *state = to_state(sd);
 int err = 0;

 if (state->std == std)
  return 0;

 err = adv7393_setstd(sd, std);
 if (!err)
  state->std = std;

 return err;
}

static int adv7393_s_routing(struct v4l2_subdev *sd,
  u32 input, u32 output, u32 config)
{
 struct adv7393_state *state = to_state(sd);
 int err = 0;

 if (state->output == output)
  return 0;

 err = adv7393_setoutput(sd, output);
 if (!err)
  state->output = output;

 return err;
}

static const struct v4l2_subdev_video_ops adv7393_video_ops = {
 .s_std_output = adv7393_s_std_output,
 .s_routing = adv7393_s_routing,
};

static const struct v4l2_subdev_ops adv7393_ops = {
 .core = &adv7393_core_ops,
 .video = &adv7393_video_ops,
};

static int adv7393_initialize(struct v4l2_subdev *sd)
{
 struct adv7393_state *state = to_state(sd);
 int err = 0;
 int i;

 for (i = 0; i < ARRAY_SIZE(adv7393_init_reg_val); i += 2) {

  err = adv7393_write(sd, adv7393_init_reg_val[i],
     adv7393_init_reg_val[i+1]);
  if (err) {
   v4l2_err(sd, "Error initializing\n");
   return err;
  }
 }

 /* Configure for default video standard */
 err = adv7393_setoutput(sd, state->output);
 if (err < 0) {
  v4l2_err(sd, "Error setting output during init\n");
  return -EINVAL;
 }

 err = adv7393_setstd(sd, state->std);
 if (err < 0) {
  v4l2_err(sd, "Error setting std during init\n");
  return -EINVAL;
 }

 return err;
}

static int adv7393_probe(struct i2c_client *client)
{
 struct adv7393_state *state;
 int err;

 if (!i2c_check_functionality(client->adapter, I2C_FUNC_SMBUS_BYTE_DATA))
  return -ENODEV;

 v4l_info(client, "chip found @ 0x%x (%s)\n",
   client->addr << 1, client->adapter->name);

 state = devm_kzalloc(&client->dev, sizeof(*state), GFP_KERNEL);
 if (state == NULL)
  return -ENOMEM;

 state->reg00 = ADV7393_POWER_MODE_REG_DEFAULT;
 state->reg01 = 0x00;
 state->reg02 = 0x20;
 state->reg35 = ADV7393_HD_MODE_REG6_DEFAULT;
 state->reg80 = ADV7393_SD_MODE_REG1_DEFAULT;
 state->reg82 = ADV7393_SD_MODE_REG2_DEFAULT;

 state->output = ADV7393_COMPOSITE_ID;
 state->std = V4L2_STD_NTSC;

 v4l2_i2c_subdev_init(&state->sd, client, &adv7393_ops);

 v4l2_ctrl_handler_init(&state->hdl, 3);
 v4l2_ctrl_new_std(&state->hdl, &adv7393_ctrl_ops,
   V4L2_CID_BRIGHTNESS, ADV7393_BRIGHTNESS_MIN,
          ADV7393_BRIGHTNESS_MAX, 1,
          ADV7393_BRIGHTNESS_DEF);
 v4l2_ctrl_new_std(&state->hdl, &adv7393_ctrl_ops,
   V4L2_CID_HUE, ADV7393_HUE_MIN,
          ADV7393_HUE_MAX, 1,
          ADV7393_HUE_DEF);
 v4l2_ctrl_new_std(&state->hdl, &adv7393_ctrl_ops,
   V4L2_CID_GAIN, ADV7393_GAIN_MIN,
           ADV7393_GAIN_MAX, 1,
           ADV7393_GAIN_DEF);
 state->sd.ctrl_handler = &state->hdl;
 if (state->hdl.error) {
  int err = state->hdl.error;

  v4l2_ctrl_handler_free(&state->hdl);
  return err;
 }
 v4l2_ctrl_handler_setup(&state->hdl);

 err = adv7393_initialize(&state->sd);
 if (err)
  v4l2_ctrl_handler_free(&state->hdl);
 return err;
}

static void adv7393_remove(struct i2c_client *client)
{
 struct v4l2_subdev *sd = i2c_get_clientdata(client);
 struct adv7393_state *state = to_state(sd);

 v4l2_device_unregister_subdev(sd);
 v4l2_ctrl_handler_free(&state->hdl);
}

static const struct i2c_device_id adv7393_id[] = {
 { "adv7393" },
 {}
};
MODULE_DEVICE_TABLE(i2c, adv7393_id);

static struct i2c_driver adv7393_driver = {
 .driver = {
  .name = "adv7393",
 },
 .probe  = adv7393_probe,
 .remove  = adv7393_remove,
 .id_table = adv7393_id,
};
module_i2c_driver(adv7393_driver);