env = Environment()

# config

# just comment following line to disable CUDA
cuda          = 'cuda-9.2'
try: cuda
except NameError: cuda = ''
cuda_path      = '/opt/' + cuda
cuda_bin       = cuda_path + '/bin'
cuda_pkgconfig = cuda_path + '/pkgconfig'
cuda_flags     = '-O3 -use_fast_math'

libs = ['gl', 'x11', 'OpenCL']


# compute build options

flags = ' -O3 -Wall -fmessage-length=0 -DGL_GLEXT_PROTOTYPES'
cuda_flags = ' '

if cuda:
	flags += ' -DCUDA'
	
	# remember pkg-config path
	old = env['ENV']['PKG_CONFIG_PATH'] if 'PKG_CONFIG_PATH' in env['ENV'] else None

	env['ENV']['PKG_CONFIG_PATH'] = cuda_pkgconfig
	env.ParseConfig('pkg-config --cflags --libs ' + cuda)
	
	# restore pkg-config path
	if old: env['ENV']['PKG_CONFIG_PATH'] = old
	
	env['BUILDERS']['Cuda'] = Builder(
		action = cuda_bin + '/nvcc ' + cuda_flags + '-ptx $SOURCE -o $TARGET',
		suffix = '.ptx',
		src_suffix = '.cu' )
	

# files lists

target = 'contourgl'

sources = [
	'contourgl.cpp',
	'clcontext.cpp',
	'clrender.cpp',
	'contour.cpp',
	'contourbuilder.cpp',
	'environment.cpp',
	'geometry.cpp',
	'glcontext.cpp',
	'measure.cpp',
	'polyspan.cpp',
	'shaders.cpp',
	'swrender.cpp',
	'test.cpp',
	'triangulator.cpp',
	'utils.cpp' ]

if cuda:
	sources += [
		'cudacontext.cpp',
		'cudarender.cpp' ]
	cuda_sources = [
		'cuda/contour.cu',
		'cuda/hello.cu' ]


# build

env.ParseConfig('pkg-config --cflags --libs ' + ' '.join(libs))

if cuda:
	for cuda_source in cuda_sources:
		env.Cuda(cuda_source)
	
env.Program(
	target = target,
	source = sources,
	parse_flags = flags )