You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

82 lines
1.9 KiB
Go

package ent
import (
8 years ago
"fmt"
"github.com/golang/protobuf/proto"
"github.com/jordanorelli/hyperstone/bit"
8 years ago
"github.com/jordanorelli/hyperstone/dota"
)
type Env struct {
8 years ago
symbols symbolTable
source bit.BufReader
classes map[string]classHistory
fields []field
}
func (e *Env) Handle(m proto.Message) error {
8 years ago
switch v := m.(type) {
case *dota.CDemoSendTables:
return e.mergeSendTables(v)
}
return nil
}
func (e *Env) setSource(buf []byte) {
e.source.SetSource(buf)
}
8 years ago
func (e *Env) mergeSendTables(m *dota.CDemoSendTables) error {
Debug.Printf("merge send tables")
flat, err := getSerializers(m)
if err != nil {
return wrap(err, "unable to get serializers in sendtables")
}
e.symbols = symbolTable(flat.GetSymbols())
e.stubClasses(flat)
if err := e.parseFields(flat); err != nil {
return wrap(err, "unable to parse serializer fields")
}
if err := e.parseClasses(flat); err != nil {
return wrap(err, "unable to parse serializers")
}
return nil
}
// stubs out the classes to be created later. we do this to create empty class
// structs that fields may point to.
func (e *Env) stubClasses(flat *dota.CSVCMsg_FlattenedSerializer) {
serializers := flat.GetSerializers()
if e.classes == nil {
e.classes = make(map[string]classHistory, len(serializers))
}
for _, s := range serializers {
name := e.symbol(int(s.GetSerializerNameSym()))
v := int(s.GetSerializerVersion())
if e.classes[name] == nil {
e.classes[name] = make(classHistory, 4)
}
8 years ago
e.classes[name][v] = &class{name: name, version: v}
8 years ago
}
}
func (e *Env) parseFields(flat *dota.CSVCMsg_FlattenedSerializer) error {
e.fields = make([]field, len(flat.GetFields()))
for i, ff := range flat.GetFields() {
f := &e.fields[i]
if err := f.fromProto(ff, e); err != nil {
return err
}
}
return nil
}
func (e *Env) parseClasses(flat *dota.CSVCMsg_FlattenedSerializer) error {
return fmt.Errorf("nope, not yet")
}
func (e *Env) symbol(id int) string { return e.symbols[id] }