o :ήc @sVddlZddlZddlmZddlmZgdZdddZd d Z d d Z d dZ dS)N)color)_convert) adapt_rgb hsv_value each_channelcCs|jdko |j|dvS)zReturn True if the image *looks* like it's RGB. This function should not be public because it is only intended to be used for functions that don't accept volumes as input, since checking an image's shape is fragile. )r )ndimshape)image channel_axisr>/tmp/pip-target-vg8gfxp4/lib/python/skimage/color/adapt_rgb.py is_rgb_like srcsfdd}|S)aReturn decorator that adapts to RGB images to a gray-scale filter. This function is only intended to be used for functions that don't accept volumes as input, since checking an image's shape is fragile. Parameters ---------- apply_to_rgb : function Function that returns a filtered image from an image-filter and RGB image. This will only be called if the image is RGB-like. cstfdd}|S)Ncs6t|r|g|Ri|S|g|Ri|S)N)r)r argskwargs) apply_to_rgb image_filterrrimage_filter_adapted#sz:adapt_rgb..decorator..image_filter_adapted) functoolswraps)rrr)rr decorator"szadapt_rgb..decoratorr)rrrrrrs rcOsxt|ddddddf}|dddddf}||g|Ri|}t||j|dddddf<t|S)aLReturn color image by applying `image_filter` on HSV-value of `image`. Note that this function is intended for use with `adapt_rgb`. Parameters ---------- image_filter : function Function that filters a gray-scale image. image : array Input image. Note that RGBA images are treated as RGB. Nr r)rrgb2hsvcopyrdtypehsv2rgb)rr rrhsvvaluerrrr-s  rcs.fddt|ddD}tj|ddS)a!Return color image by applying `image_filter` on channels of `image`. Note that this function is intended for use with `adapt_rgb`. Parameters ---------- image_filter : function Function that filters a gray-scale image. image : array Input image. cs"g|] }|gRiqSrr).0crrrrr Msz each_channel..rr)axis)npmoveaxisstack)rr rrc_newrr#rrAs r)r) rnumpyr&r util.dtyper__all__rrrrrrrrs