3 d*@sdZddlZddlZddlZddlmZddlmZddlm Z ddl Z ddl Z ddZ ddd Zd d Zd d ZddZdde eefddZddZddZddZddZd ddZd!ddZdS)"z Miscellaneous utility functions N)Iterator)warn) LooseVersioncs ddfdd}t||dS)zASorts string in human order (i.e. 'stat10' will go after 'stat2')cSs|jrt|S|S)N)isdigitint)textr3/tmp/pip-build-7vycvbft/nipype/nipype/utils/misc.pyatoisz human_order_sorted..atoics,t|tr|d}fddtjd|DS)Nrcsg|] }|qSrr).0c)r rr sz.natural_keys..z(\d+)) isinstancetupleresplit)r)r rr natural_keyss z(human_order_sorted..natural_keys)key)sorted)lrr)r r human_order_sorteds rcsFt|trt|d}|sdS|jj}tj}x6|ddD]&}|jr.: )r)rbytesstr expandtabs splitlinessysmaxsizelstripminlenstriprstripallreplaceappendpopjoin)Z docstringmarkerlinesindentlineZtrimmedr)rr trim!s2      r2cCstjtj|\}|S)z1Return the indices where ravel(condition) is true)npZnonzeroZravel) conditionresrrr find_indicesGsr6cCs$t|trdSt|drdSdSdS)zChecks if item is a container (list, tuple, dict, set) Parameters ---------- item : object object to check for .__iter__ Returns ------- output : Boolean True if container False if not (eg string) F__iter__TN)rrhasattr)itemrrr is_containerMs   r:cCs(t|dr t|t r dj|}t|S)aConvert a container to a command line string. Elements of the container are joined with a space between them, suitable for a command line parameter. If the container `cont` is only a sequence, like a string and not a container, it is returned unmodified. Parameters ---------- cont : container A container object like a list, tuple, dict, or a set. Returns ------- cont_str : string Container elements joined into a string. r7 )r8rrr-)Zcontrrr container_to_stringcs r<c Cs|rd||f}nd|}|r,|d|f7}y t|}Wn,tk rd}z|||WYdd}~XnX|sndSy |j} Wn0tk r}z|d||WYdd}~XnX|| ||kr||dS)aCheck that the minimal version of the required package is installed. Parameters ---------- pkg_name : string Name of the required package. version : string, optional Minimal version number for required package. app : string, optional Application that is performing the check. For instance, the name of the tutorial being executed that depends on specific packages. Default is *Nipype*. checker : object, optional The class that will perform the version checking. Default is distutils.version.LooseVersion. exc_failed_import : Exception, optional Class of the exception to be thrown if import failed. exc_failed_check : Exception, optional Class of the exception to be thrown if version check failed. Examples -------- package_check('numpy', '1.3') package_check('scipy', '0.7', 'tutorial1') z%s requires %szNipype requires %sz with version >= %sNzCannot find version for %s) __import__ ImportError __version__AttributeError) pkg_nameversionZappcheckerZexc_failed_importZexc_failed_checkmsgmodeZ have_versionrrr package_check~s"#   rGcCs\t|tr|St|tr"|jd}t|trL|j}|dkr@dS|dkrLdStd|dS)a Convert strings (and bytearrays) to boolean values >>> all([str2bool(v) for v in (True, "yes", "true", ... "y", "t", "Yes", "True", "1", "on", "On")]) True >>> all([str2bool(v.encode('utf-8')) ... for v in ("yes", "true", "y", "t", "1", "Yes", "on", "On")]) True >>> any([str2bool(v) for v in (False, "no", "false", "n", "f", ... "False", "0", "off", "Off")]) False >>> any([str2bool(v.encode('utf-8')) ... for v in ("no", "false", "n", "f", "0", "off", "Off")]) False >>> str2bool(None) # doctest: +ELLIPSIS Traceback (most recent call last): ... ValueError: ... >>> str2bool('/some/path') # doctest: +ELLIPSIS Traceback (most recent call last): ... ValueError: ... >>> str2bool('Agg') # doctest: +ELLIPSIS Traceback (most recent call last): ... ValueError: ... >>> str2bool('INFO') # doctest: +ELLIPSIS Traceback (most recent call last): ... ValueError: ... >>> str2bool('/some/bytes/path'.encode('utf-8')) # doctest: +ELLIPSIS Traceback (most recent call last): ... ValueError: ... zutf-8yestrueyt1onTnofalsenf0offFz%r cannot be converted to boolN)rHrIrJrKrLrM)rNrOrPrQrRrS)rboolrdecoderlower ValueError)vrVrrr str2bools&    rYcCsR|gkr |St|dtr6t|dt|ddS|ddt|ddS)Nrr)rlistflatten)Srrr r[s r[cCsJt|tst|}t|ts$t|Sg}x|D]}|jt||q.W|S)N)rriterrZnextr+ unflatten)Zin_listZprev_structureoutr9rrr r_s   r_cCs|jdkr"|ddddddg}n|jdkrt|tjddddddgt|d k}|dd tjd |dd <nX|jd krddlm}m}||}tjd }|d ddf|d d<|||ddd<|S)aN Normalize a single row of motion parameters to the SPM format. SPM saves motion parameters as: x Right-Left (mm) y Anterior-Posterior (mm) z Superior-Inferior (mm) rx Pitch (rad) ry Roll (rad) rz Yaw (rad) ZFSLrrAFNIFSFASTNgf@ZNIPY) to_matrix44 aff2euler)rerfrr) upperr3Zasarrayr&piZnipy.algorithms.registrationrhrizeros)paramssourcerhriZmatrixrrr normalize_mc_paramss  &   roc s@yt|t|}}Wn,tk rBtjd|d|d|SXt|j}t|j}g}||rz|d||g7}||r|d||g7}|r|jddt|}dd }fd d xN|j|D]@}||} ||} | | kr|d ||| || fg7}qWt||kr*|j|d tjdj |d|S)zHelper to log what actually changed from old to new values of dictionaries. typical use -- log difference for hashed_inputs z4Diff between nipype inputs failed: * Cached inputs: z * New inputs: r;z * keys not previously seen: %sz * keys not presently seen: %srz Dictionaries had differing keys:cSs|t|tr4t|dkr4|ddd|ddSt|ttfrxt|dkrxtt|dddgt|ddS|S)N2 z...rdi)rrr&rrZ)valuerrr _shorten?s *zdict_diff.._shortencsHt|tr fdd|jDSt|ttfrDtfdd|DS|S)Ncsi|]\}}||qSrr)r krX) _uniformizerr Hsz2dict_diff.._uniformize..c3s|]}|VqdS)Nr)r el)rvrr Jsz1dict_diff.._uniformize..)rdictitemsrZr)val)rvrr rvFs  zdict_diff.._uniformizez * %s: %r != %rz-Some dictionary entries had differing values:r) rz Exceptiontextwrapr0setkeysinsertr& intersectionr-) ZdoldZdnewr0Znew_keysZold_keysZdiffZdiffkeysrtrunewoldr)rvr dict_diffs4         rTcCsl|r tjSy tj}WnNtk rf}z2tjd}|dkrJt|jdftd|WYdd}~XnX|S)zz Robust replacement for getcwd when folders get removed If error==True, this is just an alias for os.getcwd() ZPWDNzbCurrent directory does not exist anymore, and nipype was not able to guess it from the environmentz;Current folder does not exist, replacing with "%s" instead.)osgetcwdOSErrorgetenverrnor)errorcwdexcrrr rgetcwd_s  r)N)r)T)__doc__rr"rcollections.abcrwarningsrdistutils.versionrnumpyr3r~rr2r6r:r<r> RuntimeErrorrGrYr[r_rorrrrrr s0    & 16  C