3 d@srdZddlZddlmZddlmZmZmZmZm Z m Z m Z ddZ e dkZ Gd d d eZGd d d eZdS) z!Interfaces to run MATLAB scripts.N)config)CommandLineInputSpecInputMultiPath isdefined CommandLinetraitsFile DirectorycCs*dtjkr&ddlm}|tjddSdS)z:Determine whether Matlab is installed and can be executed.ZNIPYPE_NO_MATLABr)whichZ MATLABCMDmatlabN)osenvironZnipype.utils.filemanipr getenv)r r:/tmp/pip-build-7vycvbft/nipype/nipype/interfaces/matlab.pyget_matlab_commands  rc@seZdZdZejdddd(dZejddd d gdd Zejdd dd ddZ ejddddddZ e dddZ ejddddZ ejddddZe ddddZeeddZejddgdd dZejd!d"d#d$d%gd&ddZd'S))MatlabInputSpecz)Basic expected inputs to Matlab interfacez -r "%s;exit"z m-code to runTr)argstrdesc mandatorypositionzuse MCR interface nodesktopnosplashsingle_comp_thread)rxornohashz -nodesktopz)Switch off desktop mode on unix platforms)r usedefaultrrz -nosplashzSwitch of splash screenz -logfile %szSave matlab output to log)rrz-singleCompThreadzforce single threaded operation)rrrzRun m-code using m-file)rrz pyscript.mzName of file to write m-code to)rrzPaths to add to matlabpath)rzver,ztry,z!prescript to be added before codez ,catch ME,z/fprintf(2,'MATLAB code threw an exception:\n');zfprintf(2,'%s\n',ME.message);zsif length(ME.stack) ~= 0, fprintf(2,'File:%s\nName:%s\nLine:%d\n',ME.stack.file,ME.stack.name,ME.stack.line);, end;zend;zscript added after codeN)__name__ __module__ __qualname____doc__r StrscriptZBooluses_mcrrrr logfilermfile script_filerr pathsList prescript postscriptrrrrrsH   rcszeZdZdZdZdZdZdZeZ dfdd Z e ddZ e dd Z e d d Zfd d ZfddZddZZS) MatlabCommandzInterface that runs matlab code >>> import nipype.interfaces.matlab as matlab >>> mlab = matlab.MatlabCommand(mfile=False) # don't write script file >>> mlab.inputs.script = "which('who')" >>> out = mlab.run() # doctest: +SKIP r Nc stt|jf||r&t|r&||_n|jr4|j|_|jrRt|jj rR|j|j_|j rpt|jj  rp|j |j_ t|jj  rt|jj  rt jddrd|j_ d|_dS)zXinitializes interface to matlab (default 'matlab -nodesktop -nosplash') Z executionZsingle_thread_matlabT allatonceN)superr.__init__r_cmd_default_matlab_cmd_default_mfileinputsr(_default_pathsr*rr&r getbooleanterminal_output)self matlab_cmdr5) __class__rrr1`s     zMatlabCommand.__init__cCs ||_dS)a&Set the default MATLAB command line for MATLAB classes. This method is used to set values for all MATLAB subclasses. However, setting this will not update the output type for any existing instances. For these, assign the .inputs.matlab_cmd. N)r3)clsr:rrrset_default_matlab_cmdys z$MatlabCommand.set_default_matlab_cmdcCs ||_dS)a'Set the default MATLAB script file format for MATLAB classes. This method is used to set values for all MATLAB subclasses. However, setting this will not update the output type for any existing instances. For these, assign the .inputs.mfile. N)r4)r<r(rrrset_default_mfiles zMatlabCommand.set_default_mfilecCs ||_dS)aSet the default MATLAB paths for MATLAB classes. This method is used to set values for all MATLAB subclasses. However, setting this will not update the output type for any existing instances. For these, assign the .inputs.paths. N)r6)r<r*rrrset_default_pathss zMatlabCommand.set_default_pathsc sJd|_tt|j|}ytjdWn YnXd|jkrF|j||S)Nr/z stty sanezMATLAB code threw an exception)r8r0r._run_interfacersystemstderrZraise_exception)r9Zruntime)r;rrr@s  zMatlabCommand._run_interfacecs:|dkr&|j}|jjrd}|j||Stt|j|||S)Nr%z%s)r%)rr5r&_gen_matlab_commandr0r. _format_arg)r9nameZ trait_specvaluer)r;rrrDs  zMatlabCommand._format_argc Cs:tj}|jjp|jj}g}t|jjr0|jj}|jj}|jj}|rR|j ddn |j ddx|D]}|j d|qdW|sdj dd|j dD}dj ||dj |}|r"t tjj ||jjd }|j|Wd QRX|jjrd tjj ||jj} nd ||jjj d df} ndj |j d} || S)z>Generates commands and, if mfile specified, writes it to disk.rzsz5MatlabCommand._gen_matlab_command.. wtNz%szaddpath('%s');%s.)rgetcwdr5r(r&rr*r,r-insertappendjoinsplitopenpathr)write) r9rZ script_linescwdr(r*r,r-rXr%rrrrCs8     z!MatlabCommand._gen_matlab_command)N)r r!r"r#r2r3r4r6rZ input_specr1 classmethodr=r>r?r@rDrC __classcell__rr)r;rr.Qs r.)r#rrQrbaserrrrr r r rZ no_matlabrr.rrrrs $  3