Source code for fury

"""Init file for visualization package."""

from __future__ import division, print_function, absolute_import
import warnings
from fury._version import get_versions

__version__ = get_versions()['version']
__revision_id__ = get_versions()['full-revisionid']
del get_versions


def get_info(verbose=False):
    """Return dict describing the context of this package.

    Parameters
    ------------
    pkg_path : str
       path containing __init__.py for package
    Returns
    ----------
    context : dict
       with named parameters of interest

    """
    from fury.optpkg import optional_package
    from os.path import dirname
    import sys
    import numpy
    import scipy
    import vtk

    mpl, have_mpl, _ = optional_package('matplotlib')
    dipy, have_dipy, _ = optional_package('dipy')

    info = dict(fury_version=__version__,
                pkg_path=dirname(__file__),
                commit_hash=__revision_id__,
                sys_version=sys.version,
                sys_executable=sys.executable,
                sys_platform=sys.platform,
                numpy_version=numpy.__version__,
                scipy_version=scipy.__version__,
                vtk_version=vtk.vtkVersion.GetVTKVersion())

    d_mpl = dict(matplotlib_version=mpl.__version__) if have_mpl else {}
    d_dipy = dict(dipy_version=dipy.__version__) if have_dipy else {}

    info.update(d_mpl)
    info.update(d_dipy)

    if verbose:
        print('\n'.join(['{0}: {1}'.format(k, v) for k, v in info.items()]))

    return info


# Ignore this specific warning below from vtk < 8.2.
# FutureWarning: Conversion of the second argument of issubdtype from
# `complex` to `np.complexfloating` is deprecated. In future, it will be
# treated as `np.complex128 == np.dtype(complex).type`.
# assert not numpy.issubdtype(z.dtype, complex), \
warnings.filterwarnings("ignore",
                        message="Conversion of the second argument of"
                                " issubdtype from `complex` to"
                                " `np.complexfloating` is deprecated.*",
                        category=FutureWarning)