File: //usr/lib/python3/dist-packages/ocrmypdf/pdfinfo/__pycache__/layout.cpython-38.pyc
U
��Z^E$ � @ s` d dl Z d dlmZ d dlmZ d dlmZ d dlZd dlZd dl Zd dl
Zd dl m Z d dl
mZ d dlmZmZmZmZ d dlmZ d d lmZmZmZ d d
lmZ d d l m Z m Z d d
l m Z e �!d�Z"ej#dk�r dd� Z$e$ej%_$ej&Z'd$dd�Z(e(e_&ej&Z)dd� Z*e*e_&dd� Z+dd� Z,dd� Z-G d d � d e�Z.G d d � d e �Z/d d!� Z0d"d#� Z1dS )%� N)�copysign)�Path)�patch)�PDFLayoutAnalyzer)�glyphname2unicode)�LAParams�LTChar�LTPage� LTTextBox)�PDFTextExtractionNotAllowed)�PDFFont�
PDFSimpleFont�PDFUnicodeNotDefined)�PDFPage)�bbox2str�
matrix2str� )�EncryptedPdfErrorz[0-9]+Z20181108c C s� | t krt | S | �d�s$| �d�r,t| ��| �d�rlztt| dd� d��W S tk
rj t| ��Y nX t�| �}|s�t| ��tt|�d���S )aY Fix pdfminer's name2unicode function
Font cids that are mapped to names of the form /g123 seem to be, by convention
characters with no corresponding Unicode entry. These can be subsetted fonts
or symbolic fonts. There seems to be no way to map /g123 fonts to Unicode,
barring a ToUnicode data structure.
�g�aZuni� N� r ) r �
startswith�KeyError�chr�int�
ValueError�
STRIP_NAME�search�group)�name�m� r" �9/usr/lib/python3/dist-packages/ocrmypdf/pdfinfo/layout.py� name2unicode- s
r$ c C s& t | |||� | jdkr"| j | _d S )Nr )�original_PDFFont_init�descent)�self�
descriptor�widthsZ
default_widthr" r" r# �PDFFont__init__G s
r* c C s& t | |||� | js"d|kr"i | _d S )NZEncoding)�original_PDFSimpleFont_initZ unicode_mapZ cid2unicode)r' r( r) �specr" r" r# �PDFSimpleFont__init__[ s r- c C s8 | j d | j d }|dkr(| j| j }|td| j� S )Nr � r � �?)�bbox�ascentr& r �vscale)r' �hr" r" r# �!PDFType3Font__PScript5_get_heightn s r4 c C s | j td| j� S �Nr/ )r& r r2 �r' r" r" r# �"PDFType3Font__PScript5_get_descentu s r7 c C s | j td| j� S r5 )r1 r r2 r6 r" r" r# �!PDFType3Font__PScript5_get_ascenty s r8 c s< e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d
� Z� Z S ) �LTStateAwareCharzDA subclass of LTChar that tracks text render mode at time of drawing)�
rendermode�_text�matrix�fontname�advZupright�size�widthZheightr0 Zx0Zx1Zy0Zy1c s* t � �||||||||| |
�
| j| _d S �N)�super�__init__Zrenderr: ) r' r< �font�fontsize�scaling�rise�text� textwidth�textdisp�ncs� graphicstate� textstate�� __class__r" r# rC � s � zLTStateAwareChar.__init__c C st t | jt�ot |jt�}z<|r,| j|jkW S | j\}}|j\}}||koR| j|jkW S ttfk
rn Y dS X dS )a
Check if characters can be combined into a textline
We consider characters compatible if:
- the Unicode mapping is known, and both have the same render mode
- the Unicode mapping is unknown but both are part of the same font
FN)�
isinstancer; �strr: r �AttributeError)r' �objZboth_unicode_mappedZfont0�_Zfont1r" r" r# �
is_compatible� s
z LTStateAwareChar.is_compatiblec C s t | jt�rdS | jS )Nu �)rP r; �tupler6 r" r" r# �get_text� s zLTStateAwareChar.get_textc C s0 d| j jt| j�t| j�| j| j| j| � � f S )Nz6<%s %s matrix=%s rendermode=%r font=%r adv=%s text=%r>)
rO �__name__r r0 r r< r: r= r> rW r6 r" r" r# �__repr__� s �zLTStateAwareChar.__repr__)
rX �
__module__� __qualname__�__doc__� __slots__rC rU rW rY �
__classcell__r" r" rN r# r9 } s r9 c sb e Zd ZdZd� fdd� Z� fdd�Zdd � Z� fd
d �Zd d
� Zdd� Z dd� Z
dd� Z � Z S )�TextPositionTrackerz=A page layout analyzer that pays attention to text visibilityr. Nc s& t � �|||� d | _d | _d | _d S rA )rB rC rM �result�cur_item)r' Zrsrcmgr�pageno�laparamsrN r" r# rC � s z TextPositionTracker.__init__c s" t � �||� t| j|j�| _d S rA )rB �
begin_pager rb Zmediaboxra )r' �pageZctmrN r" r# rd � s z TextPositionTracker.begin_pagec C sl | j rttt| j ����t| jt�s6ttt| j����| jd k rN| j� | j� | j
d7 _
| � | j� d S )Nr. ) Z_stack�AssertionErrorrQ �lenrP ra r �typerc Zanalyzerb �receive_layout)r' re r" r" r# �end_page� s
z TextPositionTracker.end_pagec s" |� � | _t� �| j|||� d S rA )�copyrM rB �
render_string)r' rM �seqrK rL rN r" r# rl � s
z!TextPositionTracker.render_stringc
C s� z(|� |�} t| t�s&ttt| ����W n tk
rH | �||�} Y nX |�|�}
|�|�} t |||||| |
| ||| j
� } | j � | � | j
S rA )Z to_unichrrP rQ rf rh r �handle_undefined_charZ
char_widthZ char_dispr9 rM ra �addr> )
r' r< rD rE rF rG �cidrK rL rH rI rJ �itemr" r" r# � render_char� s,
�
z TextPositionTracker.render_charc C s
|j |fS rA )r= )r' rD rp r" r" r# rn s z)TextPositionTracker.handle_undefined_charc C s
|| _ d S rA �r` )r' Zltpager" r" r# ri s z"TextPositionTracker.receive_layoutc C s | j S rA rs r6 r" r" r# �
get_result
s z TextPositionTracker.get_result)r. N)
rX rZ r[ r\ rC rd rj rl rr rn ri rt r^ r" r" rN r# r_ � s r_ c C s� t jjdd�}t|t� d�}t j�||�}|rJtjddtt t
d�}|� � z^z>t
| ��d��&}tj||gdd�}|�t|�� W 5 Q R X W n tk
r� t� �Y nX W 5 |r�|� � X |�� S ) NT)Zcaching)rc z pdfminer.pdffont.PDFType3Font)r, Z
get_ascentZ get_descentZ
get_height�rbr )ZpagenosZmaxpages)�pdfminerZ pdfinterpZPDFResourceManagerr_ r ZPDFPageInterpreterr Zmultipler8 r7 r4 �start�stopr �openr Z get_pagesZ process_page�nextr r rt ) Zinfilerb Z
pscript5_modeZrmanZdevZinterpZpatcher�fre r" r" r# �get_page_analysis s* �
r| c c sL | D ]B}t |t�r|V qzt|�E d H W q tk
rD Y qY qX qd S rA )rP r
�get_text_boxes� TypeError)rS Zchildr" r" r# r} * s
r} )N)2�reZmathr �pathlibr Z
unittest.mockr rv Zpdfminer.encodingdbZpdfminer.pdfdeviceZpdfminer.pdfinterpZpdfminer.converterr Zpdfminer.glyphlistr Zpdfminer.layoutr r r r
Zpdfminer.pdfdocumentr Zpdfminer.pdffontr r
r Zpdfminer.pdfpager Zpdfminer.utilsr r �
exceptionsr �compiler � __version__r$ Z
encodingdbrC r% r* r+ r- r4 r7 r8 r9 r_ r| r} r" r" r" r# �<module> s@
S>