U
���h�
� @ s� d dl Z d dlmZ d dlZd dlmZmZ dZe je e
e eej f d�dd�Z
e jeej d�d d
�Ze jdd �d d
�Ze jdd �dd�ZdS )� N)�Optional)�types�logZPRSO)�
trace_path�map_file_provided�returnc
C s� t d| � �� t| d��}|�� }W 5 Q R X |�� }t|�dk rDdS d}i }|d d dkr�|d dd� }|r|t�td� d d
� |�d �D �D ]<}||kr�t�td |� d
�t� | j
d�� |||< |d7 }q�t|�d } | |fS t|�dfS )zrGiven `trace_path`, return the inferred length of the trace and, if `return_mapping` is enabled, a signal mapping.zTrace: �r� )�����Nr �#NzHMap file given and header included in trace file; header will be ignoredc S s g | ] }|� � �qS � )�strip)�.0�sr r �o/Users/aaurandt/Desktop/IowaState/R2U2Research/Dissertation/Comparison/r2u2-modify/compiler/c2po/parse_utils.py�
<listcomp># s z$parse_trace_file.<locals>.<listcomp>�,� Signal ID 'z5' found multiple times in csv, using right-most value) �print�open�read�
splitlines�lenr �warning� MODULE_CODE�split� FileLocation�name)
r r �f�content�linesZcnt�signal_mapping�header�id� trace_lengthr r r �parse_trace_file s6 �
�
r% )�map_pathr c C s� t | d��}|�� }W 5 Q R X i }|�� }|D ]�}t�d|�r�|�d�}|d }t|d �}||kr�t�t d|� d�t�
| j |� |�d �� |||< q.t�
t d|� d �t�
| j |� |��� d
S q.|S ) z*Return the signal mapping from `map_path`.r z [a-zA-Z_][a-zA-Z0-9_\[\]]*:\d+�:r r r z6' found multiple times in map file, using latest valuez#Invalid format for map line (found z,)
Should be of the form SYMBOL ':' NUMERALN)r r r �re�matchr �intr r r r r �index�error) r& r r �mappingr �lineZstrsr# Zsidr r r �parse_map_file5 s.
�
�r/ )� bounds_pathr c C s d S �Nr �r0 r r r �parse_bounds_cW s r3 c C s d S r1 r r2 r r r �parse_bounds_rust[ s r4 )�pathlib�typingr r( �c2por r r �Path�bool�tupler* �
SignalMappingr% r/ r3 r4 r r r r �<module> s � ,"