U VcA@sLdZddlZddlZdZedddgZefddZefd d Zd d Z dS) z"Package for histogram compression.N) rii3i iii it$i'CompressedHistogramValueZ basis_pointvaluec CsR|jsdd|DSt|j}t|j}||d|p>d}g}d}|t|krtj |||dd}|t|kr||}|dkr||dnd } || kr|d7}qp|r| s|j } nt ||d|j } t |||j } t ||| || | } | t||| |d7}qNqpqqN|t|krN| t|||j |d7}q|S) aCreates fixed size histogram by adding compression to accumulated state. This routine transforms a histogram at a particular step by interpolating its variable number of buckets to represent their cumulative weight at a constant number of compression points. This significantly reduces the size of the histogram and makes it suitable for a two-dimensional area plot where the output of this routine constitutes the ranges for a single x coordinate. Args: histo: A HistogramProto object. bps: Compression points represented in basis points, 1/100ths of a percent. Defaults to normal distribution. Returns: List of values for each basis point. cSsg|]}t|dqSr.0br O/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/plugins/distribution/compressor.py 5sz,compress_histogram_proto..?rrightZsider)numnparraybucketlist bucket_limitsumcumsumlen searchsortedminmax_lerpappendr) Zhistobpsrrweightsvaluesjir cumsum_prevlhsrhsweightr r r compress_histogram_proto"s8   r)cCs|t|}|jsdd|DS|dd|dd}}|dddf}t|dddf}||d|ppd}g}d}|t|krJtj|||d d } | t|krJ|| } | dkr|| dnd } | | kr| d7} q| r| s|} nt|| d|} t || |} t ||| | | | }| t ||||d7}qqqJq|t|krx| t ||||d7}qJ|S) aCreates fixed size histogram by adding compression to accumulated state. This routine transforms a histogram at a particular step by linearly interpolating its variable number of buckets to represent their cumulative weight at a constant number of compression points. This significantly reduces the size of the histogram and makes it suitable for a two-dimensional area plot where the output of this routine constitutes the ranges for a single x coordinate. Args: buckets: A list of buckets, each of which is a 3-tuple of the form `(min, max, count)`. bps: Compression points represented in basis points, 1/100ths of a percent. Defaults to normal distribution. Returns: List of values for each basis point. cSsg|]}t|dqSrrrr r r r jsz&compress_histogram..rr rNrrrr) rrsizerrrrrrrrrr)Zbucketsr ZminminZmaxmaxcountsZ right_edgesr!resultZbp_indexr$rr%r&r'r(r r r compress_histogramTs<  r.cCs |||t||||S)z)Affinely map from [x0, x1] onto [y0, y1].)float)xZx0x1Zy0y1r r r rsr) __doc__ collectionsZnumpyrZNORMAL_HISTOGRAM_BPS namedtuplerr)r.rr r r r s 2 7