another rewrite, v1.0.0

5 files changed, 317 insertions, 0 deletions

diff --git a/pkg/classes/cmdInfo.go b/pkg/classes/cmdInfo.go
new file mode 100644
index 0000000..6e56c41
--- /dev/null
+++ b/pkg/classes/cmdInfo.go
@@ -0,0 +1,27 @@
+package classes
+
+import (
+        "github.com/pektezol/bitreader"
+)
+
+type CmdInfo struct {
+        Flags            int32
+        ViewOrigin       []float32
+        ViewAngles       []float32
+        LocalViewAngles  []float32
+        ViewOrigin2      []float32
+        ViewAngles2      []float32
+        LocalViewAngles2 []float32
+}
+
+func ParseCmdInfo(reader *bitreader.ReaderType) CmdInfo {
+        return CmdInfo{
+                Flags:            int32(reader.TryReadBits(32)),
+                ViewOrigin:       []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
+                ViewAngles:       []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
+                LocalViewAngles:  []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
+                ViewOrigin2:      []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
+                ViewAngles2:      []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
+                LocalViewAngles2: []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
+        }
+}
diff --git a/pkg/classes/sendTable.go b/pkg/classes/sendTable.go
new file mode 100644
index 0000000..4521464
--- /dev/null
+++ b/pkg/classes/sendTable.go
@@ -0,0 +1,106 @@
+package classes
+
+import (
+        "github.com/pektezol/bitreader"
+)
+
+type SendTable struct {
+        NeedsDecoder bool
+        NetTableName string
+        NumOfProps   int16
+        Props        []prop
+}
+
+type prop struct {
+        SendPropType  int8
+        SendPropName  string
+        SendPropFlags int32
+        Priority      int8
+        ExcludeDtName string
+        LowValue      float32
+        HighValue     float32
+        NumBits       int32
+        NumElements   int32
+}
+
+type sendPropFlag int
+
+const (
+        Unsigned sendPropFlag = iota
+        Coord
+        NoScale
+        RoundDown
+        RoundUp
+        Normal
+        Exclude
+        Xyze
+        InsideArray
+        ProxyAlwaysYes
+        IsVectorElem
+        Collapsible
+        CoordMp
+        CoordMpLp // low precision
+        CoordMpInt
+        CellCoord
+        CellCoordLp
+        CellCoordInt
+        ChangesOften
+)
+
+type sendPropType int
+
+const (
+        Int sendPropType = iota
+        Float
+        Vector3
+        Vector2
+        String
+        Array
+        DataTable
+)
+
+func ParseSendTable(reader *bitreader.ReaderType) SendTable {
+        sendTable := SendTable{
+                NeedsDecoder: reader.TryReadBool(),
+                NetTableName: reader.TryReadString(),
+                NumOfProps:   int16(reader.TryReadBits(10)),
+                // SendPropType:  int8(reader.TryReadBits(5)),
+                // SendPropName:  reader.TryReadString(),
+                // SendPropFlags: int16(reader.TryReadBits(16)),
+        }
+        if sendTable.NumOfProps < 0 {
+                return sendTable
+        }
+        for count := 0; count < int(sendTable.NumOfProps); count++ {
+                propType := int8(reader.TryReadBits(5))
+                if propType >= int8(7) {
+                        return sendTable
+                }
+                prop := prop{
+                        SendPropType:  propType,
+                        SendPropName:  reader.TryReadString(),
+                        SendPropFlags: int32(reader.TryReadBits(19)),
+                        Priority:      int8(reader.TryReadBits(8)),
+                }
+                if propType == int8(DataTable) || CheckBit(int64(prop.SendPropFlags), int(Exclude)) {
+                        prop.ExcludeDtName = reader.TryReadString()
+                } else {
+                        switch propType {
+                        case int8(String), int8(Int), int8(Float), int8(Vector3), int8(Vector2):
+                                prop.LowValue = reader.TryReadFloat32()
+                                prop.HighValue = reader.TryReadFloat32()
+                                prop.NumBits = int32(reader.TryReadBits(7))
+                        case int8(Array):
+                                prop.NumElements = int32(reader.TryReadBits(10))
+                        default:
+                                return sendTable
+                        }
+                }
+                sendTable.Props = append(sendTable.Props, prop)
+        }
+        return sendTable
+}
+
+func CheckBit(val int64, bit int) bool {
+        return (val & (int64(1) << bit)) != 0
+}
diff --git a/pkg/classes/serverClassInfo.go b/pkg/classes/serverClassInfo.go
new file mode 100644
index 0000000..c60dad1
--- /dev/null
+++ b/pkg/classes/serverClassInfo.go
@@ -0,0 +1,30 @@
+package classes
+
+import (
+        "github.com/pektezol/bitreader"
+)
+
+type ServerClassInfo struct {
+        ClassId       int16
+        ClassName     string
+        DataTableName string
+}
+
+func ParseServerClassInfo(reader *bitreader.ReaderType, count int, numOfClasses int) ServerClassInfo {
+        return ServerClassInfo{
+                ClassId:       int16(reader.TryReadBits(16)),
+                ClassName:     reader.TryReadString(),
+                DataTableName: reader.TryReadString(),
+        }
+}
+
+// func serverClassBits(numOfClasses int) int {
+//      return highestBitIndex(uint(numOfClasses)) + 1
+// }
+
+// func highestBitIndex(i uint) int {
+//      var j int
+//      for j = 31; j >= 0 && (i&(1<<j)) == 0; j-- {
+//      }
+//      return j
+// }
diff --git a/pkg/classes/stringTable.go b/pkg/classes/stringTable.go
new file mode 100644
index 0000000..c6709f5
--- /dev/null
+++ b/pkg/classes/stringTable.go
@@ -0,0 +1,82 @@
+package classes
+
+import (
+        "bytes"
+
+        "github.com/pektezol/bitreader"
+)
+
+type StringTable struct {
+        Name         string
+        TableEntries []StringTableEntry
+        Classes      []StringTableClass
+}
+
+type StringTableEntry struct {
+        Name      string
+        EntryData StringTableEntryData
+}
+
+type StringTableEntryData struct {
+        // TODO: Parse StringTableEntry
+}
+
+type StringTableClass struct {
+        Name string
+        Data string
+}
+
+func ParseStringTables(reader *bitreader.ReaderType) []StringTable {
+        tableCount := reader.TryReadBits(8)
+        stringTables := make([]StringTable, tableCount)
+        for i := 0; i < int(tableCount); i++ {
+                var table StringTable
+                table.ParseStream(reader)
+                stringTables[i] = table
+        }
+        return stringTables
+}
+
+func (stringTable *StringTable) ParseStream(reader *bitreader.ReaderType) {
+        stringTable.Name = reader.TryReadString()
+        entryCount := reader.TryReadBits(16)
+        stringTable.TableEntries = make([]StringTableEntry, entryCount)
+
+        for i := 0; i < int(entryCount); i++ {
+                var entry StringTableEntry
+                entry.Parse(reader)
+                stringTable.TableEntries[i] = entry
+        }
+
+        if reader.TryReadBool() {
+                classCount := reader.TryReadBits(16)
+                stringTable.Classes = make([]StringTableClass, classCount)
+
+                for i := 0; i < int(classCount); i++ {
+                        var class StringTableClass
+                        class.Parse(reader)
+                        stringTable.Classes[i] = class
+                }
+        }
+}
+
+func (stringTableEntry *StringTableEntry) Parse(reader *bitreader.ReaderType) {
+        stringTableEntry.Name = reader.TryReadString()
+        if reader.TryReadBool() {
+                byteLen, err := reader.ReadBits(16)
+                if err != nil {
+                        return
+                }
+                dataBsr := reader.TryReadBytesToSlice(int(byteLen))
+                _ = bitreader.Reader(bytes.NewReader(dataBsr), true) // TODO: Parse StringTableEntry
+                // stringTableEntry.EntryData.ParseStream(entryReader)
+        }
+}
+
+func (stringTableClass *StringTableClass) Parse(reader *bitreader.ReaderType) {
+        stringTableClass.Name = reader.TryReadString()
+        if reader.TryReadBool() {
+                dataLen := reader.TryReadBits(16)
+                stringTableClass.Data = reader.TryReadStringLen(int(dataLen))
+        }
+}
diff --git a/pkg/classes/userCmdInfo.go b/pkg/classes/userCmdInfo.go
new file mode 100644
index 0000000..a6d9091
--- /dev/null
+++ b/pkg/classes/userCmdInfo.go
@@ -0,0 +1,72 @@
+package classes
+
+import (
+        "bytes"
+
+        "github.com/pektezol/bitreader"
+)
+
+type UserCmdInfo struct {
+        CommandNumber int32
+        TickCount     int32
+        ViewAnglesX   float32
+        ViewAnglesY   float32
+        ViewAnglesZ   float32
+        ForwardMove   float32
+        SideMove      float32
+        UpMove        float32
+        Buttons       int32
+        Impulse       int8
+        WeaponSelect  int16
+        WeaponSubType int8
+        MouseDx       int16
+        MouseDy       int16
+}
+
+func ParseUserCmdInfo(data []byte) UserCmdInfo {
+        reader := bitreader.Reader(bytes.NewReader(data), true)
+        userCmdInfo := UserCmdInfo{}
+        if reader.TryReadBool() {
+                userCmdInfo.CommandNumber = int32(reader.TryReadBits(32))
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.TickCount = int32(reader.TryReadBits(32))
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.ViewAnglesX = reader.TryReadFloat32()
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.ViewAnglesY = reader.TryReadFloat32()
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.ViewAnglesZ = reader.TryReadFloat32()
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.ForwardMove = reader.TryReadFloat32()
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.SideMove = reader.TryReadFloat32()
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.UpMove = reader.TryReadFloat32()
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.Buttons = int32(reader.TryReadBits(32))
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.Impulse = int8(reader.TryReadBits(8))
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.WeaponSelect = int16(reader.TryReadBits(11))
+                if reader.TryReadBool() {
+                        userCmdInfo.WeaponSubType = int8(reader.TryReadBits(6))
+                }
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.MouseDx = int16(reader.TryReadBits(16))
+        }
+        if reader.TryReadBool() {
+                userCmdInfo.MouseDy = int16(reader.TryReadBits(16))
+        }
+        return userCmdInfo
+}