dranke/upkr
1
0
Fork 0
mirror of https://github.com/exoticorn/upkr.git synced 2026-01-20 11:36:42 +01:00
Code Issues Projects Releases Wiki Activity

use u8 context entries

Browse Source
This commit is contained in:
Dennis Ranke
2021-12-27 22:35:53 +01:00
parent f7f891e154
commit 2e7983fc65
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/context_state.rs
View File

@@ -6,7 +6,7 @@ const UPDATE_ADD: u32 = 8;
#[derive(Clone)] #[derive(Clone)]
pub struct ContextState { pub struct ContextState {
contexts: Vec<u16>, contexts: Vec<u8>,
} }
pub struct Context<'a> { pub struct Context<'a> {
@@ -17,7 +17,7 @@ pub struct Context<'a> {
impl ContextState { impl ContextState {
pub fn new(size: usize) -> ContextState { pub fn new(size: usize) -> ContextState {
ContextState { ContextState {
contexts: vec![INIT_PROB; size], contexts: vec![INIT_PROB as u8; size],
} }
} }
@@ -28,15 +28,15 @@ impl ContextState {
impl<'a> Context<'a> { impl<'a> Context<'a> {
pub fn prob(&self) -> u16 { pub fn prob(&self) -> u16 {
self.state.contexts[self.index] self.state.contexts[self.index] as u16
} }
pub fn update(&mut self, bit: bool) { pub fn update(&mut self, bit: bool) {
let old = self.state.contexts[self.index]; let old = self.state.contexts[self.index];
self.state.contexts[self.index] = if bit { self.state.contexts[self.index] = if bit {
old + ((ONE_PROB - old as u32 + UPDATE_ADD) >> UPDATE_RATE) as u16 old + ((ONE_PROB - old as u32 + UPDATE_ADD) >> UPDATE_RATE) as u8
} else { } else {
old - ((old + UPDATE_ADD as u16) >> UPDATE_RATE) old - ((old as u32 + UPDATE_ADD) >> UPDATE_RATE) as u8
}; };
} }
} }