Merge branch 'master' into default_value_2532

Author: djarecka

nipype/algorithms/metrics.py

@@ -19,11 +19,11 @@
 from scipy.ndimage.measurements import center_of_mass, label
 
 from .. import config, logging
-from ..utils.misc import package_check
 
 from ..interfaces.base import (BaseInterface, traits, TraitedSpec, File,
                                InputMultiPath, BaseInterfaceInputSpec,
                                isdefined)
+from ..interfaces.nipy.base import NipyBaseInterface
 from ..utils import NUMPY_MMAP
 
 iflogger = logging.getLogger('interface')
@@ -651,7 +651,7 @@ class SimilarityOutputSpec(TraitedSpec):
         traits.Float(desc="Similarity between volume 1 and 2, frame by frame"))
 
 
-class Similarity(BaseInterface):
+class Similarity(NipyBaseInterface):
     """Calculates similarity between two 3D or 4D volumes. Both volumes have to be in
     the same coordinate system, same space within that coordinate system and
     with the same voxel dimensions.
@@ -674,19 +674,8 @@ class Similarity(BaseInterface):
 
     input_spec = SimilarityInputSpec
     output_spec = SimilarityOutputSpec
-    _have_nipy = True
-
-    def __init__(self, **inputs):
-        try:
-            package_check('nipy')
-        except Exception:
-            self._have_nipy = False
-        super(Similarity, self).__init__(**inputs)
 
     def _run_interface(self, runtime):
-        if not self._have_nipy:
-            raise RuntimeError('nipy is not installed')
-
         from nipy.algorithms.registration.histogram_registration import HistogramRegistration
         from nipy.algorithms.registration.affine import Affine
 

nipype/interfaces/afni/preprocess.py

@@ -2510,7 +2510,7 @@ class TProjectInputSpec(AFNICommandInputSpec):
                rather than the value stored in the dataset header.""",
         argstr='-TR %g')
     mask = File(
-        exist=True,
+        exists=True,
         desc="""Only operate on voxels nonzero in the mset dataset.
                 ++ Voxels outside the mask will be filled with zeros.
                 ++ If no masking option is given, then all voxels

nipype/interfaces/afni/tests/test_auto_TProject.py

@@ -32,10 +32,7 @@ def test_TProject_inputs():
             mandatory=True,
             position=1,
         ),
-        mask=dict(
-            argstr='-mask %s',
-            exist=True,
-        ),
+        mask=dict(argstr='-mask %s', ),
         noblock=dict(argstr='-noblock', ),
         norm=dict(argstr='-norm', ),
         num_threads=dict(

nipype/interfaces/base/__init__.py

@@ -10,7 +10,7 @@
 """
 from .core import (Interface, BaseInterface, SimpleInterface, CommandLine,
                    StdOutCommandLine, MpiCommandLine, SEMLikeCommandLine,
-                   PackageInfo)
+                   LibraryBaseInterface, PackageInfo)
 
 from .specs import (BaseTraitedSpec, TraitedSpec, DynamicTraitedSpec,
                     BaseInterfaceInputSpec, CommandLineInputSpec,

nipype/interfaces/base/core.py

@@ -1267,6 +1267,35 @@ def _format_arg(self, name, spec, value):
         return super(SEMLikeCommandLine, self)._format_arg(name, spec, value)
 
 
+class LibraryBaseInterface(BaseInterface):
+    _pkg = None
+    imports = ()
+
+    def __init__(self, check_import=True, *args, **kwargs):
+        super(LibraryBaseInterface, self).__init__(*args, **kwargs)
+        if check_import:
+            import importlib
+            failed_imports = []
+            for pkg in (self._pkg,) + tuple(self.imports):
+                try:
+                    importlib.import_module(pkg)
+                except ImportError:
+                    failed_imports.append(pkg)
+            if failed_imports:
+                iflogger.warn('Unable to import %s; %s interface may fail to '
+                              'run', failed_imports, self.__class__.__name__)
+
+    @property
+    def version(self):
+        if self._version is None:
+            import importlib
+            try:
+                self._version = importlib.import_module(self._pkg).__version__
+            except (ImportError, AttributeError):
+                pass
+        return super(LibraryBaseInterface, self).version
+
+
 class PackageInfo(object):
     _version = None
     version_cmd = None

nipype/interfaces/base/tests/test_auto_LibraryBaseInterface.py

@@ -0,0 +1,17 @@
+# AUTO-GENERATED by tools/checkspecs.py - DO NOT EDIT
+from __future__ import unicode_literals
+from ..core import LibraryBaseInterface
+
+
+def test_LibraryBaseInterface_inputs():
+    input_map = dict(
+        ignore_exception=dict(
+            deprecated='1.0.0',
+            nohash=True,
+            usedefault=True,
+        ), )
+    inputs = LibraryBaseInterface.input_spec()
+
+    for key, metadata in list(input_map.items()):
+        for metakey, value in list(metadata.items()):
+            assert getattr(inputs.traits()[key], metakey) == value

nipype/interfaces/cmtk/base.py

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+# emacs: -*- mode: python; py-indent-offset: 4; indent-tabs-mode: nil -*-
+# vi: set ft=python sts=4 ts=4 sw=4 et:
+""" Base interface for cmtk """
+
+from ..base import LibraryBaseInterface
+from ...utils.misc import package_check
+
+
+class CFFBaseInterface(LibraryBaseInterface):
+    _pkg = 'cfflib'
+
+
+# Originally set in convert, nbs, nx, parcellation
+# Set here to be imported, in case anybody depends on its presence
+# Remove in 2.0
+have_cmp = True
+try:
+    package_check('cmp')
+except ImportError:
+    have_cmp = False
+
+have_cfflib = True
+try:
+    package_check('cfflib')
+except ImportError:
+    have_cfflib = False
+
+have_cv = True
+try:
+    package_check('cviewer')
+except ImportError:
+    have_cv = False

nipype/interfaces/cmtk/convert.py

@@ -8,18 +8,10 @@
 import string
 import networkx as nx
 
-from ...utils.misc import package_check
 from ...utils.filemanip import split_filename
-from ..base import (BaseInterface, BaseInterfaceInputSpec, traits, File,
+from ..base import (BaseInterfaceInputSpec, traits, File,
                     TraitedSpec, InputMultiPath, isdefined)
-
-have_cfflib = True
-try:
-    package_check('cfflib')
-except Exception as e:
-    have_cfflib = False
-else:
-    import cfflib as cf
+from .base import CFFBaseInterface, have_cfflib
 
 
 class CFFConverterInputSpec(BaseInterfaceInputSpec):
@@ -67,7 +59,7 @@ class CFFConverterOutputSpec(TraitedSpec):
     connectome_file = File(exists=True, desc='Output connectome file')
 
 
-class CFFConverter(BaseInterface):
+class CFFConverter(CFFBaseInterface):
     """
     Creates a Connectome File Format (CFF) file from input networks, surfaces, volumes, tracts, etcetera....
 
@@ -87,6 +79,7 @@ class CFFConverter(BaseInterface):
     output_spec = CFFConverterOutputSpec
 
     def _run_interface(self, runtime):
+        import cfflib as cf
         a = cf.connectome()
 
         if isdefined(self.inputs.title):
@@ -232,7 +225,7 @@ class MergeCNetworksOutputSpec(TraitedSpec):
         exists=True, desc='Output CFF file with all the networks added')
 
 
-class MergeCNetworks(BaseInterface):
+class MergeCNetworks(CFFBaseInterface):
     """ Merges networks from multiple CFF files into one new CFF file.
 
     Example
@@ -248,6 +241,7 @@ class MergeCNetworks(BaseInterface):
     output_spec = MergeCNetworksOutputSpec
 
     def _run_interface(self, runtime):
+        import cfflib as cf
         extracted_networks = []
 
         for i, con in enumerate(self.inputs.in_files):

nipype/interfaces/cmtk/nbs.py

@@ -10,19 +10,11 @@
 import networkx as nx
 
 from ... import logging
-from ...utils.misc import package_check
-from ..base import (BaseInterface, BaseInterfaceInputSpec, traits, File,
+from ..base import (LibraryBaseInterface, BaseInterfaceInputSpec, traits, File,
                     TraitedSpec, InputMultiPath, OutputMultiPath, isdefined)
+from .base import have_cv
 iflogger = logging.getLogger('interface')
 
-have_cv = True
-try:
-    package_check('cviewer')
-except Exception as e:
-    have_cv = False
-else:
-    import cviewer.libs.pyconto.groupstatistics.nbs as nbs
-
 
 def ntwks_to_matrices(in_files, edge_key):
     first = nx.read_gpickle(in_files[0])
@@ -92,7 +84,7 @@ class NetworkBasedStatisticOutputSpec(TraitedSpec):
         desc='Output network with edges identified by the NBS')
 
 
-class NetworkBasedStatistic(BaseInterface):
+class NetworkBasedStatistic(LibraryBaseInterface):
     """
     Calculates and outputs the average network given a set of input NetworkX gpickle files
 
@@ -111,11 +103,10 @@ class NetworkBasedStatistic(BaseInterface):
     """
     input_spec = NetworkBasedStatisticInputSpec
     output_spec = NetworkBasedStatisticOutputSpec
+    _pkg = 'cviewer'
 
     def _run_interface(self, runtime):
-
-        if not have_cv:
-            raise ImportError("cviewer library is not available")
+        from cviewer.libs.pyconto.groupstatistics import nbs
 
         THRESH = self.inputs.threshold
         K = self.inputs.number_of_permutations

nipype/interfaces/cmtk/nx.py

@@ -14,20 +14,12 @@
 
 from ... import logging
 from ...utils.filemanip import split_filename
-from ...utils.misc import package_check
 from ..base import (BaseInterface, BaseInterfaceInputSpec, traits, File,
                     TraitedSpec, InputMultiPath, OutputMultiPath, isdefined)
+from .base import have_cmp
 
 iflogger = logging.getLogger('interface')
 
-have_cmp = True
-try:
-    package_check('cmp')
-except Exception as e:
-    have_cmp = False
-else:
-    import cmp
-
 
 def read_unknown_ntwk(ntwk):
     if not isinstance(ntwk, nx.classes.graph.Graph):