uYfO d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZ G d d ej e Z G d de Z
d Z G d
d e Z G d d
e Z
y) )division)ABCMeta)abstractmethodNc T e Zd ZdZd dZed Zed Zed Z d Z
d Z y)
KeyMappinga3
Args:
relative_accuracy (float): the accuracy guarantee; referred to as alpha
in the paper. (0. < alpha < 1.)
offset (float): an offset that can be used to shift all bin keys
Attributes:
gamma (float): the base for the exponential buckets. gamma = (1 + alpha) / (1 - alpha)
min_possible: the smallest value the sketch can distinguish from 0
max_possible: the largest value the sketch can handle
_multiplier (float): used for calculating log_gamma(value) initially, _multiplier = 1 / log(gamma)
c d |dk s|dk\ rt d|z || _ || _ d|z d|z
z }d|z | _ dt j
| z | _ t j j | j z | _
t j j | j z | _ y )Nr z1Relative accuracy must be between 0 and 1, got %r )
ValueErrorrelative_accuracy_offsetgammamathlog1p _multipliersys
float_infomin min_possiblemax max_possible)selfr offsetgamma_mantissas P/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/ddsketch/mapping.py__init__zKeyMapping.__init__+ s !%6!%; CFWW
"3 ..!6G2GH '
tzz. 99 NN . . ; NN . . ; c , |dz
|dz z } | || S )z Constructor used by pb.proto ?r )clsr r r s r from_gamma_offsetz KeyMapping.from_gamma_offset: s% #S[US[ 9$V44r c y)zBReturn (an approximation of) the logarithm of the value base gammaNr! r values r
_log_gammazKeyMapping._log_gammaA r c y)z5Return (an approximation of) gamma to the power valueNr! r% s r
_pow_gammazKeyMapping._pow_gammaF r( r c v t t j | j | | j z S )z{
Args:
value (float)
Returns:
int: the key specifying the bucket for value
)intr ceilr' r
r% s r keyzKeyMapping.keyK s* 499T__U 34t||CDDr c d | j || j z
dd| j z z z S )z
Args:
key (int)
Returns:
float: the value represented by the bucket specified by the key
@r )r* r
r )r r. s r r& zKeyMapping.valueU s. sT\\ 12cQ^6LMMr Ng ) __name__
__module__ __qualname____doc__r classmethodr# r r' r* r. r& r! r r r r sX
< 5 5 Q Q D DENr r c 0 e Zd ZdZd fd Zd Zd Z xZS )LogarithmicMappingzA memory-optimal KeyMapping, i.e., given a targeted relative accuracy, it
requires the least number of keys to cover a given range of values. This is
done by logarithmically mapping floating-point values to integers.
c t t | || | xj t j
d z c_ y )Nr r
)superr8 r r r logr r r __class__s r r zLogarithmicMapping.__init__f s2
$01B60R DHHQK'r c H t j |d | j z S )Nr
)r r; r r% s r r' z LogarithmicMapping._log_gammak s xxq!D$4$444r c H t j d|| j z S )Nr0 )r powr r% s r r* z LogarithmicMapping._pow_gammao s xxUT%5%5 566r r1 )r2 r3 r4 r5 r r' r*
__classcell__r= s @r r8 r8 ` s
(
57r r8 c D t t | dz }| dk r| S |S )NgUUUUUU?r )floatabs)xys r _cbrtrH t s(
c!f#$A1ur Hr c ( e Zd ZdZd Zd Zd Zd Zy)LinearlyInterpolatedMappinga A fast KeyMapping that approximates the memory-optimal
LogarithmicMapping by extracting the floor value of the logarithm to the
base 2 from the binary representations of floating-point values and
linearly interpolating the logarithm in-between.
c R t j | \ }}d|z dz
}||dz
z S )zApproximates log2 by s + f
where v = (s+1) * 2 ** f for s in [0, 1)
frexp(v) returns m and e s.t.
v = m * 2 ** e ; (m in [0.5, 1) or 0.0)
so we adjust m and e accordingly
r
r )r frexpr r& mantissaexponent significands r _log2_approxz(LinearlyInterpolatedMapping._log2_approx s4 "ZZ .((lQ& hl++r c t t j | dz }||z
dz dz }t j || S )zInverse of _log2_approxr r
r0 )r, r floorldexp)r r& rO rN s r _exp2_approxz(LinearlyInterpolatedMapping._exp2_approx sA tzz%(1,-H$q(C/zz(H--r c > | j | | j z S N)rQ r r% s r r' z&LinearlyInterpolatedMapping._log_gamma s '$*:*:::r c > | j || j z S rW )rU r r% s r r* z&LinearlyInterpolatedMapping._pow_gamma s )9)9!9::r N)r2 r3 r4 r5 rQ rU r' r* r! r r rJ rJ | s ,.;;r rJ c H e Zd ZdZdZdZdZd
fd Zd Zd Z d Z
d Z xZ S ) CubicallyInterpolatedMappinga A fast KeyMapping that approximates the memory-optimal LogarithmicMapping by
extracting the floor value of the logarithm to the base 2 from the binary
representations of floating-point values and cubically interpolating the
logarithm in-between.
More detailed documentation of this method can be found in:
sketches-java
g__?g333333gm۶m?c n t t | || | xj | j z c_ y )Nr )r: rZ r r Cr< s r r z%CubicallyInterpolatedMapping.__init__ s4
*D: f ;
DFF"r c t j | \ }}d|z dz
}| j |z | j z |z | j z |z |dz
z S )z*Approximates log2 using a cubic polynomialr
r )r rL ABr\ rM s r _cubic_log2_approxz/CubicallyInterpolatedMapping._cubic_log2_approx sa "ZZ .((lQ&
VVk
!DFF
*k 9DFF B #a<) )r c j t t j | }| j | j z d| j z | j
z z
}d| j z | j z | j z d| j z | j z | j
z z
d| j z | j z ||z
z z
}t
|||z d|z |z |z z
dz z
dz }| j |z ||z z d| j z z dz }|d z }t j ||d
z S ) N r0 g "@g ;@ g ?g @r r
r )r, r rS r_ r^ r\ rH rT )r r& rO delta_0delta_1cardanosignificand_plus_onerN s r _cubic_exp2_approxz/CubicallyInterpolatedMapping._cubic_exp2_approx s2 tzz%()&&466/AJ$77 $&&L466 !DFF *DFFlTVV#dff,
-TVVmdff$(89
:
7*Q[7-BW-LLQTT
U
ffw 7!22 3sTVV| Ds J (!+zz(HqL11r c > | j | | j z S rW )r` r r% s r r' z'CubicallyInterpolatedMapping._log_gamma s &&u-0@0@@@r c > | j || j z S rW )rh r r% s r r* z'CubicallyInterpolatedMapping._pow_gamma s &&ut/?/?'?@@r r1 )
r2 r3 r4 r5 r^ r_ r\ r r` rh r' r* rA rB s @r rZ rZ s4 AAA#)2(AAr rZ )
__future__r abcr r r r sixwith_metaclassr r8 rH rJ rZ r! r r ro sg
?N###G, ?ND7 7(
!;* !;H8A: 8Ar