o :ήc @sZddlZddlmZddlmZmZddlm Z ddl m Z Gdd d e Z d d d Z dS)N)colorexposure) PlotPlugin) RectangleToolcsNeZdZdZdfdd ZfddZddZd d Zd d Zd dZ Z S)ColorHistogramColor HistogramGz?c s0tt|jdddi|||_t|dS)Nheighti)superr __init__max_pctprinthelp)selfrkwargs __class__r M/tmp/pip-target-vg8gfxp4/lib/python/skimage/viewer/plugins/color_histogram.pyr szColorHistogram.__init__cs0tt||t||jd|_||jdS)N) on_release)rr attachr ab_selected rect_tool _on_new_imageimage)r image_viewerrr rrs zColorHistogram.attachc Cst||_|jj\}}}d\}}|d|d|d|dg}t|jt|_t ||}tj | | |dd\} } } || | | f||||fd|_ t | |jd} tj| d| fd} |jj| |tjjd |jd |jd |jd dS) N)idg?T)normed)binshistedgesextents) percentiler)in_range)extentcmapr ba)rrgb2lab lab_imageTnponesshapeboolmaskarange histogram2dflattendatapct_total_arearrrescale_intensityaximshowpltcmgray set_title set_xlabel set_ylabel) rrLr*r)leftright ab_extentsr!r"x_edgesy_edgesmax_valr r rrs$      zColorHistogram._on_new_imagecCsd}d|S)N)zColor Histogram tool:z6Select region of a-b colorspace to highlight on image. )join)rhelpstrr r rr.s zColorHistogram.helpc Csz|\}}}}||jd<|j}|j\}}} ||k||k@| |k| |k@@|_d|dddf|jj<t||j_dS)Nr$r.r) r6r,copyr-r2rlab2rgbrr) rr$x0x1y0y1 lab_maskedrAr*r)r r rr3s    "zColorHistogram.ab_selectedcCs |j|jfS)aReturn the image mask and the histogram data. Returns ------- mask : array of bool, same shape as image The selected pixels. data : dict The data describing the histogram and the selected region. The dictionary contains: - 'bins' : array of float The bin boundaries for both `a` and `b` channels. - 'hist' : 2D array of float The normalized histogram. - 'edges' : tuple of array of float The bin edges along each dimension - 'extents' : tuple of float The left and right and top and bottom of the selected region. )r2r6)rr r routput?s zColorHistogram.output)r ) __name__ __module__ __qualname__namerrrrrrR __classcell__r r rrr s  r 皙?cCs&t|jd|}t|j}||S)zReturn threshold value based on percentage of total area. The specified percent of pixels less than the given intensity threshold. r)intsizer.sortflat)rr%idx sorted_pixelsr r rr7Vs r7)rX)numpyr.matplotlib.pyplotpyplotr;rr plotpluginr canvastoolsrr r7r r r rs   N