From 33579d448e276c50429fb42cd887cb26fbc62ef9 Mon Sep 17 00:00:00 2001
From: Tnze <cjd001113@outlook.com>
Date: Tue, 14 Jun 2022 13:37:24 +0800
Subject: [PATCH] Fix Palette bugs reported by patyhank

---
 level/bitstorage.go | 12 ++++++++++++
 level/palette.go    |  2 +-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/level/bitstorage.go b/level/bitstorage.go
index 6cbef93..52aede9 100644
--- a/level/bitstorage.go
+++ b/level/bitstorage.go
@@ -182,3 +182,15 @@ func (b *BitStorage) WriteTo(w io.Writer) (int64, error) {
 	}
 	return n, nil
 }
+
+func (b *BitStorage) Fix(bits int) error {
+	b.mask = 1<<bits - 1
+	b.bits = bits
+	b.valuesPerLong = 64 / bits
+	// check data length
+	dataLen := calcBitStorageSize(bits, b.length)
+	if l := len(b.data); l != dataLen {
+		return newBitStorageErr{ArrlLen: l, WantLen: dataLen}
+	}
+	return nil
+}
diff --git a/level/palette.go b/level/palette.go
index 30128f3..70aa1f5 100644
--- a/level/palette.go
+++ b/level/palette.go
@@ -150,7 +150,7 @@ func (p *PaletteContainer[T]) ReadFrom(r io.Reader) (n int64, err error) {
 	if err != nil {
 		return n, err
 	}
-	return n, nil
+	return n, p.data.Fix(int(nBits))
 }
 
 func (p *PaletteContainer[T]) WriteTo(w io.Writer) (n int64, err error) {