Ë
    ?çi·v  ã                   ó”   — d dl Z d dlmZ d dlmZ d dlmZ d dlmZm	Z	m
Z
mZmZ dZ G d„ de«      Z G d„ d	e«      Z G d
„ de«      Z[y)é    N)ÚIntervalSet)ÚToken)ÚATNState)ÚRecognitionExceptionÚNoViableAltExceptionÚInputMismatchExceptionÚFailedPredicateExceptionÚParseCancellationExceptionc                   ó\   — e Zd Zdefd„Zdefd„Zdedefd„Zdefd„Zdefd„Z	dedefd„Z
y	)
ÚErrorStrategyÚ
recognizerc                  ó   — y ©N© ©Úselfr   s     úo/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/antlr4/error/ErrorStrategy.pyÚresetzErrorStrategy.reset   ó   € Øó    c                  ó   — y r   r   r   s     r   ÚrecoverInlinezErrorStrategy.recoverInline   r   r   Úec                  ó   — y r   r   ©r   r   r   s      r   ÚrecoverzErrorStrategy.recover   r   r   c                  ó   — y r   r   r   s     r   ÚsynczErrorStrategy.sync   r   r   c                  ó   — y r   r   r   s     r   ÚinErrorRecoveryModez!ErrorStrategy.inErrorRecoveryMode   r   r   c                  ó   — y r   r   r   s      r   ÚreportErrorzErrorStrategy.reportError"   r   r   N)Ú__name__Ú
__module__Ú__qualname__ÚParserr   r   r   r   r   r    r"   r   r   r   r   r      sZ   „ ð˜vó ð vó ð ð Ð+?ó ð˜fó ð¨Vó ð Vð Ð/Cô r   r   c                   ó4  ‡ — e Zd Zˆ fd„Zdefd„Zdefd„Zdefd„Zdefd„Zdefd„Z	dede
fd	„Zdede
fd
„Zdefd„Zdedefd„Zdedefd„Zd„ Zdefd„Zdefd„Zdefd„Zdefd„Zdefd„Zdefd„Zdefd„Zdefd„Zdefd„Zdefd„Zdedefd„Z ˆ xZ!S )ÚDefaultErrorStrategyc                 óh   •— t         ‰| �  «        d| _        d| _        d | _        d | _        d| _        y )NFéÿÿÿÿr   )ÚsuperÚ__init__ÚerrorRecoveryModeÚlastErrorIndexÚlastErrorStatesÚnextTokensContextÚnextTokenState)r   Ú	__class__s    €r   r,   zDefaultErrorStrategy.__init__+   s:   ø€ Ü‰ÑÔð "'ˆÔð !ˆÔØ#ˆÔØ!%ˆÔØˆÕr   r   c                 ó&   — | j                  |«       y r   ©ÚendErrorConditionr   s     r   r   zDefaultErrorStrategy.resetB   ó   € Ø×Ñ˜zÕ*r   c                 ó   — d| _         y )NT©r-   r   s     r   ÚbeginErrorConditionz(DefaultErrorStrategy.beginErrorConditionK   s
   € Ø!%ˆÕr   c                 ó   — | j                   S r   r8   r   s     r   r    z(DefaultErrorStrategy.inErrorRecoveryModeN   s   € Ø×%Ñ%Ð%r   c                 ó.   — d| _         d | _        d| _        y )NFr*   )r-   r/   r.   r   s     r   r5   z&DefaultErrorStrategy.endErrorConditionW   s   € Ø!&ˆÔØ#ˆÔØ ˆÕr   c                 ó&   — | j                  |«       y r   r4   r   s     r   ÚreportMatchz DefaultErrorStrategy.reportMatcha   r6   r   r   c                 ó¬  — | j                  |«      ry | j                  |«       t        |t        «      r| j	                  ||«       y t        |t
        «      r| j                  ||«       y t        |t        «      r| j                  ||«       y t        dt        |«      j                  z   «       |j                  |j                  |j                  |«       y )Nz unknown recognition error type: )r    r9   Ú
isinstancer   ÚreportNoViableAlternativer   ÚreportInputMismatchr	   ÚreportFailedPredicateÚprintÚtyper#   ÚnotifyErrorListenersÚmessageÚoffendingTokenr   s      r   r"   z DefaultErrorStrategy.reportErrorw   s¨   € ð ×#Ñ# JÔ/ØØ× Ñ  Ô,Ü�qÔ.Ô0Ø×*Ñ*¨:°qÕ9Ü˜Ô2Ô4Ø×$Ñ$ Z°Õ3Ü˜Ô4Ô6Ø×&Ñ& z°1Õ5äÐ4´t¸A³w×7GÑ7GÑGÔHØ×+Ñ+¨A¯I©I°q×7GÑ7GÈÕKr   c                 ó¦  — | j                   |j                  «       j                  k(  r4| j                  �(|j                  | j                  v r|j                  «        |j                  j                  | _         | j                  €g | _        | j                  j                  |j                  «       | j                  |«      }| j                  ||«       y r   )
r.   ÚgetInputStreamÚindexr/   ÚstateÚconsumeÚ_inputÚappendÚgetErrorRecoverySetÚconsumeUntil)r   r   r   Ú	followSets       r   r   zDefaultErrorStrategy.recoverŽ   s¯   € Ø×Ñ 
× 9Ñ 9Ó ;× AÑ AÒAØ×$Ñ$Ð0Ø× Ñ  D×$8Ñ$8Ñ8ð
 ×ÑÔ à(×/Ñ/×5Ñ5ˆÔØ×ÑÐ'Ø#%ˆDÔ Ø×Ñ×#Ñ# J×$4Ñ$4Ô5Ø×,Ñ,¨ZÓ8ˆ	Ø×Ñ˜* iÕ0r   c                 óˆ  — | j                  |«      ry |j                  j                  j                  |j                     }|j                  «       j                  d«      }|j                  j                  |«      }||v rd | _        t        j                  | _        y t        j                  |v r/| j                  €"|j                  | _        |j                  | _        y |j"                  t        j$                  t        j&                  t        j(                  t        j*                  fv r| j-                  |«      �y t/        |«      ‚|j"                  t        j0                  t        j2                  fv rT| j5                  |«       |j7                  «       }|j9                  | j;                  |«      «      }| j=                  ||«       y y ©Né   )r    Ú_interpÚatnÚstatesrK   ÚgetTokenStreamÚLAÚ
nextTokensr0   r   ÚINVALID_STATE_NUMBERr1   r   ÚEPSILONÚ_ctxÚ_stateNumberÚnextTokensStateÚ	stateTypeÚBLOCK_STARTÚSTAR_BLOCK_STARTÚPLUS_BLOCK_STARTÚSTAR_LOOP_ENTRYÚsingleTokenDeletionr   ÚPLUS_LOOP_BACKÚSTAR_LOOP_BACKÚreportUnwantedTokenÚgetExpectedTokensÚaddSetrO   rP   )r   r   ÚsÚlarZ   Ú	expectingÚwhatFollowsLoopIterationOrRules          r   r   zDefaultErrorStrategy.syncÌ   s{  € à×#Ñ# JÔ/Øà×Ñ×"Ñ"×)Ñ)¨*×*:Ñ*:Ñ;ˆØ×&Ñ&Ó(×+Ñ+¨AÓ.ˆà—^‘^×.Ñ.¨qÓ1ˆ
Ø�ÑØ%)ˆDÔ"Ü"*×"?Ñ"?ˆDÔØÜ�]‰]˜jÑ(Ø×%Ñ%Ð-ð *4¯©�Ô&Ø'1×'>Ñ'>�Ô$Øà�;‰;œ8×/Ñ/´×1JÑ1JÜ (× 9Ñ 9¼8×;SÑ;SðUñ Uð ×'Ñ'¨
Ó3Ð>Øä,¨ZÓ8Ð8à�[‰[œX×4Ñ4´h×6MÑ6MÐNÑNØ×$Ñ$ ZÔ0Ø"×4Ñ4Ó6ˆIØ-6×-=Ñ-=¸d×>VÑ>VÐWaÓ>bÓ-cÐ*Ø×Ñ˜jÐ*HÕIð r   c                 ó0  — |j                  «       }|�Q|j                  j                  t        j                  k(  rd}n)|j                  |j                  |j                  «      }nd}d| j                  |«      z   }|j                  ||j                  |«       y )Nú<EOF>z<unknown input>zno viable alternative at input )	rX   Ú
startTokenrD   r   ÚEOFÚgetTextrG   ÚescapeWSAndQuoterE   )r   r   r   ÚtokensÚinputÚmsgs         r   r@   z.DefaultErrorStrategy.reportNoViableAlternativeû   s}   € Ø×*Ñ*Ó,ˆØÐØ�|‰|× Ñ ¤%§)¡)Ò+Ø‘àŸ™ q§|¡|°Q×5EÑ5EÓF‘à%ˆEØ/°$×2GÑ2GÈÓ2NÑNˆØ×'Ñ'¨¨Q×-=Ñ-=¸qÕAr   c                 óê   — d| j                  |j                  «      z   dz   |j                  «       j                  |j                  |j
                  «      z   }|j                  ||j                  |«       y )Nzmismatched input ú expecting )ÚgetTokenErrorDisplayrG   ri   ÚtoStringÚliteralNamesÚsymbolicNamesrE   )r   r   r   rw   s       r   rA   z(DefaultErrorStrategy.reportInputMismatch  si   € Ø! D×$=Ñ$=¸a×>NÑ>NÓ$OÑOØñØ !× 3Ñ 3Ó 5× >Ñ >¸z×?VÑ?VÐXb×XpÑXpÓ qñrˆà×'Ñ'¨¨Q×-=Ñ-=¸qÕAr   c                 ó¶   — |j                   |j                  j                  «          }d|z   dz   |j                  z   }|j	                  ||j
                  |«       y )Nzrule Ú )Ú	ruleNamesr]   ÚgetRuleIndexrF   rE   rG   )r   r   r   ÚruleNamerw   s        r   rB   z*DefaultErrorStrategy.reportFailedPredicate  sO   € Ø×'Ñ'¨
¯©×(DÑ(DÓ(FÑGˆØ˜Ñ  3Ñ&¨¯©Ñ2ˆØ×'Ñ'¨¨Q×-=Ñ-=¸qÕAr   c                 ó2  — | j                  |«      ry | j                  |«       |j                  «       }| j                  |«      }| j	                  |«      }d|z   dz   |j                  |j                  |j                  «      z   }|j                  ||d «       y )Nzextraneous input ry   )	r    r9   ÚgetCurrentTokenrz   ri   r{   r|   r}   rE   )r   r   ÚtÚ	tokenNamerm   rw   s         r   rh   z(DefaultErrorStrategy.reportUnwantedToken4  s’   € Ø×#Ñ# JÔ/Øà× Ñ  Ô,Ø×&Ñ&Ó(ˆØ×-Ñ-¨aÓ0ˆ	Ø×*Ñ*¨:Ó6ˆ	Ø! IÑ-°Ñ=Ø× Ñ  ×!8Ñ!8¸*×:RÑ:RÓSñTˆà×'Ñ'¨¨Q°Õ5r   c                 ó.  — | j                  |«      ry | j                  |«       |j                  «       }| j                  |«      }d|j	                  |j
                  |j                  «      z   dz   | j                  |«      z   }|j                  ||d «       y )Nzmissing z at )	r    r9   r„   ri   r{   r|   r}   rz   rE   )r   r   r…   rm   rw   s        r   ÚreportMissingTokenz'DefaultErrorStrategy.reportMissingTokenP  s�   € Ø×#Ñ# JÔ/ØØ× Ñ  Ô,Ø×&Ñ&Ó(ˆØ×*Ñ*¨:Ó6ˆ	Ø˜9×-Ñ-¨j×.EÑ.EÀz×G_ÑG_Ó`Ñ`ØñØ×2Ñ2°1Ó5ñ6ˆà×'Ñ'¨¨Q°Õ5r   c                 ó¦   — | j                  |«      }|�|j                  «        |S | j                  |«      r| j                  |«      S t	        |«      ‚r   )re   rL   ÚsingleTokenInsertionÚgetMissingSymbolr   )r   r   ÚmatchedSymbols      r   r   z"DefaultErrorStrategy.recoverInline‰  sZ   € à×0Ñ0°Ó<ˆØÐ$ð ×ÑÔ Ø Ð ð ×$Ñ$ ZÔ0Ø×(Ñ(¨Ó4Ð4ô % ZÓ0Ð0r   c                 ó6  — |j                  «       j                  d«      }|j                  j                  }|j                  |j
                     }|j                  d   j                  }|j                  ||j                  «      }||v r| j                  |«       yy)NrT   r   TF)rX   rY   rU   rV   rW   rK   ÚtransitionsÚtargetrZ   r]   rˆ   )r   r   ÚcurrentSymbolTyperV   ÚcurrentStateÚnextÚexpectingAtLL2s          r   rŠ   z)DefaultErrorStrategy.singleTokenInsertionª  s‰   € Ø&×5Ñ5Ó7×:Ñ:¸1Ó=Ðð × Ñ ×$Ñ$ˆØ—z‘z *×"2Ñ"2Ñ3ˆØ×'Ñ'¨Ñ*×1Ñ1ˆØŸ™¨¨j¯o©oÓ>ˆØ Ñ.Ø×#Ñ# JÔ/Øàr   c                 óô   — |j                  «       j                  d«      }| j                  |«      }||v rD| j                  |«       |j	                  «        |j                  «       }| j                  |«       |S y )Né   )rX   rY   ri   rh   rL   r„   r=   )r   r   ÚnextTokenTyperm   rŒ   s        r   re   z(DefaultErrorStrategy.singleTokenDeletionË  st   € Ø"×1Ñ1Ó3×6Ñ6°qÓ9ˆØ×*Ñ*¨:Ó6ˆ	Ø˜IÑ%Ø×$Ñ$ ZÔ0ð
 ×ÑÔ à&×6Ñ6Ó8ˆMØ×Ñ˜ZÔ(Ø Ð àr   c           
      óh  — |j                  «       }| j                  |«      }|d   }|t        j                  k(  rd}ncd }|t	        |j
                  «      k  r|j
                  |   }|€'|t	        |j                  «      k  r|j                  |   }dt        |«      z   dz   }|}|j                  «       j                  d«      }|j                  t        j                  k(  r|�|}|j                  «       j                  |j                  ||t        j                  dd|j                  |j                   «      S )Nr   z<missing EOF>z	<missing Ú>r*   )r„   ri   r   rr   Úlenr|   r}   ÚstrrX   ÚLTrD   ÚgetTokenFactoryÚcreateÚsourceÚDEFAULT_CHANNELÚlineÚcolumn)	r   r   ÚcurrentSymbolrm   ÚexpectedTokenTypeÚ	tokenTextÚnameÚcurrentÚlookbacks	            r   r‹   z%DefaultErrorStrategy.getMissingSymbolï  s  € Ø"×2Ñ2Ó4ˆØ×*Ñ*¨:Ó6ˆ	Ø% a™LÐØœeŸi™iÒ'Ø'‰IàˆDØ ¤3 z×'>Ñ'>Ó#?Ò?Ø!×.Ñ.Ð/@ÑA�Øˆ|Ð 1´C¸
×8PÑ8PÓ4QÒ QØ!×/Ñ/Ð0AÑB�Ø#¤c¨$£iÑ/°#Ñ5ˆIØˆØ×,Ñ,Ó.×1Ñ1°"Ó5ˆØ�<‰<œŸ™Ò" xÐ';ØˆGØ×)Ñ)Ó+×2Ñ2°7·>±>Ø˜y¬%×*?Ñ*?Ø��G—L‘L '§.¡.ó2ð 	2r   c                 ó"   — |j                  «       S r   )ri   r   s     r   ri   z&DefaultErrorStrategy.getExpectedTokens  s   € Ø×+Ñ+Ó-Ð-r   r…   c                 ó¼   — |€y|j                   }|€;|j                  t        j                  k(  rd}ndt	        |j                  «      z   dz   }| j                  |«      S )Nz
<no token>rp   Ú<r˜   )ÚtextrD   r   rr   rš   rt   )r   r…   rk   s      r   rz   z)DefaultErrorStrategy.getTokenErrorDisplay  sU   € Øˆ9ØØ�F‰FˆØˆ9Ø�v‰v”u—y‘yÒ Ø‘àœ#˜aŸf™f›+Ñ%¨Ñ+�Ø×$Ñ$ QÓ'Ð'r   rk   c                 ó~   — |j                  dd«      }|j                  dd«      }|j                  dd«      }d|z   dz   S )NÚ
z\nÚz\rÚ	z\tÚ')Úreplace)r   rk   s     r   rt   z%DefaultErrorStrategy.escapeWSAndQuote  s@   € Ø�I‰I�d˜5Ó!ˆØ�I‰I�d˜5Ó!ˆØ�I‰I�d˜5Ó!ˆØ�Q‰w˜‰}Ðr   c                 ó¢  — |j                   j                  }|j                  }t        «       }|��|j                  dk\  rr|j
                  |j                     }|j                  d   }|j                  |j                  «      }|j                  |«       |j                  }|�|j                  dk\  rŒr|j                  t        j                  «       |S )Nr   )rU   rV   r]   r   ÚinvokingStaterW   rŽ   rZ   ÚfollowStaterj   Ú	parentCtxÚ	removeOner   r\   )r   r   rV   ÚctxÚ
recoverSetr³   ÚrtÚfollows           r   rO   z(DefaultErrorStrategy.getErrorRecoverySet|  s°   € Ø× Ñ ×$Ñ$ˆØ�o‰oˆÜ “]ˆ
Øˆo #×"3Ñ"3°QÒ"6àŸJ™J s×'8Ñ'8Ñ9ˆMØ×*Ñ*¨1Ñ-ˆBØ—^‘^ B§N¡NÓ3ˆFØ×Ñ˜fÔ%Ø—-‘-ˆCð ˆo #×"3Ñ"3°QÓ"6ð 	×ÑœUŸ]™]Ô+ØÐr   Úset_c                 ó  — |j                  «       j                  d«      }|t        j                  k7  rN||vrI|j	                  «        |j                  «       j                  d«      }|t        j                  k7  r||vrŒGy y y y rS   )rX   rY   r   rr   rL   )r   r   r»   Úttypes       r   rP   z!DefaultErrorStrategy.consumeUntil‹  sn   € Ø×)Ñ)Ó+×.Ñ.¨qÓ1ˆØ”u—y‘yÒ ¨°$©Ø×ÑÔ Ø×-Ñ-Ó/×2Ñ2°1Ó5ˆEð ”u—y‘yÒ ¨°$«Ð ¨Ð r   )"r#   r$   r%   r,   r&   r   r9   r    r5   r=   r   r"   r   r   r   r@   r   rA   rB   rh   rˆ   r   rŠ   re   r‹   ri   r   rz   rš   rt   rO   ÚsetrP   Ú__classcell__)r2   s   @r   r(   r(   )   sA  ø„ ô ð.+˜vó +ð&¨Vó &ð&¨Vó &ð!¨6ó !ð+ Vó +ð,L Vð LÐ/Có Lð.1 ð 1Ð+?ó 1ð|%˜fó %ð^
B°6ð 
BÐ=Qó 
Bð*B¨Vð BÐ7Mó BòBð,
6¨Vó 
6ð86¨Fó 6ðr1 vó 1ðB¨fó ðB¨Vó ðH2¨&ó 2ð*.¨6ó .ð	( Uó 	(ð ó ðD¨Vó ð6 fð 6°3÷ 6r   r(   c                   ó4   — e Zd Zdedefd„Zdefd„Zdefd„Zy)ÚBailErrorStrategyr   r   c                 ó`   — |j                   }|�||_        |j                  }|�Œt        |«      ‚r   )r]   Ú	exceptionrµ   r
   )r   r   r   Úcontexts       r   r   zBailErrorStrategy.recover´  s9   € Ø—/‘/ˆØÐ!Ø !ˆGÔØ×'Ñ'ˆGð Ñ!ô )¨Ó+Ð+r   c                 ó:   — | j                  |t        |«      «       y r   )r   r   r   s     r   r   zBailErrorStrategy.recoverInline¾  s   € Ø�‰�ZÔ!7¸
Ó!CÕDr   c                  ó   — y r   r   r   s     r   r   zBailErrorStrategy.syncÂ  r   r   N)r#   r$   r%   r&   r   r   r   r   r   r   r   rÁ   rÁ   ®  s0   „ ð, ð ,Ð+?ó ,ðE vó Eð˜fô r   rÁ   )ÚsysÚantlr4.IntervalSetr   Úantlr4.Tokenr   Úantlr4.atn.ATNStater   Úantlr4.error.Errorsr   r   r   r	   r
   r&   Úobjectr   r(   rÁ   r   r   r   Ú<module>rÍ      sQ   ðó Ý *å Ý (÷9õ 9ð 
€ô�Fô ô0f	6˜=ô f	6ôJÐ,ô ñ. r   