Implemented Image Instructions
This commit is contained in:
parent
bdf538bd64
commit
0318449d65
3 changed files with 219 additions and 0 deletions
|
@ -439,6 +439,86 @@ class Module {
|
||||||
|
|
||||||
Id OpInterpolateAtOffset(Id result_type, Id interpolant, Id offset);
|
Id OpInterpolateAtOffset(Id result_type, Id interpolant, Id offset);
|
||||||
|
|
||||||
|
// Image
|
||||||
|
|
||||||
|
Id OpSampledImage(Id result_type, Id image, Id sampler);
|
||||||
|
|
||||||
|
Id OpImageSampleImplicitLod(
|
||||||
|
Id result_type, Id sampled_image, Id coordinate,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageSampleExplicitLod(Id result_type, Id sampled_image, Id coordinate,
|
||||||
|
spv::ImageOperandsMask image_operands, Id lod,
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageSampleDrefImplicitLod(
|
||||||
|
Id result_type, Id sampled_image, Id coordinate, Id dref,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageSampleDrefExplicitLod(Id result_type, Id sampled_image,
|
||||||
|
Id coordinate, Id dref,
|
||||||
|
spv::ImageOperandsMask image_operands,
|
||||||
|
Id lod,
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageSampleProjImplicitLod(
|
||||||
|
Id result_type, Id sampled_image, Id coordinate,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageSampleProjExplicitLod(Id result_type, Id sampled_image,
|
||||||
|
Id coordinate,
|
||||||
|
spv::ImageOperandsMask image_operands,
|
||||||
|
Id lod,
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageSampleProjDrefImplicitLod(
|
||||||
|
Id result_type, Id sampled_image, Id coordinate, Id dref,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageSampleProjDrefExplicitLod(Id result_type, Id sampled_image,
|
||||||
|
Id coordinate, Id dref,
|
||||||
|
spv::ImageOperandsMask image_operands,
|
||||||
|
Id lod,
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageFetch(Id result_type, Id sampled_image, Id coordinate,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageGather(Id result_type, Id sampled_image, Id coordinate,
|
||||||
|
Id component,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id
|
||||||
|
OpImageDrefGather(Id result_type, Id sampled_image, Id coordinate, Id dref,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageRead(Id result_type, Id sampled_image, Id coordinate,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImageWrite(Id image, Id coordinate, Id texel,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands = {},
|
||||||
|
const std::vector<Id>& operands = {});
|
||||||
|
|
||||||
|
Id OpImage(Id result_type, Id sampled_image);
|
||||||
|
|
||||||
|
Id OpImageQuerySizeLod(Id result_type, Id image, Id level_of_detail);
|
||||||
|
|
||||||
|
Id OpImageQuerySize(Id result_type, Id image);
|
||||||
|
|
||||||
|
Id OpImageQueryLod(Id result_type, Id image, Id coordinate);
|
||||||
|
|
||||||
|
Id OpImageQueryLevels(Id result_type, Id image);
|
||||||
|
|
||||||
|
Id OpImageQuerySamples(Id result_type, Id image);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
Id AddCode(std::unique_ptr<Op> op);
|
Id AddCode(std::unique_ptr<Op> op);
|
||||||
|
|
||||||
|
|
|
@ -25,6 +25,7 @@ add_library(sirit
|
||||||
insts/bit.cpp
|
insts/bit.cpp
|
||||||
insts/arithmetic.cpp
|
insts/arithmetic.cpp
|
||||||
insts/extension.cpp
|
insts/extension.cpp
|
||||||
|
insts/image.cpp
|
||||||
)
|
)
|
||||||
target_include_directories(sirit
|
target_include_directories(sirit
|
||||||
PUBLIC ../include
|
PUBLIC ../include
|
||||||
|
|
138
src/insts/image.cpp
Normal file
138
src/insts/image.cpp
Normal file
|
@ -0,0 +1,138 @@
|
||||||
|
/* This file is part of the sirit project.
|
||||||
|
* Copyright (c) 2018 ReinUsesLisp
|
||||||
|
* This software may be used and distributed according to the terms of the GNU
|
||||||
|
* Lesser General Public License version 2.1 or any later version.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "common_types.h"
|
||||||
|
#include "op.h"
|
||||||
|
#include "sirit/sirit.h"
|
||||||
|
|
||||||
|
namespace Sirit {
|
||||||
|
|
||||||
|
Id Module::OpSampledImage(Id result_type, Id image, Id sampler) {
|
||||||
|
auto op{
|
||||||
|
std::make_unique<Op>(spv::Op::OpSampledImage, bound++, result_type)};
|
||||||
|
op->Add(image);
|
||||||
|
op->Add(sampler);
|
||||||
|
return AddCode(std::move(op));
|
||||||
|
}
|
||||||
|
|
||||||
|
void AddImageOperands(Op* op,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands,
|
||||||
|
const std::vector<Id>& operands) {
|
||||||
|
if (!image_operands)
|
||||||
|
return;
|
||||||
|
op->Add(static_cast<u32>(*image_operands));
|
||||||
|
op->Add(operands);
|
||||||
|
}
|
||||||
|
|
||||||
|
#define DEFINE_IMAGE_OP(funcname, opcode) \
|
||||||
|
Id Module::funcname(Id result_type, Id sampled_image, Id coordinate, \
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands, \
|
||||||
|
const std::vector<Id>& operands) { \
|
||||||
|
auto op{std::make_unique<Op>(opcode, bound++, result_type)}; \
|
||||||
|
op->Add(sampled_image); \
|
||||||
|
op->Add(coordinate); \
|
||||||
|
AddImageOperands(op.get(), image_operands, operands); \
|
||||||
|
return AddCode(std::move(op)); \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define DEFINE_IMAGE_EXP_OP(funcname, opcode) \
|
||||||
|
Id Module::funcname(Id result_type, Id sampled_image, Id coordinate, \
|
||||||
|
spv::ImageOperandsMask image_operands, Id lod, \
|
||||||
|
const std::vector<Id>& operands) { \
|
||||||
|
auto op{std::make_unique<Op>(opcode, bound++, result_type)}; \
|
||||||
|
op->Add(sampled_image); \
|
||||||
|
op->Add(coordinate); \
|
||||||
|
op->Add(static_cast<u32>(image_operands)); \
|
||||||
|
op->Add(lod); \
|
||||||
|
op->Add(operands); \
|
||||||
|
return AddCode(std::move(op)); \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define DEFINE_IMAGE_EXTRA_OP(funcname, opcode) \
|
||||||
|
Id Module::funcname(Id result_type, Id sampled_image, Id coordinate, \
|
||||||
|
Id extra, \
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands, \
|
||||||
|
const std::vector<Id>& operands) { \
|
||||||
|
auto op{std::make_unique<Op>(opcode, bound++, result_type)}; \
|
||||||
|
op->Add(sampled_image); \
|
||||||
|
op->Add(coordinate); \
|
||||||
|
op->Add(extra); \
|
||||||
|
AddImageOperands(op.get(), image_operands, operands); \
|
||||||
|
return AddCode(std::move(op)); \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define DEFINE_IMAGE_EXTRA_EXP_OP(funcname, opcode) \
|
||||||
|
Id Module::funcname(Id result_type, Id sampled_image, Id coordinate, \
|
||||||
|
Id extra, spv::ImageOperandsMask image_operands, \
|
||||||
|
Id lod, const std::vector<Id>& operands) { \
|
||||||
|
auto op{std::make_unique<Op>(opcode, bound++, result_type)}; \
|
||||||
|
op->Add(sampled_image); \
|
||||||
|
op->Add(coordinate); \
|
||||||
|
op->Add(extra); \
|
||||||
|
op->Add(static_cast<u32>(image_operands)); \
|
||||||
|
op->Add(lod); \
|
||||||
|
op->Add(operands); \
|
||||||
|
return AddCode(std::move(op)); \
|
||||||
|
}
|
||||||
|
|
||||||
|
DEFINE_IMAGE_OP(OpImageSampleImplicitLod, spv::Op::OpImageSampleImplicitLod)
|
||||||
|
DEFINE_IMAGE_EXP_OP(OpImageSampleExplicitLod, spv::Op::OpImageSampleExplicitLod)
|
||||||
|
DEFINE_IMAGE_EXTRA_OP(OpImageSampleDrefImplicitLod,
|
||||||
|
spv::Op::OpImageSampleDrefImplicitLod)
|
||||||
|
DEFINE_IMAGE_EXTRA_EXP_OP(OpImageSampleDrefExplicitLod,
|
||||||
|
spv::Op::OpImageSampleDrefExplicitLod)
|
||||||
|
DEFINE_IMAGE_OP(OpImageSampleProjImplicitLod,
|
||||||
|
spv::Op::OpImageSampleProjImplicitLod)
|
||||||
|
DEFINE_IMAGE_EXP_OP(OpImageSampleProjExplicitLod,
|
||||||
|
spv::Op::OpImageSampleProjExplicitLod)
|
||||||
|
DEFINE_IMAGE_EXTRA_OP(OpImageSampleProjDrefImplicitLod,
|
||||||
|
spv::Op::OpImageSampleProjDrefImplicitLod)
|
||||||
|
DEFINE_IMAGE_EXTRA_EXP_OP(OpImageSampleProjDrefExplicitLod,
|
||||||
|
spv::Op::OpImageSampleProjDrefExplicitLod)
|
||||||
|
DEFINE_IMAGE_OP(OpImageFetch, spv::Op::OpImageFetch)
|
||||||
|
DEFINE_IMAGE_EXTRA_OP(OpImageGather, spv::Op::OpImageGather)
|
||||||
|
DEFINE_IMAGE_EXTRA_OP(OpImageDrefGather, spv::Op::OpImageDrefGather)
|
||||||
|
DEFINE_IMAGE_OP(OpImageRead, spv::Op::OpImageRead)
|
||||||
|
|
||||||
|
Id Module::OpImageWrite(Id image, Id coordinate, Id texel,
|
||||||
|
std::optional<spv::ImageOperandsMask> image_operands,
|
||||||
|
const std::vector<Id>& operands) {
|
||||||
|
auto op{std::make_unique<Op>(spv::Op::OpImageWrite)};
|
||||||
|
op->Add(image);
|
||||||
|
op->Add(coordinate);
|
||||||
|
op->Add(texel);
|
||||||
|
AddImageOperands(op.get(), image_operands, operands);
|
||||||
|
return AddCode(std::move(op));
|
||||||
|
}
|
||||||
|
|
||||||
|
Id Module::OpImage(Id result_type, Id sampled_image) {
|
||||||
|
auto op{std::make_unique<Op>(spv::Op::OpImage, bound++, result_type)};
|
||||||
|
op->Add(sampled_image);
|
||||||
|
return AddCode(std::move(op));
|
||||||
|
}
|
||||||
|
|
||||||
|
#define DEFINE_IMAGE_QUERY_OP(funcname, opcode) \
|
||||||
|
Id Module::funcname(Id result_type, Id image) { \
|
||||||
|
auto op{std::make_unique<Op>(opcode, bound++, result_type)}; \
|
||||||
|
op->Add(image); \
|
||||||
|
return AddCode(std::move(op)); \
|
||||||
|
}
|
||||||
|
|
||||||
|
#define DEFINE_IMAGE_QUERY_BIN_OP(funcname, opcode) \
|
||||||
|
Id Module::funcname(Id result_type, Id image, Id extra) { \
|
||||||
|
auto op{std::make_unique<Op>(opcode, bound++, result_type)}; \
|
||||||
|
op->Add(image); \
|
||||||
|
op->Add(extra); \
|
||||||
|
return AddCode(std::move(op)); \
|
||||||
|
}
|
||||||
|
|
||||||
|
DEFINE_IMAGE_QUERY_BIN_OP(OpImageQuerySizeLod, spv::Op::OpImageQuerySizeLod)
|
||||||
|
DEFINE_IMAGE_QUERY_OP(OpImageQuerySize, spv::Op::OpImageQuerySize)
|
||||||
|
DEFINE_IMAGE_QUERY_BIN_OP(OpImageQueryLod, spv::Op::OpImageQueryLod)
|
||||||
|
DEFINE_IMAGE_QUERY_OP(OpImageQueryLevels, spv::Op::OpImageQueryLevels)
|
||||||
|
DEFINE_IMAGE_QUERY_OP(OpImageQuerySamples, spv::Op::OpImageQuerySamples)
|
||||||
|
|
||||||
|
} // namespace Sirit
|
Loading…
Reference in a new issue