int main(int argc, char **argv) { int ret = 0, n = 1; const char *filename; struct tgsi_token toks[65536]; struct tgsi_parse_context parse; struct ir3_compiler *compiler; struct ir3_shader_variant v; struct ir3_shader_key key = {}; const char *info; void *ptr; size_t size; fd_mesa_debug |= FD_DBG_DISASM; /* cmdline args which impact shader variant get spit out in a * comment on the first line.. a quick/dirty way to preserve * that info so when ir3test recompiles the shader with a new * compiler version, we use the same shader-key settings: */ debug_printf("; options:"); while (n < argc) { if (!strcmp(argv[n], "--verbose")) { fd_mesa_debug |= FD_DBG_MSGS | FD_DBG_OPTMSGS; n++; continue; } if (!strcmp(argv[n], "--binning-pass")) { debug_printf(" %s", argv[n]); key.binning_pass = true; n++; continue; } if (!strcmp(argv[n], "--color-two-side")) { debug_printf(" %s", argv[n]); key.color_two_side = true; n++; continue; } if (!strcmp(argv[n], "--half-precision")) { debug_printf(" %s", argv[n]); key.half_precision = true; n++; continue; } if (!strcmp(argv[n], "--saturate-s")) { debug_printf(" %s %s", argv[n], argv[n+1]); key.vsaturate_s = key.fsaturate_s = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--saturate-t")) { debug_printf(" %s %s", argv[n], argv[n+1]); key.vsaturate_t = key.fsaturate_t = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--saturate-r")) { debug_printf(" %s %s", argv[n], argv[n+1]); key.vsaturate_r = key.fsaturate_r = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--help")) { print_usage(); return 0; } break; } debug_printf("\n"); filename = argv[n]; memset(&v, 0, sizeof(v)); v.key = key; ret = read_file(filename, &ptr, &size); if (ret) { print_usage(); return ret; } if (fd_mesa_debug & FD_DBG_OPTMSGS) debug_printf("%s\n", (char *)ptr); if (!tgsi_text_translate(ptr, toks, Elements(toks))) errx(1, "could not parse `%s'", filename); tgsi_parse_init(&parse, toks); switch (parse.FullHeader.Processor.Processor) { case TGSI_PROCESSOR_FRAGMENT: v.type = SHADER_FRAGMENT; break; case TGSI_PROCESSOR_VERTEX: v.type = SHADER_VERTEX; break; case TGSI_PROCESSOR_COMPUTE: v.type = SHADER_COMPUTE; break; } /* TODO cmdline option to target different gpus: */ compiler = ir3_compiler_create(320); info = "NIR compiler"; ret = ir3_compile_shader_nir(compiler, &v, toks, key); if (ret) { fprintf(stderr, "compiler failed!\n"); return ret; } dump_info(&v, info); }
int main(int argc, char **argv) { int ret = 0, n = 1; char *filenames[2]; int num_files = 0; unsigned stage = 0; struct ir3_shader_variant v; struct ir3_shader s; struct ir3_shader_key key = {}; /* TODO cmdline option to target different gpus: */ unsigned gpu_id = 320; const char *info; void *ptr; size_t size; memset(&s, 0, sizeof(s)); memset(&v, 0, sizeof(v)); /* cmdline args which impact shader variant get spit out in a * comment on the first line.. a quick/dirty way to preserve * that info so when ir3test recompiles the shader with a new * compiler version, we use the same shader-key settings: */ debug_printf("; options:"); while (n < argc) { if (!strcmp(argv[n], "--verbose")) { fd_mesa_debug |= FD_DBG_MSGS | FD_DBG_OPTMSGS | FD_DBG_DISASM; n++; continue; } if (!strcmp(argv[n], "--binning-pass")) { debug_printf(" %s", argv[n]); key.binning_pass = true; n++; continue; } if (!strcmp(argv[n], "--color-two-side")) { debug_printf(" %s", argv[n]); key.color_two_side = true; n++; continue; } if (!strcmp(argv[n], "--half-precision")) { debug_printf(" %s", argv[n]); key.half_precision = true; n++; continue; } if (!strcmp(argv[n], "--saturate-s")) { debug_printf(" %s %s", argv[n], argv[n+1]); key.vsaturate_s = key.fsaturate_s = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--saturate-t")) { debug_printf(" %s %s", argv[n], argv[n+1]); key.vsaturate_t = key.fsaturate_t = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--saturate-r")) { debug_printf(" %s %s", argv[n], argv[n+1]); key.vsaturate_r = key.fsaturate_r = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--astc-srgb")) { debug_printf(" %s %s", argv[n], argv[n+1]); key.vastc_srgb = key.fastc_srgb = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--stream-out")) { struct pipe_stream_output_info *so = &s.stream_output; debug_printf(" %s", argv[n]); /* TODO more dynamic config based on number of outputs, etc * rather than just hard-code for first output: */ so->num_outputs = 1; so->stride[0] = 4; so->output[0].register_index = 0; so->output[0].start_component = 0; so->output[0].num_components = 4; so->output[0].output_buffer = 0; so->output[0].dst_offset = 2; so->output[0].stream = 0; n++; continue; } if (!strcmp(argv[n], "--ucp")) { debug_printf(" %s %s", argv[n], argv[n+1]); key.ucp_enables = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--gpu")) { debug_printf(" %s %s", argv[n], argv[n+1]); gpu_id = strtol(argv[n+1], NULL, 0); n += 2; continue; } if (!strcmp(argv[n], "--help")) { print_usage(); return 0; } break; } debug_printf("\n"); while (n < argc) { char *filename = argv[n]; char *ext = rindex(filename, '.'); if (strcmp(ext, ".tgsi") == 0) { if (num_files != 0) errx(1, "in TGSI mode, only a single file may be specified"); s.from_tgsi = true; } else if (strcmp(ext, ".frag") == 0) { if (s.from_tgsi) errx(1, "cannot mix GLSL and TGSI"); if (num_files >= ARRAY_SIZE(filenames)) errx(1, "too many GLSL files"); stage = MESA_SHADER_FRAGMENT; } else if (strcmp(ext, ".vert") == 0) { if (s.from_tgsi) errx(1, "cannot mix GLSL and TGSI"); if (num_files >= ARRAY_SIZE(filenames)) errx(1, "too many GLSL files"); stage = MESA_SHADER_VERTEX; } else { print_usage(); return -1; } filenames[num_files++] = filename; n++; } nir_shader *nir; if (s.from_tgsi) { struct tgsi_token toks[65536]; ret = read_file(filenames[0], &ptr, &size); if (ret) { print_usage(); return ret; } if (fd_mesa_debug & FD_DBG_OPTMSGS) debug_printf("%s\n", (char *)ptr); if (!tgsi_text_translate(ptr, toks, ARRAY_SIZE(toks))) errx(1, "could not parse `%s'", filenames[0]); if (fd_mesa_debug & FD_DBG_OPTMSGS) tgsi_dump(toks, 0); nir = ir3_tgsi_to_nir(toks); } else if (num_files > 0) { nir = load_glsl(num_files, filenames, stage); } else { print_usage(); return -1; } s.compiler = ir3_compiler_create(NULL, gpu_id); s.nir = ir3_optimize_nir(&s, nir, NULL); v.key = key; v.shader = &s; switch (nir->stage) { case MESA_SHADER_FRAGMENT: s.type = v.type = SHADER_FRAGMENT; break; case MESA_SHADER_VERTEX: s.type = v.type = SHADER_VERTEX; break; case MESA_SHADER_COMPUTE: s.type = v.type = SHADER_COMPUTE; break; default: errx(1, "unhandled shader stage: %d", nir->stage); } info = "NIR compiler"; ret = ir3_compile_shader_nir(s.compiler, &v); if (ret) { fprintf(stderr, "compiler failed!\n"); return ret; } dump_info(&v, info); }