o e; @sddlmZddlZddlZdedefddZddZd d Z d e d e d e fddZ de de d e fddZ de de d e fddZ  d0dejdedededef dd Zd!ejd"ed#efd$d%Zdejd&ededefd'd(Zd1d*ejdejd+e d,e d-e f d.d/ZdS)2)dequeNbbox_1bbox_2c Cs|\}}}}|\}}}} ||||t||t||} | dkr$dS||| |t|| t||} | dkrzz*solve_rects_intersection..cs|ddkSr)rr*)X1rrr.{r/cs|ddkSNr'rrr*r,rrr.|r/cr(r1rr*)X0rrr.}r/)intlistfilter_stabsolve_rects_intersection) r"r#r$Z center_posleftrightZS11ZS12ZS22ZS21r)r-r2r0rr7Xs  ,      r7S1S2c Cs|r|sdS|jddd|jdddd\}}|t|kr|t|kr||\}}}||\}} }|d| dkr}|} | t|krx|| dd|dkrxtt|d t|| d d || d7} | t|krx|| dd|dksP|d7}n@|} | t|kr|| dd| dkrtt|| d d t|d || d7} | t|kr|| dd| dks|d7}|t|kr|t|ks&dSdSdSdS) uCheck interval intersection in y-direction. Procedure ``stab(A, B)``:: i := 1; j := 1 while i ≤ |A| and j ≤ |B| if ai.y0 < bj.y0 then k := j while k ≤ |B| and bk.y0 < ai.y1 reportPair(air, bks) k := k + 1 i := i + 1 else k := i while k ≤ |A| and ak.y0 < bj.y1 reportPair(bjs, akr) k := k + 1 j := j + 1 NcS |ddSNr'rr*rrrr. z_stab..)keycSr<r=rr*rrrr.r>rrr'r%r)sortr _report_pairr3) r:r;r$rjma_nbkrrrr6s* $$$ $$$(r6rrDcCs ||||||dS)z add pair (i,j) to adjacent list.N)r)rrDr$rrrrCsrC.@ img_binarymin_wmin_hmin_dxmin_dycsJdtjdtdtdtdtdtdtffdd g}|d |||||d |S) aSplit image with recursive xy-cut algorithm. Args: img_binary (np.array): Binarized image with interesting region (255) and empty region (0). min_w (float): Ignore bbox if the width is less than this value. min_h (float): Ignore bbox if the height is less than this value. min_dx (float): Merge two bbox-es if the x-gap is less than this value. min_dy (float): Merge two bbox-es if the y-gap is less than this value. Returns: list: bbox (x0, y0, x1, y1) of split blocks. arrtop_leftresrNrOrPrQc s|\}}|j\} } tj|dkdd} t| ||} | sdS| \} }t| |D]f\}}|||d| f}tj|dkdd} t| ||}|sEq%|\}}t|dkre|||d||||d||fq%t||D] \}}|||||f}||||f}|||||||qjq%dS)Nr'Zaxisr)shapenp count_nonzero_split_projection_profileziprr)rRrSrTrNrOrPrQrrrrZ projectionZpos_yZarr_y0Zarr_y1Zr0r1Zx_arrZpos_xZarr_x0Zarr_x1Zc0c1Zy_arrxy_cutrrr_s*    *z recursive_xy_cut..xy_cutr@)rRrSrTrNrOrPrQ)rXarraytupler4float)rMrNrOrPrQrTrr^rrecursive_xy_cutsrc arr_values min_valuemin_gapc Cst||kd}t|sdS|dd|dd}t||kd}||}||d}t|d|d}t||d} | d7} || fS)uUSplit projection profile: ``` ┌──┐ arr_values │ │ ┌─┐─── ┌──┐ │ │ │ │ | │ │ │ │ ┌───┐ │ │min_value │ │<- min_gap ->│ │ │ │ │ │ | ────┴──┴─────────────┴──┴─┴───┴─┴─┴─┴─── 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 ``` Args: arr_values (np.array): 1-d array representing the projection profile. min_value (float): Ignore the profile if `arr_value` is less than `min_value`. min_gap (float): Ignore the gap if less than this value. Returns: tuple: Start indexes and end indexes of split groups. rNr'r&)rXwhererinsertr) rdrerfZ arr_indexZarr_diffZarr_diff_indexZarr_zero_intvl_startZarr_zero_intvl_endZ arr_startZarr_endrrrrZs  rZbboxcs|\}}}}tj|jtjd}|||||f|||||f<t|tjtj\} } t| ddddfdkd} tt | ddddf| d} tt | ddddf| d} ddggg}}| D]#}t | |\}}}}||ks||krqn| ||||||fqn|D]}|D]}||krq||r || |qqfdd }| D]*}t | |\}}}}||||||f}||ks||krq||rq| |q|S) aInner contours of current region, especially level 2 contours of the default opencv tree hirerachy. Args: img_binary (np.array): Binarized image with intresting region (255) and empty region (0). bbox (tuple): The external bbox. min_w (float): Ignore contours if the bbox width is less than this value. min_h (float): Ignore contours if the bbox height is less than this value. Returns: list: A list of bbox-es of inner contours. ZdtyperNrAr&c Ss8|\}}}}|\}}}} ||ko||ko||ko| |kS)Nr) bbox1bbox2rrr r r r r rrrrcontainsAs   z inner_contours..containscsD] }||r dSqdS)NTFr)riZ level_1_bboxrmZ res_level_1rrcontained_in_concerned_level_1Tsz6inner_contours..contained_in_concerned_level_1) rXZzerosrWuint8cvZ findContoursZ RETR_TREEZCHAIN_APPROX_SIMPLErgisinZ boundingRectr)rMrirNrOrrr r rRZcontoursZ hierarchyZlevel_0Zlevel_1Zlevel_2Zlevel_1_bbox_listrTrxyrrrkrlroZ level_2_bboxrrnrinner_contours!s: $ $$      ru img_sourcegapdwdhc Cs:|j\}}|ptt|dd}|ptt|dd}dtj||||||dftjd}|||||||d|ddf<tj|dkdd}t|D]#\} } t| ||} d|| ||||||t| ddf<qPtj|dkdd}t|D]\} } t| ||} d||| || ddf<q|S) uProjection profile along x and y direction. ``` ┌────────────────┐ dh │ │ └────────────────┘ gap ┌────────────────┐ ┌───┐ │ │ │ │ h │ image │ │ │ │ │ │ │ └────────────────┘ └───┘ w dw ``` Args: img_source (np.array): Source image, e.g. RGB mode. img_binary (np.array): Binarized image. gap (int, optional): Gap between sub-graph. Defaults to 5. dw (int, optional): Graph height of x projection profile. Defaults to None. dh (int, optional): Graph height of y projection profile. Defaults to None. Returns: np.array: The combined graph data. rArUrjrNr'rV)rWrr3rXZonesrprY enumerate) rwrMrxryrzrrrRvalsrvalcrrrrxy_project_profilecs *&0r)rKrKrLrL)rvNN) collectionsrnumpyrXZcv2rqrarrrr4r3r7r6rCr`rbrcrZrurrrrrs. &1+  3*(B