TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-11-29 13:41:47 +00:00
Code Issues Projects Releases Packages Wiki Activity

compiler: create arrays/structs more effectively

Browse source 
Use PACK/PACKSTRUCT.
This commit is contained in:
Roman Khimov 2021-11-12 16:13:41 +03:00
parent 3e6ce3c221
commit 909ea477f4
1 changed files with 7 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
pkg/compiler/codegen.go
View file

@ -1828,13 +1828,6 @@ func (c *codegen) convertStruct(lit *ast.CompositeLit, ptr bool) {
} }
emit.Opcodes(c.prog.BinWriter, opcode.NOP) emit.Opcodes(c.prog.BinWriter, opcode.NOP)
emit.Int(c.prog.BinWriter, int64(strct.NumFields()))
if ptr {
emit.Opcodes(c.prog.BinWriter, opcode.NEWARRAY)
} else {
emit.Opcodes(c.prog.BinWriter, opcode.NEWSTRUCT)
}
keyedLit := len(lit.Elts) > 0 keyedLit := len(lit.Elts) > 0
if keyedLit { if keyedLit {
_, ok := lit.Elts[0].(*ast.KeyValueExpr) _, ok := lit.Elts[0].(*ast.KeyValueExpr)
@ -1842,13 +1835,10 @@ func (c *codegen) convertStruct(lit *ast.CompositeLit, ptr bool) {
} }
// We need to locally store all the fields, even if they are not initialized. // We need to locally store all the fields, even if they are not initialized.
// We will initialize all fields to their "zero" value. // We will initialize all fields to their "zero" value.
for i := 0; i < strct.NumFields(); i++ { for i := strct.NumFields() - 1; i >= 0; i-- {
sField := strct.Field(i) sField := strct.Field(i)
var initialized bool var initialized bool
emit.Opcodes(c.prog.BinWriter, opcode.DUP)
emit.Int(c.prog.BinWriter, int64(i))
if !keyedLit { if !keyedLit {
if len(lit.Elts) > i { if len(lit.Elts) > i {
ast.Walk(c, lit.Elts[i]) ast.Walk(c, lit.Elts[i])
@ -1870,7 +1860,12 @@ func (c *codegen) convertStruct(lit *ast.CompositeLit, ptr bool) {
if !initialized { if !initialized {
c.emitDefault(sField.Type()) c.emitDefault(sField.Type())
} }
emit.Opcodes(c.prog.BinWriter, opcode.SETITEM) }
emit.Int(c.prog.BinWriter, int64(strct.NumFields()))
if ptr {
emit.Opcodes(c.prog.BinWriter, opcode.PACK)
} else {
emit.Opcodes(c.prog.BinWriter, opcode.PACKSTRUCT)
} }
} }