wp
This commit is contained in:
@@ -2,22 +2,49 @@ package chunkenc
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"math"
|
||||
|
||||
"gordenko.dev/dima/qb/bin"
|
||||
"gordenko.dev/dima/qb/conbuf"
|
||||
)
|
||||
|
||||
// REVERSE
|
||||
// COMPRESSOR REVERSE
|
||||
|
||||
/*
|
||||
Формат:
|
||||
base value (var u64)
|
||||
delta (var u64)
|
||||
q (qty of previous deltas; msb=0 - series, msb=1 - non series, low 7 bits - qty 1..128)
|
||||
delta
|
||||
delta
|
||||
delta
|
||||
q
|
||||
|
||||
Дельта рахується від base value.
|
||||
Декодування у зворотньому порядку.
|
||||
|
||||
Після base value слідують run або literal блоки.
|
||||
Run блок - це delta + header byte в кінці.
|
||||
Literal блок - це від одної до N дельт + header byte в кінці.
|
||||
Спочатку створюється literal блок.
|
||||
Якщо для останної дельти додається дублікат, literal блок модифікується -
|
||||
лічильник зменшується до 1. А остання дельта переміщюється в новий run блок.
|
||||
Причому лічильник 0 - означає 2 елементи. Приклад:
|
||||
До:
|
||||
v1 v2 v3 h-byte(literal, 3) <- v3
|
||||
Після:
|
||||
v1 v2 h-byte(literal, 2) v3 h-byte(run, 2)
|
||||
*/
|
||||
|
||||
type ReverseCumulativeDeltaCompressor struct {
|
||||
buf *conbuf.ContinuousBuffer
|
||||
coef float64
|
||||
pos int
|
||||
firstValue float64
|
||||
lastDelta uint64
|
||||
length uint16
|
||||
numIdx int
|
||||
buf *conbuf.ContinuousBuffer
|
||||
coef float64
|
||||
pos int
|
||||
baseValue float64
|
||||
lastDelta uint64
|
||||
lastDeltaSize int
|
||||
h byte
|
||||
state *CumulativeDeltaBound
|
||||
}
|
||||
|
||||
func NewReverseCumulativeDeltaCompressor(buf *conbuf.ContinuousBuffer, size int, fracDigits byte) *ReverseCumulativeDeltaCompressor {
|
||||
@@ -27,236 +54,166 @@ func NewReverseCumulativeDeltaCompressor(buf *conbuf.ContinuousBuffer, size int,
|
||||
}
|
||||
s := &ReverseCumulativeDeltaCompressor{
|
||||
buf: buf,
|
||||
pos: size,
|
||||
pos: size, // перший вільний байт
|
||||
coef: coef,
|
||||
}
|
||||
if size > 0 {
|
||||
s.restoreState()
|
||||
u64, _, err := s.buf.GetVarUint64(0)
|
||||
if err != nil {
|
||||
log.Fatalf("bug: get base value: %s", err)
|
||||
}
|
||||
s.baseValue = float64(u64) / s.coef
|
||||
s.h = s.buf.GetByte(s.pos - 1)
|
||||
s.lastDelta, s.lastDeltaSize, err = s.buf.ReverseGetVarUint64(s.pos - 2)
|
||||
if err != nil {
|
||||
log.Fatalf("bug: get last delta: %s", err)
|
||||
}
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) restoreState() {
|
||||
u64, n, err := s.buf.GetVarUint64(0)
|
||||
if err != nil {
|
||||
panic(fmt.Sprintf("bug: get first value: %s", err))
|
||||
}
|
||||
s.firstValue = float64(u64) / s.coef
|
||||
|
||||
if s.pos > n {
|
||||
pos := s.pos - 1
|
||||
idxOf8 := uint(8 - s.buf.GetByte(pos))
|
||||
pos--
|
||||
s8 := s.buf.GetByte(pos)
|
||||
pos--
|
||||
|
||||
var n int
|
||||
s.lastDelta, n, err = s.buf.ReverseGetVarUint64(pos)
|
||||
if err != nil {
|
||||
panic(fmt.Sprintf("bug: get last delta: %s", err))
|
||||
}
|
||||
pos -= n
|
||||
s.numIdx = pos + 1
|
||||
|
||||
var flag byte = 1 << idxOf8
|
||||
if (s8 & flag) == flag {
|
||||
s.length, _ = s.buf.DecodeRunLength(pos)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) Size() int {
|
||||
return s.pos
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) CalcRequiredSpace(value float64) int {
|
||||
if s.pos == 0 {
|
||||
n := bin.CalcVarUint64Length(uint64(value * s.coef))
|
||||
return n + 3
|
||||
}
|
||||
delta := uint64((value-s.firstValue)*s.coef + eps)
|
||||
if delta == s.lastDelta {
|
||||
if s.length == 0 {
|
||||
return 1
|
||||
} else {
|
||||
newLength := s.length + 1
|
||||
if newLength < 130 {
|
||||
return 0
|
||||
} else if newLength == 130 {
|
||||
return 1
|
||||
} else {
|
||||
if newLength < 32769 {
|
||||
return 0
|
||||
} else {
|
||||
n := bin.CalcVarUint64Length(delta)
|
||||
n += 2
|
||||
s8q := s.buf.GetByte(s.pos - 1)
|
||||
if s8q == 8 {
|
||||
n -= 1
|
||||
} else {
|
||||
n -= 2
|
||||
}
|
||||
return n
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
n := bin.CalcVarUint64Length(delta)
|
||||
n += 2
|
||||
s8q := s.buf.GetByte(s.pos - 1)
|
||||
if s8q == 8 {
|
||||
n -= 1
|
||||
} else {
|
||||
n -= 2
|
||||
}
|
||||
return n
|
||||
}
|
||||
}
|
||||
//func (s *ReverseCumulativeDeltaCompressor) CalcRequiredSpace(value float64) int {
|
||||
//}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) Append(value float64) {
|
||||
if s.pos == 0 {
|
||||
n := s.buf.PutVarUint64(s.pos, uint64(value*s.coef))
|
||||
s.pos += n
|
||||
s.firstValue = value
|
||||
s.encodeNewDelta(0, 0, 1)
|
||||
// base value
|
||||
s.pos += s.buf.PutVarUint64(s.pos, uint64(value*s.coef))
|
||||
s.baseValue = value
|
||||
s.appendNewLiteral(0)
|
||||
} else {
|
||||
delta := uint64((value-s.firstValue)*s.coef + eps)
|
||||
delta := uint64((value-s.baseValue)*s.coef + eps)
|
||||
if delta == s.lastDelta {
|
||||
if s.length == 0 {
|
||||
s.length = 2
|
||||
s.shiftOnePosToRight()
|
||||
s.buf.SetByte(s.numIdx-1, 0)
|
||||
s8q := s.buf.GetByte(s.pos - 1)
|
||||
s.buf.SetFlag(s.pos-2, 1<<(8-s8q))
|
||||
} else {
|
||||
s.length++
|
||||
if s.length < 130 {
|
||||
s.buf.SetByte(s.numIdx-1, byte(s.length-2))
|
||||
} else if s.length == 130 {
|
||||
s.shiftOnePosToRight()
|
||||
s.encode2bLength()
|
||||
if s.h < 128 {
|
||||
// run блок - отже треба збільшити лічильник
|
||||
if s.h < 127 {
|
||||
// increase counter
|
||||
s.h++
|
||||
s.buf.SetByte(s.pos-1, s.h)
|
||||
} else {
|
||||
if s.length < 32769 {
|
||||
s.encode2bLength()
|
||||
} else {
|
||||
s.appendNewDelta(delta)
|
||||
}
|
||||
// не можу збільшити - буде переповнення. Додаю новий literal блок
|
||||
// counter overflow
|
||||
s.appendNewLiteral(delta)
|
||||
}
|
||||
} else {
|
||||
// literal блок.
|
||||
// Якщо в ньому лише одне значення - перетворюю його на run блок.
|
||||
// Інакше забираю останнє значення із literal блока і додаю новий run блок.
|
||||
q := s.h & 127
|
||||
if q == 0 { // 1 кодується як 0
|
||||
s.convertLiteralToRun()
|
||||
} else {
|
||||
// забираю останнє значення із literal блоку щоб зробити run блок
|
||||
s.convertLastFromLiteralToRun()
|
||||
}
|
||||
}
|
||||
} else {
|
||||
s.appendNewDelta(delta)
|
||||
if s.h < 127 { // fix 128?
|
||||
// end of run
|
||||
s.appendNewLiteral(delta)
|
||||
} else {
|
||||
if s.h < 255 {
|
||||
// encode value from pos - 1, then append h byte
|
||||
s.appendDeltaToLiteral(delta)
|
||||
} else {
|
||||
// overflowed - encode new
|
||||
s.appendNewLiteral(delta)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) appendNewDelta(delta uint64) {
|
||||
s.length = 0
|
||||
|
||||
s8 := s.buf.GetByte(s.pos - 2)
|
||||
s8q := s.buf.GetByte(s.pos - 1)
|
||||
|
||||
if s8q == 8 {
|
||||
s.pos -= 1
|
||||
s8 = 0
|
||||
s8q = 1
|
||||
} else {
|
||||
s.pos -= 2
|
||||
s8q++
|
||||
}
|
||||
|
||||
s.encodeNewDelta(delta, s8, s8q)
|
||||
func (s *ReverseCumulativeDeltaCompressor) convertLastFromLiteralToRun() {
|
||||
// Зменшую кількість елементів в literal блоці
|
||||
s.h--
|
||||
s.pos -= 1 + s.lastDeltaSize
|
||||
s.buf.SetByte(s.pos, s.h) // закриваю literal блок
|
||||
s.pos++
|
||||
s.lastDeltaSize = s.buf.ReversePutVarUint64(s.pos, s.lastDelta)
|
||||
s.pos += s.lastDeltaSize
|
||||
s.h = 0 // run блок, довжини 2
|
||||
s.buf.SetByte(s.pos, s.h)
|
||||
s.pos++
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) encodeNewDelta(delta uint64, s8 byte, s8q byte) {
|
||||
func (s *ReverseCumulativeDeltaCompressor) convertLiteralToRun() {
|
||||
// Знімаю flagLiteral, а лічильник 0 дорівнює 2 елементам в серії.
|
||||
s.h = 0
|
||||
s.buf.SetByte(s.pos-1, s.h)
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) appendDeltaToLiteral(delta uint64) {
|
||||
s.h++ // збільшую к-сть дельт
|
||||
s.lastDelta = delta
|
||||
s.numIdx = s.pos
|
||||
n := s.buf.ReversePutVarUint64(s.pos, s.lastDelta)
|
||||
s.pos += n
|
||||
s.buf.SetByte(s.pos, s8)
|
||||
s.pos++
|
||||
s.buf.SetByte(s.pos, s8q)
|
||||
s.pos--
|
||||
s.lastDeltaSize = s.buf.ReversePutVarUint64(s.pos, delta)
|
||||
s.pos += s.lastDeltaSize
|
||||
s.buf.SetByte(s.pos, s.h)
|
||||
s.pos++
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) shiftOnePosToRight() {
|
||||
s.buf.ShiftOnePosToRight(s.numIdx, s.pos)
|
||||
func (s *ReverseCumulativeDeltaCompressor) appendNewLiteral(delta uint64) {
|
||||
s.h = flagLiteral
|
||||
s.lastDelta = delta
|
||||
s.lastDeltaSize = s.buf.ReversePutVarUint64(s.pos, delta)
|
||||
s.pos += s.lastDeltaSize
|
||||
s.buf.SetByte(s.pos, flagLiteral) // literal, length = 1
|
||||
s.pos++
|
||||
s.numIdx++
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) encode2bLength() {
|
||||
num := s.length - 2
|
||||
s.buf.SetByte(s.numIdx-1, byte(num&127)|128)
|
||||
s.buf.SetByte(s.numIdx-2, byte(num>>7))
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) DeleteLast() {
|
||||
var (
|
||||
s8q = s.buf.GetByte(s.pos - 1)
|
||||
s8 = s.buf.GetByte(s.pos - 2)
|
||||
flag byte = 1 << uint(8-s8q)
|
||||
)
|
||||
|
||||
if s.length > 0 {
|
||||
if s.length == 2 {
|
||||
s.length = 0
|
||||
s.buf.UnsetFlag(s.pos-2, flag)
|
||||
s.buf.ShiftOnePosToLeft(s.numIdx, s.pos)
|
||||
s.numIdx--
|
||||
s.pos--
|
||||
} else if s.length < 130 {
|
||||
s.length--
|
||||
s.buf.SetByte(s.numIdx-1, byte(s.length)-2)
|
||||
} else if s.length == 130 {
|
||||
s.length--
|
||||
s.buf.ShiftOnePosToLeft(s.numIdx, s.pos)
|
||||
s.numIdx--
|
||||
s.pos--
|
||||
s.buf.SetByte(s.numIdx-1, byte(s.length)-2)
|
||||
} else {
|
||||
s.length--
|
||||
s.encode2bLength()
|
||||
}
|
||||
} else {
|
||||
if s8q > 1 {
|
||||
s8q--
|
||||
flag = 1 << uint(8-s8q)
|
||||
s.pos = s.numIdx + 2
|
||||
s.buf.SetByte(s.pos-2, s8)
|
||||
s.buf.SetByte(s.pos-1, s8q)
|
||||
} else {
|
||||
s.pos = s.numIdx + 1
|
||||
s.buf.SetByte(s.pos-1, 8)
|
||||
s8 = s.buf.GetByte(s.pos - 2)
|
||||
flag = 1
|
||||
}
|
||||
var (
|
||||
pos = s.pos - 3
|
||||
n int
|
||||
err error
|
||||
)
|
||||
s.lastDelta, n, err = s.buf.ReverseGetVarUint64(pos)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
s.numIdx = pos - n
|
||||
if (s8 & flag) == flag {
|
||||
s.length, _ = s.buf.DecodeRunLength(s.numIdx - 1)
|
||||
}
|
||||
}
|
||||
|
||||
type CumulativeDeltaBound struct {
|
||||
Pos int
|
||||
H byte
|
||||
LastDelta uint64
|
||||
Chunks [][]byte
|
||||
}
|
||||
|
||||
// delta h
|
||||
func (s *ReverseCumulativeDeltaCompressor) Lock() {
|
||||
s.state = &CumulativeDeltaBound{
|
||||
Pos: s.pos - 1 - s.lastDeltaSize,
|
||||
H: s.h,
|
||||
LastDelta: s.lastDelta,
|
||||
Chunks: s.buf.Chunks(),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) CreateDecompressor(fracDigits byte) *ReverseCumulativeDeltaDecompressor {
|
||||
if s.state == nil {
|
||||
d := NewReverseCumulativeDeltaDecompressor(s.buf, s.Size(), fracDigits)
|
||||
d.RestoreFromEnd()
|
||||
return d
|
||||
}
|
||||
d := NewReverseCumulativeDeltaDecompressor(s.buf, s.Size(), fracDigits)
|
||||
d.RestoreFromBound(*s.state)
|
||||
return d
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaCompressor) Unlock() {
|
||||
s.state = nil
|
||||
}
|
||||
|
||||
// DECOMPRESSOR
|
||||
|
||||
type ReverseCumulativeDeltaDecompressor struct {
|
||||
buf *conbuf.ContinuousBuffer
|
||||
pos int
|
||||
bound int
|
||||
firstValue float64
|
||||
lastValue float64
|
||||
length uint16
|
||||
coef float64
|
||||
idxOf8 uint
|
||||
s8 byte
|
||||
step byte
|
||||
buf *conbuf.ContinuousBuffer
|
||||
coef float64
|
||||
pos int
|
||||
bound int
|
||||
baseValue float64
|
||||
lastValue float64
|
||||
isRun bool
|
||||
pending int
|
||||
done bool
|
||||
}
|
||||
|
||||
func NewReverseCumulativeDeltaDecompressor(buf *conbuf.ContinuousBuffer, size int, fracDigits byte) *ReverseCumulativeDeltaDecompressor {
|
||||
@@ -264,67 +221,91 @@ func NewReverseCumulativeDeltaDecompressor(buf *conbuf.ContinuousBuffer, size in
|
||||
if fracDigits > 0 {
|
||||
coef = math.Pow(10, float64(fracDigits))
|
||||
}
|
||||
return &ReverseCumulativeDeltaDecompressor{
|
||||
buf: buf,
|
||||
coef: coef,
|
||||
pos: size,
|
||||
u64, n, err := buf.GetVarUint64(0)
|
||||
if err != nil {
|
||||
log.Fatalf("bug: get base value: %s", err)
|
||||
}
|
||||
//fmt.Printf("baseValue: %.2f, bound: %d, pos: %d\n", float64(u64)/coef, n, size)
|
||||
return &ReverseCumulativeDeltaDecompressor{
|
||||
buf: buf,
|
||||
coef: coef,
|
||||
pos: size - 1, // last elem
|
||||
baseValue: float64(u64) / coef,
|
||||
bound: n,
|
||||
}
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaDecompressor) RestoreFromEnd() {
|
||||
if s.pos > s.bound {
|
||||
// читаю заголовок наступної серії
|
||||
s.readHeader()
|
||||
s.readValue()
|
||||
} else {
|
||||
s.done = true
|
||||
}
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaDecompressor) RestoreFromBound(bound CumulativeDeltaBound) {
|
||||
s.pos = bound.Pos - 1
|
||||
s.lastValue = s.baseValue + float64(bound.LastDelta)
|
||||
s.decodeHeaderByte(bound.H)
|
||||
fmt.Printf("restore from bound: isRun=%t, pending=%d\n", s.isRun, s.pending)
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaDecompressor) NextValue() (value float64, done bool) {
|
||||
if s.step > 0 {
|
||||
if s.length > 0 {
|
||||
s.length--
|
||||
return s.lastValue, false
|
||||
}
|
||||
if s.pos < s.bound {
|
||||
return 0, true
|
||||
}
|
||||
if s.idxOf8 == 0 {
|
||||
s.s8 = s.buf.GetByte(s.pos)
|
||||
s.pos--
|
||||
}
|
||||
s.readVar()
|
||||
if s.length > 0 {
|
||||
s.length--
|
||||
}
|
||||
return s.lastValue, false
|
||||
//fmt.Printf("NextValue(): bound: %d, pos: %d, pending: %d\n", s.bound, s.pos, s.pending)
|
||||
if s.done {
|
||||
return 0, true
|
||||
}
|
||||
u64, n, err := s.buf.GetVarUint64(0)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
// повертаю значення, що було прочитано в методі RestoreFromBound/RestoreFromEnd
|
||||
value = s.lastValue
|
||||
s.pending--
|
||||
if s.pending > 0 {
|
||||
// якщо в серії залишаються елементи
|
||||
if !s.isRun {
|
||||
s.readValue()
|
||||
}
|
||||
} else if s.pos > s.bound {
|
||||
// в серії більше немає елементів, отже перевіряє чи є ще дані в буфері.
|
||||
// дані є - читаю заголовок наступної серії
|
||||
s.readHeader()
|
||||
s.readValue()
|
||||
} else {
|
||||
s.done = true
|
||||
}
|
||||
s.firstValue = float64(u64) / s.coef
|
||||
s.bound = n
|
||||
s.pos--
|
||||
s.idxOf8 = uint(8 - s.buf.GetByte(s.pos))
|
||||
s.pos--
|
||||
s.s8 = s.buf.GetByte(s.pos)
|
||||
s.pos--
|
||||
s.readVar()
|
||||
if s.length > 0 {
|
||||
s.length--
|
||||
}
|
||||
s.step = 1
|
||||
return s.lastValue, false
|
||||
// серія завершена - перевіряю чи є ще серії
|
||||
return value, false
|
||||
}
|
||||
|
||||
func (s *ReverseCumulativeDeltaDecompressor) readVar() {
|
||||
func (s *ReverseCumulativeDeltaDecompressor) readHeader() {
|
||||
h := s.buf.GetByte(s.pos)
|
||||
s.pos--
|
||||
s.decodeHeaderByte(h)
|
||||
// fmt.Println("h:", h)
|
||||
// fmt.Println("isRun:", s.isRun)
|
||||
// fmt.Println("pending:", s.pending)
|
||||
}
|
||||
func (s *ReverseCumulativeDeltaDecompressor) readValue() {
|
||||
u64, n, err := s.buf.ReverseGetVarUint64(s.pos)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
log.Fatalln(err)
|
||||
}
|
||||
// fmt.Println()
|
||||
// fmt.Println("read from pos:", s.pos)
|
||||
// fmt.Println("read delta:", u64)
|
||||
// fmt.Println("read delta n:", n)
|
||||
// fmt.Println()
|
||||
s.pos -= n
|
||||
s.lastValue = s.firstValue + float64(u64)/s.coef
|
||||
s.lastValue = s.baseValue + float64(u64)/s.coef
|
||||
}
|
||||
|
||||
var flag byte = 1 << s.idxOf8
|
||||
if (s.s8 & flag) == flag {
|
||||
s.length, n = s.buf.DecodeRunLength(s.pos)
|
||||
s.pos -= n
|
||||
}
|
||||
if s.idxOf8 == 7 {
|
||||
s.idxOf8 = 0
|
||||
func (s *ReverseCumulativeDeltaDecompressor) decodeHeaderByte(h byte) {
|
||||
s.isRun = h < 128
|
||||
if s.isRun {
|
||||
s.pending = int(h&127) + 2
|
||||
} else {
|
||||
s.idxOf8++
|
||||
s.pending = int(h&127) + 1
|
||||
}
|
||||
}
|
||||
|
||||
//
|
||||
|
||||
Reference in New Issue
Block a user