3
d @ s d Z ddlmZ ddlZddlmZ ddlmZmZm Z m
Z
m Z ddl m
Z
ejd Zdd d ZG d
d de ZG dd deZG dd deZdS )an
The niftyreg module provides classes for interfacing with `niftyreg
`_ command line tools.
These are the base tools for working with niftyreg.
Registration tools are found in niftyreg/reg.py
Every other tool is found in niftyreg/regutils.py
Examples
--------
See the docstrings of the individual classes for examples.
)
StrictVersionN )logging ) CommandLineCommandLineInputSpectraits Undefined PackageInfo)split_filenameznipype.interface NIFTYREGDIRc C s t jjt j|d| S )N )ospathjoingetenv)commandZenv_dir r A/tmp/pip-build-7vycvbft/nipype/nipype/interfaces/niftyreg/base.pyget_custom_path s r c @ s$ e Zd Zedd Zedd ZdS )InfoZ
reg_aladinz
--versionc C s | S )Nr )Zraw_infor r r
parse_version# s zInfo.parse_versionN)__name__
__module__ __qualname__r Z version_cmd staticmethodr r r r r r s r c @ s0 e Zd ZdZejeejj ddddddZ
dS ) NiftyRegCommandInputSpecz#Input Spec for niftyreg interfaces.OMP_NUM_THREADS1z Number of openmp thread to usez-omp %iT)descZargstrZ
usedefaultN) r r r __doc__r ZIntintr environget omp_core_valr r r r r ( s r c sp e Zd ZdZdZdZeZd fdd Zdd Z d d
Z
d d Z e d
d Z
dd Z fddZdddZ ZS )NiftyRegCommandz7
Base support interface for NiftyReg commands.
Z_nrz1.5.30Nc s d| _ tt| jf | || _| j}|r| jd k rXt|t| jk rXd}tj ||| j |d k rt|t|krd}|d7 }tj ||| j | j
j | j d | j
j | j
d | j d S )N z1A later version of Niftyreg is required (%s < %s)z1The version of NiftyReg differs from the requiredz
(%s != %s)r$ r" ) num_threadssuperr% __init__required_versionversion _min_versionr ifloggerwarninginputsZon_trait_change _omp_update_environ_update)selfr* r/ _versionmsg) __class__r r r) > s$ zNiftyRegCommand.__init__c C sH | j jr(t| j j| j jd< | j j| _n d| j jkr>| j jd= d| _d S )Nr r& )r/ r$ strr" r' )r2 r r r r0 R s
zNiftyRegCommand._omp_updatec C s@ | j jr4d| j jkr*t| j jd | j _q s