zworld/engine/object/serialize.go

package object

import (
	"encoding/gob"
	"errors"
	"fmt"
	"io"
	"reflect"
	"zworld/plugins/math/quat"
	"zworld/plugins/math/vec3"
)

type Decoder interface {
	Decode(e any) error
}

type Encoder interface {
	Encode(data any) error
}

type Serializable interface {
	Serialize(Encoder) error
}

type MemorySerializer struct {
	stream []any
	index  int
}

func (m *MemorySerializer) Encode(data any) error {
	m.stream = append(m.stream, data)
	return nil
}

func (m *MemorySerializer) Decode(target any) error {
	if m.index >= len(m.stream) {
		return io.EOF
	}
	reflect.ValueOf(target).Elem().Set(reflect.ValueOf(m.stream[m.index]))
	m.index++
	return nil
}

func Copy(obj Component) Component {
	buffer := &MemorySerializer{}

	err := Serialize(buffer, obj)
	if err != nil {
		panic(err)
	}

	kopy, err := Deserialize(buffer)
	if err != nil {
		panic(err)
	}

	return kopy
}

func Save(writer io.Writer, obj Component) error {
	enc := gob.NewEncoder(writer)
	return Serialize(enc, obj)
}

func Load(reader io.Reader) (Component, error) {
	dec := gob.NewDecoder(reader)
	return Deserialize(dec)
}

type ComponentState struct {
	ID      uint
	Name    string
	Enabled bool
}

func NewComponentState(c Component) ComponentState {
	return ComponentState{
		ID:      c.ID(),
		Name:    c.Name(),
		Enabled: c.Enabled(),
	}
}

func (c ComponentState) New() Component {
	return &component{
		id:      c.ID,
		name:    c.Name,
		enabled: c.Enabled,
	}
}

type ObjectState struct {
	ComponentState
	Position vec3.T
	Rotation quat.T
	Scale    vec3.T
	Children int
}

type DeserializeFn func(Decoder) (Component, error)

var ErrSerialize = errors.New("serialization error")

var types = map[string]DeserializeFn{}

func typeName(obj any) string {
	t := reflect.TypeOf(obj).Elem()
	return t.PkgPath() + "/" + t.Name()
}

func init() {
	Register[*object](DeserializeObject)
}

func Register[T Serializable](deserializer DeserializeFn) {
	var empty T
	kind := typeName(empty)
	types[kind] = deserializer
}

func Serialize(enc Encoder, obj Component) error {
	kind := typeName(obj)
	serializable, ok := obj.(Serializable)
	if !ok {
		return fmt.Errorf("%w: %s is not serializable", ErrSerialize, kind)
	}
	if err := enc.Encode(kind); err != nil {
		return err
	}
	return serializable.Serialize(enc)
}

func Deserialize(decoder Decoder) (Component, error) {
	var kind string
	if err := decoder.Decode(&kind); err != nil {
		return nil, err
	}
	deserializer, exists := types[kind]
	if !exists {
		return nil, fmt.Errorf("%w: no deserializer for %s", ErrSerialize, kind)
	}
	return deserializer(decoder)
}

func (o *object) Serialize(enc Encoder) error {
	children := 0
	for _, child := range o.children {
		if _, ok := child.(Serializable); ok {
			children++
		}
	}

	if err := enc.Encode(ObjectState{
		ComponentState: NewComponentState(o),
		Position:       o.transform.Position(),
		Rotation:       o.transform.Rotation(),
		Scale:          o.transform.Scale(),
		Children:       children,
	}); err != nil {
		return err
	}

	// serialize children
	for _, child := range o.children {
		if err := Serialize(enc, child); err != nil {
			if errors.Is(err, ErrSerialize) {
				continue
			}
			return err
		}
	}
	return nil
}

func DeserializeObject(dec Decoder) (Component, error) {
	var data ObjectState
	if err := dec.Decode(&data); err != nil {
		return nil, err
	}
	obj := Empty(data.Name)
	obj.setEnabled(data.Enabled)
	obj.Transform().SetPosition(data.Position)
	obj.Transform().SetRotation(data.Rotation)
	obj.Transform().SetScale(data.Scale)

	// deserialize children
	for i := 0; i < data.Children; i++ {
		child, err := Deserialize(dec)
		if err != nil {
			return nil, err
		}
		Attach(obj, child)
	}
	return obj, nil
}