o 8ήcb ã@s<ddlmZmZddlmZmZedƒGdd„deƒƒZdS)é)ÚerrorsÚir)Úregister_rewriteÚRewritezbefore-inferencec@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚRewriteConstRaisesa? Rewrite IR statements of the kind `raise(value)` where `value` is the result of instantiating an exception with constant arguments into `static_raise(exception_type, constant args)`. This allows lowering in nopython mode, where one can't instantiate exception instances from runtime data. cCst|tƒo t|tƒS)N)Ú isinstanceÚtypeÚ issubclassÚ Exception)ÚselfÚconst©r úG/tmp/pip-target-vg8gfxp4/lib/python/numba/core/rewrites/static_raise.pyÚ_is_exception_typesz%RewriteConstRaises._is_exception_typecCsxt|tƒr#| |d¡sd}t ||df|¡‚|dt|dƒfS| |¡r,|dfSt|tƒr4d}nd}t ||¡‚)z0 Break down constant exception. rz-Encountered unsupported exception constant %réNzDDirectly raising a string constant as an exception is not supported.z8Encountered unsupported constant type used for exception)rÚtuplerrÚUnsupportedErrorÚstr)r r ÚlocÚmsgr r rÚ_break_constants    z"RewriteConstRaises._break_constantc CsÂi|_}i|_}||_| tjtjf¡D]@}|jdur"d\}} n| |j¡} |jj } |  | | ¡\}} t |tjƒrA|| f||<qt |tjƒrN|| f||<qt d  t|ƒ¡ƒ‚t|ƒt|ƒdkS)N)NNzunexpected: {}r)ÚraisesÚ tryraisesÚblockÚ find_instsrÚRaiseÚTryRaiseÚ exceptionÚinfer_constantrrrÚ ValueErrorÚformatrÚlen) r Úfunc_irrÚtypemapÚ calltypesrrÚinstÚexc_typeÚexc_argsr rr r rÚmatch's       zRewriteConstRaises.matchcCs–|j ¡}| ¡|jjD];}||jvr)|j|\}}t |||j¡}| |¡q ||j vrC|j |\}}t  |||j¡}| |¡q | |¡q |S)zC Rewrite all matching setitems as static_setitems. ) rÚcopyÚclearÚbodyrrÚ StaticRaiserÚappendrÚStaticTryRaise)r Ú new_blockr%r&r'Únew_instr r rÚapply>s       zRewriteConstRaises.applyN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr(r1r r r rrs   rN)Ú numba.corerrÚnumba.core.rewritesrrrr r r rÚs