B \^@sddlmZmZmZddlZddlmZmZmZddl m Z m Z m Z m Z mZddZdd Zd d Ze ejejejee ejejejee ejejejee ejd d ZeejddZe ejejddZe ejejddZe ejeje ejejddZdS))print_functionabsolute_importdivisionN)typescgutilstyping) lower_cast lower_builtinlower_getattr_genericimpl_ret_untrackedlower_setattr_genericcCstjS)N)rtrue_bit)contextbuildersigargsr5lib/python3.7/site-packages/numba/targets/optional.pyalways_return_true_impl srcCstjS)N)r false_bit)rrrrrrralways_return_false_implsrc Csj|j\}}|\}}|tjkr0||}}||}}|}|} |||| } |t|| j} t|||j | S)z/ Check if an Optional value is invalid ) rrnone make_helpernot_r as_bool_bitvalidr return_type) rrrrZltyZrtyZlvalZrvalZopt_typeZopt_valZoptZresrrroptional_is_nones    rcCs2|j}|||||}|||}||||||S)z? Optional.__getattr__ => redirect to the wrapped type. )typecastZ get_getattr)rrtypvalueattrZ inner_typevalimprrroptional_getattr/s r&c CsR|j\}}|\}}|j} ||||| }t|j| |} ||| } | |||fS)z? Optional.__setattr__ => redirect to the wrapped type. )rrr rZ signaturerZ get_setattr) rrrrr#ZbasetyZvaltytargetr$Z target_typeZnewsigr%rrroptional_setattr:s  r(c Cs|j|||d}t||j}|||}||b\}} |&tj|_|||j|j|j|_WdQRX| tj |_t |jj|_WdQRXWdQRX| S)a The handling of optional->optional cast must be special cased for correct propagation of None value. Given type T and U. casting of T? to U? (? denotes optional) should always succeed. If the from-value is None, the None value the casted value (U?) should be None; otherwise, the from-value is casted to U. This is different from casting T? to U, which requires the from-value must not be None. )r"N) rrrrZif_elserr datarrZget_null_valueZ _getvalue) rrfromtytotyr$optvalvalidbitZ outoptvalZis_validZ is_not_validrrroptional_to_optionalIs    r.cCs>|tjkr|||jS|||||j}|||j|SdS)N)rrZmake_optional_nonerr Zmake_optional_value)rrr*r+r$rrrany_to_optionalhs r/c Csp|j|||d}t||j}|j||dd$d|jf}|j|t |fWdQRX| ||j |j|S)N)r"F)Zlikelyzexpected %s, got None) rrrrZif_thenrrZ call_convZreturn_user_exc TypeErrorr r))rrr*r+r$r,r-msgrrroptional_to_anyqs  r2)Z __future__rrroperatorZnumbarrrZimputilsr r r r r rrris_rZOptionalr&r(r.ZAnyr/ZBooleanr2rrrrs