gigatron/rom/Contrib/at67/tools/gtasm/gtasm.cpp
2025-01-28 19:17:01 +03:00

100 lines
3.1 KiB
C++

#include <stdio.h>
#include <stdlib.h>
#include <sstream>
#include "../../memory.h"
#include "../../loader.h"
#include "../../assembler.h"
#include "../../expression.h"
#define GTASM_MAJOR_VERSION "1.0"
#define GTASM_MINOR_VERSION "6R"
#define GTASM_VERSION_STR "gtasm v" GTASM_MAJOR_VERSION "." GTASM_MINOR_VERSION
int main(int argc, char* argv[])
{
if(argc != 2 && argc != 3)
{
fprintf(stderr, "%s\n", GTASM_VERSION_STR);
fprintf(stderr, "Usage: gtasm <input filename> <optional include path>\n");
return 1;
}
std::string name = std::string(argv[1]);
if(name.find(".vasm") == name.npos && name.find(".gasm") == name.npos && name.find(".asm") == name.npos && name.find(".s") == name.npos)
{
fprintf(stderr, "Wrong file extension in %s : must be one of : '.vasm' or '.gasm' or '.asm' or '.s'\n", name.c_str());
return 1;
}
Loader::initialise();
Expression::initialise();
Assembler::initialise();
// Optional include path
uint16_t address = DEFAULT_EXEC_ADDRESS;
std::string includepath = (argc == 3) ? std::string(argv[2]) : ".";
Assembler::setIncludePath(includepath);
// Path and name
size_t slash = name.find_last_of("\\/");
std::string path = (slash != std::string::npos) ? name.substr(0, slash) : ".";
Expression::replaceText(path, "\\", "/");
name = (slash != std::string::npos) ? name.substr(slash + 1) : name;
std::string filename = path + "/" + name;
Loader::setFilePath(filename);
if(!Assembler::assemble(filename, address)) return 1;
// Create gt1 format
Loader::Gt1File gt1File;
gt1File._loStart = LO_BYTE(address);
gt1File._hiStart = HI_BYTE(address);
Loader::Gt1Segment gt1Segment;
gt1Segment._loAddress = LO_BYTE(address);
gt1Segment._hiAddress = HI_BYTE(address);
bool hasRomCode = false;
Assembler::ByteCode byteCode;
while(!Assembler::getNextAssembledByte(byteCode))
{
if(byteCode._isRomAddress) hasRomCode = true;
// Custom address
if(byteCode._isCustomAddress)
{
if(gt1Segment._dataBytes.size())
{
// Previous segment
gt1Segment._segmentSize = uint8_t(gt1Segment._dataBytes.size());
gt1File._segments.push_back(gt1Segment);
gt1Segment._dataBytes.clear();
}
address = byteCode._address;
gt1Segment._isRomAddress = byteCode._isRomAddress;
gt1Segment._loAddress = LO_BYTE(address);
gt1Segment._hiAddress = HI_BYTE(address);
}
gt1Segment._dataBytes.push_back(byteCode._data);
}
// Last segment
if(gt1Segment._dataBytes.size())
{
gt1Segment._segmentSize = uint8_t(gt1Segment._dataBytes.size());
gt1File._segments.push_back(gt1Segment);
}
// Don't save gt1 file for any asm files that contain native rom code
std::string gt1FileName;
if(!hasRomCode && !saveGt1File(filename, gt1File, gt1FileName)) return 1;
Loader::printGt1Stats(gt1FileName, gt1File, false);
return 0;
}