fmt: move asm related code into it's own file (#10874)
Lukas Neubert
GitHub
parent
f6402eae10
commit
05d0288e8d
|
|
@ -0,0 +1,176 @@
|
||||||
|
// Copyright (c) 2019-2021 Alexander Medvednikov. All rights reserved.
|
||||||
|
// Use of this source code is governed by an MIT license
|
||||||
|
// that can be found in the LICENSE file.
|
||||||
|
module fmt
|
||||||
|
|
||||||
|
import v.ast
|
||||||
|
|
||||||
|
fn (mut f Fmt) asm_stmt(stmt ast.AsmStmt) {
|
||||||
|
f.write('asm ')
|
||||||
|
if stmt.is_volatile {
|
||||||
|
f.write('volatile ')
|
||||||
|
} else if stmt.is_goto {
|
||||||
|
f.write('goto ')
|
||||||
|
}
|
||||||
|
f.writeln('$stmt.arch {')
|
||||||
|
f.indent++
|
||||||
|
for template in stmt.templates {
|
||||||
|
if template.is_directive {
|
||||||
|
f.write('.')
|
||||||
|
}
|
||||||
|
f.write('$template.name')
|
||||||
|
if template.is_label {
|
||||||
|
f.write(':')
|
||||||
|
} else {
|
||||||
|
if template.args.len > 0 {
|
||||||
|
f.write(' ')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for i, arg in template.args {
|
||||||
|
f.asm_arg(arg)
|
||||||
|
if i + 1 < template.args.len {
|
||||||
|
f.write(', ')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if template.comments.len == 0 {
|
||||||
|
f.writeln('')
|
||||||
|
} else {
|
||||||
|
f.comments(template.comments, inline: false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if stmt.output.len != 0 || stmt.input.len != 0 || stmt.clobbered.len != 0 {
|
||||||
|
f.write('; ')
|
||||||
|
}
|
||||||
|
f.asm_ios(stmt.output)
|
||||||
|
|
||||||
|
if stmt.input.len != 0 || stmt.clobbered.len != 0 {
|
||||||
|
f.write('; ')
|
||||||
|
}
|
||||||
|
f.asm_ios(stmt.input)
|
||||||
|
|
||||||
|
if stmt.clobbered.len != 0 {
|
||||||
|
f.write('; ')
|
||||||
|
}
|
||||||
|
for i, clob in stmt.clobbered {
|
||||||
|
if i != 0 {
|
||||||
|
f.write(' ')
|
||||||
|
}
|
||||||
|
f.write(clob.reg.name)
|
||||||
|
|
||||||
|
if clob.comments.len == 0 {
|
||||||
|
f.writeln('')
|
||||||
|
} else {
|
||||||
|
f.comments(clob.comments, inline: false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
f.indent--
|
||||||
|
f.writeln('}')
|
||||||
|
if f.indent == 0 {
|
||||||
|
f.writeln('')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn (mut f Fmt) asm_arg(arg ast.AsmArg) {
|
||||||
|
match arg {
|
||||||
|
ast.AsmRegister {
|
||||||
|
f.asm_reg(arg)
|
||||||
|
}
|
||||||
|
ast.AsmAlias {
|
||||||
|
f.write('$arg.name')
|
||||||
|
}
|
||||||
|
ast.IntegerLiteral, ast.FloatLiteral, ast.CharLiteral {
|
||||||
|
f.write(arg.val)
|
||||||
|
}
|
||||||
|
ast.BoolLiteral {
|
||||||
|
f.write(arg.val.str())
|
||||||
|
}
|
||||||
|
string {
|
||||||
|
f.write(arg)
|
||||||
|
}
|
||||||
|
ast.AsmAddressing {
|
||||||
|
f.write('[')
|
||||||
|
base := arg.base
|
||||||
|
index := arg.index
|
||||||
|
displacement := arg.displacement
|
||||||
|
scale := arg.scale
|
||||||
|
match arg.mode {
|
||||||
|
.base {
|
||||||
|
f.asm_arg(base)
|
||||||
|
}
|
||||||
|
.displacement {
|
||||||
|
f.asm_arg(displacement)
|
||||||
|
}
|
||||||
|
.base_plus_displacement {
|
||||||
|
f.asm_arg(base)
|
||||||
|
f.write(' + ')
|
||||||
|
f.asm_arg(displacement)
|
||||||
|
}
|
||||||
|
.index_times_scale_plus_displacement {
|
||||||
|
f.asm_arg(index)
|
||||||
|
f.write(' * $scale + ')
|
||||||
|
f.asm_arg(displacement)
|
||||||
|
}
|
||||||
|
.base_plus_index_plus_displacement {
|
||||||
|
f.asm_arg(base)
|
||||||
|
f.write(' + ')
|
||||||
|
f.asm_arg(index)
|
||||||
|
f.write(' + ')
|
||||||
|
f.asm_arg(displacement)
|
||||||
|
}
|
||||||
|
.base_plus_index_times_scale_plus_displacement {
|
||||||
|
f.asm_arg(base)
|
||||||
|
f.write(' + ')
|
||||||
|
f.asm_arg(index)
|
||||||
|
f.write(' * $scale + ')
|
||||||
|
f.asm_arg(displacement)
|
||||||
|
}
|
||||||
|
.rip_plus_displacement {
|
||||||
|
f.asm_arg(base)
|
||||||
|
f.write(' + ')
|
||||||
|
f.asm_arg(displacement)
|
||||||
|
}
|
||||||
|
.invalid {
|
||||||
|
panic('fmt: invalid addressing mode')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
f.write(']')
|
||||||
|
}
|
||||||
|
ast.AsmDisp {
|
||||||
|
if arg.val.len >= 2 && arg.val[arg.val.len - 1] in [`b`, `f`]
|
||||||
|
&& arg.val[..arg.val.len - 1].bytes().all(it.is_digit()) {
|
||||||
|
f.write(arg.val[arg.val.len - 1].ascii_str())
|
||||||
|
f.write(arg.val[..arg.val.len - 1])
|
||||||
|
} else {
|
||||||
|
f.write(arg.val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn (mut f Fmt) asm_reg(reg ast.AsmRegister) {
|
||||||
|
f.write(reg.name)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn (mut f Fmt) asm_ios(ios []ast.AsmIO) {
|
||||||
|
for i, io in ios {
|
||||||
|
if i != 0 {
|
||||||
|
f.write(' ')
|
||||||
|
}
|
||||||
|
|
||||||
|
f.write('$io.constraint ($io.expr)')
|
||||||
|
mut as_block := true
|
||||||
|
if io.expr is ast.Ident {
|
||||||
|
if io.expr.name == io.alias {
|
||||||
|
as_block = false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if as_block && io.alias != '' {
|
||||||
|
f.write(' as $io.alias')
|
||||||
|
}
|
||||||
|
if io.comments.len == 0 {
|
||||||
|
f.writeln('')
|
||||||
|
} else {
|
||||||
|
f.comments(io.comments, inline: false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
199
vlib/v/fmt/fmt.v
199
vlib/v/fmt/fmt.v
|
|
@ -3,7 +3,7 @@
|
||||||
// that can be found in the LICENSE file.
|
// that can be found in the LICENSE file.
|
||||||
module fmt
|
module fmt
|
||||||
|
|
||||||
import math.mathutil as mu
|
import math.mathutil
|
||||||
import v.ast
|
import v.ast
|
||||||
import strings
|
import strings
|
||||||
import v.util
|
import v.util
|
||||||
|
|
@ -17,9 +17,12 @@ const (
|
||||||
|
|
||||||
pub struct Fmt {
|
pub struct Fmt {
|
||||||
pub mut:
|
pub mut:
|
||||||
|
file ast.File
|
||||||
table &ast.Table
|
table &ast.Table
|
||||||
out_imports strings.Builder
|
pref &pref.Preferences
|
||||||
|
is_debug bool
|
||||||
out strings.Builder
|
out strings.Builder
|
||||||
|
out_imports strings.Builder
|
||||||
indent int
|
indent int
|
||||||
empty_line bool
|
empty_line bool
|
||||||
line_len int // the current line length, NB: it counts \t as 4 spaces, and starts at 0 after f.writeln
|
line_len int // the current line length, NB: it counts \t as 4 spaces, and starts at 0 after f.writeln
|
||||||
|
|
@ -29,15 +32,13 @@ pub mut:
|
||||||
array_init_depth int // current level of hierarchy in array init
|
array_init_depth int // current level of hierarchy in array init
|
||||||
single_line_if bool
|
single_line_if bool
|
||||||
cur_mod string
|
cur_mod string
|
||||||
file ast.File
|
|
||||||
did_imports bool
|
did_imports bool
|
||||||
is_assign bool
|
is_assign bool
|
||||||
is_struct_init bool
|
is_struct_init bool
|
||||||
auto_imports []string // automatically inserted imports that the user forgot to specify
|
auto_imports []string // automatically inserted imports that the user forgot to specify
|
||||||
import_pos int // position of the imports in the resulting string for later autoimports insertion
|
import_pos int // position of the imports in the resulting string for later autoimports insertion
|
||||||
used_imports []string // to remove unused imports
|
used_imports []string // to remove unused imports
|
||||||
import_syms_used map[string]bool // to remove unused import symbols.
|
import_syms_used map[string]bool // to remove unused import symbols.
|
||||||
is_debug bool
|
|
||||||
mod2alias map[string]string // for `import time as t`, will contain: 'time'=>'t'
|
mod2alias map[string]string // for `import time as t`, will contain: 'time'=>'t'
|
||||||
use_short_fn_args bool
|
use_short_fn_args bool
|
||||||
single_line_fields bool // should struct fields be on a single line
|
single_line_fields bool // should struct fields be on a single line
|
||||||
|
|
@ -45,18 +46,16 @@ pub mut:
|
||||||
inside_lambda bool
|
inside_lambda bool
|
||||||
inside_const bool
|
inside_const bool
|
||||||
is_mbranch_expr bool // match a { x...y { } }
|
is_mbranch_expr bool // match a { x...y { } }
|
||||||
pref &pref.Preferences
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn fmt(file ast.File, table &ast.Table, pref &pref.Preferences, is_debug bool) string {
|
pub fn fmt(file ast.File, table &ast.Table, pref &pref.Preferences, is_debug bool) string {
|
||||||
mut f := Fmt{
|
mut f := Fmt{
|
||||||
|
file: file
|
||||||
|
table: table
|
||||||
|
pref: pref
|
||||||
|
is_debug: is_debug
|
||||||
out: strings.new_builder(1000)
|
out: strings.new_builder(1000)
|
||||||
out_imports: strings.new_builder(200)
|
out_imports: strings.new_builder(200)
|
||||||
table: table
|
|
||||||
indent: 0
|
|
||||||
file: file
|
|
||||||
is_debug: is_debug
|
|
||||||
pref: pref
|
|
||||||
}
|
}
|
||||||
f.process_file_imports(file)
|
f.process_file_imports(file)
|
||||||
f.set_current_module_name('main')
|
f.set_current_module_name('main')
|
||||||
|
|
@ -69,7 +68,7 @@ pub fn fmt(file ast.File, table &ast.Table, pref &pref.Preferences, is_debug boo
|
||||||
if res.len == 1 {
|
if res.len == 1 {
|
||||||
return f.out_imports.str().trim_space() + '\n'
|
return f.out_imports.str().trim_space() + '\n'
|
||||||
}
|
}
|
||||||
bounded_import_pos := mu.min(res.len, f.import_pos)
|
bounded_import_pos := mathutil.min(res.len, f.import_pos)
|
||||||
return res[..bounded_import_pos] + f.out_imports.str() + res[bounded_import_pos..]
|
return res[..bounded_import_pos] + f.out_imports.str() + res[bounded_import_pos..]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -670,176 +669,6 @@ fn expr_is_single_line(expr ast.Expr) bool {
|
||||||
|
|
||||||
//=== Specific Stmt methods ===//
|
//=== Specific Stmt methods ===//
|
||||||
|
|
||||||
fn (mut f Fmt) asm_stmt(stmt ast.AsmStmt) {
|
|
||||||
f.write('asm ')
|
|
||||||
if stmt.is_volatile {
|
|
||||||
f.write('volatile ')
|
|
||||||
} else if stmt.is_goto {
|
|
||||||
f.write('goto ')
|
|
||||||
}
|
|
||||||
f.writeln('$stmt.arch {')
|
|
||||||
f.indent++
|
|
||||||
for template in stmt.templates {
|
|
||||||
if template.is_directive {
|
|
||||||
f.write('.')
|
|
||||||
}
|
|
||||||
f.write('$template.name')
|
|
||||||
if template.is_label {
|
|
||||||
f.write(':')
|
|
||||||
} else {
|
|
||||||
if template.args.len > 0 {
|
|
||||||
f.write(' ')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for i, arg in template.args {
|
|
||||||
f.asm_arg(arg)
|
|
||||||
if i + 1 < template.args.len {
|
|
||||||
f.write(', ')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if template.comments.len == 0 {
|
|
||||||
f.writeln('')
|
|
||||||
} else {
|
|
||||||
f.comments(template.comments, inline: false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if stmt.output.len != 0 || stmt.input.len != 0 || stmt.clobbered.len != 0 {
|
|
||||||
f.write('; ')
|
|
||||||
}
|
|
||||||
f.asm_ios(stmt.output)
|
|
||||||
|
|
||||||
if stmt.input.len != 0 || stmt.clobbered.len != 0 {
|
|
||||||
f.write('; ')
|
|
||||||
}
|
|
||||||
f.asm_ios(stmt.input)
|
|
||||||
|
|
||||||
if stmt.clobbered.len != 0 {
|
|
||||||
f.write('; ')
|
|
||||||
}
|
|
||||||
for i, clob in stmt.clobbered {
|
|
||||||
if i != 0 {
|
|
||||||
f.write(' ')
|
|
||||||
}
|
|
||||||
f.write(clob.reg.name)
|
|
||||||
|
|
||||||
if clob.comments.len == 0 {
|
|
||||||
f.writeln('')
|
|
||||||
} else {
|
|
||||||
f.comments(clob.comments, inline: false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
f.indent--
|
|
||||||
f.writeln('}')
|
|
||||||
if f.indent == 0 {
|
|
||||||
f.writeln('')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn (mut f Fmt) asm_arg(arg ast.AsmArg) {
|
|
||||||
match arg {
|
|
||||||
ast.AsmRegister {
|
|
||||||
f.asm_reg(arg)
|
|
||||||
}
|
|
||||||
ast.AsmAlias {
|
|
||||||
f.write('$arg.name')
|
|
||||||
}
|
|
||||||
ast.IntegerLiteral, ast.FloatLiteral, ast.CharLiteral {
|
|
||||||
f.write(arg.val)
|
|
||||||
}
|
|
||||||
ast.BoolLiteral {
|
|
||||||
f.write(arg.val.str())
|
|
||||||
}
|
|
||||||
string {
|
|
||||||
f.write(arg)
|
|
||||||
}
|
|
||||||
ast.AsmAddressing {
|
|
||||||
f.write('[')
|
|
||||||
base := arg.base
|
|
||||||
index := arg.index
|
|
||||||
displacement := arg.displacement
|
|
||||||
scale := arg.scale
|
|
||||||
match arg.mode {
|
|
||||||
.base {
|
|
||||||
f.asm_arg(base)
|
|
||||||
}
|
|
||||||
.displacement {
|
|
||||||
f.asm_arg(displacement)
|
|
||||||
}
|
|
||||||
.base_plus_displacement {
|
|
||||||
f.asm_arg(base)
|
|
||||||
f.write(' + ')
|
|
||||||
f.asm_arg(displacement)
|
|
||||||
}
|
|
||||||
.index_times_scale_plus_displacement {
|
|
||||||
f.asm_arg(index)
|
|
||||||
f.write(' * $scale + ')
|
|
||||||
f.asm_arg(displacement)
|
|
||||||
}
|
|
||||||
.base_plus_index_plus_displacement {
|
|
||||||
f.asm_arg(base)
|
|
||||||
f.write(' + ')
|
|
||||||
f.asm_arg(index)
|
|
||||||
f.write(' + ')
|
|
||||||
f.asm_arg(displacement)
|
|
||||||
}
|
|
||||||
.base_plus_index_times_scale_plus_displacement {
|
|
||||||
f.asm_arg(base)
|
|
||||||
f.write(' + ')
|
|
||||||
f.asm_arg(index)
|
|
||||||
f.write(' * $scale + ')
|
|
||||||
f.asm_arg(displacement)
|
|
||||||
}
|
|
||||||
.rip_plus_displacement {
|
|
||||||
f.asm_arg(base)
|
|
||||||
f.write(' + ')
|
|
||||||
f.asm_arg(displacement)
|
|
||||||
}
|
|
||||||
.invalid {
|
|
||||||
panic('fmt: invalid addressing mode')
|
|
||||||
}
|
|
||||||
}
|
|
||||||
f.write(']')
|
|
||||||
}
|
|
||||||
ast.AsmDisp {
|
|
||||||
if arg.val.len >= 2 && arg.val[arg.val.len - 1] in [`b`, `f`]
|
|
||||||
&& arg.val[..arg.val.len - 1].bytes().all(it.is_digit()) {
|
|
||||||
f.write(arg.val[arg.val.len - 1].ascii_str())
|
|
||||||
f.write(arg.val[..arg.val.len - 1])
|
|
||||||
} else {
|
|
||||||
f.write(arg.val)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn (mut f Fmt) asm_reg(reg ast.AsmRegister) {
|
|
||||||
f.write(reg.name)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn (mut f Fmt) asm_ios(ios []ast.AsmIO) {
|
|
||||||
for i, io in ios {
|
|
||||||
if i != 0 {
|
|
||||||
f.write(' ')
|
|
||||||
}
|
|
||||||
|
|
||||||
f.write('$io.constraint ($io.expr)')
|
|
||||||
mut as_block := true
|
|
||||||
if io.expr is ast.Ident {
|
|
||||||
if io.expr.name == io.alias {
|
|
||||||
as_block = false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if as_block && io.alias != '' {
|
|
||||||
f.write(' as $io.alias')
|
|
||||||
}
|
|
||||||
if io.comments.len == 0 {
|
|
||||||
f.writeln('')
|
|
||||||
} else {
|
|
||||||
f.comments(io.comments, inline: false)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn (mut f Fmt) assert_stmt(node ast.AssertStmt) {
|
pub fn (mut f Fmt) assert_stmt(node ast.AssertStmt) {
|
||||||
f.write('assert ')
|
f.write('assert ')
|
||||||
if node.expr is ast.ParExpr {
|
if node.expr is ast.ParExpr {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue