<_9Y� � @ 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 Gd 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 � S)
�
SJISProberc sH t t | � j � t t � | _ t � | _ t � | _ | j
� d S)N) �superr �__init__r r � coding_smr �distribution_analyzerr �context_analyzer�reset)�self)� __class__� �/sjisprober.pyr % s
zSJISProber.__init__c s$ t t | � j � | j j � d S)N)r
r r r )r )r r r r , s zSJISProber.resetc C s
| j j S)N)r � charset_name)r r r r r 0 s zSJISProber.charset_namec C s d S)NZJapaneser )r r r r �language4 s zSJISProber.languagec C s� xWt t | � � D]C} | j j | | � } | t j k rm | j j d | j | j | � t
j | _ Pq | t j
k r� t
j | _ Pq | t j k r | j j � } | d k r| d | j d <| j j | j d | d � | � | j j | j | � q | j j | | d | | d | � | � | j j | | d | d � | � q W| d | j d <| j t
j k r�| j j � r�| j � | j k r�t
j | _ | j S)Nz!%s %s prober hit error at byte %s� r � � ���)�range�lenr Z
next_stater �ERROR�logger�debugr r r ZNOT_ME�_stateZITS_MEZFOUND_ITZSTARTZget_current_charlenZ
_last_charr �feedr
�stateZ DETECTINGZgot_enough_data�get_confidenceZSHORTCUT_THRESHOLD)r Zbyte_str�iZ coding_stateZchar_lenr r r r 8 s6 zSJISProber.feedc C s+ | j j � } | j j � } t | | � S)N)r r" r
�max)r Z context_confZ distrib_confr r r r" Y s zSJISProber.get_confidence)
�__name__�
__module__� __qualname__r r �propertyr r r r" r r )r r r $ s !r N)Zmbcharsetproberr Zcodingstatemachiner Zchardistributionr Zjpcntxr Zmbcssmr Zenumsr r r r r r r �<module> s