B \_;@sddlZddlZddlmZddlZddlmZmZddl m Z ddl m Z ddl mZmZddlmZddlZdd lmZejZed d ee DZ ed e fe Zd dZddZddZddZddZGdddeZ GdddeZ!e"dkrddl#Z#e#$dS)N) namedtuple)io img_as_ubyte)resize) color_dict)file_or_url_contextis_url)ImageCollection)requestccs(|] \}}|tdd|DfVqdS)css|]}td|dVqdS)g?N)int).0cr5lib/python3.7/site-packages/skimage/novice/_novice.py sz.N)tuple)rnamergbrrrrsrcolorscCst|S)z0Return Picture object from the given image path.)Picture)pathrrropensrcCst|trt|dks$td|tdd|Dr:|St|}x4t|D](\}}t|trLt ||d||<}qLWt|S)z5Raise error if picture index is not a 2D index/slice.rzExpected 2D index but got {0!r}css|]}t|tVqdS)N) isinstancer )rirrrr!sz(_verify_picture_index..) rrlen IndexErrorformatalllist enumerater slice)indexrZ dim_slicerrr_verify_picture_indexs r%cCs t|dS)z.Return RGB array with first 2 axes transposed.)rrr)npZ transpose)arrayrrr rgb_transpose/sr(cCst|dddS)z@Return view of image transformed from array to Cartesian origin.N)r()imagerrrarray_to_xy_origin4sr+cCst|dddddfS)z@Return view of image transformed from Cartesian to array origin.Nr))r()r*rrrxy_to_array_origin9sr,c@seZdZdZd$ddZeddZeddZed d Zej d d Zed d Z e j dd Z eddZ e j ddZ eddZ e j ddZ eddZ e j ddZ eddZej ddZddZddZdd Zd!d"Zd#S)%PixelaA single pixel in a Picture. Attributes ---------- pic : Picture The Picture object that this pixel references. array : array_like Byte array with raw image data (RGB). x : int Horizontal coordinate of this pixel (left = 0). y : int Vertical coordinate of this pixel (bottom = 0). rgb : tuple RGB tuple with red, green, and blue components (0-255) alpha : int Transparency component (0-255), 255 (opaque) by default r cCsR||_||_||_||d|_||d|_||d|_|||_dS)Nrrr)_picture_x_y _validate_red_green_blue_alpha)selfZpicr'xyralpharrr__init__RszPixel.__init__cCs|jS)z@Horizontal location of this pixel in the parent image(left = 0).)r/)r6rrrr7[szPixel.xcCs|jS)zAVertical location of this pixel in the parent image (bottom = 0).)r0)r6rrrr8`szPixel.ycCs|jS)z'The red component of the pixel (0-255).)r2)r6rrrredesz Pixel.redcCs|||_|dS)N)r1r2 _setpixel)r6valuerrrr;js cCs|jS)z)The green component of the pixel (0-255).)r3)r6rrrgreenosz Pixel.greencCs|||_|dS)N)r1r3r<)r6r=rrrr>ts cCs|jS)z(The blue component of the pixel (0-255).)r4)r6rrrblueysz Pixel.bluecCs|||_|dS)N)r1r4r<)r6r=rrrr?~s cCs|jS)z0The transparency component of the pixel (0-255).)r5)r6rrrr9sz Pixel.alphacCs|||_|dS)N)r1r5r<)r6r=rrrr9s cCs|j|j|jfS)z7The RGB color components of the pixel (3 values 0-255).)r;r>r?)r6rrrrsz Pixel.rgbcsDt|dkr|_n,fdd|D\___d_dS)Nc3s|]}|VqdS)N)r1)rv)r6rrrszPixel.rgb..r )rrgbar2r3r4r5r<)r6r=r)r6rrs  cCs|j|j|j|jfS)zYThe RGB color and transparency components of the pixel (4 values 0-255). )r;r>r?r9)r6rrrrBsz Pixel.rgbacs.fdd|D\____dS)Nc3s|]}|VqdS)N)r1)rrA)r6rrrszPixel.rgba..)r2r3r4r5r<)r6r=r)r6rrBs"cCsNy"t|}|dks|dkr tWn&tk rHd}t||YnX|S)z-Verifies that the pixel value is in [0, 255].rr z;Expected an integer between 0 and 255, but got {0} instead!)r ValueErrorr)r6r=msgrrrr1s zPixel._validatecCs$|j|jj|j|jf<|jdS)N)rBr.xy_arrayr/r0_array_modified)r6rrrr<szPixel._setpixelcCst|tr|j|jkSdS)N)rr-rB)r6otherrrr__eq__s z Pixel.__eq__cCs|j|j|j|jf}dj|S)Nz.Pixel(red={0}, green={1}, blue={2}, alpha={3}))r;r>r?r9r)r6argsrrr__repr__szPixel.__repr__N)r )__name__ __module__ __qualname____doc__r:propertyr7r8r;setterr>r?r9rrBr1r<rHrJrrrrr->s(         r-c@seZdZdZdOddZedPddZedd Zej d d Zed d Z e j d d Z ddZ ddZ eddZ eddZddZeddZeddZej ddZeddZej ddZed d!Zej d"d!Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zed-d.Zej d/d.Zed0d1Zej d2d1Zed3d4Zej d5d4Zed6d7Zej d8d7Zed9d:Zej d;d:Zedd=Zd?d@ZdAdBZ dCdDZ!dEdFZ"dGdHZ#dIdJZ$dKdLZ%dMdNZ&dS)QraA 2-D picture made up of pixels. Attributes ---------- path : str Path to an image file to load / URL of an image array : array Raw RGB or RGBA image data [0-255], with origin at top-left. xy_array : array Raw RGB or RGBA image data [0-255], with origin at bottom-left. Examples -------- Load an image from a file: >>> from skimage import novice >>> from skimage import data >>> picture = novice.open(data.data_dir + '/chelsea.png') Load an image from a URL (the URL must start with ``http(s)://`` or ``ftp(s)://``): >>> picture = novice.open('https://scikit-image.org/_static/img/logo.png') # doctest: +SKIP Create a blank 100 pixel wide, 200 pixel tall white image: >>> pic = Picture.from_size((100, 200), color=(255, 255, 255)) Use numpy to make an RGB byte array (shape is height x width x 3): >>> import numpy as np >>> data = np.zeros(shape=(200, 100, 3), dtype=np.uint8) >>> data[:, :, 0] = 255 # Set red component to maximum >>> pic = Picture(array=data) Get the bottom-left pixel: >>> pic[0, 0] Pixel(red=255, green=0, blue=0, alpha=255) Get the top row of the picture: >>> pic[:, pic.height-1] Picture(100 x 1) Set the bottom-left pixel to black: >>> pic[0, 0] = (0, 0, 0) Set the top row to red: >>> pic[:, pic.height-1] = (255, 0, 0) Nc Csd|_d|_d|_tdd|||gD}|dkr@d}t|nr|dk rt|s\tj|}||_t |"}t t ||_ t||_WdQRXn|dk r||_ n|dk r||_|j jddkrtj|j ddd d |_ dS) NFcSsg|]}|dk r|qS)Nr)rarrr sz$Picture.__init__..rz:Must provide a single keyword arg (path, array, xy_array).r)r r)valuesaxis) _modified_path_formatrrCr osrabspathrrrZimreadr'imghdrwhatrEshaper&insert)r6rr'rEZn_argsrDcontextrrrr:s(   zPicture.__init__blackcCstt|tjrt|}t|t|f}tj|tjd}tj |tjd|}|j ddkrjtj |dddd}t |dS)aJReturn a Picture of the specified size and a uniform color. Parameters ---------- size : tuple Width and height of the picture in pixels. color : tuple or str RGB or RGBA tuple with the fill color for the picture [0-255] or a valid key in `color_dict`. )Zdtyper)rSr r)rTrU)r') rsixZ string_typesrrrr&r'uint8Zonesr]r^r)sizecolorZrgb_sizer'rrr from_sizes zPicture.from_sizecCs|jS)z!Image data stored as numpy array.)_array)r6rrrr',sz Picture.arraycCs*|tj|_t|j|_|j|_dS)N)astyper&rbrfr+ _xy_arraycopy _array_backup)r6r'rrrr'1s cCs|jS)z@Image data stored as numpy array with origin at the bottom-left.)rh)r6rrrrE7szPicture.xy_arraycCs||_t||_dS)N)rhr,rf)r6r'rrrrE<scCsz|jjdkrH|jjddkrHtj|ddkrH|jdddf|_t||jd|_ tj ||_ t ||_dS)zSaves the picture to the given path. Parameters ---------- path : str Path (with file extension) where the picture is saved. rSr)r@)z.jpgz.jpeg.NF)r'ndimr]rYrsplitextlowerrimsaverVrZrWr[r\rX)r6rrrrsaveAsz Picture.savecCs |j|_dS)zDReset image to its original state, removing modifications. N)rjr')r6rrrresetQsz Picture.resetcCs|jS)zThe path to the picture.)rW)r6rrrrWsz Picture.pathcCs|jS)z True if the picture has changed.)rV)r6rrrmodified\szPicture.modifiedcCsd|_d|_dS)NT)rVrW)r6rrrrFaszPicture._array_modifiedcCs|jS)z The image format of the picture.)rX)r6rrrreszPicture.formatcCs|jjddS)z(The size (width, height) of the picture.Nr)rEr])r6rrrrcjsz Picture.sizecCsd|d|jks|d|jkr`t|dt|df}t|j|ddddd}|tj|_|dS)NrrTZconstantF)orderZpreserve_rangemodeZ anti_aliasing) widthheightr rr'rgr&rbrF)r6r=Znew_sizeZ new_arrayrrrrcos cCs |jdS)zThe width of the picture.r)rc)r6rrrrt|sz Picture.widthcCs||jf|_dS)N)rurc)r6r=rrrrtscCs |jdS)zThe height of the picture.r)rc)r6rrrruszPicture.heightcCs|j|f|_dS)N)rtrc)r6r=rrrruscCst|jtdS)zDisplay the image.N)rZimshowr'show)r6rrrrvs z Picture.showcs:|j|jgtddgfddd}ttdS)z,Compare the image to its unmodified version.rrcs|S)Nr)r7)imagesrrsz!Picture.compare..)Z load_funcN)rjr'r rZimshow_collectionrv)r6Zicr)rwrcompares  zPicture.comparecCs |j||f}t||j|||S)z0Create a Pixel object for a given x, y location.)rEr-r')r6r7r8rrrr _makepixelszPicture._makepixelcCs|jdddd|fS)zsz Picture.greencCs|d|dS)Nr)r})r6r=rrrr>scCs|dS)z(The blue component of the pixel (0-255).r)r|r~)r6rrrr?sz Picture.bluecCs|d|dS)Nr)r})r6r=rrrr?scCs|dS)z0The transparency component of the pixel (0-255).rS)r|r~)r6rrrr9sz Picture.alphacCs|d|dS)NrS)r})r6r=rrrr9scCs|jddddddfS)z7The RGB color components of the pixel (3 values 0-255).NrS)rE)r6rrrrsz Picture.rgbcCs ||jddddddf<dS)NrS)rE)r6r=rrrrscCs|jS)z8The RGBA color components of the pixel (4 values 0-255).)rE)r6rrrrBsz Picture.rgbacCs||jdd<dS)N)rE)r6r=rrrrBsccs:x4t|jD]&}x t|jD]}|||VqWq WdS)z&Iterates over all pixels in the image.N)rangertrurz)r6r7r8rrr__iter__szPicture.__iter__cCs8t|}tdd|Dr$|j|St|j|dSdS)z6Return `Picture`s for slices and `Pixel`s for indexes.css|]}t|tVqdS)N)rr )rr$rrrrsz&Picture.__getitem__..)rEN)r%r rzrrE)r6xy_indexrrr __getitem__s zPicture.__getitem__cCsJt|}t|tr|||_n t|tr6|j|j|<ntd|dS)NzInvalid value type)r%rrrrrE TypeErrorrF)r6rr=rrr __setitem__s   zPicture.__setitem__cCs"t|tstt|j|jkS)N)rrNotImplementedErrorr&r r')r6rGrrrrHs zPicture.__eq__cCs dj|jS)NzPicture({0} x {1}))rrc)r6rrrrJszPicture.__repr__cCs |dS)NZpng)_repr_image_format)r6rrr _repr_png_szPicture._repr_png_cCs |dS)NZjpeg)r)r6rrr _repr_jpeg_szPicture._repr_jpeg_cCs.t}tj||j|d|}||S)N) format_str)raBytesIOrrnr'getvalueclose)r6rZ str_bufferZ return_strrrrrs zPicture._repr_image_format)NNN)r`)'rKrLrMrNr: staticmethodrerOr'rPrErorprrqrFrrcrtrurvryrzr|r}r;r>r?r9rrBrrrrHrJrrrrrrrrsZ6                 r__main__)%rYr[ collectionsrZnumpyr&rrZ transformrrdrZio.utilrr Z io.collectionr raZsix.moves.urllibr ZurlopendictZ iteritemskeysrrr%r(r+r,objectr-rrKZdoctestZtestmodrrrrs6     J