mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-29 22:41:03 +00:00
9a6f4121f4
This fixes the errors: validate_ray_tracing_reorder.cpp:25:49: error: 'numeric_limits' is not a member of 'std'
626 lines
25 KiB
C++
626 lines
25 KiB
C++
// Copyright (c) 2022 The Khronos Group Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// Validates ray tracing instructions from SPV_NV_shader_execution_reorder
|
|
|
|
#include "source/opcode.h"
|
|
#include "source/val/instruction.h"
|
|
#include "source/val/validate.h"
|
|
#include "source/val/validation_state.h"
|
|
|
|
#include <limits>
|
|
|
|
namespace spvtools {
|
|
namespace val {
|
|
|
|
static const uint32_t KRayParamInvalidId = std::numeric_limits<uint32_t>::max();
|
|
|
|
spv_result_t ValidateHitObjectPointer(ValidationState_t& _,
|
|
const Instruction* inst,
|
|
uint32_t hit_object_index) {
|
|
const uint32_t hit_object_id = inst->GetOperandAs<uint32_t>(hit_object_index);
|
|
auto variable = _.FindDef(hit_object_id);
|
|
const auto var_opcode = variable->opcode();
|
|
if (!variable || (var_opcode != spv::Op::OpVariable &&
|
|
var_opcode != spv::Op::OpFunctionParameter &&
|
|
var_opcode != spv::Op::OpAccessChain)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hit Object must be a memory object declaration";
|
|
}
|
|
auto pointer = _.FindDef(variable->GetOperandAs<uint32_t>(0));
|
|
if (!pointer || pointer->opcode() != spv::Op::OpTypePointer) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hit Object must be a pointer";
|
|
}
|
|
auto type = _.FindDef(pointer->GetOperandAs<uint32_t>(2));
|
|
if (!type || type->opcode() != spv::Op::OpTypeHitObjectNV) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Type must be OpTypeHitObjectNV";
|
|
}
|
|
return SPV_SUCCESS;
|
|
}
|
|
|
|
spv_result_t ValidateHitObjectInstructionCommonParameters(
|
|
ValidationState_t& _, const Instruction* inst,
|
|
uint32_t acceleration_struct_index, uint32_t instance_id_index,
|
|
uint32_t primtive_id_index, uint32_t geometry_index,
|
|
uint32_t ray_flags_index, uint32_t cull_mask_index, uint32_t hit_kind_index,
|
|
uint32_t sbt_index, uint32_t sbt_offset_index, uint32_t sbt_stride_index,
|
|
uint32_t sbt_record_offset_index, uint32_t sbt_record_stride_index,
|
|
uint32_t miss_index, uint32_t ray_origin_index, uint32_t ray_tmin_index,
|
|
uint32_t ray_direction_index, uint32_t ray_tmax_index,
|
|
uint32_t payload_index, uint32_t hit_object_attr_index) {
|
|
auto isValidId = [](uint32_t spvid) { return spvid < KRayParamInvalidId; };
|
|
if (isValidId(acceleration_struct_index) &&
|
|
_.GetIdOpcode(_.GetOperandTypeId(inst, acceleration_struct_index)) !=
|
|
spv::Op::OpTypeAccelerationStructureKHR) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected Acceleration Structure to be of type "
|
|
"OpTypeAccelerationStructureKHR";
|
|
}
|
|
|
|
if (isValidId(instance_id_index)) {
|
|
const uint32_t instance_id = _.GetOperandTypeId(inst, instance_id_index);
|
|
if (!_.IsIntScalarType(instance_id) || _.GetBitWidth(instance_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Instance Id must be a 32-bit int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(primtive_id_index)) {
|
|
const uint32_t primitive_id = _.GetOperandTypeId(inst, primtive_id_index);
|
|
if (!_.IsIntScalarType(primitive_id) || _.GetBitWidth(primitive_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Primitive Id must be a 32-bit int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(geometry_index)) {
|
|
const uint32_t geometry_index_id = _.GetOperandTypeId(inst, geometry_index);
|
|
if (!_.IsIntScalarType(geometry_index_id) ||
|
|
_.GetBitWidth(geometry_index_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Geometry Index must be a 32-bit int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(miss_index)) {
|
|
const uint32_t miss_index_id = _.GetOperandTypeId(inst, miss_index);
|
|
if (!_.IsUnsignedIntScalarType(miss_index_id) ||
|
|
_.GetBitWidth(miss_index_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Miss Index must be a 32-bit int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(cull_mask_index)) {
|
|
const uint32_t cull_mask_id = _.GetOperandTypeId(inst, cull_mask_index);
|
|
if (!_.IsUnsignedIntScalarType(cull_mask_id) ||
|
|
_.GetBitWidth(cull_mask_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Cull mask must be a 32-bit int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(sbt_index)) {
|
|
const uint32_t sbt_index_id = _.GetOperandTypeId(inst, sbt_index);
|
|
if (!_.IsUnsignedIntScalarType(sbt_index_id) ||
|
|
_.GetBitWidth(sbt_index_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "SBT Index must be a 32-bit unsigned int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(sbt_offset_index)) {
|
|
const uint32_t sbt_offset_id = _.GetOperandTypeId(inst, sbt_offset_index);
|
|
if (!_.IsUnsignedIntScalarType(sbt_offset_id) ||
|
|
_.GetBitWidth(sbt_offset_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "SBT Offset must be a 32-bit unsigned int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(sbt_stride_index)) {
|
|
const uint32_t sbt_stride_index_id =
|
|
_.GetOperandTypeId(inst, sbt_stride_index);
|
|
if (!_.IsUnsignedIntScalarType(sbt_stride_index_id) ||
|
|
_.GetBitWidth(sbt_stride_index_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "SBT Stride must be a 32-bit unsigned int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(sbt_record_offset_index)) {
|
|
const uint32_t sbt_record_offset_index_id =
|
|
_.GetOperandTypeId(inst, sbt_record_offset_index);
|
|
if (!_.IsUnsignedIntScalarType(sbt_record_offset_index_id) ||
|
|
_.GetBitWidth(sbt_record_offset_index_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "SBT record offset must be a 32-bit unsigned int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(sbt_record_stride_index)) {
|
|
const uint32_t sbt_record_stride_index_id =
|
|
_.GetOperandTypeId(inst, sbt_record_stride_index);
|
|
if (!_.IsUnsignedIntScalarType(sbt_record_stride_index_id) ||
|
|
_.GetBitWidth(sbt_record_stride_index_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "SBT record stride must be a 32-bit unsigned int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(ray_origin_index)) {
|
|
const uint32_t ray_origin_id = _.GetOperandTypeId(inst, ray_origin_index);
|
|
if (!_.IsFloatVectorType(ray_origin_id) ||
|
|
_.GetDimension(ray_origin_id) != 3 ||
|
|
_.GetBitWidth(ray_origin_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray Origin must be a 32-bit float 3-component vector";
|
|
}
|
|
}
|
|
|
|
if (isValidId(ray_tmin_index)) {
|
|
const uint32_t ray_tmin_id = _.GetOperandTypeId(inst, ray_tmin_index);
|
|
if (!_.IsFloatScalarType(ray_tmin_id) || _.GetBitWidth(ray_tmin_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray TMin must be a 32-bit float scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(ray_direction_index)) {
|
|
const uint32_t ray_direction_id =
|
|
_.GetOperandTypeId(inst, ray_direction_index);
|
|
if (!_.IsFloatVectorType(ray_direction_id) ||
|
|
_.GetDimension(ray_direction_id) != 3 ||
|
|
_.GetBitWidth(ray_direction_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray Direction must be a 32-bit float 3-component vector";
|
|
}
|
|
}
|
|
|
|
if (isValidId(ray_tmax_index)) {
|
|
const uint32_t ray_tmax_id = _.GetOperandTypeId(inst, ray_tmax_index);
|
|
if (!_.IsFloatScalarType(ray_tmax_id) || _.GetBitWidth(ray_tmax_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray TMax must be a 32-bit float scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(ray_flags_index)) {
|
|
const uint32_t ray_flags_id = _.GetOperandTypeId(inst, ray_flags_index);
|
|
if (!_.IsIntScalarType(ray_flags_id) || _.GetBitWidth(ray_flags_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray Flags must be a 32-bit int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(payload_index)) {
|
|
const uint32_t payload_id = inst->GetOperandAs<uint32_t>(payload_index);
|
|
auto variable = _.FindDef(payload_id);
|
|
const auto var_opcode = variable->opcode();
|
|
if (!variable || var_opcode != spv::Op::OpVariable ||
|
|
(variable->GetOperandAs<spv::StorageClass>(2) !=
|
|
spv::StorageClass::RayPayloadKHR &&
|
|
variable->GetOperandAs<spv::StorageClass>(2) !=
|
|
spv::StorageClass::IncomingRayPayloadKHR)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "payload must be a OpVariable of storage "
|
|
"class RayPayloadKHR or IncomingRayPayloadKHR";
|
|
}
|
|
}
|
|
|
|
if (isValidId(hit_kind_index)) {
|
|
const uint32_t hit_kind_id = _.GetOperandTypeId(inst, hit_kind_index);
|
|
if (!_.IsUnsignedIntScalarType(hit_kind_id) ||
|
|
_.GetBitWidth(hit_kind_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hit Kind must be a 32-bit unsigned int scalar";
|
|
}
|
|
}
|
|
|
|
if (isValidId(hit_object_attr_index)) {
|
|
const uint32_t hit_object_attr_id =
|
|
inst->GetOperandAs<uint32_t>(hit_object_attr_index);
|
|
auto variable = _.FindDef(hit_object_attr_id);
|
|
const auto var_opcode = variable->opcode();
|
|
if (!variable || var_opcode != spv::Op::OpVariable ||
|
|
(variable->GetOperandAs<spv::StorageClass>(2)) !=
|
|
spv::StorageClass::HitObjectAttributeNV) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hit Object Attributes id must be a OpVariable of storage "
|
|
"class HitObjectAttributeNV";
|
|
}
|
|
}
|
|
|
|
return SPV_SUCCESS;
|
|
}
|
|
|
|
spv_result_t RayReorderNVPass(ValidationState_t& _, const Instruction* inst) {
|
|
const spv::Op opcode = inst->opcode();
|
|
const uint32_t result_type = inst->type_id();
|
|
|
|
auto RegisterOpcodeForValidModel = [](ValidationState_t& vs,
|
|
const Instruction* rtinst) {
|
|
std::string opcode_name = spvOpcodeString(rtinst->opcode());
|
|
vs.function(rtinst->function()->id())
|
|
->RegisterExecutionModelLimitation(
|
|
[opcode_name](spv::ExecutionModel model, std::string* message) {
|
|
if (model != spv::ExecutionModel::RayGenerationKHR &&
|
|
model != spv::ExecutionModel::ClosestHitKHR &&
|
|
model != spv::ExecutionModel::MissKHR) {
|
|
if (message) {
|
|
*message = opcode_name +
|
|
" requires RayGenerationKHR, ClosestHitKHR and "
|
|
"MissKHR execution models";
|
|
}
|
|
return false;
|
|
}
|
|
return true;
|
|
});
|
|
return;
|
|
};
|
|
|
|
switch (opcode) {
|
|
case spv::Op::OpHitObjectIsMissNV:
|
|
case spv::Op::OpHitObjectIsHitNV:
|
|
case spv::Op::OpHitObjectIsEmptyNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (!_.IsBoolScalarType(result_type)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "expected Result Type to be bool scalar type";
|
|
}
|
|
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 2)) return error;
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectGetShaderRecordBufferHandleNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 2)) return error;
|
|
|
|
if (!_.IsIntVectorType(result_type) ||
|
|
(_.GetDimension(result_type) != 2) ||
|
|
(_.GetBitWidth(result_type) != 32))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected 32-bit integer type 2-component vector as Result "
|
|
"Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectGetHitKindNV:
|
|
case spv::Op::OpHitObjectGetPrimitiveIndexNV:
|
|
case spv::Op::OpHitObjectGetGeometryIndexNV:
|
|
case spv::Op::OpHitObjectGetInstanceIdNV:
|
|
case spv::Op::OpHitObjectGetInstanceCustomIndexNV:
|
|
case spv::Op::OpHitObjectGetShaderBindingTableRecordIndexNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 2)) return error;
|
|
|
|
if (!_.IsIntScalarType(result_type) || !_.GetBitWidth(result_type))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected 32-bit integer type scalar as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectGetCurrentTimeNV:
|
|
case spv::Op::OpHitObjectGetRayTMaxNV:
|
|
case spv::Op::OpHitObjectGetRayTMinNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 2)) return error;
|
|
|
|
if (!_.IsFloatScalarType(result_type) || _.GetBitWidth(result_type) != 32)
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected 32-bit floating-point type scalar as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectGetObjectToWorldNV:
|
|
case spv::Op::OpHitObjectGetWorldToObjectNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 2)) return error;
|
|
|
|
uint32_t num_rows = 0;
|
|
uint32_t num_cols = 0;
|
|
uint32_t col_type = 0;
|
|
uint32_t component_type = 0;
|
|
|
|
if (!_.GetMatrixTypeInfo(result_type, &num_rows, &num_cols, &col_type,
|
|
&component_type)) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "expected matrix type as Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
if (num_cols != 4) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "expected Result Type matrix to have a Column Count of 4"
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
|
|
if (!_.IsFloatScalarType(component_type) ||
|
|
_.GetBitWidth(result_type) != 32 || num_rows != 3) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "expected Result Type matrix to have a Column Type of "
|
|
"3-component 32-bit float vectors: "
|
|
<< spvOpcodeString(opcode);
|
|
}
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectGetObjectRayOriginNV:
|
|
case spv::Op::OpHitObjectGetObjectRayDirectionNV:
|
|
case spv::Op::OpHitObjectGetWorldRayDirectionNV:
|
|
case spv::Op::OpHitObjectGetWorldRayOriginNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 2)) return error;
|
|
|
|
if (!_.IsFloatVectorType(result_type) ||
|
|
(_.GetDimension(result_type) != 3) ||
|
|
(_.GetBitWidth(result_type) != 32))
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Expected 32-bit floating-point type 3-component vector as "
|
|
"Result Type: "
|
|
<< spvOpcodeString(opcode);
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectGetAttributesNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
|
|
const uint32_t hit_object_attr_id = inst->GetOperandAs<uint32_t>(1);
|
|
auto variable = _.FindDef(hit_object_attr_id);
|
|
const auto var_opcode = variable->opcode();
|
|
if (!variable || var_opcode != spv::Op::OpVariable ||
|
|
variable->GetOperandAs<spv::StorageClass>(2) !=
|
|
spv::StorageClass::HitObjectAttributeNV) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hit Object Attributes id must be a OpVariable of storage "
|
|
"class HitObjectAttributeNV";
|
|
}
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectExecuteShaderNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
|
|
const uint32_t hit_object_attr_id = inst->GetOperandAs<uint32_t>(1);
|
|
auto variable = _.FindDef(hit_object_attr_id);
|
|
const auto var_opcode = variable->opcode();
|
|
if (!variable || var_opcode != spv::Op::OpVariable ||
|
|
(variable->GetOperandAs<spv::StorageClass>(2)) !=
|
|
spv::StorageClass::RayPayloadKHR) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hit Object Attributes id must be a OpVariable of storage "
|
|
"class RayPayloadKHR";
|
|
}
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectRecordEmptyNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectRecordMissNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
|
|
const uint32_t miss_index = _.GetOperandTypeId(inst, 1);
|
|
if (!_.IsUnsignedIntScalarType(miss_index) ||
|
|
_.GetBitWidth(miss_index) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Miss Index must be a 32-bit int scalar";
|
|
}
|
|
|
|
const uint32_t ray_origin = _.GetOperandTypeId(inst, 2);
|
|
if (!_.IsFloatVectorType(ray_origin) || _.GetDimension(ray_origin) != 3 ||
|
|
_.GetBitWidth(ray_origin) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray Origin must be a 32-bit float 3-component vector";
|
|
}
|
|
|
|
const uint32_t ray_tmin = _.GetOperandTypeId(inst, 3);
|
|
if (!_.IsFloatScalarType(ray_tmin) || _.GetBitWidth(ray_tmin) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray TMin must be a 32-bit float scalar";
|
|
}
|
|
|
|
const uint32_t ray_direction = _.GetOperandTypeId(inst, 4);
|
|
if (!_.IsFloatVectorType(ray_direction) ||
|
|
_.GetDimension(ray_direction) != 3 ||
|
|
_.GetBitWidth(ray_direction) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray Direction must be a 32-bit float 3-component vector";
|
|
}
|
|
|
|
const uint32_t ray_tmax = _.GetOperandTypeId(inst, 5);
|
|
if (!_.IsFloatScalarType(ray_tmax) || _.GetBitWidth(ray_tmax) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Ray TMax must be a 32-bit float scalar";
|
|
}
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectRecordHitWithIndexNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
|
|
if (auto error = ValidateHitObjectInstructionCommonParameters(
|
|
_, inst, 1 /* Acceleration Struct */, 2 /* Instance Id */,
|
|
3 /* Primtive Id */, 4 /* Geometry Index */,
|
|
KRayParamInvalidId /* Ray Flags */,
|
|
KRayParamInvalidId /* Cull Mask */, 5 /* Hit Kind*/,
|
|
6 /* SBT index */, KRayParamInvalidId /* SBT Offset */,
|
|
KRayParamInvalidId /* SBT Stride */,
|
|
KRayParamInvalidId /* SBT Record Offset */,
|
|
KRayParamInvalidId /* SBT Record Stride */,
|
|
KRayParamInvalidId /* Miss Index */, 7 /* Ray Origin */,
|
|
8 /* Ray TMin */, 9 /* Ray Direction */, 10 /* Ray TMax */,
|
|
KRayParamInvalidId /* Payload */, 11 /* Hit Object Attribute */))
|
|
return error;
|
|
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectRecordHitNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
|
|
if (auto error = ValidateHitObjectInstructionCommonParameters(
|
|
_, inst, 1 /* Acceleration Struct */, 2 /* Instance Id */,
|
|
3 /* Primtive Id */, 4 /* Geometry Index */,
|
|
KRayParamInvalidId /* Ray Flags */,
|
|
KRayParamInvalidId /* Cull Mask */, 5 /* Hit Kind*/,
|
|
KRayParamInvalidId /* SBT index */,
|
|
KRayParamInvalidId /* SBT Offset */,
|
|
KRayParamInvalidId /* SBT Stride */, 6 /* SBT Record Offset */,
|
|
7 /* SBT Record Stride */, KRayParamInvalidId /* Miss Index */,
|
|
8 /* Ray Origin */, 9 /* Ray TMin */, 10 /* Ray Direction */,
|
|
11 /* Ray TMax */, KRayParamInvalidId /* Payload */,
|
|
12 /* Hit Object Attribute */))
|
|
return error;
|
|
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectTraceRayMotionNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
|
|
if (auto error = ValidateHitObjectInstructionCommonParameters(
|
|
_, inst, 1 /* Acceleration Struct */,
|
|
KRayParamInvalidId /* Instance Id */,
|
|
KRayParamInvalidId /* Primtive Id */,
|
|
KRayParamInvalidId /* Geometry Index */, 2 /* Ray Flags */,
|
|
3 /* Cull Mask */, KRayParamInvalidId /* Hit Kind*/,
|
|
KRayParamInvalidId /* SBT index */, 4 /* SBT Offset */,
|
|
5 /* SBT Stride */, KRayParamInvalidId /* SBT Record Offset */,
|
|
KRayParamInvalidId /* SBT Record Stride */, 6 /* Miss Index */,
|
|
7 /* Ray Origin */, 8 /* Ray TMin */, 9 /* Ray Direction */,
|
|
10 /* Ray TMax */, 12 /* Payload */,
|
|
KRayParamInvalidId /* Hit Object Attribute */))
|
|
return error;
|
|
// Current Time
|
|
const uint32_t current_time_id = _.GetOperandTypeId(inst, 11);
|
|
if (!_.IsFloatScalarType(current_time_id) ||
|
|
_.GetBitWidth(current_time_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Current Times must be a 32-bit float scalar type";
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpHitObjectTraceRayNV: {
|
|
RegisterOpcodeForValidModel(_, inst);
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
|
|
if (auto error = ValidateHitObjectInstructionCommonParameters(
|
|
_, inst, 1 /* Acceleration Struct */,
|
|
KRayParamInvalidId /* Instance Id */,
|
|
KRayParamInvalidId /* Primtive Id */,
|
|
KRayParamInvalidId /* Geometry Index */, 2 /* Ray Flags */,
|
|
3 /* Cull Mask */, KRayParamInvalidId /* Hit Kind*/,
|
|
KRayParamInvalidId /* SBT index */, 4 /* SBT Offset */,
|
|
5 /* SBT Stride */, KRayParamInvalidId /* SBT Record Offset */,
|
|
KRayParamInvalidId /* SBT Record Stride */, 6 /* Miss Index */,
|
|
7 /* Ray Origin */, 8 /* Ray TMin */, 9 /* Ray Direction */,
|
|
10 /* Ray TMax */, 11 /* Payload */,
|
|
KRayParamInvalidId /* Hit Object Attribute */))
|
|
return error;
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpReorderThreadWithHitObjectNV: {
|
|
std::string opcode_name = spvOpcodeString(inst->opcode());
|
|
_.function(inst->function()->id())
|
|
->RegisterExecutionModelLimitation(
|
|
[opcode_name](spv::ExecutionModel model, std::string* message) {
|
|
if (model != spv::ExecutionModel::RayGenerationKHR) {
|
|
if (message) {
|
|
*message = opcode_name +
|
|
" requires RayGenerationKHR execution model";
|
|
}
|
|
return false;
|
|
}
|
|
return true;
|
|
});
|
|
|
|
if (auto error = ValidateHitObjectPointer(_, inst, 0)) return error;
|
|
|
|
if (inst->operands().size() > 1) {
|
|
if (inst->operands().size() != 3) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hint and Bits are optional together i.e "
|
|
<< " Either both Hint and Bits should be provided or neither.";
|
|
}
|
|
|
|
// Validate the optional opreands Hint and Bits
|
|
const uint32_t hint_id = _.GetOperandTypeId(inst, 1);
|
|
if (!_.IsIntScalarType(hint_id) || _.GetBitWidth(hint_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hint must be a 32-bit int scalar";
|
|
}
|
|
const uint32_t bits_id = _.GetOperandTypeId(inst, 2);
|
|
if (!_.IsIntScalarType(bits_id) || _.GetBitWidth(bits_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "bits must be a 32-bit int scalar";
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
case spv::Op::OpReorderThreadWithHintNV: {
|
|
std::string opcode_name = spvOpcodeString(inst->opcode());
|
|
_.function(inst->function()->id())
|
|
->RegisterExecutionModelLimitation(
|
|
[opcode_name](spv::ExecutionModel model, std::string* message) {
|
|
if (model != spv::ExecutionModel::RayGenerationKHR) {
|
|
if (message) {
|
|
*message = opcode_name +
|
|
" requires RayGenerationKHR execution model";
|
|
}
|
|
return false;
|
|
}
|
|
return true;
|
|
});
|
|
|
|
const uint32_t hint_id = _.GetOperandTypeId(inst, 0);
|
|
if (!_.IsIntScalarType(hint_id) || _.GetBitWidth(hint_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "Hint must be a 32-bit int scalar";
|
|
}
|
|
|
|
const uint32_t bits_id = _.GetOperandTypeId(inst, 1);
|
|
if (!_.IsIntScalarType(bits_id) || _.GetBitWidth(bits_id) != 32) {
|
|
return _.diag(SPV_ERROR_INVALID_DATA, inst)
|
|
<< "bits must be a 32-bit int scalar";
|
|
}
|
|
}
|
|
|
|
default:
|
|
break;
|
|
}
|
|
return SPV_SUCCESS;
|
|
}
|
|
} // namespace val
|
|
} // namespace spvtools
|