\;j<ddlZddlmZmZdZdZGddZy)N) deal_eb_tokendeal_bbc&|\}}}}|\}}}} t||z t||z zt||z zt| |z z} t||z t||z z} t||z t| |z z} | t| | zSN)absmin) box_1box_2x1y1x2y2x3y3x4y4disdis_2dis_3s lG:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddleocr/ppstructure/table/matcher.pydistancersNBBNBB b2g,R"W %BG 4s27| CC RL3rBw< 'E RL3rBw< 'E UE" ""c:|d|dz |d|dz z}|d|dz |d|dz z}||z}t|d|d}t|d|d}t|d|d}t|d|d}||k\s||k\ry||z ||z z} | || z z dzS)z computing IoU :param rec1: (y0, x0, y1, x1), which reflects (top, left, bottom, right) :param rec2: (y0, x0, y1, x1) :return: scala value of IoU rg?)maxr) rec1rec2S_rec1S_rec2sum_area left_line right_linetop_line bottom_line intersects r compute_iour)s1gQDGd1g$5 6F1gQDGd1g$5 6FHDGT!W%IT!Wd1g&J47DG$Hd1gtAw'KJ(k"9)+ h0FG X 12c99rc2eZdZddZdZdZdZdZdZy) TableMatchc ||_||_yr)filter_ocr_result use_master)selfr-r.s r__init__zTableMatch.__init__:s!2$rc|\}}|jr|j|||\}}|j||}|jr|j |||\}}|S|j |||\}}|Sr)r-_filter_ocr_result match_resultr.get_pred_html_master get_pred_html) r/ structure_resdt_boxesrec_respred_structures pred_bboxes matched_index pred_htmlpreds r__call__zTableMatch.__call__>s'4$  ! ! $ 7 7 Xw W Hg))(K@ ??"77OIt#00OItrc i}t|D]?\}}g}t|D]\}}t|dk(rjtj|dddtj|dddtj|dddtj|dddg}|j t ||dt||z f|j} t| d} |j| d|jvr|g||j| d<||j| dj |B|S)Nrrrrc|d|dfS)Nrr)items rz)TableMatch.match_result.._sDGT!W3Er)key) enumeratelennprrappendrr)copysortedindexkeys) r/r7r:matchedigt_box distancesjpred_boxsorted_distancess rr3zTableMatch.match_resultMsK"8,IAvI(5 8x=A%x1~.x1~.x1~.x1~.  H   fh/{687T1TU 6 )~~/ % &E  /237<<>IAB (8(;<= (8(;<=DDQG+-,rcg}d}|D]T}d|vr;d|k(r|jd||jvrd}d|||dvr$t||dkDrd}|jdt||D]\}} || d} t||dkDr^t| dk(r.| dd k(r| dd} d| vr| d d} d | vr| dd } t| dk(r\|t||dz k7r d | d k7r| d z } |j| |r|jd d|k(r|j dn|j ||dz }D|j |Wdj ||fS)Nr zFrT r)extendrMrGrFrIjoin) r/r9r; ocr_contentsend_htmltd_indextagb_withrOtd_index_indexcontents rr5zTableMatch.get_pred_htmlgs"C#~#%OOF+}1133"FmH.Ea.H!II h 781".~">q"A}X67!;"7|q0 (&qzS0*1!"+$/*1!"+%0*1#2,"7|q0 ( !Sx)@%AA%E E$'72;$6 '3 0%.P& /#%OOG,OOC(A $O#Pwwx (**rcg}d}|D],}d|vrd}d}||jvrd|||dvrt||dkDrd}t||D]\} } || d} t||dkDr^t| dk(r.| ddk(r| dd} d| vr| d d} d | vr| dd } t| dk(r\| t||dz k7r d| d k7r| dz } || z }|rd j|}d|k(rdj|}ndj|}|dz }t |}|j |/dj |} t| } | |fS)NrrVr]FrXrTrYrrZr[r\z {}rWz {}z{})rMrGrFformatrrIr_r) r/r9r;r`rarbtokentxtrdrOrerfhtmls rr4zTableMatch.get_pred_html_masters$E%}1133mH.Ea.H!II h 781".~">q"A}X67!;"7|q0 (&qzS0*1!"+$/*1!"+%0*1#2,"7|q0 ( !Sx)@%AA%E E$'72;$6 '3w%.P&%,,S1C%')005E%,,S1EA !%(E OOE "K%Lwwx t}X~rc|dddddfj}g}g}t||D]F\}}tj|ddd|kr%|j ||j |H||fS)Nrr)rziprHrrI) r/r:r7r8r new_dt_boxes new_rec_resboxrecs rr2zTableMatch._filter_ocr_results ADqD ! % % '  Hg.HCvvc!$Q$i 2%    $   s # / [((rN)FF) __name__ __module__ __qualname__r0r>r3r5r4r2rBrrr+r+9s"% 4++Z+Z )rr+)numpyrH$ppstructure.table.table_master_matchrrrr)r+rBrrrws#G#::R)R)r