mirror of
https://github.com/SourMesen/Mesen2.git
synced 2025-04-02 10:21:44 -04:00
131 lines
3.5 KiB
C++
131 lines
3.5 KiB
C++
#pragma once
|
|
#include "pch.h"
|
|
#include "Shared/CpuType.h"
|
|
#include "Debugger/IAssembler.h"
|
|
#include "Utilities/StringUtilities.h"
|
|
|
|
class LabelManager;
|
|
|
|
template<class T>
|
|
class Base6502Assembler : public IAssembler
|
|
{
|
|
protected:
|
|
enum class OperandType
|
|
{
|
|
None,
|
|
A,
|
|
X,
|
|
Y,
|
|
S,
|
|
Custom
|
|
};
|
|
|
|
enum class OperandValueType
|
|
{
|
|
None,
|
|
Decimal,
|
|
Hex,
|
|
Binary,
|
|
Label
|
|
};
|
|
|
|
struct AssemblerOperand
|
|
{
|
|
OperandType Type = OperandType::None;
|
|
OperandValueType ValueType = OperandValueType::None;
|
|
int64_t Value = 0;
|
|
int ByteCount = 0;
|
|
int OperandSize = 0;
|
|
bool IsImmediate = false;
|
|
bool HasOpeningParenthesis = false;
|
|
bool HasClosingParenthesis = false;
|
|
bool HasOpeningBracket = false;
|
|
bool HasClosingBracket = false;
|
|
|
|
bool HasParenOrBracket()
|
|
{
|
|
return HasOpeningBracket || HasClosingParenthesis || HasOpeningBracket || HasClosingBracket;
|
|
}
|
|
};
|
|
|
|
struct AssemblerLineData
|
|
{
|
|
string OpCode;
|
|
T AddrMode = {};
|
|
AssemblerOperand Operands[3] = {};
|
|
int OperandCount = 0;
|
|
};
|
|
|
|
private:
|
|
unordered_map<string, unordered_map<T, uint8_t>> _availableModesByOpName;
|
|
bool _needSecondPass = false;
|
|
CpuType _cpuType;
|
|
|
|
LabelManager* _labelManager;
|
|
void ProcessLine(string code, uint32_t& instructionAddress, vector<int16_t>& output, unordered_map<string, uint32_t>& labels, bool firstPass, unordered_map<string, uint32_t>& currentPassLabels);
|
|
AssemblerSpecialCodes ParseOperand(AssemblerLineData& lineData, string operand, bool firstPass, unordered_map<string, uint32_t>& labels);
|
|
void AssembleInstruction(AssemblerLineData& lineData, uint32_t& instructionAddress, vector<int16_t>& output, bool firstPass);
|
|
|
|
protected:
|
|
bool IsOpModeAvailable(string& opCode, T mode);
|
|
int16_t GetOpByteCode(string& opCode, T addrMode);
|
|
|
|
virtual string GetOpName(uint8_t opcode) = 0;
|
|
virtual T GetOpMode(uint8_t opcode) = 0;
|
|
virtual bool IsOfficialOp(uint8_t opcode) = 0;
|
|
virtual AssemblerSpecialCodes ResolveOpMode(AssemblerLineData& op, uint32_t instructionAddress, bool firstPass) = 0;
|
|
|
|
public:
|
|
Base6502Assembler(LabelManager* labelManager, CpuType cpuType)
|
|
{
|
|
_labelManager = labelManager;
|
|
_cpuType = cpuType;
|
|
}
|
|
|
|
virtual ~Base6502Assembler()
|
|
{
|
|
}
|
|
|
|
uint32_t AssembleCode(string code, uint32_t startAddress, int16_t* assembledCode)
|
|
{
|
|
for(int i = 0; i < 256; i++) {
|
|
string opName = StringUtilities::ToUpper(GetOpName(i));
|
|
if(_availableModesByOpName.find(opName) == _availableModesByOpName.end()) {
|
|
_availableModesByOpName[opName] = unordered_map<T, uint8_t>();
|
|
}
|
|
|
|
T addrMode = GetOpMode(i);
|
|
if(IsOfficialOp(i) || _availableModesByOpName[opName].find(addrMode) == _availableModesByOpName[opName].end()) {
|
|
_availableModesByOpName[opName][addrMode] = i;
|
|
}
|
|
}
|
|
|
|
unordered_map<string, uint32_t> temporaryLabels;
|
|
unordered_map<string, uint32_t> currentPassLabels;
|
|
|
|
vector<int16_t> output;
|
|
output.reserve(1000);
|
|
|
|
uint32_t originalStartAddr = startAddress;
|
|
|
|
vector<string> codeLines = StringUtilities::Split(code, '\n');
|
|
|
|
//Make 2 passes - first one to find all labels, second to assemble
|
|
_needSecondPass = false;
|
|
for(string& line : codeLines) {
|
|
ProcessLine(line, startAddress, output, temporaryLabels, true, currentPassLabels);
|
|
}
|
|
|
|
if(_needSecondPass) {
|
|
currentPassLabels.clear();
|
|
output.clear();
|
|
startAddress = originalStartAddr;
|
|
for(string& line : codeLines) {
|
|
ProcessLine(line, startAddress, output, temporaryLabels, false, currentPassLabels);
|
|
}
|
|
}
|
|
|
|
memcpy(assembledCode, output.data(), std::min<int>(100000, (int)output.size()) * sizeof(uint16_t));
|
|
return (uint32_t)output.size();
|
|
}
|
|
};
|