Zack Buhman
118942521e
This also adds support for "#" characters prior to immediates. nop may also now appear in an op_t. The parser no longer generates nop_t--this is instead now represented as an op_t with a zero-length ops vector.
112 lines
2.6 KiB
C++
112 lines
2.6 KiB
C++
#include <iostream>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <optional>
|
|
#include <bitset>
|
|
#include <sstream>
|
|
|
|
#include "lexer.hpp"
|
|
#include "token.hpp"
|
|
#include "ast_printer.hpp"
|
|
#include "ast_resolver.hpp"
|
|
#include "ast_emitter.hpp"
|
|
#include "parser.hpp"
|
|
|
|
namespace dsp {
|
|
|
|
bool had_error = false;
|
|
|
|
}
|
|
|
|
static void write(std::ostream& os, uint32_t v)
|
|
{
|
|
const char out[4] = {
|
|
static_cast<char>((v >> 24) & 0xff),
|
|
static_cast<char>((v >> 16) & 0xff),
|
|
static_cast<char>((v >> 8 ) & 0xff),
|
|
static_cast<char>((v >> 0 ) & 0xff),
|
|
};
|
|
os.write(out, 4);
|
|
}
|
|
|
|
static void run(std::ostream& os, std::string source)
|
|
{
|
|
using namespace dsp;
|
|
|
|
std::string_view buf(source);
|
|
lexer_t lexer(buf);
|
|
std::vector<token_t> tokens = lexer.lex_tokens();
|
|
parser_t pass1(tokens);
|
|
parser_t pass2(tokens);
|
|
ast::printer_t printer(std::cout);
|
|
ast::pc_t pc;
|
|
ast::variables_t variables;
|
|
ast::resolver_t resolver(pc, variables);
|
|
while (auto stmt_o = pass1.statement()) {
|
|
(*stmt_o)->accept(&printer);
|
|
std::cout << std::endl << std::flush;
|
|
(*stmt_o)->accept(&resolver);
|
|
}
|
|
ast::emitter_t emitter(variables);
|
|
while (auto stmt_o = pass2.statement()) {
|
|
uint32_t output = (*stmt_o)->accept(&emitter);
|
|
if (output != 0xffff'ffff) {
|
|
std::cout << std::bitset<32>(output) << std::endl;
|
|
write(os, output);
|
|
}
|
|
}
|
|
}
|
|
|
|
static void run_prompt()
|
|
{
|
|
constexpr auto prompt = "> ";
|
|
std::string line;
|
|
std::cout << prompt << std::flush;
|
|
while (std::getline(std::cin, line)) {
|
|
std::ostringstream os;
|
|
run(os, line);
|
|
std::cout << prompt << std::flush;
|
|
}
|
|
}
|
|
|
|
static int run_file(char const * const input_filename, char const * const output_filename)
|
|
{
|
|
std::ifstream is {input_filename, std::ios::binary | std::ios::ate};
|
|
if (!is.is_open()) {
|
|
std::cerr << "failed to open " << input_filename << std::endl;
|
|
return -1;
|
|
}
|
|
const std::streampos size = is.tellg();
|
|
std::string buf(size, '\0');
|
|
is.seekg(0);
|
|
if (!is.read(&buf[0], size)) {
|
|
std::cerr << "read failed" << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
std::ostringstream os;
|
|
run(os, buf);
|
|
|
|
if (!dsp::had_error) {
|
|
std::ofstream ofs {output_filename, std::ios::binary | std::ios::trunc};
|
|
if (!ofs.is_open()) {
|
|
std::cerr << "failed to open " << output_filename << std::endl;
|
|
return -1;
|
|
}
|
|
ofs << os.str();
|
|
}
|
|
|
|
return dsp::had_error;
|
|
}
|
|
|
|
int main(const int argc, char const * const argv[])
|
|
{
|
|
switch (argc) {
|
|
case 1: run_prompt(); return dsp::had_error;
|
|
case 3: return run_file(argv[1], argv[2]);
|
|
default:
|
|
std::cerr << "Usage: " << argv[0] << " [input-filename output-filename]" << std::endl;
|
|
return -1;
|
|
}
|
|
}
|