100 lines
2.9 KiB
C
100 lines
2.9 KiB
C
/* Spa */
|
|
/* SPDX-FileCopyrightText: Copyright © 2019 Wim Taymans */
|
|
/* SPDX-License-Identifier: MIT */
|
|
|
|
#include <vulkan/vulkan.h>
|
|
|
|
#include <spa/buffer/buffer.h>
|
|
#include <spa/param/video/format.h>
|
|
#include <spa/node/node.h>
|
|
#include <spa/pod/builder.h>
|
|
|
|
#include "vulkan-utils.h"
|
|
|
|
#define MAX_STREAMS 2
|
|
#define WORKGROUP_SIZE 32
|
|
|
|
struct pixel {
|
|
float r, g, b, a;
|
|
};
|
|
|
|
struct push_constants {
|
|
float time;
|
|
int frame;
|
|
int width;
|
|
int height;
|
|
};
|
|
|
|
struct vulkan_stream {
|
|
enum spa_direction direction;
|
|
|
|
uint32_t pending_buffer_id;
|
|
uint32_t current_buffer_id;
|
|
uint32_t busy_buffer_id;
|
|
uint32_t ready_buffer_id;
|
|
|
|
uint32_t format;
|
|
|
|
struct vulkan_buffer buffers[MAX_BUFFERS];
|
|
struct spa_buffer *spa_buffers[MAX_BUFFERS];
|
|
uint32_t n_buffers;
|
|
};
|
|
|
|
struct vulkan_compute_state {
|
|
struct spa_log *log;
|
|
|
|
struct push_constants constants;
|
|
|
|
struct vulkan_base base;
|
|
|
|
struct vulkan_format_infos formatInfos;
|
|
|
|
VkPipeline pipeline;
|
|
VkPipelineLayout pipelineLayout;
|
|
const char *shaderName;
|
|
VkShaderModule computeShaderModule;
|
|
|
|
VkCommandPool commandPool;
|
|
VkCommandBuffer commandBuffer;
|
|
struct vulkan_staging_buffer staging_buffer;
|
|
|
|
VkFence fence;
|
|
VkSemaphore pipelineSemaphore;
|
|
unsigned int initialized:1;
|
|
unsigned int prepared:1;
|
|
unsigned int started:1;
|
|
|
|
VkDescriptorPool descriptorPool;
|
|
VkDescriptorSetLayout descriptorSetLayout;
|
|
|
|
VkSampler sampler;
|
|
|
|
uint32_t n_streams;
|
|
VkDescriptorSet descriptorSet;
|
|
struct vulkan_stream streams[MAX_STREAMS];
|
|
};
|
|
|
|
int spa_vulkan_compute_init_stream(struct vulkan_compute_state *s, struct vulkan_stream *stream, enum spa_direction,
|
|
struct spa_dict *props);
|
|
|
|
int spa_vulkan_compute_fixate_modifier(struct vulkan_compute_state *s, struct vulkan_stream *p, struct spa_video_info_dsp *dsp_info,
|
|
uint32_t modifierCount, uint64_t *modifiers, uint64_t *modifier);
|
|
int spa_vulkan_compute_use_buffers(struct vulkan_compute_state *s, struct vulkan_stream *stream, uint32_t flags,
|
|
struct spa_video_info_dsp *dsp_info, uint32_t n_buffers, struct spa_buffer **buffers);
|
|
int spa_vulkan_compute_enumerate_formats(struct vulkan_compute_state *s, uint32_t index, uint32_t caps,
|
|
struct spa_pod **param, struct spa_pod_builder *builder);
|
|
int spa_vulkan_compute_prepare(struct vulkan_compute_state *s);
|
|
int spa_vulkan_compute_unprepare(struct vulkan_compute_state *s);
|
|
|
|
int spa_vulkan_compute_start(struct vulkan_compute_state *s);
|
|
int spa_vulkan_compute_stop(struct vulkan_compute_state *s);
|
|
int spa_vulkan_compute_ready(struct vulkan_compute_state *s);
|
|
int spa_vulkan_compute_process(struct vulkan_compute_state *s);
|
|
int spa_vulkan_compute_cleanup(struct vulkan_compute_state *s);
|
|
|
|
int spa_vulkan_compute_get_buffer_caps(struct vulkan_compute_state *s, enum spa_direction direction);
|
|
struct vulkan_modifier_info *spa_vulkan_compute_get_modifier_info(struct vulkan_compute_state *s,
|
|
struct spa_video_info_dsp *dsp_info);
|
|
|
|
int spa_vulkan_compute_init(struct vulkan_compute_state *s);
|
|
void spa_vulkan_compute_deinit(struct vulkan_compute_state *s);
|