API追加

Author: Tengu712

include/orge.h

@@ -111,6 +111,23 @@ API_EXPORT uint8_t orgeUpdateBufferDescriptor(
 	uint32_t offset
 );
 
+/// バッファディスクリプタを更新する関数 (コンピュートパイプライン)
+///
+/// - pipelineId: パイプラインID
+/// - id: バッファID
+/// - set: ディスクリプタセット番号
+/// - index: 何個目のディスクリプタセットか
+/// - binding: バインディング番号
+/// - offset: 配列上のオフセット (ディスクリプタが配列でないなら0)
+API_EXPORT uint8_t orgeUpdateComputeBufferDescriptor(
+	const char *pipelineId,
+	const char *id,
+	uint32_t set,
+	uint32_t index,
+	uint32_t binding,
+	uint32_t offset
+);
+
 /// orgeにイメージを追加する関数
 ///
 /// - file: アセットファイル名
@@ -138,6 +155,23 @@ API_EXPORT uint8_t orgeUpdateImageDescriptor(
 	uint32_t offset
 );
 
+/// イメージディスクリプタを更新する関数 (コンピュートパイプライン)
+///
+/// - pipelineId: パイプラインID
+/// - id: イメージID (アセットファイル名)
+/// - set: ディスクリプタセット番号
+/// - index: 何個目のディスクリプタセットか
+/// - binding: バインディング番号
+/// - offset: 配列上のオフセット (ディスクリプタが配列でないなら0)
+API_EXPORT uint8_t orgeUpdateComputeImageDescriptor(
+	const char *pipelineId,
+	const char *id,
+	uint32_t set,
+	uint32_t index,
+	uint32_t binding,
+	uint32_t offset
+);
+
 /// orgeにサンプラを追加する関数
 ///
 /// - id: サンプラID
@@ -174,6 +208,23 @@ API_EXPORT uint8_t orgeUpdateSamplerDescriptor(
 	uint32_t offset
 );
 
+/// サンプラディスクリプタを更新する関数 (コンピュート)
+///
+/// - pipelineId: パイプラインID
+/// - id: サンプラID
+/// - set: ディスクリプタセット番号
+/// - index: 何個目のディスクリプタセットか
+/// - binding: バインディング番号
+/// - offset: 配列上のオフセット (ディスクリプタが配列でないなら0)
+API_EXPORT uint8_t orgeUpdateComputeSamplerDescriptor(
+	const char *pipelineId,
+	const char *id,
+	uint32_t set,
+	uint32_t index,
+	uint32_t binding,
+	uint32_t offset
+);
+
 /// アタッチメントディスクリプタを更新する関数
 ///
 /// - renderPassId: レンダーパスID
@@ -330,6 +381,21 @@ API_EXPORT uint8_t orgeDraw(uint32_t instanceCount, uint32_t instanceOffset);
 /// WARN: パイプラインがバインドされていること。
 API_EXPORT uint8_t orgeDrawDirectly(uint32_t vertexCount, uint32_t instanceCount, uint32_t instanceOffset);
 
+// ================================================================================================================== //
+//     Compute                                                                                                        //
+// ================================================================================================================== //
+
+/// コンピュートパイプラインをバインドする関数
+///
+/// WARN: 描画が開始されていること。
+/// WARN: レンダーパスがバインドされていないこと。
+uint8_t orgeBindComputePipeline(const char *pipelineId, uint32_t const *indices);
+
+/// コンピュートパイプラインを実行する関数
+///
+/// WARN: パイプラインがバインドされていること。
+uint8_t orgeDispatch(uint32_t x, uint32_t y, uint32_t z);
+
 // ================================================================================================================== //
 //     Input                                                                                                          //
 // ================================================================================================================== //

src/graphics/compute/pipeline.cpp

@@ -48,7 +48,7 @@ DEFINE_UPDATE_DESC_METHOD(updateBufferDescriptor) {
 	core::device().updateDescriptorSets(1, &ds, 0, nullptr);
 }
 
-DEFINE_UPDATE_DESC_METHOD(updateUserImageDescriptor) {
+DEFINE_UPDATE_DESC_METHOD(updateImageDescriptor) {
 	const auto &descSets = error::at(_descSetss, set, "descriptor sets");
 	const auto &descSet = error::at(descSets, index, "descriptor sets allocated");
 	const auto &image = resource::getUserImage(id);

src/graphics/compute/pipeline.hpp

@@ -51,7 +51,7 @@ class ComputePipeline {
 	) const
 
 	DECLARE_UPDATE_DESC_METHOD(updateBufferDescriptor);
-	DECLARE_UPDATE_DESC_METHOD(updateUserImageDescriptor);
+	DECLARE_UPDATE_DESC_METHOD(updateImageDescriptor);
 	DECLARE_UPDATE_DESC_METHOD(updateSamplerDescriptor);
 
 #undef DECLARE_UPDATE_DESC_METHOD

src/graphics/renderer/context.hpp

@@ -134,10 +134,14 @@ class RenderContext {
 		if (_renderPass) {
 			throw std::format("render pass '{}' not ended.", _renderPass->id());
 		}
-		const auto &computePipeline = compute::getComputePipeline(pipelineId);
-		computePipeline.bind(_commandBuffer);
-		computePipeline.bindDescriptorSets(_commandBuffer, indices);
-		_computePipeline = &computePipeline;
+		if (!_computePipeline || _computePipeline->id() != pipelineId) {
+			const auto &computePipeline = compute::getComputePipeline(pipelineId);
+			computePipeline.bind(_commandBuffer);
+			if (indices) {
+				computePipeline.bindDescriptorSets(_commandBuffer, indices);
+			}
+			_computePipeline = &computePipeline;
+		}
 	}
 
 	void dispatch(uint32_t x, uint32_t y, uint32_t z) const {

src/orge-compute.cpp

@@ -0,0 +1,38 @@
+#include <orge.h>
+
+#include "graphics/compute/pipeline.hpp"
+#include "graphics/renderer/renderer.hpp"
+#include "orge-private.hpp"
+
+#define DEFINE_UPDATE_DESC_FUNC(n) \
+	uint8_t orgeUpdateCompute##n##Descriptor( \
+		const char *pipelineId, \
+		const char *id, \
+		uint32_t set, \
+		uint32_t index, \
+		uint32_t binding, \
+		uint32_t offset \
+	) { \
+		TRY( \
+			graphics::compute::getComputePipeline(pipelineId) \
+				.update##n##Descriptor(id, set, index, binding, offset) \
+		); \
+	}
+
+DEFINE_UPDATE_DESC_FUNC(Buffer)
+DEFINE_UPDATE_DESC_FUNC(Image)
+DEFINE_UPDATE_DESC_FUNC(Sampler)
+
+#define TRY_OR(n) \
+	bool result = false; \
+	CHECK(n); \
+	if (!result) graphics::renderer::renderer().reset(); \
+	return static_cast<uint8_t>(result);
+
+uint8_t orgeBindComputePipeline(const char *pipelineId, uint32_t const *indices) {
+	TRY_OR(graphics::renderer::renderer().getContext().bindComputePipeline(pipelineId, indices));
+}
+
+uint8_t orgeDispatch(uint32_t x, uint32_t y, uint32_t z) {
+	TRY_OR(graphics::renderer::renderer().getContext().dispatch(x, y, z));
+}