/*
 * Linux driver for TerraTec DMX 6Fire USB
 *
 * Main routines and module definitions.
 *
 * Author:      Torsten Schenk <torsten.schenk@zoho.com>
 * Created:     Jan 01, 2011
 * Copyright:   (C) Torsten Schenk
 *
 * 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.
 */

#include "chip.h"
#include "firmware.h"
#include "pcm.h"
#include "control.h"
#include "comm.h"
#include "midi.h"

#include <linux/moduleparam.h>
#include <linux/interrupt.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/gfp.h>
#include <sound/initval.h>

MODULE_AUTHOR("Torsten Schenk <torsten.schenk@zoho.com>");
MODULE_DESCRIPTION("TerraTec DMX 6Fire USB audio driver");
MODULE_LICENSE("GPL v2");
MODULE_SUPPORTED_DEVICE("{{TerraTec, DMX 6Fire USB}}");

static int index[SNDRV_CARDS] = SNDRV_DEFAULT_IDX; /* Index 0-max */
static char *id[SNDRV_CARDS] = SNDRV_DEFAULT_STR; /* Id for card */
static bool enable[SNDRV_CARDS] = SNDRV_DEFAULT_ENABLE_PNP; /* Enable card */
static struct sfire_chip *chips[SNDRV_CARDS] = SNDRV_DEFAULT_PTR;
static struct usb_device *devices[SNDRV_CARDS] = SNDRV_DEFAULT_PTR;

module_param_array(index, int, NULL, 0444);
MODULE_PARM_DESC(index, "Index value for the 6fire sound device");
module_param_array(id, charp, NULL, 0444);
MODULE_PARM_DESC(id, "ID string for the 6fire sound device.");
module_param_array(enable, bool, NULL, 0444);
MODULE_PARM_DESC(enable, "Enable the 6fire sound device.");

static DEFINE_MUTEX(register_mutex);

static void usb6fire_chip_abort(struct sfire_chip *chip)
{
        if (chip) {
                if (chip->pcm)
                        usb6fire_pcm_abort(chip);
                if (chip->midi)
                        usb6fire_midi_abort(chip);
                if (chip->comm)
                        usb6fire_comm_abort(chip);
                if (chip->control)
                        usb6fire_control_abort(chip);
                if (chip->card) {
                        snd_card_disconnect(chip->card);
                        snd_card_free_when_closed(chip->card);
                        chip->card = NULL;
                }
        }
}

static void usb6fire_chip_destroy(struct sfire_chip *chip)
{
        if (chip) {
                if (chip->pcm)
                        usb6fire_pcm_destroy(chip);
                if (chip->midi)
                        usb6fire_midi_destroy(chip);
                if (chip->comm)
                        usb6fire_comm_destroy(chip);
                if (chip->control)
                        usb6fire_control_destroy(chip);
                if (chip->card)
                        snd_card_free(chip->card);
        }
}

static int __devinit usb6fire_chip_probe(struct usb_interface *intf,
                const struct usb_device_id *usb_id)
{
        int ret;
        int i;
        struct sfire_chip *chip = NULL;
        struct usb_device *device = interface_to_usbdev(intf);
        int regidx = -1; /* index in module parameter array */
        struct snd_card *card = NULL;

        /* look if we already serve this card and return if so */
        mutex_lock(&register_mutex);
        for (i = 0; i < SNDRV_CARDS; i++) {
                if (devices[i] == device) {
                        if (chips[i])
                                chips[i]->intf_count++;
                        usb_set_intfdata(intf, chips[i]);
                        mutex_unlock(&register_mutex);
                        return 0;
                } else if (regidx < 0)
                        regidx = i;
        }
        if (regidx < 0) {
                mutex_unlock(&register_mutex);
                snd_printk(KERN_ERR PREFIX "too many cards registered.\n");
                return -ENODEV;
        }
        devices[regidx] = device;
        mutex_unlock(&register_mutex);

        /* check, if firmware is present on device, upload it if not */
        ret = usb6fire_fw_init(intf);
        if (ret < 0)
                return ret;
        else if (ret == FW_NOT_READY) /* firmware update performed */
                return 0;

        /* if we are here, card can be registered in alsa. */
        if (usb_set_interface(device, 0, 0) != 0) {
                snd_printk(KERN_ERR PREFIX "can't set first interface.\n");
                return -EIO;
        }
        ret = snd_card_create(index[regidx], id[regidx], THIS_MODULE,
                        sizeof(struct sfire_chip), &card);
        if (ret < 0) {
                snd_printk(KERN_ERR PREFIX "cannot create alsa card.\n");
                return ret;
        }
        strcpy(card->driver, "6FireUSB");
        strcpy(card->shortname, "TerraTec DMX6FireUSB");
        sprintf(card->longname, "%s at %d:%d", card->shortname,
                        device->bus->busnum, device->devnum);
        snd_card_set_dev(card, &intf->dev);

        chip = card->private_data;
        chips[regidx] = chip;
        chip->dev = device;
        chip->regidx = regidx;
        chip->intf_count = 1;
        chip->card = card;

        ret = usb6fire_comm_init(chip);
        if (ret < 0) {
                usb6fire_chip_destroy(chip);
                return ret;
        }

        ret = usb6fire_midi_init(chip);
        if (ret < 0) {
                usb6fire_chip_destroy(chip);
                return ret;
        }

        ret = usb6fire_pcm_init(chip);
        if (ret < 0) {
                usb6fire_chip_destroy(chip);
                return ret;
        }

        ret = usb6fire_control_init(chip);
        if (ret < 0) {
                usb6fire_chip_destroy(chip);
                return ret;
        }

        ret = snd_card_register(card);
        if (ret < 0) {
                snd_printk(KERN_ERR PREFIX "cannot register card.");
                usb6fire_chip_destroy(chip);
                return ret;
        }
        usb_set_intfdata(intf, chip);
        return 0;
}

static void usb6fire_chip_disconnect(struct usb_interface *intf)
{
        struct sfire_chip *chip;
        struct snd_card *card;

        chip = usb_get_intfdata(intf);
        if (chip) { /* if !chip, fw upload has been performed */
                card = chip->card;
                chip->intf_count--;
                if (!chip->intf_count) {
                        mutex_lock(&register_mutex);
                        devices[chip->regidx] = NULL;
                        chips[chip->regidx] = NULL;
                        mutex_unlock(&register_mutex);

                        chip->shutdown = true;
                        usb6fire_chip_abort(chip);
                        usb6fire_chip_destroy(chip);
                }
        }
}

static struct usb_device_id device_table[] = {
        {
                .match_flags = USB_DEVICE_ID_MATCH_DEVICE,
                .idVendor = 0x0ccd,
                .idProduct = 0x0080
        },
        {}
};

MODULE_DEVICE_TABLE(usb, device_table);

static struct usb_driver usb_driver = {
        .name = "snd-usb-6fire",
        .probe = usb6fire_chip_probe,
        .disconnect = usb6fire_chip_disconnect,
        .id_table = device_table,
};

module_usb_driver(usb_driver);