NgwddlmZmZmZmZmZmZmZddlmZddl 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,de,DZ-de.e,DZ/e'eGddeZ0dS))ListUnionDictrAnyOptionalTuple)Iterable)copy) getmembers isfunctionN)Image)getPerspectiveTransform)warpPerspective)BaseCoordElementBaseLayoutElement)cvt_coordinates_to_pointscvt_points_to_coordinatesperspective_transformationvertice_in_polygon polygon_area)NotSupportedShapeErrorInvalidShapeErrorcFtjfd}|S)Ncz|g|Ri|}t|trt|}||_|SdSN) isinstancerr block)selfargskwargsoutfuncs a/var/www/html/ai-engine/env/lib/python3.11/site-packages/layoutparser/elements/layout_elements.pywrapz"mixin_textblock_meta..wrap(sVd4)$)))&)) c+ , , ::DDJK   functoolswrapsr#r%s` r$mixin_textblock_metar+'s8_T Kr& base_classcd|tjt|St|tD]|\}}|jr |dkr?|jddD].}t||rt||j|_n/Rt||rt||j|_}|S)Nr,r) r(partialinherit_docstringsr r __doc____mro__hasattrgetattr)clsr-namer#parents r$r0r03s { !3 KKKK j11 A A d <     +abb/  64((#*64#8#8#@DLEz4(( A&z488@ Jr&cFtjfd}|S)NcZt|tr|j}||g|Ri|}|Sr)r TextBlockr)rotherr r!r"r#s r$r%zsupport_textblock..wrapIsB eY ' ' KEd4000000 r&r'r*s` r$support_textblockr<Hs8_T Kr&cbeZdZdZdZgdZddZedZedZ edZ ed Z ed Z ed Z d Zed ZedZeidfdZeddedefdZeddedefdZddZdZdZdZdZdZdS) 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||ks Jd||_||_|dvsJd|d||_|pd|_|pd|_dS)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__isp|||S||| z!!!#U4#U#U#U!!! */a(-Ar&cD|jdkr|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) 9  % %8dj( (r&cD|jdkr|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) 9  $ $8dj( (r&cr|jdkr|jd|j|jf}nd|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.coordinatessB 9  j!TXt/ABFFT%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:::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 TX%,,r&c |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{TZ''r&c*t|tjr|jdd\}}nPt|tr|j|j}}n,t|tjr |j\}}nt| ||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_canvass fbj ) ) &<#DAqq  0 1 1 &=&,qAA  , , &;DAqq% %xxaax888r&c<t|trR|j|jkr3|j}||j|z|j|z|jSt |St|tr:|| |St|tr:|| |Std|jdNInvalid input type for other)rr>rCrA __class__rBr Rectanglerg to_rectangle condition_on Quadrilateralto_quadrilateral Exceptionrr;ds r$rozInterval.condition_ons eX & & OzTY&&K~~dj1ndhlDINNNDzz! y ) ) O%%e,,99;;HHOO O } - - O%%e,,==??LLUSS SM%/MMMNN Nr&c<t|trR|j|jkr3|j}||j|z |j|z |jSt |St|tr:|| |St|tr:|| |Std|jdri)rr>rCrArlrBr rmrgrn relative_torprqrrrss r$rvzInterval.relative_tos eX & & OzTY&&K~~dj1ndhlDINNNDzz! y ) ) O%%e,,99;;GGNN N } - - O%%e,,==??KKERR RM%/MMMNN Nr&Fc|jdi|}t|tr^|j|jkrdS|s+|j|jcxko|jcxko |jkncS|j|jcxko |jkncSt|tst|tr|j \}}}}|r5|jdkr||jcxko|kncS||jcxko|kncS|jdkr!||jcxko|jcxko|kncS||jcxko|jcxko|kncStd|j d)NFrGrjrk) padrr>rCrArBrYrmrprQrrrl)rr; soft_marginrYx_1y_1x_2y_2s r$is_inzInterval.is_in s ((K(( eX & & OyEJ&&uC ;$*MMMMMMMMEIMMMMM ;$+BBBBBBBBB y ) ) OZ}-M-M O!&!2 Cc3 @9##$+444444444$+4444444449##$*????????C?????$*????????C?????M%/MMMNN Nr&Tr;strictc &t|tr|j|jkrb|jdkr1|jdkr&t|j|j|j|jSt|j|j|j|jS|t|j|jt|j|j|j|j |j St|tr|j \}}}}|jdkr8tt||j|t||j|S|jdkr8t|t||j|t||jSd St|tr]|rtdtjd|jdtd||St%d|jd) 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 & &" OyEJ&&9## c(9(9$TZdh RRR$U[$*eiRRR~~ EK00%),,I&% y ) ) O!&!2 Cc3yC S$*!5!5sCTXrCrrlrrArrBrDrErmrQrprrrunionrnrrrs r$rzInterval.unionSs eX & &! OyEJ&&'O~~ EK00%),,I&% y ) ) O!&!2 Cc3yC S$*!5!5sCTX!;L  M`PTPY```    ^+h'xxex---r&ct|tr8|jdkr|dn|d}tjd|jd|j|z}|j|z}|||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.scales lH - - .2i3.>.><??LQROL M^di^^^    \)h%xxex---r&c||j\}}}}|t|t|t|t|fSr)rgrQintrimager{r|r}r~s r$ crop_imagezInterval.crop_imagesS!//66BS#sSXXC(#c((SXX*==>>r&ct|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>Ss6" EIIII . . . . ) )X ) ) )X )X" ; ;X ;--X-((X(9994OOO*OOO*')%OOOO<%O%O/%O%O%O%O%ON#O#O+#OT#O#O#O#OJ....,.......*???,,,*****r&r>c^eZdZdZdZgdZdZedZedZ edZ edZ ed Z ed Z ed Zed Zeid fdZeddedefdZeddedefdZddZddZd dZdZdZdZdS)!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>||_||_||_||_dSrr)rr{r|r}r~s r$rIzRectangle.__init__s"r&c |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.heightx$(""r&c |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&c6|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$(DHdh77r&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&cJ|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.centers+48#s*TX-@C,GGGr&c |j|jzS)z3 Return the area of the rectangle. )rNrLrKs r$r[zRectangle.area*s zDK''r&c^t|trV|jdkr |jd}}n d|j}}||j|z|j|z|j|z|j|zSt|trD|j \}}}}||j|z|j|z|j|z|j|zSt|tr=t|j |jd}|||j|jSt#d|jdNrGrTis_invrjrkrr>rCrArlr{r|r}r~rmrQrprperspective_matrixrTrLrNrrrr;dxdy_transformed_pointss r$rozRectangle.condition_on1s@ eX & & OzS  aBEKB>>2 tx"}dhmTX] y ) ) O ,LBAq>>2 tx"}dhmTX] } - - O!;($+d""" ??#5t{DJOO OM%/MMMNN Nr&c^t|trV|jdkr |jd}}n d|j}}||j|z |j|z |j|z |j|z St|trD|j \}}}}||j|z |j|z |j|z |j|z St|tr=t|j |jd}|||j|jSt#d|jdNrGrFrrjrkrrs r$rvzRectangle.relative_toOs> eX & & OzS  aBEKB>>2 tx"}dhmTX] y ) ) O ,LBAq>>2 tx"}dhmTX] } - - O!;($+e""" ??#5t{DJOO OM%/MMMNN Nr&Fc8jd i|ttr|sIjdkr|j|j}}n|j|j}}j|cxko|cxko j kncSjdkr |j dn |j d}j|cxko j kncSttrZ d} d}| ||o| ||SttrR|s"fd|jD} t!| St#j|j }t'|jSt)djd ) NrGrrrCrHrYc:g|]}t|jSrxrrT.0verticer;s r$ z#Rectangle.is_in..3!!!BI&w ==!!!r&rjrkrx)ryrr>rCr{r}r|r~rArBrYrm to_intervalrrprTallr^arrayrrrrl rr;rzrYrArBc x_interval y_interval is_vertice_ins ` r$rzRectangle.is_inls ((K(( eX & & O 5:$$!%483EE!%483E{e????s????ei?????&+jC&7&7DKNNT[^{a44445944444 y ) ) O***44J***44J::j:88TZZ6>H>> } - - O @!!!!MQ[!!! =)))$+..)&%,???M%/MMMNN Nr&Tr;rc t|tr||St|trx|t |j|jt |j|jt|j |j t|j |j St|tr]|rtdtjd|jdtd||St!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.intersectsJ eX & & O??4(( ( y ) ) O>>DHei((DHei((DHei((DHei((  } - - O <,[ QuQQfoQQQ~~e&8&8&:&:;;;M%/MMMNN Nr&c t|tr||St|trx|t |j|jt |j|jt|j |j t|j |j St|tr]|rtdtjd|jdtd||St#d|jd)rrrrrrjrk)rr>rrmrlrr{r|rr}r~rprrrrrnrrrs r$rzRectangle.unionsJ eX & & O??4(( ( y ) ) O>>DHei((DHei((DHei((DHei((  } - - O 8,[ QuQQfoQQQzz%"4"4"6"6777M%/MMMNN Nr&rc|j|z }|j|z }|j|z}|j|z} |r t d|}t d|}||||| S)Nr)r{r|r}r~rrl) rrrrrrr{r|r}r~s r$ryz Rectangle.padsjhohnhh  a++Ca++C~~c3S111r&ct|ts|}|}n"t|dks Jd|\}}|j|z}|j|z}|j|z}|j|z}|||||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.(33 .$G$GGN##q(((c)((- GWh h h h ~~c3S111r&rct|ts|}|}n"t|dks Jd|\}}|j|z}|j|z}|j|z}|j|z}|||||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,11 ,"G"GGL!!Q&&&a'&&+ GWh h h h ~~c3S111r&c|j\}}}}|t|t|t|t|fSr)rQrrs r$rzRectangle.crop_imagesE!-S#sSXXC(#c((SXX*==>>r&c h|dkr|j|j}}n|j|j}}t ||fd|i|SNrGrC)r{r}r|r~r>)rrCr!rArBs r$rzRectangle.to_intervalsD 3;;483EE483Es888888r&c*t|jSrrrKs r$rqzRectangle.to_quadrilateralsT[)))r&Nrrrr)rrrr1rrrIrrLrNrQrTrYr[r<rorvrrrrrryrrrrrqrxr&r$rmrms.( E,,,I##X###X#88X8 ; ;X ;HHXH((X( OOO:OOO8')%$O$O$O$OLOO/OOOOO:OO+OTOOOO6 2 2 2 22222"2222"???999*****r&rmceZdZdZdZgdZ d%deeje e e ffdZ e dZ e dZ e d Ze d Ze d Ze d Ze d Ze dZdZedZedZeidfdZed&dedefdZed&dedefdZd'dZd(dZd)dZdZ dZ!d Z"d!Z#d"Z$d#e%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)rTrLrNNrTcTt|tjr$|jdkrt d|jdnt|t rt |dkr)tj|dd}nt |dkr0t|dt rtj|}n@t dt |d t d t|d ||_ ||_ ||_ dS) 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__,s0 fbj ) ) |v%% !I&,!I!I!IJJJ&  % % 6{{a&))11!Q77V!!jD&A&A!&)) SF SSS?f???     r&c|j|jS|jdddf|jdddfz 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.heightFsU < #< {111a4 $$&&QQQT):)>)>)@)@@@r&c|j|jS|jdddf|jdddfz 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.widthSsU ; "; {111a4 $$&&QQQT):)>)>)@)@@@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)555r&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&cvt|jdS)z Calculate the center of the quadrilateral. Returns: :obj:`Tuple(numeric)`: Returns of coordinate of the center. rr)tuplerTmeantolistrKs r$rYzQuadrilateral.centerxs1T[%%1%--4466777r&c^t|jdddf|jdddfS)z7 Return the area of the quadrilateral. Nrr)rrTrKs r$r[zQuadrilateral.areas2 DK1-t{111a4/@AAAr&crdd|j|jd}dd|j|jd}|||S)Nrrrr])rNrLmap_to_points_ordering)rx_mapy_maps r$mapped_rectangle_pointsz%Quadrilateral.mapped_rectangle_pointssD! tz::!  <<**5%888r&ct|jd|jdS)Nfloat32)_getPerspectiveTransformrTastyper rKs r$rz Quadrilateral.perspective_matrixs;' K  y ) )  ( / / : :   r&c>|jdd}tjtj|j|dddftj|j|dddfgjS)Nrrr)rTargsortr^vstack vectorizegetT)rrr points_orderings r$rz$Quadrilateral.map_to_points_orderings+--1-55==1=EEy' UY''1(=>>' UY''1(=>>    r&ct|trC|jdkr||jdgS|d|jgSt|t r!||j|jgSt|tr=t|j |j d}| ||j |jStd|j drrr>rCrrArmr{r|rprrrTrlrLrNrrrr;rs r$rozQuadrilateral.condition_ons eX & & OzS  zz5;"2333zz1ek"2333 y ) ) O::uy%)455 5 } - - O!;($+d""" >>"4dk4:NN NM%/MMMNN Nr&ct|trE|jdkr||j dgS|d|j gSt|t r#||j |j gSt|tr=t|j |j d}| ||j |jStd|j drrrs r$rvzQuadrilateral.relative_tos eX & & OzS  zzEK<"3444zz1u{l"3444 y ) ) O:: zEI:677 7 } - - O!;($+e""" >>"4dk4:NN NM%/MMMNN Nr&Fchjd i|ttr|sajdkr|jd|jd}}n|jd|jd}}j|cxko|cxko jkncSjdkr |jdn |jd}j|cxko jkncSttrZ d} d}| ||o| ||SttrR|s"fd |j D} t| Stj|j}t!|j St#d jd ) NrGrr]rrrrHrc:g|]}t|jSrxrrs r$rz'Quadrilateral.is_in..rr&rjrkrx)ryrr>rCrQrArBrYrmrrrprTrr^rrrrrlrs ` r$rzQuadrilateral.is_ins ((K(( eX & & O 5:$$!%!1!!4d6Fq6I3EE!%!1!!4d6Fq6I3E{e????s????ei?????&+jC&7&7DKNNT[^{a44445944444 y ) ) O***44J***44J::j:88TZZ6>H>> } - - O @!!!!MQ[!!! =)))$+..)&%,???M%/MMMNN Nr&Tr;rc|rtdt|tst|trDt jdtd||St|trVt jdtd||Std|j draThe intersection between a Quadrilateral and other objects 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.zKWith `strict=False`, the current Quadrilateral object will be converted to rzKWith `strict=False`, both input Quadrilateral objects will be converted to rjrk) rrr>rmrrrrnrprrrlrs r$rzQuadrilateral.intersects'  S(Y %** Sj .J.J S MbkMMMt'8'8':':;;;E=11 S MbkMMM((**44U5G5G5I5IJJJ Qeo Q Q QRRRr&c|rtdt|tst|trDt jdtd||St|trVt jdtd||Std|j dr) rrr>rmrrrrnrprrrlrs r$rzQuadrilateral.unions'  S(Y %** Sj .J.J S MbkMMM{{4#4#4#6#6777E=11 S MbkMMM((**001C1C1E1EFFF Qeo Q Q QRRRr&rc| | ||d}| | ||d}|||}|j|z} |rtj| d} || S)NrrrT)rrTr^maximumrc) rrrrrrrr  padding_matrTs r$ryzQuadrilateral.pad'szEte%88DcTf8811%?? {*  +Z**Fxxvx&&&r&ct|ts||g}nt|dks Jd|}|jt j|z}||S)Nr]rr rr rrTr^rrc)rr shift_matrTs r$rzQuadrilateral.shift4su.(33 ''8IIN##q(((c)((&Irx 222xxvx&&&r&rct|ts||g}nt|dks Jd|}|jt j|z}||S)Nr]rr r$)rr scale_matrTs r$rzQuadrilateral.scaleBsu,11 %%|4IIL!!Q&&&a'&&$Irx 222xxvx&&&r&czt||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_imagePs7 4*S__c$+>N>N,O   r&c X|j\}}}}|dkr||}}n||}}t||fd|i|Sr)rQr>) rrCr!r{r|r}r~rArBs r$rzQuadrilateral.to_interval_sM!-S#s 3;;c3EEc3Es888888r&ct|jSrrrKs r$rnzQuadrilateral.to_rectangleis$*++r&c|j|jurdStj|j|jS)NF)rlr^iscloserTrrr;s r$__eq__zQuadrilateral.__eq__ls8 ?$. 0 05z$+u|4488:::r&crgd}dfd|D}jjd|dS)N)rTrNrLz, c:g|]}|dt|S)=)r4)rkeyrs r$rz*Quadrilateral.__repr__..ss1LLL;;wtS'9'9;;LLLr&())joinrlr)rkeysinfo_strs` r$__repr__zQuadrilateral.__repr__qsP,,,99LLLLtLLLMM.)77H7777r&returnct}|dd|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$r?zQuadrilateral.to_dictvsD"ww  h//33::<<X r&rrrrr)*rrrr1rrrr^r_rrIrrLrNrQrTrYr[r rrr<rorvrrrrrryrrrrrnr0r:rstrrr? __classcell__)rls@r$rprp s6 E---IOSBJd4j894 A AX A A AX A 6 6X 6  X 88X8BBXB 99X9  X    OOO.OOO.')%$O$O$O$OLSS/SSSSS*SS+STSSSS( ' ' ' ' ' ' ' ' ' ' ' '    999,,,;;; 888 c3hr&rpci|] }|j| Srxr)reles r$ rFsNNNSYNNNr&c$i|] \}}|j|SrxrD)ridxrEs r$rFrFs-sCCIsr&ceZdZdZdZgdZ d!dZedZedZ edZ ed Z ed Z e d Ze d Zid fdZe d"dedefdZe d"dedefdZe dZe d#dZe dZdZd$deefdZdZdZdeeeffdZ e!deeefddfd Z"dS)%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||_||_||_||_||_||_||_dSr) rrrrKrLrr7rMrN)rrrKrLrr7rMrNs r$rIzTextBlock.__init__sP%!122222      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.heightsz  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.widthszr&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.coordinatessz%%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.pointssz  r&c|jjS)z6 Return the area of associated block. )rr[rKs r$r[zTextBlock.areas zr&c6|j|Sr)rror/s r$rozTextBlock.condition_onsz&&u---r&c6|j|Sr)rrvr/s r$rvzTextBlock.relative_tosz%%e,,,r&Fc:|j|||Sr)rr)rr;rzrYs r$rzTextBlock.is_insz{F;;;r&Tr;rc:|j||SN)r)rrrs r$rzTextBlock.unionszf555r&c:|j||SrY)rrrs r$rzTextBlock.intersectsz##E&#999r&c6|j|Sr)rr)rrs r$rzTextBlock.shiftsz///r&rc>|j|||||Sr)rry)rrrrrrs r$ryz TextBlock.padsz~~dE3 BBBr&c6|j|Sr)rr)rrs r$rzTextBlock.scale sz ---r&c6|j|Sr)rrr*s r$rzTextBlock.crop_imagesz$$U+++r&rCc t|jtr|S|stddd||jjdd|i|S)Nz#Please provide valid `axis` values rGrHz as the inputrCrrx)rrr>rrcr)rrCr!s r$rzTextBlock.to_intervalsu dj( + + OK  S#SSS88"8$*"8"M"Md"Mf"M"M8NN Nr&ct|jtr|S||jSNr`)rrrmrcrnrKs r$rnzTextBlock.to_rectangles= dj) , , =K88$*"9"9";";8<< dj- 0 0 AK88$*"="="?"?8@@ @r&r;c|j}|jD]'}t||}|t||||<(|S)a Generate a dictionary representation of the current textblock of the format:: { "block_type": , } )rr?rr4)r base_dictfvals r$r?zTextBlock.to_dict(sVJ&&((  0 0A$""C&tQ// ! r&r@cdtvsJddtd}||fifd|jDS)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 c>i|]}||dSr)r)rrfr@s r$rFz'TextBlock.from_dict..Hs)IIIaQD 1 1IIIr&)BASECOORD_ELEMENT_NAMEMAP from_dictr)r5r@rs ` r$rlzTextBlock.from_dict9s  "; ; ; ; 5l!3 5 5 < ; ;*$|*<=GGMMs5JJIIII3=IIIJJJr&)NNNNNNrrr)#rrrr1rrrIrrLrNrQrTr[r+rorvrrrrrrryrrrrArrnrqrrr? classmethodrlrxr&r$r:r:s* EAAAISW    !!X!  X &&X& ! !X !X ...---(*%<<<<66+6T6666::/:::::000CCCC...,,,OO OOOO=== AAA c3h"KT#s(^K KKK[KKKr&r:r)1typingrrrrrrcollections.abcr r inspectr r rr(numpyr^pandaspdPILr cv2rr rr)baserrutilsrrrrrerrorsrrr+r0r<r>rmrpALL_BASECOORD_ELEMENTSrk enumerateBASECOORD_ELEMENT_INDEXMAPr:rxr&r$r|s A@@@@@@@@@@@@@@@@@$$$$$$********CCCCCC33333355555555>=======   t*{*{*{*{*{*{*{*{*| v*v*v*v*v* v*v*v*r |||||$|||~ #I}=NN7MNNN#,9-C#D#D /000sKsKsKsKsK!sKsK10sKsKsKr&