/** * MojoShader; generate shader programs from bytecode of compiled * Direct3D shaders. * * Please see the file LICENSE.txt in the source's root directory. * * This file written by Ryan C. Gordon. */ #define __MOJOSHADER_INTERNAL__ 1 #include "mojoshader_profile.h" #pragma GCC visibility push(hidden) // !!! FIXME: A lot of this is cut-and-paste from the GLSL version. #if SUPPORT_PROFILE_METAL #define EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(op) \ void emit_METAL_##op(Context *ctx) { \ fail(ctx, #op " unimplemented in Metal profile"); \ } static inline const char *get_METAL_register_string(Context *ctx, const RegisterType regtype, const int regnum, char *regnum_str, const size_t regnum_size) { // turns out these are identical at the moment. return get_D3D_register_string(ctx,regtype,regnum,regnum_str,regnum_size); } // get_METAL_register_string const char *get_METAL_uniform_type(Context *ctx, const RegisterType rtype) { switch (rtype) { case REG_TYPE_CONST: return "float4"; case REG_TYPE_CONSTINT: return "int4"; case REG_TYPE_CONSTBOOL: return "bool"; default: fail(ctx, "BUG: used a uniform we don't know how to define."); } // switch return NULL; } // get_METAL_uniform_type const char *get_METAL_varname_in_buf(Context *ctx, RegisterType rt, int regnum, char *buf, const size_t len) { char regnum_str[16]; const char *regtype_str = get_METAL_register_string(ctx, rt, regnum, regnum_str, sizeof (regnum_str)); // We don't separate vars with vs_ or ps_ here, because, for the most part, // there are only local vars in Metal shaders. snprintf(buf, len, "%s%s", regtype_str, regnum_str); return buf; } // get_METAL_varname_in_buf const char *get_METAL_varname(Context *ctx, RegisterType rt, int regnum) { char buf[64]; get_METAL_varname_in_buf(ctx, rt, regnum, buf, sizeof (buf)); return StrDup(ctx, buf); } // get_METAL_varname static inline const char *get_METAL_const_array_varname_in_buf(Context *ctx, const int base, const int size, char *buf, const size_t buflen) { snprintf(buf, buflen, "const_array_%d_%d", base, size); return buf; } // get_METAL_const_array_varname_in_buf const char *get_METAL_const_array_varname(Context *ctx, int base, int size) { char buf[64]; get_METAL_const_array_varname_in_buf(ctx, base, size, buf, sizeof (buf)); return StrDup(ctx, buf); } // get_METAL_const_array_varname static inline const char *get_METAL_input_array_varname(Context *ctx, char *buf, const size_t buflen) { snprintf(buf, buflen, "%s", "vertex_input_array"); return buf; } // get_METAL_input_array_varname const char *get_METAL_uniform_array_varname(Context *ctx, const RegisterType regtype, char *buf, const size_t len) { const char *shadertype = ctx->shader_type_str; const char *type = get_METAL_uniform_type(ctx, regtype); snprintf(buf, len, "uniforms.uniforms_%s", type); return buf; } // get_METAL_uniform_array_varname const char *get_METAL_destarg_varname(Context *ctx, char *buf, size_t len) { const DestArgInfo *arg = &ctx->dest_arg; return get_METAL_varname_in_buf(ctx, arg->regtype, arg->regnum, buf, len); } // get_METAL_destarg_varname const char *get_METAL_srcarg_varname(Context *ctx, const size_t idx, char *buf, size_t len) { if (idx >= STATICARRAYLEN(ctx->source_args)) { fail(ctx, "Too many source args"); *buf = '\0'; return buf; } // if const SourceArgInfo *arg = &ctx->source_args[idx]; return get_METAL_varname_in_buf(ctx, arg->regtype, arg->regnum, buf, len); } // get_METAL_srcarg_varname const char *make_METAL_destarg_assign(Context *, char *, const size_t, const char *, ...) ISPRINTF(4,5); const char *make_METAL_destarg_assign(Context *ctx, char *buf, const size_t buflen, const char *fmt, ...) { int need_parens = 0; const DestArgInfo *arg = &ctx->dest_arg; if (arg->writemask == 0) { *buf = '\0'; return buf; // no writemask? It's a no-op. } // if char clampbuf[32] = { '\0' }; const char *clampleft = ""; const char *clampright = ""; if (arg->result_mod & MOD_SATURATE) { ctx->metal_need_header_common = 1; const int vecsize = vecsize_from_writemask(arg->writemask); clampleft = "clamp("; if (vecsize == 1) clampright = ", 0.0, 1.0)"; else { snprintf(clampbuf, sizeof (clampbuf), ", float%d(0.0), float%d(1.0))", vecsize, vecsize); clampright = clampbuf; } // else } // if // MSDN says MOD_PP is a hint and many implementations ignore it. So do we. // CENTROID only allowed in DCL opcodes, which shouldn't come through here. assert((arg->result_mod & MOD_CENTROID) == 0); if (ctx->predicated) { fail(ctx, "predicated destinations unsupported"); // !!! FIXME *buf = '\0'; return buf; } // if char operation[256]; va_list ap; va_start(ap, fmt); const int len = vsnprintf(operation, sizeof (operation), fmt, ap); va_end(ap); if (len >= sizeof (operation)) { fail(ctx, "operation string too large"); // I'm lazy. :P *buf = '\0'; return buf; } // if const char *result_shift_str = ""; switch (arg->result_shift) { case 0x1: result_shift_str = " * 2.0"; break; case 0x2: result_shift_str = " * 4.0"; break; case 0x3: result_shift_str = " * 8.0"; break; case 0xD: result_shift_str = " / 8.0"; break; case 0xE: result_shift_str = " / 4.0"; break; case 0xF: result_shift_str = " / 2.0"; break; } // switch need_parens |= (result_shift_str[0] != '\0'); char regnum_str[16]; const char *regtype_str = get_METAL_register_string(ctx, arg->regtype, arg->regnum, regnum_str, sizeof (regnum_str)); char writemask_str[6]; size_t i = 0; const int scalar = isscalar(ctx, ctx->shader_type, arg->regtype, arg->regnum); if (!scalar && !writemask_xyzw(arg->writemask)) { writemask_str[i++] = '.'; if (arg->writemask0) writemask_str[i++] = 'x'; if (arg->writemask1) writemask_str[i++] = 'y'; if (arg->writemask2) writemask_str[i++] = 'z'; if (arg->writemask3) writemask_str[i++] = 'w'; } // if writemask_str[i] = '\0'; assert(i < sizeof (writemask_str)); const char *leftparen = (need_parens) ? "(" : ""; const char *rightparen = (need_parens) ? ")" : ""; snprintf(buf, buflen, "%s%s%s = %s%s%s%s%s%s;", regtype_str, regnum_str, writemask_str, clampleft, leftparen, operation, rightparen, result_shift_str, clampright); // !!! FIXME: make sure the scratch buffer was large enough. return buf; } // make_METAL_destarg_assign char *make_METAL_swizzle_string(char *swiz_str, const size_t strsize, const int swizzle, const int writemask) { size_t i = 0; if ( (!no_swizzle(swizzle)) || (!writemask_xyzw(writemask)) ) { const int writemask0 = (writemask >> 0) & 0x1; const int writemask1 = (writemask >> 1) & 0x1; const int writemask2 = (writemask >> 2) & 0x1; const int writemask3 = (writemask >> 3) & 0x1; const int swizzle_x = (swizzle >> 0) & 0x3; const int swizzle_y = (swizzle >> 2) & 0x3; const int swizzle_z = (swizzle >> 4) & 0x3; const int swizzle_w = (swizzle >> 6) & 0x3; swiz_str[i++] = '.'; if (writemask0) swiz_str[i++] = swizzle_channels[swizzle_x]; if (writemask1) swiz_str[i++] = swizzle_channels[swizzle_y]; if (writemask2) swiz_str[i++] = swizzle_channels[swizzle_z]; if (writemask3) swiz_str[i++] = swizzle_channels[swizzle_w]; } // if assert(i < strsize); swiz_str[i] = '\0'; return swiz_str; } // make_METAL_swizzle_string const char *make_METAL_srcarg_string(Context *ctx, const size_t idx, const int writemask, char *buf, const size_t buflen) { *buf = '\0'; if (idx >= STATICARRAYLEN(ctx->source_args)) { fail(ctx, "Too many source args"); return buf; } // if const SourceArgInfo *arg = &ctx->source_args[idx]; const char *premod_str = ""; const char *postmod_str = ""; switch (arg->src_mod) { case SRCMOD_NEGATE: premod_str = "-"; break; case SRCMOD_BIASNEGATE: premod_str = "-("; postmod_str = " - 0.5)"; break; case SRCMOD_BIAS: premod_str = "("; postmod_str = " - 0.5)"; break; case SRCMOD_SIGNNEGATE: premod_str = "-(("; postmod_str = " - 0.5) * 2.0)"; break; case SRCMOD_SIGN: premod_str = "(("; postmod_str = " - 0.5) * 2.0)"; break; case SRCMOD_COMPLEMENT: premod_str = "(1.0 - "; postmod_str = ")"; break; case SRCMOD_X2NEGATE: premod_str = "-("; postmod_str = " * 2.0)"; break; case SRCMOD_X2: premod_str = "("; postmod_str = " * 2.0)"; break; case SRCMOD_DZ: fail(ctx, "SRCMOD_DZ unsupported"); return buf; // !!! FIXME postmod_str = "_dz"; break; case SRCMOD_DW: fail(ctx, "SRCMOD_DW unsupported"); return buf; // !!! FIXME postmod_str = "_dw"; break; case SRCMOD_ABSNEGATE: ctx->metal_need_header_math = 1; premod_str = "-abs("; postmod_str = ")"; break; case SRCMOD_ABS: ctx->metal_need_header_math = 1; premod_str = "abs("; postmod_str = ")"; break; case SRCMOD_NOT: premod_str = "!"; break; case SRCMOD_NONE: case SRCMOD_TOTAL: break; // stop compiler whining. } // switch const char *regtype_str = NULL; if (!arg->relative) { regtype_str = get_METAL_varname_in_buf(ctx, arg->regtype, arg->regnum, (char *) alloca(64), 64); } // if const char *rel_lbracket = ""; char rel_offset[32] = { '\0' }; const char *rel_rbracket = ""; char rel_swizzle[4] = { '\0' }; const char *rel_regtype_str = ""; if (arg->relative) { if (arg->regtype == REG_TYPE_INPUT) regtype_str=get_METAL_input_array_varname(ctx,(char*)alloca(64),64); else { assert(arg->regtype == REG_TYPE_CONST); const int arrayidx = arg->relative_array->index; const int offset = arg->regnum - arrayidx; assert(offset >= 0); if (arg->relative_array->constant) { const int arraysize = arg->relative_array->count; regtype_str = get_METAL_const_array_varname_in_buf(ctx, arrayidx, arraysize, (char *) alloca(64), 64); if (offset != 0) snprintf(rel_offset, sizeof (rel_offset), "%d + ", offset); } // if else { regtype_str = get_METAL_uniform_array_varname(ctx, arg->regtype, (char *) alloca(64), 64); if (offset == 0) { snprintf(rel_offset, sizeof (rel_offset), "ARRAYBASE_%d + ", arrayidx); } // if else { snprintf(rel_offset, sizeof (rel_offset), "(ARRAYBASE_%d + %d) + ", arrayidx, offset); } // else } // else } // else rel_lbracket = "["; rel_regtype_str = get_METAL_varname_in_buf(ctx, arg->relative_regtype, arg->relative_regnum, (char *) alloca(64), 64); rel_swizzle[0] = '.'; rel_swizzle[1] = swizzle_channels[arg->relative_component]; rel_swizzle[2] = '\0'; rel_rbracket = "]"; } // if char swiz_str[6] = { '\0' }; if (!isscalar(ctx, ctx->shader_type, arg->regtype, arg->regnum)) { make_METAL_swizzle_string(swiz_str, sizeof (swiz_str), arg->swizzle, writemask); } // if if (regtype_str == NULL) { fail(ctx, "Unknown source register type."); return buf; } // if snprintf(buf, buflen, "%s%s%s%s%s%s%s%s%s", premod_str, regtype_str, rel_lbracket, rel_offset, rel_regtype_str, rel_swizzle, rel_rbracket, swiz_str, postmod_str); // !!! FIXME: make sure the scratch buffer was large enough. return buf; } // make_METAL_srcarg_string // generate some convenience functions. #define MAKE_METAL_SRCARG_STRING_(mask, bitmask) \ static inline const char *make_METAL_srcarg_string_##mask(Context *ctx, \ const size_t idx, char *buf, \ const size_t buflen) { \ return make_METAL_srcarg_string(ctx, idx, bitmask, buf, buflen); \ } MAKE_METAL_SRCARG_STRING_(x, (1 << 0)) MAKE_METAL_SRCARG_STRING_(y, (1 << 1)) MAKE_METAL_SRCARG_STRING_(z, (1 << 2)) MAKE_METAL_SRCARG_STRING_(w, (1 << 3)) MAKE_METAL_SRCARG_STRING_(scalar, (1 << 0)) MAKE_METAL_SRCARG_STRING_(full, 0xF) MAKE_METAL_SRCARG_STRING_(masked, ctx->dest_arg.writemask) MAKE_METAL_SRCARG_STRING_(vec3, 0x7) MAKE_METAL_SRCARG_STRING_(vec2, 0x3) #undef MAKE_METAL_SRCARG_STRING_ // special cases for comparison opcodes... const char *get_METAL_comparison_string_scalar(Context *ctx) { const char *comps[] = { "", ">", "==", ">=", "<", "!=", "<=" }; if (ctx->instruction_controls >= STATICARRAYLEN(comps)) { fail(ctx, "unknown comparison control"); return ""; } // if return comps[ctx->instruction_controls]; } // get_METAL_comparison_string_scalar const char *get_METAL_comparison_string_vector(Context *ctx) { return get_METAL_comparison_string_scalar(ctx); // standard C operators work for vectors in Metal. } // get_METAL_comparison_string_vector void emit_METAL_start(Context *ctx, const char *profilestr) { if (!shader_is_vertex(ctx) && !shader_is_pixel(ctx)) { failf(ctx, "Shader type %u unsupported in this profile.", (uint) ctx->shader_type); return; } // if if (!ctx->mainfn) { if (shader_is_vertex(ctx)) ctx->mainfn = StrDup(ctx, "VertexShader"); else if (shader_is_pixel(ctx)) ctx->mainfn = StrDup(ctx, "FragmentShader"); } // if set_output(ctx, &ctx->mainline); ctx->indent++; } // emit_METAL_start void emit_METAL_RET(Context *ctx); void emit_METAL_end(Context *ctx) { // !!! FIXME: maybe handle this at a higher level? // ps_1_* writes color to r0 instead oC0. We move it to the right place. // We don't have to worry about a RET opcode messing this up, since // RET isn't available before ps_2_0. if (shader_is_pixel(ctx) && !shader_version_atleast(ctx, 2, 0)) { set_used_register(ctx, REG_TYPE_COLOROUT, 0, 1); output_line(ctx, "oC0 = r0;"); } // if // !!! FIXME: maybe handle this at a higher level? // force a RET opcode if we're at the end of the stream without one. if (ctx->previous_opcode != OPCODE_RET) emit_METAL_RET(ctx); } // emit_METAL_end void emit_METAL_phase(Context *ctx) { // no-op in Metal. } // emit_METAL_phase void emit_METAL_finalize(Context *ctx) { // If we had a relative addressing of REG_TYPE_INPUT, we need to build // an array for it at the start of main(). GLSL doesn't let you specify // arrays of attributes. //float4 blah_array[BIGGEST_ARRAY]; if (ctx->have_relative_input_registers) // !!! FIXME fail(ctx, "Relative addressing of input registers not supported."); // Insert header includes we need... push_output(ctx, &ctx->preflight); #define INC_METAL_HEADER(name) \ if (ctx->metal_need_header_##name) { \ output_line(ctx, "#include "); \ } INC_METAL_HEADER(common); INC_METAL_HEADER(math); INC_METAL_HEADER(relational); INC_METAL_HEADER(geometric); INC_METAL_HEADER(graphics); INC_METAL_HEADER(texture); #undef INC_METAL_HEADER output_blank_line(ctx); output_line(ctx, "using namespace metal;"); output_blank_line(ctx); pop_output(ctx); // Fill in the shader's mainline function signature. push_output(ctx, &ctx->mainline_intro); output_line(ctx, "%s %s%s %s (", shader_is_vertex(ctx) ? "vertex" : "fragment", ctx->outputs ? ctx->mainfn : "void", ctx->outputs ? "_Output" : "", ctx->mainfn); pop_output(ctx); push_output(ctx, &ctx->mainline_arguments); ctx->indent++; const int uniform_count = ctx->uniform_float4_count + ctx->uniform_int4_count + ctx->uniform_bool_count; int commas = 0; if (uniform_count) commas++; if (ctx->inputs) commas++; if (commas) commas--; if (uniform_count > 0) { push_output(ctx, &ctx->globals); output_line(ctx, "struct %s_Uniforms", ctx->mainfn); output_line(ctx, "{"); ctx->indent++; if (ctx->uniform_float4_count > 0) output_line(ctx, "float4 uniforms_float4[%d];", ctx->uniform_float4_count); if (ctx->uniform_int4_count > 0) output_line(ctx, "int4 uniforms_int4[%d];", ctx->uniform_int4_count); if (ctx->uniform_bool_count > 0) output_line(ctx, "bool uniforms_bool[%d];", ctx->uniform_bool_count); ctx->indent--; output_line(ctx, "};"); pop_output(ctx); output_line(ctx, "constant %s_Uniforms &uniforms [[buffer(16)]]%s", ctx->mainfn, commas ? "," : ""); commas--; } // if if (ctx->inputs) { output_line(ctx, "%s_Input input [[stage_in]]%s", ctx->mainfn, commas ? "," : ""); commas--; } // if ctx->indent--; output_line(ctx, ") {"); if (ctx->outputs) { ctx->indent++; output_line(ctx, "%s_Output output;", ctx->mainfn); push_output(ctx, &ctx->mainline); ctx->indent++; output_line(ctx, "return output;"); pop_output(ctx); } // if pop_output(ctx); if (ctx->inputs) { push_output(ctx, &ctx->inputs); output_line(ctx, "};"); output_blank_line(ctx); pop_output(ctx); } // if if (ctx->outputs) { push_output(ctx, &ctx->outputs); output_line(ctx, "};"); output_blank_line(ctx); pop_output(ctx); } // if // throw some blank lines around to make source more readable. if (ctx->globals) // don't add a blank line if the section is empty. { push_output(ctx, &ctx->globals); output_blank_line(ctx); pop_output(ctx); } // if } // emit_METAL_finalize void emit_METAL_global(Context *ctx, RegisterType regtype, int regnum) { char varname[64]; get_METAL_varname_in_buf(ctx, regtype, regnum, varname, sizeof (varname)); // These aren't actually global in metal, set them up at top of mainline. push_output(ctx, &ctx->mainline_top); ctx->indent++; switch (regtype) { case REG_TYPE_ADDRESS: if (shader_is_vertex(ctx)) output_line(ctx, "int4 %s;", varname); else if (shader_is_pixel(ctx)) // actually REG_TYPE_TEXTURE. { // We have to map texture registers to temps for ps_1_1, since // they work like temps, initialize with tex coords, and the // ps_1_1 TEX opcode expects to overwrite it. if (!shader_version_atleast(ctx, 1, 4)) output_line(ctx, "float4 %s = input.%s;",varname,varname); } // else if break; case REG_TYPE_PREDICATE: output_line(ctx, "bool4 %s;", varname); break; case REG_TYPE_TEMP: output_line(ctx, "float4 %s;", varname); break; case REG_TYPE_LOOP: break; // no-op. We declare these in for loops at the moment. case REG_TYPE_LABEL: break; // no-op. If we see it here, it means we optimized it out. default: fail(ctx, "BUG: we used a register we don't know how to define."); break; } // switch pop_output(ctx); } // emit_METAL_global void emit_METAL_array(Context *ctx, VariableList *var) { // All uniforms (except constant arrays, which are literally constant // data embedded in Metal shaders) are now packed into a single array, // so we can batch the uniform transfers. So this doesn't actually // define an array here; the one, big array is emitted during // finalization instead. // However, we need to #define the offset into the one, big array here, // and let dereferences use that #define. const int base = var->index; const int metalbase = ctx->uniform_float4_count; push_output(ctx, &ctx->mainline_top); ctx->indent++; output_line(ctx, "const int ARRAYBASE_%d = %d;", base, metalbase); pop_output(ctx); var->emit_position = metalbase; } // emit_METAL_array void emit_METAL_const_array(Context *ctx, const ConstantsList *clist, int base, int size) { char varname[64]; get_METAL_const_array_varname_in_buf(ctx,base,size,varname,sizeof(varname)); const char *cstr = NULL; push_output(ctx, &ctx->mainline_top); ctx->indent++; output_line(ctx, "const float4 %s[%d] = {", varname, size); ctx->indent++; int i; for (i = 0; i < size; i++) { while (clist->constant.type != MOJOSHADER_UNIFORM_FLOAT) clist = clist->next; assert(clist->constant.index == (base + i)); char val0[32]; char val1[32]; char val2[32]; char val3[32]; floatstr(ctx, val0, sizeof (val0), clist->constant.value.f[0], 1); floatstr(ctx, val1, sizeof (val1), clist->constant.value.f[1], 1); floatstr(ctx, val2, sizeof (val2), clist->constant.value.f[2], 1); floatstr(ctx, val3, sizeof (val3), clist->constant.value.f[3], 1); output_line(ctx, "float4(%s, %s, %s, %s)%s", val0, val1, val2, val3, (i < (size-1)) ? "," : ""); clist = clist->next; } // for ctx->indent--; output_line(ctx, "};"); output_line(ctx, "(void) %s[0];", varname); // stop compiler warnings. pop_output(ctx); } // emit_METAL_const_array void emit_METAL_uniform(Context *ctx, RegisterType regtype, int regnum, const VariableList *var) { // Now that we're pushing all the uniforms as one struct, pack these // down, so if we only use register c439, it'll actually map to // uniforms.uniforms_float4[0]. As we push one big struct, this will // prevent uploading unused data. const char *utype = get_METAL_uniform_type(ctx, regtype); char varname[64]; char name[64]; int index = 0; get_METAL_varname_in_buf(ctx, regtype, regnum, varname, sizeof (varname)); push_output(ctx, &ctx->mainline_top); ctx->indent++; if (var == NULL) { get_METAL_uniform_array_varname(ctx, regtype, name, sizeof (name)); if (regtype == REG_TYPE_CONST) index = ctx->uniform_float4_count; else if (regtype == REG_TYPE_CONSTINT) index = ctx->uniform_int4_count; else if (regtype == REG_TYPE_CONSTBOOL) index = ctx->uniform_bool_count; else // get_METAL_uniform_array_varname() would have called fail(). assert(!(ctx->isfail)); // !!! FIXME: can cause unused var warnings in Clang... //output_line(ctx, "constant %s &%s = %s[%d];", utype, varname, name, index); output_line(ctx, "#define %s %s[%d]", varname, name, index); push_output(ctx, &ctx->mainline); ctx->indent++; output_line(ctx, "#undef %s", varname); // !!! FIXME: gross. pop_output(ctx); } // if else { const int arraybase = var->index; if (var->constant) { get_METAL_const_array_varname_in_buf(ctx, arraybase, var->count, name, sizeof (name)); index = (regnum - arraybase); } // if else { assert(var->emit_position != -1); get_METAL_uniform_array_varname(ctx, regtype, name, sizeof (name)); index = (regnum - arraybase) + var->emit_position; } // else // !!! FIXME: might trigger unused var warnings in Clang. //output_line(ctx, "constant %s &%s = %s[%d];", utype, varname, name, index); output_line(ctx, "#define %s %s[%d];", varname, name, index); push_output(ctx, &ctx->mainline); ctx->indent++; output_line(ctx, "#undef %s", varname); // !!! FIXME: gross. pop_output(ctx); } // else pop_output(ctx); } // emit_METAL_uniform void emit_METAL_sampler(Context *ctx,int stage,TextureType ttype,int tb) { char var[64]; const char *texsuffix = NULL; switch (ttype) { case TEXTURE_TYPE_2D: texsuffix = "2d"; break; case TEXTURE_TYPE_CUBE: texsuffix = "cube"; break; case TEXTURE_TYPE_VOLUME: texsuffix = "3d"; break; default: assert(!"unexpected texture type"); return; } // switch get_METAL_varname_in_buf(ctx, REG_TYPE_SAMPLER, stage, var, sizeof (var)); push_output(ctx, &ctx->mainline_arguments); ctx->indent++; output_line(ctx, "texture%s %s_texture [[texture(%d)]],", texsuffix, var, stage); output_line(ctx, "sampler %s [[sampler(%d)]],", var, stage); pop_output(ctx); if (tb) // This sampler used a ps_1_1 TEXBEM opcode? { push_output(ctx, &ctx->mainline_top); ctx->indent++; char name[64]; const int index = ctx->uniform_float4_count; ctx->uniform_float4_count += 2; get_METAL_uniform_array_varname(ctx, REG_TYPE_CONST, name, sizeof (name)); output_line(ctx, "constant float4 &%s_texbem = %s[%d];", var, name, index); output_line(ctx, "constant float4 &%s_texbeml = %s[%d];", var, name, index+1); pop_output(ctx); } // if } // emit_METAL_sampler void emit_METAL_attribute(Context *ctx, RegisterType regtype, int regnum, MOJOSHADER_usage usage, int index, int wmask, int flags) { // !!! FIXME: this function doesn't deal with write masks at all yet! const char *usage_str = NULL; char index_str[16] = { '\0' }; char var[64]; get_METAL_varname_in_buf(ctx, regtype, regnum, var, sizeof (var)); //assert((flags & MOD_PP) == 0); // !!! FIXME: is PP allowed? if (index != 0) // !!! FIXME: a lot of these MUST be zero. snprintf(index_str, sizeof (index_str), "%u", (uint) index); if (shader_is_vertex(ctx)) { // pre-vs3 output registers. // these don't ever happen in DCL opcodes, I think. Map to vs_3_* // output registers. if (!shader_version_atleast(ctx, 3, 0)) { if (regtype == REG_TYPE_RASTOUT) { regtype = REG_TYPE_OUTPUT; index = regnum; switch ((const RastOutType) regnum) { case RASTOUT_TYPE_POSITION: usage = MOJOSHADER_USAGE_POSITION; break; case RASTOUT_TYPE_FOG: usage = MOJOSHADER_USAGE_FOG; break; case RASTOUT_TYPE_POINT_SIZE: usage = MOJOSHADER_USAGE_POINTSIZE; break; } // switch } // if else if (regtype == REG_TYPE_ATTROUT) { regtype = REG_TYPE_OUTPUT; usage = MOJOSHADER_USAGE_COLOR; index = regnum; } // else if else if (regtype == REG_TYPE_TEXCRDOUT) { regtype = REG_TYPE_OUTPUT; usage = MOJOSHADER_USAGE_TEXCOORD; index = regnum; } // else if } // if if (regtype == REG_TYPE_INPUT) { push_output(ctx, &ctx->inputs); if (buffer_size(ctx->inputs) == 0) { output_line(ctx, "struct %s_Input", ctx->mainfn); output_line(ctx, "{"); } // if ctx->indent++; output_line(ctx, "float4 %s [[attribute(%d)]];", var, regnum); pop_output(ctx); push_output(ctx, &ctx->mainline_top); ctx->indent++; // !!! FIXME: might trigger unused var warnings in Clang. //output_line(ctx, "constant float4 &%s = input.%s;", var, var); output_line(ctx, "#define %s input.%s", var, var); pop_output(ctx); push_output(ctx, &ctx->mainline); ctx->indent++; output_line(ctx, "#undef %s", var); // !!! FIXME: gross. pop_output(ctx); } // if else if (regtype == REG_TYPE_OUTPUT) { push_output(ctx, &ctx->outputs); if (buffer_size(ctx->outputs) == 0) { output_line(ctx, "struct %s_Output", ctx->mainfn); output_line(ctx, "{"); } // if ctx->indent++; switch (usage) { case MOJOSHADER_USAGE_POSITION: output_line(ctx, "float4 %s [[position]];", var); break; case MOJOSHADER_USAGE_POINTSIZE: output_line(ctx, "float %s [[point_size]];", var); break; case MOJOSHADER_USAGE_COLOR: output_line(ctx, "float4 %s [[user(color%d)]];", var, index); break; case MOJOSHADER_USAGE_FOG: output_line(ctx, "float4 %s [[user(fog)]];", var); break; case MOJOSHADER_USAGE_TEXCOORD: output_line(ctx, "float4 %s [[user(texcoord%d)]];", var, index); break; case MOJOSHADER_USAGE_NORMAL: output_line(ctx, "float4 %s [[user(normal)]];", var); default: // !!! FIXME: we need to deal with some more built-in varyings here. break; } // switch pop_output(ctx); push_output(ctx, &ctx->mainline_top); ctx->indent++; // !!! FIXME: this doesn't work. //output_line(ctx, "float4 &%s = output.%s;", var, var); output_line(ctx, "#define %s output.%s", var, var); pop_output(ctx); push_output(ctx, &ctx->mainline); ctx->indent++; output_line(ctx, "#undef %s", var); // !!! FIXME: gross. pop_output(ctx); } // else if else { fail(ctx, "unknown vertex shader attribute register"); } // else } // if else if (shader_is_pixel(ctx)) { // samplers DCLs get handled in emit_METAL_sampler(). if (flags & MOD_CENTROID) // !!! FIXME { failf(ctx, "centroid unsupported in %s profile", ctx->profile->name); return; } // if if ((regtype == REG_TYPE_COLOROUT) || (regtype == REG_TYPE_DEPTHOUT)) { push_output(ctx, &ctx->outputs); if (buffer_size(ctx->outputs) == 0) { output_line(ctx, "struct %s_Output", ctx->mainfn); output_line(ctx, "{"); } // if ctx->indent++; if (regtype == REG_TYPE_COLOROUT) output_line(ctx, "float4 %s [[color(%d)]];", var, regnum); else if (regtype == REG_TYPE_DEPTHOUT) output_line(ctx, "float %s [[depth(any)]];", var); pop_output(ctx); push_output(ctx, &ctx->mainline_top); ctx->indent++; // !!! FIXME: this doesn't work. //output_line(ctx, "float%s &%s = output.%s;", (regtype == REG_TYPE_DEPTHOUT) ? "" : "4", var, var); output_line(ctx, "#define %s output.%s", var, var); pop_output(ctx); push_output(ctx, &ctx->mainline); ctx->indent++; output_line(ctx, "#undef %s", var); // !!! FIXME: gross. pop_output(ctx); } // if // !!! FIXME: can you actualy have a texture register with COLOR usage? else if ((regtype == REG_TYPE_TEXTURE) || (regtype == REG_TYPE_INPUT) || (regtype == REG_TYPE_MISCTYPE)) { int skipreference = 0; push_output(ctx, &ctx->inputs); if (buffer_size(ctx->inputs) == 0) { output_line(ctx, "struct %s_Input", ctx->mainfn); output_line(ctx, "{"); } // if ctx->indent++; if (regtype == REG_TYPE_MISCTYPE) { const MiscTypeType mt = (MiscTypeType) regnum; if (mt == MISCTYPE_TYPE_FACE) output_line(ctx, "bool %s [[front_facing]];", var); else if (mt == MISCTYPE_TYPE_POSITION) output_line(ctx, "float4 %s [[position]];", var); else fail(ctx, "BUG: unhandled misc register"); } // else if else { if (usage == MOJOSHADER_USAGE_TEXCOORD) { // ps_1_1 does a different hack for this attribute. // Refer to emit_METAL_global()'s REG_TYPE_ADDRESS code. if (!shader_version_atleast(ctx, 1, 4)) skipreference = 1; output_line(ctx, "float4 %s [[user(texcoord%d)]];", var, index); } // if else if (usage == MOJOSHADER_USAGE_COLOR) output_line(ctx, "float4 %s [[user(color%d)]];", var, index); else if (usage == MOJOSHADER_USAGE_FOG) output_line(ctx, "float4 %s [[user(fog)]];", var); else if (usage == MOJOSHADER_USAGE_NORMAL) output_line(ctx, "float4 %s [[user(normal)]];", var); } // else pop_output(ctx); // !!! FIXME: can cause unused var warnings in Clang... #if 0 push_output(ctx, &ctx->mainline_top); ctx->indent++; if ((regtype == REG_TYPE_MISCTYPE)&&(regnum == MISCTYPE_TYPE_FACE)) output_line(ctx, "constant bool &%s = input.%s;", var, var); else if (!skipreference) output_line(ctx, "constant float4 &%s = input.%s;", var, var); pop_output(ctx); #endif if (!skipreference) { push_output(ctx, &ctx->mainline_top); ctx->indent++; output_line(ctx, "#define %s input.%s", var, var); pop_output(ctx); push_output(ctx, &ctx->mainline); ctx->indent++; output_line(ctx, "#undef %s", var); // !!! FIXME: gross. pop_output(ctx); } // if } // else if else { fail(ctx, "unknown pixel shader attribute register"); } // else } // else if else { fail(ctx, "Unknown shader type"); // state machine should catch this. } // else } // emit_METAL_attribute void emit_METAL_NOP(Context *ctx) { // no-op is a no-op. :) } // emit_METAL_NOP void emit_METAL_MOV(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; make_METAL_destarg_assign(ctx, code, sizeof (code), "%s", src0); output_line(ctx, "%s", code); } // emit_METAL_MOV void emit_METAL_ADD(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; make_METAL_destarg_assign(ctx, code, sizeof (code), "%s + %s", src0, src1); output_line(ctx, "%s", code); } // emit_METAL_ADD void emit_METAL_SUB(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; make_METAL_destarg_assign(ctx, code, sizeof (code), "%s - %s", src0, src1); output_line(ctx, "%s", code); } // emit_METAL_SUB void emit_METAL_MAD(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char src2[64]; make_METAL_srcarg_string_masked(ctx, 2, src2, sizeof (src2)); char code[128]; make_METAL_destarg_assign(ctx, code, sizeof (code), "(%s * %s) + %s", src0, src1, src2); output_line(ctx, "%s", code); } // emit_METAL_MAD void emit_METAL_MUL(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; make_METAL_destarg_assign(ctx, code, sizeof (code), "%s * %s", src0, src1); output_line(ctx, "%s", code); } // emit_METAL_MUL void emit_METAL_RCP(Context *ctx) { const int vecsize = vecsize_from_writemask(ctx->dest_arg.writemask); char cast[16] = { '\0' }; if (vecsize != 1) snprintf(cast, sizeof (cast), "float%d", vecsize); char src0[64]; make_METAL_srcarg_string_scalar(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "%s((%s == 0.0) ? FLT_MAX : 1.0 / %s)", cast, src0, src0); output_line(ctx, "%s", code); } // emit_METAL_RCP void emit_METAL_RSQ(Context *ctx) { const int vecsize = vecsize_from_writemask(ctx->dest_arg.writemask); char cast[16] = { '\0' }; if (vecsize != 1) snprintf(cast, sizeof (cast), "float%d", vecsize); char src0[64]; make_METAL_srcarg_string_scalar(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "%s((%s == 0.0) ? FLT_MAX : rsqrt(abs(%s)))", cast, src0, src0); output_line(ctx, "%s", code); } // emit_METAL_RSQ void emit_METAL_dotprod(Context *ctx, const char *src0, const char *src1, const char *extra) { const int vecsize = vecsize_from_writemask(ctx->dest_arg.writemask); char castleft[16] = { '\0' }; const char *castright = ""; if (vecsize != 1) { snprintf(castleft, sizeof (castleft), "float%d(", vecsize); castright = ")"; } // if char code[128]; ctx->metal_need_header_geometric = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "%sdot(%s, %s)%s%s", castleft, src0, src1, extra, castright); output_line(ctx, "%s", code); } // emit_METAL_dotprod void emit_METAL_DP3(Context *ctx) { char src0[64]; make_METAL_srcarg_string_vec3(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_vec3(ctx, 1, src1, sizeof (src1)); emit_METAL_dotprod(ctx, src0, src1, ""); } // emit_METAL_DP3 void emit_METAL_DP4(Context *ctx) { char src0[64]; make_METAL_srcarg_string_full(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_full(ctx, 1, src1, sizeof (src1)); emit_METAL_dotprod(ctx, src0, src1, ""); } // emit_METAL_DP4 void emit_METAL_MIN(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "min(%s, %s)", src0, src1); output_line(ctx, "%s", code); } // emit_METAL_MIN void emit_METAL_MAX(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "max(%s, %s)", src0, src1); output_line(ctx, "%s", code); } // emit_METAL_MAX void emit_METAL_SLT(Context *ctx) { const int vecsize = vecsize_from_writemask(ctx->dest_arg.writemask); char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; // float(bool) or vec(bvec) results in 0.0 or 1.0, like SLT wants. if (vecsize == 1) make_METAL_destarg_assign(ctx, code, sizeof (code), "float(%s < %s)", src0, src1); else { make_METAL_destarg_assign(ctx, code, sizeof (code), "float%d(%s < %s)", vecsize, src0, src1); } // else output_line(ctx, "%s", code); } // emit_METAL_SLT void emit_METAL_SGE(Context *ctx) { const int vecsize = vecsize_from_writemask(ctx->dest_arg.writemask); char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; // float(bool) or vec(bvec) results in 0.0 or 1.0, like SGE wants. if (vecsize == 1) { make_METAL_destarg_assign(ctx, code, sizeof (code), "float(%s >= %s)", src0, src1); } // if else { make_METAL_destarg_assign(ctx, code, sizeof (code), "float%d(%s >= %s)", vecsize, src0, src1); } // else output_line(ctx, "%s", code); } // emit_METAL_SGE void emit_METAL_EXP(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "exp2(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_EXP void emit_METAL_LOG(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "log2(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_LOG void emit_METAL_LIT_helper(Context *ctx) { const char *maxp = "127.9961"; // value from the dx9 reference. if (ctx->glsl_generated_lit_helper) return; ctx->glsl_generated_lit_helper = 1; ctx->metal_need_header_common = 1; ctx->metal_need_header_math = 1; push_output(ctx, &ctx->helpers); output_line(ctx, "float4 LIT(const float4 src)"); output_line(ctx, "{"); ctx->indent++; output_line(ctx, "const float power = clamp(src.w, -%s, %s);",maxp,maxp); output_line(ctx, "float4 retval = float4(1.0, 0.0, 0.0, 1.0);"); output_line(ctx, "if (src.x > 0.0) {"); ctx->indent++; output_line(ctx, "retval.y = src.x;"); output_line(ctx, "if (src.y > 0.0) {"); ctx->indent++; output_line(ctx, "retval.z = pow(src.y, power);"); ctx->indent--; output_line(ctx, "}"); ctx->indent--; output_line(ctx, "}"); output_line(ctx, "return retval;"); ctx->indent--; output_line(ctx, "}"); output_blank_line(ctx); pop_output(ctx); } // emit_METAL_LIT_helper void emit_METAL_LIT(Context *ctx) { char src0[64]; make_METAL_srcarg_string_full(ctx, 0, src0, sizeof (src0)); char code[128]; emit_METAL_LIT_helper(ctx); make_METAL_destarg_assign(ctx, code, sizeof (code), "LIT(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_LIT void emit_METAL_DST(Context *ctx) { // !!! FIXME: needs to take ctx->dst_arg.writemask into account. char src0_y[64]; make_METAL_srcarg_string_y(ctx, 0, src0_y, sizeof (src0_y)); char src1_y[64]; make_METAL_srcarg_string_y(ctx, 1, src1_y, sizeof (src1_y)); char src0_z[64]; make_METAL_srcarg_string_z(ctx, 0, src0_z, sizeof (src0_z)); char src1_w[64]; make_METAL_srcarg_string_w(ctx, 1, src1_w, sizeof (src1_w)); char code[128]; make_METAL_destarg_assign(ctx, code, sizeof (code), "float4(1.0, %s * %s, %s, %s)", src0_y, src1_y, src0_z, src1_w); output_line(ctx, "%s", code); } // emit_METAL_DST void emit_METAL_LRP(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char src2[64]; make_METAL_srcarg_string_masked(ctx, 2, src2, sizeof (src2)); char code[128]; ctx->metal_need_header_common = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "mix(%s, %s, %s)", src2, src1, src0); output_line(ctx, "%s", code); } // emit_METAL_LRP void emit_METAL_FRC(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "fract(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_FRC void emit_METAL_M4X4(Context *ctx) { char src0[64]; make_METAL_srcarg_string_full(ctx, 0, src0, sizeof (src0)); char row0[64]; make_METAL_srcarg_string_full(ctx, 1, row0, sizeof (row0)); char row1[64]; make_METAL_srcarg_string_full(ctx, 2, row1, sizeof (row1)); char row2[64]; make_METAL_srcarg_string_full(ctx, 3, row2, sizeof (row2)); char row3[64]; make_METAL_srcarg_string_full(ctx, 4, row3, sizeof (row3)); char code[256]; ctx->metal_need_header_geometric = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "float4(dot(%s, %s), dot(%s, %s), dot(%s, %s), dot(%s, %s))", src0, row0, src0, row1, src0, row2, src0, row3); output_line(ctx, "%s", code); } // emit_METAL_M4X4 void emit_METAL_M4X3(Context *ctx) { char src0[64]; make_METAL_srcarg_string_full(ctx, 0, src0, sizeof (src0)); char row0[64]; make_METAL_srcarg_string_full(ctx, 1, row0, sizeof (row0)); char row1[64]; make_METAL_srcarg_string_full(ctx, 2, row1, sizeof (row1)); char row2[64]; make_METAL_srcarg_string_full(ctx, 3, row2, sizeof (row2)); char code[256]; ctx->metal_need_header_geometric = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "float3(dot(%s, %s), dot(%s, %s), dot(%s, %s))", src0, row0, src0, row1, src0, row2); output_line(ctx, "%s", code); } // emit_METAL_M4X3 void emit_METAL_M3X4(Context *ctx) { char src0[64]; make_METAL_srcarg_string_vec3(ctx, 0, src0, sizeof (src0)); char row0[64]; make_METAL_srcarg_string_vec3(ctx, 1, row0, sizeof (row0)); char row1[64]; make_METAL_srcarg_string_vec3(ctx, 2, row1, sizeof (row1)); char row2[64]; make_METAL_srcarg_string_vec3(ctx, 3, row2, sizeof (row2)); char row3[64]; make_METAL_srcarg_string_vec3(ctx, 4, row3, sizeof (row3)); char code[256]; ctx->metal_need_header_geometric = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "float4(dot(%s, %s), dot(%s, %s), " "dot(%s, %s), dot(%s, %s))", src0, row0, src0, row1, src0, row2, src0, row3); output_line(ctx, "%s", code); } // emit_METAL_M3X4 void emit_METAL_M3X3(Context *ctx) { char src0[64]; make_METAL_srcarg_string_vec3(ctx, 0, src0, sizeof (src0)); char row0[64]; make_METAL_srcarg_string_vec3(ctx, 1, row0, sizeof (row0)); char row1[64]; make_METAL_srcarg_string_vec3(ctx, 2, row1, sizeof (row1)); char row2[64]; make_METAL_srcarg_string_vec3(ctx, 3, row2, sizeof (row2)); char code[256]; ctx->metal_need_header_geometric = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "float3(dot(%s, %s), dot(%s, %s), dot(%s, %s))", src0, row0, src0, row1, src0, row2); output_line(ctx, "%s", code); } // emit_METAL_M3X3 void emit_METAL_M3X2(Context *ctx) { char src0[64]; make_METAL_srcarg_string_vec3(ctx, 0, src0, sizeof (src0)); char row0[64]; make_METAL_srcarg_string_vec3(ctx, 1, row0, sizeof (row0)); char row1[64]; make_METAL_srcarg_string_vec3(ctx, 2, row1, sizeof (row1)); char code[256]; ctx->metal_need_header_geometric = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "float2(dot(%s, %s), dot(%s, %s))", src0, row0, src0, row1); output_line(ctx, "%s", code); } // emit_METAL_M3X2 void emit_METAL_CALL(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); if (ctx->loops > 0) output_line(ctx, "%s(aL);", src0); else output_line(ctx, "%s();", src0); } // emit_METAL_CALL void emit_METAL_CALLNZ(Context *ctx) { // !!! FIXME: if src1 is a constbool that's true, we can remove the // !!! FIXME: if. If it's false, we can make this a no-op. char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); if (ctx->loops > 0) output_line(ctx, "if (%s) { %s(aL); }", src1, src0); else output_line(ctx, "if (%s) { %s(); }", src1, src0); } // emit_METAL_CALLNZ void emit_METAL_LOOP(Context *ctx) { // !!! FIXME: swizzle? char var[64]; get_METAL_srcarg_varname(ctx, 1, var, sizeof (var)); assert(ctx->source_args[0].regnum == 0); // in case they add aL1 someday. output_line(ctx, "{"); ctx->indent++; output_line(ctx, "const int aLend = %s.x + %s.y;", var, var); output_line(ctx, "for (int aL = %s.y; aL < aLend; aL += %s.z) {", var, var); ctx->indent++; } // emit_METAL_LOOP void emit_METAL_RET(Context *ctx) { // thankfully, the MSDN specs say a RET _has_ to end a function...no // early returns. So if you hit one, you know you can safely close // a high-level function. push_output(ctx, &ctx->postflight); output_line(ctx, "}"); output_blank_line(ctx); set_output(ctx, &ctx->subroutines); // !!! FIXME: is this for LABEL? Maybe set it there so we don't allocate unnecessarily. } // emit_METAL_RET void emit_METAL_ENDLOOP(Context *ctx) { ctx->indent--; output_line(ctx, "}"); ctx->indent--; output_line(ctx, "}"); } // emit_METAL_ENDLOOP void emit_METAL_LABEL(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); const int label = ctx->source_args[0].regnum; RegisterList *reg = reglist_find(&ctx->used_registers, REG_TYPE_LABEL, label); assert(ctx->output == ctx->subroutines); // not mainline, etc. assert(ctx->indent == 0); // we shouldn't be in the middle of a function. // MSDN specs say CALL* has to come before the LABEL, so we know if we // can ditch the entire function here as unused. if (reg == NULL) set_output(ctx, &ctx->ignore); // Func not used. Parse, but don't output. // !!! FIXME: it would be nice if we could determine if a function is // !!! FIXME: only called once and, if so, forcibly inline it. // !!! FIXME: this worked in GLSL because all our state is global to the shader, // !!! FIXME: but in metal we kept it local to the shader mainline. // !!! FIXME: Can we do C++11 lambdas in Metal to have nested functions? :) const char *uses_loopreg = ((reg) && (reg->misc == 1)) ? "int aL" : ""; output_line(ctx, "void %s(%s)", src0, uses_loopreg); output_line(ctx, "{"); ctx->indent++; } // emit_METAL_LABEL void emit_METAL_DCL(Context *ctx) { // no-op. We do this in our emit_attribute() and emit_uniform(). } // emit_METAL_DCL void emit_METAL_POW(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "pow(abs(%s), %s)", src0, src1); output_line(ctx, "%s", code); } // emit_METAL_POW void emit_METAL_CRS(Context *ctx) { // !!! FIXME: needs to take ctx->dst_arg.writemask into account. char src0[64]; make_METAL_srcarg_string_vec3(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_vec3(ctx, 1, src1, sizeof (src1)); char code[128]; ctx->metal_need_header_geometric = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "cross(%s, %s)", src0, src1); output_line(ctx, "%s", code); } // emit_METAL_CRS void emit_METAL_SGN(Context *ctx) { // (we don't need the temporary registers specified for the D3D opcode.) char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_common = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "sign(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_SGN void emit_METAL_ABS(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_math = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "abs(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_ABS void emit_METAL_NRM(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_geometric = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "normalize(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_NRM void emit_METAL_SINCOS(Context *ctx) { // we don't care about the temp registers that <= sm2 demands; ignore them. // sm2 also talks about what components are left untouched vs. undefined, // but we just leave those all untouched with Metal write masks (which // would fulfill the "undefined" requirement, too). const int mask = ctx->dest_arg.writemask; char src0[64]; make_METAL_srcarg_string_scalar(ctx, 0, src0, sizeof (src0)); char code[128] = { '\0' }; ctx->metal_need_header_math = 1; if (writemask_x(mask)) make_METAL_destarg_assign(ctx, code, sizeof (code), "cos(%s)", src0); else if (writemask_y(mask)) make_METAL_destarg_assign(ctx, code, sizeof (code), "sin(%s)", src0); else if (writemask_xy(mask)) { // !!! FIXME: can use sincos(), but need to assign cos to a temp, since it needs a reference. make_METAL_destarg_assign(ctx, code, sizeof (code), "float2(cos(%s), sin(%s))", src0, src0); } // else if output_line(ctx, "%s", code); } // emit_METAL_SINCOS void emit_METAL_REP(Context *ctx) { // !!! FIXME: // msdn docs say legal loop values are 0 to 255. We can check DEFI values // at parse time, but if they are pulling a value from a uniform, do // we clamp here? // !!! FIXME: swizzle is legal here, right? char src0[64]; make_METAL_srcarg_string_x(ctx, 0, src0, sizeof (src0)); const uint rep = (uint) ctx->reps; output_line(ctx, "for (int rep%u = 0; rep%u < %s; rep%u++) {", rep, rep, src0, rep); ctx->indent++; } // emit_METAL_REP void emit_METAL_ENDREP(Context *ctx) { ctx->indent--; output_line(ctx, "}"); } // emit_METAL_ENDREP void emit_METAL_IF(Context *ctx) { char src0[64]; make_METAL_srcarg_string_scalar(ctx, 0, src0, sizeof (src0)); output_line(ctx, "if (%s) {", src0); ctx->indent++; } // emit_METAL_IF void emit_METAL_IFC(Context *ctx) { const char *comp = get_METAL_comparison_string_scalar(ctx); char src0[64]; make_METAL_srcarg_string_scalar(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_scalar(ctx, 1, src1, sizeof (src1)); output_line(ctx, "if (%s %s %s) {", src0, comp, src1); ctx->indent++; } // emit_METAL_IFC void emit_METAL_ELSE(Context *ctx) { ctx->indent--; output_line(ctx, "} else {"); ctx->indent++; } // emit_METAL_ELSE void emit_METAL_ENDIF(Context *ctx) { ctx->indent--; output_line(ctx, "}"); } // emit_METAL_ENDIF void emit_METAL_BREAK(Context *ctx) { output_line(ctx, "break;"); } // emit_METAL_BREAK void emit_METAL_BREAKC(Context *ctx) { const char *comp = get_METAL_comparison_string_scalar(ctx); char src0[64]; make_METAL_srcarg_string_scalar(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_scalar(ctx, 1, src1, sizeof (src1)); output_line(ctx, "if (%s %s %s) { break; }", src0, comp, src1); } // emit_METAL_BREAKC void emit_METAL_MOVA(Context *ctx) { const int vecsize = vecsize_from_writemask(ctx->dest_arg.writemask); char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_math = 1; ctx->metal_need_header_common = 1; if (vecsize == 1) { make_METAL_destarg_assign(ctx, code, sizeof (code), "int(floor(abs(%s) + 0.5) * sign(%s))", src0, src0); } // if else { make_METAL_destarg_assign(ctx, code, sizeof (code), "int%d(floor(abs(%s) + float%d(0.5)) * sign(%s))", vecsize, src0, vecsize, src0); } // else output_line(ctx, "%s", code); } // emit_METAL_MOVA void emit_METAL_DEFB(Context *ctx) { char varname[64]; get_METAL_destarg_varname(ctx, varname, sizeof (varname)); push_output(ctx, &ctx->mainline_top); ctx->indent++; output_line(ctx, "const bool %s = %s;", varname, ctx->dwords[0] ? "true" : "false"); pop_output(ctx); } // emit_METAL_DEFB void emit_METAL_DEFI(Context *ctx) { char varname[64]; get_METAL_destarg_varname(ctx, varname, sizeof (varname)); const int32 *x = (const int32 *) ctx->dwords; push_output(ctx, &ctx->mainline_top); ctx->indent++; output_line(ctx, "const int4 %s = int4(%d, %d, %d, %d);", varname, (int) x[0], (int) x[1], (int) x[2], (int) x[3]); pop_output(ctx); } // emit_METAL_DEFI EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(TEXCRD) void emit_METAL_TEXKILL(Context *ctx) { char dst[64]; get_METAL_destarg_varname(ctx, dst, sizeof (dst)); ctx->metal_need_header_relational = 1; ctx->metal_need_header_graphics = 1; output_line(ctx, "if (any(%s.xyz < float3(0.0))) discard_fragment();", dst); } // emit_METAL_TEXKILL static void metal_texld(Context *ctx, const int texldd) { ctx->metal_need_header_texture = 1; if (!shader_version_atleast(ctx, 1, 4)) { DestArgInfo *info = &ctx->dest_arg; char dst[64]; char sampler[64]; char code[128] = {0}; assert(!texldd); RegisterList *sreg; sreg = reglist_find(&ctx->samplers, REG_TYPE_SAMPLER, info->regnum); const TextureType ttype = (TextureType) (sreg ? sreg->index : 0); char swizzle[4] = { 'x', 'y', 'z', '\0' }; if (ttype == TEXTURE_TYPE_2D) swizzle[2] = '\0'; // "xy" instead of "xyz". // !!! FIXME: this code counts on the register not having swizzles, etc. get_METAL_destarg_varname(ctx, dst, sizeof (dst)); get_METAL_varname_in_buf(ctx, REG_TYPE_SAMPLER, info->regnum, sampler, sizeof (sampler)); make_METAL_destarg_assign(ctx, code, sizeof (code), "%s_texture.sample(%s, %s.%s)", sampler, sampler, dst, swizzle); output_line(ctx, "%s", code); } // if else if (!shader_version_atleast(ctx, 2, 0)) { // ps_1_4 is different, too! fail(ctx, "TEXLD == Shader Model 1.4 unimplemented."); // !!! FIXME return; } // else if else { const SourceArgInfo *samp_arg = &ctx->source_args[1]; RegisterList *sreg = reglist_find(&ctx->samplers, REG_TYPE_SAMPLER, samp_arg->regnum); const char *funcname = NULL; char src0[64] = { '\0' }; char src1[64]; get_METAL_srcarg_varname(ctx, 1, src1, sizeof (src1)); // !!! FIXME: SRC_MOD? char src2[64] = { '\0' }; char src3[64] = { '\0' }; if (sreg == NULL) { fail(ctx, "TEXLD using undeclared sampler"); return; } // if const char *grad = ""; if (texldd) { switch ((const TextureType) sreg->index) { case TEXTURE_TYPE_2D: grad = "2d"; make_METAL_srcarg_string_vec2(ctx, 2, src2, sizeof (src2)); make_METAL_srcarg_string_vec2(ctx, 3, src3, sizeof (src3)); break; case TEXTURE_TYPE_VOLUME: grad = "3d"; make_METAL_srcarg_string_vec3(ctx, 2, src2, sizeof (src2)); make_METAL_srcarg_string_vec3(ctx, 3, src3, sizeof (src3)); break; case TEXTURE_TYPE_CUBE: grad = "cube"; make_METAL_srcarg_string_vec3(ctx, 2, src2, sizeof (src2)); make_METAL_srcarg_string_vec3(ctx, 3, src3, sizeof (src3)); break; } // switch } // if // !!! FIXME: can TEXLDD set instruction_controls? // !!! FIXME: does the d3d bias value map directly to Metal? const char *biasleft = ""; const char *biasright = ""; char bias[64] = { '\0' }; if (ctx->instruction_controls == CONTROL_TEXLDB) { biasleft = ", bias("; make_METAL_srcarg_string_w(ctx, 0, bias, sizeof (bias)); biasright = ")"; } // if // Metal doesn't have a texture2DProj() function, but you just divide // your texcoords by texcoords.w to achieve it anyhow, so DIY. const char *projop = ""; char proj[64] = { '\0' }; if (ctx->instruction_controls == CONTROL_TEXLDP) { if (sreg->index == TEXTURE_TYPE_CUBE) fail(ctx, "TEXLDP on a cubemap"); // !!! FIXME: is this legal? projop = " / "; make_METAL_srcarg_string_w(ctx, 0, proj, sizeof (proj)); } // if switch ((const TextureType) sreg->index) { case TEXTURE_TYPE_2D: make_METAL_srcarg_string_vec2(ctx, 0, src0, sizeof (src0)); break; case TEXTURE_TYPE_CUBE: case TEXTURE_TYPE_VOLUME: make_METAL_srcarg_string_vec3(ctx, 0, src0, sizeof (src0)); break; default: fail(ctx, "unknown texture type"); return; } // switch assert(!isscalar(ctx, ctx->shader_type, samp_arg->regtype, samp_arg->regnum)); char swiz_str[6] = { '\0' }; make_METAL_swizzle_string(swiz_str, sizeof (swiz_str), samp_arg->swizzle, ctx->dest_arg.writemask); char code[128]; if (texldd) { make_METAL_destarg_assign(ctx, code, sizeof (code), "%s_texture.sample(%s, %s, gradient%s(%s, %s))%s", src1, src1, src0, grad, src2, src3, swiz_str); } // if else { make_METAL_destarg_assign(ctx, code, sizeof (code), "%s_texture.sample(%s, %s%s%s%s%s%s)%s", src1, src1, src0, projop, proj, biasleft, bias, biasright, swiz_str); } // else output_line(ctx, "%s", code); } // else } // metal_texld void emit_METAL_TEXLD(Context *ctx) { metal_texld(ctx, 0); } // emit_METAL_TEXLD void emit_METAL_TEXBEM(Context *ctx) { DestArgInfo *info = &ctx->dest_arg; char dst[64]; get_METAL_destarg_varname(ctx, dst, sizeof (dst)); char src[64]; get_METAL_srcarg_varname(ctx, 0, src, sizeof (src)); char sampler[64]; char code[512]; ctx->metal_need_header_texture = 1; // !!! FIXME: this code counts on the register not having swizzles, etc. get_METAL_varname_in_buf(ctx, REG_TYPE_SAMPLER, info->regnum, sampler, sizeof (sampler)); make_METAL_destarg_assign(ctx, code, sizeof (code), "%s_texture.sample(%s, float2(%s.x + (%s_texbem.x * %s.x) + (%s_texbem.z * %s.y)," " %s.y + (%s_texbem.y * %s.x) + (%s_texbem.w * %s.y)))", sampler, sampler, dst, sampler, src, sampler, src, dst, sampler, src, sampler, src); output_line(ctx, "%s", code); } // emit_METAL_TEXBEM void emit_METAL_TEXBEML(Context *ctx) { // !!! FIXME: this code counts on the register not having swizzles, etc. DestArgInfo *info = &ctx->dest_arg; char dst[64]; get_METAL_destarg_varname(ctx, dst, sizeof (dst)); char src[64]; get_METAL_srcarg_varname(ctx, 0, src, sizeof (src)); char sampler[64]; char code[512]; ctx->metal_need_header_texture = 1; get_METAL_varname_in_buf(ctx, REG_TYPE_SAMPLER, info->regnum, sampler, sizeof (sampler)); make_METAL_destarg_assign(ctx, code, sizeof (code), "(%s_texture.sample(%s, float2(%s.x + (%s_texbem.x * %s.x) + (%s_texbem.z * %s.y)," " %s.y + (%s_texbem.y * %s.x) + (%s_texbem.w * %s.y)))) *" " ((%s.z * %s_texbeml.x) + %s_texbem.y)", sampler, sampler, dst, sampler, src, sampler, src, dst, sampler, src, sampler, src, src, sampler, sampler); output_line(ctx, "%s", code); } // emit_METAL_TEXBEML EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(TEXREG2AR) // !!! FIXME EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(TEXREG2GB) // !!! FIXME void emit_METAL_TEXM3X2PAD(Context *ctx) { // no-op ... work happens in emit_METAL_TEXM3X2TEX(). } // emit_METAL_TEXM3X2PAD void emit_METAL_TEXM3X2TEX(Context *ctx) { if (ctx->texm3x2pad_src0 == -1) return; DestArgInfo *info = &ctx->dest_arg; char dst[64]; char src0[64]; char src1[64]; char src2[64]; char sampler[64]; char code[512]; ctx->metal_need_header_texture = 1; ctx->metal_need_header_geometric = 1; // !!! FIXME: this code counts on the register not having swizzles, etc. get_METAL_varname_in_buf(ctx, REG_TYPE_SAMPLER, info->regnum, sampler, sizeof (sampler)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x2pad_src0, src0, sizeof (src0)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x2pad_dst0, src1, sizeof (src1)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->source_args[0].regnum, src2, sizeof (src2)); get_METAL_destarg_varname(ctx, dst, sizeof (dst)); make_METAL_destarg_assign(ctx, code, sizeof (code), "%s_texture.sample(%s, float2(dot(%s.xyz, %s.xyz), dot(%s.xyz, %s.xyz)))", sampler, sampler, src0, src1, src2, dst); output_line(ctx, "%s", code); } // emit_METAL_TEXM3X2TEX void emit_METAL_TEXM3X3PAD(Context *ctx) { // no-op ... work happens in emit_METAL_TEXM3X3*(). } // emit_METAL_TEXM3X3PAD void emit_METAL_TEXM3X3TEX(Context *ctx) { if (ctx->texm3x3pad_src1 == -1) return; DestArgInfo *info = &ctx->dest_arg; char dst[64]; char src0[64]; char src1[64]; char src2[64]; char src3[64]; char src4[64]; char sampler[64]; char code[512]; ctx->metal_need_header_texture = 1; ctx->metal_need_header_geometric = 1; // !!! FIXME: this code counts on the register not having swizzles, etc. get_METAL_varname_in_buf(ctx, REG_TYPE_SAMPLER, info->regnum, sampler, sizeof (sampler)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_dst0, src0, sizeof (src0)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_src0, src1, sizeof (src1)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_dst1, src2, sizeof (src2)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_src1, src3, sizeof (src3)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->source_args[0].regnum, src4, sizeof (src4)); get_METAL_destarg_varname(ctx, dst, sizeof (dst)); RegisterList *sreg = reglist_find(&ctx->samplers, REG_TYPE_SAMPLER, info->regnum); const TextureType ttype = (TextureType) (sreg ? sreg->index : 0); const char *ttypestr = (ttype == TEXTURE_TYPE_CUBE) ? "Cube" : "3D"; make_METAL_destarg_assign(ctx, code, sizeof (code), "texture%s(%s," " float3(dot(%s.xyz, %s.xyz)," " dot(%s.xyz, %s.xyz)," " dot(%s.xyz, %s.xyz)))", ttypestr, sampler, src0, src1, src2, src3, dst, src4); output_line(ctx, "%s", code); } // emit_METAL_TEXM3X3TEX void emit_METAL_TEXM3X3SPEC_helper(Context *ctx) { if (ctx->glsl_generated_texm3x3spec_helper) return; ctx->glsl_generated_texm3x3spec_helper = 1; push_output(ctx, &ctx->helpers); output_line(ctx, "float3 TEXM3X3SPEC_reflection(const float3 normal, const float3 eyeray)"); output_line(ctx, "{"); ctx->indent++; output_line(ctx, "return (2.0 * ((normal * eyeray) / (normal * normal)) * normal) - eyeray;"); ctx->indent--; output_line(ctx, "}"); output_blank_line(ctx); pop_output(ctx); } // emit_METAL_TEXM3X3SPEC_helper void emit_METAL_TEXM3X3SPEC(Context *ctx) { if (ctx->texm3x3pad_src1 == -1) return; DestArgInfo *info = &ctx->dest_arg; char dst[64]; char src0[64]; char src1[64]; char src2[64]; char src3[64]; char src4[64]; char src5[64]; char sampler[64]; char code[512]; ctx->metal_need_header_texture = 1; ctx->metal_need_header_geometric = 1; emit_METAL_TEXM3X3SPEC_helper(ctx); // !!! FIXME: this code counts on the register not having swizzles, etc. get_METAL_varname_in_buf(ctx, REG_TYPE_SAMPLER, info->regnum, sampler, sizeof (sampler)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_dst0, src0, sizeof (src0)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_src0, src1, sizeof (src1)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_dst1, src2, sizeof (src2)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_src1, src3, sizeof (src3)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->source_args[0].regnum, src4, sizeof (src4)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->source_args[1].regnum, src5, sizeof (src5)); get_METAL_destarg_varname(ctx, dst, sizeof (dst)); RegisterList *sreg = reglist_find(&ctx->samplers, REG_TYPE_SAMPLER, info->regnum); const TextureType ttype = (TextureType) (sreg ? sreg->index : 0); const char *ttypestr = (ttype == TEXTURE_TYPE_CUBE) ? "Cube" : "3D"; make_METAL_destarg_assign(ctx, code, sizeof (code), "texture%s(%s, " "TEXM3X3SPEC_reflection(" "float3(" "dot(%s.xyz, %s.xyz), " "dot(%s.xyz, %s.xyz), " "dot(%s.xyz, %s.xyz)" ")," "%s.xyz," ")" ")", ttypestr, sampler, src0, src1, src2, src3, dst, src4, src5); output_line(ctx, "%s", code); } // emit_METAL_TEXM3X3SPEC void emit_METAL_TEXM3X3VSPEC(Context *ctx) { if (ctx->texm3x3pad_src1 == -1) return; DestArgInfo *info = &ctx->dest_arg; char dst[64]; char src0[64]; char src1[64]; char src2[64]; char src3[64]; char src4[64]; char sampler[64]; char code[512]; ctx->metal_need_header_texture = 1; ctx->metal_need_header_geometric = 1; emit_METAL_TEXM3X3SPEC_helper(ctx); // !!! FIXME: this code counts on the register not having swizzles, etc. get_METAL_varname_in_buf(ctx, REG_TYPE_SAMPLER, info->regnum, sampler, sizeof (sampler)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_dst0, src0, sizeof (src0)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_src0, src1, sizeof (src1)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_dst1, src2, sizeof (src2)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_src1, src3, sizeof (src3)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->source_args[0].regnum, src4, sizeof (src4)); get_METAL_destarg_varname(ctx, dst, sizeof (dst)); RegisterList *sreg = reglist_find(&ctx->samplers, REG_TYPE_SAMPLER, info->regnum); const TextureType ttype = (TextureType) (sreg ? sreg->index : 0); const char *ttypestr = (ttype == TEXTURE_TYPE_CUBE) ? "Cube" : "3D"; make_METAL_destarg_assign(ctx, code, sizeof (code), "texture%s(%s, " "TEXM3X3SPEC_reflection(" "float3(" "dot(%s.xyz, %s.xyz), " "dot(%s.xyz, %s.xyz), " "dot(%s.xyz, %s.xyz)" "), " "float3(%s.w, %s.w, %s.w)" ")" ")", ttypestr, sampler, src0, src1, src2, src3, dst, src4, src0, src2, dst); output_line(ctx, "%s", code); } // emit_METAL_TEXM3X3VSPEC void emit_METAL_EXPP(Context *ctx) { // !!! FIXME: msdn's asm docs don't list this opcode, I'll have to check the driver documentation. emit_METAL_EXP(ctx); // I guess this is just partial precision EXP? } // emit_METAL_EXPP void emit_METAL_LOGP(Context *ctx) { // LOGP is just low-precision LOG, but we'll take the higher precision. emit_METAL_LOG(ctx); } // emit_METAL_LOGP // common code between CMP and CND. void emit_METAL_comparison_operations(Context *ctx, const char *cmp) { int i, j; DestArgInfo *dst = &ctx->dest_arg; const SourceArgInfo *srcarg0 = &ctx->source_args[0]; const int origmask = dst->writemask; int used_swiz[4] = { 0, 0, 0, 0 }; const int writemask[4] = { dst->writemask0, dst->writemask1, dst->writemask2, dst->writemask3 }; const int src0swiz[4] = { srcarg0->swizzle_x, srcarg0->swizzle_y, srcarg0->swizzle_z, srcarg0->swizzle_w }; for (i = 0; i < 4; i++) { int mask = (1 << i); if (!writemask[i]) continue; if (used_swiz[i]) continue; // This is a swizzle we haven't checked yet. used_swiz[i] = 1; // see if there are any other elements swizzled to match (.yyyy) for (j = i + 1; j < 4; j++) { if (!writemask[j]) continue; if (src0swiz[i] != src0swiz[j]) continue; mask |= (1 << j); used_swiz[j] = 1; } // for // okay, (mask) should be the writemask of swizzles we like. //return make_METAL_srcarg_string(ctx, idx, (1 << 0)); char src0[64]; char src1[64]; char src2[64]; make_METAL_srcarg_string(ctx, 0, (1 << i), src0, sizeof (src0)); make_METAL_srcarg_string(ctx, 1, mask, src1, sizeof (src1)); make_METAL_srcarg_string(ctx, 2, mask, src2, sizeof (src2)); set_dstarg_writemask(dst, mask); char code[128]; make_METAL_destarg_assign(ctx, code, sizeof (code), "((%s %s) ? %s : %s)", src0, cmp, src1, src2); output_line(ctx, "%s", code); } // for set_dstarg_writemask(dst, origmask); } // emit_METAL_comparison_operations void emit_METAL_CND(Context *ctx) { emit_METAL_comparison_operations(ctx, "> 0.5"); } // emit_METAL_CND void emit_METAL_DEF(Context *ctx) { const float *val = (const float *) ctx->dwords; // !!! FIXME: could be int? char varname[64]; get_METAL_destarg_varname(ctx, varname, sizeof (varname)); char val0[32]; floatstr(ctx, val0, sizeof (val0), val[0], 1); char val1[32]; floatstr(ctx, val1, sizeof (val1), val[1], 1); char val2[32]; floatstr(ctx, val2, sizeof (val2), val[2], 1); char val3[32]; floatstr(ctx, val3, sizeof (val3), val[3], 1); push_output(ctx, &ctx->mainline_top); ctx->indent++; // The "(void) %s;" is to make the compiler not warn if this isn't used. output_line(ctx, "const float4 %s = float4(%s, %s, %s, %s); (void) %s;", varname, val0, val1, val2, val3, varname); pop_output(ctx); } // emit_METAL_DEF EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(TEXREG2RGB) // !!! FIXME EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(TEXDP3TEX) // !!! FIXME EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(TEXM3X2DEPTH) // !!! FIXME EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(TEXDP3) // !!! FIXME void emit_METAL_TEXM3X3(Context *ctx) { if (ctx->texm3x3pad_src1 == -1) return; char dst[64]; char src0[64]; char src1[64]; char src2[64]; char src3[64]; char src4[64]; char code[512]; ctx->metal_need_header_geometric = 1; // !!! FIXME: this code counts on the register not having swizzles, etc. get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_dst0, src0, sizeof (src0)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_src0, src1, sizeof (src1)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_dst1, src2, sizeof (src2)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->texm3x3pad_src1, src3, sizeof (src3)); get_METAL_varname_in_buf(ctx, REG_TYPE_TEXTURE, ctx->source_args[0].regnum, src4, sizeof (src4)); get_METAL_destarg_varname(ctx, dst, sizeof (dst)); make_METAL_destarg_assign(ctx, code, sizeof (code), "float4(dot(%s.xyz, %s.xyz), dot(%s.xyz, %s.xyz), dot(%s.xyz, %s.xyz), 1.0)", src0, src1, src2, src3, dst, src4); output_line(ctx, "%s", code); } // emit_METAL_TEXM3X3 EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(TEXDEPTH) // !!! FIXME void emit_METAL_CMP(Context *ctx) { emit_METAL_comparison_operations(ctx, ">= 0.0"); } // emit_METAL_CMP EMIT_METAL_OPCODE_UNIMPLEMENTED_FUNC(BEM) // !!! FIXME void emit_METAL_DP2ADD(Context *ctx) { char src0[64]; make_METAL_srcarg_string_vec2(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_vec2(ctx, 1, src1, sizeof (src1)); char src2[64]; make_METAL_srcarg_string_scalar(ctx, 2, src2, sizeof (src2)); char extra[64]; snprintf(extra, sizeof (extra), " + %s", src2); emit_METAL_dotprod(ctx, src0, src1, extra); } // emit_METAL_DP2ADD void emit_METAL_DSX(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_graphics = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "dfdx(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_DSX void emit_METAL_DSY(Context *ctx) { char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char code[128]; ctx->metal_need_header_graphics = 1; make_METAL_destarg_assign(ctx, code, sizeof (code), "dfdy(%s)", src0); output_line(ctx, "%s", code); } // emit_METAL_DSY void emit_METAL_TEXLDD(Context *ctx) { metal_texld(ctx, 1); } // emit_METAL_TEXLDD void emit_METAL_SETP(Context *ctx) { const int vecsize = vecsize_from_writemask(ctx->dest_arg.writemask); char src0[64]; make_METAL_srcarg_string_masked(ctx, 0, src0, sizeof (src0)); char src1[64]; make_METAL_srcarg_string_masked(ctx, 1, src1, sizeof (src1)); char code[128]; // destination is always predicate register (which is type bvec4). const char *comp = (vecsize == 1) ? get_METAL_comparison_string_scalar(ctx) : get_METAL_comparison_string_vector(ctx); make_METAL_destarg_assign(ctx, code, sizeof (code), "(%s %s %s)", src0, comp, src1); output_line(ctx, "%s", code); } // emit_METAL_SETP void emit_METAL_TEXLDL(Context *ctx) { // !!! FIXME: The spec says we can't use GLSL's texture*Lod() built-ins // !!! FIXME: from fragment shaders for some inexplicable reason. // !!! FIXME: Maybe Metal can do it, but I haven't looked into it yet. emit_METAL_TEXLD(ctx); } // emit_METAL_TEXLDL void emit_METAL_BREAKP(Context *ctx) { char src0[64]; make_METAL_srcarg_string_scalar(ctx, 0, src0, sizeof (src0)); output_line(ctx, "if (%s) { break; }", src0); } // emit_METAL_BREAKP void emit_METAL_RESERVED(Context *ctx) { // do nothing; fails in the state machine. } // emit_METAL_RESERVED #endif // SUPPORT_PROFILE_METAL #pragma GCC visibility pop