a dM@sbdZddlZddlZddlmZdZejddGdddZefd d Z efd d Z d dZ dS)z"Package for histogram compression.N)Tuple) rii3i iii it$i'T)frozenc@s8eZdZUdZeed<eed<eeefdddZdS)CompressedHistogramValuezRepresents a value in a compressed histogram. Attributes: basis_point: Compression point represented in basis point, 1/100th of a percent. value: Cumulative weight at the basis point. basis_pointvalue)returncCs |j|jfS)z1Returns the basis point and the value as a tuple.)rr)selfr t/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/tensorboard/plugins/distribution/compressor.pyas_tuple*sz!CompressedHistogramValue.as_tupleN)__name__ __module__ __qualname____doc__float__annotations__rr r r r r rs rc CsZ|jsdd|DSt|j}t|j}||d|p>d}g}d}|t|kr"tj |||dd}|t|kr"||}|dkr||dnd } || kr|d7}qp|r| s|j } nt ||d|j } t |||j } t ||| || | } | t||| |d7}qNqpq"qN|t|krV| 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|dqSrr .0br r r Cz,compress_histogram_proto..?rrightZsider)numnparraybucketlist bucket_limitsumcumsumlen searchsortedminmax_lerpappendrr ) Zhistobpsr"r$weightsvaluesjir& cumsum_prevlhsrhsweightr r r compress_histogram_proto0s8   r6cCst|}|jsdd|DS|dd|dd}}|dddf}t|dddf}||d|ppd}g}d}|t|krNtj|||d d } | t|krN|| } | dkr|| dnd } | | kr| d7} q| r| s|} nt|| d|} t || |} t ||| | | | }| t ||| |d7}qqqNq|t|kr| t ||| |d7}qN|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 rxrz&compress_histogram..rrrNrrrr)r r!sizer#r%r&r'r(r*r)r+r,rr )Zbucketsr-ZminminZmaxmaxcountsZ right_edgesr.resultZbp_indexr1r&r2r3r4r5r r r compress_histogrambsD  r;cCs |||t||||S)z)Affinely map from [x0, x1] onto [y0, y1].)r)xZx0x1Zy0y1r r r r+sr+) rZ dataclassesnumpyr typingrZNORMAL_HISTOGRAM_BPSZ dataclassrr6r;r+r r r r s   2 ;