/*
* Copyright 2014 Ilia Mirkin
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
#include <errno.h>
#include "tgsi/tgsi_text.h"
#include "util/u_debug.h"
#include "codegen/nv50_ir_driver.h"
#include "nv50/nv50_context.h"
/* these headers weren't really meant to be included together */
#undef SB_DATA
#include "nv30/nv30_state.h"
#include "nv30/nvfx_shader.h"
static int
nv30_fp(int chipset, struct tgsi_token tokens[],
unsigned *size, unsigned **code) {
struct nv30_fragprog fp;
memset(&fp, 0, sizeof(fp));
fp.pipe.tokens = tokens;
tgsi_scan_shader(fp.pipe.tokens, &fp.info);
_nvfx_fragprog_translate(chipset >= 0x40 ? 0x4097 : 0x3097, &fp);
*size = fp.insn_len * 4;
*code = fp.insn;
return !fp.translated;
}
static int
nv30_vp(int chipset, struct tgsi_token tokens[],
unsigned *size, unsigned **code) {
struct nv30_vertprog vp;
memset(&vp, 0, sizeof(vp));
vp.pipe.tokens = tokens;
tgsi_scan_shader(vp.pipe.tokens, &vp.info);
_nvfx_vertprog_translate(chipset >= 0x40 ? 0x4097 : 0x3097, &vp);
*size = vp.nr_insns * 16;
*code = (unsigned *)vp.insns;
return !vp.translated;
}
static int
nv30_codegen(int chipset, int type, struct tgsi_token tokens[],
unsigned *size, unsigned **code) {
switch (type) {
case PIPE_SHADER_FRAGMENT:
return nv30_fp(chipset, tokens, size, code);
case PIPE_SHADER_VERTEX:
return nv30_vp(chipset, tokens, size, code);
}
_debug_printf("Unexpected shader type: %d\n", type);
return 1;
}
static int
dummy_assign_slots(struct nv50_ir_prog_info *info)
{
unsigned i, n, c;
n = 0;
for (i = 0; i < info->numInputs; ++i) {
for (c = 0; c < 4; ++c)
if (info->in[i].mask & (1 << c))
info->in[i].slot[c] = n++;
}
/* VertexID before InstanceID */
if (info->io.vertexId < info->numSysVals)
info->sv[info->io.vertexId].slot[0] = n++;
if (info->io.instanceId < info->numSysVals)
info->sv[info->io.instanceId].slot[0] = n++;
n = 0;
for (i = 0; i < info->numOutputs; ++i) {
for (c = 0; c < 4; ++c)
if (info->out[i].mask & (1 << c))
info->out[i].slot[c] = n++;
}
return 0;
}
static int
nouveau_codegen(int chipset, int type, struct tgsi_token tokens[],
unsigned *size, unsigned **code) {
struct nv50_ir_prog_info info = {0};
int ret;
info.type = type;
info.target = chipset;
info.bin.sourceRep = PIPE_SHADER_IR_TGSI;
info.bin.source = tokens;
info.io.auxCBSlot = 15;
info.io.ucpBase = NV50_CB_AUX_UCP_OFFSET;
info.io.suInfoBase = NV50_CB_AUX_TEX_MS_OFFSET;
info.io.msInfoCBSlot = 15;
info.io.msInfoBase = NV50_CB_AUX_MS_OFFSET;
info.assignSlots = dummy_assign_slots;
info.optLevel = debug_get_num_option("NV50_PROG_OPTIMIZE", 3);
info.dbgFlags = debug_get_num_option("NV50_PROG_DEBUG", 0);
info.omitLineNum = debug_get_num_option("NV50_PROG_DEBUG_OMIT_LINENUM", 0);
ret = nv50_ir_generate_code(&info);
if (ret) {
_debug_printf("Error compiling program: %d\n", ret);
return ret;
}
*size = info.bin.codeSize;
*code = info.bin.code;
return 0;
}
int
main(int argc, char *argv[])
{
struct tgsi_token tokens[4096];
int i, chipset = 0, type = -1;
const char *filename = NULL;
FILE *f;
char text[65536] = {0};
unsigned size = 0, *code = NULL;
for (i = 1; i < argc; i++) {
if (!strcmp(argv[i], "-a"))
chipset = strtol(argv[++i], NULL, 16);
else
filename = argv[i];
}
if (!chipset) {
_debug_printf("Must specify a chipset (-a)\n");
return 1;
}
if (!filename) {
_debug_printf("Must specify a filename\n");
return 1;
}
if (!strcmp(filename, "-"))
f = stdin;
else
f = fopen(filename, "r");
if (!f) {
_debug_printf("Error opening file '%s': %s\n", filename, strerror(errno));
return 1;
}
if (!fread(text, 1, sizeof(text), f) || ferror(f)) {
_debug_printf("Error reading file '%s'\n", filename);
fclose(f);
return 1;
}
fclose(f);
_debug_printf("Compiling for NV%X\n", chipset);
if (!strncmp(text, "FRAG", 4))
type = PIPE_SHADER_FRAGMENT;
else if (!strncmp(text, "VERT", 4))
type = PIPE_SHADER_VERTEX;
else if (!strncmp(text, "GEOM", 4))
type = PIPE_SHADER_GEOMETRY;
else if (!strncmp(text, "COMP", 4))
type = PIPE_SHADER_COMPUTE;
else if (!strncmp(text, "TESS_CTRL", 9))
type = PIPE_SHADER_TESS_CTRL;
else if (!strncmp(text, "TESS_EVAL", 9))
type = PIPE_SHADER_TESS_EVAL;
else {
_debug_printf("Unrecognized TGSI header\n");
return 1;
}
if (!tgsi_text_translate(text, tokens, ARRAY_SIZE(tokens))) {
_debug_printf("Failed to parse TGSI shader\n");
return 1;
}
if (chipset >= 0x50) {
i = nouveau_codegen(chipset, type, tokens, &size, &code);
} else if (chipset >= 0x30) {
i = nv30_codegen(chipset, type, tokens, &size, &code);
} else {
_debug_printf("chipset NV%02X not supported\n", chipset);
i = 1;
}
if (i)
return i;
_debug_printf("program binary (%d bytes)\n", size);
for (i = 0; i < size; i += 4) {
printf("%08x ", code[i / 4]);
if (i % (8 * 4) == (7 * 4))
printf("\n");
}
if (i % (8 * 4) != 0)
printf("\n");
return 0;
}