SConstruct

#!python

import os
import platform

# Configure build environment.
env = None
if (platform.system() == 'Windows'):

    #
    # Check for user defined compiler.
    # i.e. > scons.bat -Q compiler=mingw
    #
    # The compiler needs to be in the users path.
    #
    compiler = ARGUMENTS.get('compiler', '')
    print("Using compiler", compiler)
    if (len(compiler) > 0):
        env = DefaultEnvironment(tools = [compiler],
                                 ENV = {'PATH' : os.environ['PATH'],
                                        'TMP' : os.environ['TMP'],
                                        'TEMP' : os.environ['TEMP']})
        
# Use the current environment if nothing was specified.
if env is None:
    env = Environment(ENV = os.environ)


# C compiler flags.
#
# FIXME: Visual C flags?
if (env['CC'] == "gcc"):
    if (platform.system() == 'Linux'):
        if True:
            env.Append(CCFLAGS = ['-O3','-Wall'],
                       LINKFLAGS = ['-Wl,-z,defs'])
        else: # Build with debugging.
            env.Append(CCFLAGS = ['-Og','-Wall'],
                       LINKFLAGS = ['-Wl,-z,defs'])
    else:
        env.Append(CCFLAGS = ['-O3','-Wall'])

# Library names and paths.
fftw_lib = 'fftw3'
fftw_lib_path = []
lapack_lib_path = []

#
# OS-X specific settings, FFTW is in /usr/local/?
#
if (platform.system() == "Darwin"):
    fftw_lib='libfftw3'
    fftw_lib_path = ['/usr/local/lib']
    env.Append(CCFLAGS='-I/usr/local/include')
    env.Append(LDFLAGS='-L/usr/local/include')

#
# Windows specific settings library setting. Basically we are trying
# to figure out if FFTW and LAPACK exist in the build environment or
# if we should use the versions included in this package.
#
if (platform.system() == 'Windows'):
    fftw_lib = 'fftw3-3'
    conf = Configure(env)
    if not conf.CheckLib(fftw_lib):
        print("FFTW3 library not found, using storm-analysis version.")
        fftw_lib_path = ['#/storm_analysis/c_libraries/']
    if not conf.CheckLib('lapack'):
        print("LAPACK library not found, using storm-analysis version.")
        lapack_lib_path = ['#/storm_analysis/c_libraries/']

#
# This is for linking libraries that use both FFTW and LAPACK.
#
fftw_lapack_cpp_path = []
fftw_lapack_lib_path = []
if fftw_lib_path is not None:
    fftw_lapack_cpp_path += fftw_lib_path
    fftw_lapack_lib_path += fftw_lib_path
    
if lapack_lib_path is not None:
    if not (lapack_lib_path in fftw_lapack_lib_path):
        fftw_lapack_lib_path += lapack_lib_path

#
# storm_analysis/admm
#
if True:
    Default(env.SharedObject(source = './storm_analysis/sa_library/ft_math.c',
                             target = './storm_analysis/c_libraries/ft_math.o',
                             CPPPATH = fftw_lapack_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/admm_lasso',
                              ['./storm_analysis/c_libraries/ft_math.o',
                               './storm_analysis/admm/admm_lasso.c'],
                              LIBS = [fftw_lib, 'm'], 
                              LIBPATH = fftw_lib_path, 
                              CPPPATH = fftw_lib_path))


#
# storm_analysis/dbscan
#
if True:
    Default(env.SharedObject(source = './storm_analysis/dbscan/kdtree.c',
                             target = './storm_analysis/c_libraries/kdtree.o'))
        
    Default(env.SharedLibrary('./storm_analysis/c_libraries/dbscan',
	                      ['./storm_analysis/c_libraries/kdtree.o',
                               './storm_analysis/dbscan/dbscan.c']))

    
#
# storm_analysis/densestorm
#
if True:
    Default(env.SharedObject(source = './storm_analysis/sa_library/ft_math.c',
                             target = './storm_analysis/c_libraries/ft_math.o',
                             CPPPATH = fftw_lapack_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/densestorm',
                              ['./storm_analysis/c_libraries/ft_math.o',
                               './storm_analysis/densestorm/densestorm.c'],
                              LIBS = [fftw_lib, 'm'], 
                              LIBPATH = fftw_lib_path, 
                              CPPPATH = fftw_lib_path))


#
# storm_analysis/fista
#
if True:
    Default(env.SharedObject(source = './storm_analysis/sa_library/ft_math.c',
                             target = './storm_analysis/c_libraries/ft_math.o',
                             CPPPATH = fftw_lapack_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/fista_fft',
                              ['./storm_analysis/c_libraries/ft_math.o',
                               './storm_analysis/fista/fista_fft.c'],
                              LIBS = [fftw_lib, 'm'], 
                              LIBPATH = fftw_lib_path, 
                              CPPPATH = fftw_lib_path))

    
#
# storm_analysis/frc
#
if True:
    Default(env.SharedLibrary('./storm_analysis/c_libraries/frc',
	                      ['./storm_analysis/frc/frc.c'],
                              LIBS = ['m']))


#
# storm_analysis/L1H
#
if True:
    l1h_libs = ['lapack', 'rt', 'm']
    if (platform.system() == "Darwin"):
        # OS-X apparently does not have and doesn't need the rt library.
        l1h_libs = ['lapack', 'm']
    
    if (platform.system() == 'Windows'):
        # Windows (MINGW) apparently also does not have it?
        l1h_libs = ['lapack', 'm']

    Default(env.SharedObject(source = './storm_analysis/L1H/homotopy_common.c',
                             target = './storm_analysis/c_libraries/homotopy_common.o'))
    
    Default(env.SharedObject(source = './storm_analysis/L1H/homotopy_imagea.c',
                             target = './storm_analysis/c_libraries/homotopy_imagea.o'))
    
    Default(env.SharedObject(source = './storm_analysis/L1H/homotopy_imagea_common.c',
                             target = './storm_analysis/c_libraries/homotopy_imagea_common.o'))

    Default(env.SharedObject(source = './storm_analysis/L1H/homotopy_sse.c',
                             target = './storm_analysis/c_libraries/homotopy_sse.o'))
    
    Default(env.SharedObject(source = './storm_analysis/L1H/homotopy_storm.c',
                             target = './storm_analysis/c_libraries/homotopy_storm.o'))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/homotopy_general',
                              ['./storm_analysis/L1H/homotopy_general.c',
                               './storm_analysis/c_libraries/homotopy_common.o'],
                              LIBS = l1h_libs, 
                              LIBPATH = lapack_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/homotopy_ia_sse',
                              ['./storm_analysis/c_libraries/homotopy_imagea.o',
                               './storm_analysis/c_libraries/homotopy_sse.o',
                               './storm_analysis/c_libraries/homotopy_imagea_common.o',
                               './storm_analysis/c_libraries/homotopy_common.o'],
                              LIBS = l1h_libs, 
                              LIBPATH = lapack_lib_path))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/homotopy_ia_storm',
                              ['./storm_analysis/c_libraries/homotopy_imagea.o',
                               './storm_analysis/c_libraries/homotopy_storm.o',
                               './storm_analysis/c_libraries/homotopy_imagea_common.o',
                               './storm_analysis/c_libraries/homotopy_common.o'],
                              LIBS = l1h_libs, 
                              LIBPATH = lapack_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/homotopy_sse',
                              ['./storm_analysis/c_libraries/homotopy_sse.o',
                               './storm_analysis/c_libraries/homotopy_common.o'],
                              LIBS = l1h_libs, 
                              LIBPATH = lapack_lib_path))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/homotopy_storm',
                              ['./storm_analysis/c_libraries/homotopy_storm.o',
                               './storm_analysis/c_libraries/homotopy_common.o'],
                              LIBS = l1h_libs, 
                              LIBPATH = lapack_lib_path))

#
# storm_analysis/multi_plane
#
if True:
    Default(env.SharedObject(source = './storm_analysis/multi_plane/mp_fit.c',
                             target = './storm_analysis/c_libraries/mp_fit.o',
                             CPPPATH = fftw_lapack_cpp_path))

    # 3D-DAOSTORM multiplane fitting.
    Default(env.SharedObject(source = './storm_analysis/multi_plane/mp_fit_dao.c',
                             target = './storm_analysis/c_libraries/mp_fit_dao.o',
                             CPPPATH = fftw_lapack_cpp_path))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/mp_fit_dao',
                              ['./storm_analysis/c_libraries/dao_fit.o',
                               './storm_analysis/c_libraries/mp_fit.o',
                               './storm_analysis/c_libraries/mp_fit_dao.o',
                               './storm_analysis/c_libraries/multi_fit.o'],
                              LIBS = [fftw_lib, 'lapack', 'm'], 
                              LIBPATH = fftw_lapack_lib_path, 
                              CPPPATH = fftw_lapack_cpp_path))
    
    # Spline / Pupil Function / PSF FFT multiplane fitting.
    Default(env.SharedObject(source = './storm_analysis/multi_plane/mp_fit_arb.c',
                             target = './storm_analysis/c_libraries/mp_fit_arb.o',
                             CPPPATH = fftw_lapack_cpp_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/mp_fit_arb',
                              ['./storm_analysis/c_libraries/cubic_fit.o',
                               './storm_analysis/c_libraries/cubic_spline.o',
                               './storm_analysis/c_libraries/fft_fit.o',
                               './storm_analysis/c_libraries/psf_fft.o',
                               './storm_analysis/c_libraries/pupil_fit.o',
                               './storm_analysis/c_libraries/pupil_function.o',
                               './storm_analysis/c_libraries/mp_fit.o',
                               './storm_analysis/c_libraries/mp_fit_arb.o',
                               './storm_analysis/c_libraries/multi_fit.o'],
                              LIBS = [fftw_lib, 'lapack', 'm'], 
                              LIBPATH = fftw_lapack_lib_path, 
                              CPPPATH = fftw_lapack_cpp_path))
 
#
# storm_analysis/psf_fft
#
if True:
    Default(env.SharedObject(source = './storm_analysis/psf_fft/psf_fft.c',
                             target = './storm_analysis/c_libraries/psf_fft.o',
                             CPPPATH = fftw_lapack_lib_path))

    Default(env.SharedObject(source = './storm_analysis/psf_fft/fft_fit.c',
                             target = './storm_analysis/c_libraries/fft_fit.o',
                             CPPPATH = fftw_lapack_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/psf_fft',
                              ['./storm_analysis/c_libraries/psf_fft.o'],
                              LIBS = [fftw_lib, 'm'], 
                              LIBPATH = fftw_lib_path, 
                              CPPPATH = fftw_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/fft_fit',
                              ['./storm_analysis/c_libraries/multi_fit.o',
                               './storm_analysis/c_libraries/psf_fft.o',
                               './storm_analysis/c_libraries/fft_fit.o'],
                              LIBS = [fftw_lib, 'lapack', 'm'], 
                              LIBPATH = fftw_lapack_lib_path, 
                              CPPPATH = fftw_lapack_cpp_path))

#
# storm_analysis/pupilfn
#
if True:
    Default(env.SharedObject(source = './storm_analysis/pupilfn/otf_scaling.c',
                             target = './storm_analysis/c_libraries/otf_scaling.o',
                             CPPPATH = fftw_lapack_lib_path))

    Default(env.SharedObject(source = './storm_analysis/pupilfn/pupil_fit.c',
                             target = './storm_analysis/c_libraries/pupil_fit.o',
                             CPPPATH = fftw_lapack_lib_path))

    Default(env.SharedObject(source = './storm_analysis/pupilfn/pupil_function.c',
                             target = './storm_analysis/c_libraries/pupil_function.o',
                             CPPPATH = fftw_lapack_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/otf_scaling',
                              ['./storm_analysis/c_libraries/otf_scaling.o'],
                              LIBS = [fftw_lib, 'm'], 
                              LIBPATH = fftw_lib_path, 
                              CPPPATH = fftw_lib_path))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/pupil_function',
                              ['./storm_analysis/c_libraries/pupil_function.o'],
                              LIBS = [fftw_lib, 'm'], 
                              LIBPATH = fftw_lib_path, 
                              CPPPATH = fftw_lib_path))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/pupil_fit',
                              ['./storm_analysis/c_libraries/multi_fit.o',
                               './storm_analysis/c_libraries/pupil_function.o',
                               './storm_analysis/c_libraries/pupil_fit.o'],
                              LIBS = [fftw_lib, 'lapack', 'm'], 
                              LIBPATH = fftw_lapack_lib_path, 
                              CPPPATH = fftw_lapack_cpp_path))

#
# storm_analysis/rolling_ball_bgr
#
if True:
    Default(env.SharedLibrary('./storm_analysis/c_libraries/rolling_ball_lib',
	                      ['./storm_analysis/rolling_ball_bgr/rolling_ball_lib.c']))

#
# storm_analysis/sa_library
#
if True:
    Default(env.SharedObject(source = './storm_analysis/sa_library/dao_fit.c',
                             target = './storm_analysis/c_libraries/dao_fit.o'))

    Default(env.SharedObject(source = './storm_analysis/sa_library/ft_math.c',
                             target = './storm_analysis/c_libraries/ft_math.o',
                             CPPPATH = fftw_lapack_lib_path))

    Default(env.SharedObject(source = './storm_analysis/sa_library/multi_fit.c',
                             target = './storm_analysis/c_libraries/multi_fit.o'))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/dao_fit',
                              ['./storm_analysis/c_libraries/dao_fit.o',
                               './storm_analysis/c_libraries/multi_fit.o'],
                              LIBS = ['lapack', 'm'], 
                              LIBPATH = lapack_lib_path))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/affine_transform',
	                      ['./storm_analysis/sa_library/affine_transform.c']))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/cs_decon_utilities',
                              ['./storm_analysis/sa_library/cs_decon_utilities.c']))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/grid',
	                      ['./storm_analysis/sa_library/grid.c']))
    
    Default(env.SharedLibrary('./storm_analysis/c_libraries/ia_utilities',
	                      ['./storm_analysis/c_libraries/kdtree.o',
                               './storm_analysis/sa_library/ia_utilities.c'],
                              LIBS = ['m']))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/matched_filter',
                              ['./storm_analysis/c_libraries/ft_math.o',
                               './storm_analysis/sa_library/matched_filter.c'],
                              LIBS = [fftw_lib], 
                              LIBPATH = fftw_lib_path, 
                              CPPPATH = fftw_lib_path))

#
# storm_analysis/sa_utilities
#
if True:
    Default(env.SharedLibrary('./storm_analysis/c_libraries/fitz',
	                      ['./storm_analysis/sa_utilities/fitz.c'],
                              LIBS = ['m']))


#
# storm_analysis/simulator
#
if True:
    Default(env.SharedLibrary('./storm_analysis/c_libraries/draw_gaussians',
	                      ['./storm_analysis/simulator/draw_gaussians.c'],
                              LIBS = ['m']))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/pf_math',
                             ['./storm_analysis/simulator/pf_math.c'],
                              LIBS = [fftw_lib, 'm'],
                              LIBPATH = fftw_lib_path,
                              CPPPATH = fftw_lib_path))

#
# storm_analysis/spliner
#
if True:
    Default(env.SharedObject(source = './storm_analysis/spliner/cubic_fit.c',
                             target = './storm_analysis/c_libraries/cubic_fit.o'))

    Default(env.SharedObject(source = './storm_analysis/spliner/cubic_spline.c',
                             target = './storm_analysis/c_libraries/cubic_spline.o'))

    Default(env.SharedLibrary('./storm_analysis/c_libraries/cubic_spline',
	                      ['./storm_analysis/c_libraries/cubic_spline.o']))
                           
    Default(env.SharedLibrary('./storm_analysis/c_libraries/cubic_fit',
                              ['./storm_analysis/c_libraries/cubic_fit.o',
                               './storm_analysis/c_libraries/cubic_spline.o',
                               './storm_analysis/c_libraries/multi_fit.o'],
                              LIBS = ['lapack', 'm'], 
                              LIBPATH = lapack_lib_path))