129 lines
2.7 KiB
Go
129 lines
2.7 KiB
Go
package save
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/Tnze/go-mc/data/packetid"
|
|
pk "github.com/Tnze/go-mc/net/packet"
|
|
"github.com/Tnze/go-mc/save/region"
|
|
"math/rand"
|
|
"testing"
|
|
"unsafe"
|
|
)
|
|
|
|
func TestColumn(t *testing.T) {
|
|
var c Column
|
|
r, err := region.Open("testdata/region/r.0.0.mca")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer r.Close()
|
|
|
|
data, err := r.ReadSector(0, 0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = c.Load(data)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
t.Logf("%+v", c)
|
|
}
|
|
|
|
func BenchmarkColumn_Load(b *testing.B) {
|
|
// Test how many time we load a chunk
|
|
var c Column
|
|
r, err := region.Open("testdata/region/r.-1.-1.mca")
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
defer r.Close()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
//x, y := (i%1024)/32, (i%1024)%32
|
|
x, y := rand.Intn(32), rand.Intn(32)
|
|
|
|
data, err := r.ReadSector(x, y)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
err = c.Load(data)
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func ExampleColumn_send() {
|
|
r, err := region.Open("/path/to/r.0.0.mca")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
chunkPos := [2]int{0, 0}
|
|
data, err := r.ReadSector(chunkPos[0], chunkPos[1])
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
var c Column
|
|
if err := c.Load(data); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
var buf bytes.Buffer
|
|
var PrimaryBitMask pk.VarInt
|
|
for _, v := range c.Level.Sections {
|
|
if int8(v.Y) >= 0 && int8(v.Y) < 16 {
|
|
PrimaryBitMask |= 1 << v.Y
|
|
|
|
bpb := len(v.BlockStates) * 64 / (16 * 16 * 16)
|
|
hasPalette := pk.Boolean(bpb >= 9)
|
|
paletteLength := pk.VarInt(len(v.Palette))
|
|
dataArrayLength := pk.VarInt(len(v.BlockStates))
|
|
dataArray := (*[]pk.Long)(unsafe.Pointer(&v.BlockStates))
|
|
_, err := pk.Tuple{
|
|
pk.Short(0), // Block count
|
|
pk.UnsignedByte(bpb), // Bits Per Block
|
|
hasPalette, pk.Opt{
|
|
Has: &hasPalette,
|
|
Field: pk.Tuple{
|
|
paletteLength, pk.Ary{
|
|
Len: &paletteLength,
|
|
Ary: nil, // TODO: We need translate v.Palette (with type of []Block) to state ID
|
|
},
|
|
},
|
|
}, // Palette
|
|
dataArrayLength, pk.Ary{
|
|
Len: &dataArrayLength,
|
|
Ary: dataArray,
|
|
}, // Data Array
|
|
}.WriteTo(&buf)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
size := pk.VarInt(buf.Len())
|
|
bal := pk.VarInt(len(c.Level.Biomes))
|
|
_ = pk.Marshal(
|
|
packetid.WorldParticles,
|
|
pk.Int(chunkPos[0]), // Chunk X
|
|
pk.Int(chunkPos[1]), // Chunk Y
|
|
pk.Boolean(true), // Full chunk
|
|
PrimaryBitMask, // PrimaryBitMask
|
|
pk.NBT(c.Level.Heightmaps), // Heightmaps
|
|
bal, pk.Ary{
|
|
Len: bal, // Biomes array length
|
|
Ary: c.Level.Biomes, // Biomes
|
|
},
|
|
size, pk.Ary{
|
|
Len: size, // Size
|
|
Ary: pk.ByteArray(buf.Bytes()), // Data
|
|
},
|
|
pk.VarInt(0), // Block entities array length
|
|
)
|
|
}
|