khk*FdZddlZddlZddlmZddlmZGddeZy)aObject with a bounding box, e.g. Block, Line, Span. Based on ``PyMuPDF``, the coordinates (e.g. bbox of ``page.get_text('rawdict')``) are generally provided relative to the un-rotated page; while this ``pdf2docx`` library works under real page coordinate system, i.e. with rotation considered. So, any instances created by this Class are always applied a rotation matrix automatically. Therefore, the bbox parameter used to create ``Element`` instance MUST be relative to un-rotated CS. If final coordinates are provided, should update it after creating an empty object:: Element().update_bbox(final_bbox) .. note:: An exception is ``page.get_drawings()``, the coordinates are converted to real page CS already. N)IText) constantsc *eZdZdZej dZedZedZ d de fdZ dZ d Z ed Zej d Zd Zd efdZdZdZd!dddefdZd"defdZd#dedefdZd#dedefdZdZdZd$dedededefdZy)%Elementz.Boundary box with attribute in fitz.Rect type.cN|r#t|tjr||_yyy)zlSet global rotation matrix. Args: Rotation_matrix (fitz.Matrix): target matrix N) isinstancefitzMatrixROTATION_MATRIX)clsrotation_matrixs S/var/www/teggl/fontify/venv/lib/python3.12/site-packages/pdf2docx/common/Element.pyset_rotation_matrixzElement.set_rotation_matrixs$ z/4;;G"1C  H?c\|j\}}}}}}tj||||ddS)zHPure rotation matrix used for calculating text direction after rotation.r)r r r )rabcdefs rpure_rotation_matrixzElement.pure_rotation_matrix*s4)) !Aa!{{1Qq1Q''rNrawctj|_||_d|xsivr;tj|dtj z}|j |yy)zJ Initialize Element and convert to the real (rotation considered) page CS.bboxN)r Rectr_parentrr update_bbox)selfrparentrects r__init__zElement.__init__1sTIIK   ciR 99S[)G,C,CCD   T " !rc,t|jS)z!Real object when bbox is defined.)boolrr!s r__bool__zElement.__bool__=sDIIrc`|jjdt|jdS)N()) __class____name__tuplerr's r__repr__zElement.__repr__Is)$.."9"9!:!E$))|jjt j z|jjk\S|jjt j z|jjk\S)a5Whether given element is contained in this instance, with margin considered. Args: e (Element): Target element threshold (float, optional): Intersection rate. Defaults to 1.0. The larger, the stricter. Returns: bool: [description] F)rget_arear<widthheightr MINOR_DIST)r!rr@S intersectionfactors rcontainszElement.containss FFOO  yy166) |,,.q0!4 ) E 99??dii.. .99??9#7#77166<<G Gyy 4 44 EErc6|j}t|dr |jntj|}||z}|jry|j |j |j }}}|r|t ||z nd} | |k\r||zSdS)aCIf the intersection with ``e`` exceeds the threshold, return the union of these two elements; else return None. Args: e (Element): Target element. threshold (float, optional): Intersection rate. Defaults to 0.95. Returns: fitz.Rect: Union bbox or None. rNgư>)rhasattrr ris_emptyrCmin) r!rr@bbox_1bbox_2ra1a2rrIs r get_main_bboxzElement.get_main_bboxs"1f-499Q< VO ::dOO%v'8!**,B!"3r":"(I"5v?4?rrItext_directionc|r t|sy|r|jrdnd}|j|dz|j|z }|j|dz|j|z }t|j|dz|j|dzt |j||j|z }d}||z|z |z|t ||zk\S)aCheck whether two Element instances have enough intersection in vertical direction, i.e. perpendicular to reading direction. Args: e (Element): Object to check with factor (float, optional): Threshold of overlap ratio, the larger it is, the higher probability the two bbox-es are aligned. text_direction (bool, optional): Consider text direction or not. True by default. Returns: bool: [description] Examples:: +--------------+ | | +--------------+ L1 +-------------------+ | | +-------------------+ L2 An enough intersection is defined based on the minimum width of two boxes:: L1+L2-L>factor*min(L1,L2) FrrrBMbP?r&is_vertical_textrmaxrN r!rrIrTidxL1L2Lepss rvertically_align_withzElement.vertically_align_withs8T 5"d&;&;a YYs1u diin , VVCE]166#; &  #a% !&&Q- 03tyy~qvvc{3S S"uQws{fSBZ///rc|r t|sy|r|jrdnd}|j|dz|j|z }|j|dz|j|z }t|j|dz|j|dzt |j||j|z }d}||z|z |z|t ||zk\S)aCheck whether two Element instances have enough intersection in horizontal direction, i.e. along the reading direction. Args: e (Element): Element to check with factor (float, optional): threshold of overlap ratio, the larger it is, the higher probability the two bbox-es are aligned. text_direction (bool, optional): consider text direction or not. True by default. Examples:: +--------------+ | | L1 +--------------------+ +--------------+ | | L2 +--------------------+ An enough intersection is defined based on the minimum width of two boxes:: L1+L2-L>factor*min(L1,L2) FrrrBrVrWrZs rhorizontally_align_withzElement.horizontally_align_withs*T 5"d&;&;a YYs1u diin , VVCE]166#; &  #a% !&&Q- 03tyy~qvvc{3S S"uQws{fSBZ///rcJ|r|j|jk7ry|jrdnd}|j||j|dzzdz }|j||j|dzzdz }||j|dzkxr||j|dzk}|S)aCheck whether in same row/line with specified Element instance. With text direction considered. Taking horizontal text as an example: * yes: the bottom edge of each box is lower than the centerline of the other one; * otherwise, not in same row. Args: e (Element): Target object. .. note:: The difference to method ``horizontally_align_with``: they may not in same line, though aligned horizontally. FrrrBg@)is_horizontal_textr)r!rr[c1c2ress r in_same_rowzElement.in_same_row s D++q/C/CC**aiintyyQ//3 6ffSkAFF3q5M)S 0!&&Q-8B #a%(8$8 rc>dtd|jDiS)zStore properties in raw dict.rc3 K|]}|ywr1).0r=s r z Element.store..-s4Qq4s )r.rr's rstorez Element.store+s4$))4466rstrokerDfilldashesc H|j|j||||ddy)z(Plot bbox in PDF page for debug purpose.F?)colorrprDrqoverlay fill_opacityN) draw_rectr)r!pagerorDrprqs rplotz Element.plot0s+ tyy# "$$$'  )r)NN)g?)gffffff?)rT))rrrrsNN) r- __module__ __qualname____doc__r r r classmethodrrdictr$r(r/propertyr"setterr5floatr:r r?rJrSr&r`rbrhrnr.strryrkrrrrs 8"dkk#&O22(( #4 # P )) ]]33  . . 4FFeF8@@2&0e&0&0R00$0D>7 ))U)U)X[)rr)r|r5r sharerrrrkrrrs&  a)ea)r