#include <mruby.h>
#include <mruby/irep.h>
#include <mruby/debug.h>
#include <mruby/opcode.h>
#include <mruby/string.h>
#include <mruby/proc.h>
#include <mruby/dump.h>
#include <mruby/internal.h>

#ifndef MRB_NO_STDIO
static mrb_bool
print_r_p(mrb_state *mrb, const mrb_irep *irep, size_t n)
{
  if (n == 0) return FALSE;
  if (!irep->lv) return FALSE;
  if (n >= irep->nlocals) return FALSE;
  if (!irep->lv[n-1]) return FALSE;
  return TRUE;
}

static void
print_r(mrb_state *mrb, const mrb_irep *irep, size_t n, FILE *out)
{
  if (n == 0) return;
  if (n >= irep->nlocals) return;
  if (!irep->lv[n-1]) return;
  fprintf(out, " R%d:%s", (int)n, mrb_sym_dump(mrb, irep->lv[n-1]));
}

static void
print_lv_a(mrb_state *mrb, const mrb_irep *irep, uint16_t a, FILE *out)
{
  if (print_r_p(mrb, irep, a)) {
    fprintf(out, "\t;");
    print_r(mrb, irep, a, out);
  }
  fprintf(out, "\n");
}

static void
print_lv_ab(mrb_state *mrb, const mrb_irep *irep, uint16_t a, uint16_t b, FILE *out)
{
  if (print_r_p(mrb, irep, a) || print_r_p(mrb, irep, b)) {
    fprintf(out, "\t;");
    print_r(mrb, irep, a, out);
    print_r(mrb, irep, b, out);
  }
  fprintf(out, "\n");
}

static void
print_header(mrb_state *mrb, const mrb_irep *irep, ptrdiff_t i, FILE *out)
{
  int32_t line;

  mrb_assert(i <= UINT32_MAX);
  line = mrb_debug_get_line(mrb, irep, (uint32_t)i);
  if (line < 0) {
    fprintf(out, "      ");
  }
  else {
    fprintf(out, "%5d ", line);
  }

  fprintf(out, "%03d ", (int)i);
}

static void
print_args(uint16_t i, FILE *out)
{
  mrb_assert(i <= 255);
  uint8_t n = i&0xf;
  uint8_t nk = (i>>4)&0xf;

  if (n == 15) {
    fprintf(out, "n=*");
  }
  else {
    fprintf(out, "n=%d", n);
  }
  if (nk > 0) {
    fprintf(out, "|");
    if (nk == 15) {
      fprintf(out, "nk=*");
    }
    else {
      fprintf(out, "nk=%d", nk);
    }
  }
  fprintf(out, "\n");
}

#define CASE(insn,ops) case insn: FETCH_ ## ops (); L_ ## insn

static void
codedump(mrb_state *mrb, const mrb_irep *irep, FILE *out)
{
  const char *file = NULL;

  if (!irep) return;
  fprintf(out, "irep %p nregs=%d nlocals=%d pools=%d syms=%d reps=%d ilen=%d\n", (void*)irep,
          irep->nregs, irep->nlocals, (int)irep->plen, (int)irep->slen, (int)irep->rlen, (int)irep->ilen);

  if (irep->lv) {
    int head = FALSE;

    for (int i = 1; i < irep->nlocals; i++) {
      char const *s = mrb_sym_dump(mrb, irep->lv[i - 1]);
      if (s) {
        if (!head) {
          head = TRUE;
          fprintf(out, "local variable names:\n");
        }
        fprintf(out, "  R%d:%s\n", i, s);
      }
    }
  }

  if (irep->clen > 0) {
    const struct mrb_irep_catch_handler *e = mrb_irep_catch_handler_table(irep);

    for (int i = irep->clen; i > 0; i--,e++) {
      uint32_t begin = mrb_irep_catch_handler_unpack(e->begin);
      uint32_t end = mrb_irep_catch_handler_unpack(e->end);
      uint32_t target = mrb_irep_catch_handler_unpack(e->target);
      char buf[20];
      const char *type;

      switch (e->type) {
        case MRB_CATCH_RESCUE:
          type = "rescue";
          break;
        case MRB_CATCH_ENSURE:
          type = "ensure";
          break;
        default:
          buf[0] = '\0';
          snprintf(buf, sizeof(buf), "0x%02x <unknown>", (int)e->type);
          type = buf;
          break;
      }
      fprintf(out, "catch type: %-8s begin: %04" PRIu32 " end: %04" PRIu32 " target: %04" PRIu32 "\n", type, begin, end, target);
    }
  }

  const mrb_code *pc = irep->iseq;
  const mrb_code *pcend = pc + irep->ilen;
  while (pc < pcend) {
    uint32_t a;
    uint16_t b;
    uint16_t c;
    mrb_code ins;

    int ai = mrb_gc_arena_save(mrb);
    ptrdiff_t i = pc - irep->iseq;

    const char *next_file = mrb_debug_get_filename(mrb, irep, (uint32_t)i);
    if (next_file && file != next_file) {
      fprintf(out, "file: %s\n", next_file);
      file = next_file;
    }
    print_header(mrb, irep, i, out);
    ins = READ_B();
    switch (ins) {
    CASE(OP_NOP, Z):
      fprintf(out, "NOP\n");
      break;
    CASE(OP_MOVE, BB):
      fprintf(out, "MOVE\t\tR%d\tR%d", a, b);
      print_lv_ab(mrb, irep, a, b, out);
      break;

    CASE(OP_LOADL, BB):
      switch (irep->pool[b].tt) {
#ifndef MRB_NO_FLOAT
      case IREP_TT_FLOAT:
        fprintf(out, "LOADL\t\tR%d\tL[%d]\t; %f", a, b, (double)irep->pool[b].u.f);
        break;
#endif
      case IREP_TT_INT32:
        fprintf(out, "LOADL\t\tR%d\tL[%d]\t; %" PRId32, a, b, irep->pool[b].u.i32);
        break;
#ifdef MRB_64BIT
      case IREP_TT_INT64:
        fprintf(out, "LOADL\t\tR%d\tL[%d]\t; %" PRId64, a, b, irep->pool[b].u.i64);
        break;
#endif
      default:
        fprintf(out, "LOADL\t\tR%d\tL[%d]", a, b);
        break;
      }
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADI8, BB):
      fprintf(out, "LOADI8\tR%d\t%d", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADINEG, BB):
      fprintf(out, "LOADINEG\tR%d\t-%d", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADI16, BS):
      fprintf(out, "LOADI16\tR%d\t%d", a, (int)(int16_t)b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADI32, BSS):
      fprintf(out, "LOADI32\tR%d\t%d", a, (int32_t)(((uint32_t)b<<16)+c));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADI__1, B):
      fprintf(out, "LOADI__1\tR%d\t(-1)", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADI_0, B): goto L_LOADI;
    CASE(OP_LOADI_1, B): goto L_LOADI;
    CASE(OP_LOADI_2, B): goto L_LOADI;
    CASE(OP_LOADI_3, B): goto L_LOADI;
    CASE(OP_LOADI_4, B): goto L_LOADI;
    CASE(OP_LOADI_5, B): goto L_LOADI;
    CASE(OP_LOADI_6, B): goto L_LOADI;
    CASE(OP_LOADI_7, B):
    L_LOADI:
      b = ins-(int)OP_LOADI_0;
      fprintf(out, "LOADI_%d\tR%d\t(%d)", b, a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADSYM, BB):
      fprintf(out, "LOADSYM\tR%d\t:%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADNIL, B):
      fprintf(out, "LOADNIL\tR%d\t(nil)", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADSELF, B):
      fprintf(out, "LOADSELF\tR%d\t(R0)", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADT, B):
      fprintf(out, "LOADT\t\tR%d\t(true)", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LOADF, B):
      fprintf(out, "LOADF\t\tR%d\t(false)", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_GETGV, BB):
      fprintf(out, "GETGV\t\tR%d\t%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SETGV, BB):
      fprintf(out, "SETGV\t\t%s\tR%d", mrb_sym_dump(mrb, irep->syms[b]), a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_GETSV, BB):
      fprintf(out, "GETSV\t\tR%d\t%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SETSV, BB):
      fprintf(out, "SETSV\t\t%s\tR%d", mrb_sym_dump(mrb, irep->syms[b]), a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_GETCONST, BB):
      fprintf(out, "GETCONST\tR%d\t%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SETCONST, BB):
      fprintf(out, "SETCONST\t%s\tR%d", mrb_sym_dump(mrb, irep->syms[b]), a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_GETMCNST, BB):
      fprintf(out, "GETMCNST\tR%d\t(R%d)::%s", a, a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SETMCNST, BB):
      fprintf(out, "SETMCNST\t(R%d)::%s\tR%d", a+1, mrb_sym_dump(mrb, irep->syms[b]), a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_GETIV, BB):
      fprintf(out, "GETIV\t\tR%d\t%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SETIV, BB):
      fprintf(out, "SETIV\t\t%s\tR%d", mrb_sym_dump(mrb, irep->syms[b]), a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_GETUPVAR, BBB):
      fprintf(out, "GETUPVAR\tR%d\t%d\t%d", a, b, c);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SETUPVAR, BBB):
      fprintf(out, "SETUPVAR\tR%d\t%d\t%d", a, b, c);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_GETCV, BB):
      fprintf(out, "GETCV\t\tR%d\t%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SETCV, BB):
      fprintf(out, "SETCV\t\t%s\tR%d", mrb_sym_dump(mrb, irep->syms[b]), a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_GETIDX, B):
      fprintf(out, "GETIDX\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_SETIDX, B):
      fprintf(out, "SETIDX\tR%d\t(R%d)\t(R%d)\n", a, a+1, a+2);
      break;
    CASE(OP_JMP, S):
      i = pc - irep->iseq;
      fprintf(out, "JMP\t\t%03d\n", (int)i+(int16_t)a);
      break;
    CASE(OP_JMPUW, S):
      i = pc - irep->iseq;
      fprintf(out, "JMPUW\t\t%03d\n", (int)i+(int16_t)a);
      break;
    CASE(OP_JMPIF, BS):
      i = pc - irep->iseq;
      fprintf(out, "JMPIF\t\tR%d\t%03d", a, (int)i+(int16_t)b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_JMPNOT, BS):
      i = pc - irep->iseq;
      fprintf(out, "JMPNOT\tR%d\t%03d", a, (int)i+(int16_t)b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_JMPNIL, BS):
      i = pc - irep->iseq;
      fprintf(out, "JMPNIL\tR%d\t%03d", a, (int)i+(int16_t)b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SSEND, BBB):
      fprintf(out, "SSEND\t\tR%d\t:%s\t", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_args(c, out);
      break;
    CASE(OP_SSENDB, BBB):
      fprintf(out, "SSENDB\tR%d\t:%s\t", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_args(c, out);
      break;
    CASE(OP_SEND, BBB):
      fprintf(out, "SEND\t\tR%d\t:%s\t", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_args(c, out);
      break;
    CASE(OP_SENDB, BBB):
      fprintf(out, "SENDB\t\tR%d\t:%s\t", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_args(c, out);
      break;
    CASE(OP_CALL, Z):
      fprintf(out, "CALL\n");
      break;
    CASE(OP_SUPER, BB):
      fprintf(out, "SUPER\t\tR%d\t", a);
      print_args(b, out);
      break;
    CASE(OP_ARGARY, BS):
      fprintf(out, "ARGARY\tR%d\t%d:%d:%d:%d (%d)", a,
             (b>>11)&0x3f,
             (b>>10)&0x1,
             (b>>5)&0x1f,
             (b>>4)&0x1,
             (b>>0)&0xf);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_ENTER, W):
      fprintf(out, "ENTER\t\t%d:%d:%d:%d:%d:%d:%d (0x%x)\n",
              MRB_ASPEC_REQ(a),
              MRB_ASPEC_OPT(a),
              MRB_ASPEC_REST(a),
              MRB_ASPEC_POST(a),
              MRB_ASPEC_KEY(a),
              MRB_ASPEC_KDICT(a),
              MRB_ASPEC_BLOCK(a), a);
      break;
    CASE(OP_KEY_P, BB):
      fprintf(out, "KEY_P\t\tR%d\t:%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_KEYEND, Z):
      fprintf(out, "KEYEND\n");
      break;
    CASE(OP_KARG, BB):
      fprintf(out, "KARG\t\tR%d\t:%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_RETURN, B):
      fprintf(out, "RETURN\tR%d\t", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_RETURN_BLK, B):
      fprintf(out, "RETURN_BLK\tR%d\t", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_BREAK, B):
      fprintf(out, "BREAK\t\tR%d\t", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_BLKPUSH, BS):
      fprintf(out, "BLKPUSH\tR%d\t%d:%d:%d:%d (%d)", a,
             (b>>11)&0x3f,
             (b>>10)&0x1,
             (b>>5)&0x1f,
             (b>>4)&0x1,
             (b>>0)&0xf);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_LAMBDA, BB):
      fprintf(out, "LAMBDA\tR%d\tI[%d]\n", a, b);
      break;
    CASE(OP_BLOCK, BB):
      fprintf(out, "BLOCK\t\tR%d\tI[%d]\n", a, b);
      break;
    CASE(OP_METHOD, BB):
      fprintf(out, "METHOD\tR%d\tI[%d]\n", a, b);
      break;
    CASE(OP_RANGE_INC, B):
      fprintf(out, "RANGE_INC\tR%d\n", a);
      break;
    CASE(OP_RANGE_EXC, B):
      fprintf(out, "RANGE_EXC\tR%d\n", a);
      break;
    CASE(OP_DEF, BB):
      fprintf(out, "DEF\t\tR%d\t:%s\t(R%d)\n", a, mrb_sym_dump(mrb, irep->syms[b]),a+1);
      break;
    CASE(OP_UNDEF, B):
      fprintf(out, "UNDEF\t\t:%s\n", mrb_sym_dump(mrb, irep->syms[a]));
      break;
    CASE(OP_ALIAS, BB):
      fprintf(out, "ALIAS\t\t:%s\t%s\n", mrb_sym_dump(mrb, irep->syms[a]), mrb_sym_dump(mrb, irep->syms[b]));
      break;
    CASE(OP_ADD, B):
      fprintf(out, "ADD\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_ADDI, BB):
      fprintf(out, "ADDI\t\tR%d\t%d", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SUB, B):
      fprintf(out, "SUB\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_SUBI, BB):
      fprintf(out, "SUBI\t\tR%d\t%d", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_MUL, B):
      fprintf(out, "MUL\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_DIV, B):
      fprintf(out, "DIV\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_LT, B):
      fprintf(out, "LT\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_LE, B):
      fprintf(out, "LE\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_GT, B):
      fprintf(out, "GT\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_GE, B):
      fprintf(out, "GE\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_EQ, B):
      fprintf(out, "EQ\t\tR%d\t(R%d)\n", a, a+1);
      break;
    CASE(OP_ARRAY, BB):
      fprintf(out, "ARRAY\t\tR%d\t%d", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_ARRAY2, BBB):
      fprintf(out, "ARRAY\t\tR%d\tR%d\t%d", a, b, c);
      print_lv_ab(mrb, irep, a, b, out);
      break;
    CASE(OP_ARYCAT, B):
      fprintf(out, "ARYCAT\tR%d\t(R%d)", a, a+1);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_ARYPUSH, BB):
      fprintf(out, "ARYPUSH\tR%d\t%d", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_ARYSPLAT, B):
      fprintf(out, "ARYSPLAT\tR%d", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_AREF, BBB):
      fprintf(out, "AREF\t\tR%d\tR%d\t%d", a, b, c);
      print_lv_ab(mrb, irep, a, b, out);
      break;
    CASE(OP_ASET, BBB):
      fprintf(out, "ASET\t\tR%d\tR%d\t%d", a, b, c);
      print_lv_ab(mrb, irep, a, b, out);
      break;
    CASE(OP_APOST, BBB):
      fprintf(out, "APOST\t\tR%d\t%d\t%d", a, b, c);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_INTERN, B):
      fprintf(out, "INTERN\tR%d\t", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SYMBOL, BB):
      mrb_assert((irep->pool[b].tt&IREP_TT_NFLAG)==0);
      fprintf(out, "SYMBOL\tR%d\tL[%d]\t; %s", a, b, irep->pool[b].u.str);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_STRING, BB):
      mrb_assert((irep->pool[b].tt&IREP_TT_NFLAG)==0);
      fprintf(out, "STRING\tR%d\tL[%d]", a, b);
      if (irep->pool[b].u.str[0]) {
        fprintf(out, "\t; %s", irep->pool[b].u.str);
      }
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_STRCAT, B):
      fprintf(out, "STRCAT\tR%d\t(R%d)", a, a+1);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_HASH, BB):
      fprintf(out, "HASH\t\tR%d\t%d", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_HASHADD, BB):
      fprintf(out, "HASHADD\tR%d\t%d", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_HASHCAT, B):
      fprintf(out, "HASHCAT\tR%d\t(R%d)", a, a+1);
      print_lv_a(mrb, irep, a, out);
      break;

    CASE(OP_OCLASS, B):
      fprintf(out, "OCLASS\tR%d\t", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_CLASS, BB):
      fprintf(out, "CLASS\t\tR%d\t:%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_MODULE, BB):
      fprintf(out, "MODULE\tR%d\t:%s", a, mrb_sym_dump(mrb, irep->syms[b]));
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_EXEC, BB):
      fprintf(out, "EXEC\t\tR%d\tI[%d]", a, b);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_SCLASS, B):
      fprintf(out, "SCLASS\tR%d", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_TCLASS, B):
      fprintf(out, "TCLASS\tR%d\t", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_ERR, B):
      if ((irep->pool[a].tt & IREP_TT_NFLAG) == 0) {
        fprintf(out, "ERR\t\t%s\n", irep->pool[a].u.str);
      }
      else {
        fprintf(out, "ERR\tL[%d]\n", a);
      }
      break;
    CASE(OP_EXCEPT, B):
      fprintf(out, "EXCEPT\tR%d\t", a);
      print_lv_a(mrb, irep, a, out);
      break;
    CASE(OP_RESCUE, BB):
      fprintf(out, "RESCUE\tR%d\tR%d", a, b);
      print_lv_ab(mrb, irep, a, b, out);
      break;
    CASE(OP_RAISEIF, B):
      fprintf(out, "RAISEIF\tR%d\t", a);
      print_lv_a(mrb, irep, a, out);
      break;

    CASE(OP_DEBUG, BBB):
      fprintf(out, "DEBUG\t\t%d\t%d\t%d\n", a, b, c);
      break;

    CASE(OP_STOP, Z):
      fprintf(out, "STOP\n");
      break;

    CASE(OP_EXT1, Z):
      fprintf(out, "EXT1\n");
      print_header(mrb, irep, pc-irep->iseq, out);
      ins = READ_B();
      switch (ins) {
#define OPCODE(i,x) case OP_ ## i: FETCH_ ## x ## _1 (); goto L_OP_ ## i;
#include <mruby/ops.h>
#undef OPCODE
      }
      break;
    CASE(OP_EXT2, Z):
      fprintf(out, "EXT2\n");
      print_header(mrb, irep, pc-irep->iseq, out);
      ins = READ_B();
      switch (ins) {
#define OPCODE(i,x) case OP_ ## i: FETCH_ ## x ## _2 (); goto L_OP_ ## i;
#include <mruby/ops.h>
#undef OPCODE
      }
      break;
    CASE(OP_EXT3, Z):
      fprintf(out, "EXT3\n");
      print_header(mrb, irep, pc-irep->iseq, out);
      ins = READ_B();
      switch (ins) {
#define OPCODE(i,x) case OP_ ## i: FETCH_ ## x ## _3 (); goto L_OP_ ## i;
#include <mruby/ops.h>
#undef OPCODE
      }
      break;

    default:
      fprintf(out, "unknown_op (0x%x)\n", ins);
      break;
    }
    mrb_gc_arena_restore(mrb, ai);
  }
  fprintf(out, "\n");
}

static void
codedump_recur(mrb_state *mrb, const mrb_irep *irep, FILE *out)
{
  codedump(mrb, irep, out);
  if (irep->reps) {
    for (int i=0; i<irep->rlen; i++) {
      codedump_recur(mrb, irep->reps[i], out);
    }
  }
}

void
mrb_codedump_all_file(mrb_state *mrb, struct RProc *proc, FILE *out)
{
  codedump_recur(mrb, proc->body.irep, out);
  fflush(out);
}

#endif

void
mrb_codedump_all(mrb_state *mrb, struct RProc *proc)
{
#ifndef MRB_NO_STDIO
  mrb_codedump_all_file(mrb, proc, stdout);
#endif
}
#undef CASE
