1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

array: map() method

This commit is contained in:
Alexander Medvednikov 2019-10-26 14:25:03 +03:00
parent f6f5b8cfb1
commit 4ef10c92f4
2 changed files with 43 additions and 0 deletions

View File

@ -374,3 +374,42 @@ fn (p mut Parser) gen_array_filter(str_typ string, method_ph int) {
p.check(.rpar) p.check(.rpar)
p.close_scope() p.close_scope()
} }
fn (p mut Parser) gen_array_map(str_typ string, method_ph int) {
/*
// V
a := [1,2,3,4]
b := a.map(it * 2)
// C
array_int a = ...;
array_int tmp2 = new_array(0, 4, 4);
for (int i = 0; i < a.len; i++) {
int it = ((int*)a.data)[i];
_PUSH(tmp2, it * 2, tmp3, int)
}
array_int b = tmp2;
*/
val_type:=str_typ.right(6)
p.open_scope()
p.register_var(Var{
name: 'it'
typ: val_type
})
p.next()
p.check(.lpar)
p.cgen.resetln('')
tmp := p.get_tmp()
tmp_elm := p.get_tmp()
a := p.expr_var.name
map_type, expr := p.tmp_expr()
p.cgen.set_placeholder(method_ph,'\narray $tmp = new_array(0, $a .len, ' +
'sizeof($map_type));\n')
p.genln('for (int i = 0; i < ${a}.len; i++) {')
p.genln('$val_type it = (($val_type*)${a}.data)[i];')
p.genln('_PUSH(&$tmp, $expr, $tmp_elm, $map_type)')
p.genln('}')
p.gen(tmp) // TODO why does this `gen()` work?
p.check(.rpar)
p.close_scope()
}

View File

@ -1947,6 +1947,10 @@ fn (p mut Parser) dot(str_typ_ string, method_ph int) string {
p.gen_array_filter(str_typ, method_ph) p.gen_array_filter(str_typ, method_ph)
return str_typ return str_typ
} }
else if field_name == 'map' && str_typ.starts_with('array_') {
p.gen_array_map(str_typ, method_ph)
return str_typ
}
fname_tidx := p.cur_tok_index() fname_tidx := p.cur_tok_index()
p.fgen(field_name) p.fgen(field_name)