fix: Fixed reversed offsets for unity files
This commit is contained in:
parent
81915ff4f3
commit
07b676c571
@ -39,6 +39,23 @@ func CreateUnityFile(fileName string) (UnityFile, error) {
|
|||||||
return u, nil
|
return u, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func OpenUnityFile(fileName, metadataName string) (UnityFile, error) {
|
||||||
|
var u UnityFile
|
||||||
|
|
||||||
|
// Read the file
|
||||||
|
f, err := os.Open(fileName)
|
||||||
|
if err != nil {
|
||||||
|
return u, err
|
||||||
|
}
|
||||||
|
u.fd = f
|
||||||
|
|
||||||
|
if err := u.ReadMetadataFile(metadataName); err != nil {
|
||||||
|
return u, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return u, nil
|
||||||
|
}
|
||||||
|
|
||||||
func (u UnityFile) Size() int {
|
func (u UnityFile) Size() int {
|
||||||
return u.fileSize
|
return u.fileSize
|
||||||
}
|
}
|
||||||
@ -54,7 +71,7 @@ func (u *UnityFile) WriteChunk(data world.ChunkData) error {
|
|||||||
encodedSize := encoded.Len()
|
encodedSize := encoded.Len()
|
||||||
|
|
||||||
// Go to the end of the file
|
// Go to the end of the file
|
||||||
u.fd.Seek(0, u.fileSize)
|
u.fd.Seek(int64(u.fileSize), io.SeekStart)
|
||||||
// Write the encoded contents to the file
|
// Write the encoded contents to the file
|
||||||
if _, err := u.fd.Write(encoded.Bytes()); err != nil {
|
if _, err := u.fd.Write(encoded.Bytes()); err != nil {
|
||||||
return err
|
return err
|
||||||
@ -100,7 +117,7 @@ func (u *UnityFile) ReadMetadataFile(fileName string) error {
|
|||||||
func (u UnityFile) ReadChunk(pos world.ChunkPos) (world.ChunkData, error) {
|
func (u UnityFile) ReadChunk(pos world.ChunkPos) (world.ChunkData, error) {
|
||||||
m := u.metadata[pos]
|
m := u.metadata[pos]
|
||||||
|
|
||||||
u.fd.Seek(0, m.StartOffset)
|
u.fd.Seek(int64(m.StartOffset), io.SeekStart)
|
||||||
|
|
||||||
fileReader := io.LimitReader(u.fd, int64(m.FileSize))
|
fileReader := io.LimitReader(u.fd, int64(m.FileSize))
|
||||||
|
|
||||||
@ -112,6 +129,20 @@ func (u UnityFile) ReadChunk(pos world.ChunkPos) (world.ChunkData, error) {
|
|||||||
return data, nil
|
return data, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (u UnityFile) ReadAllChunks() ([]world.ChunkData, error) {
|
||||||
|
chunks := []world.ChunkData{}
|
||||||
|
|
||||||
|
for pos := range u.metadata {
|
||||||
|
chunk, err := u.ReadChunk(pos)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
chunks = append(chunks, chunk)
|
||||||
|
}
|
||||||
|
|
||||||
|
return chunks, nil
|
||||||
|
}
|
||||||
|
|
||||||
func (u *UnityFile) Close() error {
|
func (u *UnityFile) Close() error {
|
||||||
return u.fd.Close()
|
return u.fd.Close()
|
||||||
}
|
}
|
||||||
|
@ -1,6 +1,7 @@
|
|||||||
package storage
|
package storage
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"reflect"
|
"reflect"
|
||||||
@ -69,6 +70,7 @@ func TestWriteMultipleFiles(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("Error creating temporary directory: %v", err)
|
t.Fatalf("Error creating temporary directory: %v", err)
|
||||||
}
|
}
|
||||||
|
fmt.Println(tempDir)
|
||||||
defer os.RemoveAll(tempDir)
|
defer os.RemoveAll(tempDir)
|
||||||
|
|
||||||
u, err := CreateUnityFile(path.Join(tempDir, "test-unity"))
|
u, err := CreateUnityFile(path.Join(tempDir, "test-unity"))
|
||||||
@ -119,3 +121,60 @@ func TestWriteMultipleFiles(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestReadAllChunks(t *testing.T) {
|
||||||
|
tempDir, err := os.MkdirTemp("", "unity")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Error creating temporary directory: %v", err)
|
||||||
|
}
|
||||||
|
fmt.Println(tempDir)
|
||||||
|
defer os.RemoveAll(tempDir)
|
||||||
|
|
||||||
|
u, err := CreateUnityFile(path.Join(tempDir, "test-unity"))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Error creating unity file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
chunk1 world.ChunkData
|
||||||
|
chunk2 world.ChunkData
|
||||||
|
chunk3 world.ChunkData
|
||||||
|
)
|
||||||
|
chunk1.Pos = world.ChunkPos{
|
||||||
|
X: 0,
|
||||||
|
Z: 0,
|
||||||
|
}
|
||||||
|
chunk1.Sections[0].BlockStates[0] = 2
|
||||||
|
chunk2.Sections[0].BlockStates[0] = 3
|
||||||
|
chunk2.Pos = world.ChunkPos{
|
||||||
|
X: 1,
|
||||||
|
Z: 0,
|
||||||
|
}
|
||||||
|
chunk3.Sections[0].BlockStates[0] = 4
|
||||||
|
chunk3.Pos = world.ChunkPos{
|
||||||
|
X: 2,
|
||||||
|
Z: 0,
|
||||||
|
}
|
||||||
|
|
||||||
|
chunks := []world.ChunkData{chunk1, chunk2, chunk3}
|
||||||
|
|
||||||
|
// Write all chunks
|
||||||
|
for _, data := range chunks {
|
||||||
|
if err := u.WriteChunk(data); err != nil {
|
||||||
|
t.Fatalf("Error writing chunk: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Log(chunks)
|
||||||
|
|
||||||
|
readChunks, err := u.ReadAllChunks()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Error reading chunks: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for index, chunk := range readChunks {
|
||||||
|
if !reflect.DeepEqual(chunk, chunks[index]) {
|
||||||
|
t.Fatalf("Chunks were not equal")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user