{-j - nddlZddlZddlmZddlZddlmZmZddl m Z m Z m Z ddl mZmZmZddlmZmZe d rddlZe d Gd d eZe d d ed ejdededejf dZe d ddZdejdejfdZdS)N)Dict)Image ImageDraw)class_requires_depsfunction_requires_depsis_dep_available) SIMFANG_FONT create_fontcreate_font_vertical) BaseCVResult JsonMixinzopencv-contrib-pythoncpeZdZdZdeeejffdZdeeeffdZdeeeffdZ dS) OCRResultz OCR resultreturnc & d|vr?g}g}d|dD}d|dD}t|D]\}}|}ttj|dd|ddz dz|dd|ddz dzz}ttj|dd|ddz dz|dd|ddz dzz} |dks| dkr|||||n|d }|d }|d d } | jdd\} } t j| t j} tj | }tj | | dftj dz}tjdt!j|}t$}|dr|dd}tt'||D]'\}\}} tjddtjddtjddf}tj|}t-|dkrd|D}|||d|t3|}tdt5|dddft7|dddfz z}tj|dddf|dddf<tj|dddft7d|z|dddf<n5d|D}|||t;|t<r|d}t?| | f|||j }tj|tj!"d}t j#||gd|dt j$||}!#Y&xYwtj%tj | |d}tj&d| dz| fd}|'|dd| | f|'tj || d| dz| f|d}d|i}|dr|j(d i|d j)|S)!z Converts the internal data to a PIL Image with detection and recognition results. Returns: Dict[Image.Image]: A dictionary containing two images: 'doc_preprocessor_res' and 'ocr_res_img'. text_word_regioncg|] }|D]}| Sr.0sublistitems f/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/inference/pipelines/ocr/result.py z%OCRResult._to_img..-s9      AE    cg|] }|D]}| Srrrs rrz%OCRResult._to_img..0s%SSS'7SS4SSSSr text_wordrr  rec_polys rec_textsdoc_preprocessor_res output_imgdtype vis_fontsrcg|] \}}||f Srrrxys rrz%OCRResult._to_img..Ys ;;;daAq6;;;r)outlinewidthfillg?NcPg|]#\}}t|t|f$Sr)intr+s rrz%OCRResult._to_img..`s-IIIDAqAA/IIIr)r1)r!r TRGBr(r(r(model_settings ocr_res_imguse_doc_preprocessorr)* enumerater4mathsqrtappendshapecv2cvtColor COLOR_BGR2RGBr fromarraynponesuint8randomseedrDrawr ziprandintarraylentolistpolygonget_minarea_rectmaxminmean isinstancetupledraw_box_txt_finepathint32reshape polylines bitwise_andblendnewpasteupdateimg)selfboxestxtsrridx word_regionchar_box box_height box_widthimagehw image_rgbimg_left img_right draw_leftvis_fontboxtxtcolorptsheightbox_ptsimg_right_textimg_showr8 res_img_dicts r_to_imgzOCRResult._to_img"s&  % %ED  $();$<    TST+->SSSI$-.>$?$? , , [& I!!Q(1+a.8Q>#A;q>HQKN:q@A  I!!Q(1+a.8Q>#A;q>HQKN:q@A ??i1nn [))) IcN++++# ,&%E $D+,\:{1Q31L(9:: ?9--GQ1IRX666<  AN8,,    ,K(+H(UD)9)9::  OC#s N1c**N1c**N1c** hsmms88a<<;;cjjll;;;C%%c5%NNN*3//C C1ISAY(G!HIIF!#QQQT!3!3CAJ!#QQQT!3!3c"foo!ECAJJIICJJLLIIIG%%gE%:::c5))!a&C!2Aq63X]!S!ShsBH--55jAA nseT5!DDDOI~FF  ;uy998SII9UQUAJ@@x!Q1...uy11Aq!a%3CDDD./&1 0 1 D L  C C$'=">"B C C Cs :HQQci}|d|d<|d|d<|d|d<|ddr|djd|d<|ddkr|d ntj|d |d <|d |d <|d|d<d |vrtj|d |d <|d |d <|d |d <|d|d<tj|d|d<|ddkr|dntj|d|d<tj|d|d<d|vr|d|d<|d|d<tj|g|Ri|S)axConverts the instance's attributes to a dictionary and then to a string. Args: *args: Additional positional arguments passed to the base class method. **kwargs: Additional keyword arguments passed to the base class method. Returns: Dict[str, str]: A dictionary with the instance's attributes converted to strings. input_path page_indexr8r:r$res text_typesealdt_polystext_det_paramstextline_orientation_anglestext_rec_score_threshreturn_word_boxr# rec_scoresr" rec_boxestext_word_boxesr)strrDrLr_to_strraargskwargsdatas rrzOCRResult._to_strvs!,/\!,/\!%&6!7   !"8 9 S+/0F+G+KE+RD' (K F**   $z*++ Z #''8"9  -[ (D 0 024(2333D. /)--D(E $%"&'8"9  -[Xd<&899\K F**   ${+,, [ HT+%677[  $ $&*+<&=D" # $[ 1D  7777777rci}|d|d<|d|d<|d|d<|ddr|djd|d<|d|d<|d|d<|d |d <d |vr |d |d <|d |d <|d |d <|d |d <|d|d<|d|d<|d|d<d|vr|d|d<|d|d<tj|g|Ri|S)a[ Converts the object's data to a JSON dictionary. Args: *args: Positional arguments passed to the JsonMixin._to_json method. **kwargs: Keyword arguments passed to the JsonMixin._to_json method. Returns: Dict[str, str]: A dictionary containing the object's data in JSON format. r|r}r8r:r$r~rrrrrrr#rr"rrr)jsonr_to_jsonrs rrzOCRResult._to_jsons[!,/\!,/\!%&6!7   !"8 9 T+/0F+G+LU+SD' ( +Z"&'8"9  -[ (D 0 0267T2UD. /(,-D(E $%"&'8"9  -[!,/\ -[ -[  $ $&*+<&=D" # $[ 1D !$8888888rN) __name__ __module__ __qualname____doc__rrrrzrrrrrrrsRc5;./RRRRh)8$sCx.)8)8)8)8V94S>999999rrimg_sizerqrr font_pathrc ttjt|dd|ddz dzt|dd|ddz dzz}ttjt|dd|ddz dzt|dd|ddz dzz}|d|zkr]|dkrWt jd||fd}t j|}|r(t|||f|}t|d||d d n\t jd||fd}t j|}|r.t|||f|}| ddg|d | tj ddg|dg||gd|gg} tj|tj } tj| | } tj|tj }tj|| |tjtjd } | S)a Draws text in a box on an image with fine control over size and orientation. Args: img_size (tuple): The size of the output image (width, height). box (np.ndarray): A 4x2 numpy array defining the corners of the box in (x, y) order. txt (str): The text to draw inside the box. font_path (str): The path to the font file to use for drawing the text. Returns: np.ndarray: An image with the text drawn in the specified box. rr r r!r6r7)rrrrr)r1 line_spacing)r1fontr&)flags borderMode borderValue)r4r<r=floatrr]rrIr draw_vertical_textr textrDfloat32rLr@getPerspectiveTransformrFwarpPerspective INTER_NEARESTBORDER_CONSTANT) rrqrrrrgrhimg_text draw_textrpts1pts2Mrws rrVrVs`  %Aq CF1I-..!3eCF1IAq z"get_minarea_rect..s QqTr)key)rr!r r r!rr r ) r@ minAreaRectsortedlist boxPointsrDrLastyperX)r bounding_boxindex_aindex_bindex_cindex_drqs rrPrPs?6**L D|4455>> J J JF)3&GWgw ay|fQil"" ay|fQil"" ( &/6'?F7OL   fRX Jr)rr )r<rGtypingrnumpyrDPILrr utils.depsrrr utils.fontsr r r common.resultrrr@rrUndarrayrrVrrPrrrrs  XXXXXXXXXXKKKKKKKKKK44444444+,,JJJ,--a9a9a9a9a9 a9a9.-a9J/0044*4+.4;>4Z444104n/00(((10( RZ BJ      r