B d @s0dZd ddZddZddZedkr,edS) zRun Python code using ``mpi4py``. Run Python code (scripts, modules, zip files) using the ``runpy`` module. In case of an unhandled exception, abort execution of the MPI program by calling ``MPI.COMM_WORLD.Abort()``. Ncsddlddlm}m}dfdd }|dk r2|n jddjdd<jdd krtj}||d d d d njddkrjd}||d ddd nvjdd kr̈jd=|jdd ddnJddlm }m }t j dds||jdj d<|jdd ddS)agRun command line ``[pyfile | -m mod | -c cmd | -] [arg] ...``. * ``pyfile`` : program read from script file * ``-m mod`` : run library module as a script * ``-c cmd`` : program passed in as a command string * ``-`` : program read from standard input (``sys.stdin``) * ``arg ...``: arguments passed to program in ``sys.argv[1:]`` N) run_modulerun_path-ccsBddlm}jdkrdnd}t||ddd}||||f||iS)Nr)_run_module_code) script_nameZ mod_fnameexec)runpyr version_infocompile)stringZ init_globalsrun_namefilenameargv0rZkargcode)sysW/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/mpi4py/run.py run_strings z$run_command_line..run_stringr -__main__z)rrrz-mT)rZ alter_sys)realpathdirnameisolated)r)NNrr)rr rrargvstdinreadpopos.pathrrgetattrflagspath)argsrrrcmdrrr)rrrun_command_line s$ $  r'cCshddl}t|tr|n|dkr"dnd}tjp8tdd}|j|d}|dk r`|r`| || S)aTerminate MPI execution environment at Python exit. Terminate MPI execution environment at Python exit by calling ``MPI.COMM_WORLD.Abort(status)``. This function should be called within an ``except`` block. Afterwards, exceptions should be re-raised. rNr .z.MPI) r isinstanceint__spec__parent__name__ rpartitionmodulesgetZ_set_abort_statusexc_info)statusrpkgZmpirrrset_abort_status2s   r4c sddlddltjfddd fdd d fdd }d d }|\}}||y t|WnBtk r}zt|jWdd}~XYntd YnXdS)z-Entry-point for ``python -m mpi4py.run ...``.rNcs*ddlm}t|jdddS)Nr ) __version__)filer)r5printstdoutexit)r5)packagerrrversionPs zmain..versioncsddlm}tdkrd}n}jj}t||d}|djf|}|djf|}|d}|rt |j d t |j d t |j d  d n&t |j d t |j d  ddS) Nr)dedentrz.run)progpythonz usage: {python} -m {prog} [options] [arg] ... or: {python} -m {prog} [options] -m [arg] ... or: {python} -m {prog} [options] -c [arg] ... or: {python} -m {prog} [options] - [arg] ... zC Try `{python} -m {prog} -h` for more information. an options: --version show version number and exit -h|--help show this help message and exit -rc set 'mpi4py.rc.key=value' -p|--profile use for profiling --mpe profile with MPE --vt profile with VampirTrace )r6r ) textwrapr=r-r$basename executabledictstripformatr8stderrr:r9)Zerrmessr=Z prog_nameZ python_exesubsZcmdlineZhelptipoptions)osr;rrrusageUs(     zmain..usagec s,Gddd}fdd}|}|dkr6jddn |dd}x|r|ddr|ddkrjP|dd kr||dd kry,|d}|d rd |kr|ddd \}}}|d kr||g}|dd<n |dd}|dkrnx||dD]`}|d \} }} | r(| s0t|yt| i|j| <Wn tk rd| |j| <YnXqWnF|dkr||pd|_n*|dkr|dd|_nd|d|d=WqFt k rd|dYqFXqFW|sdn,|ddkr$t |dkr$d|d||fS)Nc@seZdZiZdZdS)z1main..parse_command_line..OptionsN)r- __module__ __qualname__rc_argsprofilerrrrOptionssrOcs4t|dks|ddr*d|d|dS)Nr rzArgument expected for option: r)len startswithr )r%)rJrrpopargsz0main..parse_command_line..popargr rr)z-mz-cr)z-hz-helpz--help)z-versionz --versionz--=)z-rcz-profilez-rc,)z-pz-profile)z-mpez-vtzUnknown option: zCannot parse option: zNo path specified for execution)z-mz-crPzArgument expected for option: ) rrR partitionsplit ValueErrorevalrM NameErrorrN ExceptionrQ) r%rOrSrHZarg0opt_argentrykeyval)rrJr<rrparse_command_line}sR "           z main..parse_command_linecSs>|jrddlm}|f|j|jr:ddlm}||jdS)Nr )rc)rN)rMr7rcrN)rHrcrNrrr bootstraps    zmain..bootstrapr )N)N)rIrr+r,r' SystemExitr4r)rbrdrHr%excr)rIr;rrJr<rmainFs"(:   rgr)N)__doc__r'r4rgr-rrrrs  '