pico-sdk/tools/pioasm/pio_assembler.h
graham sanderson 26653ea81e Initial Release
2021-01-20 10:44:27 -06:00

103 lines
3.2 KiB
C++

/*
* Copyright (c) 2020 Raspberry Pi (Trading) Ltd.
*
* SPDX-License-Identifier: BSD-3-Clause
*/
#ifndef _PIO_ASSEMBLER_H
#define _PIO_ASSEMBLER_H
#include <algorithm>
#include "parser.hpp"
#include "output_format.h"
// Give Flex the prototype of yylex we want ...
# define YY_DECL \
yy::parser::symbol_type yylex (pio_assembler& pioasm)
// ... and declare it for the parser's sake.
YY_DECL;
struct pio_assembler {
public:
using syntax_error = yy::parser::syntax_error;
using location_type = yy::parser::location_type;
std::shared_ptr<program> dummy_global_program;
std::vector<program> programs;
int error_count = 0;
pio_assembler();
std::shared_ptr<output_format> format;
// The name of the file being parsed.
std::string source;
// name of the output file or "-" for stdout
std::string dest;
std::vector<std::string> options;
int write_output();
bool add_program(const yy::location &l, const std::string &name) {
if (std::find_if(programs.begin(), programs.end(), [&](const program &p) { return p.name == name; }) ==
programs.end()) {
programs.emplace_back(this, l, name);
return true;
} else {
return false;
}
}
program &get_dummy_global_program() {
if (!dummy_global_program) {
dummy_global_program = std::shared_ptr<program>(new program(this, yy::location(&source), ""));
}
return *dummy_global_program;
}
program &get_current_program(const location_type &l, const std::string &requiring_program,
bool before_any_instructions = false, bool disallow_global = true) {
if (programs.empty()) {
if (disallow_global) {
std::stringstream msg;
msg << requiring_program << " is invalid outside of a program";
throw syntax_error(l, msg.str());
}
return get_dummy_global_program();
}
auto &p = programs[programs.size() - 1];
if (before_any_instructions && !p.instructions.empty()) {
std::stringstream msg;
msg << requiring_program << " must preceed any program instructions";
throw syntax_error(l, msg.str());
}
return p;
}
// note p may be null for global symbols only
std::shared_ptr<symbol> get_symbol(const std::string &name, const program *p) {
const auto &i = get_dummy_global_program().symbols.find(name);
if (i != get_dummy_global_program().symbols.end())
return i->second;
if (p) {
const auto &i2 = p->symbols.find(name);
if (i2 != p->symbols.end())
return i2->second;
}
return nullptr;
}
std::vector<compiled_source::symbol> public_symbols(program &program);
int generate(std::shared_ptr<output_format> _format, const std::string &_source, const std::string &_dest,
const std::vector<std::string> &_options = std::vector<std::string>());
// Handling the scanner.
void scan_begin();
void scan_end();
// The token's location used by the scanner.
yy::location location;
};
#endif