Merge pull request #27 from Galaco/refactor

Refactor
Galaco · Mar 28, 2019 · 4bfb611 · 4bfb611
2 parents 208dcd7 + 26a12a6
commit 4bfb611
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lumps/leaf.go b/lumps/leaf.go
@@ -42,11 +42,13 @@ func (lump *Leaf) Unmarshall(raw []byte) (err error) {
 	i := 0
 
 	for i < numLeafs {
-		rawLeaf := bytes.NewBuffer(raw[(structSize * i) : (structSize*i)+structSize])
+		leafBuf := make([]byte, structSize)
+		copy(leafBuf, raw[(structSize * i) : (structSize*i)+structSize])
 		// Pad the raw data to the correct size of a leaf
 		if lump.Version() > maxBspVersionOfV0Leaf {
-			rawLeaf.Write(make([]byte, int(unsafe.Sizeof(common.CompressedLightCube{}))))
+			leafBuf = append(leafBuf, make([]byte, int(unsafe.Sizeof(common.CompressedLightCube{})))...)
 		}
+		rawLeaf := bytes.NewBuffer(leafBuf)
 		err = binary.Read(rawLeaf, binary.LittleEndian, &lump.data[i])
 		if err != nil {
 			return err