[GitHub/LineageOS/android_kernel_motorola_exynos9610.git] / drivers / input / rmi4 / rmi_f54.c

/*
 * Copyright (c) 2012-2015 Synaptics Incorporated
 * Copyright (C) 2016 Zodiac Inflight Innovations
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 as published by
 * the Free Software Foundation.
 */

#include <linux/kernel.h>
#include <linux/rmi.h>
#include <linux/input.h>
#include <linux/slab.h>
#include <linux/delay.h>
#include <linux/i2c.h>
#include <media/v4l2-device.h>
#include <media/v4l2-ioctl.h>
#include <media/videobuf2-v4l2.h>
#include <media/videobuf2-vmalloc.h>
#include "rmi_driver.h"

#define F54_NAME		"rmi4_f54"

/* F54 data offsets */
#define F54_REPORT_DATA_OFFSET  3
#define F54_FIFO_OFFSET         1
#define F54_NUM_TX_OFFSET       1
#define F54_NUM_RX_OFFSET       0

/* F54 commands */
#define F54_GET_REPORT          1
#define F54_FORCE_CAL           2

/* F54 capabilities */
#define F54_CAP_BASELINE	(1 << 2)
#define F54_CAP_IMAGE8		(1 << 3)
#define F54_CAP_IMAGE16		(1 << 6)

/**
 * enum rmi_f54_report_type - RMI4 F54 report types
 *
 * @F54_8BIT_IMAGE:	Normalized 8-Bit Image Report. The capacitance variance
 *			from baseline for each pixel.
 *
 * @F54_16BIT_IMAGE:	Normalized 16-Bit Image Report. The capacitance variance
 *			from baseline for each pixel.
 *
 * @F54_RAW_16BIT_IMAGE:
 *			Raw 16-Bit Image Report. The raw capacitance for each
 *			pixel.
 *
 * @F54_TRUE_BASELINE:	True Baseline Report. The baseline capacitance for each
 *			pixel.
 *
 * @F54_FULL_RAW_CAP:   Full Raw Capacitance Report. The raw capacitance with
 *			low reference set to its minimum value and high
 *			reference set to its maximum value.
 *
 * @F54_FULL_RAW_CAP_RX_OFFSET_REMOVED:
 *			Full Raw Capacitance with Receiver Offset Removed
 *			Report. Set Low reference to its minimum value and high
 *			references to its maximum value, then report the raw
 *			capacitance for each pixel.
 */
enum rmi_f54_report_type {
	F54_REPORT_NONE = 0,
	F54_8BIT_IMAGE = 1,
	F54_16BIT_IMAGE = 2,
	F54_RAW_16BIT_IMAGE = 3,
	F54_TRUE_BASELINE = 9,
	F54_FULL_RAW_CAP = 19,
	F54_FULL_RAW_CAP_RX_OFFSET_REMOVED = 20,
	F54_MAX_REPORT_TYPE,
};

const char *rmi_f54_report_type_names[] = {
	[F54_REPORT_NONE]		= "Unknown",
	[F54_8BIT_IMAGE]		= "Normalized 8-Bit Image",
	[F54_16BIT_IMAGE]		= "Normalized 16-Bit Image",
	[F54_RAW_16BIT_IMAGE]		= "Raw 16-Bit Image",
	[F54_TRUE_BASELINE]		= "True Baseline",
	[F54_FULL_RAW_CAP]		= "Full Raw Capacitance",
	[F54_FULL_RAW_CAP_RX_OFFSET_REMOVED]
					= "Full Raw Capacitance RX Offset Removed",
};

struct rmi_f54_reports {
	int start;
	int size;
};

struct f54_data {
	struct rmi_function *fn;

	u8 num_rx_electrodes;
	u8 num_tx_electrodes;
	u8 capabilities;
	u16 clock_rate;
	u8 family;

	enum rmi_f54_report_type report_type;
	u8 *report_data;
	int report_size;
	struct rmi_f54_reports standard_report[2];

	bool is_busy;
	struct mutex status_mutex;
	struct mutex data_mutex;

	struct workqueue_struct *workqueue;
	struct delayed_work work;
	unsigned long timeout;

	struct completion cmd_done;

	/* V4L2 support */
	struct v4l2_device v4l2;
	struct v4l2_pix_format format;
	struct video_device vdev;
	struct vb2_queue queue;
	struct mutex lock;
	int input;
	enum rmi_f54_report_type inputs[F54_MAX_REPORT_TYPE];
};

/*
 * Basic checks on report_type to ensure we write a valid type
 * to the sensor.
 */
static bool is_f54_report_type_valid(struct f54_data *f54,
				     enum rmi_f54_report_type reptype)
{
	switch (reptype) {
	case F54_8BIT_IMAGE:
		return f54->capabilities & F54_CAP_IMAGE8;
	case F54_16BIT_IMAGE:
	case F54_RAW_16BIT_IMAGE:
		return f54->capabilities & F54_CAP_IMAGE16;
	case F54_TRUE_BASELINE:
		return f54->capabilities & F54_CAP_IMAGE16;
	case F54_FULL_RAW_CAP:
	case F54_FULL_RAW_CAP_RX_OFFSET_REMOVED:
		return true;
	default:
		return false;
	}
}

static enum rmi_f54_report_type rmi_f54_get_reptype(struct f54_data *f54,
						unsigned int i)
{
	if (i >= F54_MAX_REPORT_TYPE)
		return F54_REPORT_NONE;

	return f54->inputs[i];
}

static void rmi_f54_create_input_map(struct f54_data *f54)
{
	int i = 0;
	enum rmi_f54_report_type reptype;

	for (reptype = 1; reptype < F54_MAX_REPORT_TYPE; reptype++) {
		if (!is_f54_report_type_valid(f54, reptype))
			continue;

		f54->inputs[i++] = reptype;
	}

	/* Remaining values are zero via kzalloc */
}

static int rmi_f54_request_report(struct rmi_function *fn, u8 report_type)
{
	struct f54_data *f54 = dev_get_drvdata(&fn->dev);
	struct rmi_device *rmi_dev = fn->rmi_dev;
	int error;

	/* Write Report Type into F54_AD_Data0 */
	if (f54->report_type != report_type) {
		error = rmi_write(rmi_dev, f54->fn->fd.data_base_addr,
				  report_type);
		if (error)
			return error;
		f54->report_type = report_type;
	}

	/*
	 * Small delay after disabling interrupts to avoid race condition
	 * in firmare. This value is a bit higher than absolutely necessary.
	 * Should be removed once issue is resolved in firmware.
	 */
	usleep_range(2000, 3000);

	mutex_lock(&f54->data_mutex);

	error = rmi_write(rmi_dev, fn->fd.command_base_addr, F54_GET_REPORT);
	if (error < 0)
		goto unlock;

	init_completion(&f54->cmd_done);

	f54->is_busy = 1;
	f54->timeout = jiffies + msecs_to_jiffies(100);

	queue_delayed_work(f54->workqueue, &f54->work, 0);

unlock:
	mutex_unlock(&f54->data_mutex);

	return error;
}

static size_t rmi_f54_get_report_size(struct f54_data *f54)
{
	struct rmi_device *rmi_dev = f54->fn->rmi_dev;
	struct rmi_driver_data *drv_data = dev_get_drvdata(&rmi_dev->dev);
	u8 rx = drv_data->num_rx_electrodes ? : f54->num_rx_electrodes;
	u8 tx = drv_data->num_tx_electrodes ? : f54->num_tx_electrodes;
	size_t size;

	switch (rmi_f54_get_reptype(f54, f54->input)) {
	case F54_8BIT_IMAGE:
		size = rx * tx;
		break;
	case F54_16BIT_IMAGE:
	case F54_RAW_16BIT_IMAGE:
	case F54_TRUE_BASELINE:
	case F54_FULL_RAW_CAP:
	case F54_FULL_RAW_CAP_RX_OFFSET_REMOVED:
		size = sizeof(u16) * rx * tx;
		break;
	default:
		size = 0;
	}

	return size;
}

static int rmi_f54_get_pixel_fmt(enum rmi_f54_report_type reptype, u32 *pixfmt)
{
	int ret = 0;

	switch (reptype) {
	case F54_8BIT_IMAGE:
		*pixfmt = V4L2_TCH_FMT_DELTA_TD08;
		break;

	case F54_16BIT_IMAGE:
		*pixfmt = V4L2_TCH_FMT_DELTA_TD16;
		break;

	case F54_RAW_16BIT_IMAGE:
	case F54_TRUE_BASELINE:
	case F54_FULL_RAW_CAP:
	case F54_FULL_RAW_CAP_RX_OFFSET_REMOVED:
		*pixfmt = V4L2_TCH_FMT_TU16;
		break;

	case F54_REPORT_NONE:
	case F54_MAX_REPORT_TYPE:
		ret = -EINVAL;
		break;
	}

	return ret;
}

static const struct v4l2_file_operations rmi_f54_video_fops = {
	.owner = THIS_MODULE,
	.open = v4l2_fh_open,
	.release = vb2_fop_release,
	.unlocked_ioctl = video_ioctl2,
	.read = vb2_fop_read,
	.mmap = vb2_fop_mmap,
	.poll = vb2_fop_poll,
};

static int rmi_f54_queue_setup(struct vb2_queue *q, unsigned int *nbuffers,
			       unsigned int *nplanes, unsigned int sizes[],
			       struct device *alloc_devs[])
{
	struct f54_data *f54 = q->drv_priv;

	if (*nplanes)
		return sizes[0] < rmi_f54_get_report_size(f54) ? -EINVAL : 0;

	*nplanes = 1;
	sizes[0] = rmi_f54_get_report_size(f54);

	return 0;
}

static void rmi_f54_buffer_queue(struct vb2_buffer *vb)
{
	struct f54_data *f54 = vb2_get_drv_priv(vb->vb2_queue);
	u16 *ptr;
	enum vb2_buffer_state state;
	enum rmi_f54_report_type reptype;
	int ret;

	mutex_lock(&f54->status_mutex);

	reptype = rmi_f54_get_reptype(f54, f54->input);
	if (reptype == F54_REPORT_NONE) {
		state = VB2_BUF_STATE_ERROR;
		goto done;
	}

	if (f54->is_busy) {
		state = VB2_BUF_STATE_ERROR;
		goto done;
	}

	ret = rmi_f54_request_report(f54->fn, reptype);
	if (ret) {
		dev_err(&f54->fn->dev, "Error requesting F54 report\n");
		state = VB2_BUF_STATE_ERROR;
		goto done;
	}

	/* get frame data */
	mutex_lock(&f54->data_mutex);

	while (f54->is_busy) {
		mutex_unlock(&f54->data_mutex);
		if (!wait_for_completion_timeout(&f54->cmd_done,
						 msecs_to_jiffies(1000))) {
			dev_err(&f54->fn->dev, "Timed out\n");
			state = VB2_BUF_STATE_ERROR;
			goto done;
		}
		mutex_lock(&f54->data_mutex);
	}

	ptr = vb2_plane_vaddr(vb, 0);
	if (!ptr) {
		dev_err(&f54->fn->dev, "Error acquiring frame ptr\n");
		state = VB2_BUF_STATE_ERROR;
		goto data_done;
	}

	memcpy(ptr, f54->report_data, f54->report_size);
	vb2_set_plane_payload(vb, 0, rmi_f54_get_report_size(f54));
	state = VB2_BUF_STATE_DONE;

data_done:
	mutex_unlock(&f54->data_mutex);
done:
	vb2_buffer_done(vb, state);
	mutex_unlock(&f54->status_mutex);
}

/* V4L2 structures */
static const struct vb2_ops rmi_f54_queue_ops = {
	.queue_setup            = rmi_f54_queue_setup,
	.buf_queue              = rmi_f54_buffer_queue,
	.wait_prepare           = vb2_ops_wait_prepare,
	.wait_finish            = vb2_ops_wait_finish,
};

static const struct vb2_queue rmi_f54_queue = {
	.type = V4L2_BUF_TYPE_VIDEO_CAPTURE,
	.io_modes = VB2_MMAP | VB2_USERPTR | VB2_DMABUF | VB2_READ,
	.buf_struct_size = sizeof(struct vb2_buffer),
	.ops = &rmi_f54_queue_ops,
	.mem_ops = &vb2_vmalloc_memops,
	.timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC,
	.min_buffers_needed = 1,
};

static int rmi_f54_vidioc_querycap(struct file *file, void *priv,
				   struct v4l2_capability *cap)
{
	struct f54_data *f54 = video_drvdata(file);

	strlcpy(cap->driver, F54_NAME, sizeof(cap->driver));
	strlcpy(cap->card, SYNAPTICS_INPUT_DEVICE_NAME, sizeof(cap->card));
	snprintf(cap->bus_info, sizeof(cap->bus_info),
		"rmi4:%s", dev_name(&f54->fn->dev));

	return 0;
}

static int rmi_f54_vidioc_enum_input(struct file *file, void *priv,
				     struct v4l2_input *i)
{
	struct f54_data *f54 = video_drvdata(file);
	enum rmi_f54_report_type reptype;

	reptype = rmi_f54_get_reptype(f54, i->index);
	if (reptype == F54_REPORT_NONE)
		return -EINVAL;

	i->type = V4L2_INPUT_TYPE_TOUCH;

	strlcpy(i->name, rmi_f54_report_type_names[reptype], sizeof(i->name));
	return 0;
}

static int rmi_f54_set_input(struct f54_data *f54, unsigned int i)
{
	struct rmi_device *rmi_dev = f54->fn->rmi_dev;
	struct rmi_driver_data *drv_data = dev_get_drvdata(&rmi_dev->dev);
	u8 rx = drv_data->num_rx_electrodes ? : f54->num_rx_electrodes;
	u8 tx = drv_data->num_tx_electrodes ? : f54->num_tx_electrodes;
	struct v4l2_pix_format *f = &f54->format;
	enum rmi_f54_report_type reptype;
	int ret;

	reptype = rmi_f54_get_reptype(f54, i);
	if (reptype == F54_REPORT_NONE)
		return -EINVAL;

	ret = rmi_f54_get_pixel_fmt(reptype, &f->pixelformat);
	if (ret)
		return ret;

	f54->input = i;

	f->width = rx;
	f->height = tx;
	f->field = V4L2_FIELD_NONE;
	f->colorspace = V4L2_COLORSPACE_RAW;
	f->bytesperline = f->width * sizeof(u16);
	f->sizeimage = f->width * f->height * sizeof(u16);

	return 0;
}

static int rmi_f54_vidioc_s_input(struct file *file, void *priv, unsigned int i)
{
	return rmi_f54_set_input(video_drvdata(file), i);
}

static int rmi_f54_vidioc_g_input(struct file *file, void *priv,
				  unsigned int *i)
{
	struct f54_data *f54 = video_drvdata(file);

	*i = f54->input;

	return 0;
}

static int rmi_f54_vidioc_fmt(struct file *file, void *priv,
			      struct v4l2_format *f)
{
	struct f54_data *f54 = video_drvdata(file);

	f->fmt.pix = f54->format;

	return 0;
}

static int rmi_f54_vidioc_enum_fmt(struct file *file, void *priv,
				   struct v4l2_fmtdesc *fmt)
{
	if (fmt->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
		return -EINVAL;

	switch (fmt->index) {
	case 0:
		fmt->pixelformat = V4L2_TCH_FMT_DELTA_TD16;
		break;

	case 1:
		fmt->pixelformat = V4L2_TCH_FMT_DELTA_TD08;
		break;

	case 2:
		fmt->pixelformat = V4L2_TCH_FMT_TU16;
		break;

	default:
		return -EINVAL;
	}

	return 0;
}

static int rmi_f54_vidioc_g_parm(struct file *file, void *fh,
				 struct v4l2_streamparm *a)
{
	if (a->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
		return -EINVAL;

	a->parm.capture.readbuffers = 1;
	a->parm.capture.timeperframe.numerator = 1;
	a->parm.capture.timeperframe.denominator = 10;
	return 0;
}

static const struct v4l2_ioctl_ops rmi_f54_video_ioctl_ops = {
	.vidioc_querycap	= rmi_f54_vidioc_querycap,

	.vidioc_enum_fmt_vid_cap = rmi_f54_vidioc_enum_fmt,
	.vidioc_s_fmt_vid_cap	= rmi_f54_vidioc_fmt,
	.vidioc_g_fmt_vid_cap	= rmi_f54_vidioc_fmt,
	.vidioc_try_fmt_vid_cap	= rmi_f54_vidioc_fmt,
	.vidioc_g_parm		= rmi_f54_vidioc_g_parm,

	.vidioc_enum_input	= rmi_f54_vidioc_enum_input,
	.vidioc_g_input		= rmi_f54_vidioc_g_input,
	.vidioc_s_input		= rmi_f54_vidioc_s_input,

	.vidioc_reqbufs		= vb2_ioctl_reqbufs,
	.vidioc_create_bufs	= vb2_ioctl_create_bufs,
	.vidioc_querybuf	= vb2_ioctl_querybuf,
	.vidioc_qbuf		= vb2_ioctl_qbuf,
	.vidioc_dqbuf		= vb2_ioctl_dqbuf,
	.vidioc_expbuf		= vb2_ioctl_expbuf,

	.vidioc_streamon	= vb2_ioctl_streamon,
	.vidioc_streamoff	= vb2_ioctl_streamoff,
};

static const struct video_device rmi_f54_video_device = {
	.name = "Synaptics RMI4",
	.fops = &rmi_f54_video_fops,
	.ioctl_ops = &rmi_f54_video_ioctl_ops,
	.release = video_device_release_empty,
	.device_caps = V4L2_CAP_VIDEO_CAPTURE | V4L2_CAP_TOUCH |
		       V4L2_CAP_READWRITE | V4L2_CAP_STREAMING,
};

static void rmi_f54_work(struct work_struct *work)
{
	struct f54_data *f54 = container_of(work, struct f54_data, work.work);
	struct rmi_function *fn = f54->fn;
	u8 fifo[2];
	struct rmi_f54_reports *report;
	int report_size;
	u8 command;
	u8 *data;
	int error;

	data = f54->report_data;
	report_size = rmi_f54_get_report_size(f54);
	if (report_size == 0) {
		dev_err(&fn->dev, "Bad report size, report type=%d\n",
				f54->report_type);
		error = -EINVAL;
		goto error;     /* retry won't help */
	}
	f54->standard_report[0].size = report_size;
	report = f54->standard_report;

	mutex_lock(&f54->data_mutex);

	/*
	 * Need to check if command has completed.
	 * If not try again later.
	 */
	error = rmi_read(fn->rmi_dev, f54->fn->fd.command_base_addr,
			 &command);
	if (error) {
		dev_err(&fn->dev, "Failed to read back command\n");
		goto error;
	}
	if (command & F54_GET_REPORT) {
		if (time_after(jiffies, f54->timeout)) {
			dev_err(&fn->dev, "Get report command timed out\n");
			error = -ETIMEDOUT;
		}
		report_size = 0;
		goto error;
	}

	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "Get report command completed, reading data\n");

	report_size = 0;
	for (; report->size; report++) {
		fifo[0] = report->start & 0xff;
		fifo[1] = (report->start >> 8) & 0xff;
		error = rmi_write_block(fn->rmi_dev,
					fn->fd.data_base_addr + F54_FIFO_OFFSET,
					fifo, sizeof(fifo));
		if (error) {
			dev_err(&fn->dev, "Failed to set fifo start offset\n");
			goto abort;
		}

		error = rmi_read_block(fn->rmi_dev, fn->fd.data_base_addr +
				       F54_REPORT_DATA_OFFSET, data,
				       report->size);
		if (error) {
			dev_err(&fn->dev, "%s: read [%d bytes] returned %d\n",
				__func__, report->size, error);
			goto abort;
		}
		data += report->size;
		report_size += report->size;
	}

abort:
	f54->report_size = error ? 0 : report_size;
error:
	if (error)
		report_size = 0;

	if (report_size == 0 && !error) {
		queue_delayed_work(f54->workqueue, &f54->work,
				   msecs_to_jiffies(1));
	} else {
		f54->is_busy = false;
		complete(&f54->cmd_done);
	}

	mutex_unlock(&f54->data_mutex);
}

static int rmi_f54_attention(struct rmi_function *fn, unsigned long *irqbits)
{
	return 0;
}

static int rmi_f54_config(struct rmi_function *fn)
{
	struct rmi_driver *drv = fn->rmi_dev->driver;

	drv->set_irq_bits(fn->rmi_dev, fn->irq_mask);

	return 0;
}

static int rmi_f54_detect(struct rmi_function *fn)
{
	int error;
	struct f54_data *f54;
	u8 buf[6];

	f54 = dev_get_drvdata(&fn->dev);

	error = rmi_read_block(fn->rmi_dev, fn->fd.query_base_addr,
			       buf, sizeof(buf));
	if (error) {
		dev_err(&fn->dev, "%s: Failed to query F54 properties\n",
			__func__);
		return error;
	}

	f54->num_rx_electrodes = buf[0];
	f54->num_tx_electrodes = buf[1];
	f54->capabilities = buf[2];
	f54->clock_rate = buf[3] | (buf[4] << 8);
	f54->family = buf[5];

	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 num_rx_electrodes: %d\n",
		f54->num_rx_electrodes);
	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 num_tx_electrodes: %d\n",
		f54->num_tx_electrodes);
	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 capabilities: 0x%x\n",
		f54->capabilities);
	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 clock rate: 0x%x\n",
		f54->clock_rate);
	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 family: 0x%x\n",
		f54->family);

	f54->is_busy = false;

	return 0;
}

static int rmi_f54_probe(struct rmi_function *fn)
{
	struct f54_data *f54;
	int ret;
	u8 rx, tx;

	f54 = devm_kzalloc(&fn->dev, sizeof(struct f54_data), GFP_KERNEL);
	if (!f54)
		return -ENOMEM;

	f54->fn = fn;
	dev_set_drvdata(&fn->dev, f54);

	ret = rmi_f54_detect(fn);
	if (ret)
		return ret;

	mutex_init(&f54->data_mutex);
	mutex_init(&f54->status_mutex);

	rx = f54->num_rx_electrodes;
	tx = f54->num_tx_electrodes;
	f54->report_data = devm_kzalloc(&fn->dev,
					sizeof(u16) * tx * rx,
					GFP_KERNEL);
	if (f54->report_data == NULL)
		return -ENOMEM;

	INIT_DELAYED_WORK(&f54->work, rmi_f54_work);

	f54->workqueue = create_singlethread_workqueue("rmi4-poller");
	if (!f54->workqueue)
		return -ENOMEM;

	rmi_f54_create_input_map(f54);

	/* register video device */
	strlcpy(f54->v4l2.name, F54_NAME, sizeof(f54->v4l2.name));
	ret = v4l2_device_register(&fn->dev, &f54->v4l2);
	if (ret) {
		dev_err(&fn->dev, "Unable to register video dev.\n");
		goto remove_wq;
	}

	/* initialize the queue */
	mutex_init(&f54->lock);
	f54->queue = rmi_f54_queue;
	f54->queue.drv_priv = f54;
	f54->queue.lock = &f54->lock;
	f54->queue.dev = &fn->dev;

	ret = vb2_queue_init(&f54->queue);
	if (ret)
		goto remove_v4l2;

	f54->vdev = rmi_f54_video_device;
	f54->vdev.v4l2_dev = &f54->v4l2;
	f54->vdev.lock = &f54->lock;
	f54->vdev.vfl_dir = VFL_DIR_RX;
	f54->vdev.queue = &f54->queue;
	video_set_drvdata(&f54->vdev, f54);

	ret = video_register_device(&f54->vdev, VFL_TYPE_TOUCH, -1);
	if (ret) {
		dev_err(&fn->dev, "Unable to register video subdevice.");
		goto remove_v4l2;
	}

	return 0;

remove_v4l2:
	v4l2_device_unregister(&f54->v4l2);
remove_wq:
	cancel_delayed_work_sync(&f54->work);
	flush_workqueue(f54->workqueue);
	destroy_workqueue(f54->workqueue);
	return ret;
}

static void rmi_f54_remove(struct rmi_function *fn)
{
	struct f54_data *f54 = dev_get_drvdata(&fn->dev);

	video_unregister_device(&f54->vdev);
	v4l2_device_unregister(&f54->v4l2);
}

struct rmi_function_handler rmi_f54_handler = {
	.driver = {
		.name = F54_NAME,
	},
	.func = 0x54,
	.probe = rmi_f54_probe,
	.config = rmi_f54_config,
	.attention = rmi_f54_attention,
	.remove = rmi_f54_remove,
};
Commit	Line	Data
3a762dbd ND	1	/*
3a762dbd ND	2	* Copyright (c) 2012-2015 Synaptics Incorporated
f3c4a8f8	3	* Copyright (C) 2016 Zodiac Inflight Innovations
3a762dbd ND	4	*
	5	* This program is free software; you can redistribute it and/or modify it
	6	* under the terms of the GNU General Public License version 2 as published by
	7	* the Free Software Foundation.
	8	*/
	9
	10	#include <linux/kernel.h>
	11	#include <linux/rmi.h>
	12	#include <linux/input.h>
	13	#include <linux/slab.h>
	14	#include <linux/delay.h>
	15	#include <linux/i2c.h>
	16	#include <media/v4l2-device.h>
	17	#include <media/v4l2-ioctl.h>
	18	#include <media/videobuf2-v4l2.h>
	19	#include <media/videobuf2-vmalloc.h>
	20	#include "rmi_driver.h"
	21
	22	#define F54_NAME "rmi4_f54"
	23
	24	/* F54 data offsets */
	25	#define F54_REPORT_DATA_OFFSET 3
	26	#define F54_FIFO_OFFSET 1
	27	#define F54_NUM_TX_OFFSET 1
	28	#define F54_NUM_RX_OFFSET 0
	29
	30	/* F54 commands */
	31	#define F54_GET_REPORT 1
	32	#define F54_FORCE_CAL 2
	33
3a762dbd ND	34	/* F54 capabilities */
	35	#define F54_CAP_BASELINE (1 << 2)
	36	#define F54_CAP_IMAGE8 (1 << 3)
	37	#define F54_CAP_IMAGE16 (1 << 6)
	38
	39	/**
	40	* enum rmi_f54_report_type - RMI4 F54 report types
	41	*
	42	* @F54_8BIT_IMAGE: Normalized 8-Bit Image Report. The capacitance variance
	43	* from baseline for each pixel.
	44	*
	45	* @F54_16BIT_IMAGE: Normalized 16-Bit Image Report. The capacitance variance
	46	* from baseline for each pixel.
	47	*
	48	* @F54_RAW_16BIT_IMAGE:
	49	* Raw 16-Bit Image Report. The raw capacitance for each
	50	* pixel.
	51	*
	52	* @F54_TRUE_BASELINE: True Baseline Report. The baseline capacitance for each
	53	* pixel.
	54	*
	55	* @F54_FULL_RAW_CAP: Full Raw Capacitance Report. The raw capacitance with
	56	* low reference set to its minimum value and high
	57	* reference set to its maximum value.
	58	*
	59	* @F54_FULL_RAW_CAP_RX_OFFSET_REMOVED:
	60	* Full Raw Capacitance with Receiver Offset Removed
	61	* Report. Set Low reference to its minimum value and high
	62	* references to its maximum value, then report the raw
	63	* capacitance for each pixel.
	64	*/
	65	enum rmi_f54_report_type {
	66	F54_REPORT_NONE = 0,
	67	F54_8BIT_IMAGE = 1,
	68	F54_16BIT_IMAGE = 2,
	69	F54_RAW_16BIT_IMAGE = 3,
	70	F54_TRUE_BASELINE = 9,
	71	F54_FULL_RAW_CAP = 19,
	72	F54_FULL_RAW_CAP_RX_OFFSET_REMOVED = 20,
	73	F54_MAX_REPORT_TYPE,
	74	};
	75
	76	const char *rmi_f54_report_type_names[] = {
	77	[F54_REPORT_NONE] = "Unknown",
	78	[F54_8BIT_IMAGE] = "Normalized 8-Bit Image",
	79	[F54_16BIT_IMAGE] = "Normalized 16-Bit Image",
	80	[F54_RAW_16BIT_IMAGE] = "Raw 16-Bit Image",
	81	[F54_TRUE_BASELINE] = "True Baseline",
	82	[F54_FULL_RAW_CAP] = "Full Raw Capacitance",
	83	[F54_FULL_RAW_CAP_RX_OFFSET_REMOVED]
	84	= "Full Raw Capacitance RX Offset Removed",
	85	};
	86
	87	struct rmi_f54_reports {
	88	int start;
	89	int size;
	90	};
	91
	92	struct f54_data {
	93	struct rmi_function *fn;
	94
3a762dbd ND	95	u8 num_rx_electrodes;
	96	u8 num_tx_electrodes;
	97	u8 capabilities;
	98	u16 clock_rate;
	99	u8 family;
	100
	101	enum rmi_f54_report_type report_type;
	102	u8 *report_data;
	103	int report_size;
	104	struct rmi_f54_reports standard_report[2];
	105
	106	bool is_busy;
	107	struct mutex status_mutex;
	108	struct mutex data_mutex;
	109
	110	struct workqueue_struct *workqueue;
	111	struct delayed_work work;
	112	unsigned long timeout;
	113
	114	struct completion cmd_done;
	115
	116	/* V4L2 support */
	117	struct v4l2_device v4l2;
	118	struct v4l2_pix_format format;
	119	struct video_device vdev;
	120	struct vb2_queue queue;
	121	struct mutex lock;
	122	int input;
	123	enum rmi_f54_report_type inputs[F54_MAX_REPORT_TYPE];
	124	};
	125
	126	/*
	127	* Basic checks on report_type to ensure we write a valid type
	128	* to the sensor.
	129	*/
	130	static bool is_f54_report_type_valid(struct f54_data *f54,
	131	enum rmi_f54_report_type reptype)
	132	{
	133	switch (reptype) {
	134	case F54_8BIT_IMAGE:
	135	return f54->capabilities & F54_CAP_IMAGE8;
	136	case F54_16BIT_IMAGE:
	137	case F54_RAW_16BIT_IMAGE:
	138	return f54->capabilities & F54_CAP_IMAGE16;
	139	case F54_TRUE_BASELINE:
	140	return f54->capabilities & F54_CAP_IMAGE16;
	141	case F54_FULL_RAW_CAP:
	142	case F54_FULL_RAW_CAP_RX_OFFSET_REMOVED:
	143	return true;
	144	default:
	145	return false;
	146	}
	147	}
	148
	149	static enum rmi_f54_report_type rmi_f54_get_reptype(struct f54_data *f54,
	150	unsigned int i)
	151	{
	152	if (i >= F54_MAX_REPORT_TYPE)
	153	return F54_REPORT_NONE;
	154
	155	return f54->inputs[i];
	156	}
	157
	158	static void rmi_f54_create_input_map(struct f54_data *f54)
159	{
160	int i = 0;
161	enum rmi_f54_report_type reptype;
162
163	for (reptype = 1; reptype < F54_MAX_REPORT_TYPE; reptype++) {
164	if (!is_f54_report_type_valid(f54, reptype))
165	continue;
166
167	f54->inputs[i++] = reptype;
168	}
169
170	/* Remaining values are zero via kzalloc */
171	}
172
173	static int rmi_f54_request_report(struct rmi_function *fn, u8 report_type)
174	{
175	struct f54_data *f54 = dev_get_drvdata(&fn->dev);
176	struct rmi_device *rmi_dev = fn->rmi_dev;
177	int error;
178
179	/* Write Report Type into F54_AD_Data0 */
180	if (f54->report_type != report_type) {
181	error = rmi_write(rmi_dev, f54->fn->fd.data_base_addr,
182	report_type);
183	if (error)
184	return error;
185	f54->report_type = report_type;
186	}
187
188	/*
189	* Small delay after disabling interrupts to avoid race condition
190	* in firmare. This value is a bit higher than absolutely necessary.
191	* Should be removed once issue is resolved in firmware.
192	*/
193	usleep_range(2000, 3000);
194
195	mutex_lock(&f54->data_mutex);
196
197	error = rmi_write(rmi_dev, fn->fd.command_base_addr, F54_GET_REPORT);
198	if (error < 0)
792f497b	199	goto unlock;
3a762dbd ND	200
	201	init_completion(&f54->cmd_done);
	202
	203	f54->is_busy = 1;
	204	f54->timeout = jiffies + msecs_to_jiffies(100);
	205
	206	queue_delayed_work(f54->workqueue, &f54->work, 0);
	207
792f497b	208	unlock:
3a762dbd ND	209	mutex_unlock(&f54->data_mutex);
3a762dbd ND	210
792f497b	211	return error;
3a762dbd ND	212	}
	213
	214	static size_t rmi_f54_get_report_size(struct f54_data *f54)
	215	{
c762cc68 GR	216	struct rmi_device *rmi_dev = f54->fn->rmi_dev;
	217	struct rmi_driver_data *drv_data = dev_get_drvdata(&rmi_dev->dev);
	218	u8 rx = drv_data->num_rx_electrodes ? : f54->num_rx_electrodes;
	219	u8 tx = drv_data->num_tx_electrodes ? : f54->num_tx_electrodes;
3a762dbd ND	220	size_t size;
	221
	222	switch (rmi_f54_get_reptype(f54, f54->input)) {
	223	case F54_8BIT_IMAGE:
	224	size = rx * tx;
	225	break;
	226	case F54_16BIT_IMAGE:
	227	case F54_RAW_16BIT_IMAGE:
	228	case F54_TRUE_BASELINE:
	229	case F54_FULL_RAW_CAP:
	230	case F54_FULL_RAW_CAP_RX_OFFSET_REMOVED:
	231	size = sizeof(u16) * rx * tx;
	232	break;
	233	default:
	234	size = 0;
	235	}
	236
	237	return size;
	238	}
	239
	240	static int rmi_f54_get_pixel_fmt(enum rmi_f54_report_type reptype, u32 *pixfmt)
	241	{
	242	int ret = 0;
	243
	244	switch (reptype) {
	245	case F54_8BIT_IMAGE:
	246	*pixfmt = V4L2_TCH_FMT_DELTA_TD08;
	247	break;
	248
	249	case F54_16BIT_IMAGE:
	250	*pixfmt = V4L2_TCH_FMT_DELTA_TD16;
	251	break;
	252
	253	case F54_RAW_16BIT_IMAGE:
	254	case F54_TRUE_BASELINE:
	255	case F54_FULL_RAW_CAP:
	256	case F54_FULL_RAW_CAP_RX_OFFSET_REMOVED:
	257	*pixfmt = V4L2_TCH_FMT_TU16;
	258	break;
	259
	260	case F54_REPORT_NONE:
	261	case F54_MAX_REPORT_TYPE:
	262	ret = -EINVAL;
	263	break;
	264	}
	265
	266	return ret;
	267	}
	268
	269	static const struct v4l2_file_operations rmi_f54_video_fops = {
	270	.owner = THIS_MODULE,
	271	.open = v4l2_fh_open,
	272	.release = vb2_fop_release,
	273	.unlocked_ioctl = video_ioctl2,
	274	.read = vb2_fop_read,
	275	.mmap = vb2_fop_mmap,
	276	.poll = vb2_fop_poll,
	277	};
	278
	279	static int rmi_f54_queue_setup(struct vb2_queue q, unsigned int nbuffers,
	280	unsigned int *nplanes, unsigned int sizes[],
	281	struct device *alloc_devs[])
	282	{
	283	struct f54_data *f54 = q->drv_priv;
284
285	if (*nplanes)
286	return sizes[0] < rmi_f54_get_report_size(f54) ? -EINVAL : 0;
287
288	*nplanes = 1;
289	sizes[0] = rmi_f54_get_report_size(f54);
290
291	return 0;
292	}
293
294	static void rmi_f54_buffer_queue(struct vb2_buffer *vb)
295	{
296	struct f54_data *f54 = vb2_get_drv_priv(vb->vb2_queue);
297	u16 *ptr;
298	enum vb2_buffer_state state;
299	enum rmi_f54_report_type reptype;
300	int ret;
301
302	mutex_lock(&f54->status_mutex);
303
304	reptype = rmi_f54_get_reptype(f54, f54->input);
305	if (reptype == F54_REPORT_NONE) {
306	state = VB2_BUF_STATE_ERROR;
307	goto done;
308	}
309
310	if (f54->is_busy) {
311	state = VB2_BUF_STATE_ERROR;
312	goto done;
313	}
314
315	ret = rmi_f54_request_report(f54->fn, reptype);
316	if (ret) {
317	dev_err(&f54->fn->dev, "Error requesting F54 report\n");
318	state = VB2_BUF_STATE_ERROR;
319	goto done;
320	}
321
322	/* get frame data */
323	mutex_lock(&f54->data_mutex);
324
325	while (f54->is_busy) {
326	mutex_unlock(&f54->data_mutex);
327	if (!wait_for_completion_timeout(&f54->cmd_done,
328	msecs_to_jiffies(1000))) {
329	dev_err(&f54->fn->dev, "Timed out\n");
330	state = VB2_BUF_STATE_ERROR;
331	goto done;
332	}
333	mutex_lock(&f54->data_mutex);
334	}
335
336	ptr = vb2_plane_vaddr(vb, 0);
337	if (!ptr) {
338	dev_err(&f54->fn->dev, "Error acquiring frame ptr\n");
339	state = VB2_BUF_STATE_ERROR;
340	goto data_done;
341	}
342
343	memcpy(ptr, f54->report_data, f54->report_size);
344	vb2_set_plane_payload(vb, 0, rmi_f54_get_report_size(f54));
345	state = VB2_BUF_STATE_DONE;
346
347	data_done:
348	mutex_unlock(&f54->data_mutex);
349	done:
350	vb2_buffer_done(vb, state);
351	mutex_unlock(&f54->status_mutex);
352	}
353
354	/* V4L2 structures */
355	static const struct vb2_ops rmi_f54_queue_ops = {
356	.queue_setup = rmi_f54_queue_setup,
357	.buf_queue = rmi_f54_buffer_queue,
358	.wait_prepare = vb2_ops_wait_prepare,
359	.wait_finish = vb2_ops_wait_finish,
360	};
361
362	static const struct vb2_queue rmi_f54_queue = {
363	.type = V4L2_BUF_TYPE_VIDEO_CAPTURE,
364	.io_modes = VB2_MMAP \| VB2_USERPTR \| VB2_DMABUF \| VB2_READ,
365	.buf_struct_size = sizeof(struct vb2_buffer),
366	.ops = &rmi_f54_queue_ops,
367	.mem_ops = &vb2_vmalloc_memops,
368	.timestamp_flags = V4L2_BUF_FLAG_TIMESTAMP_MONOTONIC,
369	.min_buffers_needed = 1,
370	};
371
372	static int rmi_f54_vidioc_querycap(struct file file, void priv,
373	struct v4l2_capability *cap)
374	{
375	struct f54_data *f54 = video_drvdata(file);
376
377	strlcpy(cap->driver, F54_NAME, sizeof(cap->driver));
378	strlcpy(cap->card, SYNAPTICS_INPUT_DEVICE_NAME, sizeof(cap->card));
379	snprintf(cap->bus_info, sizeof(cap->bus_info),
380	"rmi4:%s", dev_name(&f54->fn->dev));
381
382	return 0;
383	}
384
385	static int rmi_f54_vidioc_enum_input(struct file file, void priv,
386	struct v4l2_input *i)
387	{
388	struct f54_data *f54 = video_drvdata(file);
389	enum rmi_f54_report_type reptype;
390
391	reptype = rmi_f54_get_reptype(f54, i->index);
392	if (reptype == F54_REPORT_NONE)
393	return -EINVAL;
394
395	i->type = V4L2_INPUT_TYPE_TOUCH;
396
397	strlcpy(i->name, rmi_f54_report_type_names[reptype], sizeof(i->name));
398	return 0;
399	}
400
401	static int rmi_f54_set_input(struct f54_data *f54, unsigned int i)
402	{
c762cc68 GR	403	struct rmi_device *rmi_dev = f54->fn->rmi_dev;
	404	struct rmi_driver_data *drv_data = dev_get_drvdata(&rmi_dev->dev);
	405	u8 rx = drv_data->num_rx_electrodes ? : f54->num_rx_electrodes;
	406	u8 tx = drv_data->num_tx_electrodes ? : f54->num_tx_electrodes;
3a762dbd ND	407	struct v4l2_pix_format *f = &f54->format;
	408	enum rmi_f54_report_type reptype;
	409	int ret;
	410
	411	reptype = rmi_f54_get_reptype(f54, i);
	412	if (reptype == F54_REPORT_NONE)
	413	return -EINVAL;
	414
	415	ret = rmi_f54_get_pixel_fmt(reptype, &f->pixelformat);
	416	if (ret)
	417	return ret;
	418
	419	f54->input = i;
	420
c762cc68 GR	421	f->width = rx;
c762cc68 GR	422	f->height = tx;
3a762dbd ND	423	f->field = V4L2_FIELD_NONE;
	424	f->colorspace = V4L2_COLORSPACE_RAW;
	425	f->bytesperline = f->width * sizeof(u16);
	426	f->sizeimage = f->width * f->height * sizeof(u16);
	427
	428	return 0;
	429	}
	430
	431	static int rmi_f54_vidioc_s_input(struct file file, void priv, unsigned int i)
	432	{
	433	return rmi_f54_set_input(video_drvdata(file), i);
	434	}
	435
	436	static int rmi_f54_vidioc_g_input(struct file file, void priv,
	437	unsigned int *i)
	438	{
	439	struct f54_data *f54 = video_drvdata(file);
	440
	441	*i = f54->input;
	442
	443	return 0;
	444	}
	445
	446	static int rmi_f54_vidioc_fmt(struct file file, void priv,
	447	struct v4l2_format *f)
	448	{
	449	struct f54_data *f54 = video_drvdata(file);
	450
	451	f->fmt.pix = f54->format;
	452
	453	return 0;
	454	}
	455
	456	static int rmi_f54_vidioc_enum_fmt(struct file file, void priv,
	457	struct v4l2_fmtdesc *fmt)
	458	{
	459	if (fmt->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
	460	return -EINVAL;
	461
	462	switch (fmt->index) {
	463	case 0:
	464	fmt->pixelformat = V4L2_TCH_FMT_DELTA_TD16;
	465	break;
	466
	467	case 1:
	468	fmt->pixelformat = V4L2_TCH_FMT_DELTA_TD08;
	469	break;
	470
	471	case 2:
	472	fmt->pixelformat = V4L2_TCH_FMT_TU16;
	473	break;
	474
	475	default:
	476	return -EINVAL;
	477	}
	478
	479	return 0;
	480	}
	481
	482	static int rmi_f54_vidioc_g_parm(struct file file, void fh,
	483	struct v4l2_streamparm *a)
	484	{
	485	if (a->type != V4L2_BUF_TYPE_VIDEO_CAPTURE)
	486	return -EINVAL;
487
488	a->parm.capture.readbuffers = 1;
489	a->parm.capture.timeperframe.numerator = 1;
490	a->parm.capture.timeperframe.denominator = 10;
491	return 0;
492	}
493
494	static const struct v4l2_ioctl_ops rmi_f54_video_ioctl_ops = {
495	.vidioc_querycap = rmi_f54_vidioc_querycap,
496
497	.vidioc_enum_fmt_vid_cap = rmi_f54_vidioc_enum_fmt,
498	.vidioc_s_fmt_vid_cap = rmi_f54_vidioc_fmt,
499	.vidioc_g_fmt_vid_cap = rmi_f54_vidioc_fmt,
500	.vidioc_try_fmt_vid_cap = rmi_f54_vidioc_fmt,
501	.vidioc_g_parm = rmi_f54_vidioc_g_parm,
502
503	.vidioc_enum_input = rmi_f54_vidioc_enum_input,
504	.vidioc_g_input = rmi_f54_vidioc_g_input,
505	.vidioc_s_input = rmi_f54_vidioc_s_input,
506
507	.vidioc_reqbufs = vb2_ioctl_reqbufs,
508	.vidioc_create_bufs = vb2_ioctl_create_bufs,
509	.vidioc_querybuf = vb2_ioctl_querybuf,
510	.vidioc_qbuf = vb2_ioctl_qbuf,
511	.vidioc_dqbuf = vb2_ioctl_dqbuf,
512	.vidioc_expbuf = vb2_ioctl_expbuf,
513
514	.vidioc_streamon = vb2_ioctl_streamon,
515	.vidioc_streamoff = vb2_ioctl_streamoff,
516	};
517
518	static const struct video_device rmi_f54_video_device = {
519	.name = "Synaptics RMI4",
520	.fops = &rmi_f54_video_fops,
521	.ioctl_ops = &rmi_f54_video_ioctl_ops,
522	.release = video_device_release_empty,
523	.device_caps = V4L2_CAP_VIDEO_CAPTURE \| V4L2_CAP_TOUCH \|
524	V4L2_CAP_READWRITE \| V4L2_CAP_STREAMING,
525	};
526
527	static void rmi_f54_work(struct work_struct *work)
528	{
529	struct f54_data *f54 = container_of(work, struct f54_data, work.work);
530	struct rmi_function *fn = f54->fn;
531	u8 fifo[2];
532	struct rmi_f54_reports *report;
533	int report_size;
534	u8 command;
535	u8 *data;
536	int error;
537
538	data = f54->report_data;
539	report_size = rmi_f54_get_report_size(f54);
540	if (report_size == 0) {
541	dev_err(&fn->dev, "Bad report size, report type=%d\n",
542	f54->report_type);
543	error = -EINVAL;
544	goto error; /* retry won't help */
545	}
546	f54->standard_report[0].size = report_size;
547	report = f54->standard_report;
548
549	mutex_lock(&f54->data_mutex);
550
551	/*
552	* Need to check if command has completed.
553	* If not try again later.
554	*/
555	error = rmi_read(fn->rmi_dev, f54->fn->fd.command_base_addr,
556	&command);
557	if (error) {
558	dev_err(&fn->dev, "Failed to read back command\n");
559	goto error;
560	}
561	if (command & F54_GET_REPORT) {
562	if (time_after(jiffies, f54->timeout)) {
563	dev_err(&fn->dev, "Get report command timed out\n");
564	error = -ETIMEDOUT;
565	}
566	report_size = 0;
567	goto error;
568	}
569
570	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "Get report command completed, reading data\n");
571
572	report_size = 0;
573	for (; report->size; report++) {
574	fifo[0] = report->start & 0xff;
575	fifo[1] = (report->start >> 8) & 0xff;
576	error = rmi_write_block(fn->rmi_dev,
577	fn->fd.data_base_addr + F54_FIFO_OFFSET,
578	fifo, sizeof(fifo));
579	if (error) {
580	dev_err(&fn->dev, "Failed to set fifo start offset\n");
581	goto abort;
582	}
583
584	error = rmi_read_block(fn->rmi_dev, fn->fd.data_base_addr +
585	F54_REPORT_DATA_OFFSET, data,
586	report->size);
587	if (error) {
588	dev_err(&fn->dev, "%s: read [%d bytes] returned %d\n",
589	__func__, report->size, error);
590	goto abort;
591	}
592	data += report->size;
593	report_size += report->size;
594	}
595
596	abort:
597	f54->report_size = error ? 0 : report_size;
598	error:
599	if (error)
600	report_size = 0;
601
602	if (report_size == 0 && !error) {
603	queue_delayed_work(f54->workqueue, &f54->work,
604	msecs_to_jiffies(1));
605	} else {
606	f54->is_busy = false;
607	complete(&f54->cmd_done);
608	}
609
610	mutex_unlock(&f54->data_mutex);
611	}
612
613	static int rmi_f54_attention(struct rmi_function fn, unsigned long irqbits)
614	{
615	return 0;
616	}
617
618	static int rmi_f54_config(struct rmi_function *fn)
619	{
620	struct rmi_driver *drv = fn->rmi_dev->driver;
621
622	drv->set_irq_bits(fn->rmi_dev, fn->irq_mask);
623
624	return 0;
625	}
626
627	static int rmi_f54_detect(struct rmi_function *fn)
628	{
629	int error;
630	struct f54_data *f54;
97689352	631	u8 buf[6];
3a762dbd ND	632
	633	f54 = dev_get_drvdata(&fn->dev);
	634
	635	error = rmi_read_block(fn->rmi_dev, fn->fd.query_base_addr,
97689352	636	buf, sizeof(buf));
3a762dbd ND	637	if (error) {
	638	dev_err(&fn->dev, "%s: Failed to query F54 properties\n",
	639	__func__);
	640	return error;
	641	}
	642
97689352 AD	643	f54->num_rx_electrodes = buf[0];
	644	f54->num_tx_electrodes = buf[1];
	645	f54->capabilities = buf[2];
	646	f54->clock_rate = buf[3] \| (buf[4] << 8);
	647	f54->family = buf[5];
3a762dbd ND	648
	649	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 num_rx_electrodes: %d\n",
	650	f54->num_rx_electrodes);
	651	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 num_tx_electrodes: %d\n",
	652	f54->num_tx_electrodes);
	653	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 capabilities: 0x%x\n",
	654	f54->capabilities);
	655	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 clock rate: 0x%x\n",
	656	f54->clock_rate);
	657	rmi_dbg(RMI_DEBUG_FN, &fn->dev, "F54 family: 0x%x\n",
	658	f54->family);
	659
	660	f54->is_busy = false;
	661
	662	return 0;
	663	}
	664
	665	static int rmi_f54_probe(struct rmi_function *fn)
	666	{
	667	struct f54_data *f54;
	668	int ret;
	669	u8 rx, tx;
	670
	671	f54 = devm_kzalloc(&fn->dev, sizeof(struct f54_data), GFP_KERNEL);
	672	if (!f54)
	673	return -ENOMEM;
	674
	675	f54->fn = fn;
	676	dev_set_drvdata(&fn->dev, f54);
	677
	678	ret = rmi_f54_detect(fn);
	679	if (ret)
	680	return ret;
	681
	682	mutex_init(&f54->data_mutex);
	683	mutex_init(&f54->status_mutex);
	684
	685	rx = f54->num_rx_electrodes;
	686	tx = f54->num_tx_electrodes;
	687	f54->report_data = devm_kzalloc(&fn->dev,
	688	sizeof(u16) * tx * rx,
	689	GFP_KERNEL);
	690	if (f54->report_data == NULL)
	691	return -ENOMEM;
	692
	693	INIT_DELAYED_WORK(&f54->work, rmi_f54_work);
	694
	695	f54->workqueue = create_singlethread_workqueue("rmi4-poller");
	696	if (!f54->workqueue)
	697	return -ENOMEM;
	698
	699	rmi_f54_create_input_map(f54);
	700
	701	/* register video device */
	702	strlcpy(f54->v4l2.name, F54_NAME, sizeof(f54->v4l2.name));
	703	ret = v4l2_device_register(&fn->dev, &f54->v4l2);
	704	if (ret) {
	705	dev_err(&fn->dev, "Unable to register video dev.\n");
	706	goto remove_wq;
	707	}
	708
	709	/* initialize the queue */
	710	mutex_init(&f54->lock);
	711	f54->queue = rmi_f54_queue;
712	f54->queue.drv_priv = f54;
713	f54->queue.lock = &f54->lock;
714	f54->queue.dev = &fn->dev;
715
716	ret = vb2_queue_init(&f54->queue);
717	if (ret)
718	goto remove_v4l2;
719
720	f54->vdev = rmi_f54_video_device;
721	f54->vdev.v4l2_dev = &f54->v4l2;
722	f54->vdev.lock = &f54->lock;
723	f54->vdev.vfl_dir = VFL_DIR_RX;
724	f54->vdev.queue = &f54->queue;
725	video_set_drvdata(&f54->vdev, f54);
726
727	ret = video_register_device(&f54->vdev, VFL_TYPE_TOUCH, -1);
728	if (ret) {
729	dev_err(&fn->dev, "Unable to register video subdevice.");
730	goto remove_v4l2;
731	}
732
733	return 0;
734
735	remove_v4l2:
736	v4l2_device_unregister(&f54->v4l2);
737	remove_wq:
738	cancel_delayed_work_sync(&f54->work);
739	flush_workqueue(f54->workqueue);
740	destroy_workqueue(f54->workqueue);
741	return ret;
742	}
743
744	static void rmi_f54_remove(struct rmi_function *fn)
745	{
746	struct f54_data *f54 = dev_get_drvdata(&fn->dev);
747
748	video_unregister_device(&f54->vdev);
749	v4l2_device_unregister(&f54->v4l2);
750	}
751
752	struct rmi_function_handler rmi_f54_handler = {
753	.driver = {
754	.name = F54_NAME,
755	},
756	.func = 0x54,
757	.probe = rmi_f54_probe,
758	.config = rmi_f54_config,
759	.attention = rmi_f54_attention,
760	.remove = rmi_f54_remove,
761	};