U
    X                     @   s`   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZmZ G dd deZd	S )
   )MultiByteCharSetProber)CodingStateMachine)SJISDistributionAnalysis)SJISContextAnalysis)SJIS_SM_MODEL)ProbingStateMachineStatec                       sP   e Zd Z fddZ fddZedd Zedd Zd	d
 Zdd Z	  Z
S )
SJISProberc                    s4   t t|   tt| _t | _t | _	| 
  d S N)superr	   __init__r   r   	coding_smr   distribution_analyzerr   context_analyzerresetself	__class__ 4/usr/lib/python3/dist-packages/chardet/sjisprober.pyr   %   s
    
zSJISProber.__init__c                    s   t t|   | j  d S r
   )r   r	   r   r   r   r   r   r   r   ,   s    zSJISProber.resetc                 C   s   | j jS r
   )r   charset_namer   r   r   r   r   0   s    zSJISProber.charset_namec                 C   s   dS )NZJapaneser   r   r   r   r   language4   s    zSJISProber.languagec                 C   sN  t t|D ]}| j|| }|tjkrP| jd| j| j	| t
j| _ qq|tjkrjt
j| _ qq|tjkr| j }|dkr|d | jd< | j| jd| d  | | j| j| q| j||d | |d |  | | j||d |d  | q|d | jd< | jt
jkrH| j rH|  | jkrHt
j| _| jS )Nz!%s %s prober hit error at byte %s    r         )rangelenr   Z
next_stater   ZERRORZloggerdebugr   r   r   ZNOT_MEZ_stateZITS_MEZFOUND_ITZSTARTZget_current_charlenZ
_last_charr   feedr   stateZ	DETECTINGZgot_enough_dataget_confidenceZSHORTCUT_THRESHOLD)r   Zbyte_striZcoding_stateZchar_lenr   r   r   r    8   sH    
  


zSJISProber.feedc                 C   s   | j  }| j }t||S r
   )r   r"   r   max)r   Zcontext_confZdistrib_confr   r   r   r"   Y   s    

zSJISProber.get_confidence)__name__
__module____qualname__r   r   propertyr   r   r    r"   __classcell__r   r   r   r   r	   $   s   

!r	   N)Zmbcharsetproberr   Zcodingstatemachiner   Zchardistributionr   Zjpcntxr   Zmbcssmr   Zenumsr   r   r	   r   r   r   r   <module>   s   