Import('PRJ_KERNEL')
from building import *

cwd = GetCurrentDir()
path = [cwd + '/common']
path += [cwd + '/core']
path += [cwd]
src = []

CPPDEFINES = []

# USB DEVICE
if GetDepend(['LPKG_CHERRYUSB_DEVICE']) and GetDepend(['AIC_USB_DEVICE_DRV']):
    src += Glob('core/usbd_core.c')

    if GetDepend(['LPKG_CHERRYUSB_DEVICE_HS']):
        CPPDEFINES+=['CONFIG_USB_HS']

    if GetDepend(['LPKG_CHERRYUSB_DEVICE_CDC']):
        path += [cwd + '/class/cdc']
        src += Glob('class/cdc/usbd_cdc.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_HID']):
        path += [cwd + '/class/hid']
        src += Glob('class/hid/usbd_hid.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_MSC']):
        path += [cwd + '/class/msc']
        src += Glob('class/msc/usbd_msc.c')
        if GetDepend(['LPKG_CHERRYUSB_EXMAPLE']):
            src += Glob('example/usbd_msc_ramdisk.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_AUDIO']):
        path += [cwd + '/class/audio']
        src += Glob('class/audio/usbd_audio.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_VIDEO']):
        path += [cwd + '/class/video']
        src += Glob('class/video/usbd_video.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_RNDIS']):
        path += [cwd + '/class/wireless']
        src += Glob('class/wireless/usbd_rndis.c')
    if GetDepend(['LPKG_CHERRYUSB_USING_DFU']):
        path += [cwd + '/class/dfu']
        src += Glob('class/dfu/usbd_dfu.c')
    if GetDepend(['LPKG_CHERRYUSB_USING_HUB']):
        path += [cwd + '/class/hub']
        src += Glob('class/hub/usbd_hub.c')

    if GetDepend(['LPKG_CHERRYUSB_DEVICE_CDC_TEMPLATE']):
        src += Glob('demo/cdc_acm_template.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_HID_MOUSE_TEMPLATE']):
        src += Glob('demo/hid_mouse_template.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_HID_KEYBOARD_TEMPLATE']):
        src += Glob('demo/hid_keyboard_template.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_MSC_TEMPLATE']):
        src += Glob('demo/msc_ram_template.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_AUDIO_V1_TEMPLATE']):
        src += Glob('demo/audio_v1_mic_speaker_multichan_template.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_AUDIO_V2_TEMPLATE']):
        src += Glob('demo/audio_v2_mic_speaker_multichan_template.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_VIDEO_TEMPLATE']):
        src += Glob('demo/video_static_mjpeg_template.c')

    if GetDepend(['LPKG_CHERRYUSB_DEVICE_AIC']):
        src += Glob('port/aic/usb_dc_aic.c')
        src += Glob('port/aic/usb_dc_aic_ll.c')

    if GetDepend(['LPKG_CHERRYUSB_DEVICE_FSDEV']):
        src += Glob('port/fsdev/usb_dc_fsdev.c')
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_DWC2']):
        src += Glob('port/dwc2/usb_dc_dwc2.c')
        if GetDepend(['LPKG_CHERRYUSB_DEVICE_DWC2_PORT_FS']):
            CPPDEFINES += ['CONFIG_USB_DWC2_PORT=FS_PORT']
        elif GetDepend(['LPKG_CHERRYUSB_DEVICE_DWC2_PORT_HS']):
            CPPDEFINES += ['CONFIG_USB_DWC2_PORT=HS_PORT']
    if GetDepend(['LPKG_CHERRYUSB_DEVICE_MUSB']):
        src += Glob('port/musb/usb_dc_musb.c')
        if GetDepend(['LPKG_CHERRYUSB_DEVICE_MUSB_SUNXI']):
            CPPDEFINES += ['CONFIG_USB_MUSB_SUNXI']

# USB HOST
if GetDepend(['LPKG_CHERRYUSB_HOST']) and GetDepend(['AIC_USB_HOST_EHCI_DRV']):
    path += [cwd + '/osal']
    path += [cwd + '/class/hub']
    src += Glob('core/usbh_core.c')
    if PRJ_KERNEL == 'rt-thread':
        src += Glob('osal/usb_osal_rtthread.c')
    elif PRJ_KERNEL == 'freertos':
        src += Glob('osal/usb_osal_freertos.c')
    src += Glob('osal/usb_workq.c')

    if GetDepend(['LPKG_CHERRYUSB_HOST_CDC']):
        path += [cwd + '/class/cdc']
        src += Glob('class/cdc/usbh_cdc_acm.c')
    if GetDepend(['LPKG_CHERRYUSB_HOST_HID']):
        path += [cwd + '/class/hid']
        src += Glob('class/hid/usbh_hid.c')
    if GetDepend(['LPKG_CHERRYUSB_HOST_MSC']):
        path += [cwd + '/class/msc']
        src += Glob('class/msc/usbh_msc.c')
        if GetDepend(['LPKG_CHERRYUSB_EXMAPLE']):
            src += Glob('example/usbh_msc.c')
    if GetDepend(['LPKG_CHERRYUSB_HOST_RNDIS']):
        src += Glob('class/wireless/usbh_rndis.c')

    if GetDepend(['LPKG_CHERRYUSB_HOST_HUB']):
        src += Glob('class/hub/usbh_hub.c')
        CPPDEFINES += ['CONFIG_USBHOST_HUB']

    if GetDepend(['LPKG_CHERRYUSB_HOST_DWC2']):
        src += Glob('port/dwc2/usb_hc_dwc2.c')
        CPPDEFINES += ['CONFIG_USBHOST_HIGH_WORKQ']

    if GetDepend(['LPKG_CHERRYUSB_HOST_MUSB']):
        src += Glob('port/musb/usb_hc_musb.c')
        if GetDepend(['LPKG_CHERRYUSB_HOST_MUSB_SUNXI']):
            CPPDEFINES += ['CONFIG_USB_MUSB_SUNXI']

    if GetDepend(['LPKG_CHERRYUSB_HOST_EHCI']):
        src += Glob('port/ehci/usb_ehci.c')
        src += Glob('port/ehci/usb_ehci_aic_ll.c')

    if GetDepend(['LPKG_CHERRYUSB_HOST_TEMPLATE']):
        src += Glob('demo/usb_host.c')

group = DefineGroup('CherryUSB', src, depend = ['LPKG_USING_CHERRYUSB'], CPPPATH = path, CPPDEFINES = CPPDEFINES)

Return('group')