File: //home/arjun/.local/lib/python3.10/site-packages/numpy/f2py/__pycache__/auxfuncs.cpython-310.pyc
o
���g�g � @ s| d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddl m Z g d�Z e
j
ZejjZejZi Zg ZdZd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Z dd� Z dd� Z dd � Z d d � Z d d � Z!d!d"� Z"d#d$� Z#d%d&� Z$d'd(� Z%d)d*� Z&d+d,� Z'd-d.� Z(d/d0� Z)d1d2� Z*d3d4� Z+d5d6� Z,d7d8� Z-d9d:� Z.d;d<� Z/d=d>� Z0d?d@� Z1dAdB� Z2dCdD� Z3dEdF� Z4dGdH� Z5dIdJ� Z6dKdL� Z7dMdN� Z8dOdP� Z9dQdR� Z:dSdT� Z;dUdV� Z<dWdX� Z=dYdZ� Z>d[d\� Z?d]d^� Z@d_d`� ZAdadb� ZBdcdd� ZCdedf� ZDdgdh� ZEdidj� ZFdkdl� ZGdmdn� ZHdodp� ZIdqdr� ZJdsdt� ZKdudv� ZLdwdx� ZMdydz� ZNd{d|� ZOd}d~� ZPdd�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d�� ZbeTd�eUd�eVd�eWd�eZd�eYd�ePd�e^d�e`d�ead�ebd�i Zcd�d�� Zdd�d�� Zed�d�� Zfd�d�� Zgd�d�� Zhd�d�� Zid�d�� Zjd�d�� Zkd�d�� Zld�d�� Zmd�dÄ Znd�dń ZoG d�dDŽ d�ep�ZqG d�dɄ dɃZrd�d˄ Zsd�d̈́ Ztd�dτ Zud�dф Zvd�dӄ Zwd�d�dՄZxd�dׄ Zyi fd�dلZzd�dۄ Z{d�d݄ Z|d�d߄ Z}d�d� Z~d�d� Zd�d� Z�d�d� Z�d�d� Z�d�d� Z�d�d�d�Z�d�d� Z�i fd�d�Z�e��d�ej��j�Z�e��d�ej��j�Z�d�d�� Z�d�d�� Z�d�d�d��Z�dS )�a9
Auxiliary functions for f2py2e.
Copyright 1999 -- 2011 Pearu Peterson all rights reserved.
Copyright 2011 -- present NumPy Developers.
Permission to use, modify, and distribute this software is given under the
terms of the NumPy (BSD style) LICENSE.
NO WARRANTY IS EXPRESSED OR IMPLIED. USE AT YOUR OWN RISK.
� N)�reduce)�deepcopy� )� __version__)�cfuncs)\�
applyrules� debugcapi�
dictappend�errmess�gentitle�getargs2�getcallprotoargument�getcallstatement�getfortranname�getpymethoddef�
getrestdoc� getusercode� getusercode1� getdimension�hasbody�hascallstatement� hascommon� hasexternals� hasinitvalue�hasnote�
hasresultnote�
isallocatable�isarray�isarrayofstrings� ischaracter�ischaracterarray� ischaracter_or_characterarray� iscomplex�iscomplexarray�iscomplexfunction�iscomplexfunction_warn�isdouble�isdummyroutine�
isexternal�
isfunction�isfunction_wrap�isint1� isint1array� isinteger� isintent_aux�
isintent_c�isintent_callback�
isintent_copy�
isintent_dict�
isintent_hide� isintent_in�isintent_inout�isintent_inplace�isintent_nothide� isintent_out�isintent_overwrite� islogical�islogicalfunction�islong_complex�
islong_double�islong_doublefunction� islong_long�islong_longfunction�ismodule�ismoduleroutine�
isoptional� isprivate�
isrequired� isroutine�isscalar�issigned_long_longarray�isstring�
isstringarray�isstring_or_stringarray�isstringfunction� issubroutine�get_f2py_modulename�issubroutine_wrap� isthreadsafe�
isunsigned�isunsigned_char�isunsigned_chararray�isunsigned_long_long�isunsigned_long_longarray�isunsigned_short�isunsigned_shortarray�l_and�l_not�l_or�outmess�replace�show�
stripcomma� throw_error� isattr_value� getuseblocks�process_f2cmap_dictc C s t �dd�rtj�| � d S d S )N�verboser )�options�get�sys�stdout�write)�t� rj �F/home/arjun/.local/lib/python3.10/site-packages/numpy/f2py/auxfuncs.pyr[ > s �r[ c C s dt v S )N�capi)� debugoptions��varrj rj rk r C � r c C � d| v o| d dkot | � S �N�typespec� character�r( rn rj rj rk � _ischaracterG � �rv c C rq rr ru rn rj rj rk � _isstringL rw rx c C s t | �od| vS �N� charselector�rv rn rj rj rk r! Q � r! c C � t | �ot| � S �N�r! r rn rj rj rk r U � r c C � t | �ot| �S r~ r rn rj rj rk r Y r| r c C s t | �od| v S ry r{ rn rj rj rk rK ] r| rK c C r} r~ �rK r rn rj rj rk rI a r� rI c C r� r~ r� rn rj rj rk rJ e r| rJ c C s t | �o | d d dkS )N� dimension���z(*))rJ rn rj rj rk r i s r c C s d| v ot | � S �Nr� ru rn rj rj rk r o r� r c C s t | �p t| �p t| � S r~ )r rI r( rn rj rj rk rG s s rG c C � t | �o
| �d�dv S �Nrs )�complexzdouble complex�rG re rn rj rj rk r" w � �r" c C � t | �o
| �d�dkS )Nrs �logicalr� rn rj rj rk r: | � r: c C r� )Nrs �integerr� rn rj rj rk r- � r� r- c C r� )Nrs �realr� rn rj rj rk �isreal� r� r� c C sH z| d d W S t y# z | d d W Y S t y" Y Y d S w w )N� kindselector�*�kind��KeyErrorrn rj rj rk �get_kind� s ��r� c C s$ | � d�dkot| �dkot| � S �Nrs r� �1)re r� r rn rj rj rk r+ � s
��r+ c C s* t | �sdS | �d�dvrdS t| �dkS )Nr rs �r� r� �8�rG re r� rn rj rj rk r? � �
r? c C �* t | �sdS | �d�dkrdS t| �dkS )Nr rs r� �-1r� rn rj rj rk rR � r� rR c C r� )Nr rs r� �-2r� rn rj rj rk rV � r� rV c C r� )Nr rs r� �-4r� rn rj rj rk rQ � r� rQ c C r� )Nr rs r� �-8r� rn rj rj rk rT � r� rT c C �* t | �sdS | �d�dksdS t| �dkS )Nr rs r� r� r� rn rj rj rk r& � r� r& c C r� )Nr rs r� �16r� rn rj rj rk r= � r� r= c C s t | �sdS t| �dkS )Nr �32)r" r� rn rj rj rk r< � s r< c C r� r� )r re rn rj rj rk r# � r� r# c C s" t | �o| �d�dkot| �dkS r� �r re r� rn rj rj rk r, � �
�r, c C �" t | �o| �d�dv ot| �dkS )Nrs r� r� r� rn rj rj rk rS � r� rS c C r� )Nrs r� r� r� rn rj rj rk rW � r� rW c C r� )Nrs r� r� r� rn rj rj rk �isunsignedarray� r� r� c C r� )Nrs r� r� r� rn rj rj rk rU � r� rU c C r� )Nrs r� r� r� rn rj rj rk �issigned_chararray� r� r� c C r� )Nrs r� �2r� rn rj rj rk �issigned_shortarray� r� r� c C r� )Nrs r� �4r� rn rj rj rk �issigned_array� r� r� c C r� )Nrs r� r� r� rn rj rj rk rH r� rH c C � d| v o d| d v S )N�attrspec� allocatablerj rn rj rj rk r � r c C s d| vpt | � S r� )rI rn rj rj rk � ismutable r� r� c C � d| v S )N�
modulenamerj ��routrj rj rk rB rp rB c C � d| v o d| d kS )N�block�modulerj r� rj rj rk rA r� rA c C r� )Nr� �functionrj r� rj rj rk r) r� r) c C s" t | �rdS tot| �ot| � S �Nr )r/ � wrapfuncsr) r( r� rj rj rk r* s r* c C r� )Nr� �
subroutinerj r� rj rj rk rM ! r� rM c C s t | �rdS t| �o
t| �S r� )r/ rM �hasassumedshaper� rj rj rk rO % � rO c C � d| � dg �v S )N�valuer� �re rn rj rj rk r` * r| r` c C sV | � d�rdS | d D ] }| d � |i �� dg �D ]}|dkr'd| d< dS qq dS )Nr� T�args�varsr� �:Fr� )r� �a�drj rj rk r� . s
��r� c C � t | �pt| �S r~ )rB r� r� rj rj rk �requiresf90wrapper9 r| r� c C r� r~ )r) rM r� rj rj rk rF = r| rF c C �F t | �sdS d| v r| d }n| d }|| d v r!t| d | �S dS �Nr �result�namer� )r) r: �r� r� rj rj rk r; A �
r; c C r� r� )r) r? r� rj rj rk r@ M r� r@ c C r� r� )r) r= r� rj rj rk r> Y r� r> c C r� r� )r) r" r� rj rj rk r$ e r� r$ c C s t | �r
td� dS dS )Na� **************************************************************
Warning: code with a function returning complex value
may not work correctly with your Fortran compiler.
When using GNU gcc/g77 compilers, codes should work
correctly for callbacks with:
f2py -c -DF2PY_CB_RETURNCOMPLEX
**************************************************************
r r )r$ r[ r� rj rj rk r% q s r% c C r� r� )r) rI r� rj rj rk rL r� rL c C � d| v o| d S )N� externalsrj r� rj rj rk r � r| r c C r� )N�f2pyenhancements�
threadsaferj r� rj rj rk rP � s
�rP c C r� )Nr� rj r� rj rj rk � hasvariables� r| r� c C s( d| v od| d v od| d vot | �S )Nr� �optional�required)r7 rn rj rj rk rC � s
��rC c C r� )Nr� �externalrj rn rj rj rk r( � r� r( c sH d� d| � � v r tdd� | d D ��r"� fdd�| d D �d S d S d S )Nz \((.*?)\)r� c s s � | ]}d |v V qdS )r� Nrj )�.0�srj rj rk � <genexpr>� s � z getdimension.<locals>.<genexpr>c s g | ]}t �� |��qS rj )�re�findall)r� �v��
dimpatternrj rk �
<listcomp>� � z getdimension.<locals>.<listcomp>r )�keys�anyrn rj r� rk r � s �r c C s t | � ot| �S r~ )rC r7 rn rj rj rk rE � r� rE c C sp d| vrdS d| d v rdS d| d v rdS d| d v r dS d| d v r&dS d| d v r.dS d | d v r6dS dS )
N�intentr �hider �inplace�in�out�inout�outinrj rn rj rj rk r4 � s r4 c C sD d| v o!d| d v sd| d v o!d| d vo!d| d vo!d| d vS )Nr� r� r� r� r� r� rj rn rj rj rk r5 � s
�
�
�
�r5 c C r� )Nr� r� r� rn rj rj rk r8 � r| r8 c C s< d| v o d| d v p d| d v o d| d vo t tt�| � S )Nr� r� r� r� )rZ r5 r6 rn rj rj rk r3 � s �r3 c C s
t | � S r~ )r3 rn rj rj rk r7 � �
r7 c C r� )N�cr� r� rn rj rj rk r/ � r| r/ c C r� )N�cacher� r� rn rj rj rk �isintent_cache� r| r� c C r� )N�copyr� r� rn rj rj rk r1 � r| r1 c C r� )N� overwriter� r� rn rj rj rk r9 � r| r9 c C r� )N�callbackr� r� rn rj rj rk r0 � r| r0 c C r� )Nr� r� r� rn rj rj rk r6 � r| r6 c C r� )N�auxr� r� rn rj rj rk r. � r| r. c C r� )N�aligned4r� r� rn rj rj rk �isintent_aligned4� r| r� c C r� )N�aligned8r� r� rn rj rj rk �isintent_aligned8� r| r� c C r� )N� aligned16r� r� rn rj rj rk �isintent_aligned16� r| r� � INTENT_IN� INTENT_INOUT�
INTENT_OUT� INTENT_HIDE� INTENT_CACHE�INTENT_C�OPTIONAL�INTENT_INPLACE�INTENT_ALIGNED4�INTENT_ALIGNED8�INTENT_ALIGNED16c C r� )Nr� �privaterj rn rj rj rk rD r� rD c C r� )N�=rj rn rj rj rk r rp r c C s t | �sdS | d d dv S )Nr r )�"�')r rn rj rj rk �hasinitvalueasstring
r� r c C r� )N�noterj rn rj rj rk r rp r c C r� r� )r) r r� rj rj rk r r� r c C r� )N�commonrj r� rj rj rk r # rp r c C s4 t | �rdS t| �r| d D ] }t|�r dS qdS )Nr �bodyr )r r �containscommon)r� �brj rj rk r ' s �r c C s8 t | �rdS t| �s dS | d D ] }t|�r dS qdS )Nr r r )rA r �containsmodule)r� r
rj rj rk r 1 s �r c C r� )Nr rj r� rj rj rk r < rp r c C s t | �d uS r~ )r r� rj rj rk r @ s r c C � dS )Nr rj rn rj rj rk �istrueD � r c C r r� rj rn rj rj rk �isfalseH r r c @ s e Zd ZdS )� F2PYErrorN)�__name__�
__module__� __qualname__rj rj rj rk r L s r c @ s e Zd Zdd� Zdd� ZdS )r_ c C s
|| _ d S r~ )�mess)�selfr rj rj rk �__init__R r� zthrow_error.__init__c C s d|| j f }t|��)Nz
var = %s
Message: %s
)r r )r ro r rj rj rk �__call__U s zthrow_error.__call__N)r r r r r rj rj rj rk r_ P s r_ c G �N dg }}t t| ��D ]}d|||f }|�d| � q td|d�|�f �S )N�lambda v� %s,f%d=f[%d]�f%d(v)�%s:%sz and ��range�len�append�eval�join��f�l1�l2�irj rj rk rX Z �
rX c G r )Nr r r r z or r r& rj rj rk rZ b r+ rZ c C s t d�S )Nzlambda v,f=f:not f(v))r$ )r' rj rj rk rY j rp rY c C s( z | d d dkW S t y Y dS w )Nr� � fortranname� r r� r� rj rj rk r' n s
�r' c C sT z| d d }|dkr
t �|std| d � t �W |S t y) | d }Y |S w )Nr� r, r- z"Failed to use fortranname from %s
r� )r� r
)r� r� rj rj rk r u s �� �
�r c C s
z | d � |�}W n
ty Y d S w |sd S |dkr#t|t�r#d S t|t�r4|t|�kr0d S || }|d d� dkr�|rQd| d t|� d |dd � }n|dd � }|dd � dkry|rq|d d� d t|� d
}|S |d d� }|S td |t|�f � |S ) Nr� r � z'''z
/* start z multiline (z) */
���z
/* end multiline (z)*/z-%s multiline block should end with `'''`: %s
)re r� �
isinstance�str�listr" �reprr
)r� � blockname�comment�counter�rrj rj rk �getmultilineblock� sF �
���
� ��
�r8 c C �
t | d�S )N�
callstatement�r8 r� rj rj rk r � r� r c
C s t | ddd�}|r |S t| �rtd� d S ddlm} g g }}tttt��| �r0|� ddg� | d D ]J}| d
| }t
|�rAq4||v rL|| d }n-||�}tt t t
t��|�r[n
t|�r`nt|�sh|d }t|�stt|�stt|�ry|�d� |�|� q4d
�|| �} | s�d} | S )N�callprotoargumentr )r5 z<warning: callstatement is defined without callprotoargument
r )�getctypezchar*�size_tr� r� �_typedefr� �,�void)r8 r r[ � capi_mapsr= rX rL rY r* �extendr0 r/ rZ rG r" rI r` r rJ r# r% )
r� �cb_mapr7 r= � arg_types�
arg_types2�nro �ctype�
proto_argsrj rj rk r
� sH �
��
r
c C r9 )N�usercoder; r� rj rj rk r � r� r c C s t | ddd�S )NrJ r )r6 r; r� rj rj rk r � s r c C r9 )N� pymethoddefr; r� rj rj rk r � r� r c C sv g g }}d| v r7| d }d| v r3| d D ] }||v r |� |� q|D ] }||vr.|� |� q#||fS | d }||fS )Nr� �sortvars)r# )r� �sortargsr� r� rj rj rk �getargs� s
�
��rN c s� g �� dg �}� � �fdd��d �� D �}|� � d�v rA�d D ] }|� v r.|�|� q#� D ] }||vr<|�|� q1� |fS |�d }� |fS )Nr� c s( g | ]}t �d | �r|� vr|�qS )r� )r. )r� r� �r� r� rj rk r� � s �z getargs2.<locals>.<listcomp>r� rL )re r� r# )r� rM �auxvarsr� rj rO rk r � s
�
� �r c C s< d| vrd S d }| d dkr| d | d f}| d � |d �S )N�f2pymultilinesr� z
python moduler� r� )r� �krj rj rk r � s r c C s* dt | � d d }d|d | |d f S )N�P � � z /*%s %s %s*/r� )r" )r� �lnrj rj rk r s r c C s$ t | t�rttfdd�| g �S | gS )Nc S s | ||� S r~ rj )�x�yr' rj rj rk �<lambda> s zflatlist.<locals>.<lambda>)r0 r2 r �flatlist)�lstrj rj rk rZ s
rZ c C s | r| d dkr| d d� S | S )Nr� r@ rj )r� rj rj rk r^
s r^ r- c s� t � t�r��fdd�� D �S t �t�r � �fdd��D �S dt� �� � D ];}|dkr-q&d� v r>|� d v r>� d | }n�}t � | t�rW��d| |�t� | ����q&��d| � | ��q&�S )Nc s g | ]}t �|� ��qS rj �r\ �r� �_m)�
defaultsepr1 rj rk r� r� zreplace.<locals>.<listcomp>c s g | ]}t |� ���qS rj r\ r] )r� r_ rj rk r� r� rU �
separatorsforz#%s#)r0 r2 r� r\ r% rZ )r1 r� r_ rR �seprj )r� r_ r1 rk r\ s
r\ c C s( t |t�r|D ]}t| |�} q| S |�� D ]|}|d dkr q|| v r�t | | t�r0| | g| |< t | | t�rSt || t�rI| | || | |< q| | �|| � qt | | t�r�t || t�r�|dkr|| �� D ]}|| | vr}|| | | | |< qkqt| | || �| |< q|| | |< q| S )Nr �_r` )r0 r2 r r� r1 r# �dict)�rd�arr� rR �k1rj rj rk r & s4
���r c C s� i }t | t�r | D ]}t|||�}t||�}d|v r |S q |S d| v r,| d |�s,|S d| v rEtd| d i||�}d|v rEt�|d � | �� D ]�}|dkrV| | ||< qIt | | t�rgt| | |�||< n�t | | t�r�g ||< | | D ]}t||i||�} || v r�|| � | | � qvn�|d dkr�qIt | | t
��rg ||< | | �� D ]l}
t |
t j ��r|
|��rt | | |
t�r�| | |
D ]$}t |t
�r�td|i||�}d|v r�|d }nd }|| � t||�� q�q�| | |
}t |t
��r
td|i|�}d|v �r|d }nd }|| � t||�� q�n
t
d
t| | � � t || t��rCt|| �d k�r9|| d ||< || g k�rC||= qI|S ) N�_break�_check�need�needsr` r rb � supertextr- z applyrules: ignoring rule %s.
r )r0 r2 r r r � append_needsr� r1 r\ r# rc �types� FunctionTyper
r3 r" ) �rulesr� ro �retr7 �rr�resrR r* re rf rj rj rk r C s|
� ��
�
���r z%\s*python\s*module\s*(?P<name>[\w_]+)z4\s*python\s*module\s*(?P<name>[\w_]*?__user__[\w_]*)c C sp d }t | ��(}|D ]}t|�}|r t|�rq |�d�} n q W d � |S W d � |S 1 s1w Y |S )Nr� )�open�_f2py_module_name_match� _f2py_user_module_name_match�group)�sourcer� r' �line�mrj rj rk rN � s$
��
��
��rN c C sL g }| d D ] }|d D ]}|� d�r"|�dd� |� d��� D �� q q|S )Nr �usec S s g | ]}d |vr|�qS )�__rj )r� rW rj rj rk r� � r� z getuseblocks.<locals>.<listcomp>)re rC r� )�pymod�all_uses�inner�modblockrj rj rk ra � s
��ra Fc C s� g }i }|� � D ]\}}dd� |� � D �}|||�� < q|� � D ]U\}}|| vr,i | |< |� � D ]D\} }
|
|v rc| | | v rLtd|| | | | |
f � |
| | | < |r]td|| |
f � |�|
� q0|rttd|| |
|
t|�� �f � q0q | |fS )a�
Update the Fortran-to-C type mapping dictionary with new mappings and
return a list of successfully mapped C types.
This function integrates a new mapping dictionary into an existing
Fortran-to-C type mapping dictionary. It ensures that all keys are in
lowercase and validates new entries against a given C-to-Python mapping
dictionary. Redefinitions and invalid entries are reported with a warning.
Parameters
----------
f2cmap_all : dict
The existing Fortran-to-C type mapping dictionary that will be updated.
It should be a dictionary of dictionaries where the main keys represent
Fortran types and the nested dictionaries map Fortran type specifiers
to corresponding C types.
new_map : dict
A dictionary containing new type mappings to be added to `f2cmap_all`.
The structure should be similar to `f2cmap_all`, with keys representing
Fortran types and values being dictionaries of type specifiers and their
C type equivalents.
c2py_map : dict
A dictionary used for validating the C types in `new_map`. It maps C
types to corresponding Python types and is used to ensure that the C
types specified in `new_map` are valid.
verbose : boolean
A flag used to provide information about the types mapped
Returns
-------
tuple of (dict, list)
The updated Fortran-to-C type mapping dictionary and a list of
successfully mapped C types.
c S s i | ] \}}|� � |�qS rj )�lower)r� rf �v1rj rj rk �
<dictcomp>� s z'process_f2cmap_dict.<locals>.<dictcomp>z3 Warning: redefinition of {'%s':{'%s':'%s'->'%s'}}
z Mapping "%s(kind=%s)" to "%s"
z5 Ignoring map {'%s':{'%s':'%s'}}: '%s' must be in %s
)�itemsr� r[ r# r
r2 r� ) �
f2cmap_all�new_map�c2py_maprc �
f2cmap_mapped�
new_map_lowerrR �d1�d1_lowerrf r� rj rj rk rb � s: & �� ����rb )r r )r- )F)��__doc__�pprintrf r� rm � functoolsr r� r r- r r �__all__�version� f2py_version�stderrrh r
r] rd rm r� r[ r rv rx r! r r rK rI rJ r r rG r" r: r- r� r� r+ r? rR rV rQ rT r& r= r< r# r, rS rW r� rU r� r� r� rH r r� rB rA r) r* rM rO r` r� r� rF r; r@ r> r$ r% rL r rP r� rC r( r rE r4 r5 r8 r3 r7 r/ r� r1 r9 r0 r6 r. r� r� r� r2 rD r r r r r r r r r r r � Exceptionr r_ rX rZ rY r' r r8 r r
r r r rN r r r rZ r^ r\ r r �compile�I�matchrt ru rN ra rb rj rj rj rk �<module> s*
�
'
=����