ADPCM: Improved comment consistancy.

This commit is contained in:
Trek H 2019-03-07 16:08:00 +10:30
parent d9baabdf02
commit 6cfd7c5104
4 changed files with 36 additions and 36 deletions

View File

@ -46,14 +46,14 @@ func main() {
flag.StringVar(&outPath, "out", "decoded.pcm", "file path of output data") flag.StringVar(&outPath, "out", "decoded.pcm", "file path of output data")
flag.Parse() flag.Parse()
// read adpcm // Read adpcm.
comp, err := ioutil.ReadFile(inPath) comp, err := ioutil.ReadFile(inPath)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
fmt.Println("Read", len(comp), "bytes from file", inPath) fmt.Println("Read", len(comp), "bytes from file", inPath)
// decode adpcm // Decode adpcm.
inBSize := 256 inBSize := 256
numBlocks := len(comp) / inBSize numBlocks := len(comp) / inBSize
outBSize := 2 + (inBSize-4)*4 // 2 bytes are copied, 2 are used as block header info, the remaining bytes are decompressed 1:4 outBSize := 2 + (inBSize-4)*4 // 2 bytes are copied, 2 are used as block header info, the remaining bytes are decompressed 1:4
@ -67,7 +67,7 @@ func main() {
} }
} }
// save pcm to file // Save pcm to file.
err = ioutil.WriteFile(outPath, decoded.Bytes(), 0644) err = ioutil.WriteFile(outPath, decoded.Bytes(), 0644)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)

View File

@ -46,17 +46,17 @@ func main() {
flag.StringVar(&adpcmPath, "out", "encoded.adpcm", "file path of output") flag.StringVar(&adpcmPath, "out", "encoded.adpcm", "file path of output")
flag.Parse() flag.Parse()
//read pcm // Read pcm.
pcm, err := ioutil.ReadFile(inPath) pcm, err := ioutil.ReadFile(inPath)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
fmt.Println("Read", len(pcm), "bytes from file", inPath) fmt.Println("Read", len(pcm), "bytes from file", inPath)
//encode adpcm // Encode adpcm.
inBSize := 1010 inBSize := 1010
numBlocks := len(pcm) / inBSize numBlocks := len(pcm) / inBSize
outBSize := int(float64(inBSize)/4 + 3.5) // compression is 4:1 and 3.5 bytes of info are added to each block outBSize := int(float64(inBSize)/4 + 3.5) // Compression is 4:1 and 3.5 bytes of info are added to each block.
comp := bytes.NewBuffer(make([]byte, 0, outBSize*numBlocks)) comp := bytes.NewBuffer(make([]byte, 0, outBSize*numBlocks))
enc := adpcm.NewEncoder(comp) enc := adpcm.NewEncoder(comp)
for i := 0; i < numBlocks; i++ { for i := 0; i < numBlocks; i++ {
@ -67,7 +67,7 @@ func main() {
} }
} }
// save adpcm to file // Save adpcm to file.
err = ioutil.WriteFile(adpcmPath, comp.Bytes(), 0644) err = ioutil.WriteFile(adpcmPath, comp.Bytes(), 0644)
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)

View File

@ -57,13 +57,13 @@ type Decoder struct {
step int16 step int16
} }
// table of index changes (see spec) // Table of index changes (see spec).
var indexTable = []int16{ var indexTable = []int16{
-1, -1, -1, -1, 2, 4, 6, 8, -1, -1, -1, -1, 2, 4, 6, 8,
-1, -1, -1, -1, 2, 4, 6, 8, -1, -1, -1, -1, 2, 4, 6, 8,
} }
// quantize step size table (see spec) // Quantize step size table (see spec).
var stepTable = []int16{ var stepTable = []int16{
7, 8, 9, 10, 11, 12, 13, 14, 7, 8, 9, 10, 11, 12, 13, 14,
16, 17, 19, 21, 23, 25, 28, 31, 16, 17, 19, 21, 23, 25, 28, 31,
@ -97,12 +97,12 @@ func NewDecoder(dst *bytes.Buffer) *Decoder {
} }
// encodeSample takes a single 16 bit PCM sample and // encodeSample takes a single 16 bit PCM sample and
// returns a byte of which the last 4 bits are an encoded ADPCM nibble // returns a byte of which the last 4 bits are an encoded ADPCM nibble.
func (e *Encoder) encodeSample(sample int16) byte { func (e *Encoder) encodeSample(sample int16) byte {
// find difference of actual sample from encoder's prediction // Find difference of actual sample from encoder's prediction.
delta := sample - e.pred delta := sample - e.pred
// create and set sign bit for nibble and find absolute value of difference // Create and set sign bit for nibble and find absolute value of difference.
var nib byte var nib byte
if delta < 0 { if delta < 0 {
nib = 8 nib = 8
@ -123,7 +123,7 @@ func (e *Encoder) encodeSample(sample int16) byte {
step >>= 1 step >>= 1
} }
// adjust predicted sample based on calculated difference // Adjust predicted sample based on calculated difference.
if nib&8 != 0 { if nib&8 != 0 {
e.pred -= diff e.pred -= diff
} else { } else {
@ -131,7 +131,7 @@ func (e *Encoder) encodeSample(sample int16) byte {
} }
e.index += indexTable[nib&7] e.index += indexTable[nib&7]
// check for underflow and overflow // Check for underflow and overflow.
if e.index < 0 { if e.index < 0 {
e.index = 0 e.index = 0
} else if e.index > int16(len(stepTable)-1) { } else if e.index > int16(len(stepTable)-1) {
@ -142,9 +142,9 @@ func (e *Encoder) encodeSample(sample int16) byte {
} }
// decodeSample takes a byte, the last 4 bits of which contain a single // decodeSample takes a byte, the last 4 bits of which contain a single
// 4 bit ADPCM nibble, and returns a 16 bit decoded PCM sample // 4 bit ADPCM nibble, and returns a 16 bit decoded PCM sample.
func (d *Decoder) decodeSample(nibble byte) int16 { func (d *Decoder) decodeSample(nibble byte) int16 {
// calculate difference // Calculate difference.
var diff int16 var diff int16
if nibble&4 != 0 { if nibble&4 != 0 {
diff += d.step diff += d.step
@ -157,34 +157,34 @@ func (d *Decoder) decodeSample(nibble byte) int16 {
} }
diff += d.step >> 3 diff += d.step >> 3
// account for sign bit // Account for sign bit.
if nibble&8 != 0 { if nibble&8 != 0 {
diff = -diff diff = -diff
} }
// adjust predicted sample based on calculated difference // Adjust predicted sample based on calculated difference.
d.pred += diff d.pred += diff
// adjust index into step size lookup table using nibble // Adjust index into step size lookup table using nibble.
d.index += indexTable[nibble] d.index += indexTable[nibble]
// check for overflow and underflow // Check for overflow and underflow.
if d.index < 0 { if d.index < 0 {
d.index = 0 d.index = 0
} else if d.index > int16(len(stepTable)-1) { } else if d.index > int16(len(stepTable)-1) {
d.index = int16(len(stepTable) - 1) d.index = int16(len(stepTable) - 1)
} }
// find new quantizer step size // Find new quantizer step size.
d.step = stepTable[d.index] d.step = stepTable[d.index]
return d.pred return d.pred
} }
// calcHead sets the state for the encoder by running the first sample through // calcHead sets the state for the encoder by running the first sample through
// the encoder, and writing the first sample // the encoder, and writing the first sample.
func (e *Encoder) calcHead(sample []byte) error { func (e *Encoder) calcHead(sample []byte) error {
// check that we are given 1 16-bit sample (2 bytes) // Check that we are given 1 16-bit sample (2 bytes).
sampSize := 2 sampSize := 2
if len(sample) != sampSize { if len(sample) != sampSize {
return fmt.Errorf("length of given byte array is: %v, expected: %v", len(sample), sampSize) return fmt.Errorf("length of given byte array is: %v, expected: %v", len(sample), sampSize)
@ -202,10 +202,10 @@ func (e *Encoder) calcHead(sample []byte) error {
// EncodeBlock takes a slice of 1010 bytes (505 16-bit PCM samples). // EncodeBlock takes a slice of 1010 bytes (505 16-bit PCM samples).
// It outputs encoded (compressed) bytes (each byte containing two ADPCM nibbles) to the encoder's dest writer. // It outputs encoded (compressed) bytes (each byte containing two ADPCM nibbles) to the encoder's dest writer.
// note: nibbles are output in little endian order, eg. n1n0 n3n2 n5n4... // Note: nibbles are output in little endian order, eg. n1n0 n3n2 n5n4...
// note: first 4 bytes are for initializing the decoder before decoding a block. // Note: first 4 bytes are for initializing the decoder before decoding a block.
// - first two bytes contain the first 16-bit sample uncompressed, // - First two bytes contain the first 16-bit sample uncompressed.
// - third byte is the decoder's starting index for the block, the fourth is padding and ignored // - Third byte is the decoder's starting index for the block, the fourth is padding and ignored.
func (e *Encoder) EncodeBlock(block []byte) error { func (e *Encoder) EncodeBlock(block []byte) error {
bSize := 1010 bSize := 1010
if len(block) != bSize { if len(block) != bSize {
@ -234,14 +234,14 @@ func (d *Decoder) DecodeBlock(block []byte) error {
return fmt.Errorf("unsupported block size. Given: %v, expected: %v", len(block), bSize) return fmt.Errorf("unsupported block size. Given: %v, expected: %v", len(block), bSize)
} }
// initialize decoder with first 4 bytes of the block // Initialize decoder with first 4 bytes of the block.
d.pred = int16(binary.LittleEndian.Uint16(block[0:2])) d.pred = int16(binary.LittleEndian.Uint16(block[0:2]))
d.index = int16(block[2]) d.index = int16(block[2])
d.step = stepTable[d.index] d.step = stepTable[d.index]
d.dest.Write(block[0:2]) d.dest.Write(block[0:2])
// for each byte, seperate it into two nibbles (each nibble is a compressed sample), // For each byte, seperate it into two nibbles (each nibble is a compressed sample),
// then decode each nibble and output the resulting 16-bit samples // then decode each nibble and output the resulting 16-bit samples.
for i := 4; i < bSize; i++ { for i := 4; i < bSize; i++ {
twoNibs := block[i] twoNibs := block[i]
nib2 := byte(twoNibs >> 4) nib2 := byte(twoNibs >> 4)

View File

@ -37,13 +37,13 @@ import (
// TestEncodeBlock will read PCM data, encode it in blocks and generate ADPCM // TestEncodeBlock will read PCM data, encode it in blocks and generate ADPCM
// then compare the result with expected ADPCM. // then compare the result with expected ADPCM.
func TestEncodeBlock(t *testing.T) { func TestEncodeBlock(t *testing.T) {
//read input pcm // Read input pcm.
pcm, err := ioutil.ReadFile("../../../test/test-data/av/input/raw-voice.pcm") pcm, err := ioutil.ReadFile("../../../test/test-data/av/input/raw-voice.pcm")
if err != nil { if err != nil {
t.Errorf("Unable to read input PCM file: %v", err) t.Errorf("Unable to read input PCM file: %v", err)
} }
//encode adpcm // Encode adpcm.
inBSize := 1010 inBSize := 1010
numBlocks := len(pcm) / inBSize numBlocks := len(pcm) / inBSize
outBSize := int(float64(inBSize)/4 + 3.5) // compression is 4:1 and 3.5 bytes of info are added to each block outBSize := int(float64(inBSize)/4 + 3.5) // compression is 4:1 and 3.5 bytes of info are added to each block
@ -57,7 +57,7 @@ func TestEncodeBlock(t *testing.T) {
} }
} }
//read expected adpcm file // Read expected adpcm file.
exp, err := ioutil.ReadFile("../../../test/test-data/av/output/encoded-voice.adpcm") exp, err := ioutil.ReadFile("../../../test/test-data/av/output/encoded-voice.adpcm")
if err != nil { if err != nil {
t.Errorf("Unable to read expected ADPCM file: %v", err) t.Errorf("Unable to read expected ADPCM file: %v", err)
@ -71,13 +71,13 @@ func TestEncodeBlock(t *testing.T) {
// TestDecodeBlock will read encoded ADPCM, decode it in blocks and then compare the // TestDecodeBlock will read encoded ADPCM, decode it in blocks and then compare the
// resulting PCM with the expected decoded PCM. // resulting PCM with the expected decoded PCM.
func TestDecodeBlock(t *testing.T) { func TestDecodeBlock(t *testing.T) {
//read adpcm // Read adpcm.
comp, err := ioutil.ReadFile("../../../test/test-data/av/input/encoded-voice.adpcm") comp, err := ioutil.ReadFile("../../../test/test-data/av/input/encoded-voice.adpcm")
if err != nil { if err != nil {
t.Errorf("Unable to read input ADPCM file: %v", err) t.Errorf("Unable to read input ADPCM file: %v", err)
} }
//decode adpcm // Decode adpcm.
inBSize := 256 inBSize := 256
numBlocks := len(comp) / inBSize numBlocks := len(comp) / inBSize
outBSize := 2 + (inBSize-4)*4 // 2 bytes are copied, 2 are used as block header info, the remaining bytes are decompressed 1:4 outBSize := 2 + (inBSize-4)*4 // 2 bytes are copied, 2 are used as block header info, the remaining bytes are decompressed 1:4
@ -91,7 +91,7 @@ func TestDecodeBlock(t *testing.T) {
} }
} }
//read expected pcm file // Read expected pcm file.
exp, err := ioutil.ReadFile("../../../test/test-data/av/output/decoded-voice.pcm") exp, err := ioutil.ReadFile("../../../test/test-data/av/output/decoded-voice.pcm")
if err != nil { if err != nil {
t.Errorf("Unable to read expected PCM file: %v", err) t.Errorf("Unable to read expected PCM file: %v", err)