91 lines
1.7 KiB
Go
91 lines
1.7 KiB
Go
package block
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
_ "embed"
|
|
"fmt"
|
|
"math/bits"
|
|
|
|
"git.konjactw.dev/falloutBot/go-mc/nbt"
|
|
)
|
|
|
|
type Block interface {
|
|
ID() string
|
|
}
|
|
|
|
// This file stores all possible block states into a TAG_List with gzip compressed.
|
|
//
|
|
//go:generate go run ./generator/blocks/main.go
|
|
//go:embed block_states.nbt
|
|
var blockStates []byte
|
|
|
|
var (
|
|
ToStateID map[Block]StateID
|
|
StateList []Block
|
|
)
|
|
|
|
// BitsPerBlock indicates how many bits are needed to represent all possible
|
|
// block states. This value is used to determine the size of the global palette.
|
|
var BitsPerBlock int
|
|
|
|
type (
|
|
StateID int
|
|
State struct {
|
|
Name string
|
|
Properties nbt.RawMessage
|
|
}
|
|
)
|
|
|
|
func (s *State) Block() (Block, error) {
|
|
// Get an empty Block object
|
|
block, ok := FromID[s.Name]
|
|
if !ok {
|
|
return nil, UnknownBlockErr{s.Name}
|
|
}
|
|
|
|
// Set property values
|
|
if s.Properties.Type != nbt.TagEnd {
|
|
err := s.Properties.Unmarshal(&block)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return block, nil
|
|
}
|
|
|
|
type UnknownBlockErr struct {
|
|
Name string
|
|
}
|
|
|
|
func (u UnknownBlockErr) Error() string {
|
|
return "unknown block: " + u.Name
|
|
}
|
|
|
|
func init() {
|
|
var states []State
|
|
// decompress
|
|
z, err := gzip.NewReader(bytes.NewReader(blockStates))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// decode all states
|
|
if _, err = nbt.NewDecoder(z).Decode(&states); err != nil {
|
|
panic(err)
|
|
}
|
|
ToStateID = make(map[Block]StateID, len(states))
|
|
StateList = make([]Block, 0, len(states))
|
|
for _, state := range states {
|
|
block, err := state.Block()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if _, ok := ToStateID[block]; ok {
|
|
panic(fmt.Errorf("state %#v already exists", block))
|
|
}
|
|
ToStateID[block] = StateID(len(StateList))
|
|
StateList = append(StateList, block)
|
|
}
|
|
BitsPerBlock = bits.Len(uint(len(StateList)))
|
|
}
|