falloutBot/go-mc
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

try read ChunkData

Browse Source
This commit is contained in:
JunDao
2019-05-21 00:11:27 +08:00
parent 6be465640f
commit 837bb0b213
7 changed files with 46247 additions and 20499 deletions
Download Patch File Download Diff File Expand all files Collapse all files

217
bot/world/chunk.go
View File

@ -1,122 +1,65 @@
package world
import (
"github.com/Tnze/go-mc/nbt"
// "fmt"
pk "github.com/Tnze/go-mc/net/packet"
"bytes"
"fmt"
// "io"
"github.com/Tnze/go-mc/data"
pk "github.com/Tnze/go-mc/net/packet"
)
func UnpackChunkDataPacket(p pk.Packet, hasSkyLight bool) (c *Chunk, x, z int, err error) {
var (
X, Z pk.Int
FullChunk pk.Boolean
PrimaryBitMask pk.VarInt
Heightmaps struct{}
Data chunkData
BlockEntities blockEntities
)
p.Scan(&X, &Z, &FullChunk, &PrimaryBitMask, pk.NBT{V: &Heightmaps}, &Data, &BlockEntities)
//解析区块数据
cc, err := readChunkColumn(bool(FullChunk), int32(PrimaryBitMask), []byte(Data), hasSkyLight)
if err != nil {
panic(err)
}
return cc, int(X), int(Z), err
}
type chunkData []byte
type blockEntities []blockEntitie
type blockEntitie struct {
}
func (c *chunkData) Decode(r pk.DecodeReader) error {
var Size pk.VarInt
if err := Size.Decode(r); err != nil {
return err
}
*c = make([]byte, Size)
if _, err := r.Read(*c); err != nil {
return err
}
return nil
}
func (b *blockEntities) Decode(r pk.DecodeReader) error {
var NumberofBlockEntities pk.VarInt
if err := NumberofBlockEntities.Decode(r); err != nil {
return err
}
*b = make(blockEntities, NumberofBlockEntities)
decoder := nbt.NewDecoder(r)
for i := 0; i < int(NumberofBlockEntities); i++ {
if err := decoder.Decode(&(*b)[i]); err != nil {
return err
}
}
return nil
}
func readChunkColumn(isFull bool, mask int32, data []byte, hasSkyLight bool) (*Chunk, error) {
//DecodeChunkColumn decode the chunk data structure
func DecodeChunkColumn(isFull bool, mask int32, data []byte) (*Chunk, error) {
var c Chunk
// for sectionY := 0; sectionY < 16; sectionY++ {
// if (mask & (1 << uint(sectionY))) != 0 { // Is the given bit set in the mask?
// BitsPerBlock, err := data.ReadByte()
// if err != nil {
// return nil, fmt.Errorf("read BitsPerBlock fail: %v", err)
// }
// //读调色板
// var palette []uint
// if BitsPerBlock < 9 {
// length, err := pk.UnpackVarInt(data)
// if err != nil {
// return nil, fmt.Errorf("read palette (id len) fail: %v", err)
// }
// palette = make([]uint, length)
r := bytes.NewReader(data)
for sectionY := 0; sectionY < 16; sectionY++ {
if (mask & (1 << uint(sectionY))) == 0 { // Is the given bit set in the mask?
continue
}
var (
BlockCount pk.Short
BitsPerBlock pk.Byte
)
if err := BlockCount.Decode(r); err != nil {
return nil, err
}
if err := BitsPerBlock.Decode(r); err != nil {
return nil, err
}
//读调色板
var palette []uint
if BitsPerBlock < 9 {
var length pk.VarInt
if err := length.Decode(r); err != nil {
return nil, fmt.Errorf("read palette (id len) fail: %v", err)
}
palette = make([]uint, length)
// for id := uint(0); id < uint(length); id++ {
// stateID, err := pk.UnpackVarInt(data)
// if err != nil {
// return nil, fmt.Errorf("read palette (id) fail: %v", err)
// }
for id := uint(0); id < uint(length); id++ {
var stateID pk.VarInt
if err := stateID.Decode(r); err != nil {
return nil, fmt.Errorf("read palette (id) fail: %v", err)
}
// palette[id] = uint(stateID)
// }
// }
palette[id] = uint(stateID)
}
}
// //Section数据
// DataArrayLength, err := pk.UnpackVarInt(data)
// if err != nil {
// return nil, fmt.Errorf("read DataArrayLength fail: %v", err)
// }
//Section数据
var DataArrayLength pk.VarInt
if err := DataArrayLength.Decode(r); err != nil {
return nil, fmt.Errorf("read DataArrayLength fail: %v", err)
}
// DataArray := make([]int64, DataArrayLength)
// for i := 0; i < int(DataArrayLength); i++ {
// DataArray[i], err = pk.UnpackInt64(data)
// if err != nil {
// return nil, fmt.Errorf("read DataArray fail: %v", err)
// }
// }
// //用数据填充区块
// fillSection(&c.sections[sectionY], perBits(BitsPerBlock), DataArray, palette)
// //throw BlockLight data
// _, err = pk.ReadNBytes(data, 2048)
// if err != nil {
// return nil, fmt.Errorf("read BlockLight fail: %v", err)
// }
// if hasSkyLight {
// //throw SkyLight data
// _, err = pk.ReadNBytes(data, 2048)
// if err != nil {
// return nil, fmt.Errorf("read SkyLight fail: %v", err)
// }
// }
// }
// }
DataArray := make([]int64, DataArrayLength)
for i := 0; i < int(DataArrayLength); i++ {
if err := (*pk.Long)(&DataArray[i]).Decode(r); err != nil {
return nil, fmt.Errorf("read DataArray fail: %v", err)
}
}
//用数据填充区块
fillSection(&c.sections[sectionY], perBits(byte(BitsPerBlock)), DataArray, palette)
}
// if isFull { //need recive Biomes datas
// _, err := pk.ReadNBytes(data, 256*4)
// if err != nil {
@ -128,35 +71,33 @@ func readChunkColumn(isFull bool, mask int32, data []byte, hasSkyLight bool) (*C
return &c, nil
}
// const defaultBitsPerBlock = 14
func perBits(BitsPerBlock byte) uint {
switch {
case BitsPerBlock <= 4:
return 4
case BitsPerBlock < 9:
return uint(BitsPerBlock)
default:
return uint(data.BitsPerBlock) // DefaultBitsPerBlock
}
}
// func perBits(BitsPerBlock byte) uint {
// switch {
// case BitsPerBlock <= 4:
// return 4
// case BitsPerBlock < 9:
// return uint(BitsPerBlock)
// default:
// return defaultBitsPerBlock
// }
// }
func fillSection(s *Section, bpb uint, DataArray []int64, palette []uint) {
mask := uint(1<<bpb - 1)
for n := 0; n < 16*16*16; n++ {
offset := uint(n * int(bpb))
data := uint(DataArray[offset/64])
data >>= offset % 64
if offset%64 > 64-bpb {
l := bpb + offset%64 - 64
data &= uint(DataArray[offset/64+1] << l)
}
data &= mask
// func fillSection(s *Section, bpb uint, DataArray []int64, palette []uint) {
// mask := uint(1<<bpb - 1)
// for n := 0; n < 16*16*16; n++ {
// offset := uint(n * int(bpb))
// data := uint(DataArray[offset/64])
// data >>= offset % 64
// if offset%64 > 64-bpb {
// l := bpb + offset%64 - 64
// data &= uint(DataArray[offset/64+1] << l)
// }
// data &= mask
// if bpb < 9 {
// s.blocks[n%16][n/(16*16)][n%(16*16)/16].id = palette[data]
// } else {
// s.blocks[n%16][n/(16*16)][n%(16*16)/16].id = data
// }
// }
// }
if bpb < 9 {
s.blocks[n%16][n/(16*16)][n%(16*16)/16].id = palette[data]
} else {
s.blocks[n%16][n/(16*16)][n%(16*16)/16].id = data
}
}
}