def base.leaf

bend-parallel@0.1.0.0 · sort.bend:204

import Base
import bend-parallel@0.1.0.0/histogram.bend as Hist

204# Lane i: rows group i / 2^gc, columns group i % 2^gc (as Hist.grp.leaf).
205def base.leaf(+i: U32, +gc: Nat, +w: Nat, +m: Nat, +kb: Nat, +k: U32, p: Hist.Two) -> Hist.Two:
206  +g = U32.shrn(i, gc)
207  +j = (i - U32.shln(g, gc) : U32)
208  +last = (U32.shln(g, Nat.add(m, kb)) + U32.shln((U32.shln(1, m) - 1 : U32), kb) : U32)
209  base.cols(U32.to_nat((U32.shln(1, w) - 1 : U32)), U32.shln(j, w), k, last, U32.shln(1, kb),
210    U32.to_nat((U32.shln(1, m) - 1 : U32)), p)