U hd2@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 ||_dSNvalselfrrW/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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)align)dtype)r r z array([...], dtype=%s, shape=%s)zNULL/Uninitializedzarray([, ])z%array[Exception: Failed to parse. %s])numpy ImportErrorrstrr replacerrecompilematchgroupssplitlen TypeErrorr)appendarraysumrgdbselected_inferior read_memoryZ frombufferlibZ stride_tricksZ as_stridedlistjoin Exception))rnpZ HAVE_NUMPYNULLr r ZrshpZrstridesZ is_alignedZty_strZmatcherZarr_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 rKrrrr r src@seZdZddZddZdS)NumbaComplexPrintercCs ||_dSrrrrrr r szNumbaComplexPrinter.__init__cCsd|jd|jdfS)Nz%s+%sjrealimagr)rrrr rKszNumbaComplexPrinter.to_stringNrLrrrr rPsrPc@seZdZddZddZdS)NumbaTuplePrintercCs ||_dSrrrrrr r szNumbaTuplePrinter.__init__cCs>g}|jj}|D]}|t|j|jqdd|S)N(%s)r+)rr r!r9r0rFrA)rrIr!rErrr rKs  zNumbaTuplePrinter.to_stringNrLrrrr rSsrSc@seZdZddZddZdS)NumbaUniTuplePrintercCs ||_dSrrrrrr r szNumbaUniTuplePrinter.__init__cCsX|jj}|dj\}}g}t||dD]}|t|j|q0dd|S)NrrrTr+)rr r!r"r9r0rA)rr!r$r%rIirrr rKs  zNumbaUniTuplePrinter.to_stringNrLrrrr rUsrUc@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 memoryviewbytesdecoder0)rrDr ZnitemsrYrGrHrIrrr rKs     z!NumbaUnicodeTypePrinter.to_stringNrLrrrr rWsrWcCsdtjd}|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_printerrrPrSrUrW)printerrrr _create_printerss$ r`)__doc__r2Z gdb.printingr<r/rrPrSrUrWr`r^Zregister_pretty_printerZcurrent_objfilerrrr s s