File: //lib/python3/dist-packages/ocrmypdf/__pycache__/leptonica.cpython-38.pyc
U
��Z^dv �
@ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d dlmZ d d
lmZ d d lmZ d d lmZ e�e�Zejd
kr�dZ e� ej d< ndZ e
e �Z e s�ed��ze� e �Z e �!e j"� W n8 ej#k
�r0 Z$ zede � d��e$�W 5 dZ$[$X Y nX G dd� d�Z%G dd� de&�Z'G dd� de'�Z(G dd� d�Z)G d d � d e)�Z*G d d � d e)�Z+G d d!� d!e)e�Z,G d"d#� d#e)�Z-G d$d%� d%e)e�Z.G d&d'� d'e)e�Z/G d(d)� d)e)�Z0e d d*�d+d,� �Z1d-d.� Z2dHd3d4�Z3ed5k�r�e j4d6d7�Z5e5j6d8d9d:�Z7e7�8d.�Z9e9j:d;d<d=d>e;d?d@dA� e9j:dBdCdD� e9j:dEdFdD� e9j<e2dG� e5�=� Z>e>�?e>� dS )I� N)�Sequence)�suppress)� find_library)� lru_cache)�BytesIO)�fspath)�
TemporaryFile� )�MissingDependencyError)� shim_paths_with_program_files)�ffi�ntz liblept-5�PATH�lepta�
---------------------------------------------------------------------
This error normally occurs when ocrmypdf can't find the Leptonica
library, which is usually installed with Tesseract OCR. It could be that
Tesseract is not installed properly, we can't find the installation
on your system PATH environment variable.
The library we are looking for is usually called:
liblept-5.dll (Windows)
liblept*.dylib (macOS)
liblept*.so (Linux/BSD)
Please review our installation procedures to find a solution:
https://ocrmypdf.readthedocs.io/en/latest/installation.html
---------------------------------------------------------------------
zLeptonica library found at z , but we could not access itc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �_LeptonicaErrorTrapa+
Context manager to trap errors reported by Leptonica.
Leptonica's error return codes don't provide much informatino about what
went wrong. Leptonica does, however, write more detailed errors to stderr
(provided this is not disabled at compile time). The Leptonica source
code is very consistent in its use of macros to generate errors.
This context manager redirects stderr to a temporary file which is then
read and parsed for error messages. As a side benefit, debug messages
from Leptonica are also suppressed.
c C s d | _ d| _d| _d S )N���F)�tmpfile�copy_of_stderr� no_stderr��self� r �4/usr/lib/python3/dist-packages/ocrmypdf/leptonica.py�__init__] s z _LeptonicaErrorTrap.__init__c C s� ddl m} t� | _tt�� tj�� W 5 Q R X z2t �
tj� � �| _ t j
| j� � tj� � dd� W nN tk
r� d | _ d| _t j
| j� � ddd� Y n |k
r� d | _ Y nX d S )Nr )�UnsupportedOperationF)Z inheritableT� )�ior r r r �AttributeError�sys�stderr�flush�os�dup�filenor �dup2r )r r r r r � __enter__b s
z _LeptonicaErrorTrap.__enter__c C s� t t�� tj�� W 5 Q R X | jd k rHt�| jtj�� � t� | j� | j
rXt� d� | j � d� | j �
� jdd�}| j � � |r�t�|� d|kr�d|kr�t� �d|kr�t� �d|kr�t� �t|��d S )
Nr r �replace)�errors�Errorzimage file not foundzpixWrite: stream not openedzindex not validF)r r r r r r r! r$ r# �closer r �seek�read�decode�loggerZwarning�FileNotFoundError�LeptonicaIOError�
IndexError�LeptonicaError)r �exc_type� exc_value� tracebackZleptonica_outputr r r �__exit__x s*
z _LeptonicaErrorTrap.__exit__N)�__name__�
__module__� __qualname__�__doc__r r% r5 r r r r r N s r c @ s e Zd ZdS )r1 N�r6 r7 r8 r r r r r1 � s r1 c @ s e Zd ZdS )r/ Nr: r r r r r/ � s r/ c @ s0 e Zd ZdZdd� ZdZdd� Zedd� �Zd S )
�LeptonicaObjecta� General wrapper for Leptonica objects
When Leptonica returns an object, we bundled it in a wrapper class, which
manages its memory. The wrapper class assumes that it will be calling some
sort of lept.thingDestroy() function when the instance is deleted. Most
Leptonica objects are reference counted, and destroy decrements the
refcount.
Most of the time, when Leptonica returns something, we wrap and it the job
is done. When wrapping objects that came from a Leptonica container, like
a PIXA returning PIX, the subclass must clone the object before passing it
here, to maintain the reference count.
CFFI ensures that the destroy function is called at garbage collection time
so we do not need to mess with __del__.
c C s d S �Nr )�cdatar r r �<lambda>� � zLeptonicaObject.<lambda>� c C s& |st d| j ��t�|| j�| _d S )NzTried to wrap a NULL )�
ValueError�LEPTONICA_TYPENAMEr �gc�_destroy�_cdata)r r= r r r r � s zLeptonicaObject.__init__c C s" t �d�| j�|�}| �|� dS )zDestroy some cdataz{} **N)r �new�formatrB �
cdata_destroy)�clsr= Zppr r r rD � s zLeptonicaObject._destroyN) r6 r7 r8 r9 rH rB r � classmethodrD r r r r r; � s r; c @ s� e Zd ZdZdZejZdd� Zdd� Z dd� Z
d d
� Z d d � Z e
d
d� �Ze
dd� �Ze
dd� �Ze
dd� �Ze
dd� �Ze
dd� �Zedd� �Zedd � �Zdad d �Zed d!� �Zd"d#� Zd$d%� Zdbd&d'�Zd(d)� Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z!dcd7d8�Z"ddd?d@�Z#dedAdB�Z$dCdDdEd d d e%j&fdFdG�Z'dfdIdJ�Z(dgdKdL�Z)dhdQdR�Z*e+e,dSdT�dUdV� ��Z-e+dWdX� �Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� Z2d9S )i�Pixa�
Wrapper around leptonica's PIX object.
Leptonica uses referencing counting on PIX objects. Also, many Leptonica
functions return the original object with an increased reference count
if the operation had no effect (for example, image skew was found to be 0).
This has complications for memory management in Python. Whenever Leptonica
returns a PIX object (new or old), we wrap it in this class, which
registers it with the FFI garbage collector. pixDestroy() decrements the
reference count and only destroys when the last reference is removed.
Leptonica's reference counting is not threadsafe. This class can be used
in a threadsafe manner if a Python threading.Lock protects the data.
This class treats Pix objects as immutable. All methods return new
modified objects. This allows convenient chaining:
>>> Pix.open('filename.jpg').scale((0.5, 0.5)).deskew().show()
ZPIXc
C sJ | j rBd}|�| j j| j j| j jtt�d| j ��| j jr<dnd�S dS d S )Nz:<leptonica.Pix image size={0}x{1} depth={2}{4} at 0x{3:x}>Zintptr_tz
(colormapped)r@ z<leptonica.Pix image NULL>) rE rG �w�h�d�intr �cast�colormap)r �sr r r �__repr__� s �z Pix.__repr__c C s^ t �d�}t �d�}t�||| jd�}|dkr6td��t �d|d �}t �||d �dd� S )zCiPython display hook
returns png version of image
z
l_uint8 **�size_t *r �pixWriteMemPng�char *N)r rF r rU rE r/ rP �buffer)r �data�size�err� char_datar r r �
_repr_png_� s
zPix._repr_png_c C sp t �d�}t �d�}t�| j||�}|dkr4td��t �d|d �}t �||d �d d � }t�|� t |d�S )Nz l_uint32 **rT r �pixSerializeToMemoryrV )rX )
r rF r r] rE r/ rP rW � lept_free�dict)r rX rY rZ r[ Z
data_bytesr r r � __getstate__� s
zPix.__getstate__c C s@ t �d|d �}t �d|�}t�|t|d ��}t�| |� d S )Nzchar[]rX z
l_uint32 *)r rF rP r ZpixDeserializeFromMemory�lenrK r )r �stateZ cdata_bytesZ cdata_uint32�pixr r r � __setstate__
s zPix.__setstate__c C sT t |t�stS t�dd�}t� �" t�| j|j|�}|r>t � �W 5 Q R X t
|d �S )Nz l_int32 *r ) �
isinstancerK �NotImplementedr rF r r ZpixEqualrE � TypeError�bool)r �otherZsamerZ r r r �__eq__ s
z
Pix.__eq__c C s | j jS r< �rE rL r r r r �width s z Pix.widthc C s | j jS r< �rE rM r r r r �height s z
Pix.heightc C s | j jS r< )rE rN r r r r �depth# s z Pix.depthc C s | j j| j jfS r< )rE rL rM r r r r rY ' s zPix.sizec C s d| j j| j jfiS )N�dpi)rE ZxresZyresr r r r �info+ s zPix.infoc C s0 | j dkrdS | j dkr dS | jjs(dS dS dS )zReturn mode like PIL.Imager �1� �RGB�L�PN)ro rE rQ r r r r �mode/ s
zPix.modec C s t �dt� | �|�S )NzUse Pix.open() instead)�warnings�warn�DeprecationWarning�open)rI �pathr r r r+ ; s zPix.readc C sf t |d��R}|�� }t�|�}t� �. | t�|t|���W 5 Q R � W 5 Q R � S Q R X W 5 Q R X dS )z�Load an image file into a PIX object.
Leptonica can load TIFF, PNM (PBM, PGM, PPM), PNG, and JPEG. If
loading fails then the object will wrap a C null pointer.
�rbN)r{ r+ r � from_bufferr r �
pixReadMemra )rI r| �py_filerX rW r r r r{ @ s
zPix.openr c
C s� t �t�|��}t|d��|}t�d�}t�d�}t� �8 |t jkrXt � ||| j
||� nt � ||| j
|� W 5 Q R X t� |d |d �}|�
|� W 5 Q R X dS )z�Write pix to the filename, with the extension indicating format.
jpeg_quality -- quality (iff JPEG; 1 - 100, 0 for default)
jpeg_progressive -- (iff JPEG; 0 for baseline seq., 1 for progressive)
�wbzl_uint8 **pdataz
size_t *psizer N)r ZgetImpliedFileFormatr! �fsencoder{ r rF r Z
L_JPEG_ENCODEZpixWriteMemJpegrE Z pixWriteMemrW �write) r r| � jpeg_qualityZjpeg_progressiveZ lept_formatr� rX rY rW r r r �write_implied_formatM s
� zPix.write_implied_formatc
C s^ t � }|j|ddd� |�� }t�|�}t� �& tt�|t |���}|W 5 Q R � S Q R X dS )z*Create a copy of a PIL.Image from this PixZpngr )rG Zcompress_levelN)
r Zsave� getbufferr r~ r rK r r ra )r Z pillow_imageZbioZ py_bufferZc_bufferrc r r r �frompila s
z Pix.frompilc C s� ddl m} | }tjdkrr| jdkr*d}qx| jdkr:d}qx| jdkrZd }tt�|j��}qx| j}tt�|j��}n| j}|jj |jj
f}|jj d
|jj
}t �
|jj|�}|jj d
}|�| j||d ||�}|S ) z'Returns a PIL.Image version of this Pixr )�Image�littlert ZXBGRZRGBAZABGRrr z1;I� �raw)ZPILr� r � byteorderrw rK r ZpixEndianByteSwapNewrE rL rM Zwplr rW rX Z frombytes) r r� rc Zraw_moderY Z bytecount�bufZstrideZimr r r �topill s&
z Pix.topilc C s | � � �� S r< )r� �showr r r r r� � s zPix.showc
C s0 t � � tt�| j|��W 5 Q R � S Q R X dS )a Returns the deskewed pix object.
A clone of the original is returned when the algorithm cannot find a
skew angle with sufficient confidence.
reduction_factor -- amount to downsample (0 for default) when searching
for skew angle
N)r rK r Z pixDeskewrE )r �reduction_factorr r r �deskew� s z
Pix.deskewc
C s: t � �* tt�| j|d |d ��W 5 Q R � S Q R X dS )zCReturns the pix object rescaled according to the proportions given.r r N)r rK r ZpixScalerE )r Zscale_xyr r r �scale� s z Pix.scalec
C s2 t � �" tt�tj| j��W 5 Q R � S Q R X d S r< )r rK r Z pixRotate180r �NULLrE r r r r � rotate180� s z
Pix.rotate180c
C s0 t � � tt�| j|��W 5 Q R � S Q R X dS )z@Orthographic rotation, quads: 0-3, number of clockwise rotationsN)r rK r Z
pixRotateOrthrE )r Zquadsr r r � rotate_orth� s zPix.rotate_orthc
C sp t � �` t�dd�}t�dd�}t�| j||�}|dkrT|d |d fW 5 Q R � S W 5 Q R � dS W 5 Q R X dS )z}Returns a tuple (deskew angle in degrees, confidence value).
Returns (None, None) if no angle is available.
�float *� r )NNN)r r rF r Z pixFindSkewrE )r ZangleZ
confidence�resultr r r � find_skew� s z
Pix.find_skewc
C sD t � �4 t�| j�}|r,t|�W 5 Q R � S W 5 Q R � d S Q R X d S r< )r r ZpixConvertRGBToLuminancerE rK )r Zgray_pixr r r �convert_rgb_to_luminance� s
z Pix.convert_rgb_to_luminancec
C s4 t � �$ tt�| j|tj��W 5 Q R � S Q R X dS )z�Remove a palette (colormap); if no colormap, returns a copy of this
image
removal_type - any of lept.REMOVE_CMAP_*
N)r rK r ZpixRemoveColormapGeneralrE ZL_COPY)r Z removal_typer r r �remove_colormap� s �zPix.remove_colormap��, r� �r� r� 皙�����?c C s� t � �| |\}}|\}}t�d�}tt�| jd��} t�| j|||||tj|�}
|
dkrpt|d �W 5 Q R � S W 5 Q R � d S W 5 Q R X d S )NzPIX **r ) r r rF rK r �
pixConvertTo8rE ZpixOtsuAdaptiveThresholdr� ) r � tile_size� kernel_size�
scorefract�sx�sy�smoothx�smoothyZp_pixrc r� r r r �otsu_adaptive_threshold� s$
�zPix.otsu_adaptive_thresholdN��
� �d �2 � �r r c C s� t � �p |\}} |\}
} tj}t|t�r.|j}tt�| jd��} t�| j||| ||||
| |tj� }
t|
�W 5 Q R � S Q R X d S �Nr ) r r r� re rK rE r r� Z pixOtsuThreshOnBackgroundNorm)r �maskr� �thresh�mincountZbgvalr� r� r� r� r� r� rc �
thresh_pixr r r �!otsu_threshold_on_background_norm� s*
�
z%Pix.otsu_threshold_on_background_normc
C s~ t � �n |\}}|\} }
tj}t|t�r.|j}tt�| jd��} t�| j|||||| |
|tj�
} t| �W 5 Q R � S Q R X d S r� ) r r r� re rK rE r r� Z pixMaskedThreshOnBackgroundNorm)
r r� r� r� r� r� r� r� r� r� r� rc r� r r r �#masked_threshold_on_background_norm� s(
� z'Pix.masked_threshold_on_background_norm� �F � c
C sh t � dk rtd��t� �F tt�| j||||tj��}t� | j|jtj�} t
| �W 5 Q R � S Q R X d S )Nzleptonica-1.76z*Not available in this version of Leptonica) �get_leptonica_versionr1 r �Boxr ZpixFindPageForegroundrE r r� ZpixClipRectanglerK )
r Z thresholdZmindistZ erasedistZpagenumZ showmorphZdisplayZpdfdirZcropboxZ cropped_pixr r r �crop_to_foreground s
��zPix.crop_to_foreground� �?c
C sD t � �4 tt�| j|ptj|p"tj|||��W 5 Q R � S Q R X d S r< )r rK r ZpixCleanBackgroundToWhiterE r r� )r r� � grayscale�gammaZblackZwhiter r r �clean_background_to_white2 s ��z Pix.clean_background_to_whitec
C s8 t � �( tt�tj| j|||��W 5 Q R � S Q R X d S r< )r rK r Z pixGammaTRCr r� rE )r r� ZminvalZmaxvalr r r � gamma_trcA s z
Pix.gamma_trc�< �( �� �r r c C sh | � tj�}t� �L tt�|j|p&tj|p.tj|d |d ||||d |d �
�W 5 Q R � S Q R X d S )Nr r ) r� r ZREMOVE_CMAP_BASED_ON_SRCr rK ZpixBackgroundNormrE r r� ) r r� r� r� Z fg_thresholdZ min_countZbg_valZ
smooth_kernelZ
target_pixr r r �background_normE s ��zPix.background_normr ��maxsizec C s t �� S r< )r ZmakePixelSumTab8r r r r �make_pixel_sum_tab8a s zPix.make_pixel_sum_tab8c C sH t � dk rtd��t�dd�}t�| j|j|�}|dkr@td��|d S )Nzleptonica-1.72z Leptonica version is too oldr� r� r zCorrelation failed)r� r1 r rF r ZpixCorrelationBinaryrE )Zpix1Zpix2Z correlationr� r r r �correlation_binaryf s
zPix.correlation_binaryc C s: t �d�}t�| j||d|�}|dkr.td��t|d �S )z%Convert to PDF data, with transcoding�L_COMP_DATA **r zGenerate PDF data failed)r rF r ZpixGenerateCIDatarE r1 �CompressedData)r �type_Zquality�
p_compdatar� r r r �generate_pdf_ci_datau s
zPix.generate_pdf_ci_datac C s t t�tj| j��S r< )rK r Z pixInvertr r� rE r r r r �invert} s z
Pix.invertc c s6 z�zpt� �` tt�| jd��}t t�
|jd��}|sHW 5 Q R � W W �Td S t t� |jtj
tjtjd��}W 5 Q R X W n tttfk
r� Y W �
d S X W 5 d}|D ]"}t t�� t�|� W 5 Q R X q�X t|�D ]b\}}|�� }|�� dkr�q�|�|�} | j| j }
} | j| j | j| j } }
||
| | |
ffV q�d S )N)z junkpixt.pngZjunkpixtr r@ ) r r. r! �unlinkr rK r r� rE �PixArrayZpixExtractBarcodes� StringArrayZpixReadBarcodesZL_BF_ANYZ L_USE_WIDTHSr r� r1 rA r0 � enumerater, �strip�get_box�x�yrL rM )r Zleptonica_junkZjunkrc Zpixa_candidatesZsarray�nrR Zdecoded�box�left�top�rightZbottomr r r �locate_barcodes� s: ��
zPix.locate_barcodesc C s� |dkr.d}t �|d�}t �ddddtj�}n6|dkr\d}t �|d�}t �ddddtj�}nt|��tt�tj | j
|j
��}tt� tj |j
|j
��}tt� tj | j
|j
��}|S )Nr ze
oooo
oC o
o o
oooo
�speckle2r � z
ooooo
oC o
o o
o o
ooooo
�speckle3)
�Sel� from_selstr� create_brickr ZSEL_HITrA rK ZpixHMTr r� rE Z pixDilateZ pixSubtract) r rY r� Zsel1Zsel2r� ZpixhmtZ
pixdilatedZpixsubr r r � despeckle� s z
Pix.despeckle)r r )r )r� r� r� )Nr� r� r� r� r� r� )Nr� r� r� r� r� )NNr� r r� )r� r r� )NNr� r� r� r� r� )3r6 r7 r8 r9 rB r Z
pixDestroyrH rS r\ r` rd rj �propertyrl rn ro rY rq rw rJ r+ r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r� � staticmethodr r� r� r� r� r� r� r r r r rK � s�
�
�
# �
!�
�
�
!rK c @ sH e Zd ZdZdZejZeddd��Z dd� Z
dd � Z d
d � Z d d
� Z
dS )r� z8Wrapper for L_COMP_DATA - abstract compressed image dataZ L_COMP_DATA�K c C sF t |�}t�d�}t�t�|�tj||�}|dkr:td��t |d �S )z)Open compressed data, without transcodingr� r �CompressedData.open)
r r rF r Zl_generateCIDataForPdfr! r� r� r1 r� )rI r| r� �filenamer� r� r r r r{ � s
�r� c C s | j jS r< )rE �
nbytescompr r r r �__len__� s zCompressedData.__len__c C s t �| jj| jj�}t|�S r< )r rW rE Zdatacompr� �bytes)r r� r r r r+ � s zCompressedData.readc C s$ t | j|�rt| j|�S t|��d S r< )�hasattrrE �getattrr )r �namer r r � __getattr__� s zCompressedData.__getattr__c C s8 t d�t d�| jj t d� }t�| jj|�}t|�S )z,Returns palette pre-formatted for use in PDFz< z rrggbb�>)ra rE Zncolorsr rW Z cmapdatahexr� )r Zbuflenr� r r r �get_palette_pdf_string� s z%CompressedData.get_palette_pdf_stringN)r� )r6 r7 r8 r9 rB r Zl_CIDataDestroyrH rJ r{ r� r+ r� r� r r r r r� � s r� c @ s2 e Zd ZdZdZejZdd� Zdd� Z dd� Z
d S )
r� z"Wrapper around PIXA (array of PIX)ZPIXAc C s | j d jS r� �rE r� r r r r r� � s zPixArray.__len__c
C s4 t � �$ tt�| j|tj��W 5 Q R � S Q R X d S r< )r rK r Z
pixaGetPixrE �L_CLONE�r r� r r r � __getitem__� s zPixArray.__getitem__c
C s4 t � �$ tt�| j|tj��W 5 Q R � S Q R X d S r< )r r� r Z
pixaGetBoxrE r� r� r r r r� � s zPixArray.get_boxN) r6 r7 r8 r9 rB r Z pixaDestroyrH r� r� r� r r r r r� � s r� c @ sR e Zd ZdZdZejZdd� Ze dd� �Z
e dd� �Z e d d
� �Z e d d � �Z
d
S )r� zaWrapper around Leptonica's BOX objects (a pixel rectangle)
Uses x, y, w, h coordinates.
ZBOXc C s" | j r d�| j| j| j| j�S dS )Nz'<leptonica.Box x={0} y={1} w={2} h={3}>z<leptonica.Box NULL>)rE rG r� r� rL rM r r r r rS s �z Box.__repr__c C s | j jS r< )rE r� r r r r r� s zBox.xc C s | j jS r< )rE r� r r r r r� s zBox.yc C s | j jS r< rk r r r r rL s zBox.wc C s | j jS r< rm r r r r rM s zBox.hN)r6 r7 r8 r9 rB r Z
boxDestroyrH rS r� r� r� rL rM r r r r r� � s
r� c @ s2 e Zd ZdZdZejZdd� Zdd� Z dd� Z
d S )
�BoxArrayz7Wrapper around Leptonica's BOXA (Array of BOX) objects.ZBOXAc C s* | j s
dS dd� | D �}dd�|� d S )Nz
<BoxArray>c s s | ]}t |�V qd S r< )�repr)�.0r� r r r � <genexpr>! s z$BoxArray.__repr__.<locals>.<genexpr>z <BoxArray [z, z]>)rE �join)r Zboxesr r r rS s zBoxArray.__repr__c C s | j jS r< r� r r r r r� $ s zBoxArray.__len__c C sP t |t�std��d| kr*t| �k rDn ntt�| j|tj��S t |��d S )Nz list indices must be integersr )
re rO rg ra r� r Z
boxaGetBoxrE r� r0 r� r r r r� ' s
zBoxArray.__getitem__N) r6 r7 r8 r9 rB r Z boxaDestroyrH rS r� r� r r r r r� s r� c @ s* e Zd ZdZdZejZdd� Zdd� Z dS )r� z Leptonica SARRAY/string arrayZSARRAYc C s | j jS r< r� r r r r r� 5 s zStringArray.__len__c C s: d| krt | �k r.n nt�| jj| �S t|��d S r� )ra r �stringrE Zarrayr0 r� r r r r� 8 s zStringArray.__getitem__N)
r6 r7 r8 r9 rB r Z
sarrayDestroyrH r� r� r r r r r� / s
r� c @ s: e Zd ZdZdZejZedd� �Z edd� �Z
dd� Z d S )
r� z8Leptonica 'sel'/selection element for hit-miss transformZSELc C s� dd� |� d�D �}t|�}t|d �}tdd� |D ��}t|�dkrNtd��d �d
d� |D ��}t�|�d ��}t�|�d ��} t� |||| �}
| |
�S ) Nc S s g | ]}|� � r|� � �qS r �r� �r� �liner r r �
<listcomp>G s z#Sel.from_selstr.<locals>.<listcomp>�
r c s s | ]}t |�V qd S r< )ra r� r r r r� J s z"Sel.from_selstr.<locals>.<genexpr>r z'All lines in selstr must be same lengthr@ c s s | ]}|� � V qd S r< r� r� r r r r� N s �ascii)
�splitra �setrA r� r r~ �encoder ZselCreateFromString) rI �selstrr� �linesrM rL ZlengthsZrepackedZ
buf_selstrZbuf_name�selr r r r� D s zSel.from_selstrc C s t �|||||�}| |�S r< )r ZselCreateBrick)rI rM rL ZcyZcxr� r r r r r� T s zSel.create_brickc C s. t �t�| j�tj�}dt �|��d� d S )Nz<Sel
r z
>)r rC r ZselPrintToStringrE r^ r� r, )r r r r r rS Y s z Sel.__repr__N) r6 r7 r8 r9 rB r Z
selDestroyrH rJ r� r� rS r r r r r� > s
r� r� c C s t �t�� ��� S )z�Get Leptonica version string.
Caveat: Leptonica expects the caller to free this memory. We don't,
since that would involve binding to libc to access libc.free(),
a pointless effort to reclaim 100 bytes of memory.
)r r� r ZgetLeptonicaVersionr, r r r r r� ^ s r� c C s� zt �| �}W n tk
r. td| ��Y nX |dk r>d}nd}|�|�}z|�|� W n tk
rz td| ��Y nX d S )N�Failed to open file: %s� r r �#Failed to open destination file: %s)rK r{ r/ r� r� )�infile�outfilerp Z
pix_sourcer� Z pix_deskewedr r r r� i s
r� �r� r� r� r� � c C sz zt �| �}W n tk
r. td| ��Y nX |j|d��|||�}z|�|� W n tk
rt td| ��Y nX d S )Nr
)r� r )rK r{ r/ r� r� r� )r
r r� r� Zblack_thresholdZwhite_thresholdrc r r r �remove_background{ s �r �__main__z"Python wrapper to access Leptonica)� descriptionZcommandszsupported operations)�titler z-rz--dpirp Zstorer� zinput resolution)�dest�action�type�default�helpr
zimage to deskew)r r zdeskewed output image)�func)r r� r� r )@�argparseZloggingr! r rx Zcollections.abcr �
contextlibr Z ctypes.utilr � functoolsr r r r Ztempfiler �
exceptionsr
�execr Zlib._leptonicar Z getLoggerr6 r- r� Zlibname�environZ_libpathZdlopenr ZsetMsgSeverityZL_SEVERITY_WARNING�error�er � Exceptionr1 r/ r; rK r� r� r� r� r� r� r� r� r �ArgumentParser�parserZadd_subparsersZ
subparsersZ
add_parserZ
parser_deskew� add_argumentrO Z set_defaults�
parse_args�argsr r r r r �<module> s�
�
��J$ ~&!
�
�
�