o eR@sdZddlmZmZmZddlZddlZddl Z ddl Z ddl Z ddl mZddlmZddlmZmZmZmZmZGd d d eZGd d d eZdddZddZddZdS)zClasses representing lines.)print_functiondivisionabsolute_importN)imgaug) IAugmentable)normalize_shapeproject_coords_interpolate_points_remove_out_of_image_fraction__normalize_shift_argsc@seZdZdZdiddZeddZeddZed d Zed d Z ed dZ eddZ eddZ ddZ ddZdiddZdiddZdjddZddZd d!Zd"d#Zdkd%d&Zdkd'd(Zdld*d+Zd,d-Zd.d/Zdmd1d2Zdnd3d4Z 0 $dod6d7Z 8 ) $dpd9d:Z 8 $dqd;d<Z 8 ) $drd=d>Z ? @ ) $dsdAdBZ C @ $dtdDdEZ!    ) $dudFdGZ"  )dvdHdIZ#dJdKZ$dLdMZ%dNdOZ&dPdQZ'dRdSZ( 0 $dwdTdUZ) 0 $dodVdWZ*dxdYdZZ+dxd[d\Z,dyd]d^Z-dyd_d`Z.dadbZ/dcddZ0dedfZ1dgdhZ2dS)z LineStringaiClass representing line strings. A line string is a collection of connected line segments, each having a start and end point. Each point is given as its ``(x, y)`` absolute (sub-)pixel coordinates. The end point of each segment is also the start point of the next segment. The line string is not closed, i.e. start and end point are expected to differ and will not be connected in drawings. Parameters ---------- coords : iterable of tuple of number or ndarray The points of the line string. label : None or str, optional The label of the line string. NcCst|r|jjdkr|tj}n4t|dkr!tjdtjd}n%t |s/Jdt |ft dd|DsAJdt |ft|}|j d krR|jd d ksZJd |jf||_||_d S) z!Create a new LineString instance.float32rrrdtypez1Expected 'coords' to be an iterable, got type %s.cSsg|]}t|dkqS)rlen).0Zcoords_irID:\Projects\ConvertPro\env\Lib\site-packages\imgaug/augmentables/lines.py 9z'LineString.__init__..z2Expected 'coords' to contain (x,y) tuples, got %s.rz5Expected 'coords' to have shape (N, 2), got shape %s.N)iaZ is_np_arrayrnameastypenprrzeros is_iterabletypeallstrndimshapecoordslabelselfr&r'rrr__init__-s0       zLineString.__init__cCs t|jdkr dSt|S)zCompute the total euclidean length of the line string. Returns ------- float The length based on euclidean distance, i.e. the sum of the lengths of each line segment. r)rr&rsumcompute_neighbour_distancesr)rrrlengthEs zLineString.lengthcC|jdddfS)zGet an array of x-coordinates of all points of the line string. Returns ------- ndarray ``float32`` x-coordinates of the line string points. Nrr&r-rrrxxT z LineString.xxcCr/)zGet an array of y-coordinates of all points of the line string. Returns ------- ndarray ``float32`` y-coordinates of the line string points. Nrr0r-rrryy`r2z LineString.yycCt|jtjS)axGet an array of discrete x-coordinates of all points. The conversion from ``float32`` coordinates to ``int32`` is done by first rounding the coordinates to the closest integer and then removing everything after the decimal point. Returns ------- ndarray ``int32`` x-coordinates of the line string points. )rroundr1rint32r-rrrxx_intlzLineString.xx_intcCr4)axGet an array of discrete y-coordinates of all points. The conversion from ``float32`` coordinates to ``int32`` is done by first rounding the coordinates to the closest integer and then removing everything after the decimal point. Returns ------- ndarray ``int32`` y-coordinates of the line string points. )rr5r3rr6r-rrryy_int|r8zLineString.yy_intcC*t|jdkr dSt|jt|jS)aCompute the height of a bounding box encapsulating the line. The height is computed based on the two points with lowest and largest y-coordinates. Returns ------- float The height of the line string. rr)rr&rmaxr3minr-rrrheight zLineString.heightcCr:)aCompute the width of a bounding box encapsulating the line. The width is computed based on the two points with lowest and largest x-coordinates. Returns ------- float The width of the line string. rr)rr&rr;r1r<r-rrrwidthr>zLineString.widthcCspt|jdkrtjdtdSt|}|dd\}}td|jk|j|k}td|jk|j|k}t||S)aDetermine per point whether it is inside of a given image plane. Parameters ---------- image : ndarray or tuple of int Either an image with shape ``(H,W,[C])`` or a ``tuple`` denoting such an image shape. Returns ------- ndarray ``(N,) ``bool`` array with one value for each of the ``N`` points indicating whether it is inside of the provided image plane (``True``) or not (``False``). rrrr) rr&rrboolr logical_andr1r3)r)imager%r=r?Zx_withinZy_withinrrrget_pointwise_inside_image_masks z*LineString.get_pointwise_inside_image_maskcCs^t|jdkrtjdtjdSttj|jddddf|jddddfdddS)zCompute the euclidean distance between each two consecutive points. Returns ------- ndarray ``(N-1,)`` ``float32`` array of euclidean distances between point pairs. Same order as in `coords`. rr@rNrrZaxis)rr&rrrsqrtr+r-rrrr,s .z&LineString.compute_neighbour_distancescsddlddlm}t|rjjjfnOttrEt j dkr(|St j dkr=jj dddfn,jj n$tt r`t dksYJdt fjn t dt ffdd |j DS) aRCompute min distances between points of this and another line string. Parameters ---------- other : tuple of number or imgaug.augmentables.kps.Keypoint or imgaug.augmentables.LineString Other object to which to compute the distances. default : any Value to return if `other` contains no points. Returns ------- list of float or any For each coordinate of this line string, the distance to any closest location on `other`. `default` if no distance could be computed. rNrKeypointrz>Expected tuple 'other' to contain exactly two entries, got %d.z.)shapely.geometrykpsrH isinstancerIrJxyrrr&tuple ValueErrorr!)r)rMdefaultrHrrLrcompute_pointwise_distancess2     z&LineString.compute_pointwise_distancescCs&|j|gd}t|dkr|St|S)aCompute the minimal distance between the line string and `other`. Parameters ---------- other : tuple of number or imgaug.augmentables.kps.Keypoint or imgaug.augmentables.LineString Other object to which to compute the distance. default : any Value to return if this line string or `other` contain no points. Returns ------- float or any Minimal distance to `other` or `default` if no distance could be computed. rVr)rWrr<)r)rMrVZ distancesrrrcompute_distance s zLineString.compute_distance-C6?cCs|j|tjd|kS)aEstimate whether a point is on this line string. This method uses a maximum distance to estimate whether a point is on a line string. Parameters ---------- other : tuple of number or imgaug.augmentables.kps.Keypoint Point to check for. max_distance : float Maximum allowed euclidean distance between the point and the closest point on the line. If the threshold is exceeded, the point is not considered to fall on the line. Returns ------- bool ``True`` if the point is on the line string, ``False`` otherwise. rX)rYrinf)r)rM max_distancerrrcontains#szLineString.containscCst|j|||_|S)aProject the line string onto a differently shaped image in-place. E.g. if a point of the line string is on its original image at ``x=(10 of 100 pixels)`` and ``y=(20 of 100 pixels)`` and is projected onto a new image with size ``(width=200, height=200)``, its new position will be ``(x=20, y=40)``. This is intended for cases where the original image is resized. It cannot be used for more complex changes (e.g. padding, cropping). Added in 0.4.0. Parameters ---------- from_shape : tuple of int or ndarray Shape of the original image. (Before resize.) to_shape : tuple of int or ndarray Shape of the new image. (After resize.) Returns ------- imgaug.augmentables.lines.LineString Line string with new coordinates. The object may have been modified in-place. )r r&r)Z from_shapeZto_shaperrrproject_;szLineString.project_cCs|||S)aLProject the line string onto a differently shaped image. E.g. if a point of the line string is on its original image at ``x=(10 of 100 pixels)`` and ``y=(20 of 100 pixels)`` and is projected onto a new image with size ``(width=200, height=200)``, its new position will be ``(x=20, y=40)``. This is intended for cases where the original image is resized. It cannot be used for more complex changes (e.g. padding, cropping). Parameters ---------- from_shape : tuple of int or ndarray Shape of the original image. (Before resize.) to_shape : tuple of int or ndarray Shape of the new image. (After resize.) Returns ------- imgaug.augmentables.lines.LineString Line string with new coordinates. )deepcopyr_r^rrrprojectZszLineString.projectcCsj|j}|dkrt|jdkrdS||}t|rdSdS||}tdd|D}||}d|S)aCompute fraction of polygon area outside of the image plane. This estimates ``f = A_ooi / A``, where ``A_ooi`` is the area of the polygon that is outside of the image plane, while ``A`` is the total area of the bounding box. Added in 0.4.0. Parameters ---------- image : (H,W,...) ndarray or tuple of int Image dimensions to use. If an ``ndarray``, its shape will be used. If a ``tuple``, it is assumed to represent the image shape and must contain at least two integers. Returns ------- float Fraction of the polygon area that is outside of the image plane. Returns ``0.0`` if the polygon is fully inside of the image plane. If the polygon has an area of zero, the polygon is treated similarly to a :class:`LineString`, i.e. the fraction of the line that is inside the image plane is returned. r?cSg|]}|jqSr)r.rlsrrrrz.)r.rr&rDrr"clip_out_of_imager+)r)rCr.Z points_ooiZ lss_clippedZlength_after_clipZinside_image_factorrrrcompute_out_of_image_fractionus  z(LineString.compute_out_of_image_fractionFcCs"t|jdkr |St||S)aQEstimate whether the line string is fully inside an image plane. Parameters ---------- image : ndarray or tuple of int Either an image with shape ``(H,W,[C])`` or a ``tuple`` denoting such an image shape. default : any Default value to return if the line string contains no points. Returns ------- bool or any ``True`` if the line string is fully inside the image area. ``False`` otherwise. Will return `default` if this line string contains no points. r)rr&rr"rD)r)rCrVrrris_fully_within_imagesz LineString.is_fully_within_imagecCs<t|jdkr |S||}t|rdSt||dkS)ao Estimate whether the line string is at least partially inside the image. Parameters ---------- image : ndarray or tuple of int Either an image with shape ``(H,W,[C])`` or a ``tuple`` denoting such an image shape. default : any Default value to return if the line string contains no points. Returns ------- bool or any ``True`` if the line string is at least partially inside the image area. ``False`` otherwise. Will return `default` if this line string contains no points. rT)rr&rDranyrh)r)rCrVmaskrrris_partly_within_images   z!LineString.is_partly_within_imageTcCs2t|jdkr |S||rdS||r|S|S)a Estimate whether the line is partially/fully outside of the image area. Parameters ---------- image : ndarray or tuple of int Either an image with shape ``(H,W,[C])`` or a tuple denoting such an image shape. fully : bool, optional Whether to return ``True`` if the line string is fully outside of the image area. partly : bool, optional Whether to return ``True`` if the line string is at least partially outside fo the image area. default : any Default value to return if the line string contains no points. Returns ------- bool or any ``True`` if the line string is partially/fully outside of the image area, depending on defined parameters. ``False`` otherwise. Will return `default` if this line string contains no points. rF)rr&rjrm)r)rCfullypartlyrVrrris_out_of_images  zLineString.is_out_of_imagec s tjdkr gS|}|}tjdkr$t|sgSgSt|r.gSt|dd\}}d}td||dfgt||df||||fgt||||fd||fgtd||fdgg} |}g} t t jddjdd|dd|dd|} | D]3\} \} } }}}| | d |f|D] }| |d d fq| tjdk}|r|s| | d |fqg}g}t | D]\} \}}}|rt|dkr| |g}q|s| |q| t| dk}| ot | | ddd u}t|dkr|dnd}d}|s!|s!| | dd}d}d}|dur8tjt|t|}|durKtjt|t|}|dupS|d k}|dup\|d k}|rh|rh| |qt|dkru| |d d |D}fdd |DS)aClip off all parts of the line string that are outside of the image. Parameters ---------- image : ndarray or tuple of int Either an image with shape ``(H,W,[C])`` or a ``tuple`` denoting such an image shape. Returns ------- list of imgaug.augmentables.lines.LineString Line strings, clipped to the image shape. The result may contain any number of line strins, including zero. rrrgMbP?)rbrbrbNrFTg{Gz?cSsg|] }t|dkr|qSr@rrlinerrrrez0LineString.clip_out_of_image..csg|]}j|dqS)r0r`rqr-rrrfr)rr&rDrrkcopyr"r rfind_intersections_with enumeratezipappendrAlinalgnormr) r)rCZinside_image_maskZooi_maskr=r?epsedges intersectionsZpointsgeni line_startline_endZ ooi_startZooi_endZ inter_lineZp_interZis_lastlinesrrcoordZ was_addedZooiZ is_last_pointZ is_next_ooiZp_prevZp_nextZ dist_prevZ dist_nextZ dist_prev_okZ dist_next_okrr-rrhs                zLineString.clip_out_of_imagecCs,ddl}t|}g}t|jdd|jddD]z\}}|j||g}||}tt|}g} |D]O} t | |jj jrit | jdkrh| jdd| jddf} | jdd| jddf} | | | gq4t | |jj jszJdt| f| | j| jfq4t| |fddd} || q|S) a&Find all intersection points between this line string and `other`. Parameters ---------- other : tuple of number or list of tuple of number or list of LineString or LineString The other geometry to use during intersection tests. Returns ------- list of list of tuple of number All intersection points. One list per pair of consecutive start and end point, i.e. `N-1` lists of `N` points. Each list may be empty or may contain multiple points. rNrrzxExpected to find shapely.geometry.point.Point or shapely.geometry.linestring.LineString intersection, actually found %s.cSstjt||SNrrzr{r)pZpsrrrz4LineString.find_intersections_with..)key)rO _convert_var_to_shapely_geometryrxr&rIr intersectionlist_flatten_shapely_collectionrQZ linestringrextendrKrJr!ryrRrSsorted)r)rMrNgeomresultZp_startZp_endrfr~Zintersections_pointsinterZ inter_startZ inter_endZ inter_sortedrrrrvjs6&    z"LineString.find_intersections_withrcCs8|jdddf|7<|jdddf|7<|S)aMove this line string along the x/y-axis in-place. The origin ``(0, 0)`` is at the top left of the image. Added in 0.4.0. Parameters ---------- x : number, optional Value to be added to all x-coordinates. Positive values shift towards the right images. y : number, optional Value to be added to all y-coordinates. Positive values shift towards the bottom images. Returns ------- result : imgaug.augmentables.lines.LineString Shifted line string. The object may have been modified in-place. Nrrr0)r)rRrSrrrshift_szLineString.shift_cC*t||||||d\}}|j||dS)aMove this line string along the x/y-axis. The origin ``(0, 0)`` is at the top left of the image. Parameters ---------- x : number, optional Value to be added to all x-coordinates. Positive values shift towards the right images. y : number, optional Value to be added to all y-coordinates. Positive values shift towards the bottom images. top : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift this object *from* the top (towards the bottom). right : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift this object *from* the right (towards the left). bottom : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift this object *from* the bottom (towards the top). left : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift this object *from* the left (towards the right). Returns ------- result : imgaug.augmentables.lines.LineString Shifted line string. toprightbottomleftrRrSr r`rr)rRrSrrrrrrrshift)  zLineString.shiftrc Cs |j|dd||d|d}|dkS)aDraw this line segment as a binary image mask. Parameters ---------- image_shape : tuple of int The shape of the image onto which to draw the line mask. size_lines : int, optional Thickness of the line segments. size_points : int, optional Size of the points in pixels. raise_if_out_of_image : bool, optional Whether to raise an error if the line string is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. Returns ------- ndarray Boolean line mask of shape `image_shape` (no channel axis). rcF) alpha_lines alpha_points size_lines size_points antialiasedraise_if_out_of_image?)draw_heatmap_array)r) image_shaperrrheatmaprrr draw_maskszLineString.draw_maskrccCsdt|dkst|dkr|ddksJd|f|jtj|tjdd||||d}|tjd S) aDraw the line segments of this line string as a heatmap array. Parameters ---------- image_shape : tuple of int The shape of the image onto which to draw the line mask. alpha : float, optional Opacity of the line string. Higher values denote a more visible line string. size : int, optional Thickness of the line segments. antialiased : bool, optional Whether to draw the line with anti-aliasing activated. raise_if_out_of_image : bool, optional Whether to raise an error if the line string is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. Returns ------- ndarray ``float32`` array of shape `image_shape` (no channel axis) with drawn line string. All values are in the interval ``[0.0, 1.0]``. rrr1Expected (H,W) or (H,W,1) as image_shape, got %s.rcoloralphasizerro@)rdraw_lines_on_imagerruint8rr)r)rrrrrarrrrrdraw_lines_heatmap_array s z#LineString.draw_lines_heatmap_arraycCsbt|dkst|dkr|ddksJd|f|jtj|tjdd|||d}|tjd S) aDraw the points of this line string as a heatmap array. Parameters ---------- image_shape : tuple of int The shape of the image onto which to draw the point mask. alpha : float, optional Opacity of the line string points. Higher values denote a more visible points. size : int, optional Size of the points in pixels. raise_if_out_of_image : bool, optional Whether to raise an error if the line string is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. Returns ------- ndarray ``float32`` array of shape `image_shape` (no channel axis) with drawn line string points. All values are in the interval ``[0.0, 1.0]``. rrrrrrr)rrrrr)rdraw_points_on_imagerrrrr)r)rrrrrrrrdraw_points_heatmap_array9s z$LineString.draw_points_heatmap_arrayc CsN|j|||||d}|dkr|S|j||||d} t|| g} tj| ddS)a Draw the line segments and points of the line string as a heatmap array. Parameters ---------- image_shape : tuple of int The shape of the image onto which to draw the line mask. alpha_lines : float, optional Opacity of the line string. Higher values denote a more visible line string. alpha_points : float, optional Opacity of the line string points. Higher values denote a more visible points. size_lines : int, optional Thickness of the line segments. size_points : int, optional Size of the points in pixels. antialiased : bool, optional Whether to draw the line with anti-aliasing activated. raise_if_out_of_image : bool, optional Whether to raise an error if the line string is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. Returns ------- ndarray ``float32`` array of shape `image_shape` (no channel axis) with drawn line segments and points. All values are in the interval ``[0.0, 1.0]``. )rrrrr)rrrrrE)rrrZdstackr;) r)rrrrrrrZ heatmap_linesZheatmap_pointsrrrrrbs")zLineString.draw_heatmap_arrayrrrrcCsddlm}ddlm}d} t|trd} tj|tjd}|j dvs*Jd|j ft |j d ks9|d ks9|d kr>t |S|rS|j|ddd rStd ||j f|j dkrjt|sfJd t|f|g}n|j dkr|t|r||g|j d}|tj}|j dd\} } g} t|j dd|j d dD]\} }| | d | d|d |dfqtt| tj} t|}tj|j ddtjd}| D]T}|rtjj|\}}}n tjj|\}}d}td|k|| k}td|k|| k}t||}t|r&||}||}t|s||n|}|||||f<q|d kr._assert_not_nonerrrNrrrFr)rrrarrayrrr)r)rCr color_lines color_pointsrrrrrrrrrrrrrRs>F        zLineString.draw_on_imagecCsrddlm}|jdvsJd|jft|jdks|dkr>|r/tjd|jdd|jd Stjd |jdd|jd S|j }|j } |dd} t || } t | | } t |d| } t | d| }|| | | |d }t|jdkr|j ||||d S|j|jddd ||d}|jdkrt|}|tj|}|j ||||d }tt|ddtjS)aExtract all image pixels covered by the line string. This will only extract pixels overlapping with the line string. As a rectangular image array has to be returned, non-overlapping pixels will be set to zero. This function will by default zero-pad the image if the line string is partially/fully outside of the image. This is for consistency with the same methods for bounding boxes and polygons. Parameters ---------- image : ndarray The image of shape `(H,W,[C])` from which to extract the pixels within the line string. size : int, optional Thickness of the line. pad : bool, optional Whether to zero-pad the image if the object is partially/fully outside of it. pad_max : None or int, optional The maximum number of pixels that may be zero-paded on any side, i.e. if this has value ``N`` the total maximum of added pixels is ``4*N``. This option exists to prevent extremely large images as a result of single points being moved very far away during augmentation. antialiased : bool, optional Whether to apply anti-aliasing to the line string. prevent_zero_size : bool, optional Whether to prevent height or width of the extracted image from becoming zero. If this is set to ``True`` and height or width of the line string is below ``1``, the height/width will be increased to ``1``. This can be useful to prevent problems, e.g. with image saving or plotting. If it is set to ``False``, images will be returned as ``(H', W')`` or ``(H', W', 3)`` with ``H`` or ``W`` potentially being ``0``. Returns ------- (H',W') ndarray or (H',W',C) ndarray Pixels overlapping with the line string. Zero-padded if the line string is partially/fully outside of the image and ``pad=True``. If `prevent_zero_size` is activated, it is guarantueed that ``H'>0`` and ``W'>0``, otherwise only ``H'>=0`` and ``W'>=0``. r BoundingBoxrz0Expected image of shape (H,W,[C]), got shape %s.r)rrrNrrr)x1y1x2y2)padpad_maxprevent_zero_sizerc)rrrrr)bbsrr$r%rr&rrrr7r9r<r;extract_from_imagerZ atleast_3drrZclipr5r)r)rCrrrrrrr1r3ZsizehrrrrZbbrZ image_maskedextractrrrrs@ 6      zLineString.extract_from_imagecCs0t|ts t|}|jtj|j|jgdddS)a'Concatenate this line string with another one. This will add a line segment between the end point of this line string and the start point of `other`. Parameters ---------- other : imgaug.augmentables.lines.LineString or ndarray or iterable of tuple of number The points to add to this line string. Returns ------- imgaug.augmentables.lines.LineString New line string with concatenated points. The `label` of this line string will be kept. rrEr0)rQrr`r concatenater&)r)rMrrrrs zLineString.concatenatecCs:t|jdks |dkr|St|j|dd}|j|dS)a;Derive a new line string with ``N`` interpolated points per edge. The interpolated points have (per edge) regular distances to each other. For each edge between points ``A`` and ``B`` this adds points at ``A + (i/(1+N)) * (B - A)``, where ``i`` is the index of the added point and ``N`` is the number of points to add per edge. Calling this method two times will split each edge at its center and then again split each newly created edge at their center. It is equivalent to calling `subdivide(3)`. Parameters ---------- points_per_edge : int Number of points to interpolate on each edge. Returns ------- imgaug.augmentables.lines.LineString Line string with subdivided edges. rF)Znb_stepsclosedr0)rr&r`r )r)points_per_edger&rrr subdivide3s  zLineString.subdividecs ddlmfdd|jDS)zConvert the line string points to keypoints. Returns ------- list of imgaug.augmentables.kps.Keypoint Points of the line string as keypoints. rrGcsg|] \}}||dqS)rrrrRrSrGrrr]rsz+LineString.to_keypoints..)Zimgaug.augmentables.kpsrHr&r-rrGr to_keypointsRs zLineString.to_keypointscCsRddlm}t|jdkrdS|t|jt|jt|jt|j|j dS)a Generate a bounding box encapsulating the line string. Returns ------- None or imgaug.augmentables.bbs.BoundingBox Bounding box encapsulating the line string. ``None`` if the line string contained no points. rrrN)rrrrr') rrrr&rr<r1r3r;r')r)rrrrto_bounding_box_s zLineString.to_bounding_boxcCsddlm}||j|jdS)aHGenerate a polygon from the line string points. Returns ------- imgaug.augmentables.polys.Polygon Polygon with the same corner points as the line string. Note that the polygon might be invalid, e.g. contain less than ``3`` points or have self-intersections. r)Polygon)r')Zpolysrr&r')r)rrrr to_polygonrs zLineString.to_polygoncCs(ddlm}||j|||||d|dS)aGenerate a heatmap object from the line string. This is similar to :func:`~imgaug.augmentables.lines.LineString.draw_lines_heatmap_array`, executed with ``alpha=1.0``. The result is wrapped in a :class:`~imgaug.augmentables.heatmaps.HeatmapsOnImage` object instead of just an array. No points are drawn. Parameters ---------- image_shape : tuple of int The shape of the image onto which to draw the line mask. size_lines : int, optional Thickness of the line. size_points : int, optional Size of the points in pixels. antialiased : bool, optional Whether to draw the line with anti-aliasing activated. raise_if_out_of_image : bool, optional Whether to raise an error if the line string is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. Returns ------- imgaug.augmentables.heatmaps.HeatmapsOnImage Heatmap object containing drawn line string. r)HeatmapsOnImage)rrrrr)Zheatmapsrr)r)rrrrrrrrr to_heatmaps #zLineString.to_heatmapcCs&ddlm}||j||||d|dS)aGenerate a segmentation map object from the line string. This is similar to :func:`~imgaug.augmentables.lines.LineString.draw_mask`. The result is wrapped in a ``SegmentationMapsOnImage`` object instead of just an array. Parameters ---------- image_shape : tuple of int The shape of the image onto which to draw the line mask. size_lines : int, optional Thickness of the line. size_points : int, optional Size of the points in pixels. raise_if_out_of_image : bool, optional Whether to raise an error if the line string is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. Returns ------- imgaug.augmentables.segmaps.SegmentationMapsOnImage Segmentation map object containing drawn line string. r)SegmentationMapsOnImage)rrrr)Zsegmapsrr)r)rrrrrrrrto_segmentation_maps zLineString.to_segmentation_mapc Cst|trnt|trt|g}nt|}t|jdkr%t|jdkr%dSdt|jt|jfvr3dS||}||}||}||}tt|t|}||kS)aCompare this and another LineString's coordinates. This is an approximate method based on pointwise distances and can in rare corner cases produce wrong outputs. Parameters ---------- other : imgaug.augmentables.lines.LineString or tuple of number or ndarray or list of ndarray or list of tuple of number The other line string or its coordinates. max_distance : float, optional Max distance of any point from the other line string before the two line strings are evaluated to be unequal. points_per_edge : int, optional How many points to interpolate on each edge. Returns ------- bool Whether the two LineString's coordinates are almost identical, i.e. the max distance is below the threshold. If both have no coordinates, ``True`` is returned. If only one has no coordinates, ``False`` is returned. Beyond that, the number of points is not evaluated. rTF) rQrrTrr&rrWr;r) r)rMr\rZ self_subdZ other_subdZdist_self2otherZdist_other2selfdistrrrcoords_almost_equalss       zLineString.coords_almost_equalscCs |j|jkrdS|j|||dS)aCompare this and another line string. Parameters ---------- other: imgaug.augmentables.lines.LineString The other object to compare against. Expected to be a ``LineString``. max_distance : float, optional See :func:`~imgaug.augmentables.lines.LineString.coords_almost_equals`. points_per_edge : int, optional See :func:`~imgaug.augmentables.lines.LineString.coords_almost_equals`. Returns ------- bool ``True`` if the coordinates are almost equal and additionally the labels are equal. Otherwise ``False``. F)r\r)r'r)r)rMr\rrrr almost_equalss zLineString.almost_equalscCs,t|dur|jn||dur|jdS|dS)aCreate a shallow copy of this line string. Parameters ---------- coords : None or iterable of tuple of number or ndarray If not ``None``, then the coords of the copied object will be set to this value. label : None or str If not ``None``, then the label of the copied object will be set to this value. Returns ------- imgaug.augmentables.lines.LineString Shallow copy. Nr&r')rr&r'r(rrrru s  zLineString.copycCs8t|dur t|jn||durt|jdS|dS)aCreate a deep copy of this line string. Parameters ---------- coords : None or iterable of tuple of number or ndarray If not ``None``, then the coords of the copied object will be set to this value. label : None or str If not ``None``, then the label of the copied object will be set to this value. Returns ------- imgaug.augmentables.lines.LineString Deep copy. Nr)rrrur&copylibr`r'r(rrrr`6s zLineString.deepcopycC |j|S)zGet the coordinate(s) with given indices. Added in 0.4.0. Returns ------- ndarray xy-coordinate(s) as ``ndarray``. r0r)indicesrrr __getitem__M zLineString.__getitem__cC t|jS)zIterate over the coordinates of this instance. Added in 0.4.0. Yields ------ ndarray An ``(2,)`` ``ndarray`` denoting an xy-coordinate pair. )iterr&r-rrr__iter__ZrzLineString.__iter__cC|Srrr-rrr__repr__gzLineString.__repr__cCs$ddd|jD}d||jfS)N, cSsg|] \}}d||fqS)z (%.2f, %.2f)rrrrrrlrsz&LineString.__str__..zLineString([%s], label=%s))joinr&r')r)Z points_strrrrrjszLineString.__str__rrZ)F)TFTrrrNNNN)rrF)rcrTF)rcrF)rcrcrrTF)rrcrTF)rrcrTF rNNrcNNrNNTF)rTNTT)rrTF)rZrNN)3__name__ __module__ __qualname____doc__r*propertyr.r1r3r7r9r=r?rDr,rWrYr]r_rarirjrmrprhrvrrrrrrrrrrrrrrrrrrrrur`rrrrrrrrrs          ,  &  'r 7 - " - ) =  5 j _  , ( 1    rc@s eZdZdZddZeddZejddZeddZd d Z d d Z e d dZ e jfddZ     d@ddZdAddZdAddZddZdd Zd!d"Zd#d$ZdBd&d'ZdCd(d)Zd*d+Zd,d-Zd.d/Zd0d1ZdDd2d3ZdDd4d5Zd6d7Zd8d9Z d:d;Z!dd?Z#dS)ELineStringsOnImageaObject that represents all line strings on a single image. Parameters ---------- line_strings : list of imgaug.augmentables.lines.LineString List of line strings on the image. shape : tuple of int or ndarray The shape of the image on which the objects are placed. Either an image with shape ``(H,W,[C])`` or a ``tuple`` denoting such an image shape. Examples -------- >>> import numpy as np >>> from imgaug.augmentables.lines import LineString, LineStringsOnImage >>> >>> image = np.zeros((100, 100)) >>> lss = [ >>> LineString([(0, 0), (10, 0)]), >>> LineString([(10, 20), (30, 30), (50, 70)]) >>> ] >>> lsoi = LineStringsOnImage(lss, shape=image.shape) cCs^t|sJdt|ftdd|Ds%Jdddd|D||_t||_dS)Nz9Expected 'line_strings' to be an iterable, got type '%s'.cSg|]}t|tqSrrQrrvrrrrrz/LineStringsOnImage.__init__..z/Expected iterable of LineString, got types: %s.rcSg|]}tt|qSrr#r!rrrrrr)rr r!r"r line_stringsr r%r)rr%rrrr*s zLineStringsOnImage.__init__cCs|jS)zGet the line strings in this container. Added in 0.4.0. Returns ------- list of LineString Line strings within this container. rr-rrritemss zLineStringsOnImage.itemscCs ||_dS)zSet the line strings in this container. Added in 0.4.0. Parameters ---------- value : list of LineString Line strings within this container. Nr)r)valuerrrrs cCst|jdkS)zEstimate whether this object contains zero line strings. Returns ------- bool ``True`` if this object contains zero line strings. r)rrr-rrremptys zLineStringsOnImage.emptycCs^t|}|dd|jddkr||_|St|jD]\}}||j||j|<q||_|S)a Project the line strings from one image shape to a new one in-place. Added in 0.4.0. Parameters ---------- image : ndarray or tuple of int The new image onto which to project. Either an image with shape ``(H,W,[C])`` or a tuple denoting such an image shape. Returns ------- imgaug.augmentables.lines.LineStrings Object containing all projected line strings. The object and its items may have been modified in-place. rr)r r%rwrr_r)r)rCZon_shaperitemrrron_szLineStringsOnImage.on_cC||S)aProject the line strings from one image shape to a new one. Parameters ---------- image : ndarray or tuple of int The new image onto which to project. Either an image with shape ``(H,W,[C])`` or a tuple denoting such an image shape. Returns ------- imgaug.augmentables.lines.LineStrings Object containing all projected line strings. )r`r)r)rCrrronszLineStringsOnImage.oncCs&g}|D] }|t|q|||S)agConvert an ``(N,M,2)`` ndarray to a ``LineStringsOnImage`` object. This is the inverse of :func:`~imgaug.augmentables.lines.LineStringsOnImage.to_xy_array`. Parameters ---------- xy : (N,M,2) ndarray or iterable of (M,2) ndarray Array containing the point coordinates ``N`` line strings with each ``M`` points given as ``(x,y)`` coordinates. ``M`` may differ if an iterable of arrays is used. Each array should usually be of dtype ``float32``. shape : tuple of int ``(H,W,[C])`` shape of the image on which the line strings are placed. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Object containing a list of ``LineString`` objects following the provided point coordinates. )ryr)clsxyr%lssZxy_lsrrrfrom_xy_arrayss z!LineStringsOnImage.from_xy_arrayscs"ddlmfdd|jDS)aConvert this object to an iterable of ``(M,2)`` arrays of points. This is the inverse of :func:`~imgaug.augmentables.lines.LineStringsOnImage.from_xy_array`. Parameters ---------- dtype : numpy.dtype, optional Desired output datatype of the ndarray. Returns ------- list of ndarray The arrays of point coordinates, each given as ``(M,2)``. rrcs g|] }t|jqSr)rrrur&rerrrrr.sz3LineStringsOnImage.to_xy_arrays..)rrr)r)rrr"r to_xy_arrayss  zLineStringsOnImage.to_xy_arraysrNrcrTFc Cs2|jD]} | j||||||||| | | | d }q|S)az Draw all line strings onto a given image. Parameters ---------- image : ndarray The ``(H,W,C)`` ``uint8`` image onto which to draw the line strings. color : iterable of int, optional Color to use as RGB, i.e. three values. The color of the lines and points are derived from this value, unless they are set. color_lines : None or iterable of int Color to use for the line segments as RGB, i.e. three values. If ``None``, this value is derived from `color`. color_points : None or iterable of int Color to use for the points as RGB, i.e. three values. If ``None``, this value is derived from ``0.5 * color``. alpha : float, optional Opacity of the line strings. Higher values denote more visible points. The alphas of the line and points are derived from this value, unless they are set. alpha_lines : None or float, optional Opacity of the line strings. Higher values denote more visible line string. If ``None``, this value is derived from `alpha`. alpha_points : None or float, optional Opacity of the line string points. Higher values denote more visible points. If ``None``, this value is derived from `alpha`. size : int, optional Size of the line strings. The sizes of the line and points are derived from this value, unless they are set. size_lines : None or int, optional Thickness of the line segments. If ``None``, this value is derived from `size`. size_points : None or int, optional Size of the points in pixels. If ``None``, this value is derived from ``3 * size``. antialiased : bool, optional Whether to draw the lines with anti-aliasing activated. This does currently not affect the point drawing. raise_if_out_of_image : bool, optional Whether to raise an error if a line string is fully outside of the image. If set to ``False``, no error will be raised and only the parts inside the image will be drawn. Returns ------- ndarray Image with line strings drawn on it. ) rrrrrrrrrrr)rr)r)rCrrrrrrrrrrrrfrrrr1s H z LineStringsOnImage.draw_on_imagecsfddjD_S)a Remove all LS that are fully/partially outside of an image in-place. Added in 0.4.0. Parameters ---------- fully : bool, optional Whether to remove line strings that are fully outside of the image. partly : bool, optional Whether to remove line strings that are partially outside of the image. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Reduced set of line strings. Those that are fully/partially outside of the given image plane are removed. The object and its items may have been modified in-place. cs"g|] }|jjds|qS)rnro)rpr%rernror)rrrs z;LineStringsOnImage.remove_out_of_image_..rr)rnrorr%rremove_out_of_image_sz'LineStringsOnImage.remove_out_of_image_cCs|j||dS)aG Remove all line strings that are fully/partially outside of an image. Parameters ---------- fully : bool, optional Whether to remove line strings that are fully outside of the image. partly : bool, optional Whether to remove line strings that are partially outside of the image. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Reduced set of line strings. Those that are fully/partially outside of the given image plane are removed. r$)rur'r&rrrremove_out_of_imagesz&LineStringsOnImage.remove_out_of_imagecCs t||S)aRemove all LS with an OOI fraction of at least `fraction` in-place. 'OOI' is the abbreviation for 'out of image'. Added in 0.4.0. Parameters ---------- fraction : number Minimum out of image fraction that a line string has to have in order to be removed. A fraction of ``1.0`` removes only line strings that are ``100%`` outside of the image. A fraction of ``0.0`` removes all line strings. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Reduced set of line strings, with those that had an out of image fraction greater or equal the given one removed. The object and its items may have been modified in-place. )r r)fractionrrrremove_out_of_image_fraction_s z0LineStringsOnImage.remove_out_of_image_fraction_cCr)a{Remove all LS with an out of image fraction of at least `fraction`. Parameters ---------- fraction : number Minimum out of image fraction that a line string has to have in order to be removed. A fraction of ``1.0`` removes only line strings that are ``100%`` outside of the image. A fraction of ``0.0`` removes all line strings. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Reduced set of line strings, with those that had an out of image fraction greater or equal the given one removed. )rur+r)rrrremove_out_of_image_fractionsz/LineStringsOnImage.remove_out_of_image_fractioncsfddjD_S)a Clip off all parts of the LSs that are outside of an image in-place. .. note:: The result can contain fewer line strings than the input did. That happens when a polygon is fully outside of the image plane. .. note:: The result can also contain *more* line strings than the input did. That happens when distinct parts of a line string are only connected by line segments that are outside of the image plane and hence will be clipped off, resulting in two or more unconnected line string parts that are left in the image plane. Added in 0.4.0. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Line strings, clipped to fall within the image dimensions. The count of output line strings may differ from the input count. cs"g|] }|jD]}|q qSr)rhr%)rrfZ ls_clippedr-rrrs z9LineStringsOnImage.clip_out_of_image_..rr-rr-rclip_out_of_image_s z%LineStringsOnImage.clip_out_of_image_cCs |S)as Clip off all parts of the line strings that are outside of an image. .. note:: The result can contain fewer line strings than the input did. That happens when a polygon is fully outside of the image plane. .. note:: The result can also contain *more* line strings than the input did. That happens when distinct parts of a line string are only connected by line segments that are outside of the image plane and hence will be clipped off, resulting in two or more unconnected line string parts that are left in the image plane. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Line strings, clipped to fall within the image dimensions. The count of output line strings may differ from the input count. )rur-r-rrrrhs z$LineStringsOnImage.clip_out_of_imagercCs,t|jD]\}}|j||d|j|<q|S)aMove the line strings along the x/y-axis in-place. The origin ``(0, 0)`` is at the top left of the image. Added in 0.4.0. Parameters ---------- x : number, optional Value to be added to all x-coordinates. Positive values shift towards the right images. y : number, optional Value to be added to all y-coordinates. Positive values shift towards the bottom images. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Shifted line strings. The object and its items may have been modified in-place. r)rwrr)r)rRrSrrfrrrrszLineStringsOnImage.shift_cCr)aMove the line strings along the x/y-axis. The origin ``(0, 0)`` is at the top left of the image. Parameters ---------- x : number, optional Value to be added to all x-coordinates. Positive values shift towards the right images. y : number, optional Value to be added to all y-coordinates. Positive values shift towards the bottom images. top : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift all objects *from* the top (towards the bottom). right : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift all objects *from* the right (towads the left). bottom : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift all objects *from* the bottom (towards the top). left : None or int, optional Deprecated since 0.4.0. Amount of pixels by which to shift all objects *from* the left (towards the right). Returns ------- imgaug.augmentables.lines.LineStringsOnImage Shifted line strings. rrrrrrrr:rzLineStringsOnImage.shiftcCs,|jr tjdtjdStdd|jDS)aConvert all line string coordinates to one array of shape ``(N,2)``. Added in 0.4.0. Returns ------- (N, 2) ndarray Array containing all xy-coordinates of all line strings within this instance. rrcSrdrr0rerrrrurgz2LineStringsOnImage.to_xy_array..)rrrrrrr-rrr to_xy_arraygs zLineStringsOnImage.to_xy_arraycCstj|tjd}|jddks#|jdkr|jddks#Jd|jfd}|jD]4}t|j}||t|ksIJdt|tdd|jDf|||||jd d d f<||7}q(|t|ksmJd t||f|S) aModify the corner coordinates of all line strings in-place. .. note:: This currently expects that `xy` contains exactly as many coordinates as the line strings within this instance have corner points. Otherwise, an ``AssertionError`` will be raised. Added in 0.4.0. Parameters ---------- xy : (N, 2) ndarray or iterable of iterable of number XY-Coordinates of ``N`` corner points. ``N`` must match the number of corner points in all line strings within this instance. Returns ------- LineStringsOnImage This instance itself, with updated coordinates. Note that the instance was modified in-place. rrrrz7Expected input array to have shape (N,2), got shape %s.zcReceived fewer points than there are corner points in all line strings. Got %d points, expected %d.cSsg|]}t|jqSr)rr&)rZls_rrrrrz:LineStringsOnImage.fill_from_xy_array_..N.zExpected to get exactly as many xy-coordinates as there are points in all line strings polygons within this instance. Got %d points, could only assign %d points.) rrrr%r$rrr&r+)r)rcounterrfZ nb_pointsrrrfill_from_xy_array_ws2(   z&LineStringsOnImage.fill_from_xy_array_cCsJddlm}|jr|g|jdStjdd|jDdd}|j||jdS)aConvert the line strings to one ``KeypointsOnImage`` instance. Added in 0.4.0. Returns ------- imgaug.augmentables.kps.KeypointsOnImage A keypoints instance containing ``N`` coordinates for a total of ``N`` points in the ``coords`` attributes of all line strings. Order matches the order in ``line_strings`` and ``coords`` attributes. rrrcSrdrr0rerrrrrgz.rrE)rrrr%rrrr)r)rr&rrrto_keypoints_on_images z(LineStringsOnImage.to_keypoints_on_imagec Cs|j}dd|D}tdd|D}t|j|ks%Jd|t|jf|}d}|D]!}|j}|||t|ddf|ddddf<|t|7}q-|j|_|S)aBInvert the output of ``to_keypoints_on_image()`` in-place. This function writes in-place into this ``LineStringsOnImage`` instance. Added in 0.4.0. Parameters ---------- kpsoi : imgaug.augmentables.kps.KeypointsOnImages Keypoints to convert back to line strings, i.e. the outputs of ``to_keypoints_on_image()``. Returns ------- LineStringsOnImage Line strings container with updated coordinates. Note that the instance is also updated in-place. cSrdrr0rerrrrrgzDLineStringsOnImage.invert_to_keypoints_on_image_..cSsg|]}t|qSrr)rr&rrrrz Expected %d coordinates, got %d.rN)rr+rZ keypointsr.r&r%) r)rr ZcoordssZ nb_points_expZxy_arrr/rfr&rrrinvert_to_keypoints_on_image_s  ,z0LineStringsOnImage.invert_to_keypoints_on_image_cCs2|dur |jdd}|durt|j}t||S)a'Create a shallow copy of this object. Parameters ---------- line_strings : None or list of imgaug.augmentables.lines.LineString, optional List of line strings on the image. If not ``None``, then the ``line_strings`` attribute of the copied object will be set to this value. shape : None or tuple of int or ndarray, optional The shape of the image on which the objects are placed. Either an image with shape ``(H,W,[C])`` or a tuple denoting such an image shape. If not ``None``, then the ``shape`` attribute of the copied object will be set to this value. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Shallow copy. NrrTr%r rrrrrus   zLineStringsOnImage.copycCs4|dur dd|jD}|durt|j}t||S)a Create a deep copy of the object. Parameters ---------- line_strings : None or list of imgaug.augmentables.lines.LineString, optional List of line strings on the image. If not ``None``, then the ``line_strings`` attribute of the copied object will be set to this value. shape : None or tuple of int or ndarray, optional The shape of the image on which the objects are placed. Either an image with shape ``(H,W,[C])`` or a tuple denoting such an image shape. If not ``None``, then the ``shape`` attribute of the copied object will be set to this value. Returns ------- imgaug.augmentables.lines.LineStringsOnImage Deep copy. NcSsg|]}|qSrrtrerrrr r2z/LineStringsOnImage.deepcopy..r4rrrrr` s   zLineStringsOnImage.deepcopycCr)zGet the line string(s) with given indices. Added in 0.4.0. Returns ------- list of imgaug.augmentables.lines.LineString Line string(s) with given indices. rrrrrr& rzLineStringsOnImage.__getitem__cCr)a*Iterate over the line strings in this container. Added in 0.4.0. Yields ------ LineString A line string in this container. The order is identical to the order in the line string list provided upon class initialization. )rrr-rrrr3 s zLineStringsOnImage.__iter__cCr)zGet the number of items in this instance. Added in 0.4.0. Returns ------- int Number of items in this instance. )rrr-rrr__len__B rzLineStringsOnImage.__len__cCrrrr-rrrrO rzLineStringsOnImage.__repr__cCsdt|j|jfS)Nz LineStringsOnImage(%s, shape=%s))r#rr%r-rrrrR s zLineStringsOnImage.__str__r)TFrrr)$rr r r r*r rsetterrrr classmethodr!rrr#rr'r(r+r,r-rhrrr.r0r1r3rur`rrr5rrrrrrr ~sN      T   -3 &    r rZcCsrtjt|t|}tjt|t|tjt|t|}| ||ko6|kSSrr)rrrKr|Zdist_s2eZ dist_s2p2errr_is_point_on_lineW s r8ccsvddl}t|ts |g}|D])}t|dr5t|jD]}t||jjjr.|D]}|Vq'qt|Vqq|VqdS)Nrgeoms) rOrQrhasattrrr9rIbaseZGeometrySequence)Z collectionrNrsubitemZsubsubelrrrr` s   rcsddlt|trj|d|d}|St|tr`t|dks*Jdt|ft|dtr9j|}|Stdd|DrQj fdd|D}|St dd d d|Dft|trnj|j }|St d t |f) NrrzEExpected list to contain at least one coordinate, got %d coordinates.cSrrrrrrrr{ rz4_convert_var_to_shapely_geometry..csg|] }j|jqSr)rIrr&rerNrrr| szdCould not convert list-input to shapely geometry. Invalid datatype. List elements had datatypes: %s.rcSrrrrrrrr rzFCould not convert input to shapely geometry. Invalid datatype. Got: %s)rOrQrTrIrJrrrr"ZMultiLineStringrUrr&r!)varrrr=rrq s@    rr)r  __future__rrrrurnumpyrZ skimage.drawrZskimage.measurerrrrr;rutilsr r r r r objectrr r8rrrrrrs@   r ^