# Copyright 2013-2022 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

from spack import *


class Openfast(CMakePackage):
    """Wind turbine simulation package from NREL"""

    homepage = "https://openfast.readthedocs.io/en/latest/"
    git      = "https://github.com/OpenFAST/openfast.git"

    maintainers = ['jrood-nrel']

    version('develop', branch='dev')
    version('master', branch='main')
    version('3.1.0', tag='v3.1.0')
    version('3.0.0', tag='v3.0.0')
    version('2.6.0', tag='v2.6.0')
    version('2.5.0', tag='v2.5.0')
    version('2.4.0', tag='v2.4.0')
    version('2.3.0', tag='v2.3.0')
    version('2.2.0', tag='v2.2.0')
    version('2.1.0', tag='v2.1.0')
    version('2.0.0', tag='v2.0.0')
    version('1.0.0', tag='v1.0.0')

    variant('shared', default=True,
            description="Build shared libraries")
    variant('double-precision', default=True,
            description="Treat REAL as double precision")
    variant('dll-interface', default=True,
            description="Enable dynamic library loading interface")
    variant('cxx', default=False,
            description="Enable C++ bindings")
    variant('pic', default=True,
            description="Position independent code")
    variant('openmp', default=False,
            description="Enable OpenMP support")

    # Dependencies for OpenFAST Fortran
    depends_on('blas')
    depends_on('lapack')

    # Additional dependencies when compiling C++ library
    depends_on('mpi', when='+cxx')
    depends_on('yaml-cpp', when='+cxx')
    depends_on('hdf5+mpi+cxx+hl', when='+cxx')
    depends_on('zlib', when='+cxx')
    depends_on('libxml2', when='+cxx')

    def cmake_args(self):
        spec = self.spec

        options = []

        options.extend([
            '-DBUILD_DOCUMENTATION:BOOL=OFF',
            '-DBUILD_TESTING:BOOL=OFF',
            self.define_from_variant('BUILD_SHARED_LIBS', 'shared'),
            self.define_from_variant('DOUBLE_PRECISION', 'double-precision'),
            self.define_from_variant('USE_DLL_INTERFACE', 'dll-interface'),
            self.define_from_variant('BUILD_OPENFAST_CPP_API', 'cxx'),
            self.define_from_variant('CMAKE_POSITION_INDEPENDENT_CODE', 'pic'),
        ])

        # Make sure we use Spack's blas/lapack:
        blas_libs = spec['lapack'].libs + spec['blas'].libs
        options.extend([
            '-DBLAS_LIBRARIES=%s' % blas_libs.joined(';'),
            '-DLAPACK_LIBRARIES=%s' % blas_libs.joined(';')
        ])

        if '+cxx' in spec:
            options.extend([
                '-DCMAKE_C_COMPILER=%s' % spec['mpi'].mpicc,
                '-DCMAKE_CXX_COMPILER=%s' % spec['mpi'].mpicxx,
                '-DCMAKE_Fortran_COMPILER=%s' % spec['mpi'].mpifc,
                '-DMPI_CXX_COMPILER:PATH=%s' % spec['mpi'].mpicxx,
                '-DMPI_C_COMPILER:PATH=%s' % spec['mpi'].mpicc,
                '-DMPI_Fortran_COMPILER:PATH=%s' % spec['mpi'].mpifc,
                '-DHDF5_ROOT:PATH=%s' % spec['hdf5'].prefix,
                '-DYAML_ROOT:PATH=%s' % spec['yaml-cpp'].prefix,
            ])

            if '~shared' in spec:
                options.extend([
                    '-DHDF5_USE_STATIC_LIBRARIES=ON',
                ])

        if '+openmp' in spec:
            options.extend([
                '-DOPENMP:BOOL=ON',
            ])

        if 'darwin' in spec.architecture:
            options.append('-DCMAKE_MACOSX_RPATH:BOOL=ON')

        return options

    def flag_handler(self, name, flags):
        spec = self.spec
        if name in ['cflags', 'cxxflags', 'cppflags', 'fflags']:
            if '+openmp' in spec:
                flags.append(self.compiler.openmp_flag)
            return (None, flags, None)
        return(flags, None, None)
