Ng2ddlmZmZmZmZmZmZddlmZmZddl m Z ddl Z ddl m Z mZddlmZmZmZmZmZmZmZGdd eZdS) )ListUnionDictrAnyOptional)MutableSequenceIterable)copyN)BaseCoordElementBaseLayoutElement)Interval Rectangle Quadrilateral TextBlockALL_BASECOORD_ELEMENTSBASECOORD_ELEMENT_NAMEMAPBASECOORD_ELEMENT_INDEXMAPc"eZdZdZd#dddeedefdZdZdZ d Z d Z d Z d Z d ZdZdZdZdZdZidfdZd$dedfdZidfdZdZd%dZdZdZdZdZdeeeffd Zdee fd!Z!d&de"j#fd"Z$dS)'LayoutaQ The :obj:`Layout` class id designed for processing a list of layout elements on a page. It stores the layout elements in a list and the related `page_data`, and provides handy APIs for processing all the layout elements in batch. ` Args: blocks (:obj:`list`): A list of layout element blocks page_data (Dict, optional): A dictionary storing the page (canvas) related information like `height`, `width`, etc. It should be passed in as a keyword argument to avoid any confusion. Defaults to None. N page_datablocksrc|Mt|tr|jjdks(|jjdkrd}nd|d}t |t|t rt |}||ng|_|pi|_dS)NrzUPlease check the input: it should be lp.Layout([layout]) instead of lp.Layout(layout)zHBlocks should be a list of layout elements or empty (None), instead got z. ) isinstancer __class____name__ ValueErrortuplelist_blocksr)selfrr error_msgs X/var/www/html/ai-engine/env/lib/python3.11/site-packages/layoutparser/elements/layout.py__init__zLayout.__init__1s^68,,171A1Jh1V1V(H44t rgmrrr Y'' ' fe $ $ "&\\F!'!3vv "bc|j|}t|tr'||j||jS|SNr)r!rslicerr)r"keyrs r$ __getitem__zLayout.__getitem__DsDc" c5 ! ! >>$,s"3t~>NN NMr&c||j|<dSNr!)r"r*newvalues r$ __setitem__zLayout.__setitem__Ks$ Sr&c|j|=dSr-r.)r"r*s r$ __delitem__zLayout.__delitem__Ns L   r&c*t|jSr-)lenr!r"s r$__len__zLayout.__len__Qs4<   r&c#&K|jD]}|VdSr-r.)r"eles r$__iter__zLayout.__iter__Ts,<  CIIII  r&cddt|D}|jjd|dS)Nz, c"g|] \}}|d| S)=).0r*vals r$ z#Layout.__repr__..Ys&PPPcnnsnnPPPr&())joinvarsitemsrr)r"info_strs r$__repr__zLayout.__repr__XsP99PPT$ZZ=M=M=O=OPPPQQ.)77H7777r&cpt|tr |j|jko|j|jkSdS)NF)rrr!rr"others r$__eq__z Layout.__eq__\s6 eV $ $ <5=0VT^u5V V5r&ct|tr|j|jkr)||j|jz|jS|jiks |jikr0||j|jz|jp|jSt d|jd|jdt|t r$||j|z|jSt d|jjd)Nrz(Incompatible page_data for two innputs: z vs .zInvalid input type for other )rrrrr!rr rrIs r$__add__zLayout.__add__bs$ eV $ $ ~00~~L5=0DN&2%%B)>)>~~L5=0"n?& !et~eeSXSbeeet $ $ >>$,"6$.>QQ QK0HKKK r&c<|j||dSr-)r!insert)r"r*values r$rPz Layout.insertxs  C'''''r&c^|t|j|jSr()rr r!rr5s r$r z Layout.copy{s$~~d4<00DN~KKKr&cT|fd|D|jS)Nc:g|]}|Sr=) relative_tor>r8rJs r$r@z&Layout.relative_to..s% 4 4 4S__U # # 4 4 4r&rrrrIs `r$rUzLayout.relative_to~s9~~ 4 4 4 4t 4 4 4   r&cT|fd|D|jS)Nc:g|]}|Sr=) condition_onrVs r$r@z'Layout.condition_on..s' 5 5 5S  e $ $ 5 5 5r&rrWrIs `r$rZzLayout.condition_ons9~~ 5 5 5 5 5 5 5   r&Fc\|fd|D|jS)Nc>g|]}|Sr=is_inr>r8centerrJ soft_margins r$r@z Layout.is_in..s) C C CsSYYuk6 2 2 C C Cr&rrWr"rJrar`s ```r$r^z Layout.is_insB~~ C C C C C Cd C C Cn   r&returnc|s1|t|j|||jS|j||dS)apSort the list of blocks based on the given Args: key ([type], optional): key specifies a function of one argument that is used to extract a comparison key from each list element. Defaults to None. reverse (bool, optional): reverse is a boolean value. If set to True, then the list elements are sorted as if each comparison were reversed. Defaults to False. inplace (bool, optional): whether to perform the sort inplace. If set to False, it will return another object instance with _block sorted in the order. Defaults to False. Examples:: >>> import layoutparser as lp >>> i = lp.Interval(4, 5, axis="y") >>> l = lp.Layout([i, i.shift(2)]) >>> l.sort(key=lambda x: x.coordinates[1], reverse=True) )r*reverserN)rsortedr!rsort)r"r*reinplaces r$rgz Layout.sortsb* 8>>t|g>>>$."  L  #w  7 7 7 7 7r&c\|fd|D|jS)a: Return a `Layout` object containing the elements that are in the `other` object. Args: other (:obj:`BaseCoordElement`): The block to filter the current elements. Returns: :obj:`Layout`: A new layout object after filtering. cBg|]}||Sr=r]r_s r$r@z$Layout.filter_by..s. J J JSCIIe[&$I$I JS J J Jr&rrWrbs ```r$ filter_byzLayout.filter_bysD~~ J J J J J JD J J Jn   r&cT|fd|D|jS)a Shift all layout elements by user specified amounts on x and y axis respectively. If shift_distance is one numeric value, the element will by shifted by the same specified amount on both x and y axis. Args: shift_distance (:obj:`numeric` or :obj:`Tuple(numeric)` or :obj:`List[numeric]`): The number of pixels used to shift the element. Returns: :obj:`Layout`: A new layout object with all the elements shifted in the specified values. c:g|]}|Sr=)shift)r>r8shift_distances r$r@z Layout.shift..s% 7 7 73SYY~ & & 7 7 7r&rrW)r"ros `r$rnz Layout.shifts;~~ 7 7 7 7$ 7 7 74>   r&rTcd|fd|D|jS)aPad all layout elements on the four sides of the polygon with the user-defined pixels. If safe_mode is set to True, the function will cut off the excess padding that falls on the negative side of the coordinates. Args: left (:obj:`int`, `optional`, defaults to 0): The number of pixels to pad on the upper side of the polygon. right (:obj:`int`, `optional`, defaults to 0): The number of pixels to pad on the lower side of the polygon. top (:obj:`int`, `optional`, defaults to 0): The number of pixels to pad on the left side of the polygon. bottom (:obj:`int`, `optional`, defaults to 0): The number of pixels to pad on the right side of the polygon. safe_mode (:obj:`bool`, `optional`, defaults to True): A bool value to toggle the safe_mode. Returns: :obj:`Layout`: A new layout object with all the elements padded in the specified values. c Bg|]}|Sr=)pad)r>r8bottomleftright safe_modetops r$r@zLayout.pad..s- J J JcSWWT5#vy 9 9 J J Jr&rrW)r"rtrurwrsrvs `````r$rrz Layout.padsL ~~ J J J J J J J JT J J Jn   r&cT|fd|D|jS)a Scale all layout element by a user specified amount on x and y axis respectively. If scale_factor is one numeric value, the element will by scaled by the same specified amount on both x and y axis. Args: scale_factor (:obj:`numeric` or :obj:`Tuple(numeric)` or :obj:`List[numeric]`): The amount for downscaling or upscaling the element. Returns: :obj:`Layout`: A new layout object with all the elements scaled in the specified values. c:g|]}|Sr=)scale)r>r8 scale_factors r$r@z Layout.scale..s% 5 5 5SYY| $ $ 5 5 5r&rrW)r"r{s `r$rzz Layout.scales;~~ 5 5 5 5 5 5 5   r&c fd|DS)Nc:g|]}|Sr=) crop_image)r>r8images r$r@z%Layout.crop_image..s%666#u%%666r&r=)r"rs `r$r~zLayout.crop_images66666666r&cd|DS)z Iterate through all the text blocks in the list and append their ocr'ed text results. Returns: :obj:`List[str]`: A list of text strings of the text blocks in the list of layout elements. c<g|]}t|d|jS)text)hasattrrr>r8s r$r@z$Layout.get_texts..s)AAASGC,@,@AAAAr&r=r5s r$ get_textszLayout.get_textssBADAAAAr&c fd|DS)auGiven user-provided attribute name, check all the elements in the list and return the corresponding attribute values. Args: attr_name (:obj:`str`): The text string of certain attribute name. Returns: :obj:`List`: The list of the corresponding attribute value (if exist) of each element in the list. cPg|]"}t|t|#Sr=)rgetattr)r>r8 attr_names r$r@z#Layout.get_info.. s2SSSC73 ;R;RSY''SSSr&r=)r"rs `r$get_infozLayout.get_infos TSSS4SSSSr&c*|jd|DdS)zGenerate a dict representation of the layout object with the page_data and all the blocks in its dict representation. Returns: :obj:`Dict`: The dictionary representation of the layout object. c6g|]}|Sr=to_dictrs r$r@z"Layout.to_dict..s 7V7V7V# 7V7V7Vr&)rrrr5s r$rzLayout.to_dict s$"^7V7VQU7V7V7VWWWr&c8d}d}|D]C}t|tr d}|j}n|}t|t|j}Dt |j|rg}|D]}t|trEt|}|jjkr%t|jd|_n5|jkrt|d}t|}| |nfd|D}|S)a6Convert all elements into blocks of the same type based on the type casting rule:: Interval < Rectangle < Quadrilateral < TextBlock Returns: List[BaseLayoutElement]: A list of base layout elements of the maximal compatible type FTto_c`g|]*}|jkrt|dn|+S)r)_namer)r>r8target_coord_names r$r@z1Layout.get_homogeneous_blocks..>s]9 11186#46677999r&) rrblockmaxrrrr rappend)r" has_textblockmax_coord_levelr8r new_blocksrs @r$get_homogeneous_blockszLayout.get_homogeneous_blockssj   C#y))  $  !!;EK!HOO3?CI  J ' 'c9--+s))Cy*;;;$QGCI7P=N7P7P$Q$Q$S$S y$555Egc+D1B+D+DEEGG#E**C!!#&&&& ' Jr&ct|r|}n|}tjd|D}|S)aConvert the layout object into the dataframe. Warning: the page data won't be exported. Args: enforce_same_type (:obj:`bool`, optional): If true, it will convert all the contained blocks to the maximal compatible data type. Defaults to False. Returns: pd.DataFrame: The dataframe representation of layout object c6g|]}|Sr=rrs r$r@z'Layout.to_dataframe..Zs ;;;S3;;==;;;r&)rpd DataFrame)r"enforce_same_typerdfs r$ to_dataframezLayout.to_dataframeGsG  0022FFF \;;F;;; < < r&r-)NFF)rrrrT)F)%r __module__ __qualname____doc__rrrr%r+r0r2r6r9rGrKrNrPr rUrZr^rgrkrnrrrzr~rrstrrrr rrrrr=r&r$rr!s8  )4)))x~)D))))&%%%!!!888 ,(((LLL      (*%    88hx>P88888,.e    "   "    *   777BBB T T TXc3hXXXX0->(?0000dr|r&r)typingrrrrrcollections.abcrr r pandasrbaser r layout_elementsrrrrrrrrr=r&r$rs:9999999999999995555555555555555{{{{{_{{{{{r&