NgOddlmZmZmZmZmZmZmZddlZddlZddl Z ddl m Z ddl Z ddlmZmZmZmZddlZddlmZmZmZmZmZddlmZeje jejj Z!ej"e!dd Z#d Z$d Z%d Z&e#Z'd Z(ej)e'e(Z*dZ+dZ,ddgZ- d@dZ.dZ/dAdZ0dZ1dZ2dede3dee4e4e4e4ffdZ5dZ6dZ7dZ8e8 dBd d!d"ed#ede3dd!f d$Z9e8 dCd ejd&ed'eeee4e4fd(eeee3e3fd)eeee:e:fd#eed*e;d+e;d,ee4d-ee:d.ee:d/ee:d0ee3fd1Ze;fd?Z=dS)E)ListOptionalUnionDictAnyTuplerN)cycle)Image ImageFont ImageDraw ImageColor)LayoutInterval Rectangle TextBlock Quadrilateral)cvt_coordinates_to_pointsmisczNotoSerifCJKjp-Regular.otfg{Gzt?redz'#f6bd60-#f7ede2-#f5cac3-#84a59d-#f28482blackwhitedraw_box draw_textc tdd|zfd|D}tfd|D}t||t |zz}t jd||f|} t jd||f|} tj| } tj| } t|D];\} }| dt|d| | |zzf||<| | S) zHelper function to draw text vertically. Ref: https://github.com/Belval/TextRecognitionDataGenerator/blob/7f4c782c33993d2b6f712d01e86a2f342025f2df/trdg/computer_text_generator.py  rcVg|]%}|dkr|dn&S)rrgetsize).0c image_font space_heights V/var/www/html/ai-engine/env/lib/python3.11/site-packages/layoutparser/visualization.py z'_draw_vertical_text..BsFCDAHH 1a  ,cFg|]}|dS)rr )r"r#r$s r&r'z'_draw_vertical_text..Es,===1j((++A.===r(RGBAcolorrfillfont) intr!maxsumlenr newr Draw enumeratetextcropgetbbox)r7r$ text_colortext_background_colorcharacter_spacing space_width char_heights text_width text_heighttxt_imgtxt_mask txt_img_draw txt_mask_drawir#r%s ` @r&_draw_vertical_textrF4svz))#..q1K?@@LHLL=======>>Jl##&7#d))&CCKi[ 9AVWWWGy*k!:BWXXXH>'**LN8,,M$  1 L1%&&->)>> ?       <<)) * **r(cptdtt|jtzS)Nr)r1r0minsizeDEFAULT_BOX_WIDTH_RATIO)canvass r&_calculate_default_box_widthrLYs* q#c&+&&)@@AA B BBr(c^| |tStj|pt|ptS)N)DEFAULT_FONT_OBJECTr truetypeDEFAULT_FONT_PATHDEFAULT_FONT_SIZE) font_size font_paths r&_create_font_objectrT]s9Y.""!  **I,J9J   r(cn|dkrKtjd|jdz|jf|pt}|||jdfnc|dkrKtjd|j|jdzf|pt}||d|jfnt d||S)Nlrr*rr+rudzInvalid direction )r r4widthheightDEFAULT_TEXT_BACKGROUNDpaste ValueError)rK arrangementr; new_canvass r&_create_new_canvasr_gsdY  \A v} -'B+B   &,!23333   Y  \6=1, -'B+B   !V]!34444;k;;<<< r(c dt|ttdDS)Nci|]\}}|| Srb)r"typer,s r& z)_create_color_palette..s.    D% e   r(-)zipr DEAFULT_COLOR_PALETTEsplit)typess r&_create_color_paletterjsD  ue,A,G,G,L,L&M&MNN   r( color_stringalphareturnc4|ddkrXt|dkrE|dgtfddDtd|zRS t j|}|td|zfzS#|cYSxYw)Nr#c3NK|]}t||dzdV dS)rN)r0)r"rE color_hexs r& z!_get_color_rgb..s:DDQ3yQU+R00DDDDDDr()rr)r3lstriptupler0r getrgb)rkrlrgbrss @r&_get_color_rgbr{sA##l"3"3q"8"8 '',,  DDDD)DDD D D e      #L11C#cEk**,, ,    s ')BBct|dst|jf}n|j}|}||ddz}||||dS)Npointsr)rXr.)hasattrr coordinatesr}raveltolistline)drawblockr,rXr}verticesdrawing_verticess r&_draw_box_outline_on_handlerrs 5( # #+E,=>>@||~~$$&&H(2A2,.II r(ct|dr$d|jD}nt|j}||t ||dS)Nr}c,g|]}t|Srb)rx)r"rs r&r'z4_draw_transparent_box_on_handler..sDDDUE%LLDDDr()r~r}rrrpolygonr{)rrr,rlrs r& _draw_transparent_box_on_handlerrsxuh@DDel.A.A.C.CDDD,U->??LLue$$r(cFtjfd}|S)Nct|tjr5|jdkr|d}|}n.t|t jrtj|}||g|Ri|}|S)NRGB) isinstancer modeconvertcopynpndarray fromarray)rKlayoutargskwargsoutfuncs r&wrapzimage_loader..wraps fek * * -{e##..[[]]FF  + + -_V,,Fd663D333F33 r() functoolswraps)rrs` r& image_loaderrs8_T      Kr(?rKr blocks color_mapc|(td|D}t|}|}tj|d}|D]}t ||||j| |S)zyGiven the image, draw a series of transparent boxes based on the blocks, coloring using the specified color_map. Nc<g|]}t|d|jSrcr~rcr"bs r&r'z(draw_transparent_box..s)FFFA71f3E3EFFFFr(r*)setrjrr r5rrc)rKrrrl all_typesrrs r&draw_transparent_boxrsFFFFFGG ))44 [[]]F >&& ) )DTT(ui 6KUSSSS Mr(Fr box_width box_alpha box_colorshow_element_idshow_element_type id_font_size id_font_path id_text_colorid_text_background_colorid_text_background_alphac d| cxkrdksnJtd| dtj|d} | pt} | pt} |s|rt || }|dgt |z}nt|ttfr|gt |z}t |t |kr/tdt |d t |td |Dstd |d|#t|}|gt |z}n~t|ttfr|gt |z}t |t |kr/td t |d t ||9(td |D}t|fd|D}nwt|tr|gt |z}t |t |kr/tdt |d t |t |t |cxkr't |cxkrt |ksnJtt!||||D]#\}\}}}}t|t"r||}|dkrt'| |||t)| ||||s|rd}|r|jp|}|t|z }|r0|st|jn|dzt|jz}|jdd\}}||\}}t3||||z||z}t)| || | | ||f|| |%|S)aDraw the layout region on the input canvas(image). Args: canvas (:obj:`~np.ndarray` or :obj:`~PIL.Image.Image`): The canvas to draw the layout boxes. layout (:obj:`Layout` or :obj:`list`): The layout of the canvas to show. box_width (:obj:`int` or :obj:`List[int]`, optional): Set to change the width of the drawn layout box boundary. Defaults to None, when the boundary is automatically calculated as the the :const:`DEFAULT_BOX_WIDTH_RATIO` * the maximum of (height, width) of the canvas. If box_with is a list, it will assign different widths to the corresponding layout object, and should have the same length as the number of blocks in `layout`. box_alpha (:obj:`float` or :obj:`List[float]`, optional): A float or list of floats ranging from 0 to 1. Set to change the alpha of the drawn layout box. Defaults to 0 - the layout box will be fully transparent. If box_alpha is a list of floats, it will assign different alphas to the corresponding layout object, and should have the same length as the number of blocks in `layout`. box_color (:obj:`str` or :obj:`List[str]`, optional): A string or a list of strings for box colors, e.g., `['red', 'green', 'blue']` or `'red'`. If box_color is a list of strings, it will assign different colors to the corresponding layout object, and should have the same length as the number of blocks in `layout`. Defaults to None. When `box_color` is set, it will override the `color_map`. color_map (dict, optional): A map from `block.type` to the colors, e.g., `{1: 'red'}`. You can set it to `{}` to use only the :const:`DEFAULT_OUTLINE_COLOR` for the outlines. Defaults to None, when a color palette is is automatically created based on the input layout. show_element_id (bool, optional): Whether to display `block.id` on the top-left corner of the block. Defaults to False. show_element_type (bool, optional): Whether to display `block.type` on the top-left corner of the block. Defaults to False. id_font_size (int, optional): Set to change the font size used for drawing `block.id`. Defaults to None, when the size is set to :const:`DEFAULT_FONT_SIZE`. id_font_path (:obj:`str`, optional): Set to change the font used for drawing `block.id`. Defaults to None, when the :const:`DEFAULT_FONT_OBJECT` is used. id_text_color (:obj:`str`, optional): Set to change the text color used for drawing `block.id`. Defaults to None, when the color is set to :const:`DEFAULT_TEXT_COLOR`. id_text_background_color (:obj:`str`, optional): Set to change the text region background used for drawing `block.id`. Defaults to None, when the color is set to :const:`DEFAULT_TEXT_BACKGROUND`. id_text_background_alpha (:obj:`float`, optional): A float range from 0 to 1. Set to change the alpha of the drawn text. Defaults to 1 - the text box will be solid. Returns: :obj:`PIL.Image.Image`: A Image object containing the `layout` draw upon the input `canvas`. rrz#The id_text_background_alpha value  is not within range [0,1].r*)rNzThe number of alphas z& is not equal to the number of blocks c36K|]}d|cxkodkncVdS)rrNrb)r"as r&rtzdraw_box..Js62211;;;;Q;;;;222222r(zThe box_alpha value zThe number of widths c<g|]}t|d|jSrrrs r&r'zdraw_box..]s)JJJwq&7I7IJQVJJJr(cg|]>}t|tstn|jt?Srb)rrDEFAULT_OUTLINE_COLORgetrc)r"elers r&r'zdraw_box.._sU   c9-- @ ! !sx)>??   r(zThe number of colors z: rr-)r\r r5rZDEFAULT_TEXT_COLORrTr3rfloatr0allrLrrjstrr6rfr put_on_canvasrridrcrr!rr7)rKrrrrrrrrrrrrrfont_objridxrr,rlrXr7ele_idstart_xstart_ytext_wtext_htext_box_objects ` r&rrsh ( - - - -A - - - - -zc.Fccc00 - - - >&v . . .D7R;R!7%7MC+C&|\BBC#f++% i% . . 2" c&kk1I y>>S[[ ( (kIkk^abh^i^ikk 22 22222 MyMMM 088 K#f++- i% . . 2" c&kk1I y>>S[[ ( (kIkk^abh^i^ikk   JJVJJJKKI-i88I         i % % 2" c&kk1I y>>S[[ ( (kIkk^abh^i^ikk  v;;#i.. L L L LC NN L L L Lc)nn L L L L L L+4 FIy)44,,))' 'c5% c8 $ $ ,##F++C 199 (sE5 A A A(sE5AAA  / D $3F #  R,0Qs38}}}dTkCMM6Q"rr2 GW%--d33NFF''F"2Gf4DO -((     II'""     Mr(rVr]rRrSr:r;text_background_alpha vertical_textwith_box_on_texttext_box_widthtext_box_colortext_box_alpha with_layoutc |d}| d} d|cxkrdksnJtd|dd| cxkrdksnJtd| d| r| t|} | rt||fi|}t||}| pt} |pt }|pt }t||t||}tj |d}t|D]\}}| r;| | | }t||| | t||| | t|d r |jd kr^|jdd \}}|s!|||f|j|| t%|j||t||}| r|||| z|| zf||||f|S) a< Draw the (detected) text in the `layout` according to their coordinates next to the input `canvas` (image) for better comparison. Args: canvas (:obj:`~np.ndarray` or :obj:`~PIL.Image.Image`): The canvas to draw the layout boxes. layout (:obj:`Layout` or :obj:`list`): The layout of the canvas to show. arrangement (`{'lr', 'ud'}`, optional): The arrangement of the drawn text canvas and the original image canvas: * `lr` - left and right * `ud` - up and down Defaults to 'lr'. font_size (:obj:`str`, optional): Set to change the size of the font used for drawing `block.text`. Defaults to None, when the size is set to :const:`DEFAULT_FONT_SIZE`. font_path (:obj:`str`, optional): Set to change the font used for drawing `block.text`. Defaults to None, when the :const:`DEFAULT_FONT_OBJECT` is used. text_color ([type], optional): Set to change the text color used for drawing `block.text`. Defaults to None, when the color is set to :const:`DEFAULT_TEXT_COLOR`. text_background_color ([type], optional): Set to change the text region background used for drawing `block.text`. Defaults to None, when the color is set to :const:`DEFAULT_TEXT_BACKGROUND`. text_background_alpha (:obj:`float`, optional): A float range from 0 to 1. Set to change the alpha of the background of the canvas. Defaults to 1 - the text box will be solid. vertical_text (bool, optional): Whether the text in a block should be drawn vertically. Defaults to False. with_box_on_text (bool, optional): Whether to draw the layout box boundary of a text region on the text canvas. Defaults to False. text_box_width (:obj:`int`, optional): Set to change the width of the drawn layout box boundary. Defaults to None, when the boundary is automatically calculated as the the :const:`DEFAULT_BOX_WIDTH_RATIO` * the maximum of (height, width) of the canvas. text_box_alpha (:obj:`float`, optional): A float range from 0 to 1. Set to change the alpha of the drawn text box. Defaults to 0 - the text box will be fully transparent. text_box_color (:obj:`int`, optional): Set to change the color of the drawn layout box boundary. Defaults to None, when the color is set to :const:`DEFAULT_OUTLINE_COLOR`. with_layout (bool, optional): Whether to draw the layout boxes on the input (image) canvas. Defaults to False. When set to true, you can pass in the arguments in :obj:`draw_box` to change the style of the drawn layout boxes. Returns: :obj:`PIL.Image.Image`: A Image object containing the drawn text from `layout`. Nrrz The text_background_color value rzThe text_box_alpha value r*)rightbottomr7rr)r/r.)r\rLrrTrrrZr_r{r r5r6padrrr~r7rrFr[)rKrr]rRrSr:r;rrrrrrrrrrrr modified_boxrr text_segments r&rrsh$ ! % * * * * * * * * *J]+@]]]-- * * *  # # # #! # # # # #ZONOOO&& # # #B  !9&AAN4&&33F33"9i88H#<'&& ) )Df%%%?%?S  777OOL (lNN    -lNN   sF## sx2~~  _RaR0' ? II'"     /46KLL L  ?  7^#;W~=U"V \GW+=>>>> Mr()rr)NN)Nr) NNNNFFNNNNr) rVNNNNNFFNNNF)>typingrrrrrrrossys itertoolsr numpyrPILr r r r layoutparserelementsrrrrrelements.utilsrpathdirnamemodules __package____file__ _lib_pathjoin _font_pathrJrrgrPrQrOrNrrZ__all__rFrLrTr_rjrr0r{rrrrrboolrrrbr(r&rsA@@@@@@@@@@@@@@@@@ 777777777777655555 GOOCK (@AJ K K W\\)V-I J J A(i():