j!ddlmZddlmZddlmZddlZddlZddl m Z m Z ddl m Z gdZdejd efd Z ddejdeeefdzdedededed efdZdeejdeed eejfdZdejdeed ejdzfdZded eeeffdZdeeeefeeeefzd eeeefeeeefzfdZdS) )floor) median_low)AnyN)LangDetectException detect_langs) rotate_image)estimate_orientation get_languageinvert_data_structurecontourreturnctj|\}\}}}|dks|dkrdSt||z ||z S)zGet the maximum shape ratio of a contour. Args: contour: the contour from cv2.findContour Returns: the maximum shape ratio r)cv2 minAreaRectmax)r _whs V/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/models/_utils.pyget_max_width_length_ratiorsK?7++LAv1qAvvas q1ua!e  F皙?dimggeneral_page_orientationn_ctratio_threshold_for_linesmin_confidence lower_areact|jdkr|jddvsJd|jd|jddkrjtj|tj}tj|d}tj|ddtjtjz d }n| tj }|pd \}} |d uo| |k} | r|nd} | rt|| }n|jd d \} } td t| dz }td t| dz }tjtj||f}tj||d }tj|tjtj\}}t+fd|Dt,d}g}|d |D]}tj|\}\} } }| | kr | | } } |dz}|dkr |dz }|dk |dkr |dz}|dk | dkrC| | z |kr||s| | z d |z kr||dz t|dkrd}nJt3| }t5|dkrt7| nd}t5|dkrd}| |z}|dkr |dz}|dk |dkr |dz }|dk | r@t5|dzdkr|St5|t5|kr|dkr|St9|S)a&Estimate the angle of the general document orientation based on the lines of the document and the assumption that they should be horizontal. Args: img: the img or bitmap to analyze (H, W, C) general_page_orientation: the general orientation of the page (angle [0, 90, 180, 270 (-90)], confidence) estimated by a model n_ct: the number of contours used for the orientation estimation ratio_threshold_for_lines: this is the ratio w/h used to discriminates lines min_confidence: the minimum confidence to consider the general_page_orientation lower_area: the minimum area of a contour to be considered Returns: the estimated angle of the page (clockwise, negative for left side rotation, positive for right side rotation) r)rz Image shape z not supportedr)threshmaxvaltyper%)rrNr) iterationscDg|]}tj|k|S)r contourArea).0r r"s r z(estimate_orientation..Xs-RRRWCOG,D,Dz,Q,Q,Q,Q,QrT)keyreverseZiihiL)lenshapercvtColorCOLOR_BGR2GRAY medianBlur thresholdTHRESH_BINARY_INV THRESH_OTSUastypenpuint8rrrgetStructuringElement MORPH_RECTdilate findContours RETR_LISTCHAIN_APPROX_SIMPLEsortedrrappendrabsroundint)rrrr r!r"gray_imgr(page_orientationorientation_confidence is_confident base_anglerrk_xk_ykernelcontoursranglesr angle skew_anglemedian final_angles ` rr r "s. sy>>Q  39R=F#:#:#:(A..x#CDY\_\kDklllmnoBH%%/G/S8,,#4/\4Jn4\L%18!!qJ :fzk222A2A!eAGnn&&!eAGnn&&*3>C:FFFFq999"63=#:QRRKHaRRRRRRR &H FETE?**?7336Aq5 q55aqA RKEsll SLEsllbjj SLEbjj q551u000 e$$$$Q6666 ebj))) 6{{a V$$$'*6{{a'7'7eFmm^^Q  z??b Jz)K   s       s     $ z??R 1 $ $# # z??c"233 3 38HA8M8M# #   rcrops orientationscxd|D}t|dkrdt||DngS)zRotate each crop of the list according to the predicted orientation: 0: already straight, no rotation 1: 90 ccw, rotate 3 times ccw 2: 180, rotate 2 times ccw 3: 270 ccw, rotate 1 time ccw c(g|]}|dkrd|z ndS)rr.)r0preds rr1z!rectify_crops..s(JJJT AHHqJJJrrcNg|]"\}}|dkr|ntj||#S)r)r?rot90)r0 orientationcrops rr1z!rectify_crops..s8uuuGX{TX!!rxk'B'Buuur)r6zip)rZr[s r rectify_cropsresYKJ\JJJL |  q  vu\_`lns\t\tuuuu rpage_loc_predsct|dkr.tjdt||DdndS)zOrient the quadrangle (Polygon4P) according to the predicted orientation, so that the points are in this order: top L, top R, bot R, bot L if the crop is readable rcBg|]\}}tj||dS)raxis)r?roll)r0rb page_loc_preds rr1z%rectify_loc_preds..s=   .K {;;;   rriN)r6r?stackrd)rfr[s rrectify_loc_predsrnsg |  q    25lN2S2S         rtextc t|d}n#t$rYdSwxYwt|dkst|dkr |jdkrdS|j|jfS)aGet languages of a text using langdetect model. Get the language with the highest probability or no language if only a few words or a low probability Args: text (str): text Returns: The detected language in ISO 639 code and confidence score r)unknownrr%r&r)rlowerrr6problang)rorts rr r sDJJLL))!, ~~ 4yyA~~#d))q..TY#-=-=~ 9di s '* 88xcttr`tdDdks Jdfdt DStt rfddDSt dttd) aKInvert a list of dict of elements to a dict of list of elements and the other way around Args: x: a list of dictionaries with the same keys or a dictionary of lists of the same length Returns: dictionary of list when x is a list of dictionaries or a list of dictionaries when x is dictionary of lists c,h|]}t|Sr.)r6)r0vs r z(invert_data_structure..s///qCFF///rr%z.s':::ASAYY:::rc0i|]fdDS)c g|] }| Sr.r.)r0dicks rr1z4invert_data_structure...s(((sCF(((rr.)r0rrus @r z)invert_data_structure..s1777Q((((a(((777rrz2Expected input to be either a dict or a list, got z instead.) isinstancer{r6valuesrdlist TypeErrorr*input)rus`rr r s!Te//AHHJJ///00A5557u555::::ahhjj)9:::: At  e7777!A$7777cTRW[[cccdddr)Nrrrr)mathr statisticsrtypingrrnumpyr? langdetectrrdoctr.utils.geometryr__all__ndarrayfloatrtuplerKr rrernstrr r{r r.rrrsN !!!!!! 88888888------ K K K  u    ":>'( vv v#CJ/$6v v % v  v  v vvvvr  s) "*&Js)Z$( s uS%Z0    &e DcNd3S >22e $sCx.Dd3i00eeeeeer