# JH 2020

easyblock = 'CMakeMake'

name = 'VTK'
version = '8.2.0'
versionsuffix = '-Python-%(pyver)s'

homepage = 'https://www.vtk.org'
description = """The Visualization Toolkit (VTK) is an open-source, freely available software system for
 3D computer graphics, image processing and visualization. VTK consists of a C++ class library and several
 interpreted interface layers including Tcl/Tk, Java, and Python. VTK supports a wide variety of visualization
 algorithms including: scalar, vector, tensor, texture, and volumetric methods; and advanced modeling techniques
 such as: implicit modeling, polygon reduction, mesh smoothing, cutting, contouring, and Delaunay triangulation."""

toolchain = {'name': 'intel', 'version': '2020a'}
toolchainopts = {'usempi': True}

source_urls = ['https://www.vtk.org/files/release/%(version_major_minor)s']
sources = [
    SOURCE_TAR_GZ,
    '%(name)sData-%(version)s.tar.gz',
]
patches = [('vtk-version.egg-info', '.')]
checksums = [
    '34c3dc775261be5e45a8049155f7228b6bd668106c72a3c435d95730d17d57bb',  # VTK-8.2.0.tar.gz
    'd1ff312f7a63d90d8b7033a99109801f16a462ae411d648642838aae04bcc21e',  # VTKData-8.2.0.tar.gz
    '787b82415ae7a4a1f815b4db0e25f7abc809a05fc85d7d219627f3a7e5d3867b',  # vtk-version.egg-info
]

builddependencies = [('CMake', '3.16.4')]

dependencies = [
    ('Python', '3.7.8'),
    ('XZ', '5.2.5'),
    ('HDF5', '1.10.6'),
    ('netCDF', '4.7.3'),
    ('libGLU', '9.0.1'),
    ('Mesa', '20.0.2'),
    ('X11', '20200222'),
    ('SciPy-bundle', '2020.03', versionsuffix),
]

separate_build_dir = True

# Third party modules
configopts = "-DVTK_USE_SYSTEM_MPI4PY=ON "
configopts += "-DVTK_USE_SYSTEM_LZMA=ON "
configopts += "-DVTK_USE_SYSTEM_HDF5=ON "
configopts += "-DVTK_USE_SYSTEM_NETCDF=ON "
# OpenGL
configopts += "-DOPENGL_glu_LIBRARY=$EBROOTLIBGLU/lib/libGLU.%s " % SHLIB_EXT
configopts += "-DOPENGL_gl_LIBRARY=$EBROOTMESA/lib/libGL.%s " % SHLIB_EXT
configopts += "-DOPENGL_INCLUDE_DIR=$EBROOTMESA/include "
# Python
configopts += "-DVTK_WRAP_PYTHON=ON "
configopts += "-DVTK_PYTHON_VERSION=3.7 "
configopts += "-DPYTHON_INCLUDE_DIR=$EBROOTPYTHON/include/python3.7m "
configopts += "-DPYTHON_LIBRARY=$EBROOTPYTHON/lib/libpython3.7m.%s " % SHLIB_EXT
# Other
configopts += "-DVTK_Group_MPI:BOOL=ON "
configopts += "-DCMAKE_INSTALL_LIBDIR=lib"

preinstallopts = "export PYTHONPATH=%(installdir)s/lib/python3.7/site-packages:$PYTHONPATH && "

# Install a egg-info file so VTK is more python friendly, required for mayavi
local_egg_info_src = '%(builddir)s/VTK-%(version)s/vtk-version.egg-info'
local_egg_info_dest = '%(installdir)s/lib/python3.7/site-packages/vtk-%(version)s.egg-info'
postinstallcmds = [
    'sed "s/#VTK_VERSION#/%%(version)s/" %s > %s' % (local_egg_info_src, local_egg_info_dest),
]

modextrapaths = {'PYTHONPATH': ['lib/python3.7/site-packages']}

sanity_check_paths = {
    'files': ['bin/vtk%s-%%(version_major_minor)s' % x for x in
              ['WrapJava', 'ParseJava', 'WrapPythonInit', 'WrapPython', 'WrapHierarchy']] +
             ['bin/pvtkpython', 'bin/vtkpython'],
    'dirs': ['lib/python3.7/site-packages/', 'include/vtk-%(version_major_minor)s'],
}

sanity_check_commands = [
    ('python', "-c 'import %(namelower)s'"),
    ('python', "-c 'import pkg_resources; pkg_resources.get_distribution(\"vtk\")'"),
]


moduleclass = 'vis'