khwddlmZmZmZmZmZmZmZddlmZddl m Z ddl m Z m Z ddl Z ddlZddlZddlZddlmZddlmZddlmZd d lmZmZd d lmZmZm Z m!Z!m"Z"d d l#m$Z$m%Z%d Z&ddddZ'dZ(e'GddeZ)e'GddeZ*e'GddeZ+e)e*e+gZ,e,Dcic]}|jZ|c}Z.e/e,Dcic]\}}|jZ|c}}Z0e'eGddeZ1ycc}wcc}}w))ListUnionDictrAnyOptionalTuple)Iterable)copy) getmembers isfunctionN)Image)getPerspectiveTransform)warpPerspective)BaseCoordElementBaseLayoutElement)cvt_coordinates_to_pointscvt_points_to_coordinatesperspective_transformationvertice_in_polygon polygon_area)NotSupportedShapeErrorInvalidShapeErrorcBtjfd}|S)Ncf|g|i|}t|trt|}||_|SyN) isinstancerr block)selfargskwargsoutfuncs a/var/www/teggl/fontify/venv/lib/python3.12/site-packages/layoutparser/elements/layout_elements.pywrapz"mixin_textblock_meta..wrap(s<4)$)&) c+ ,:DDJK - functoolswrapsr#r%s` r$mixin_textblock_metar+'s%__T Kr& base_classc`|tjt|St|tD]}\}}|j r|dk(r>|j ddD]+}t||st||j |_TVt||sct||j |_|S)Nr,r) r(partialinherit_docstringsr r __doc____mro__hasattrgetattr)clsr-namer#parents r$r0r03s {  !3 KK j1 A d <<   ++ab/ 64(#*64#8#@#@DL  z4(&z48@@  A Jr&cBtjfd}|S)NcZt|tr |j}||g|i|}|Sr)r TextBlockr)rotherr r!r"r#s r$r%zsupport_textblock..wrapIs1 eY 'KKE4000 r&r'r*s` r$support_textblockr<Hs%__T Kr&ceZdZdZdZgdZddZedZedZ edZ ed Z ed Z ed Z d Zed ZedZeidfdZeddedefdZeddedefdZddZdZdZdZdZdZy)Intervala This class describes the coordinate system of an interval, a block defined by a pair of start and end point on the designated axis and same length as the base canvas on the other axis. Args: start (:obj:`numeric`): The coordinate of the start point on the designated axis. end (:obj:`numeric`): The end coordinate on the same axis as start. axis (:obj:`str`): The designated axis that the end points belong to. canvas_height (:obj:`numeric`, `optional`, defaults to 0): The height of the canvas that the interval is on. canvas_width (:obj:`numeric`, `optional`, defaults to 0): The width of the canvas that the interval is on. intervalstartendaxis canvas_height canvas_widthNc||ksJd||_||_|dvs Jd|d||_|xsd|_|xsd|_y)Nz3Invalid input for start and end. Start must <= end.)xyz Invalid axis z. Axis must be in 'x' or 'y'rr@)rrArBrCrDrEs r$__init__zInterval.__init__is`|SRS| z!U]4&8T#UU! */a(-Ar&cj|jdk(r |jS|j|jz S)a- Calculate the height of the interval. If the interval is along the x-axis, the height will be the height of the canvas, otherwise, it will be the difference between the start and end point. Returns: :obj:`numeric`: Output the numeric value of the height. rG)rCrDrBrArs r$heightzInterval.heightus/ 99 %% %88djj( (r&cj|jdk(r |jS|j|jz S)a) Calculate the width of the interval. If the interval is along the y-axis, the width will be the width of the canvas, otherwise, it will be the difference between the start and end point. Returns: :obj:`numeric`: Output the numeric value of the width. rH)rCrErBrArKs r$widthzInterval.widths/ 99 $$ $88djj( (r&c|jdk(r&|jd|j|jf}|Sd|j|j|jf}|S)a6 This method considers an interval as a rectangle and calculates the coordinates of the upper left and lower right corners to define the interval. Returns: :obj:`Tuple(numeric)`: Output the numeric values of the coordinates in a Tuple of size four. rGr)rCrArBrDrE)rcoordss r$ coordinateszInterval.coordinatessV 99 jj!TXXt/A/ABF T%6%6AF r&c,t|jS)z Return the coordinates of all four corners of the interval in a clockwise fashion starting from the upper left. Returns: :obj:`Numpy array`: A Numpy array of shape 4x2 containing the coordinates. rrQrKs r$pointszInterval.points))9)9::r&c:|j|jzdz S)z Calculate the mid-point between the start and end point. Returns: :obj:`Tuple(numeric)`: Returns of coordinate of the center. @rArBrKs r$centerzInterval.centers TXX%,,r&c4|j|jzS)a.Return the area of the covered region of the interval. The area is bounded to the canvas. If the interval is put on a canvas, the area equals to interval width * canvas height (axis='x') or interval height * canvas width (axis='y'). Otherwise, the area is zero. )rLrNrKs r$areaz Interval.areas{{TZZ''r&c4t|tjr|jdd\}}nYt|tr|j |j }}n0t|tjr|j\}}nt|j||S)a Set the height and the width of the canvas that the interval is on. Args: canvas (:obj:`Numpy array` or :obj:`BaseCoordElement` or :obj:`PIL.Image.Image`): The base element that the interval is on. The numpy array should be the format of `[height, width]`. Returns: :obj:`Interval`: A copy of the current Interval with its canvas height and width set to those of the input canvas. N)rDrE) rnpndarrayshaperrLrNr sizeNotImplementedErrorset)rcanvashws r$ put_on_canvaszInterval.put_on_canvassv fbjj )<<#DAq  0 1==&,,qA  ,;;DAq% %xxaax88r&ct|trg|j|jk(rC|j}|j |j|z|j |z|jSt |St|tr.|j|jj|St|tr.|j|jj|Std|jdNInvalid input type for other)rr>rCrA __class__rBr Rectanglerg to_rectangle condition_on Quadrilateralto_quadrilateral Exceptionrr;ds r$rozInterval.condition_ons eX &zzTYY&KK~~djj1ndhhlDIINNDz! y )%%e,99;HHO O } -%%e,==?LLUS S1%//1B*MN Nr&ct|trg|j|jk(rC|j}|j |j|z |j |z |jSt |St|tr.|j|jj|St|tr.|j|jj|Std|jdri)rr>rCrArlrBr rmrgrn relative_torprqrrrss r$rvzInterval.relative_tos eX &zzTYY&KK~~djj1ndhhlDIINNDz! y )%%e,99;GGN N } -%%e,==?KKER R1%//1B*MN Nr&FcF|jdi|}t|tr|j|jk7ry|s@|j|jcxkxr!|j cxkxr|j kScS|j|j cxkxr|j kScSt|tst|tr|j\}}}}|rC|jdk(r||j cxkxr|kScS||j cxkxr|kScS|jdk(r,||jcxkxr|j cxkxr|kScS||jcxkxr|j cxkxr|kScStd|jd)NFrGrjrk) padrr>rCrArBrYrmrprQrrrl)rr; soft_marginrYx_1y_1x_2y_2s r$is_inzInterval.is_in s\ (K( eX &yyEJJ& ;;$**MMEIIMMMM ;;$++BBBBB y )Z}-M!&!2!2 Cc399#$++44444$++4444499#$**??C????$**??C????1%//1B*MN Nr&r;strictc t|tr|j|jk7r|jdk(rE|jdk(r6t|j|j|j |j St|j|j|j |j S|j t|j|jt|j |j |j|j|jSt|tr|j\}}}}|jdk(r6tt||j|t||j |S|jdk(r6t|t||j|t||j Sy t|trV|r tdtjd|j dtd|j!|j#St%d|j d) rGrHThe intersection between an Interval and a Quadrilateral might generate Polygon shapes that are not supported in the current version of layoutparser. You can pass `strict=False` in the input that converts the Quadrilateral to Rectangle to avoid this Exception.(With `strict=False`, the other of shape  will be converted to  for obtaining the intersectionrjrkN)rr>rCrmrArBrlmaxminrDrErQrprwarningswarn intersectrnrrrr;rr{r|r}r~s r$rzInterval.intersect+s eX &yyEJJ&99# c(9$TZZdhh RR$U[[$**eiiRR~~ EKK0%)),II&&%% y )!&!2!2 Cc3yyC S$**!5sCTXXu>OOefoepqPQ~~e&8&8&:;;1%//1B*MN Nr&c t|tr|j|jk7r td|j t |j |j t|j|j|j|j|jSt|tr|j\}}}}|jdk(r6tt ||j |t||j|S|jdk(r6t|t ||j |t||jSy t|trV|r tdtj d|jdtd|j#|j%St'd|jd ) rz8Unioning two intervals of different axes is not allowed.rGrHrrrrrjrkN)rr>rCrrlrrArrBrDrErmrQrprrrunionrnrrrs r$rzInterval.unionSs eX &yyEJJ&'N~~ EKK0%)),II&&%% y )!&!2!2 Cc3yyC S$**!5sCTXXu>OOefoepqPQzz%"4"4"6771%//1B*MN Nr&c|jdk(rO|j|z }|j|z}|s|r{tjd|j j nN|j|z }|j|z}|s|r,tjd|j j |r td|}|j||S)NrGz)Invalid padding top/bottom for an x axis z(Invalid padding right/left for a y axis rrX) rCrArBrrrl__name__rrc)rleftrighttopbottom safe_moderArBs r$ryz Interval.padys 99 JJ%E((U"Cf ?@W@W?XYJJ$E((V#Cu >t~~?V?V>WX 5MExxex--r&ct|tr<|jdk(r|dn|d}tjd|jd|j |z}|j |z}|j||S)a5 Shift the interval by a user specified amount along the same axis that the interval is defined on. Args: shift_distance (:obj:`numeric`): The number of pixels used to shift the interval. Returns: :obj:`BaseCoordElement`: The shifted Interval object. rGrrz=Input shift for multiple axes. Only use the distance for the  axisrXrr rCrrrArBrc)rshift_distancerArBs r$shiftzInterval.shifts nh /%)YY#%5q!>!;L  MMOPTPYPY{Z_`  ^+hh'xxex--r&ct|tr<|jdk(r|dn|d}tjd|jd|j |z}|j |z}|j||S)a6 Scale the layout element by a user specified amount the same axis that the interval is defined on. Args: scale_factor (:obj:`numeric`): The amount for downscaling or upscaling the element. Returns: :obj:`BaseCoordElement`: The scaled Interval object. rGrrz;Input scale for multiple axes. Only use the factor for the rrXr)r scale_factorrArBs r$scalezInterval.scalesw lH -.2ii3.><?LQROL MMMdii[X]^  \)hh%xxex--r&c|j|j\}}}}|t|t|t|t|fSr)rgrQintrimager{r|r}r~s r$ crop_imagezInterval.crop_imagesK!//6BBS#sSXC(#c(SX*==>>r&c&t|jS)z Convert the Interval to a Rectangle element. Returns: :obj:`Rectangle`: The converted Rectangle object. rmrQrKs r$rnzInterval.to_rectangles$**++r&c,t|jS)z Convert the Interval to a Quadrilateral element. Returns: :obj:`Quadrilateral`: The converted Quadrilateral object. rprTrKs r$rqzInterval.to_quadrilateralsT[[))r&NNTrrrrT)r __module__ __qualname__r1_name _featuresrIpropertyrLrNrQrTrYr[rgr<rorvrrboolrrryrrrrnrqrxr&r$r>r>SsJ" EII . ) ) ) )" ; ;--((94OO*OO*')%OO<%O/%O%O%ON#O+#OT#O#OJ.,...*?,*r&r>ceZdZdZdZgdZdZedZedZ edZ edZ ed Z ed Z ed Zed Zeid fdZeddedefdZeddedefdZddZddZddZdZdZdZy)rma This class describes the coordinate system of an axial rectangle box using two points as indicated below:: (x_1, y_1) ---- | | | | | | ---- (x_2, y_2) Args: x_1 (:obj:`numeric`): x coordinate on the horizontal axis of the upper left corner of the rectangle. y_1 (:obj:`numeric`): y coordinate on the vertical axis of the upper left corner of the rectangle. x_2 (:obj:`numeric`): x coordinate on the horizontal axis of the lower right corner of the rectangle. y_2 (:obj:`numeric`): y coordinate on the vertical axis of the lower right corner of the rectangle. rectangler{r|r}r~c<||_||_||_||_yrr)rr{r|r}r~s r$rIzRectangle.__init__sr&c4|j|jz S)z Calculate the height of the rectangle. Returns: :obj:`numeric`: Output the numeric value of the height. )r~r|rKs r$rLzRectangle.heightxx$((""r&c4|j|jz S)z Calculate the width of the rectangle. Returns: :obj:`numeric`: Output the numeric value of the width. )r}r{rKs r$rNzRectangle.widthrr&c^|j|j|j|jfS)z Return the coordinates of the two points that define the rectangle. Returns: :obj:`Tuple(numeric)`: Output the numeric values of the coordinates in a Tuple of size four. rrKs r$rQzRectangle.coordinatess#$((DHHdhh77r&c,t|jS)z Return the coordinates of all four corners of the rectangle in a clockwise fashion starting from the upper left. Returns: :obj:`Numpy array`: A Numpy array of shape 4x2 containing the coordinates. rSrKs r$rTzRectangle.pointsrUr&cr|j|jzdz |j|jzdz fS)z Calculate the center of the rectangle. Returns: :obj:`Tuple(numeric)`: Returns of coordinate of the center. rW)r{r}r|r~rKs r$rYzRectangle.centers3488#s*TXX-@C,GGGr&c4|j|jzS)z3 Return the area of the rectangle. )rNrLrKs r$r[zRectangle.area*s zzDKK''r&ct|trt|jdk(r|jd}}nd|j}}|j |j |z|j |z|j|z|j|zSt|trY|j\}}}}|j |j |z|j |z|j|z|j|zSt|trIt|j|jd}|j ||j|j St#d|jdNrGrTis_invrjrkrr>rCrArlr{r|r}r~rmrQrprperspective_matrixrTrLrNrrrr;dxdy_transformed_pointss r$rozRectangle.condition_on1s) eX &zzS aBEKKB>>2 txx"}dhhmTXX] y ) ,,LBAq>>2 txx"}dhhmTXX] } -!;(($++d" ??#5t{{DJJO O1%//1B*MN Nr&ct|trt|jdk(r|jd}}nd|j}}|j |j |z |j |z |j|z |j|z St|trY|j\}}}}|j |j |z |j |z |j|z |j|z St|trIt|j|jd}|j ||j|j St#d|jdNrGrFrrjrkrrs r$rvzRectangle.relative_toOs' eX &zzS aBEKKB>>2 txx"}dhhmTXX] y ) ,,LBAq>>2 txx"}dhhmTXX] } -!;(($++e" ??#5t{{DJJO O1%//1B*MN Nr&Fc|jd i|}t|tr|sl|jdk(r|j|j }}n|j |j}}|j|cxkxr|cxkxr|jkScS|jdk(r|jdn|jd}|j|cxkxr|jkScSt|trL|jd}|jd}|j||xr|j||St|tro|s8|jD cgc]} t!| |j} } t#| St%j&|j}t!||jSt)d|j*dcc} w) NrGrrrCrHrYrjrkrx)ryrr>rCr{r}r|r~rArBrYrm to_intervalrrprTrallr^arrayrrrl rr;rzrYrArBc x_interval y_intervalvertice is_vertice_ins r$rzRectangle.is_inls (K( eX &::$!%4883E!%4883E{{e?s?eii????&+jjC&7DKKNT[[^{{a45994444 y )***4J***4J::j:8TZZ6>H> } -NR[[!BI&w =! !=))$++.)&%,,??1%//1B*MN N!sGr;rc zt|tr|j|St|tr|j t |j |j t |j|jt|j|jt|j|jSt|trV|r tdtjd|jdtd|j|jSt!d|jd)raThe intersection between a Rectangle and a Quadrilateral might generate Polygon shapes that are not supported in the current version of layoutparser. You can pass `strict=False` in the input that converts the Quadrilateral to Rectangle to avoid this Exception.rrrrjrk)rr>rrmrlrr{r|rr}r~rprrrrnrrrr;rs r$rzRectangle.intersects eX &??4( ( y )>>DHHeii(DHHeii(DHHeii(DHHeii(  } -,[ >u>OOefoepqPQ~~e&8&8&:;;1%//1B*MN Nr&c zt|tr|j|St|tr|j t |j |j t |j|jt|j|jt|j|jSt|trV|r tdtjd|jdtd|j|j!St#d|jd)rrrrrrjrk)rr>rrmrlrr{r|rr}r~rprrrrrnrrrs r$rzRectangle.unions eX &??4( ( y )>>DHHeii(DHHeii(DHHeii(DHHeii(  } -,[ >u>OOefoepqPQzz%"4"4"6771%//1B*MN Nr&c|j|z }|j|z }|j|z}|j|z} |rt d|}t d|}|j |||| S)Nr)r{r|r}r~rrl) rrrrrrr{r|r}r~s r$ryz Rectangle.padsfhhohhnhhhh a+Ca+C~~c3S11r&ct|ts|}|}nt|dk(sJd|\}}|j|z}|j|z}|j |z}|j |z}|j||||S)Nr]Rshift_distance should have 2 elements, one for x dimension and one for y dimensionrr lenr{r|r}r~rl)rrshift_xshift_yr{r|r}r~s r$rzRectangle.shifts.(3$G$GN#q( dc d(- GWhh hh hh hh ~~c3S11r&ct|ts|}|}nt|dk(sJd|\}}|j|z}|j|z}|j |z}|j |z}|j||||S)Nr]Pscale_factor should have 2 elements, one for x dimension and one for y dimensionr)rrscale_xscale_yr{r|r}r~s r$rzRectangle.scales,1"G"GL!Q& ba b&+ GWhh hh hh hh ~~c3S11r&c|j\}}}}|t|t|t|t|fSr)rQrrs r$rzRectangle.crop_images@!--S#sSXC(#c(SX*==>>r&c |dk(r|j|j}}n|j|j}}t ||fd|i|SNrGrC)r{r}r|r~r>)rrCr!rArBs r$rzRectangle.to_intervalsB 3;4883E4883Es8888r&c,t|jSrrrKs r$rqzRectangle.to_quadrilateralsT[[))r&Nrrrr)rrrr1rrrIrrLrNrQrTrYr[r<rorvrrrrrryrrrrrqrxr&r$rmrmsF( E,I####88 ; ;HH(( OO:OO8')%$O$OLO/OOO:O+OTOO6 22"2"?9*r&rmceZdZdZdZgdZ d!deeje e e ffdZ e dZ e dZ e dZe d Ze d Ze d Ze d Ze d ZdZedZedZeidfdZed"dedefdZed"dedefdZd#dZd$dZd%dZdZ dZ!dZ"dZ#dZ$de%e&e'fffd Z(xZ)S)&rpa This class describes the coodinate system of a four-sided polygon. A quadrilateral is defined by the coordinates of its 4 corners in a clockwise order starting with the upper left corner (as shown below):: points[0] -...- points[1] | | . . . . . . | | points[3] -...- points[2] Args: points (:obj:`Numpy array` or `list`): A `np.ndarray` of shape 4x2 for four corner coordinates or a list of length 8 for in the format of `[p0_x, p0_y, p1_x, p1_y, p2_x, p2_y, p3_x, p3_y]` or a list of length 4 in the format of `[[p0_x, p0_y], [p1_x, p1_y], [p2_x, p2_y], [p3_x, p3_y]]`. height (:obj:`numeric`, `optional`, defaults to `None`): The height of the quadrilateral. This is to better support the perspective transformation from the OpenCV library. width (:obj:`numeric`, `optional`, defaults to `None`): The width of the quadrilateral. Similarly as height, this is to better support the perspective transformation from the OpenCV library. quadrilateral)rTrLrNrTct|tjr(|jdk7rt d|jdt|t rt |dk(r&tj|jdd}ngt |dk(r)t|dt rtj|}n0t dt |d t d t|d ||_ ||_ ||_ y) N)r]zInvalid points shape: .rr]rz!Invalid number of points element z. Should be 8.zInvalid input type for points z-.Please make sure it is a list of np.ndarray.) rr^r_r` ValueErrorlistrrreshapetype_points_width_height)rrTrLrNs r$rIzQuadrilateral.__init__,s fbjj )||v% #9&,,q!IJJ  %6{a&)11!Q7V!jD&A&) 7F }NS0f???     r&c|j |jS|jdddfj|jdddfjz S)z Return the user defined height, otherwise the height of its circumscribed rectangle. Returns: :obj:`numeric`: Output the numeric value of the height. Nr)rrTrrrKs r$rLzQuadrilateral.heightFsO << #<< {{1a4 $$&QT):)>)>)@@@r&c|j |jS|jdddfj|jdddfjz S)z Return the user defined width, otherwise the width of its circumscribed rectangle. Returns: :obj:`numeric`: Output the numeric value of the width. Nr)rrTrrrKs r$rNzQuadrilateral.widthSsO ;; ";; {{1a4 $$&QT):)>)>)@@@r&c,t|jS)a Return the coordinates of the upper left and lower right corners points that define the circumscribed rectangle. Returns :obj:`Tuple(numeric)`: Output the numeric values of the coordinates in a Tuple of size four. )rrTrKs r$rQzQuadrilateral.coordinates`s)55r&c|jS)z Return the coordinates of all four corners of the quadrilateral in a clockwise fashion starting from the upper left. Returns: :obj:`Numpy array`: A Numpy array of shape 4x2 containing the coordinates. )rrKs r$rTzQuadrilateral.pointsls||r&cht|jjdjS)z Calculate the center of the quadrilateral. Returns: :obj:`Tuple(numeric)`: Returns of coordinate of the center. rr)tuplerTmeantolistrKs r$rYzQuadrilateral.centerxs*T[[%%1%-44677r&c^t|jdddf|jdddfS)z7 Return the area of the quadrilateral. Nrr)rrTrKs r$r[zQuadrilateral.areas+ DKK1-t{{1a4/@AAr&cdd|j|jd}dd|j|jd}|j||S)Nrrrr])rNrLmap_to_points_ordering)rx_mapy_maps r$mapped_rectangle_pointsz%Quadrilateral.mapped_rectangle_pointssC! tzz:!  <**5%88r&c~t|jjd|jjdS)Nfloat32)_getPerspectiveTransformrTastyperrKs r$rz Quadrilateral.perspective_matrixs5' KK  y )  ( ( / / :  r&cD|jjdjd}tjtj|j |dddftj|j |dddfgj S)Nrrr)rTargsortr^vstack vectorizegetT)rrrpoints_orderings r$rz$Quadrilateral.map_to_points_orderings++--1-5==1=Eyy' UYY'1(=>' UYY'1(=>  !  r&ct|trI|jdk(r|j|jdgS|jd|jgSt|t r'|j|j |jgSt|trIt|j|jd}|j||j|jStd|jdrrr>rCrrArmr{r|rprrrTrlrLrNrrrr;rs r$rozQuadrilateral.condition_ons eX &zzS zz5;;"233zz1ekk"233 y )::uyy%))45 5 } -!;(($++d" >>"4dkk4::N N1%//1B*MN Nr&ct|trK|jdk(r|j|j dgS|jd|j gSt|t r)|j|j |j gSt|trIt|j|jd}|j||j|jStd|jdrrrs r$rvzQuadrilateral.relative_tos eX &zzS zzEKK<"344zz1u{{l"344 y ):: zEII:67 7 } -!;(($++e" >>"4dkk4::N N1%//1B*MN Nr&Fc|jd i|}t|tr|sx|jdk(r|jd|jd}}n|jd|jd}}|j |cxkxr|cxkxr|j kScS|jdk(r|jdn|jd}|j |cxkxr|j kScSt|trL|jd}|jd}|j||xr|j||St|tro|s8|jD cgc]} t| |j} } t| Stj |j}t||jSt#d |j$d cc} w) NrGrr]rrrrHrrjrkrx)ryrr>rCrQrArBrYrmrrrprTrrr^rrrrlrs r$rzQuadrilateral.is_ins (K( eX &::$!%!1!1!!4d6F6Fq6I3E!%!1!1!!4d6F6Fq6I3E{{e?s?eii????&+jjC&7DKKNT[[^{{a45994444 y )***4J***4J::j:8TZZ6>H> } -NR[[!BI&w =! !=))$++.)&%,,??1%//1B*MN N!s*G"r;rc|r tdt|tst|trrmrrrrnrprrrlrs r$rzQuadrilateral.intersects (Y %*j .J abkalmLMt'8'8':;;E=1 abkalmLM((*44U5G5G5IJJ"5eoo5Fj QRRr&c|r tdt|tst|trrmrrrrnrprrrlrs r$rzQuadrilateral.unions (Y %*j .J abkalmLM{{4#4#4#677E=1 abkalmLM((*001C1C1EFF"5eoo5Fj QRRr&c| | ||d}| | ||d}|j||}|j|z} |rtj| d} |j | S)NrrrT)rrTr^maximumrc) rrrrrrrr padding_matrTs r$ryzQuadrilateral.pad'sjEte%8DcTf811%? {* ZZ*Fxxvx&&r&ct|ts||g}nt|dk(sJd|}|jt j |z}|j |S)Nr]rrrr rrTr^rrc)rr shift_matrTs r$rzQuadrilateral.shift4sd.(3'8IN#q( dc d(&Irxx 22xxvx&&r&ct|ts||g}nt|dk(sJd|}|jt j |z}|j |S)Nr]rrr)rr scale_matrTs r$rzQuadrilateral.scaleBsd,1%|4IL!Q& ba b&$Irxx 22xxvx&&r&ct||jt|jt|jfS)z Crop the input image using the points of the quadrilateral instance. Args: image (:obj:`Numpy array`): The array of the input image. Returns: :obj:`Numpy array`: The array of the cropped image. )_warpPerspectiverrrNrLrrs r$rzQuadrilateral.crop_imagePs4 4**S_c$++>N,O  r&c ^|j\}}}}|dk(r||}}n||}}t||fd|i|Sr)rQr>) rrCr!r{r|r}r~rArBs r$rzQuadrilateral.to_interval_sF!--S#s 3;c3Ec3Es8888r&c&t|jSrrrKs r$rnzQuadrilateral.to_rectangleis$**++r&c|j|jurytj|j|jj S)NF)rlr^iscloserTrrr;s r$__eq__zQuadrilateral.__eq__ls7 ??$.. 0zz$++u||488::r&c gd}dj|Dcgc]}|dt||c}}|jjd|dScc}w)N)rTrNrLz, =())joinr4rlr)rkeyskeyinfo_strs r$__repr__zQuadrilateral.__repr__qsW,99tLQwtS'9&:;LM..))*!H:Q77MsAreturncnt|}|djdj|d<|S)a Generate a dictionary representation of the current object:: { "block_type": "quadrilateral", "points": [ p[0,0], p[0,1], p[1,0], p[1,1], p[2,0], p[2,1], p[3,0], p[3,1] ], "height": value, "width": value } rT)superto_dictrr)rdatarls r$r7zQuadrilateral.to_dictvs7"w h//3::<X r&rrrrr)*rrrr1rrrr^r_rrIrrLrNrQrTrYr[rrrr<rorvrrrrrryrrrrrnr)r2rstrrr7 __classcell__)rls@r$rprp s6 E-IOSBJJd4j894 A A A A 6 6  88BB 99   OO.OO.')%$O$OLS/SSS*S+STSS( ' ' '  9,; 8 c3hr&rpcleZdZdZdZgdZ ddZedZedZ edZ ed Z ed Z e d Ze d Zid fdZe d dedefdZe d dedefdZe dZe d!dZe dZdZd"deefdZdZdZdeeeffdZ e!deeefddfdZ"y)#r:a This class constructs content-related information of a layout element in addition to its coordinate definitions (i.e. Interval, Rectangle or Quadrilateral). Args: block (:obj:`BaseCoordElement`): The shape-specific coordinate systems that the text block belongs to. text (:obj:`str`, `optional`, defaults to None): The ocr'ed text results within the boundaries of the text block. id (:obj:`int`, `optional`, defaults to `None`): The id of the text block. type (:obj:`int`, `optional`, defaults to `None`): The type of the text block. parent (:obj:`int`, `optional`, defaults to `None`): The id of the parent object. next (:obj:`int`, `optional`, defaults to `None`): The id of the next block. score (:obj:`numeric`, defaults to `None`): The prediction confidence of the block textblock)textidrr7nextscoreNct|tsJ||_||_||_||_||_||_||_yr) rrrr=r>rr7r?r@)rrr=r>rr7r?r@s r$rIzTextBlock.__init__sF%!1222      r&c.|jjS)z Return the height of the shape-specific block. Returns: :obj:`numeric`: Output the numeric value of the height. )rrLrKs r$rLzTextBlock.heightszz   r&c.|jjS)z Return the width of the shape-specific block. Returns: :obj:`numeric`: Output the numeric value of the width. )rrNrKs r$rNzTextBlock.widthszzr&c.|jjS)z Return the coordinates of the two corner points that define the shape-specific block. Returns: :obj:`Tuple(numeric)`: Output the numeric values of the coordinates in a Tuple of size four. )rrQrKs r$rQzTextBlock.coordinatesszz%%%r&c.|jjS)z Return the coordinates of all four corners of the shape-specific block in a clockwise fashion starting from the upper left. Returns: :obj:`Numpy array`: A Numpy array of shape 4x2 containing the coordinates. )rrTrKs r$rTzTextBlock.pointsszz   r&c.|jjS)z6 Return the area of associated block. )rr[rKs r$r[zTextBlock.areas zzr&c8|jj|Sr)rror(s r$rozTextBlock.condition_onszz&&u--r&c8|jj|Sr)rrvr(s r$rvzTextBlock.relative_toszz%%e,,r&Fc<|jj|||Sr)rr)rr;rzrYs r$rzTextBlock.is_inszz{F;;r&r;rc<|jj||SN)r)rrrs r$rzTextBlock.unionszzf55r&c<|jj||SrK)rrrs r$rzTextBlock.intersectszz##E&#99r&c8|jj|Sr)rr)rrs r$rzTextBlock.shiftszz//r&c@|jj|||||Sr)rry)rrrrrrs r$ryz TextBlock.padszz~~dE3 BBr&c8|jj|Sr)rr)rrs r$rzTextBlock.scale szz --r&c8|jj|Sr)rrr#s r$rzTextBlock.crop_imageszz$$U++r&rCc t|jtr|S|stddd|j |jj dd|i|S)Nz#Please provide valid `axis` values rGz as the inputrCrrx)rrr>rrcr)rrCr!s r$rzTextBlock.to_intervals^ djj( +K 9#]S88"8$**"8"8"Md"Mf"M8N Nr&ct|jtr|S|j|jj SNrR)rrrmrcrnrKs r$rnzTextBlock.to_rectangles3 djj) ,K88$**"9"9";8< , } )rr7rr4)r base_dictfvals r$r7zTextBlock.to_dict(sRJJ&&(  0A$"C&tQ/ !  0r&r8c |dtvs Jd|dt|dj|}||fi|jDcic]}||j|dc}Scc}w)a/Initialize the textblock based on the dictionary representation. It generate the block based on the `block_type` and `block_attr`, and loads the textblock specific features from the dict. Args: data (:obj:`dict`): The dictionary representation of the object block_typezInvalid block_type N)BASECOORD_ELEMENT_NAMEMAP from_dictrr )r5r8rrXs r$r]zTextBlock.from_dict9s}  "; ; 6 l!3 4 5 6 ;*$|*<=GGM5J3==IaQD 11IJJIsA& )NNNNNNrrr)#rrrr1rrrIrrLrNrQrTr[r+rorvrrrrrrryrrrr9rrnrqrrr7 classmethodr]rxr&r$r:r:s* EAISW !!  && ! ! ..--(*%<6+6T66:/:::00CC..,O O= A c3h"KT#s(^K KKr&r:r)2typingrrrrrrcollections.abcr r inspectr r rr(numpyr^pandaspdPILr cv2rrrr"baserrutilsrrrrrerrorsrrr+r0r<r>rmrpALL_BASECOORD_ELEMENTSrr\ enumerateBASECOORD_ELEMENT_INDEXMAPr:)eleidxs00r$ros:A@@$*C35> t*{*{*{*| v* v*v*r |$||~ #I}=7MNSYY^N#,-C#DsCCIIsN /0sK!sK1sKOs -DD