File: //lib/python3/dist-packages/ocrmypdf/__pycache__/_graft.cpython-38.pyc
U
��Z^l) � @ s^ d dl Z d dlmZ d dlmZ d dlZdZdd� Zdd� Zd d
� Z d d � Z
G d
d� d�Z dS )� N)�suppress)�Path�d c C sv d| krt �i �| d< | d }z |d }W n tk
rH t �i �}Y nX |dk rb||krb|||< ||d< ||d< dS )z?Update this page's fonts with a reference to the Glyphless fontz
/Resources�/FontNz/ProcSet)�pikepdfZ
Dictionary�KeyError)�page�font�font_key�procsetZ resourcesZfonts� r �1/usr/lib/python3/dist-packages/ocrmypdf/_graft.py�_update_page_resources s r c s< g }d}d}g }|� � t�|d�D ]�\}}|sf|t�d�krVd}d}|�||f� q�|�||f� q$|t�d�kr||d }|�||f� |t�d�kr$d}|dkr�|�|� |�� q$d d
� � g }|D ]V\}}|t�d �kr�|d } | �� }
n$d �� fd
d�|D ��d |�� }
|�|
� q�d�|�} t� | | �|_
d S )NFr � ZBTTZTrZET� c S s2 z
| � � W S tk
r, t| ��d� Y S X d S )N�ascii)�unparse�AttributeError�str�encode)�opr r r
�convertG s
z%strip_invisible_text.<locals>.convertz INLINE IMAGE� c 3 s | ]}� |�V qd S �Nr )�.0r �r r r
� <genexpr>T s z'strip_invisible_text.<locals>.<genexpr>�
) Zpage_contents_coalescer Zparse_content_streamZOperator�append�extend�clearr �join�Stream�Contents) Zpdfr �streamZ in_text_objZ render_modeZ text_objectsZoperands�operator�linesZiim�lineZcontent_streamr r r
�strip_invisible_text. s<
$
r( c s� |� d� t|��� jdkr dS t�|���jd j�� } | j� |�� �fdd�t
d�D �}
|
d |
d |
d |
d } } � fd
d�t
d�D �}
|
d |
d |
d |
d }
}t� � � | d | d �}t� � � |
d |d �}t� � � |
d |
d �}| d }t� � �
|�}|d k�r.| | } } |
| }|| }t� � �||�}|| | | | }d
|�� | d } t�| | �}|�r�t| � � � j|dd� t� |||d� ��� dS )zAInsert the text layer from text page 0 on to pdf_base at page_numZGraftingr Nc s g | ]}t � jd j| ��qS )r )�float�pages�MediaBox�r �v)�pdf_textr r
�
<listcomp>n s z%_graft_text_layer.<locals>.<listcomp>� � r � c s g | ]}t � j| ��qS r )r) r+ r, )� base_pager r
r/ q s �h )�Z i s q %s cm
s
Q
T)Zprepend�r r r
r )�debugr �stat�st_sizer �openr* r# Z
read_bytes�p�rangeZ PdfMatrixZ
translatedZrotatedZscaledr r" r( Zpage_contents_addr �close)�pdf_base�page_num�textr r
r �rotation�strip_old_text�logZpdf_text_contentsZmediaboxZwtZhtZwpZhp� translateZ untranslateZcorner�rotateZscale_xZscale_yZscaleZctmZnew_text_layerr )r3 r. r
�_graft_text_layer[ sB
""
�rF c C s� d\}}d}z�t �| ���}z|jd j�di �}W n( tttfk
rZ Y W 5 Q R � W dS X |D ] }|�|d�}|dk r`|} q�q`|r�|�|�}||fW 5 Q R � W S Q R X W n t t j
fk
r� Y dS X dS )z0Copy a font from the filename text into pdf_base�NN)z/f-0-0z/F1r r N) r r: r* � Resources�getr �
IndexErrorr Z copy_foreign�FileNotFoundErrorZPdfError) r@ r> r r
Zpossible_font_namesr. Zpdf_text_fonts�fZ
pdf_text_fontr r r
�
_find_font� s$
rM c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
OcrGrafterc C sr || _ |j| _t|j��� | _t�| j�| _d\| _ | _
|j | _ |� d�| _
| j�tj�d��| _d| _d| _d S )NrG zgraft_layers.pdfs& [ /PDF /Text /ImageB /ImageC /ImageI ]r2 r )�contextrC r �origin�resolve� path_baser r: r> r r
�pdfinfoZget_path� output_fileZ
make_indirectZObject�parser � emplacements�
interim_count)�selfrO r r r
�__init__� s
�zOcrGrafter.__init__c C s� |\}}}}}|r,| j s,t|| j�\| _ | _d}| j| j}|rLt|��� nd } | d k r�| | jkr�| j �
d� t � |��T}
| j
d7 _
|
jd } | jj�| � | jjd } | jj| �| � | jjd= W 5 Q R X d}|r�|}|}
|
| d }| j �
d|� d |
� d
|� d
|� d
|� �
� |�rX| j �rX| jjj}t| j|d || j | j|| j|| j d � || d | jj| _| j
t dk�r�| �� d S ) NFzEmplacement updater2 r ���Tr4 zRotations for page z$: [text, auto, misalign, content] = z, ) r> r? r@ r r
rA r rB rC )r rM r> r
rS rA r rQ rR rC r7 r r: rV r* r ZemplacerO �optionsZredo_ocrrF r ZRotate�MAX_REPLACE_PAGES�save_and_reload)rX Z page_resultZpagenoZimager@ Z_sidecarZautorotate_correctionZ
emplaced_pageZcontent_rotationZ
path_imageZ pdf_imageZforeign_image_pageZlocal_image_pageZ
text_rotationZtext_misalignedZ strip_oldr r r
�
graft_page� sR
�
��zOcrGrafter.graft_pagec C s� | j jd }t|| j| j| jd� | jd| jd � d� }| jj j
sbt t �� t
�|� W 5 Q R X | jd| jd � d� }| j �|� | j �� t�|�| _ | j jd jj| _d\| _| _| jd7 _d S )Nr r6 Z_workingr2 z.pdfrG )r> r* r r r
r rT rW rO r[ Zkeep_temporary_filesr rK �os�unlink�saver= r r: rH ZProcSet)rX Zpage0Zold_fileZ next_filer r r
r] � s$ �
zOcrGrafter.save_and_reloadc C s | j �| j� | j �� | jS r )r> ra rT r= )rX r r r
�finalize s
zOcrGrafter.finalizeN)�__name__�
__module__� __qualname__rY r^ r] rb r r r r
rN � s 6 rN ) r_ �
contextlibr �pathlibr r r\ r r( rF rM rN r r r r
�<module> s -@