1 files changed, 138 insertions, 0 deletions
diff --git a/nexgb/xgbgen/go_single_field.go b/nexgb/xgbgen/go_single_field.go
new file mode 100644
index 0000000..3c354e7
--- /dev/null
+++ b/nexgb/xgbgen/go_single_field.go
@@ -0,0 +1,138 @@
+package main
+
+import (
+	"fmt"
+	"log"
+)
+
+func (f *SingleField) Define(c *Context) {
+	c.Putln("%s %s", f.SrcName(), f.Type.SrcName())
+}
+
+func ReadSimpleSingleField(c *Context, name string, typ Type) {
+	switch t := typ.(type) {
+	case *Resource:
+		c.Putln("%s = get32(buf[b:])", name)
+	case *TypeDef:
+		switch t.Size().Eval() {
+		case 1:
+			c.Putln("%s = %s(buf[b])", name, t.SrcName())
+		case 2:
+			c.Putln("%s = %s(get16(buf[b:]))", name, t.SrcName())
+		case 4:
+			c.Putln("%s = %s(get32(buf[b:]))", name, t.SrcName())
+		case 8:
+			c.Putln("%s = %s(get64(buf[b:]))", name, t.SrcName())
+		}
+	case *Base:
+		var val string
+		switch t.Size().Eval() {
+		case 1:
+			val = fmt.Sprintf("buf[b]")
+		case 2:
+			val = fmt.Sprintf("get16(buf[b:])")
+		case 4:
+			val = fmt.Sprintf("get32(buf[b:])")
+		case 8:
+			val = fmt.Sprintf("get64(buf[b:])")
+		}
+
+		// We need to convert base types if they aren't uintXX or byte
+		ty := t.SrcName()
+		if ty != "byte" && ty != "uint16" && ty != "uint32" && ty != "uint64" {
+			val = fmt.Sprintf("%s(%s)", ty, val)
+		}
+		c.Putln("%s = %s", name, val)
+	default:
+		log.Fatalf("Cannot read field '%s' as a simple field with %T type.",
+			name, typ)
+	}
+
+	c.Putln("b += %s", typ.Size())
+}
+
+func (f *SingleField) Read(c *Context) {
+	switch t := f.Type.(type) {
+	case *Resource:
+		ReadSimpleSingleField(c, fmt.Sprintf("v.%s", f.SrcName()), t)
+	case *TypeDef:
+		ReadSimpleSingleField(c, fmt.Sprintf("v.%s", f.SrcName()), t)
+	case *Base:
+		ReadSimpleSingleField(c, fmt.Sprintf("v.%s", f.SrcName()), t)
+	case *Struct:
+		c.Putln("v.%s = %s{}", f.SrcName(), t.SrcName())
+		c.Putln("b += Read%s(buf[b:], &v.%s)", t.SrcName(), f.SrcName())
+		c.Putln("")
+	default:
+		log.Fatalf("Cannot read field '%s' with %T type.", f.XmlName(), f.Type)
+	}
+}
+
+func WriteSimpleSingleField(c *Context, name string, typ Type) {
+	switch t := typ.(type) {
+	case *Resource:
+		c.Putln("put32(buf[b:], uint32(%s))", name)
+	case *TypeDef:
+		switch t.Size().Eval() {
+		case 1:
+			c.Putln("buf[b] = byte(%s)", name)
+		case 2:
+			c.Putln("put16(buf[b:], uint16(%s))", name)
+		case 4:
+			c.Putln("put32(buf[b:], uint32(%s))", name)
+		case 8:
+			c.Putln("put64(buf[b:], uint64(%s))", name)
+		}
+	case *Base:
+		switch t.Size().Eval() {
+		case 1:
+			if t.SrcName() != "byte" {
+				c.Putln("buf[b] = byte(%s)", name)
+			} else {
+				c.Putln("buf[b] = %s", name)
+			}
+		case 2:
+			if t.SrcName() != "uint16" {
+				c.Putln("put16(buf[b:], uint16(%s))", name)
+			} else {
+				c.Putln("put16(buf[b:], %s)", name)
+			}
+		case 4:
+			if t.SrcName() != "uint32" {
+				c.Putln("put32(buf[b:], uint32(%s))", name)
+			} else {
+				c.Putln("put32(buf[b:], %s)", name)
+			}
+		case 8:
+			if t.SrcName() != "uint64" {
+				c.Putln("put64(buf[b:], uint64(%s))", name)
+			} else {
+				c.Putln("put64(buf[b:], %s)", name)
+			}
+		}
+	default:
+		log.Fatalf("Cannot read field '%s' as a simple field with %T type.",
+			name, typ)
+	}
+
+	c.Putln("b += %s", typ.Size())
+}
+
+func (f *SingleField) Write(c *Context) {
+	switch t := f.Type.(type) {
+	case *Resource:
+		ReadSimpleSingleField(c, fmt.Sprintf("v.%s", f.SrcName()), t)
+	case *TypeDef:
+		ReadSimpleSingleField(c, fmt.Sprintf("v.%s", f.SrcName()), t)
+	case *Base:
+		ReadSimpleSingleField(c, fmt.Sprintf("v.%s", f.SrcName()), t)
+	case *Struct:
+		c.Putln("{")
+		c.Putln("structBytes := v.%s.Bytes()", f.SrcName())
+		c.Putln("copy(buf[b:], structBytes)")
+		c.Putln("b += len(structBytes)")
+		c.Putln("}")
+	default:
+		log.Fatalf("Cannot read field '%s' with %T type.", f.XmlName(), f.Type)
+	}
+}