U
    yc                     @   s\   d dl mZm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 G dd deZd	S )
    )OptionalUnion   )CharDistributionAnalysis)CharSetProber)CodingStateMachine)LanguageFilterMachineStateProbingStatec                       sf   e Zd ZdZejfedd fddZdd fddZee	e
f ed	d
dZedddZ  ZS )MultiByteCharSetProberz 
    MultiByteCharSetProber
    N)lang_filterreturnc                    s(   t  j|d d | _d | _td| _d S )N)r        )super__init__distribution_analyzer	coding_sm	bytearray
_last_char)selfr   	__class__ B/tmp/pip-build-0we0fmvo/pip/pip/_vendor/chardet/mbcharsetprober.pyr   +   s    zMultiByteCharSetProber.__init__)r   c                    s8   t    | jr| j  | jr*| j  td| _d S )Nr   )r   resetr   r   r   r   r   r   r   r   r   1   s    


zMultiByteCharSetProber.reset)byte_strr   c                 C   s  | j d k	st| jd k	stt|D ]\}}| j |}|tjkrd| jd| j	| j
| tj| _ q|tjkrztj| _ q|tjkr$| j  }|dkr|| jd< | j| j| q$| j||d |d  | q$|d | jd< | jtjkr| j r|  | jkrtj| _| jS )Nz!%s %s prober hit error at byte %sr   r   )r   AssertionErrorr   	enumerate
next_stater	   ERRORloggerdebugcharset_namelanguager
   NOT_ME_stateITS_MEFOUND_ITSTARTget_current_charlenr   feedstate	DETECTINGgot_enough_dataget_confidenceSHORTCUT_THRESHOLD)r   r   iZbytecoding_statechar_lenr   r   r   r,   9   s:    




 zMultiByteCharSetProber.feedc                 C   s   | j d k	st| j  S )N)r   r   r0   r   r   r   r   r0   ]   s    z%MultiByteCharSetProber.get_confidence)__name__
__module____qualname____doc__r   NONEr   r   r   bytesr   r
   r,   floatr0   __classcell__r   r   r   r   r   &   s
   $r   N)typingr   r   chardistributionr   charsetproberr   codingstatemachiner   enumsr   r	   r
   r   r   r   r   r   <module>   s
   