o
    8ήcU                     @   s   d Z ddlZddlZejdd Zz
ddlZddlZW n ey&   edw G dd dZG dd dZ	G d	d
 d
Z
G dd dZG dd dZdd Zeje e  dS )z(gdb printing extension for Numba types.
    N   z$GDB python support is not available.c                   @      e Zd Zdd Zdd ZdS )NumbaArrayPrinterc                 C   
   || _ d S Nvalselfr    r   E/tmp/pip-target-vg8gfxp4/lib/python/numba/misc/gdb_print_extension.py__init__      
zNumbaArrayPrinter.__init__c           )   
   C   s  zdd l }d}W n ty   d}Y nw zBd}| jd }| jd }| jd }| jd }d}t| jj}	|rd|	v s>d	|	v r|	d
d }	td}
dd |
	|	
 D }|\}}}d}t	||}|
 \}}}|dk}|d}g }|D ]C}|d}|d }|dd  }t|dkrtd| |d }d|v rtd| |dd dd }||}|||f qx|}n$td}
dd |
	|	
 D }|\}}}d|v r|dd dd}dd }||}||}||krG|r@|dd |D }||} |||  }!|!t|7 }!|j||d }"|"}t }#|#t||!}$|j|$|d!}%|jjj|%||d"}&d#t|& W S d$||f W S td%g}'d&d'|' d( d) W S  tyn }( z
d*|( W  Y d }(~(S d }(~(ww )+Nr   TFdataitemsizeshapestridesalignedRecordz
unaligned  z%array\((Record.*), (.*), (.*)\)\ \(.*c                 S      g | ]}|  qS r   strip.0xr   r   r   
<listcomp>:       z/NumbaArrayPrinter.to_string.<locals>.<listcomp>z(Record\((.*\[.*\]);([0-9]+);(True|False)True,[   zUnsupported sub-type: %snestedarray;=zarray\((.*),(.*),(.*)\)\ \(.*c                 S   r   r   r   r   r   r   r   r   T   r   z	unichr x z<Uc                    s<    j  }|d j  \}}t fddt||d D S )Nr   c                    s   g | ]}t  | qS r   )intr   dwarrr   r   r   ^   s    zGNumbaArrayPrinter.to_string.<locals>.dwarr2inttuple.<locals>.<listcomp>r!   )typefieldsrangetuple)r(   r*   lohir   r'   r   dwarr2inttupleZ   s   
 z3NumbaArrayPrinter.to_string.<locals>.dwarr2inttuplec                 S   s   g | ]	}t d |d qS )r   r!   )maxr   r   r   r   r   j   s    )align)dtype)r   r   
z array([...], dtype=%s, shape=%s)zNULL/Uninitializedzarray([, ])z%array[Exception: Failed to parse. %s])numpyImportErrorr   strr)   replacer   recompilematchgroupssplitlen	TypeErrorr2   appendarraysumr&   gdbselected_inferiorread_memory
frombufferlibstride_tricks
as_stridedlistjoin	Exception))r
   np
HAVE_NUMPYNULLr   r   rshprstrides
is_alignedty_strmatcherarr_info	dtype_strndim_str	order_strrstr
rstr_matchr*   balignis_aligned_str	field_dtsstruct_entriesfsplittednamedt_part	dt_as_strr2   r/   r   r   shp_arrstrd_arrextentdtype_clazz	this_procmemarr_datanew_arrbufer   r   r   	to_string   s   














zNumbaArrayPrinter.to_stringN__name__
__module____qualname__r   rp   r   r   r   r   r          r   c                   @   r   )NumbaComplexPrinterc                 C   r   r   r   r	   r   r   r   r      r   zNumbaComplexPrinter.__init__c                 C   s   d| j d | j d f S )Nz%s+%sjrealimagr   )r
   r   r   r   rp      s   zNumbaComplexPrinter.to_stringNrq   r   r   r   r   rv      ru   rv   c                   @   r   )NumbaTuplePrinterc                 C   r   r   r   r	   r   r   r   r      r   zNumbaTuplePrinter.__init__c                 C   s>   g }| j j }|D ]}|t| j |j  q
dd| S )N(%s)r4   )r   r)   r*   rB   r9   rc   rM   )r
   rn   r*   ra   r   r   r   rp      s
   zNumbaTuplePrinter.to_stringNrq   r   r   r   r   ry      ru   ry   c                   @   r   )NumbaUniTuplePrinterc                 C   r   r   r   r	   r   r   r   r      r   zNumbaUniTuplePrinter.__init__c                 C   sX   | j j }|d j \}}g }t||d D ]}|t| j |  qdd| S )Nr   r!   rz   r4   )r   r)   r*   r+   rB   r9   rM   )r
   r*   r-   r.   rn   ir   r   r   rp      s   zNumbaUniTuplePrinter.to_stringNrq   r   r   r   r   r{      ru   r{   c                   @   r   )NumbaUnicodeTypePrinterc                 C   r   r   r   r	   r   r   r   r      r   z NumbaUnicodeTypePrinter.__init__c           	   
   C   s   d}| j d }| j d }| j d }||kret }|t||| }tdk rMz
t|d}W d| S  tyL } zdt| }W Y d }~d| S d }~ww t	|t
r\t| }d| S |d}d| S t|}d| S )	Nr   r   lengthkind)   r   zutf-8z	ERROR: %sz'%s')r   rE   rF   rG   r&   
_PYVERSIONunicodeUnicodeDecodeErrorr9   
isinstance
memoryviewbytesdecode)	r
   rQ   r   nitemsr   rj   rk   rn   ro   r   r   r   rp      s.   





z!NumbaUnicodeTypePrinter.to_stringNrq   r   r   r   r   r}      ru   r}   c                  C   sd   t jd} | ddt | ddt | ddt | dd	t | d
dt | ddt | S )NNumbazNumba unaligned array printerz^unaligned array\(zNumba array printerz^array\(zNumba complex printerz^complex[0-9]+\ zNumba Tuple printerz^Tuple\(zNumba UniTuple printerz^UniTuple\(zNumba unicode_type printerz^unicode_type\s+\()	rE   printingRegexpCollectionPrettyPrinteradd_printerr   rv   ry   r{   r}   )printerr   r   r   _create_printers   s$   r   )__doc__r;   sysversion_infor   gdb.printingrE   r8   r   rv   ry   r{   r}   r   r   register_pretty_printercurrent_objfiler   r   r   r   <module>   s"    s	