Ng%rddlmZmZmZmZddlmZmZddlmZGddZGddeeZ dS) )ListDictrAny)ABCabstractmethod)copyc"eZdZddZdZdZdS)BaseLayoutElementFc |r|nt|}t|}|D]0\}}||vr|||<d||vr ||d|<td||S)N_zUnknown attribute name: )rvarsitems ValueError)selfinplacekwargsobjvar_dictkeyvals V/var/www/html/ai-engine/env/lib/python3.11/site-packages/layoutparser/elements/base.pysetzBaseLayoutElement.sets-dd4::99  C CHCh # Sh&&&)S## !AC!A!ABBB cddt|D}|jjd|dS)Nz, c"g|] \}}|d| S)=).0rrs r z.BaseLayoutElement.__repr__..$s&PPPcnnsnnPPPr())joinr r __class____name__)rinfo_strs r__repr__zBaseLayoutElement.__repr__"sP99PPT$ZZ=M=M=O=OPPPQQ.)77H7777rcb|j|jurdSt|t|kS)NF)r#r rothers r__eq__zBaseLayoutElement.__eq__'s- ?$. 0 05DzzT%[[((rN)F)r$ __module__ __qualname__rr&r*rrrr r sF    888 )))))rr cfeZdZeedefdZeedeefdZeedZ eedZ eedZ eedZ eedZ ed Zed Zeid fd ZeddddefdZeddddefdZeddZeddZeddZedZdeeeffdZedeeefddfdZdS) BaseCoordElementreturncdS)zThe name of the classNrrs r_namezBaseCoordElement._name0  rcdS)z?A list of features names used for initializing the class objectNrr1s r _featureszBaseCoordElement._features6r3rcdSNrr1s rwidthzBaseCoordElement.width@  rcdSr7rr1s rheightzBaseCoordElement.heightEr9rcdSr7rr1s r coordinateszBaseCoordElement.coordinatesJr9rcdSr7rr1s rpointszBaseCoordElement.pointsOr9rcdSr7rr1s rareazBaseCoordElement.areaTr9rcdS)aY Given the current element in relative coordinates to another element which is in absolute coordinates, generate a new element of the current element in absolute coordinates. Args: other (:obj:`BaseCoordElement`): The other layout element involved in the geometric operations. Raises: Exception: Raise error when the input type of the other element is invalid. Returns: :obj:`BaseCoordElement`: The BaseCoordElement object of the original element in the absolute coordinate system. Nrr(s r condition_onzBaseCoordElement.condition_on] $ rcdS)aC Given the current element and another element both in absolute coordinates, generate a new element of the current element in relative coordinates to the other element. Args: other (:obj:`BaseCoordElement`): The other layout element involved in the geometric operations. Raises: Exception: Raise error when the input type of the other element is invalid. Returns: :obj:`BaseCoordElement`: The BaseCoordElement object of the original element in the relative coordinate system. Nrr(s r relative_tozBaseCoordElement.relative_toq " rFcdS)a Identify whether the current element is within another element. Args: other (:obj:`BaseCoordElement`): The other layout element involved in the geometric operations. soft_margin (:obj:`dict`, `optional`, defaults to `{}`): Enlarge the other element with wider margins to relax the restrictions. center (:obj:`bool`, `optional`, defaults to `False`): The toggle to determine whether the center (instead of the four corners) of the current element is in the other element. Returns: :obj:`bool`: Returns `True` if the current element is in the other element and `False` if not. Nr)rr) soft_margincenters ris_inzBaseCoordElement.is_inrDrTr)strictcdS)zIntersect the current shape with the other object, with operations defined in :doc:`../notes/shape_operations`. Nrrr)rLs r intersectzBaseCoordElement.intersectrcdS)z|Union the current shape with the other object, with operations defined in :doc:`../notes/shape_operations`. NrrNs runionzBaseCoordElement.unionrPrrcdS)aPad the layout element 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:`BaseCoordElement`: The padded BaseCoordElement object. Nr)rleftrighttopbottom safe_modes rpadzBaseCoordElement.padrGrcdS)a Shift the layout element 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:`BaseCoordElement`: The shifted BaseCoordElement of the same shape-specific class. Nr)rshift_distances rshiftzBaseCoordElement.shifts  rcdS)a Scale the 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:`BaseCoordElement`: The scaled BaseCoordElement of the same shape-specific class. Nr)r scale_factors rscalezBaseCoordElement.scales  rcdS)z Crop the input image according to the coordinates of the element. Args: image (:obj:`Numpy array`): The array of the input image. Returns: :obj:`Numpy array`: The array of the cropped image. Nr)rimages r crop_imagezBaseCoordElement.crop_images  rcBfdjD}j|d<|S)z Generate a dictionary representation of the current object: { "block_type": <"interval", "rectangle", "quadrilateral"> , "non_empty_block_attr1": value1, ... } cRi|]#}t||t|$Sr7)getattr)rrrs r z,BaseCoordElement.to_dict..s?   tS!!- s##---r block_type)r5r2)rdatas` rto_dictzBaseCoordElement.to_dicts@    ~    "Z\ rric~|jdksJdd|difd|jDS)zInitialize an instance based on the dictionary representation Args: data (:obj:`dict`): The dictionary representation of the object rhzIncompatible block types c"i|] }|| Srr)rfris rrgz.BaseCoordElement.from_dict..s888Qaa888rr)r2r5)clsris `r from_dictzBaseCoordElement.from_dictsd Il+ + + + ;tL'9 ; ; , + +s998888#-888999rN)T)rrrrT)r)r])r$r+r,propertyrstrr2rr5r8r;r=r?rArCrFrKboolrOrRrYr\r`rcrrrj classmethodrorrrr.r./s  s   ^X  49   ^X   ^X   ^X   ^X   ^X   ^X   ^ &  ^ $')%   ^ .  1 4   ^   - t   ^    ^ $    ^      ^  $   ^  "c3h$ :T#s(^ :0B : : :[ : : :rr.N) typingrrrabcrrrr r.rrrrvs)(((((((((((########))))))))8c:c:c:c:c:s-c:c:c:c:c:r