print.cpp

#include <cstdlib>
#include <iostream>

#include "vm/program.hpp"
#include "mem/thread.hpp"
#include "vm/instr.hpp"
#include "vm/all.hpp"

using namespace vm;
using namespace std;

int
main(int argc, char **argv)
{
   if(argc < 2) {
      fprintf(stderr, "usage: print <bytecode file> [code | rules | info | prog]\n");
      return EXIT_FAILURE;
   }
   
   const string file(argv[1]);

   try {
      mem::ensure_pool();
      program prog(file);

      theProgram = &prog;

      vm::USING_MEM_ADDRESSES = false;
      if(argc == 2)
         prog.print_bytecode(cout);
      if(argc == 3) {
         const string arg(argv[2]);

         if(arg == "code") {
            prog.print_bytecode(cout);
         } else if(arg == "rules") {
            prog.print_rules(cout);
         } else if(arg == "info") {
            prog.print_predicates(cout);
         } else if(arg == "prog") {
            prog.print_program(cout);
         } else {
            cerr << "Don't know what to do" << endl;
         }
      }
   } catch(vm::load_file_error& err) {
      cerr << "File error: " << err.what() << endl;
      exit(EXIT_FAILURE);
   }
   
   return EXIT_SUCCESS;
}