U
    gZ$d                  
   @   s`  d dl Z e jjdk reefZneZeZdZdZ	dd Z
dZdZd	Zd
d ZeZdZdd Zdd Zdd Zdd Zdd Zd dlZd dlZd dlZd dlZedZddddddddd d!	Zd"d# ZG d$d% d%eZ G d&d' d'eZ!e"d(kr\d dl#m$Z$ e$$ Z%d dl Z e&e j'd) Z(e() Z*e!e%Z+e+,e*e j'd)  e+- Z.e.sLq\e/e+j0e. q:dS )*    N   )
ZCPP_IDCPP_INTEGERZ	CPP_FLOATZ
CPP_STRINGZCPP_CHARCPP_WSZCPP_COMMENT1ZCPP_COMMENT2Z	CPP_POUNDZ
CPP_DPOUNDz+-*/%|&~^<>=!?()[]{}.,;:\'"c                 C   s   | j  j| jd7  _| S )z\s+
lexerlinenovaluecountt r   5/tmp/pip-unpacked-wheel-tmpgil2w/pycparser/ply/cpp.pyt_CPP_WS   s    r   z\#z\#\#z[A-Za-z_][\w_]*c                 C   s   | S )zA(((((0x)|(0X))[0-9a-fA-F]+)|(\d+))([uU][lL]|[lL][uU]|[uU]|[lL])?)r   r   r   r   r   r   +   s    r   z?((\d+)(\.\d+)(e(\+|-)?(\d+))? | (\d+)e(\+|-)?(\d+))([lL]|[fF])?c                 C   s   | j  j| jd7  _| S )z\"([^\\\n]|(\\(.|\n)))*?\"r   r   r   r   r   r   t_CPP_STRING5   s    r   c                 C   s   | j  j| jd7  _| S )z(L)?\'([^\\\n]|(\\(.|\n)))*?\'r   r   r   r   r   r   
t_CPP_CHAR;   s    r   c                 C   s8   | j d}| j j|7  _d| _|r.d| nd| _ | S )z(/\*(.|\n)*?\*/)r   r    )r	   r
   r   r   type)r   Zncrr   r   r   t_CPP_COMMENT1A   s
     r   c                 C   s   d| _ d| _| S )z(//.*?(\n|$))r   r   )r   r	   r   r   r   r   t_CPP_COMMENT2J   s     r   c                 C   s(   | j d | _| j d | _ | jd | S Nr      )r	   r   r   skipr   r   r   r   t_errorP   s    r   z\?\?[=/\'\(\)\!<>\-]#\^[]|{}~)	=/'()!<>-c                 C   s   t dd | S )Nc                 S   s   t |  d  S N)_trigraph_repgroup)gr   r   r   <lambda>z       ztrigraph.<locals>.<lambda>)_trigraph_patsub)inputr   r   r   trigraphy   s    r6   c                   @   s   e Zd ZdddZdS )MacroNFc                 C   s0   || _ || _|| _|| _|r&|d | _d | _d S r,   )namer	   arglistvariadicvarargsource)selfr8   r	   r9   r:   r   r   r   __init__   s    
zMacro.__init__)NF)__name__
__module____qualname__r>   r   r   r   r   r7      s   r7   c                   @   s   e Zd Zd&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d Zd'ddZdd Zd(ddZdd Zdd Zd d! Zdi fd"d#Zd$d% ZdS ))PreprocessorNc                 C   sl   |d krt j}|| _i | _g | _g | _|   t }| dt	d|  | dt	d|  d | _
d S )Nz__DATE__ "%s"z%b %d %Yz__TIME__ "%s"z%H:%M:%S)lexr   macrospath	temp_pathlexprobetime	localtimedefinestrftimeparser)r=   r   tmr   r   r   r>      s    zPreprocessor.__init__c                 C   s0   g }| j | | j  }|s q,|| q|S N)r   r5   tokenappend)r=   texttokenstokr   r   r   tokenize   s    
 zPreprocessor.tokenizec                 C   s   t d|||f  d S )Nz%s:%d %s)print)r=   filelinemsgr   r   r   error   s    zPreprocessor.errorc              	   C   s  | j d | j  }|r$|jdkr.td n|j| _| j d | j  }|r^t|jdkrhtd n|j| _t|j| _	| j d | j  }|r|jdkrtd n|j| _
| j d | j  }|r|jdkrd | _n|j| _| j d	 | j  }|r|jd	krd | _td
 n|j| _| j| jf| _dddddddddg	}|D ]:}| j | | j  }|rz|j|krNtd|  qNd S )N
identifierz"Couldn't determine identifier typeZ12345i90  zCouldn't determine integer typez
"filename"zCouldn't determine string typez  r   z%Couldn't determine token for newlinesr)   r*   r   ##r   r&   r'   ,.z,Unable to lex '%s' required for preprocessor)r   r5   rO   r	   rU   r   t_IDint	t_INTEGERt_INTEGER_TYPEt_STRINGZt_SPACEZ	t_NEWLINEt_WS)r=   rS   charscr   r   r   rG      sD    









zPreprocessor.lexprobec                 C   s   | j | d S rN   )rE   rP   )r=   rE   r   r   r   add_path  s    zPreprocessor.add_pathc                 c   s   | j  }dd | D }tt|D ]T}|d }|| dr(|t|k r(|| d d ||  ||< d||< |d7 }q4q(d|}|| d|_g }|	 }|sq|
| |j| jkrd|jkr|V  g }q|r|V  d S )Nc                 S   s   g | ]}|  qS r   )rstrip.0xr   r   r   
<listcomp>  s     z,Preprocessor.group_lines.<locals>.<listcomp>r   r   r-    r   )r   clone
splitlinesxrangelenendswithjoinr5   r   rO   rP   r   rc   r	   )r=   r5   rC   linesijcurrent_linerS   r   r   r   group_lines  s*    



zPreprocessor.group_linesc                 C   st   d}|t |k r*|| j| jkr*|d7 }q|d |= t |d }|dkrb|| j| jkrb|d8 }q@||d d = |S r   )rp   r   rc   )r=   rR   rt   r   r   r   
tokenstrip8  s    


zPreprocessor.tokenstripc           	      C   s  g }g }g }d}t |}d}||k r>|| j| jkr>|d7 }q||k rd|| jdkrd||d  n | | j|d jd dg g fS |d7 }||k r`|| }|jdkr|| |d7 }n|jdkr|d8 }|dkr|r|| | || |d ||fS || nD|jdkrL|dkrL|| | ||d  g }n
|| |d7 }q| | j|d jd dg g fS )	Nr   r   r&   zMissing '(' in macro argumentsr'   r\   r-   zMissing ')' in macro arguments)	rp   r   rc   r	   rP   rY   r<   r   rx   )	r=   	tokenlistargs	positionsZcurrent_argZnestingZtokenlenrt   r   r   r   r   collect_argsT  sD    









zPreprocessor.collect_argsc                 C   s  g |_ g |_g |_d}|t|jk r |j| j| jkrh|j| j|jkrh|j|j| j}|dkr|j|d  jdkrt		|j| |j|< | j
|j| _|j|d = |j||d f qn|dkr|j|d  jdkr|j d||d f |j|d = qnZ|d t|jk rT|j|d  jdkrT|j d||f |d7 }qn|j d||f n|j| jdkr|jr|dkr|j|d  jdkr|d t|jk r|j|d  j| jkr|j|d  j|jkr|j|d  |d7 }q|j jdd	 d
d d S )Nr   r   r   r[   re   er\   c                 S   s   | d S )N   r   )rj   r   r   r   r1     r2   z,Preprocessor.macro_prescan.<locals>.<lambda>T)keyreverse)patch	str_patchvar_comma_patchrp   r	   r   r^   r9   indexcopyrb   rP   r:   r;   sort)r=   macrort   argnumr   r   r   macro_prescan  sB    ( *(
zPreprocessor.macro_prescanc           
      C   s   dd |j D }i }|jD ]X\}}||krRdddd || D  dd||< t|| ||< || || _ qd}|jr|d	 s|jD ]}d ||< d
}qi }|jD ]^\}	}}|	dkr|| |||d < q|	dkr||kr| || ||< || |||d < q|rdd |D }|S )Nc                 S   s   g | ]}t  |qS r   r   ri   _xr   r   r   rk     s     z2Preprocessor.macro_expand_args.<locals>.<listcomp>z"%s"rl   c                 S   s   g | ]
}|j qS r   r	   rh   r   r   r   rk     s     r   z\\Fr-   Tre   r   r}   c                 S   s   g | ]}|r|qS r   r   ri   Z_ir   r   r   rk     s      )	r	   r   rr   replacer   r:   r   r   expand_macros)
r=   r   rz   repZstr_expansionr   rt   Zcomma_patchexpandedptyper   r   r   macro_expand_args  s.    (
zPreprocessor.macro_expand_argsc                 C   s  |d kri }d}|t |k r|| }|j| jkr|j| jkr|j|krd||j< | j|j }|js| dd |jD |}|D ]}|j|_q||||d < |t |7 }n|d }|t |k r|| j| jkr|d7 }q|| jdkr| 	||d  \}	}
}|j
sTt |
t |jkrT| | j|jd|jt |jf  ||	 }n@|j
rt |
t |jd k rt |jdkr| | j|jd	|jt |jd f  n&| | j|jd
|jt |jd f  ||	 }n|j
rLt |
t |jd kr|
g  nD|||t |jd   ||	 d  |
t |jd < |
t |jd = | ||
}| ||}|D ]}|j|_qh|||||	 < |t |7 }||j= qn"|jdkr| j|_| |j|_|d7 }q|S )Nr   Tc                 S   s   g | ]}t  |qS r   r   r   r   r   r   rk     s     z.Preprocessor.expand_macros.<locals>.<listcomp>r   r&   zMacro %s requires %d argumentsr~   z(Macro %s must have at least %d argumentsz'Macro %s must have at least %d argumentZ__LINE__)rp   r   r^   r	   rD   r9   r   r   rc   r|   r:   rY   r<   rP   r   r`   ra   )r=   rR   r   rt   r   mexr}   ru   tokcountrz   r{   r   rr   r   r   r     s\    


" (&
4
zPreprocessor.expand_macrosc                 C   sN  d}|t |k r|| j| jkr|| jdkr|d }d}d}|t |k r|| j| jkrl|d7 }qDnn|| j| jkr|| j| jkrd}nd}|sqn<|| jdkrd}n(|| jd	krqn| | j|| jd
 |d7 }qD| j	|| _| 
||| _||d |d = |d7 }q| |}t|D ]\}}|j| jkrpt|||< | j	|| _| 
d|| _n`|j| j	kr.t|||< t|| j|| _|| jd dkr.|| jd d || _qq.ddd |D }|dd}|dd}|dd}zt|}W n0 tk
rH   | | j|d jd d}Y nX |S )Nr   Zdefinedr   FZ0LZ1Lr&   Tr'   zMalformed defined()r-   Z0123456789abcdefABCDEFrl   c                 S   s   g | ]}t |jqS r   strr	   rh   r   r   r   rk   S  s     z)Preprocessor.evalexpr.<locals>.<listcomp>z&&z and z||z or r(   z not zCouldn't evaluate expression)rp   r   r^   r	   rc   rD   rY   r<   r   r`   ra   r   	enumerater   r   rr   r   eval	Exception)r=   rR   rt   ru   Z	needparenresultr   exprr   r   r   evalexpr(  s^    " 


 
zPreprocessor.evalexprc                 c   s  t |}| |}|sd}| d|  || _g }d}d}g }|D ],}	t|	D ]\}
}|j| jkrP qjqP|jdkrb|	D ]$}|j| jkrzd|jkrz|| qz| 	|	|
d d  }|r|d j}| 	|dd  }nd}g }|d	kr|r`| 
|D ]
}|V  qg }| | qp|d
krz|r`| 
|D ]}|V  q2g }| jd }| |D ]}|V  qX|| jd< || _qp|dkr|r`| 
|D ]}|V  qg }| | qp|dkr|||f |r`|d j| jkrd}d}nd}qp|dkr8|||f |r`|d j| jkr0d}d}nd}qp|dkrv|||f |r`| |}|spd}d}nd}n|dkr|r|d d r|rd}n|s| |}|rd}d}n| | j|d jd n|dkr*|r|d d r(|rd}n|s(d}d}n| | j|d jd n6|dkrp|rH| \}}n| | j|d jd n qB|rB||	 qB| 
|D ]}|V  q|g }d S )Nrl   z__FILE__ "%s"TFr   r   r   r   rJ   includeZ__FILE__undefZifdefZifndefifelifr-   zMisplaced #elifelsezMisplaced #elseendifzMisplaced #endif)r6   rw   rJ   r<   r   r   rc   r	   rP   rx   r   rD   r   r   r   rY   r   popextend)r=   r5   r<   r   rs   chunkenableZ	iftriggerZifstackrj   rt   rS   Z	dirtokensr8   rz   Zoldfiler   r   r   r   parsegenc  s    

 


















zPreprocessor.parsegenc           
   	   c   s  |sd S |r|d j dkr4|d j| jkr4| |}|d j dkrd}|t|k rl|| j dkrbqx|d7 }qFtd d S ddd |d| D }| jdg | j }nB|d j| jkr|d j dd	 }| jdg | j }ntd
 d S |D ]}t	j||}zbt
|d }t	j|}|r0| jd| | ||D ]}	|	V  q<|rX| jd= W  qW q tk
rx   Y qX qtd|  d S )Nr   r)   r   r*   zMalformed #include <...>rl   c                 S   s   g | ]
}|j qS r   r   rh   r   r   r   rk     s     z(Preprocessor.include.<locals>.<listcomp>r-   zMalformed #include statementr   zCouldn't find '%s')r	   r   rb   r   rp   rU   rr   rE   rF   osopenreaddirnameinsertr   IOError)
r=   rR   rt   filenamerE   pZinamedataZdnamerS   r   r   r   r     sF    


zPreprocessor.includec                 C   s  t |tr| |}|}zz|d }t|dkr:|d }nd }|s^t|jg }|| j|j< n4|j| jkrt|j| 	|dd  }|| j|j< n|jdkr| 
|dd  \}}}d}	|D ]}
|	rtd  qddd	 |
D }|d
kr d}	| j|
d _d|
d _d}	|
dd = qnb|dd  d
kr|
d j| jkrd}	|
dd = |
d jdd  d
kr|
d jd d |
d _qt|
dks|
d j| jkrtd  qq| 	|d| d  }d}|t|k rX|d t|k rL|| j| jkr||d  jdkr||= qn0|| jdkrL||d  j| jkrL||d = |d7 }qt|j|dd	 |D |	}| | || j|j< ntd W n tk
r   td Y nX d S )Nr   r   r~   r&   Fz0No more arguments may follow a variadic argumentrl   c                 S   s   g | ]}t |jqS r   r   r   r   r   r   rk   1  s     z'Preprocessor.define.<locals>.<listcomp>z...TZ__VA_ARGS__zInvalid macro argumentr[   c                 S   s   g | ]}|d  j qS )r   r   rh   r   r   r   rk   O  s     zBad macro definition)
isinstanceSTRING_TYPESrT   rp   r7   r	   rD   r   rc   rx   r|   rU   rr   r^   r   LookupError)r=   rR   Zlinetokr8   mtyper   r   rz   r{   r:   aZastrZmvaluert   r   r   r   rJ     sl    





$
&&

zPreprocessor.definec                 C   s0   |d j }z| j|= W n tk
r*   Y nX d S )Nr   )r	   rD   r   )r=   rR   idr   r   r   r   ]  s
    
zPreprocessor.undefc                 C   s   || _ | ||| _d S rN   )ignorer   rL   )r=   r5   r<   r   r   r   r   parsei  s    zPreprocessor.parsec                 C   sD   z"t | j}|j| jkr|W S qW n tk
r>   d | _Y d S X d S rN   )nextrL   r   r   StopIteration)r=   rS   r   r   r   rO   r  s    
 zPreprocessor.token)N)N)N)r?   r@   rA   r>   rT   rY   rG   rf   rw   rx   r|   r   r   r   r   r   r   rJ   r   r   rO   r   r   r   r   rB      s&   
<!5+2
B;
 1F	rB   __main__r   )1sysversion_infomajorr   unicoder   rangero   rR   literalsr   Zt_CPP_POUNDZt_CPP_DPOUNDZt_CPP_IDr   Zt_CPP_INTEGERZt_CPP_FLOATr   r   r   r   r   rer   rH   os.pathr   compiler3   r.   r6   objectr7   rB   r?   Zply.lexrC   r   r   argvfr   r5   r   r   rO   rS   rU   r<   r   r   r   r   <module>
   sj   
	
     c
 