o e`-@spdZddlZddlZddlmZddlmZddl m Z ej Z e e ZGdddejZGdddejZdS)) PdfTextPagePdfTextSearcherN) PdfiumErrorcseZdZdZfddZeddZdddZdd d ZdddZ ddZ d ddZ ddZ d!ddZ ddZd"ddZZS)#rz Text page helper class. Attributes: raw (FPDF_TEXTPAGE): The underlying PDFium textpage handle. page (PdfPage): Reference to the page this textpage belongs to. c||_||_ttjdSN)rawpagesuper__init__pdfium_cFPDFText_ClosePage)selfrr __class__N/var/www/html/venv/lib/python3.10/site-packages/pypdfium2/_helpers/textpage.pyr zPdfTextPage.__init__cC|jSr)rr rrrparentzPdfTextPage.parentrcCsp||krdSt||}|dkr||d||d|St||}|dkr2|||d||dS||||fS)Nr)r "FPDFText_GetTextIndexFromCharIndex_get_active_text_range)r c_startc_end l_passive r_passivet_startt_endrrrr$s   z"PdfTextPage._get_active_text_rangerignorec Cs|dkr ||}||||d}|dkrdS|\}}}}||7}|||8}||d} t| d} t| ttj} t|||| } | | ksWJd| d| | j d| ddj d |d S) aj Extract text from a given range. See `this benchmark `_ for a performance and quality comparison with other tools. Parameters: index (int): Index of the first char to include. count (int): Number of chars to cover, relative to the internal char list. Defaults to -1 for all remaining chars after *index*. errors (str): Error handling when decoding the data (see :meth:`bytes.decode`). Returns: str: The text in the range in question, or an empty string if no text was found. Note: * The returned text's length does not have to match *count*, even if it will for most PDFs. This is because the underlying API may exclude/insert chars compared to the internal list, although rare in practice. This means, if the char at ``i`` is excluded, ``get_text_range(i, 2)[1]`` will raise an index error. Pdfium provides raw APIs ``FPDFText_GetTextIndexFromCharIndex()`` / ``FPDFText_GetCharIndexFromTextIndex()`` to translate between the two views and identify excluded/inserted chars. * In case of leading/trailing excluded characters, pypdfium2 modifies *index* and *count* accordingly to prevent pdfium from unexpectedly reading beyond ``range(index, index+count)``. rrrzBuffer size mismatch: z vs N utf-16-leerrors) count_charsrctypescreate_string_buffercastPOINTERc_ushortr FPDFText_GetTextrdecode) r indexcountr& active_rangerr rrin_sizebuffer buffer_ptrout_sizerrrget_text_range4s     zPdfTextPage.get_text_rangeNc Cs|j}|dur |d}|dur|d}|dur|d}|dur%|d}|||||f}tjg|ddR}|dkr>dSd|} t| } t| ttj} tjg|| |R| j j d|dS) a Extract text from given boundaries in PDF coordinates. If a boundary value is None, it defaults to the corresponding value of :meth:`.PdfPage.get_bbox`. Parameters: errors (str): Error treatment when decoding the data (see :meth:`bytes.decode`). Returns: str: The text on the page area in question, or an empty string if no text was found. Nrrr#r"r$r%) rget_bboxr FPDFText_GetBoundedTextr(r)r*r+r,rr.) r leftbottomrighttopr&bboxargsn_charsn_bytesr3r4rrrget_text_bounded^s$  zPdfTextPage.get_text_boundedcCst|}|dkr td|S)zV Returns: int: The number of characters on the text page. rzFailed to get character count.)r FPDFText_CountCharsr)r r@rrrr's zPdfTextPage.count_charscCs"t|||}|dkrtd|S)a Parameters: index (int): Start character index. count (int): Character count to consider (defaults to -1 for all remaining). Returns: int: The number of text rectangles in the given character range. rzFailed to count rectangles.)r FPDFText_CountRectsr)r r/r0n_rectsrrr count_rectsszPdfTextPage.count_rectscCs"t|||||}|dkrdS|S)a Get the index of a character by position. Parameters: x (float): Horizontal position (in PDF canvas units). y (float): Vertical position. x_tol (float): Horizontal tolerance. y_tol (float): Vertical tolerance. Returns: int | None: The index of the character at or nearby the point (x, y). May be None if there is no character or an error occurred. rN)r FPDFText_GetCharIndexAtPos)r xyx_toly_tolr/rrr get_indexs zPdfTextPage.get_indexFc Cs|rt}t|||}|j|j|j|jf\}}}}n&ttttf\}}}}t||||||}|j |j |j |j f\}}}}|sHt d||||fS)a Get the bounding box of a single character. Parameters: index (int): Index of the character to work with, in the page's character array. loose (bool): Get a more comprehensive box covering the entire font bounds, as opposed to the default tight box specific to the one character. Returns: Float values for left, bottom, right and top in PDF canvas units. zFailed to get charbox.) r FS_RECTFFPDFText_GetLooseCharBoxr:r;r<r=c_doubleFPDFText_GetCharBoxvaluer) r r/looserectoklbrtrrr get_charboxs  zPdfTextPage.get_charboxcCsPttttf\}}}}t||||||}|std|j|j|j|jfS)al Get the bounding box of a text rectangle at the given index. Note that :meth:`.count_rects` must be called once with default parameters before subsequent :meth:`.get_rect` calls for this function to work (due to PDFium's API). Returns: Float values for left, bottom, right and top in PDF canvas units. zzFailed to get rectangle. (Make sure count_rects() was called with default params once before subsequent get_rect() calls.))rOr FPDFText_GetRectrrQ)r r/rUrVrWrXrTrrrget_rects  zPdfTextPage.get_rectc Cst|dkr tdd}|r|tjO}|r|tjO}|r!|tjO}|dd}t|t tj }t ||||} t | |} | | | S)au Locate text on the page. Parameters: text (str): The string to search for. index (int): Character index at which to start searching. match_case (bool): If True, the search will be case-specific (upper and lower letters treated as different characters). match_whole_word (bool): If True, substring occurrences will be ignored (e. g. `cat` would not match `category`). consecutive (bool): If False (the default), :meth:`.search` will skip past the current match to look for the next match. If True, parts of the previous match may be caught again (e. g. searching for `aa` in `aaaa` would match 3 rather than 2 times). Returns: PdfTextSearcher: A helper object to search text. rz#Text length must be greater than 0.r$)len ValueErrorr FPDF_MATCHCASEFPDF_MATCHWHOLEWORDFPDF_CONSECUTIVEencoder(r*r+r,FPDFText_FindStartr_add_kid) r textr/ match_casematch_whole_word consecutiveflagsenc_text enc_text_ptr raw_searchersearcherrrrsearchs      zPdfTextPage.search)rr)rrr!)NNNNr!)rr)F)rFFF)__name__ __module__ __qualname____doc__r propertyrrr6rBr'rFrLrYr[rn __classcell__rrrrrs     *!  rcsDeZdZdZfddZeddZddZdd Zd d Z Z S) rz Text searcher helper class. Attributes: raw (FPDF_SCHHANDLE): The underlying PDFium searcher handle. textpage (PdfTextPage): Reference to the textpage this searcher belongs to. crr)rtextpager r r FPDFText_FindClose)r rrurrrr rzPdfTextSearcher.__init__cCrr)rurrrrr rzPdfTextSearcher.parentcCs,||}|sdSt|}t|}||fSr)r FPDFText_GetSchResultIndexFPDFText_GetSchCount)r find_funcrTr/r0rrr_get_occurrences   zPdfTextSearcher._get_occurrencecC |tjS)z Returns: (int, int): Start character index and count of the next occurrence, or None if the last occurrence was passed. )rzr FPDFText_FindNextrrrrget_next zPdfTextSearcher.get_nextcCr{)z Returns: (int, int): Start character index and count of the previous occurrence (i. e. the one before the last valid occurrence), or None if the last occurrence was passed. )rzr FPDFText_FindPrevrrrrget_prev!r~zPdfTextSearcher.get_prev) rorprqrrr rsrrzr}rrtrrrrrs  r)__all__r(logging pypdfium2.rawrr pypdfium2.internalinternalpdfium_ipypdfium2._helpers.miscrrO getLoggerrologger AutoCloseablerrrrrrs    n