#include <linux/delay.h>
#include <linux/gpio/consumer.h>
#include <linux/i2c.h>
#include <linux/led-class-flash.h>
#include <linux/leds.h>
#include <linux/module.h>
#include <linux/mutex.h>
#include <linux/property.h>
#include <linux/slab.h>
#include <media/v4l2-flash-led-class.h>
#define AS_TIMER_US_TO_CODE(t) (((t) / 1000 - 100) / 50)
#define AS_TIMER_CODE_TO_US(c) ((50 * (c) + 100) * 1000)
#define AS_DESIGN_INFO_REG 0x00
#define AS_DESIGN_INFO_FACTORY(x) (((x) >> 4))
#define AS_DESIGN_INFO_MODEL(x) ((x) & 0x0f)
#define AS_VERSION_CONTROL_REG 0x01
#define AS_VERSION_CONTROL_RFU(x) (((x) >> 4))
#define AS_VERSION_CONTROL_VERSION(x) ((x) & 0x0f)
#define AS_INDICATOR_AND_TIMER_REG 0x02
#define AS_INDICATOR_AND_TIMER_TIMEOUT_SHIFT 0
#define AS_INDICATOR_AND_TIMER_VREF_SHIFT 4
#define AS_INDICATOR_AND_TIMER_INDICATOR_SHIFT 6
#define AS_CURRENT_SET_REG 0x03
#define AS_CURRENT_ASSIST_LIGHT_SHIFT 0
#define AS_CURRENT_LED_DET_ON (1 << 3)
#define AS_CURRENT_FLASH_CURRENT_SHIFT 4
#define AS_CONTROL_REG 0x04
#define AS_CONTROL_MODE_SETTING_SHIFT 0
#define AS_CONTROL_STROBE_ON (1 << 2)
#define AS_CONTROL_OUT_ON (1 << 3)
#define AS_CONTROL_EXT_TORCH_ON (1 << 4)
#define AS_CONTROL_STROBE_TYPE_EDGE (0 << 5)
#define AS_CONTROL_STROBE_TYPE_LEVEL (1 << 5)
#define AS_CONTROL_COIL_PEAK_SHIFT 6
#define AS_FAULT_INFO_REG 0x05
#define AS_FAULT_INFO_INDUCTOR_PEAK_LIMIT (1 << 1)
#define AS_FAULT_INFO_INDICATOR_LED (1 << 2)
#define AS_FAULT_INFO_LED_AMOUNT (1 << 3)
#define AS_FAULT_INFO_TIMEOUT (1 << 4)
#define AS_FAULT_INFO_OVER_TEMPERATURE (1 << 5)
#define AS_FAULT_INFO_SHORT_CIRCUIT (1 << 6)
#define AS_FAULT_INFO_OVER_VOLTAGE (1 << 7)
#define AS_BOOST_REG 0x0d
#define AS_BOOST_CURRENT_DISABLE (0 << 0)
#define AS_BOOST_CURRENT_ENABLE (1 << 0)
#define AS_PASSWORD_REG 0x0f
#define AS_PASSWORD_UNLOCK_VALUE 0x55
#define AS_NAME "as3645a"
#define AS_I2C_ADDR (0x60 >> 1)
#define AS_FLASH_TIMEOUT_MIN 100000
#define AS_FLASH_TIMEOUT_MAX 850000
#define AS_FLASH_TIMEOUT_STEP 50000
#define AS_FLASH_INTENSITY_MIN 200000
#define AS_FLASH_INTENSITY_MAX_1LED 500000
#define AS_FLASH_INTENSITY_MAX_2LEDS 400000
#define AS_FLASH_INTENSITY_STEP 20000
#define AS_TORCH_INTENSITY_MIN 20000
#define AS_TORCH_INTENSITY_MAX 160000
#define AS_TORCH_INTENSITY_STEP 20000
#define AS_INDICATOR_INTENSITY_MIN 0
#define AS_INDICATOR_INTENSITY_MAX 10000
#define AS_INDICATOR_INTENSITY_STEP 2500
#define AS_PEAK_mA_MAX 2000
#define AS_PEAK_mA_TO_REG(a) \
((min_t(u32, AS_PEAK_mA_MAX, a) - 1250) / 250)
#define AS_LED_FLASH 0
#define AS_LED_INDICATOR 1
enum as_mode {
AS_MODE_EXT_TORCH = 0 << AS_CONTROL_MODE_SETTING_SHIFT,
AS_MODE_INDICATOR = 1 << AS_CONTROL_MODE_SETTING_SHIFT,
AS_MODE_ASSIST = 2 << AS_CONTROL_MODE_SETTING_SHIFT,
AS_MODE_FLASH = 3 << AS_CONTROL_MODE_SETTING_SHIFT,
};
struct as3645a_config {
u32 flash_timeout_us;
u32 flash_max_ua;
u32 assist_max_ua;
u32 indicator_max_ua;
u32 voltage_reference;
u32 peak;
};
struct as3645a {
struct i2c_client *client;
struct mutex mutex;
struct led_classdev_flash fled;
struct led_classdev iled_cdev;
struct v4l2_flash *vf;
struct v4l2_flash *vfind;
struct fwnode_handle *flash_node;
struct fwnode_handle *indicator_node;
struct as3645a_config cfg;
enum as_mode mode;
unsigned int timeout;
unsigned int flash_current;
unsigned int assist_current;
unsigned int indicator_current;
enum v4l2_flash_strobe_source strobe_source;
};
#define fled_to_as3645a(__fled) container_of(__fled, struct as3645a, fled)
#define iled_cdev_to_as3645a(__iled_cdev) \
container_of(__iled_cdev, struct as3645a, iled_cdev)
static int as3645a_write(struct as3645a *flash, u8 addr, u8 val)
{
struct i2c_client *client = flash->client;
int rval;
rval = i2c_smbus_write_byte_data(client, addr, val);
dev_dbg(&client->dev, "Write Addr:%02X Val:%02X %s\n", addr, val,
rval < 0 ? "fail" : "ok");
return rval;
}
static int as3645a_read(struct as3645a *flash, u8 addr)
{
struct i2c_client *client = flash->client;
int rval;
rval = i2c_smbus_read_byte_data(client, addr);
dev_dbg(&client->dev, "Read Addr:%02X Val:%02X %s\n", addr, rval,
rval < 0 ? "fail" : "ok");
return rval;
}
static int as3645a_set_current(struct as3645a *flash)
{
u8 val;
val = (flash->flash_current << AS_CURRENT_FLASH_CURRENT_SHIFT)
| (flash->assist_current << AS_CURRENT_ASSIST_LIGHT_SHIFT)
| AS_CURRENT_LED_DET_ON;
return as3645a_write(flash, AS_CURRENT_SET_REG, val);
}
static int as3645a_set_timeout(struct as3645a *flash)
{
u8 val;
val = flash->timeout << AS_INDICATOR_AND_TIMER_TIMEOUT_SHIFT;
val |= (flash->cfg.voltage_reference
<< AS_INDICATOR_AND_TIMER_VREF_SHIFT)
| ((flash->indicator_current ? flash->indicator_current - 1 : 0)
<< AS_INDICATOR_AND_TIMER_INDICATOR_SHIFT);
return as3645a_write(flash, AS_INDICATOR_AND_TIMER_REG, val);
}
static int
as3645a_set_control(struct as3645a *flash, enum as_mode mode, bool on)
{
u8 reg;
reg = (flash->cfg.peak << AS_CONTROL_COIL_PEAK_SHIFT)
| (on ? AS_CONTROL_OUT_ON : 0)
| mode;
if (mode == AS_MODE_FLASH &&
flash->strobe_source == V4L2_FLASH_STROBE_SOURCE_EXTERNAL)
reg |= AS_CONTROL_STROBE_TYPE_LEVEL
| AS_CONTROL_STROBE_ON;
return as3645a_write(flash, AS_CONTROL_REG, reg);
}
static int as3645a_get_fault(struct led_classdev_flash *fled, u32 *fault)
{
struct as3645a *flash = fled_to_as3645a(fled);
int rval;
rval = as3645a_read(flash, AS_FAULT_INFO_REG);
if (rval < 0)
return rval;
if (rval & AS_FAULT_INFO_INDUCTOR_PEAK_LIMIT)
*fault |= LED_FAULT_OVER_CURRENT;
if (rval & AS_FAULT_INFO_INDICATOR_LED)
*fault |= LED_FAULT_INDICATOR;
dev_dbg(&flash->client->dev, "%u connected LEDs\n",
rval & AS_FAULT_INFO_LED_AMOUNT ? 2 : 1);
if (rval & AS_FAULT_INFO_TIMEOUT)
*fault |= LED_FAULT_TIMEOUT;
if (rval & AS_FAULT_INFO_OVER_TEMPERATURE)
*fault |= LED_FAULT_OVER_TEMPERATURE;
if (rval & AS_FAULT_INFO_SHORT_CIRCUIT)
*fault |= LED_FAULT_OVER_CURRENT;
if (rval & AS_FAULT_INFO_OVER_VOLTAGE)
*fault |= LED_FAULT_INPUT_VOLTAGE;
return rval;
}
static unsigned int __as3645a_current_to_reg(unsigned int min, unsigned int max,
unsigned int step,
unsigned int val)
{
if (val < min)
val = min;
if (val > max)
val = max;
return (val - min) / step;
}
static unsigned int as3645a_current_to_reg(struct as3645a *flash, bool is_flash,
unsigned int ua)
{
if (is_flash)
return __as3645a_current_to_reg(AS_TORCH_INTENSITY_MIN,
flash->cfg.assist_max_ua,
AS_TORCH_INTENSITY_STEP, ua);
else
return __as3645a_current_to_reg(AS_FLASH_INTENSITY_MIN,
flash->cfg.flash_max_ua,
AS_FLASH_INTENSITY_STEP, ua);
}
static int as3645a_set_indicator_brightness(struct led_classdev *iled_cdev,
enum led_brightness brightness)
{
struct as3645a *flash = iled_cdev_to_as3645a(iled_cdev);
int rval;
flash->indicator_current = brightness;
rval = as3645a_set_timeout(flash);
if (rval)
return rval;
return as3645a_set_control(flash, AS_MODE_INDICATOR, brightness);
}
static int as3645a_set_assist_brightness(struct led_classdev *fled_cdev,
enum led_brightness brightness)
{
struct led_classdev_flash *fled = lcdev_to_flcdev(fled_cdev);
struct as3645a *flash = fled_to_as3645a(fled);
int rval;
if (brightness) {
flash->assist_current = brightness - 1;
rval = as3645a_set_current(flash);
if (rval)
return rval;
}
return as3645a_set_control(flash, AS_MODE_ASSIST, brightness);
}
static int as3645a_set_flash_brightness(struct led_classdev_flash *fled,
u32 brightness_ua)
{
struct as3645a *flash = fled_to_as3645a(fled);
flash->flash_current = as3645a_current_to_reg(flash, true,
brightness_ua);
return as3645a_set_current(flash);
}
static int as3645a_set_flash_timeout(struct led_classdev_flash *fled,
u32 timeout_us)
{
struct as3645a *flash = fled_to_as3645a(fled);
flash->timeout = AS_TIMER_US_TO_CODE(timeout_us);
return as3645a_set_timeout(flash);
}
static int as3645a_set_strobe(struct led_classdev_flash *fled, bool state)
{
struct as3645a *flash = fled_to_as3645a(fled);
return as3645a_set_control(flash, AS_MODE_FLASH, state);
}
static const struct led_flash_ops as3645a_led_flash_ops = {
.flash_brightness_set = as3645a_set_flash_brightness,
.timeout_set = as3645a_set_flash_timeout,
.strobe_set = as3645a_set_strobe,
.fault_get = as3645a_get_fault,
};
static int as3645a_setup(struct as3645a *flash)
{
struct device *dev = &flash->client->dev;
u32 fault = 0;
int rval;
rval = as3645a_read(flash, AS_FAULT_INFO_REG);
if (rval < 0)
return rval;
dev_dbg(dev, "Fault info: %02x\n", rval);
rval = as3645a_set_current(flash);
if (rval < 0)
return rval;
rval = as3645a_set_timeout(flash);
if (rval < 0)
return rval;
rval = as3645a_set_control(flash, AS_MODE_INDICATOR, false);
if (rval < 0)
return rval;
rval = as3645a_get_fault(&flash->fled, &fault);
if (rval < 0)
return rval;
dev_dbg(dev, "AS_INDICATOR_AND_TIMER_REG: %02x\n",
as3645a_read(flash, AS_INDICATOR_AND_TIMER_REG));
dev_dbg(dev, "AS_CURRENT_SET_REG: %02x\n",
as3645a_read(flash, AS_CURRENT_SET_REG));
dev_dbg(dev, "AS_CONTROL_REG: %02x\n",
as3645a_read(flash, AS_CONTROL_REG));
return rval & ~AS_FAULT_INFO_LED_AMOUNT ? -EIO : 0;
}
static int as3645a_detect(struct as3645a *flash)
{
struct device *dev = &flash->client->dev;
int rval, man, model, rfu, version;
const char *vendor;
rval = as3645a_read(flash, AS_DESIGN_INFO_REG);
if (rval < 0) {
dev_err(dev, "can't read design info reg\n");
return rval;
}
man = AS_DESIGN_INFO_FACTORY(rval);
model = AS_DESIGN_INFO_MODEL(rval);
rval = as3645a_read(flash, AS_VERSION_CONTROL_REG);
if (rval < 0) {
dev_err(dev, "can't read version control reg\n");
return rval;
}
rfu = AS_VERSION_CONTROL_RFU(rval);
version = AS_VERSION_CONTROL_VERSION(rval);
if (model != 0x01 || rfu != 0x00) {
dev_err(dev, "AS3645A not detected (model %d rfu %d)\n",
model, rfu);
return -ENODEV;
}
switch (man) {
case 1:
vendor = "AMS, Austria Micro Systems";
break;
case 2:
vendor = "ADI, Analog Devices Inc.";
break;
case 3:
vendor = "NSC, National Semiconductor";
break;
case 4:
vendor = "NXP";
break;
case 5:
vendor = "TI, Texas Instrument";
break;
default:
vendor = "Unknown";
}
dev_info(dev, "Chip vendor: %s (%d) Version: %d\n", vendor,
man, version);
rval = as3645a_write(flash, AS_PASSWORD_REG, AS_PASSWORD_UNLOCK_VALUE);
if (rval < 0)
return rval;
return as3645a_write(flash, AS_BOOST_REG, AS_BOOST_CURRENT_DISABLE);
}
static int as3645a_parse_node(struct device *dev, struct as3645a *flash)
{
struct as3645a_config *cfg = &flash->cfg;
int rval;
device_for_each_child_node_scoped(dev, child) {
u32 id = 0;
fwnode_property_read_u32(child, "reg", &id);
switch (id) {
case AS_LED_FLASH:
flash->flash_node = child;
fwnode_handle_get(child);
break;
case AS_LED_INDICATOR:
flash->indicator_node = child;
fwnode_handle_get(child);
break;
default:
dev_warn(&flash->client->dev,
"unknown LED %u encountered, ignoring\n", id);
break;
}
}
if (!flash->flash_node) {
dev_err(&flash->client->dev, "can't find flash node\n");
return -ENODEV;
}
rval = fwnode_property_read_u32(flash->flash_node, "flash-timeout-us",
&cfg->flash_timeout_us);
if (rval < 0) {
dev_err(&flash->client->dev,
"can't read flash-timeout-us property for flash\n");
goto out_err;
}
rval = fwnode_property_read_u32(flash->flash_node, "flash-max-microamp",
&cfg->flash_max_ua);
if (rval < 0) {
dev_err(&flash->client->dev,
"can't read flash-max-microamp property for flash\n");
goto out_err;
}
rval = fwnode_property_read_u32(flash->flash_node, "led-max-microamp",
&cfg->assist_max_ua);
if (rval < 0) {
dev_err(&flash->client->dev,
"can't read led-max-microamp property for flash\n");
goto out_err;
}
fwnode_property_read_u32(flash->flash_node, "voltage-reference",
&cfg->voltage_reference);
fwnode_property_read_u32(flash->flash_node, "ams,input-max-microamp",
&cfg->peak);
cfg->peak = AS_PEAK_mA_TO_REG(cfg->peak);
if (!flash->indicator_node) {
dev_warn(&flash->client->dev,
"can't find indicator node\n");
rval = -ENODEV;
goto out_err;
}
rval = fwnode_property_read_u32(flash->indicator_node,
"led-max-microamp",
&cfg->indicator_max_ua);
if (rval < 0) {
dev_err(&flash->client->dev,
"can't read led-max-microamp property for indicator\n");
goto out_err;
}
return 0;
out_err:
fwnode_handle_put(flash->flash_node);
fwnode_handle_put(flash->indicator_node);
return rval;
}
static int as3645a_led_class_setup(struct as3645a *flash)
{
struct led_classdev *fled_cdev = &flash->fled.led_cdev;
struct led_classdev *iled_cdev = &flash->iled_cdev;
struct led_init_data init_data = {};
struct led_flash_setting *cfg;
int rval;
iled_cdev->brightness_set_blocking = as3645a_set_indicator_brightness;
iled_cdev->max_brightness =
flash->cfg.indicator_max_ua / AS_INDICATOR_INTENSITY_STEP;
iled_cdev->flags = LED_CORE_SUSPENDRESUME;
init_data.fwnode = flash->indicator_node;
init_data.devicename = AS_NAME;
init_data.default_label = "indicator";
rval = led_classdev_register_ext(&flash->client->dev, iled_cdev,
&init_data);
if (rval < 0)
return rval;
cfg = &flash->fled.brightness;
cfg->min = AS_FLASH_INTENSITY_MIN;
cfg->max = flash->cfg.flash_max_ua;
cfg->step = AS_FLASH_INTENSITY_STEP;
cfg->val = flash->cfg.flash_max_ua;
cfg = &flash->fled.timeout;
cfg->min = AS_FLASH_TIMEOUT_MIN;
cfg->max = flash->cfg.flash_timeout_us;
cfg->step = AS_FLASH_TIMEOUT_STEP;
cfg->val = flash->cfg.flash_timeout_us;
flash->fled.ops = &as3645a_led_flash_ops;
fled_cdev->brightness_set_blocking = as3645a_set_assist_brightness;
fled_cdev->max_brightness =
as3645a_current_to_reg(flash, false,
flash->cfg.assist_max_ua) + 1;
fled_cdev->flags = LED_DEV_CAP_FLASH | LED_CORE_SUSPENDRESUME;
init_data.fwnode = flash->flash_node;
init_data.devicename = AS_NAME;
init_data.default_label = "flash";
rval = led_classdev_flash_register_ext(&flash->client->dev,
&flash->fled, &init_data);
if (rval)
goto out_err;
return rval;
out_err:
led_classdev_unregister(iled_cdev);
dev_err(&flash->client->dev,
"led_classdev_flash_register() failed, error %d\n",
rval);
return rval;
}
static int as3645a_v4l2_setup(struct as3645a *flash)
{
struct led_classdev_flash *fled = &flash->fled;
struct led_classdev *led = &fled->led_cdev;
struct v4l2_flash_config cfg = {
.intensity = {
.min = AS_TORCH_INTENSITY_MIN,
.max = flash->cfg.assist_max_ua,
.step = AS_TORCH_INTENSITY_STEP,
.val = flash->cfg.assist_max_ua,
},
};
struct v4l2_flash_config cfgind = {
.intensity = {
.min = AS_INDICATOR_INTENSITY_MIN,
.max = flash->cfg.indicator_max_ua,
.step = AS_INDICATOR_INTENSITY_STEP,
.val = flash->cfg.indicator_max_ua,
},
};
strscpy(cfg.dev_name, led->dev->kobj.name, sizeof(cfg.dev_name));
strscpy(cfgind.dev_name, flash->iled_cdev.dev->kobj.name,
sizeof(cfgind.dev_name));
flash->vf = v4l2_flash_init(
&flash->client->dev, flash->flash_node, &flash->fled, NULL,
&cfg);
if (IS_ERR(flash->vf))
return PTR_ERR(flash->vf);
flash->vfind = v4l2_flash_indicator_init(
&flash->client->dev, flash->indicator_node, &flash->iled_cdev,
&cfgind);
if (IS_ERR(flash->vfind)) {
v4l2_flash_release(flash->vf);
return PTR_ERR(flash->vfind);
}
return 0;
}
static int as3645a_probe(struct i2c_client *client)
{
struct as3645a *flash;
int rval;
if (!dev_fwnode(&client->dev))
return -ENODEV;
flash = devm_kzalloc(&client->dev, sizeof(*flash), GFP_KERNEL);
if (flash == NULL)
return -ENOMEM;
flash->client = client;
rval = as3645a_parse_node(&client->dev, flash);
if (rval < 0)
return rval;
rval = as3645a_detect(flash);
if (rval < 0)
goto out_put_nodes;
mutex_init(&flash->mutex);
i2c_set_clientdata(client, flash);
rval = as3645a_setup(flash);
if (rval)
goto out_mutex_destroy;
rval = as3645a_led_class_setup(flash);
if (rval)
goto out_mutex_destroy;
rval = as3645a_v4l2_setup(flash);
if (rval)
goto out_led_classdev_flash_unregister;
return 0;
out_led_classdev_flash_unregister:
led_classdev_flash_unregister(&flash->fled);
out_mutex_destroy:
mutex_destroy(&flash->mutex);
out_put_nodes:
fwnode_handle_put(flash->flash_node);
fwnode_handle_put(flash->indicator_node);
return rval;
}
static void as3645a_remove(struct i2c_client *client)
{
struct as3645a *flash = i2c_get_clientdata(client);
as3645a_set_control(flash, AS_MODE_EXT_TORCH, false);
v4l2_flash_release(flash->vf);
v4l2_flash_release(flash->vfind);
led_classdev_flash_unregister(&flash->fled);
led_classdev_unregister(&flash->iled_cdev);
mutex_destroy(&flash->mutex);
fwnode_handle_put(flash->flash_node);
fwnode_handle_put(flash->indicator_node);
}
static const struct i2c_device_id as3645a_id_table[] = {
{ AS_NAME },
{ }
};
MODULE_DEVICE_TABLE(i2c, as3645a_id_table);
static const struct of_device_id as3645a_of_table[] = {
{ .compatible = "ams,as3645a" },
{ },
};
MODULE_DEVICE_TABLE(of, as3645a_of_table);
static struct i2c_driver as3645a_i2c_driver = {
.driver = {
.of_match_table = as3645a_of_table,
.name = AS_NAME,
},
.probe = as3645a_probe,
.remove = as3645a_remove,
.id_table = as3645a_id_table,
};
module_i2c_driver(as3645a_i2c_driver);
MODULE_AUTHOR("Laurent Pinchart <laurent.pinchart@ideasonboard.com>");
MODULE_AUTHOR("Sakari Ailus <sakari.ailus@iki.fi>");
MODULE_DESCRIPTION("LED flash driver for AS3645A, LM3555 and their clones");
MODULE_LICENSE("GPL v2");
