From f628b40dc66467db2e299f115ea9bac75a297900 Mon Sep 17 00:00:00 2001
From: Arda Serdar Pektezol <1669855+pektezol@users.noreply.github.com>
Date: Fri, 11 Nov 2022 08:59:19 +0300
Subject: add ReadBitsToSlice() function

---
 bitreader.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

(limited to 'bitreader.go')

diff --git a/bitreader.go b/bitreader.go
index b9ac772..5c7dc04 100644
--- a/bitreader.go
+++ b/bitreader.go
@@ -152,6 +152,29 @@ func (reader *ReaderType) TryReadStringLen(length int) string {
 	return text
 }
 
+// TryReadBytesToSlice is a wrapper function that reads the specified amount of bits
+// from the parameter and puts each bit into a slice and returns this slice.
+func (reader *ReaderType) TryReadBitsToSlice(bits int) []byte {
+	bytes := (bits / 8) + (bits % 8)
+	out := make([]byte, bytes)
+	for i := 0; i < bytes; i++ {
+		if bits/8 > 0 {
+			val, err := reader.ReadBytes(1)
+			if err != nil {
+				panic(err)
+			}
+			out[i] = byte(val)
+		} else { // Not enough to fill a whole byte
+			val, err := reader.ReadBits(bits % 8)
+			if err != nil {
+				panic(err)
+			}
+			out[i] = byte(val)
+		}
+	}
+	return out
+}
+
 // TryReadBytesToSlice is a wrapper function that reads the specified amount of bytes
 // from the parameter and puts each byte into a slice and returns this slice.
 func (reader *ReaderType) TryReadBytesToSlice(bytes int) []byte {
@@ -289,6 +312,31 @@ func (reader *ReaderType) ReadBytes(bytes int) (uint64, error) {
 	return value, nil
 }
 
+// ReadBitsToSlice is a function that reads the specified amount of bits
+// from the parameter and puts each bit into a slice and returns this slice.
+//
+// Returns an error if there are no remaining bits.
+func (reader *ReaderType) ReadBitsToSlice(bits int) ([]byte, error) {
+	bytes := (bits / 8) + (bits % 8)
+	out := make([]byte, bytes)
+	for i := 0; i < bytes; i++ {
+		if bits/8 > 0 {
+			val, err := reader.ReadBytes(1)
+			if err != nil {
+				return out, err
+			}
+			out[i] = byte(val)
+		} else { // Not enough to fill a whole byte
+			val, err := reader.ReadBits(bits % 8)
+			if err != nil {
+				panic(err)
+			}
+			out[i] = byte(val)
+		}
+	}
+	return out, nil
+}
+
 // ReadBytesToSlice is a function that reads the specified amount of bytes
 // from the parameter and puts each byte into a slice and returns this slice.
 //
-- 
cgit v1.2.3