#!/usr/local/bin/perl

# 実験中。コアは完成。offset.pl を参考にした。

# C ソース用の配列 or 変数定義から、バイトオフセットを求め、
# CPP 用に #define の形で出力する。
# Usage: catbin.pl define-file
# define-file は通常 imagedata.inc などの形で、 C ソースを使う。
# (*.c だと Makefile の依存関係が書きにくいので、*.c から imagedata.inc を #include するなど工夫する)
# #define CATBIN_BEGIN tagname
# で、配列の集合体名を出す。
# (SIZEOF_tagname を求めるため)
# パースの方針は
# TYPE_SPECIFIER IDENT[] = {ELEM1, ELEM2, ...};
# TYPE_SPECIFIER IDENT = VALUE; (未対応)
# TYPE_SPECIFIER IDENT; (未対応 -> BSS に入るので動作しない!! 後述のように対処する)
# などの形がある。
# TYPE_SPECIFIER IDENT[] = {ELEM1, ELEM2}, IDENT2[]={ELEM3};
# などの複数定義形には未対応と思う。offset.pl みたいにすればいける ?
# ; が来るまでを 改行コードを無視して一つのレコードとする。
# そのレコードの最初の要素は TYPE_SPECIFIER なので削除。
# (TYPE_SPECIFIER は 1 word とは限らない。 [, ; or , の 
# 直前の識別子 の前までが TYPE_SPECIFIER である。)

# 先頭は 16byte (128bit) alignment しておくこと。
# 末尾は 自動的に 16byte alignment される。
# (圧縮、展開, DMAなど, C object linking での alignment を考慮)
# すべてが .data セクションに入らなければやばい。
# そのため、配列でなくても、C から直接代入とかしないなら
# TYPE_SPECIFIER IDENT[1] = 0; のように (0 は必要)
# 配列で宣言する。
# .bss や .rodata セクションに入ると重大な誤動作となる。

# 繰り返すが、byte offset である。十分注意のこと。

while (<>) {
# 配列の集合体名
    if (/^\#define CATBIN_BEGIN/) {
	($gomi, $gomi, $TAGNAME) = split;
    } else {
	$IN .= $_;
    }
}

# コメントを削除
$IN =~ s/\/\*.*?\*\///sg;


# 構造体定義の始まるまでを削除 (ネスト不可)
# $IN =~ s/.*?\{//gs;

# 構造体定義の終わり 以降を削除 (ネスト不可)
# $IN =~ s/\}.*//gs;

# CATBIN_END を削除
$IN =~ s/^CATBIN_END//m;

# この状態をバックアップ
$IN_ORG = $IN;

# プリプロセッサ指示行を削除
$IN =~ s/^#.*//gm;

# 配列添字の削除
# $IN =~ s/\[.+?\]//gs;

# ポインタ指定子の削除
# $IN =~ s/\*//gs;

# 空行、行頭の空白を削除
$IN =~ s/^\s+//gm;

# 行末の空白を削除
$IN =~ s/;\s+/;/gm;

@list = split(/;/, $IN);

$OUTNAME = "/tmp/offset-tmp$$.c";
open OUT, ">" . $OUTNAME;

print OUT <<EOF;
#include <ultra64.h>
#include <stdio.h>
#ifndef _64
#define _64
#endif
#ifndef _32
#define _32
#endif
#ifndef _16
#define _16
#endif
#ifndef STATIC
#define STATIC
#endif
unsigned long long int fORCe_1_sTRUCTURe_aLIGNMENt[1] = {0};
$IN_ORG
unsigned long long int fORCe_2_sTRUCTURe_aLIGNMENt[1] = {0};

void alignerr(char *member, int i, int ofs)
{
  fprintf(stderr, "Error: member \'%s\': must be %dbit aligned (current offset is 0x%x)\\n", member, i << 3, ofs);
  exit(-1);
}
void alignwarn(char *member, int ofs, int lastofs)
{
  fprintf(stderr, "Warning: member \'%s\' is aligned (current/last offset 0x%x/0x%x)\\n", member, ofs, lastofs);
}
int main() {
  int k;
  int i, j;
  j = 0;
  k = (int)&fORCe_1_sTRUCTURe_aLIGNMENt + sizeof(fORCe_1_sTRUCTURe_aLIGNMENt);
EOF

foreach $line (@list) {
# アラインメント指定子 (_64, _32 など) を抜き出す。
    $align = $line;
    $align =~ s/^\s*\_(\d+).+/\1/;
    if ($align != 0) {
	if ($align & 0x07) {
	    print STDERR "$line\n", "Error: alignment specifier ($align) is invalid\n";
	    die;
	}
	$align >>= 3;
    }
# 識別子のみ抜き出す。
# まず、型指定子を削除
    $line =~ s/(\w|\s)+\s+(\w+\s*[;,]{0,1}.*)/\2/;
# 配列添字の削除
    $line =~ s/\[.*?\]//gs;
# ポインタ指定子の削除
    $line =~ s/\*//gs;
# 初期化子の削除
    $line =~ s/\=//gs;
# 初期値定義の削除
    $line =~ s/\{.*?\}//gs;

    $word = $line;
    print OUT "  i = (int)&$word - k;";
    if ($align > 1) {
	print OUT "  if (i & ", $align - 1, ") alignerr(\"$word\", $align, i);\n";
    }
    print OUT "  if (i != j) alignwarn(\"$word\", i, j);\n";
    print OUT '  printf("#define OFFSET_', uc($word), ' 0x%x\n", i);', "\n";
    print OUT "  j = i + sizeof($word);\n";
}
# ここまでに登場した配列の大きさの総計: RSP DMA にも対応できるように。
# しかも、C object としてコンパイルされるため 16byte alignment しよう。
print OUT '  printf("#define SIZEOF_', uc($TAGNAME), ' 0x%x\n", ',
"(j + 15) & 0xfffffff0);\n";

print OUT '  if (j & 0x0f)
    fprintf(stderr, "Warning: sizeof(the structure)=0x%x, not 16byte-aligned\\n",
            j);
';

# #define のみ 抽出
@define_list = split /\n/, $IN_ORG;

print OUT '  printf("#ifndef __CATBIN_PL__\n");', "\n";
foreach (@define_list) {
    chomp;
    if (/^\#define/) {
	print OUT '  printf("', $_, '\n");', "\n";
    }
}
print OUT '  printf("#endif\n");', "\n";
print OUT '  return 0;
}
';

close OUT;

$i = system("cc -o /tmp/offset-tmp$$ $OUTNAME");
if ($i) { die; }

$i = system("/tmp/offset-tmp$$");
unlink("/tmp/offset-tmp$$");
unlink($OUTNAME);
if ($i) { die; }