3 d@szddlmZmZddlmZddlmZmZmZm Z m Z m Z ej dZ Gddde ZGdd d eZGd d d eZd S) )logging LooseVersion)which)CommandLineInputSpec CommandLinetraitsFile isdefined PackageInfoznipype.interfacec@s(eZdZdZeddZeddZdS)Infozmrconvert --versioncCsTx,|jD]}|jdr |jd}Pq WdS|jdd}dj|jdddS)Nz == mrconvert r-.r) splitlines startswithsplitjoin)Zraw_infolineZv_stringr@/tmp/pip-build-7vycvbft/nipype/nipype/interfaces/mrtrix3/base.py parse_versions  zInfo.parse_versioncCst|jp dS)zcReturn a comparable version object If no version found, use LooseVersion('0.0.0') z0.0.0)rversion)clsrrr looseversion%szInfo.looseversionN)__name__ __module__ __qualname__Z version_cmd staticmethodr classmethodrrrrrr s r c@seZdZejddddZeddddgdZejedd edd d d d gd Z ej dddddZ edd ddZ edddZ eddddZedddZdS)MRTrix3BaseInputSpecz -nthreads %dzEnumber of threads. if zero, the number of available cpus will be usedT)argstrdescZnohashz-grad %sz"dw gradient scheme (MRTrix format)grad_fsl)existsr!r"xor)r$z-fslgrad %s %sz.(bvecs, bvals) dw gradient scheme (FSL format) grad_file)r!r"r%yesnoz-bvalue_scaling %saDspecifies whether the b - values should be scaled by the square of the corresponding DW gradient norm, as often required for multishell or DSI DW acquisition schemes. The default action can also be set in the MRtrix config file, under the BValueScaling entry. Valid choices are yes / no, true / false, 0 / 1 (default: true).)r!r"zbvecs file in FSL format)r$r!r"zbvals file in FSL format)r$r"Fz-export_grad_fsl %s %szexport bvec file in FSL formatzexport bval file in FSL formatN)rrrrZIntnthreadsr r&Tupler#EnumZ bval_scalein_bvecin_bvalout_bvecout_bvalrrrrr .s<    r cs6eZdZfddZdfdd ZeddZZS) MRTrix3Basec s|dkrL|dkrLd}yddlm}|}WntjdYnX|j|S|dkrf|j||jjfS|dkr|j||jjfStt |j |||S)Nr)r) cpu_countz'Number of threads could not be computedr,r.) multiprocessingr2ifloggerwarningr!inputsr-r/superr0 _format_arg)selfnameZ trait_specvaluer2) __class__rrr8^s    zMRTrix3Base._format_argNc s|dkr g}yft|jjs&t|jjr2|ddg7}t|jj}t|jj}|sR|rp| s^| rftd|dg7}Wntk rYnXtt |j |dS)Nr,r-z6If using bvecs and bvals inputs, bothshould be defined)skip) r r6r&r#r,r- RuntimeErrorAttributeErrorr7r0 _parse_inputs)r9r=Zis_bvecZis_bval)r<rrr@qs    zMRTrix3Base._parse_inputscCstjS)N)r r)r9rrrrszMRTrix3Base.version)N)rrrr8r@propertyr __classcell__rr)r<rr0]s r0N)rrZutils.filemaniprbaserrrr r r getLoggerr4r r r0rrrrs   /