U jg2@sdZddlZzddlZddlZWnek r<edYnXGdddZGdddZGdd d ZGd d d ZGd d d Z ddZ ej e e dS)z(gdb printing extension for Numba types. Nz$GDB python support is not available.c@seZdZddZddZdS)NumbaArrayPrintercCs ||_dSNvalselfrrB/tmp/pip-unpacked-wheel-qtpwf23r/numba/misc/gdb_print_extension.py__init__szNumbaArrayPrinter.__init__c) Cszddl}d}Wntk r(d}YnXzd}|jd}|jd}|jd}|jd}d}t|jj} |rd| ksd | kr| d d } td } d d| |  D} | \} } }d}t || }| \}}}|dk}| d}g}|D]}| d}|d}|dd}t |dkr4t d|n@|d}d|krRt d|| dd dd}||}|||fq|} nJtd} dd| |  D} | \} } }d| kr| dddd} dd}||}||}||kr|r|dd|D}||} ||| }!|!t|7}!|j| |d }"|"}t}#|#t||!}$|j|$|d!}%|jjj|%||d"}&d#t|&WSd$| |fWStd%g}'d&d'|'d(d)WSWn0tk r}(zd*|(WYSd}(~(XYnXdS)+NrTFdataitemsizeshapestridesZalignedZRecordz unaligned z%array\((Record.*), (.*), (.*)\)\ \(.*cSsg|] }|qSrstrip.0xrrr 8sz/NumbaArrayPrinter.to_string..z(Record\((.*\[.*\]);([0-9]+);(True|False)True,[zUnsupported sub-type: %sZ nestedarray;=zarray\((.*),(.*),(.*)\)\ \(.*cSsg|] }|qSrrrrrr rRsz unichr x z.dwarr2inttuple..r)typefieldsrangetuple)rr!lohirrr dwarr2inttupleXs z3NumbaArrayPrinter.to_string..dwarr2inttuplecSsg|]}td|dqS)rr)maxrrrr rhs)Zalign)dtype)r r z array([...], dtype=%s, shape=%s)zNULL/Uninitializedzarray([, ])z%array[Exception: Failed to parse. %s])Znumpy ImportErrorrstrr replacerrecompilematchgroupssplitlen TypeErrorr(appendarraysumrgdbselected_inferior read_memoryZ frombufferlibZ stride_tricksZ as_stridedlistjoin Exception))rnpZ HAVE_NUMPYNULLr r ZrshpZrstridesZ is_alignedZty_strmatcherZarr_infoZ dtype_strZndim_strZ order_strZrstrZ rstr_matchr!ZbalignZis_aligned_strZ field_dtsZstruct_entriesfZsplittednameZdt_partZ dt_as_strr(r&r rZshp_arrZstrd_arrZextentZ dtype_clazz this_procmemZarr_dataZnew_arrbuferrr to_strings                       zNumbaArrayPrinter.to_stringN__name__ __module__ __qualname__r rJrrrr r src@seZdZddZddZdS)NumbaComplexPrintercCs ||_dSrrrrrr r szNumbaComplexPrinter.__init__cCsd|jd|jdfS)Nz%s+%sjrealimagr)rrrr rJszNumbaComplexPrinter.to_stringNrKrrrr rOsrOc@seZdZddZddZdS)NumbaTuplePrintercCs ||_dSrrrrrr r szNumbaTuplePrinter.__init__cCs>g}|jj}|D]}|t|j|jqdd|S)N(%s)r*)rr r!r7r.rEr?)rrHr!rDrrr rJs  zNumbaTuplePrinter.to_stringNrKrrrr rRsrRc@seZdZddZddZdS)NumbaUniTuplePrintercCs ||_dSrrrrrr r szNumbaUniTuplePrinter.__init__cCsX|jj}|dj\}}g}t||dD]}|t|j|q0dd|S)NrrrSr*)rr r!r"r7r.r?)rr!r$r%rHirrr rJs  zNumbaUniTuplePrinter.to_stringNrKrrrr rTsrTc@seZdZddZddZdS)NumbaUnicodeTypePrintercCs ||_dSrrrrrr r sz NumbaUnicodeTypePrinter.__init__cCszd}|jd}|jd}|jd}||krjt}|t|||}t|tr^t|}qr|d}nt |}d|S)Nrr lengthkindzutf-8z'%s') rr:r;r<r isinstance memoryviewbytesdecoder.)rrBr ZnitemsrXrFrGrHrrr rJs     z!NumbaUnicodeTypePrinter.to_stringNrKrrrr rVsrVcCsdtjd}|ddt|ddt|ddt|dd t|d d t|d d t|S)NZNumbazNumba unaligned array printerz^unaligned array\(zNumba array printerz^array\(zNumba complex printerz^complex[0-9]+\ zNumba Tuple printerz^Tuple\(zNumba UniTuple printerz ^UniTuple\(zNumba unicode_type printerz^unicode_type\s+\() r:printingZRegexpCollectionPrettyPrinterZ add_printerrrOrRrTrV)printerrrr _create_printerss$ r_)__doc__r0Z gdb.printingr:r-rrOrRrTrVr_r]Zregister_pretty_printerZcurrent_objfilerrrr s s