orbital/tests/test_gcn_translator.cpp
Alexandro Sanchez Bach 9a33c0dbd2
Dump GCN analyzer/translator test output and Makefile fixes
Signed-off-by: Alexandro Sanchez Bach <alexandro@phi.nz>
2019-01-29 23:23:57 +01:00

81 lines
2.1 KiB
C++

/**
* (c) 2019 Alexandro Sanchez Bach. All rights reserved.
* Released under MIT license. Read LICENSE for more details.
*/
#include "gcn_analyzer.h"
#include "gcn_translator.h"
#include "gcn_parser.h"
#include <cstdio>
#include <cstdlib>
#include <vector>
#define UNUSED(arg) (void)(arg)
static int translate_shader(const uint8_t *data, size_t size, const char *stage)
{
UNUSED(size);
gcn_parser_t parser;
gcn_analyzer_t analyzer;
gcn_translator_t *translator = NULL;
uint8_t *bc_data;
uint32_t bc_size;
gcn_parser_init(&parser);
gcn_analyzer_init(&analyzer);
gcn_parser_parse(&parser, data, &gcn_analyzer_callbacks, &analyzer);
if (!strcmp(stage, "ps"))
translator = gcn_translator_create(&analyzer, GCN_STAGE_PS);
if (!strcmp(stage, "vs"))
translator = gcn_translator_create(&analyzer, GCN_STAGE_VS);
if (!translator)
return 1;
gcn_parser_init(&parser);
gcn_parser_parse(&parser, data, &gcn_translator_callbacks, translator);
bc_data = gcn_translator_dump(translator, &bc_size);
fwrite(bc_data, 1, bc_size, stdout);
return 0;
}
int main(int argc, const char **argv)
{
size_t shader_size;
uint8_t *shader_data;
const char *name, *stage;
FILE *file;
int ret;
if (argc <= 2) {
fprintf(stderr, "Usage: %s {ps,vs} [path/to/shader.bin]\n", argv[0]);
return 0;
}
stage = argv[1];
name = argv[2];
file = fopen(name, "rb");
if (!file) {
fprintf(stderr, "File %s does not exist!\n", name);
return 1;
}
fseek(file, 0, SEEK_END);
shader_size = ftell(file);
shader_data = reinterpret_cast<uint8_t*>(malloc(shader_size));
if (!shader_data) {
fprintf(stderr, "Could not allocate 0x%zX bytes!\n", shader_size);
return 1;
}
fseek(file, 0, SEEK_SET);
if (fread(shader_data, 1, shader_size, file) != shader_size) {
fprintf(stderr, "Could not read 0x%zX bytes!\n", shader_size);
return 1;
}
ret = translate_shader(shader_data, shader_size, stage);
free(shader_data);
fclose(file);
return ret;
}