X-Git-Url: https://git.nihav.org/?p=nihav.git;a=blobdiff_plain;f=nihav-codec-support%2Fsrc%2Fcodecs%2Fimaadpcm.rs;h=c2e7fd042bb9eec7a21d09b9ae0a2e075bd0014e;hp=ac2d082aadf0d73f8a3b95eb8645ea50b69612f4;hb=4ee6b98f1975795210f654d3365aac1be0043ceb;hpb=d6c81913b6633bb6a44aad6f8ca06cee55e26660

diff --git a/nihav-codec-support/src/codecs/imaadpcm.rs b/nihav-codec-support/src/codecs/imaadpcm.rs
index ac2d082..c2e7fd0 100644
--- a/nihav-codec-support/src/codecs/imaadpcm.rs
+++ b/nihav-codec-support/src/codecs/imaadpcm.rs
@@ -56,6 +56,13 @@ impl IMAState {
         self.step = istep.max(0).min(IMA_MAX_STEP as isize) as usize;
         self.predictor as i16
     }
+    ///! Computes an encoded nibble from an input sample.
+    pub fn compress_sample(&self, sample: i16) -> u8 {
+        let diff = i32::from(sample) - self.predictor;
+        let sign = if diff >= 0 { 0 } else { 8 };
+        let nib = (diff.abs() * 4 / IMA_STEP_TABLE[self.step]).min(7) as u8;
+        nib | sign
+    }
 }
 
 impl Default for IMAState {