/*
 * File      : usbdevice.c
 * This file is part of RT-Thread RTOS
 * COPYRIGHT (C) 2012, RT-Thread Development Team
 *
 *  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.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along
 *  with this program; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 * 
 * Change Logs:
 * Date           Author       Notes
 * 2012-10-02     Yi Qiu       first version
 */

#include <rtthread.h>
#include <rtdevice.h>
#include <rtservice.h>

#ifdef RT_USING_USB_DEVICE

#define USB_DEVICE_CONTROLLER_NAME      "usbd"

#ifdef RT_USB_DEVICE_COMPOSITE
const static char* ustring[] =
{
    "Language",
    "RT-Thread Team.",
    "RTT Composite Device",
    "320219198301",
    "Configuration",
    "Interface",
    USB_STRING_OS
};

static struct udevice_descriptor compsit_desc =
{
    USB_DESC_LENGTH_DEVICE,     //bLength;
    USB_DESC_TYPE_DEVICE,       //type;
    USB_BCD_VERSION,            //bcdUSB;
    USB_CLASS_MISC,             //bDeviceClass;
    0x02,                       //bDeviceSubClass;
    0x01,                       //bDeviceProtocol;
    0x40,                       //bMaxPacketSize0;
    _VENDOR_ID,                 //idVendor;
    _PRODUCT_ID,                //idProduct;
    USB_BCD_DEVICE,             //bcdDevice;
    USB_STRING_MANU_INDEX,      //iManufacturer;
    USB_STRING_PRODUCT_INDEX,   //iProduct;
    USB_STRING_SERIAL_INDEX,    //iSerialNumber;
    USB_DYNAMIC,                //bNumConfigurations;
};

//FS and HS needed
static struct usb_qualifier_descriptor dev_qualifier =
{
    sizeof(dev_qualifier),          //bLength
    USB_DESC_TYPE_DEVICEQUALIFIER,  //bDescriptorType
    0x0200,                         //bcdUSB
    USB_CLASS_MISC,                 //bDeviceClass
    0x02,                           //bDeviceSubClass
    0x01,                           //bDeviceProtocol
    64,                             //bMaxPacketSize0
    0x01,                           //bNumConfigurations
    0,
};
#endif

struct usb_os_comp_id_descriptor usb_comp_id_desc = 
{
    //head section
    USB_DYNAMIC,
    0x0100,
    0x04,
    USB_DYNAMIC,
    {0x00,0x00,0x00,0x00,0x00,0x00,0x00},
};

rt_err_t rt_usb_device_init(void)
{
    rt_device_t udc;
    udevice_t udevice;
    uconfig_t cfg;
    ufunction_t func;

    /* create and startup usb device thread */
    rt_usbd_core_init();

    /* create a device object */
    udevice = rt_usbd_device_new();

    udc = rt_device_find(USB_DEVICE_CONTROLLER_NAME);
    if(udc == RT_NULL)
    {
        rt_kprintf("can't find usb device controller %s\n", USB_DEVICE_CONTROLLER_NAME);
        return -RT_ERROR;
    }

    /* set usb controller driver to the device */
    rt_usbd_device_set_controller(udevice, (udcd_t)udc);

    /* create a configuration object */
    cfg = rt_usbd_config_new();

    rt_usbd_device_set_os_comp_id_desc(udevice,&usb_comp_id_desc);

#ifdef RT_USB_DEVICE_MSTORAGE
    {
        extern ufunction_t rt_usbd_function_mstorage_create(udevice_t device);
        /* create a mass storage function object */
        func = rt_usbd_function_mstorage_create(udevice);

        /* add the function to the configuration */
        rt_usbd_config_add_function(cfg, func);
    }
#endif

#ifdef RT_USB_DEVICE_CDC
    {
        extern ufunction_t rt_usbd_function_cdc_create(udevice_t device);
        /* create a cdc function object */
        func = rt_usbd_function_cdc_create(udevice);

        /* add the function to the configuration */
        rt_usbd_config_add_function(cfg, func);
    }
#endif

#ifdef RT_USB_DEVICE_HID
    {
        extern ufunction_t rt_usbd_function_hid_create(udevice_t device);
        /* create a cdc function object */
        func = rt_usbd_function_hid_create(udevice);

        /* add the function to the configuration */
        rt_usbd_config_add_function(cfg, func);
    }
#endif

#ifdef RT_USB_DEVICE_RNDIS
    {
        extern ufunction_t rt_usbd_function_rndis_create(udevice_t device);
        /* create a rndis function object */
        func = rt_usbd_function_rndis_create(udevice);

        /* add the function to the configuration */
        rt_usbd_config_add_function(cfg, func);
    }
#endif

#ifdef RT_USB_DEVICE_ECM
    {
        extern ufunction_t rt_usbd_function_ecm_create(udevice_t device);
        /* create a rndis function object */
        func = rt_usbd_function_ecm_create(udevice);

        /* add the function to the configuration */
        rt_usbd_config_add_function(cfg, func);
    }
#endif
    
#ifdef RT_USB_DEVICE_WINUSB
    {
        extern ufunction_t rt_usbd_function_winusb_create(udevice_t device);
        /* create a rndis function object */
        func = rt_usbd_function_winusb_create(udevice);

        /* add the function to the configuration */
        rt_usbd_config_add_function(cfg, func);
    }
#endif

    /* set device descriptor to the device */
#ifdef RT_USB_DEVICE_COMPOSITE
    rt_usbd_device_set_descriptor(udevice, &compsit_desc);
    rt_usbd_device_set_string(udevice, ustring);
    rt_usbd_device_set_qualifier(udevice, &dev_qualifier);
#else
    rt_usbd_device_set_descriptor(udevice, func->dev_desc);
#endif

    /* add the configuration to the device */
    rt_usbd_device_add_config(udevice, cfg);

    /* initialize usb device controller */
    rt_device_init(udc);

    /* set default configuration to 1 */
    rt_usbd_set_config(udevice, 1);

    return RT_EOK;
}
#endif