{ "textureSize", "", "", 0, { }, CAP_IMAGE_QUERY, &SpirVGenerator::visit_builtin_texture_query },
{ "textureQueryLod", "", "", 0, { }, CAP_IMAGE_QUERY, &SpirVGenerator::visit_builtin_texture_query },
{ "textureQueryLevels", "", "", 0, { }, CAP_IMAGE_QUERY, &SpirVGenerator::visit_builtin_texture_query },
+ { "textureSamples", "", "", 0, { }, CAP_IMAGE_QUERY, &SpirVGenerator::visit_builtin_texture_query },
{ "texture", "", "", 0, { }, 0, &SpirVGenerator::visit_builtin_texture },
{ "textureLod", "", "", 0, { }, 0, &SpirVGenerator::visit_builtin_texture },
- { "texelFetch", "", "", 0, { }, 0, &SpirVGenerator::visit_builtin_texel_fetch },
+ { "texelFetch", "", "", 0, { }, 0, &SpirVGenerator::visit_builtin_texture_fetch },
+ { "imageSize", "", "", 0, { }, CAP_IMAGE_QUERY, &SpirVGenerator::visit_builtin_texture_query },
+ { "imageSamples", "", "", 0, { }, CAP_IMAGE_QUERY, &SpirVGenerator::visit_builtin_texture_query },
+ { "imageLoad", "", "", 0, { }, 0, &SpirVGenerator::visit_builtin_texture_fetch },
+ { "imageStore", "", "", 0, { }, 0, &SpirVGenerator::visit_builtin_texture_store },
{ "EmitVertex", "", "", OP_EMIT_VERTEX, { }, 0, 0 },
{ "EndPrimitive", "", "", OP_END_PRIMITIVE, { }, 0, 0 },
{ "dFdx", "f", "", OP_DP_DX, { 1 }, 0, 0 },
return BUILTIN_SAMPLE_POSITION;
else if(name=="gl_FragDepth")
return BUILTIN_FRAG_DEPTH;
+ else if(name=="gl_NumWorkGroups")
+ return BUILTIN_NUM_WORKGROUPS;
+ else if(name=="gl_WorkGroupSize")
+ return BUILTIN_WORKGROUP_SIZE;
+ else if(name=="gl_WorkGroupID")
+ return BUILTIN_WORKGROUP_ID;
+ else if(name=="gl_LocalInvocationID")
+ return BUILTIN_LOCAL_INVOCATION_ID;
+ else if(name=="gl_GlobalInvocationID")
+ return BUILTIN_GLOBAL_INVOCATION_ID;
+ else if(name=="gl_LocalInvocationIndex")
+ return BUILTIN_LOCAL_INVOCATION_INDEX;
else
throw invalid_argument("SpirVGenerator::get_builtin_semantic");
}
+SpirVFormat SpirVGenerator::get_format(const std::string &name)
+{
+ if(name.empty())
+ return FORMAT_UNKNOWN;
+ else if(name=="rgba32f")
+ return FORMAT_RGBA32F;
+ else if(name=="rgba16f")
+ return FORMAT_RGBA16F;
+ else if(name=="r32f")
+ return FORMAT_R32F;
+ else if(name=="rgba8")
+ return FORMAT_RGBA8;
+ else if(name=="rgba8_snorm")
+ return FORMAT_RGBA8_SNORM;
+ else if(name=="rg32f")
+ return FORMAT_RG32F;
+ else if(name=="rg16f")
+ return FORMAT_RG16F;
+ else if(name=="r16f")
+ return FORMAT_R16F;
+ else if(name=="rgba16")
+ return FORMAT_RGBA16;
+ else if(name=="rg16")
+ return FORMAT_RG16;
+ else if(name=="rg8")
+ return FORMAT_RG8;
+ else if(name=="r16")
+ return FORMAT_RG16;
+ else if(name=="r8")
+ return FORMAT_RG8;
+ else if(name=="rgba16_snorm")
+ return FORMAT_RGBA16_SNORM;
+ else if(name=="rg16_snorm")
+ return FORMAT_RG16_SNORM;
+ else if(name=="rg8_snorm")
+ return FORMAT_RG8_SNORM;
+ else if(name=="r16_snorm")
+ return FORMAT_RG16_SNORM;
+ else if(name=="r8_snorm")
+ return FORMAT_RG8_SNORM;
+ else
+ throw invalid_argument("SpirVGenerator::get_format");
+}
+
void SpirVGenerator::use_capability(Capability cap)
{
if(used_capabilities.count(cap))
writer.write_op(content.function_body, OP_SELECTION_MERGE, merge_block_id, 0); // Selection control (none)
writer.write_op(content.function_body, OP_BRANCH_CONDITIONAL, condition_id, true_label_id, false_label_id);
+ std::map<const VariableDeclaration *, Id> saved_load_ids = variable_load_ids;
+
writer.write_op_label(true_label_id);
ternary.true_expr->visit(*this);
Id true_result_id = r_expression_result_id;
true_label_id = writer.get_current_block();
writer.write_op(content.function_body, OP_BRANCH, merge_block_id);
+ swap(saved_load_ids, variable_load_ids);
writer.write_op_label(false_label_id);
ternary.false_expr->visit(*this);
Id false_result_id = r_expression_result_id;
false_label_id = writer.get_current_block();
writer.write_op_label(merge_block_id);
+ prune_loads(true_label_id);
r_expression_result_id = begin_expression(OP_PHI, get_id(*ternary.type), 4);
writer.write(true_result_id);
writer.write(true_label_id);
Opcode opcode;
if(call.name=="textureSize")
opcode = OP_IMAGE_QUERY_SIZE_LOD;
+ else if(call.name=="imageSize")
+ opcode = OP_IMAGE_QUERY_SIZE;
else if(call.name=="textureQueryLod")
opcode = OP_IMAGE_QUERY_LOD;
else if(call.name=="textureQueryLevels")
opcode = OP_IMAGE_QUERY_LEVELS;
+ else if(call.name=="textureSamples" || call.name=="imageSamples")
+ opcode = OP_IMAGE_QUERY_SAMPLES;
else
throw internal_error("invalid texture query call");
ImageTypeDeclaration &image_arg0 = dynamic_cast<ImageTypeDeclaration &>(*call.arguments[0]->type);
Id image_id;
- if(image_arg0.sampled)
+ if(image_arg0.sampled && opcode!=OP_IMAGE_QUERY_LOD)
{
Id image_type_id = get_item(image_type_ids, get_id(image_arg0));
image_id = write_expression(OP_IMAGE, image_type_id, argument_ids[0]);
end_expression(opcode);
}
-void SpirVGenerator::visit_builtin_texel_fetch(FunctionCall &call, const vector<Id> &argument_ids)
+void SpirVGenerator::visit_builtin_texture_fetch(FunctionCall &call, const vector<Id> &argument_ids)
+{
+ const ImageTypeDeclaration &image = dynamic_cast<const ImageTypeDeclaration &>(*call.arguments[0]->type);
+
+ Opcode opcode;
+ if(call.name=="texelFetch")
+ opcode = OP_IMAGE_FETCH;
+ else if(call.name=="imageLoad")
+ opcode = OP_IMAGE_READ;
+ else
+ throw internal_error("invalid texture fetch call");
+
+ bool need_sample = image.multisample;
+ bool need_lod = (opcode==OP_IMAGE_FETCH && !need_sample);
+
+ if(argument_ids.size()!=2U+need_sample+need_lod)
+ throw internal_error("invalid texture fetch call");
+
+ r_expression_result_id = begin_expression(opcode, get_id(*call.type), 2+(need_lod|need_sample)+need_lod+need_sample);
+ for(unsigned i=0; i<2; ++i)
+ writer.write(argument_ids[i]);
+ if(need_lod || need_sample)
+ {
+ writer.write(need_lod*0x02 | need_sample*0x40);
+ writer.write(argument_ids.back());
+ }
+ end_expression(opcode);
+}
+
+void SpirVGenerator::visit_builtin_texture_store(FunctionCall &call, const vector<Id> &argument_ids)
{
if(argument_ids.size()!=3)
- throw internal_error("invalid texelFetch call");
+ throw internal_error("invalid texture store call");
+
+ const ImageTypeDeclaration &image = dynamic_cast<const ImageTypeDeclaration &>(*call.arguments[0]->type);
- r_expression_result_id = begin_expression(OP_IMAGE_FETCH, get_id(*call.type), 4);
+ begin_expression(OP_IMAGE_WRITE, get_id(*call.type), 3+image.multisample*2);
for(unsigned i=0; i<2; ++i)
writer.write(argument_ids[i]);
- writer.write(2); // Lod
writer.write(argument_ids.back());
- end_expression(OP_IMAGE_FETCH);
+ if(image.multisample)
+ {
+ writer.write(0x40); // Sample
+ writer.write(argument_ids[2]);
+ }
+ end_expression(OP_IMAGE_WRITE);
+
+ r_expression_result_id = 0;
}
void SpirVGenerator::visit_builtin_interpolate(FunctionCall &call, const vector<Id> &argument_ids)
writer.write(image.dimensions-1);
writer.write(image.shadow);
writer.write(image.array);
- writer.write(false); // Multisample
+ writer.write(image.multisample);
writer.write(image.sampled ? 1 : 2);
- writer.write(0); // Format (unknown)
+ writer.write(get_format(image.format));
writer.end_op(OP_TYPE_IMAGE);
if(image.sampled)
use_capability(image.sampled ? CAP_SAMPLED_1D : CAP_IMAGE_1D);
else if(image.dimensions==ImageTypeDeclaration::CUBE && image.array)
use_capability(image.sampled ? CAP_SAMPLED_CUBE_ARRAY : CAP_IMAGE_CUBE_ARRAY);
+
+ if(image.multisample && !image.sampled)
+ use_capability(CAP_STORAGE_IMAGE_MULTISAMPLE);
}
void SpirVGenerator::visit(StructDeclaration &strct)
BuiltinSemantic semantic = get_builtin_semantic(var.name);
writer.write_op_decorate(var_id, DECO_BUILTIN, semantic);
}
+ if(var.sampling=="flat")
+ writer.write_op_decorate(var_id, DECO_FLAT);
+ if(var.sampling=="centroid")
+ writer.write_op_decorate(var_id, DECO_CENTROID);
if(init_id && current_function)
{
case Stage::VERTEX: writer.write(0); break;
case Stage::GEOMETRY: writer.write(3); break;
case Stage::FRAGMENT: writer.write(4); break;
+ case Stage::COMPUTE: writer.write(5); break;
default: throw internal_error("unknown stage");
}
writer.write(func_id);
writer.write_op(content.exec_modes, OP_EXECUTION_MODE, func_id, EXEC_INVOCATIONS, 1);
}
+ unsigned local_size[3] = { 0, 1, 1 };
+
for(const InterfaceLayout *i: interface_layouts)
{
for(const Layout::Qualifier &q: i->layout.qualifiers)
writer.write_op(content.exec_modes, OP_EXECUTION_MODE, func_id, EXEC_OUTPUT_TRIANGLE_STRIP);
else if(q.name=="max_vertices")
writer.write_op(content.exec_modes, OP_EXECUTION_MODE, func_id, EXEC_OUTPUT_VERTICES, q.value);
+ else if(q.name=="local_size_x")
+ local_size[0] = q.value;
+ else if(q.name=="local_size_y")
+ local_size[1] = q.value;
+ else if(q.name=="local_size_z")
+ local_size[2] = q.value;
}
}
+
+ if(stage->type==Stage::COMPUTE && local_size[0])
+ {
+ writer.begin_op(content.exec_modes, OP_EXECUTION_MODE);
+ writer.write(func_id);
+ writer.write(EXEC_LOCAL_SIZE);
+ for(unsigned j=0; j<3; ++j)
+ writer.write(local_size[j]);
+ writer.end_op(OP_EXECUTION_MODE);
+ }
}
void SpirVGenerator::visit(FunctionDeclaration &func)
projects / libs / gl.git / blobdiff