// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * TerraTec Cinergy T2/qanu USB2 DVB-T adapter.
 *
 * Copyright (C) 2007 Tomi Orava (tomimo@ncircle.nullnet.fi)
 *
 * Based on the dvb-usb-framework code and the
 * original Terratec Cinergy T2 driver by:
 *
 * Copyright (C) 2004 Daniel Mack <daniel@qanu.de> and
 *                  Holger Waechtler <holger@qanu.de>
 *
 *  Protocol Spec published on http://qanu.de/specs/terratec_cinergyT2.pdf
 */

#include "cinergyT2.h"


/* debug */
int dvb_usb_cinergyt2_debug;

module_param_named(debug, dvb_usb_cinergyt2_debug, int, 0644);
MODULE_PARM_DESC(debug, "set debugging level (1=info, xfer=2, rc=4 (or-able)).");

DVB_DEFINE_MOD_OPT_ADAPTER_NR(adapter_nr);

struct cinergyt2_state {
        u8 rc_counter;
        unsigned char data[64];
};

/* Forward declaration */
static const struct dvb_usb_device_properties cinergyt2_properties;

static int cinergyt2_streaming_ctrl(struct dvb_usb_adapter *adap, int enable)
{
        struct dvb_usb_device *d = adap->dev;
        struct cinergyt2_state *st = d->priv;
        int ret;

        mutex_lock(&d->data_mutex);
        st->data[0] = CINERGYT2_EP1_CONTROL_STREAM_TRANSFER;
        st->data[1] = enable ? 1 : 0;

        ret = dvb_usb_generic_rw(d, st->data, 2, st->data, 64, 0);
        mutex_unlock(&d->data_mutex);

        return ret;
}

static int cinergyt2_power_ctrl(struct dvb_usb_device *d, int enable)
{
        struct cinergyt2_state *st = d->priv;
        int ret;

        mutex_lock(&d->data_mutex);
        st->data[0] = CINERGYT2_EP1_SLEEP_MODE;
        st->data[1] = enable ? 0 : 1;

        ret = dvb_usb_generic_rw(d, st->data, 2, st->data, 3, 0);
        mutex_unlock(&d->data_mutex);

        return ret;
}

static int cinergyt2_frontend_attach(struct dvb_usb_adapter *adap)
{
        struct dvb_usb_device *d = adap->dev;
        struct cinergyt2_state *st = d->priv;
        int ret;

        adap->fe_adap[0].fe = cinergyt2_fe_attach(adap->dev);

        mutex_lock(&d->data_mutex);
        st->data[0] = CINERGYT2_EP1_GET_FIRMWARE_VERSION;

        ret = dvb_usb_generic_rw(d, st->data, 1, st->data, 3, 0);
        if (ret < 0) {
                if (adap->fe_adap[0].fe)
                        adap->fe_adap[0].fe->ops.release(adap->fe_adap[0].fe);
                deb_rc("cinergyt2_power_ctrl() Failed to retrieve sleep state info\n");
        }
        mutex_unlock(&d->data_mutex);

        return ret;
}

static struct rc_map_table rc_map_cinergyt2_table[] = {
        { 0x0401, KEY_POWER },
        { 0x0402, KEY_1 },
        { 0x0403, KEY_2 },
        { 0x0404, KEY_3 },
        { 0x0405, KEY_4 },
        { 0x0406, KEY_5 },
        { 0x0407, KEY_6 },
        { 0x0408, KEY_7 },
        { 0x0409, KEY_8 },
        { 0x040a, KEY_9 },
        { 0x040c, KEY_0 },
        { 0x040b, KEY_VIDEO },
        { 0x040d, KEY_REFRESH },
        { 0x040e, KEY_SELECT },
        { 0x040f, KEY_EPG },
        { 0x0410, KEY_UP },
        { 0x0414, KEY_DOWN },
        { 0x0411, KEY_LEFT },
        { 0x0413, KEY_RIGHT },
        { 0x0412, KEY_OK },
        { 0x0415, KEY_TEXT },
        { 0x0416, KEY_INFO },
        { 0x0417, KEY_RED },
        { 0x0418, KEY_GREEN },
        { 0x0419, KEY_YELLOW },
        { 0x041a, KEY_BLUE },
        { 0x041c, KEY_VOLUMEUP },
        { 0x041e, KEY_VOLUMEDOWN },
        { 0x041d, KEY_MUTE },
        { 0x041b, KEY_CHANNELUP },
        { 0x041f, KEY_CHANNELDOWN },
        { 0x0440, KEY_PAUSE },
        { 0x044c, KEY_PLAY },
        { 0x0458, KEY_RECORD },
        { 0x0454, KEY_PREVIOUS },
        { 0x0448, KEY_STOP },
        { 0x045c, KEY_NEXT }
};

/* Number of keypresses to ignore before detect repeating */
#define RC_REPEAT_DELAY 3

static int repeatable_keys[] = {
        KEY_UP,
        KEY_DOWN,
        KEY_LEFT,
        KEY_RIGHT,
        KEY_VOLUMEUP,
        KEY_VOLUMEDOWN,
        KEY_CHANNELUP,
        KEY_CHANNELDOWN
};

static int cinergyt2_rc_query(struct dvb_usb_device *d, u32 *event, int *state)
{
        struct cinergyt2_state *st = d->priv;
        int i, ret;

        *state = REMOTE_NO_KEY_PRESSED;

        mutex_lock(&d->data_mutex);
        st->data[0] = CINERGYT2_EP1_GET_RC_EVENTS;

        ret = dvb_usb_generic_rw(d, st->data, 1, st->data, 5, 0);
        if (ret < 0)
                goto ret;

        if (st->data[4] == 0xff) {
                /* key repeat */
                st->rc_counter++;
                if (st->rc_counter > RC_REPEAT_DELAY) {
                        for (i = 0; i < ARRAY_SIZE(repeatable_keys); i++) {
                                if (d->last_event == repeatable_keys[i]) {
                                        *state = REMOTE_KEY_REPEAT;
                                        *event = d->last_event;
                                        deb_rc("repeat key, event %x\n",
                                                   *event);
                                        goto ret;
                                }
                        }
                        deb_rc("repeated key (non repeatable)\n");
                }
                goto ret;
        }

        /* hack to pass checksum on the custom field */
        st->data[2] = ~st->data[1];
        dvb_usb_nec_rc_key_to_event(d, st->data, event, state);
        if (st->data[0] != 0) {
                if (*event != d->last_event)
                        st->rc_counter = 0;

                deb_rc("key: %*ph\n", 5, st->data);
        }

ret:
        mutex_unlock(&d->data_mutex);
        return ret;
}

static int cinergyt2_usb_probe(struct usb_interface *intf,
                                const struct usb_device_id *id)
{
        return dvb_usb_device_init(intf, &cinergyt2_properties,
                                   THIS_MODULE, NULL, adapter_nr);
}

enum {
        TERRATEC_CINERGY_T2,
};

static const struct usb_device_id cinergyt2_usb_table[] = {
        DVB_USB_DEV(TERRATEC, TERRATEC_CINERGY_T2),
        { }
};

MODULE_DEVICE_TABLE(usb, cinergyt2_usb_table);

static const struct dvb_usb_device_properties cinergyt2_properties = {
        .size_of_priv = sizeof(struct cinergyt2_state),
        .num_adapters = 1,
        .adapter = {
                {
                .num_frontends = 1,
                .fe = {{
                        .streaming_ctrl   = cinergyt2_streaming_ctrl,
                        .frontend_attach  = cinergyt2_frontend_attach,

                        /* parameter for the MPEG2-data transfer */
                        .stream = {
                                .type = USB_BULK,
                                .count = 5,
                                .endpoint = 0x02,
                                .u = {
                                        .bulk = {
                                                .buffersize = 512,
                                        }
                                }
                        },
                }},
                }
        },

        .power_ctrl       = cinergyt2_power_ctrl,

        .rc.legacy = {
                .rc_interval      = 50,
                .rc_map_table     = rc_map_cinergyt2_table,
                .rc_map_size      = ARRAY_SIZE(rc_map_cinergyt2_table),
                .rc_query         = cinergyt2_rc_query,
        },

        .generic_bulk_ctrl_endpoint = 1,

        .num_device_descs = 1,
        .devices = {
                { .name = "TerraTec/qanu USB2.0 Highspeed DVB-T Receiver",
                  .cold_ids = {NULL},
                  .warm_ids = { &cinergyt2_usb_table[TERRATEC_CINERGY_T2], NULL },
                },
                { NULL },
        }
};


static struct usb_driver cinergyt2_driver = {
        .name           = "cinergyT2",
        .probe          = cinergyt2_usb_probe,
        .disconnect     = dvb_usb_device_exit,
        .id_table       = cinergyt2_usb_table
};

module_usb_driver(cinergyt2_driver);

MODULE_DESCRIPTION("Terratec Cinergy T2 DVB-T driver");
MODULE_LICENSE("GPL");
MODULE_AUTHOR("Tomi Orava");