kh2ddlmZmZmZmZmZmZddlmZmZddl m Z ddl Z ddl m Z mZddlmZmZmZmZmZmZmZGdd eZy) )ListUnionDictrAnyOptional)MutableSequenceIterable)copyN)BaseCoordElementBaseLayoutElement)Interval Rectangle Quadrilateral TextBlockALL_BASECOORD_ELEMENTSBASECOORD_ELEMENT_NAMEMAPBASECOORD_ELEMENT_INDEXMAPceZdZdZd!dddeedefdZdZdZ d Z d Z d Z d Z d ZdZdZdZdZdZidfdZd"dedfdZidfdZdZd#dZdZdZdZdZdeeeffdZdee fdZ!d$de"jFfd Z$y)%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*|`t|tr|jjdk7s7|jjdk(r d}t |d|d}t |t|t r t |}||ng|_|xsi|_y)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/teggl/fontify/venv/lib/python3.12/site-packages/layoutparser/elements/layout.py__init__zLayout.__init__1s^68,1A1A1J1Jh1V((H4s Y' 'ggmfnnqr Y' ' fe $&\F!'!3v "bc|j|}t|tr*|j|j||jS|SNr)r!rslicerr)r"keyrs r$ __getitem__zLayout.__getitem__Ds?c" c5 !>>$,,s"3t~~>N NMr&c"||j|<yNr!)r"r*newvalues r$ __setitem__zLayout.__setitem__Ks$ Sr&c|j|=yr-r.)r"r*s r$ __delitem__zLayout.__delitem__Ns LL r&c,t|jSr-)lenr!r"s r$__len__zLayout.__len__Qs4<<  r&c#6K|jD]}|ywr-r.r"eles r$__iter__zLayout.__iter__Ts<< CI sc djt|jDcgc] \}}|d|c}}}|jjd|dScc}}w)Nz, =())joinvarsitemsrr)r"r*valinfo_strs r$__repr__zLayout.__repr__XsZ99T$Z=M=M=OPcQsenPQ..))*!H:Q77QsA ct|tr4|j|jk(xr|j|jk(Sy)NF)rrr!rr"others r$__eq__z Layout.__eq__\s4 eV $<<5==0VT^^u5V Vr&ct|tr|j|jk(r4|j|j|jz|jS|jik(s|jik(rB|j|j|jz|jxs |jSt d|jd|jdt|t r*|j|j|z|jSt d|jjd)Nrz(Incompatible page_data for two innputs: z vs .zInvalid input type for other )rrrrr!rr rrFs r$__add__zLayout.__add__bs eV $~~0~~LL5==0DNN&2%B)>~~LL5==0"nn?& !>t~~>NdSXSbSbRccdet $>>$,,"6$..>Q Q/0H0H/IK r&c<|jj||yr-)r!insert)r"r*values r$rMz Layout.insertxs C'r&cb|jt|j|jSr()rr r!rr5s r$r z Layout.copy{s"~~d4<<0DNN~KKr&c|j|Dcgc]}|j|c}|jScc}wr()r relative_torr"rGr9s r$rQzLayout.relative_to~s8~~/3 4S__U # 4  4;c|j|Dcgc]}|j|c}|jScc}wr()r condition_onrrRs r$rUzLayout.condition_ons:~~04 5S  e $ 5  5rSFc |j|Dcgc]}|j|||c}|jScc}wr(ris_inrr"rG soft_margincenterr9s r$rXz Layout.is_ins=~~>B CsSYYuk6 2 Cnn  Cs=returnc|s3|jt|j|||jS|jj ||y)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*r^inplaces r$r`z Layout.sortsL*>>t||g>$.."  LL  #w  7r&c |j|Dcgc]}|j|||s|c}|jScc}w)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. rrWrYs r$ filter_byzLayout.filter_bysA~~ JSCIIe[&$IS Jnn  Js AAc|j|Dcgc]}|j|c}|jScc}w)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. r)rshiftr)r"shift_distancer9s r$rez Layout.shifts:~~26 73SYY~ & 74>>  7rSc |j|Dcgc]}|j|||||c}|jScc}w)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. r)rpadr)r"leftrighttopbottom safe_moder9s r$rhz Layout.padsC ~~EI JcSWWT5#vy 9 Jnn  Js?c|j|Dcgc]}|j|c}|jScc}w)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. r)rscaler)r" scale_factorr9s r$roz Layout.scales:~~04 5SYY| $ 5  5rScJ|Dcgc]}|j|c}Scc}wr-) crop_image)r"imager9s r$rrzLayout.crop_images156#u%666s cZ|Dcgc]}t|ds|jc}Scc}w)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. text)hasattrrur8s r$ get_textszLayout.get_textss%%)ASGC,@AAA((cZ|Dcgc]}t||st||c}Scc}w)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. )rvgetattr)r" attr_namer9s r$get_infozLayout.get_infos(48SC73 ;RY'SSSrxcb|j|Dcgc]}|jc}dScc}w)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. )rr)rto_dictr8s r$r~zLayout.to_dict s("^^QU7V# 7VWW7Vs,c td}d}|D]@}t|trd}|j}n|}t|t|j }Bt |j }|rg}|D]}t|trHt|}|jj |k7rRt|jd||_n.|j |k7rt|d|}t}|j||S|Dcgc]'}|j |k7rt|d|n|)}}|Scc}w)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_) rrblockmaxr_namerr rzappend)r" has_textblockmax_coord_levelr9rtarget_coord_name new_blockss r$get_homogeneous_blockszLayout.get_homogeneous_blockssb  C#y) $  !!;EKK!HO 3?CII J 'c9-s)Cyy*;;$QGCII=NP88,.e " " * 7B TXc3hX0->(?0dr||r&r)typingrrrrrcollections.abcrr r pandasrbaser r layout_elementsrrrrrrrrrr&r$rs6:955{_{r&