from support.bsp_sources.target import DFBBTarget

import os

# cuda library
CUDA_API = os.path.join(os.path.dirname(__file__), '..', 'api')

def add_source_dir(holder, subdir, path):
    for f in os.listdir(path):
        full = os.path.join(path, f)
        if os.path.isfile(full):
            holder.add_source(subdir, full)

class CUDADevice(DFBBTarget):
    @property
    def name(self):
        return "cuda"

    @property
    def target(self):
        return "cuda"

    # @property
    # def readme_file(self):
    #     return os.path.join(os.path.dirname(__file__), 'README')

    @property
    def system_ads(self):
        return {'device': 'runtime/device_gnat/system.ads'}

    @property
    def compiler_switches(self):
        # The required compiler switches
        return ('-gnatp',)

    def amend_rts(self, rts_profile, cfg):
        # For simplicity pretend to be ZFP. CUDA rts_sources doesn't use any
        # ZFP RTS variables so we are safe to do this for now.
        super(CUDADevice, self).amend_rts('zfp', cfg)
        cfg.rts_vars['Cuda_Target'] = 'device'
        # Add cuda sources. Probably better to put these in their own cuda
        # directory with their own make file.
        # add_source_dir(cfg, 'gnat', os.path.join(CUDA_API, 'device_static'))


    def __init__(self):
        super(CUDADevice, self).__init__()