329 lines
12 KiB
C++
329 lines
12 KiB
C++
/*
|
|
* Copyright (c) 2021 Raspberry Pi (Trading) Ltd.
|
|
*
|
|
* SPDX-License-Identifier: BSD-3-Clause
|
|
*/
|
|
|
|
#include <array>
|
|
#include <algorithm>
|
|
#include <sstream>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
#include "output_format.h"
|
|
#include "pio_disassembler.h"
|
|
|
|
struct python_output : public output_format {
|
|
struct factory {
|
|
factory() {
|
|
output_format::add(new python_output());
|
|
}
|
|
};
|
|
|
|
python_output() : output_format("python") {}
|
|
|
|
std::string get_description() override {
|
|
return "Python file suitable for use with MicroPython";
|
|
}
|
|
|
|
void output_symbols(FILE *out, std::string prefix, const std::vector<compiled_source::symbol> &symbols) {
|
|
int count = 0;
|
|
for (const auto &s : symbols) {
|
|
if (!s.is_label) {
|
|
fprintf(out, "%s%s = %d\n", prefix.c_str(), s.name.c_str(), s.value);
|
|
count++;
|
|
}
|
|
}
|
|
if (count) {
|
|
fprintf(out, "\n");
|
|
count = 0;
|
|
}
|
|
for (const auto &s : symbols) {
|
|
if (s.is_label) {
|
|
fprintf(out, "%soffset_%s = %d\n", prefix.c_str(), s.name.c_str(), s.value);
|
|
count++;
|
|
}
|
|
}
|
|
if (count) {
|
|
fprintf(out, "\n");
|
|
}
|
|
}
|
|
|
|
void header(FILE *out, std::string msg) {
|
|
std::string dashes = std::string(msg.length(), '-');
|
|
fprintf(out, "# %s #\n", dashes.c_str());
|
|
fprintf(out, "# %s #\n", msg.c_str());
|
|
fprintf(out, "# %s #\n", dashes.c_str());
|
|
fprintf(out, "\n");
|
|
}
|
|
|
|
int output(std::string destination, std::vector<std::string> output_options,
|
|
const compiled_source &source) override {
|
|
FILE *out = open_single_output(destination);
|
|
if (!out) return 1;
|
|
|
|
header(out, "This file is autogenerated by pioasm; do not edit!");
|
|
|
|
fprintf(out, "import rp2\n");
|
|
fprintf(out, "from machine import Pin");
|
|
fprintf(out, "\n");
|
|
|
|
output_symbols(out, "", source.global_symbols);
|
|
|
|
for (const auto &program : source.programs) {
|
|
header(out, program.name);
|
|
|
|
std::string prefix = program.name + "_";
|
|
|
|
output_symbols(out, prefix, program.symbols);
|
|
|
|
int param_count = 0;
|
|
auto write_opt = [&] (std::string name, std::string value) {
|
|
if (param_count++) {
|
|
fprintf(out, ", ");
|
|
}
|
|
fprintf(out, "%s=%s", name.c_str(), value.c_str());
|
|
};
|
|
fprintf(out, "@rp2.asm_pio(");
|
|
for(const auto &p : program.lang_opts) {
|
|
if (p.first.size() >= name.size() && p.first.compare(0, name.size(), name) == 0) {
|
|
for (const auto &p2 : p.second) {
|
|
write_opt(p2.first, p2.second);
|
|
}
|
|
}
|
|
}
|
|
fprintf(out, ")\n");
|
|
fprintf(out, "def %s():\n", program.name.c_str());
|
|
|
|
std::map<uint, std::string> jmp_labels;
|
|
// for now just use numeric labels
|
|
for (int i = 0; i < (int)program.instructions.size(); i++) {
|
|
const auto &inst = program.instructions[i];
|
|
if (!(inst >> 13u)) {
|
|
// a jump
|
|
uint target = inst &0x1fu;
|
|
jmp_labels.insert(std::pair<uint,std::string>(target, std::to_string(target)));
|
|
}
|
|
}
|
|
|
|
for (uint i = 0; i < (int)program.instructions.size(); i++) {
|
|
const auto &inst = program.instructions[i];
|
|
if (i == program.wrap_target) {
|
|
fprintf(out, " wrap_target()\n");
|
|
}
|
|
auto it = jmp_labels.find(i);
|
|
if (it != jmp_labels.end()) {
|
|
fprintf(out, " label(\"%s\")\n", it->second.c_str());
|
|
}
|
|
fprintf(out, " %s # %d\n", disassemble(jmp_labels, inst, program.sideset_bits_including_opt.get(), program.sideset_opt).c_str(), i);
|
|
if (i == program.wrap) {
|
|
fprintf(out, " wrap()\n");
|
|
}
|
|
}
|
|
fprintf(out, "\n");
|
|
|
|
/*
|
|
fprintf(out, "static inline pio_sm_config %sprogram_default_config(uint offset) {\n", prefix.c_str());
|
|
fprintf(out, " pio_sm_config c = pio_sm_default_config();\n");
|
|
fprintf(out, " sm_config_wrap(&c, offset + %swrap_target, offset + %swrap);\n", prefix.c_str(),
|
|
prefix.c_str());
|
|
if (program.sideset_bits_including_opt.is_specified()) {
|
|
fprintf(out, " sm_config_sideset(&c, %d, %s, %s);\n", program.sideset_bits_including_opt.get(),
|
|
program.sideset_opt ? "true" : "false",
|
|
program.sideset_pindirs ? "true" : "false");
|
|
}
|
|
fprintf(out, " return c;\n");
|
|
fprintf(out, "}\n");
|
|
*/
|
|
// todo maybe have some code blocks inside or outside here?
|
|
for(const auto& o : program.code_blocks) {
|
|
fprintf(out, "\n");
|
|
if (o.first == name) {
|
|
for(const auto &contents : o.second) {
|
|
fprintf(out, "%s", contents.c_str());
|
|
fprintf(out, "\n");
|
|
}
|
|
}
|
|
}
|
|
|
|
fprintf(out, "\n");
|
|
}
|
|
if (out != stdout) { fclose(out); }
|
|
return 0;
|
|
}
|
|
|
|
static std::string disassemble(const std::map<uint, std::string>& jmp_labels, uint16_t inst, uint sideset_bits_including_opt, bool sideset_opt) {
|
|
std::stringstream ss;
|
|
uint major = inst >> 13u;
|
|
uint arg1 = ((uint) inst >> 5u) & 0x7u;
|
|
uint arg2 = inst & 0x1fu;
|
|
std::string op_string;
|
|
auto op = [&](const std::string &s) {
|
|
op_string = s;
|
|
};
|
|
auto op_guts = [&](const std::string &s) {
|
|
ss << std::left << std::setw(24) << (op_string + "(" + s + ")");
|
|
};
|
|
|
|
bool invalid = false;
|
|
switch (major) {
|
|
case 0b000: {
|
|
static std::array<std::string, 8> conditions{"", "not_x", "x_dec", "not_y", "y_dec", "x_not_y", "pin",
|
|
"not_osre"};
|
|
op("jmp");
|
|
auto it = jmp_labels.find(arg2);
|
|
std::string label = "?";
|
|
if (it != jmp_labels.end()) {
|
|
label = it->second;
|
|
}
|
|
if (arg1)
|
|
op_guts(conditions[arg1] + ", \"" + label +"\"");
|
|
else
|
|
op_guts("\"" + label + "\"");
|
|
break;
|
|
}
|
|
case 0b001: {
|
|
uint source = arg1 & 3u;
|
|
std::string guts;
|
|
switch (source) {
|
|
case 0b00:
|
|
guts = "gpio, " + std::to_string(arg2);
|
|
break;
|
|
case 0b01:
|
|
guts = "pin, " + std::to_string(arg2);
|
|
break;
|
|
case 0b10:
|
|
if (arg2 & 0x8u) {
|
|
invalid = true;
|
|
} else {
|
|
guts = "irq, ";
|
|
auto irq = std::to_string(arg2 & 7u);
|
|
if (arg2 & 0x10u) {
|
|
guts += "rel(" + irq + ")";
|
|
} else {
|
|
guts += irq;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
if (!invalid) {
|
|
guts = ((arg1 & 4u) ? "1, " : "0, ") + guts;
|
|
op("wait");
|
|
op_guts(guts);
|
|
}
|
|
break;
|
|
}
|
|
case 0b010: {
|
|
static std::array<std::string, 8> sources { "pins", "x", "y", "null", "", "status", "isr", "osr"};
|
|
std::string source = sources[arg1];
|
|
if (source.empty()) {
|
|
invalid = true;
|
|
} else {
|
|
op("in_");
|
|
op_guts(source + ", " + std::to_string(arg2 ? arg2 : 32));
|
|
}
|
|
break;
|
|
}
|
|
case 0b011: {
|
|
static std::array<std::string, 8> dests { "pins", "x", "y", "null", "pindirs", "pc", "isr", "exec"};
|
|
op("out");
|
|
op_guts(dests[arg1] + ", " + std::to_string(arg2 ? arg2 : 32));
|
|
break;
|
|
}
|
|
case 0b100: {
|
|
if (arg2) {
|
|
invalid = true;
|
|
} else {
|
|
std::string guts = "";
|
|
if (arg1 & 4u) {
|
|
op("pull");
|
|
if (arg1 & 2u) guts = "ifempty, ";
|
|
} else {
|
|
op("push");
|
|
if (arg1 & 2u) guts = "iffull, ";
|
|
}
|
|
guts += ((arg1 & 0x1u) ? "block" : "noblock");
|
|
op_guts(guts);
|
|
}
|
|
break;
|
|
}
|
|
case 0b101: {
|
|
static std::array<std::string, 8> dests { "pins", "x", "y", "", "exec", "pc", "isr", "osr"};
|
|
static std::array<std::string, 8> sources { "pins", "x", "y", "null", "", "status", "isr", "osr"};
|
|
std::string dest = dests[arg1];
|
|
std::string source = sources[arg2 & 7u];
|
|
uint operation = arg2 >> 3u;
|
|
if (source.empty() || dest.empty() || operation == 3) {
|
|
invalid = true;
|
|
}
|
|
if (dest == source && (arg1 == 1 || arg2 == 2) && operation == 0) {
|
|
op("nop");
|
|
op_guts("");
|
|
} else {
|
|
op("mov");
|
|
std::string guts = dest + ", ";
|
|
if (operation == 1) {
|
|
guts += "invert(";
|
|
} else if (operation == 2) {
|
|
guts += "reverse(";
|
|
}
|
|
guts += source;
|
|
if (operation == 1 || operation == 2) {
|
|
guts += ")";
|
|
}
|
|
op_guts(guts);
|
|
}
|
|
break;
|
|
}
|
|
case 0b110: {
|
|
if ((arg1 & 0x4u) || (arg2 & 0x8u)) {
|
|
invalid = true;
|
|
} else {
|
|
op("irq");
|
|
std::string guts;
|
|
if (arg1 & 0x2u) {
|
|
guts += "clear, ";
|
|
} else if (arg1 & 0x1u) {
|
|
guts += "block, ";
|
|
}
|
|
auto irq = std::to_string(arg2 & 7u);
|
|
if (arg2 & 0x10u) {
|
|
guts += "rel(" + irq + ")";
|
|
} else {
|
|
guts += irq;
|
|
}
|
|
op_guts(guts);
|
|
}
|
|
break;
|
|
}
|
|
case 0b111: {
|
|
static std::array<std::string, 8> dests{"pins", "x", "y", "", "pindirs", "", "", ""};
|
|
std::string dest = dests[arg1];
|
|
if (dest.empty()) {
|
|
invalid = true;
|
|
} else {
|
|
op("set");
|
|
op_guts(dests[arg1] + ", " + std::to_string(arg2));
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
if (invalid) {
|
|
op("word");
|
|
ss << std::hex;
|
|
op_guts(std::to_string(inst));
|
|
}
|
|
uint delay = ((uint) inst >> 8u) & 0x1f;
|
|
ss << std::left << std::setw(9);
|
|
if (sideset_bits_including_opt && (!sideset_opt || (delay & 0x10u))) {
|
|
ss << (".side("+ std::to_string((delay & (sideset_opt ? 0xfu : 0x1fu)) >> (5u - sideset_bits_including_opt))+")");
|
|
} else {
|
|
ss << "";
|
|
}
|
|
delay &= ((1u << (5 - sideset_bits_including_opt)) - 1u);
|
|
ss << std::left << std::setw(4) << (delay ? ("[" + std::to_string(delay) + "]") : "");
|
|
return ss.str();
|
|
}
|
|
};
|
|
|
|
static python_output::factory creator;
|