###############################################################################
# Copyright (c) Lawrence Livermore National Security, LLC and other Ascent
# Project developers. See top-level LICENSE AND COPYRIGHT files for dates and
# other details. No copyright assignment is required to contribute to Ascent.
###############################################################################

###############################################################################
# Setup Conduit
# This file defines:
#  CONDUIT_FOUND - If Conduit was found
#  CONDUIT_INCLUDE_DIRS - The Conduit include directories
#
#  If found, the conduit CMake targets will also be imported
###############################################################################

include(CMakeFindDependencyMacro)

# first Check for CONDUIT_DIR

if(NOT CONDUIT_DIR)
    MESSAGE(FATAL_ERROR "Conduit support needs explicit CONDUIT_DIR")
endif()

MESSAGE(STATUS "Looking for Conduit using CONDUIT_DIR = ${CONDUIT_DIR}")

###############################################################################
# Import Conduit's CMake targets
###############################################################################
find_dependency(Conduit REQUIRED
                NO_DEFAULT_PATH
                PATHS ${CONDUIT_DIR}/lib/cmake/conduit)

# if fortran is enabled, make sure conduit was built with fortran support
if(FORTRAN_FOUND)
    if(EXISTS ${CONDUIT_DIR}/include/conduit/conduit.mod)
        message(STATUS "FOUND conduit fortran module at: ${CONDUIT_DIR}/include/conduit/conduit.mod")
    else()
        message(FATAL_ERROR "Failed to find conduit fortran module. \
                             Ascent Fortran support requires Conduit with Fortran enabled.")
    endif()
endif()


if(EXISTS ${CONDUIT_DIR}/include/conduit/conduit_relay_io_hdf5.hpp)
    set(CONDUIT_HDF5_ENABLED TRUE)
    message(STATUS "FOUND conduit HDF5 support")
endif()

if(EXISTS ${CONDUIT_DIR}/include/conduit/conduit_relay_io_silo.hpp)
    set(CONDUIT_SILO_ENABLED TRUE)
    message(STATUS "FOUND conduit Silo support")
endif()

if(EXISTS ${CONDUIT_DIR}/include/conduit/conduit_relay_adios_api.hpp)
    set(CONDUIT_ADIOS_ENABLED TRUE)
    message(STATUS "FOUND conduit ADIOS support")
endif()

set(CONDUIT_FOUND TRUE)
set(CONDUIT_INCLUDE_DIRS ${CONDUIT_DIR}/include/conduit)


if(NOT CONDUIT_RELAY_WEBSERVER_ENABLED)
    # older versions of conduit may still have web support but
    # not export this cmake var, so for now, we also 
    # check for the web headers
    if(EXISTS ${CONDUIT_DIR}/include/conduit/conduit_relay_web.hpp)
        set(CONDUIT_RELAY_WEBSERVER_ENABLED TRUE)
    else()
        set(CONDUIT_RELAY_WEBSERVER_ENABLED FALSE)
    endif()
endif()

message(STATUS "CONDUIT_RELAY_WEBSERVER_ENABLED = ${CONDUIT_RELAY_WEBSERVER_ENABLED}")


if(ENABLE_PYTHON)
    set(CONDUIT_PYTHON_INCLUDE_DIR)

    # If conduit provides a python module directory, try to find the conduit C API header there
    if (EXISTS "${CONDUIT_PYTHON_MODULE_DIR}/conduit/conduit_python.hpp")
        set(CONDUIT_PYTHON_INCLUDE_DIR ${CONDUIT_PYTHON_MODULE_DIR}/conduit)
    endif ()

    # Interegate the conduit python module to find its location if the CONDUIT_PYTHON_MODULE_DIR
    # failed.
    if (NOT CONDUIT_PYTHON_INCLUDE_DIR)
        find_package(PythonInterp)
        if(PYTHONINTERP_FOUND)
            execute_process(COMMAND "${PYTHON_EXECUTABLE}" "-c"
                                    "import os; import conduit; print(os.path.split(conduit.__file__)[0]);"
                            RESULT_VARIABLE _FIND_CONDUIT_PYTHON_RESULT
                            OUTPUT_VARIABLE _FIND_CONDUIT_PYTHON_OUT
                            ERROR_VARIABLE  _FIND_CONDUIT_PYTHON_ERROR_VALUE
                            OUTPUT_STRIP_TRAILING_WHITESPACE)
                          message(STATUS "PYTHON found!! ${CONDUIT_DIR}")
            if(_FIND_CONDUIT_PYTHON_RESULT MATCHES 0)
                message(STATUS "FOUND conduit python module at: ${_FIND_CONDUIT_PYTHON_OUT}")
            else()
                # try find the path to the conduit python module assuming a standard conduit install
                execute_process(COMMAND "${PYTHON_EXECUTABLE}" "-c"
                                       "import sys; import os; sys.path.append(os.path.join('${CONDUIT_DIR}','python-modules'));  import conduit; print(os.path.split(conduit.__file__)[0]);"
                                        RESULT_VARIABLE _FIND_CONDUIT_PYTHON_RESULT
                                        OUTPUT_VARIABLE _FIND_CONDUIT_PYTHON_OUT
                                        ERROR_VARIABLE  _FIND_CONDUIT_PYTHON_ERROR_VALUE
                                        OUTPUT_STRIP_TRAILING_WHITESPACE)
                if(_FIND_CONDUIT_PYTHON_RESULT MATCHES 0)
                    # we use this to make sure we can setup unit tests env path correctly
                    list(APPEND EXTRA_PYTHON_MODULE_DIRS "${CONDUIT_DIR}/python-modules/")
                    message(STATUS "FOUND conduit python module at: ${_FIND_CONDUIT_PYTHON_OUT}")
                else()
                    message(FATAL_ERROR
                    "conduit python import failure:\n${_FIND_CONDUIT_PYTHON_OUT}")
                endif()
            endif()
        else()
            message(FATAL_ERROR "PYTHON_FOUND = TRUE, but could not find a python interpreter.")
        endif()
        set(CONDUIT_PYTHON_INCLUDE_DIR ${_FIND_CONDUIT_PYTHON_OUT})
    else()
        # we use this to make sure we can setup unit tests env path correctly
        list(APPEND EXTRA_PYTHON_MODULE_DIRS ${CONDUIT_PYTHON_MODULE_DIR})
    endif ()

    # Verify that the python header was found, if not abort
    if (NOT EXISTS "${CONDUIT_PYTHON_INCLUDE_DIR}/conduit_python.hpp")
        message(FATAL_ERROR "Cannot find conduit_python.hpp")
    endif ()

    # Add the conduit_python.h path to the include list
    message(STATUS "FOUND conduit python include dir: ${CONDUIT_PYTHON_INCLUDE_DIR}")
    list(APPEND CONDUIT_INCLUDE_DIRS ${CONDUIT_PYTHON_INCLUDE_DIR})
endif()

message(STATUS "FOUND Conduit at ${CONDUIT_DIR}")
message(STATUS "CONDUIT_INCLUDE_DIRS = ${CONDUIT_INCLUDE_DIRS}")


blt_register_library( NAME conduit
                      INCLUDES ${CONDUIT_INCLUDE_DIRS}
                      LIBRARIES  conduit conduit_relay conduit_blueprint)

# assumes conduit was built with mpi
if(MPI_FOUND)
    blt_register_library( NAME conduit_relay_mpi
                          INCLUDES ${CONDUIT_INCLUDE_DIRS}
                          LIBRARIES  conduit_relay_mpi conduit_blueprint_mpi)
endif()


if(ASCENT_ENABLE_TESTS AND WIN32 AND BUILD_SHARED_LIBS)
    # if we are running tests with dlls, we need path to dlls
    list(APPEND ASCENT_TPL_DLL_PATHS ${CONDUIT_DIR}/bin)

    # zlib dll
    if(CONDUIT_ZLIB_DIR)
        list(APPEND ASCENT_TPL_DLL_PATHS ${CONDUIT_ZLIB_DIR}/bin/)
    endif()

    # hdf5 dlls
    if(CONDUIT_HDF5_ENABLED)
        list(APPEND ASCENT_TPL_DLL_PATHS ${CONDUIT_HDF5_DIR}/bin)
    endif()

    # silo dlls
    if(CONDUIT_SILO_ENABLED)
        list(APPEND ASCENT_TPL_DLL_PATHS ${CONDUIT_SILO_DIR}/bin)
    endif()

endif()