MetzNet
/
dfhack
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

codegen ++

develop
jj 2012-04-01 01:33:30 +02:00
parent ca528fff35
commit 780d7a8b18
1 changed files with 220 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

250
plugins/ruby/codegen.pl
Unescape Escape View File

@ -21,60 +21,93 @@ sub indent_rb(&) {
sub rb_ucase { sub rb_ucase {
my ($name) = @_; my ($name) = @_;
return $name if ($name eq uc($name));
return join("", map { ucfirst $_ } (split('_', $name))); return join("", map { ucfirst $_ } (split('_', $name)));
} }
my %render_global_type = (
'enum-type' => \&render_global_enum,
'struct-type' => \&render_global_class,
'class-type' => \&render_global_class,
'bitfield-type' => \&render_global_bitfield,
);
my %render_struct_field = ( my %render_struct_field = (
'global' => \&render_field_global, 'global' => \&render_field_global,
'number' => \&render_field_number, 'number' => \&render_field_number,
'container' => \&render_field_container, 'container' => \&render_field_container,
'compound' => \&render_field_compound,
'pointer' => \&render_field_pointer,
'static-array' => \&render_field_array,
'primitive' => \&render_field_primitive,
'bytes' => \&render_field_bytes,
); );
my %render_global_type = ( my %render_item = (
'enum-type' => \&render_global_enum, 'number' => \&render_item_number,
'struct-type' => \&render_global_class, 'global' => \&render_item_global,
'class-type' => \&render_global_class, 'primitive' => \&render_item_primitive,
'bitfield-type' => \&render_global_bitfield, 'pointer' => \&render_item_pointer,
'static-array' => \&render_item_array,
'container' => \&render_item_container,
'compound' => \&render_item_compound,
); );
sub render_global_enum { sub render_global_enum {
my ($name, $type) = @_; my ($name, $type) = @_;
my $value = -1;
my $rbname = rb_ucase($name); my $rbname = rb_ucase($name);
push @lines_rb, "class $rbname"; push @lines_rb, "class $rbname";
indent_rb { indent_rb {
for my $item ($type->findnodes('child::enum-item')) { render_enum_fields($type);
$value = $item->getAttribute('value') || ($value+1);
my $elemname = $item->getAttribute('name'); # || "unk_$value";
if ($elemname) {
my $rbelemname = rb_ucase($elemname);
push @lines_rb, "$rbelemname = $value";
}
}
}; };
push @lines_rb, "end"; push @lines_rb, "end";
} }
sub render_enum_fields {
my ($type) = @_;
my $value = -1;
for my $item ($type->findnodes('child::enum-item')) {
$value = $item->getAttribute('value') || ($value+1);
my $elemname = $item->getAttribute('name'); # || "unk_$value";
if ($elemname) {
my $rbelemname = rb_ucase($elemname);
push @lines_rb, "$rbelemname = $value";
}
}
}
sub render_global_bitfield { sub render_global_bitfield {
my ($name, $type) = @_; my ($name, $type) = @_;
my $rbname = rb_ucase($name); my $rbname = rb_ucase($name);
push @lines_rb, "class $rbname < MemStruct"; push @lines_rb, "class $rbname < MemStruct";
indent_rb { indent_rb {
my $shift = 0; render_bitfield_fields($type);
for my $field ($type->findnodes('child::ld:field')) {
my $count = $field->getAttribute('count') || 1;
print "bitfield $name !number\n" if (!($field->getAttribute('ld:meta') eq 'number'));
my $fname = $field->getAttribute('name');
push @lines_rb, "bits :$fname, $shift, $count" if ($fname);
$shift += $count;
}
}; };
push @lines_rb, "end"; push @lines_rb, "end";
} }
sub render_bitfield_fields {
my ($type) = @_;
my $shift = 0;
for my $field ($type->findnodes('child::ld:field')) {
my $count = $field->getAttribute('count') || 1;
my $name = $field->getAttribute('name');
$name = $field->getAttribute('ld:anon-name') if (!$name);
print "bitfield $name !number\n" if (!($field->getAttribute('ld:meta') eq 'number'));
if ($count == 1) {
push @lines_rb, "bit :$name, $shift" if ($name);
} else {
push @lines_rb, "bits :$name, $shift, $count" if ($name);
}
$shift += $count;
}
}
sub render_global_class { sub render_global_class {
my ($name, $type) = @_; my ($name, $type) = @_;
@ -96,7 +129,7 @@ sub render_field_global {
my $rbname = rb_ucase($typename); my $rbname = rb_ucase($typename);
my $offset = "'offsetof_$name'"; my $offset = "'offsetof_$name'";
push @lines_rb, "global :$name, :$rbname, $offset"; push @lines_rb, "global :$name, $offset, :$rbname";
} }
sub render_field_number { sub render_field_number {
@ -104,8 +137,31 @@ sub render_field_number {
my $subtype = $field->getAttribute('ld:subtype'); my $subtype = $field->getAttribute('ld:subtype');
my $offset = "'offsetof_$name'"; my $offset = "'offsetof_$name'";
$subtype = 'float' if ($subtype eq 's-float');
push @lines_rb, "number :$name, :$subtype, $offset" push @lines_rb, "$subtype :$name, $offset"
}
sub render_field_compound {
my ($name, $field) = @_;
my $offset = "'offsetof_$name'";
my $subtype = $field->getAttribute('ld:subtype');
if (!$subtype || $subtype eq 'bitfield') {
push @lines_rb, "compound(:$name, $offset) {";
indent_rb {
if (!$subtype) {
render_struct_fields($field);
} else {
render_bitfield_fields($field);
}
};
push @lines_rb, "}"
} elsif ($subtype eq 'enum') {
render_enum_fields($field);
} else {
print "no render compound $subtype\n";
}
} }
sub render_field_container { sub render_field_container {
@ -114,21 +170,81 @@ sub render_field_container {
my $subtype = $field->getAttribute('ld:subtype'); my $subtype = $field->getAttribute('ld:subtype');
my $offset = "'offsetof_$name'"; my $offset = "'offsetof_$name'";
if ($subtype eq 'stl-vector') { if ($subtype eq 'stl-deque') {
my $elem = $field->findnodes('child::ld:item')->[0] or return; push @lines_rb, "stl_deque :$name, $offset";
} elsif ($subtype eq 'stl-bit-vector') {
push @lines_rb, "vector :$name, :$subtype, $offset" push @lines_rb, "stl_bitvector :$name, $offset";
} elsif ($subtype eq 'stl-vector') {
my $item = $field->findnodes('child::ld:item')->[0] or return;
my $itemdesc = render_item($item);
push @lines_rb, "stl_vector(:$name, $offset, $itemdesc";
} elsif ($subtype eq 'df-array') {
my $item = $field->findnodes('child::ld:item')->[0] or return;
my $itemdesc = render_item($item);
push @lines_rb, "df_array(:$name, $offset, $itemdesc";
} elsif ($subtype eq 'df-linked-list') {
my $item = $field->findnodes('child::ld:item')->[0] or return;
my $itemdesc = render_item($item);
push @lines_rb, "df_linklist(:$name, $offset, $itemdesc";
} elsif ($subtype eq 'df-flagarray') {
push @lines_rb, "df_flagarray :$name, $offset";
} else { } else {
print "no render field container $subtype\n"; print "no render field container $subtype\n";
} }
} }
sub render_field_pointer {
my ($name, $field) = @_;
my $offset = "'offsetof_$name'";
my $item = $field->findnodes('child::ld:item')->[0] or return;
my $itemdesc = render_item($item);
push @lines_rb, "pointer(:$name, $offset, $itemdesc";
}
sub render_field_array {
my ($name, $field) = @_;
my $offset = "'offsetof_$name'";
my $count = $field->getAttribute('count');
my $item = $field->findnodes('child::ld:item')->[0] or return;
my $itemdesc = render_item($item);
push @lines_rb, "staticarray(:$name, $offset, $count, $itemdesc";
}
sub render_field_primitive {
my ($name, $field) = @_;
my $offset = "'offsetof_$name'";
my $subtype = $field->getAttribute('ld:subtype');
if ($subtype eq 'stl-string') {
push @lines_rb, "string :$name, $offset";
} else {
print "no render primitive $subtype $name\n";
}
}
sub render_field_bytes {
my ($name, $field) = @_;
my $offset = "'offsetof_$name'";
my $subtype = $field->getAttribute('ld:subtype');
if ($subtype eq 'padding') {
} elsif ($subtype eq 'static-string') {
my $size = $field->getAttribute('ld:subtype');
push @lines_rb, "static_string :$name, $offset, $size";
} else {
print "no render field bytes $subtype $name\n";
}
}
sub render_struct_fields { sub render_struct_fields {
my ($type) = @_; my ($type) = @_;
for my $field ($type->findnodes('child::ld:field')) { for my $field ($type->findnodes('child::ld:field')) {
my $name = $field->getAttribute('name'); my $name = $field->getAttribute('name');
#$name = $field->getAttribute('ld:anon-name') if (!$name); $name = $field->getAttribute('ld:anon-name') if (!$name);
next if (!$name); next if (!$name);
my $meta = $field->getAttribute('ld:meta'); my $meta = $field->getAttribute('ld:meta');
@ -141,7 +257,81 @@ sub render_struct_fields {
} }
} }
sub render_item_number {
my ($item) = @_;
my $sz = $item->getAttribute('ld:bits') / 8;
my $subtype = $item->getAttribute('ld:subtype');
$subtype = 'float' if ($subtype eq 's-float');
return "$sz) { a_$subtype }";
}
sub render_item_global {
my ($item) = @_;
my $typename = $item->getAttribute('type-name');
my $rbname = rb_ucase($typename);
return "nil) { a_global :$rbname }";
}
sub render_item_primitive {
my ($item) = @_;
my $subtype = $item->getAttribute('ld:subtype');
if ($subtype eq 'stl-string') {
return "nil) { a_string }";
} else {
print "no render a_primitive $subtype\n";
return "nil)";
}
}
sub render_item_pointer {
my ($item) = @_;
my $pitem = $item->findnodes('child::ld:item')->[0];
my $desc;
if ($pitem) {
$desc = render_item($pitem);
} else {
$desc = 'nil)';
}
return "4) { a_pointer($desc }";
}
sub render_item_array {
my ($item) = @_;
print "no render item array\n";
return "nil)";
}
sub render_item_container {
my ($item) = @_;
print "no render item container\n";
return "nil)";
}
sub render_item_compound {
my ($item) = @_;
print "no render item compound\n";
return "nil)";
}
sub render_item {
my ($item) = @_;
my $meta = $item->getAttribute('ld:meta');
my $renderer = $render_item{$meta};
if ($renderer) {
return $renderer->($item);
} else {
print "no render field $meta\n";
return "nil)";
}
}
my $input = $ARGV[0] || '../../library/include/df/codegen.out.xml'; my $input = $ARGV[0] || '../../library/include/df/codegen.out.xml';
my $output_rb = $ARGV[1] || 'ruby-autogen.rb'; my $output_rb = $ARGV[1] || 'ruby-autogen.rb';