#
# RTEMS Project (https://www.rtems.org/)
#
# Copyright (c) 2019 Vijay Kumar Banerjee <vijaykumar9597@gmail.com>.
# All rights reserved.
#
#  Redistribution and use in source and binary forms, with or without
#  modification, are permitted provided that the following conditions
#  are met:
#  1. Redistributions of source code must retain the above copyright
#     notice, this list of conditions and the following disclaimer.
#  2. Redistributions in binary form must reproduce the above copyright
#     notice, this list of conditions and the following disclaimer in the
#     documentation and/or other materials provided with the distribution.
#
#  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
#  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
#  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
#  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
#  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
#  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
#  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
#  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
#  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
#  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
#  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

import shlex
import os
import re
import rtems_waf.rtems as rtems

def source_list(bld):
    sources = []
    cflags = []
    mk_files = ['lvgl/src/lv_core/lv_core.mk',
                'lvgl/src/lv_hal/lv_hal.mk',
                'lvgl/src/lv_widgets/lv_widgets.mk',
                'lvgl/src/lv_font/lv_font.mk',
                'lvgl/src/lv_misc/lv_misc.mk',
                'lvgl/src/lv_themes/lv_themes.mk',
                'lvgl/src/lv_draw/lv_draw.mk',
                'lvgl/src/lv_gpu/lv_gpu.mk',
               ]

    if bld.env.DRIVERS:
        drv_path = ['lv_drivers/display', 'lv_drivers/indev']
        sources = [os.path.join(d, s) for d in drv_path for s in os.listdir(d) if (s[-2:] == '.c')]

    for filename in mk_files:
      lexer = shlex.shlex(open(filename, 'rt').read())
      lexer.whitespace += '+='
      lexer.whitespace_split = True

      for token in lexer:
          if token == 'CSRCS':
              source_path = os.path.dirname(filename)
              source_path = os.path.join(source_path, next(lexer))
              sources.append(source_path)

          if token == 'CFLAGS':
              cflag = next(lexer)
              if(cflag[14] == '/'):
                  cflags.append(cflag[15:-1])
              else:
                  cflags.append(cflag[14:-1])

    return (sources, cflags)

def build(bld):

    sources, includes = source_list(bld)
    includes.append('.')
    include_paths = []

    def write_stuff(stuff):
        def stuff_writer(task):
            task.outputs[0].write(stuff)
        return stuff_writer

    lv_conf_h='lv_conf.h'
    lv_drv_conf_h='lv_drv_conf.h'

    bld(rule=write_stuff(bld.env.LV_CONF), target=lv_conf_h)
    bld(rule=write_stuff(bld.env.LV_DRV_CONF), target=lv_drv_conf_h)

    for source in sources:
        source_dir = os.path.dirname(source)
        if source_dir not in include_paths:
            include_paths.append(source_dir)

    bld.stlib(target = 'lvgl',
              features = 'c',
              cflags = ['-O2', '-g', '-DLV_CONF_INCLUDE_SIMPLE'],
              includes = includes,
              source = sources)

    arch_lib_path = rtems.arch_bsp_lib_path(bld.env.RTEMS_VERSION,
                                            bld.env.RTEMS_ARCH_BSP)
    arch_inc_path = rtems.arch_bsp_include_path(bld.env.RTEMS_VERSION,
                                                bld.env.RTEMS_ARCH_BSP)

    include_paths.extend(['lvgl/', 'lvgl/src', '.'])
    if bld.env.DRIVERS:
        include_paths.extend(['lv_drivers/display', 'lv_drivers/indev'])

    for include_path in include_paths:
        files = os.listdir(include_path)
        include_headers = [os.path.join(include_path, x) for x in files if (x[-2:] == '.h')]
        bld.install_files(os.path.join("${PREFIX}", arch_inc_path, include_path),
                          include_headers)
    bld.install_files(os.path.join('${PREFIX}', arch_lib_path), ["liblvgl.a"])
    bld.install_files(os.path.join('${PREFIX}', arch_inc_path),
        [lv_conf_h, lv_drv_conf_h])