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.
87 lines
2.2 KiB
Go
87 lines
2.2 KiB
Go
8 years ago
|
package ent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"math"
|
||
|
|
|
||
|
|
"github.com/golang/protobuf/proto"
|
||
|
|
|
||
|
|
"github.com/jordanorelli/hyperstone/bit"
|
||
|
|
"github.com/jordanorelli/hyperstone/dota"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Namespace registers the names of known classess, associating their ids to
|
||
|
|
// their network types.
|
||
|
|
type Namespace struct {
|
||
|
|
SymbolTable
|
||
|
|
idBits int
|
||
|
|
|
||
|
|
// maps ClassInfo ids to class names
|
||
|
|
classIds map[int]string
|
||
|
|
|
||
|
|
// maps classes to their {name, version} id pairs
|
||
|
|
classes map[classId]*Class
|
||
|
|
|
||
|
|
// maps a class name to every version of the class
|
||
|
|
classesByName map[string]map[int]*Class
|
||
|
|
}
|
||
|
|
|
||
|
|
// Merges in the ClassInfo data found in the replay protobufs
|
||
|
|
func (n *Namespace) MergeClassInfo(ci *dota.CDemoClassInfo) {
|
||
|
|
if n.classIds == nil {
|
||
|
|
n.classIds = make(map[int]string, len(ci.GetClasses()))
|
||
|
|
}
|
||
|
|
for _, class := range ci.GetClasses() {
|
||
|
|
n.classIds[int(class.GetClassId())] = class.GetNetworkName()
|
||
|
|
}
|
||
|
|
n.idBits = int(math.Floor(math.Log2(float64(len(n.classIds))))) + 1
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *Namespace) MergeSendTables(st *dota.CDemoSendTables) {
|
||
|
|
// sendtables only has one field, a binary data field.
|
||
|
|
data := st.GetData()
|
||
|
|
br := bit.NewBytesReader(data)
|
||
|
|
|
||
|
|
// body is length-prefixed
|
||
|
|
size := int(bit.ReadVarInt(br))
|
||
|
|
|
||
|
|
buf := make([]byte, size)
|
||
|
|
br.Read(buf)
|
||
|
|
|
||
|
|
flat := dota.CSVCMsg_FlattenedSerializer{}
|
||
|
|
if err := proto.Unmarshal(buf, &flat); err != nil {
|
||
|
|
fmt.Printf("error: %v\n", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
n.SymbolTable = SymbolTable(flat.GetSymbols())
|
||
|
|
|
||
|
|
fields := make([]Field, len(flat.GetFields()))
|
||
|
|
for i, f := range flat.GetFields() {
|
||
|
|
fields[i].fromProto(f, &n.SymbolTable)
|
||
|
|
}
|
||
|
|
|
||
|
|
n.classes = make(map[classId]*Class, len(flat.GetSerializers()))
|
||
|
|
n.classesByName = make(map[string]map[int]*Class, len(flat.GetSerializers()))
|
||
|
|
|
||
|
|
for _, c := range flat.GetSerializers() {
|
||
|
|
class := Class{}
|
||
|
|
class.fromProto(c, fields)
|
||
|
|
|
||
|
|
name := n.Symbol(int(c.GetSerializerNameSym()))
|
||
|
|
version := int(c.GetSerializerVersion())
|
||
|
|
id := classId{name: name, version: version}
|
||
|
|
n.classes[id] = &class
|
||
|
|
|
||
|
|
if n.classesByName[name.String()] == nil {
|
||
|
|
n.classesByName[name.String()] = map[int]*Class{version: &class}
|
||
|
|
} else {
|
||
|
|
n.classesByName[name.String()][version] = &class
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *Namespace) ReadClassId(r bit.Reader) int {
|
||
|
|
return int(r.ReadBits(uint(n.idBits)))
|
||
|
|
}
|