3 d<@s,dZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z mZd d lmZmZmZd d lmZmZmZmZmZmZmZd d lmZmZdjZe eZ Gdddej!Z"Gddde"Z#Gddde#Z$Gddde"Z%Gddde$Z&Gddde&Z'Gddde&Z(ddZ)dS)z Base I/O specifications for Nipype interfaces ............................................. Define the API for the I/O of interfaces N)isclass)deepcopy)warn)Version) TraitError)TraitDictObjectTraitListObject)md5 hash_infilehash_timestamp)traitsFileStr Undefined isdefined has_metadataOutputMultiObject)config __version__z{:.10f}cseZdZdZeZfddZddZddZdd Z d d Z d d Z fddZ e Z fddZd!ddZd"ddZd#ddZd$ddZeddZfdd ZZS)%BaseTraitedSpeca Provide a few methods necessary to support nipype interface api The inputs attribute of interfaces call certain methods that are not available in traits.HasTraits. These are provided here. new metadata: * usedefault : set this to True if the default value of the trait should be used. Unless this is set, the attributes are set to traits.Undefined new attribute: * get_hashval : returns a tuple containing the state of the trait as a dict and hashvalue corresponding to dict. XXX Reconsider this in the long run, but it seems like the best solution to move forward on the refactoring. c svtt|jf|tjddi}x&|jD]}|j|js,t||<q,W|jfddi||j |jf|dS)zInitialize handlers and inputsT)Zreraise_exceptionstrait_change_notifyFN) superr__init__rZpush_exception_handlercopyable_trait_names usedefaultr trait_set_generate_handlers)selfkwargsZundefined_traitstrait) __class__>/tmp/pip-build-7vycvbft/nipype/nipype/interfaces/base/specs.pyr<s  zBaseTraitedSpec.__init__ccs,x&t|jD]}||j|fVqWdS)z0Name, trait generator for user modifiable traitsN)sortedrr)rnamer#r#r$itemsLszBaseTraitedSpec.itemscCsDg}x.t|jjD]\}}|jd||fqWdjdj|S)z4Return a well-formatted representation of the traitsz%s = %sz {}  )r% trait_getr'appendformatjoin)rZoutstrr&valuer#r#r$__repr__QszBaseTraitedSpec.__repr__cCsptddd}|jf|}x|D]}|j|j|q Wtddd}|jf|}x|D]}|j|j|qVWdS)z]Find all traits with the 'xor' metadata and attach an event handler to them. cSs|dk S)Nr#)tr#r#r$\sz4BaseTraitedSpec._generate_handlers..)xorcSs|dk S)Nr#)r/r#r#r$r0`s) deprecatedN)dictZ trait_namesZon_trait_change _xor_warn_deprecated_warn)rZhas_xorZxorselemZhas_deprecationr2r#r#r$rXs    z"BaseTraitedSpec._generate_handlerscCspt|rl|j|}xV|jD]L}||kr*qtt||r|jfddid|tid||f}t|qWdS)z!Generates warnings for xor traitsrFz%szFInput "%s" is mutually exclusive with input "%s", which is already setN)rrr1getattrrrIOError)robjr&oldnew trait_specZ trait_namemsgr#r#r$r4es   zBaseTraitedSpec._xor_warnc Cst|r|j|}d||jjjddf}d|j}|jrj|j|jkr^t|d|jd|j}nd}dj |||f} t t |j|j krt| nL|jr| d ||jf7} t | |jr|jfd d id |td |j|id S)z6Checks if a user assigns a value to a deprecated traitz'Input %s in interface %s is deprecated.Z InputSpecrz2Will be removed or raise an error as of release %sz Replacement trait %s not foundzIt has been replaced by %s. z-Unsetting old value %s; setting new value %s.rFz%sN)rrr"__name__splitr2new_namerrr,rstrpackage_versionrrr) rr9r&r:r;r<Zmsg1Zmsg2Zmsg3r=r#r#r$r5xs2    z BaseTraitedSpec._deprecated_warnc s"tt|jf|}|j|t}|S)zReturns traited class as a dict Augments the trait get function to return a dictionary without notification handles )rrr)_clean_containerr)rr out)r"r#r$r)s zBaseTraitedSpec.trait_getc s$tt|jf|}|j|dd}|S)zReturns traited class as a dict Augments the trait get function to return a dictionary without any traits. The dictionary does not contain any attributes that were Undefined T) skipundefined)rrr)rE)rr rF)r"r#r$get_traitsfreeszBaseTraitedSpec.get_traitsfreeNFcCst|tst|trZi}xt|jD].\}}t|rH|j||||<q&|s&|||<q&Wnt|tsxt|tsxt|trg}xB|D]:}t|r|j |j||q|s|j |q|j dqWt|trt|}nd}t|r|}n|s|}|S)z;Convert a traited obejct into a pure python representation.N) isinstancerr3listr'rrErtupler*)robjektZ undefinedvalrGrFkeyvalr#r#r$rEs2       z BaseTraitedSpec._clean_containerTcCst|j|j|||S)z\ Return has_metadata for the requested trait name in this interface )rr!Z trait_type)rr&metadatar- recursiver#r#r$rszBaseTraitedSpec.has_metadatac Csg}g}xt|jjD]v\}}t| s|j|ddr>> class OutputSpec(TraitedSpec): ... out = OutputMultiObject(traits.List(traits.Int)) >>> spec = OutputSpec() >>> spec.out = [[4]] >>> spec.out [4] >>> spec.__getstate__()['out'] [[4]] >>> spec.__setstate__(spec.__getstate__()) >>> spec.out [4] ) rr __getstate__rdr!Z is_trait_typerhandler get_value)rstaterMZ _trait_spec)r"r#r$reLs    zBaseTraitedSpec.__getstate__)NF)NT)N)FNT)r@ __module__ __qualname____doc__nipype_versionrDrr'r.rr4r5r)r^rHrErrWrTpropertyrdre __classcell__r#r#)r"r$r%s"  $  #  1 : rc@seZdZdZejZdS) TraitedSpeczQCreate a subclass with strict traits. This is used in 90% of the cases. N)r@rirjrkrZDisallow_r#r#r#r$rogsroc@s eZdZdS)BaseInterfaceInputSpecN)r@rirjr#r#r#r$rqpsrqc@seZdZdZddZdS)DynamicTraitedSpeczA subclass to handle dynamic traits This class is a workaround for add_traits and clone_traits not functioning well together. c Cst|}||kr||St|j|}x(|jD]}||jjkr0t||}q0W|j|d}x,|jD] }yt||}WqfYqfXqfW|j|d}|jf||S)z Replace the ``__deepcopy__`` member with a traits-friendly implementation. A bug in ``__deepcopy__`` for ``HasTraits`` results in weird cloning behaviors. )memo) idrr)r__dict__keysr7Z clone_traitsr)rrsZid_selfZdup_dictrMrpdupr#r#r$ __deepcopy__{s     zDynamicTraitedSpec.__deepcopy__N)r@rirjrkrxr#r#r#r$rrtsrrc@s(eZdZedddZejddddZdS)CommandLineInputSpecz%sz$Additional parameters to the command)argstrdesczEnvironment variablesT)r{rrQN)r@rirjrargsrZ DictStrStrenvironr#r#r#r$rys ryc@seZdZeddddZdS)StdOutCommandLineInputSpecz> %sr T)rzpositionZgenfileN)r@rirjrout_filer#r#r#r$r~sr~c@s(eZdZejddddZejddZdS)MpiCommandLineInputSpecFz.Whether or not to run the command with mpiexecT)r{rzeNum processors to specify to mpiexec. Do not specify if this is managed externally (e.g. through SGE))r{N)r@rirjrZBoolZuse_mpiZIntZn_procsr#r#r#r$rs rcCs|jdkrdSt| r*t|dr*|jg}t|r>|jn|j}tddd}x4t|jf|jD]\}}|j t||j dqhW|S)zhProvides information about file inputs to copy or link to cwd. Necessary for pipeline operation Nnormalize_filenamescSs|dk S)Nr#)r/r#r#r$r0sz#get_filecopy_info..)copyfile)rMcopy) Z input_specrhasattrrinputsr3r%rr'r*r)clsinforrOr&specr#r#r$get_filecopy_infos r)*rkr[inspectrrrwarningsrZpackaging.versionrZtraits.trait_errorsrZtraits.trait_handlersrrZutils.filemanipr r r Ztraits_extensionrrrrrrrr>rrr+rbrlZ HasTraitsrrorqrrryr~rrr#r#r#r$ s,     $ D "