vulkan: add vulkan compute source

Add a source that runs a compute shader and exports the GPU buffer
as a DmaBuf to the clients.

12 files changed, 1759 insertions, 0 deletions

diff --git a/meson_options.txt b/meson_options.txt
index 76a7f8dc..f573d64f 100644
--- a/meson_options.txt
+++ b/meson_options.txt
@@ -86,3 +86,7 @@ option('volume',
        description: 'Enable volume spa plugin integration',
        type: 'boolean',
        value: false)
+option('vulkan',
+       description: 'Enable vulkan spa plugin integration',
+       type: 'boolean',
+       value: true)
diff --git a/spa/meson.build b/spa/meson.build
index c6edc8b2..6fcfcaa0 100644
--- a/spa/meson.build
+++ b/spa/meson.build
@@ -27,6 +27,9 @@ if get_option('spa-plugins')
     avcodec_dep = dependency('libavcodec')
     avformat_dep = dependency('libavformat')
   endif
+  if get_option('vulkan')
+    vulkan_dep = dependency('vulkan')
+  endif
 
   subdir('plugins')
 endif
diff --git a/spa/plugins/meson.build b/spa/plugins/meson.build
index f2948f6a..71bacba3 100644
--- a/spa/plugins/meson.build
+++ b/spa/plugins/meson.build
@@ -31,6 +31,9 @@ endif
 if get_option('volume')
   subdir('volume')
 endif
+if get_option('vulkan')
+  subdir('vulkan')
+endif
 if get_option('v4l2')
   subdir('v4l2')
 endif
diff --git a/spa/plugins/vulkan/meson.build b/spa/plugins/vulkan/meson.build
new file mode 100644
index 00000000..4bd79563
--- /dev/null
+++ b/spa/plugins/vulkan/meson.build
@@ -0,0 +1,10 @@
+spa_vulkan_sources = ['plugin.c',
+                'vulkan-compute-source.c',
+                'vulkan-utils.c']
+
+spa_vulkan = shared_library('spa-vulkan',
+                           spa_vulkan_sources,
+                           include_directories : [spa_inc],
+                           dependencies : [ vulkan_dep, mathlib ],
+                           install : true,
+                           install_dir : '@0@/spa/vulkan'.format(get_option('libdir')))
diff --git a/spa/plugins/vulkan/plugin.c b/spa/plugins/vulkan/plugin.c
new file mode 100644
index 00000000..dbaca127
--- /dev/null
+++ b/spa/plugins/vulkan/plugin.c
@@ -0,0 +1,46 @@
+/* Spa vulkan plugin
+ *
+ * Copyright © 2019 Wim Taymans
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ */
+
+#include <errno.h>
+
+#include <spa/support/plugin.h>
+
+extern const struct spa_handle_factory spa_vulkan_compute_source_factory;
+
+SPA_EXPORT
+int spa_handle_factory_enum(const struct spa_handle_factory **factory, uint32_t *index)
+{
+	spa_return_val_if_fail(factory != NULL, -EINVAL);
+	spa_return_val_if_fail(index != NULL, -EINVAL);
+
+	switch (*index) {
+	case 0:
+		*factory = &spa_vulkan_compute_source_factory;
+		break;
+	default:
+		return 0;
+	}
+	(*index)++;
+	return 1;
+}
diff --git a/spa/plugins/vulkan/shaders/main.comp b/spa/plugins/vulkan/shaders/main.comp
new file mode 100644
index 00000000..cc14c64d
--- /dev/null
+++ b/spa/plugins/vulkan/shaders/main.comp
@@ -0,0 +1,57 @@
+#version 450
+#extension GL_ARB_separate_shader_objects : enable
+
+#define WORKGROUP_SIZE 32
+layout (local_size_x = WORKGROUP_SIZE, local_size_y = WORKGROUP_SIZE, local_size_z = 1 ) in;
+
+struct Pixel{
+  vec4 value;
+};
+
+layout(std140, binding = 0) buffer buf
+{
+   Pixel imageData[];
+};
+
+layout( push_constant ) uniform Constants {
+  float time;
+  int frame;
+  int width;
+  int height;
+} PushConstant;
+
+float iTime;
+int iFrame;
+vec3 iResolution;
+vec4 iMouse;
+
+void mainImage( out vec4 fragColor, in vec2 fragCoord );
+
+void main()
+{
+	iTime = PushConstant.time;
+	iFrame = PushConstant.frame;
+	iResolution = vec3(float(PushConstant.width), float(PushConstant.height), 0.0);
+	iMouse = vec4(0.0, 0.0, 0.0, 0.0);
+	vec2 coord = vec2(float(gl_GlobalInvocationID.x),
+			iResolution.y - float(gl_GlobalInvocationID.y));
+	vec4 outColor;
+
+	if(coord.x >= iResolution.x || coord.y >= iResolution.y)
+		return;
+
+	mainImage(outColor, coord);
+
+	imageData[PushConstant.width * gl_GlobalInvocationID.y +
+			gl_GlobalInvocationID.x].value = outColor;
+}
+
+//#include "plasma-globe.comp"
+#include "mandelbrot-distance.comp"
+//#include "ring-twister.comp"
+//#include "gears.comp"
+//#include "protean-clouds.comp"
+//#include "flame.comp"
+//#include "shader.comp"
+//#include "raymarching-primitives.comp"
+//#include "3d-primitives.comp"
diff --git a/spa/plugins/vulkan/shaders/main.spv b/spa/plugins/vulkan/shaders/main.spv
new file mode 100644
index 00000000..2a388dbc
--- /dev/null
+++ b/spa/plugins/vulkan/shaders/main.spv
diff --git a/spa/plugins/vulkan/shaders/mandelbrot-distance.comp b/spa/plugins/vulkan/shaders/mandelbrot-distance.comp
new file mode 100644
index 00000000..69d5bfde
--- /dev/null
+++ b/spa/plugins/vulkan/shaders/mandelbrot-distance.comp
@@ -0,0 +1,58 @@
+// Created by inigo quilez - iq/2013
+// License Creative Commons Attribution-NonCommercial-ShareAlike 3.0 Unported License.
+
+
+// This shader computes the distance to the Mandelbrot Set for everypixel, and colorizes
+// it accoringly.
+//
+// Z -> Z²+c, Z0 = 0.
+// therefore Z' -> 2·Z·Z' + 1
+//
+// The Hubbard-Douady potential G(c) is G(c) = log Z/2^n
+// G'(c) = Z'/Z/2^n
+//
+// So the distance is |G(c)|/|G'(c)| = |Z|·log|Z|/|Z'|
+//
+// More info here: http://www.iquilezles.org/www/articles/distancefractals/distancefractals.htm
+
+
+void mainImage( out vec4 fragColor, in vec2 fragCoord )
+{
+    vec2 p = -1.0 + 2.0 * fragCoord.xy / iResolution.xy;
+    p.x *= iResolution.x/iResolution.y;
+
+    // animation
+	float tz = 0.5 - 0.5*cos(0.225*iTime);
+    float zoo = pow( 0.5, 13.0*tz );
+	vec2 c = vec2(-0.05,.6805) + p*zoo;
+
+    // iterate
+    float di =  1.0;
+    vec2 z  = vec2(0.0);
+    float m2 = 0.0;
+    vec2 dz = vec2(0.0);
+    for( int i=0; i<300; i++ )
+    {
+        if( m2>1024.0 ) { di=0.0; break; }
+
+		// Z' -> 2·Z·Z' + 1
+        dz = 2.0*vec2(z.x*dz.x-z.y*dz.y, z.x*dz.y + z.y*dz.x) + vec2(1.0,0.0);
+
+        // Z -> Z² + c
+        z = vec2( z.x*z.x - z.y*z.y, 2.0*z.x*z.y ) + c;
+
+        m2 = dot(z,z);
+    }
+
+    // distance
+	// d(c) = |Z|·log|Z|/|Z'|
+	float d = 0.5*sqrt(dot(z,z)/dot(dz,dz))*log(dot(z,z));
+    if( di>0.5 ) d=0.0;
+
+    // do some soft coloring based on distance
+	d = clamp( pow(4.0*d/zoo,0.2), 0.0, 1.0 );
+
+    vec3 col = vec3( d );
+
+    fragColor = vec4( col, 1.0 );
+}
diff --git a/spa/plugins/vulkan/vulkan-compute-source.c b/spa/plugins/vulkan/vulkan-compute-source.c
new file mode 100644
index 00000000..839fb394
--- /dev/null
+++ b/spa/plugins/vulkan/vulkan-compute-source.c
@@ -0,0 +1,972 @@
+/* Spa
+ *
+ * Copyright © 2019 Wim Taymans
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ */
+
+#include <errno.h>
+#include <stddef.h>
+#include <unistd.h>
+#include <string.h>
+#include <stdio.h>
+#include <sys/timerfd.h>
+
+#include <spa/support/log.h>
+#include <spa/support/loop.h>
+#include <spa/utils/list.h>
+#include <spa/utils/keys.h>
+#include <spa/node/node.h>
+#include <spa/node/utils.h>
+#include <spa/node/io.h>
+#include <spa/param/video/format-utils.h>
+#include <spa/param/param.h>
+#include <spa/pod/filter.h>
+
+#include "vulkan-utils.h"
+
+#define NAME "vulkan-compute-source"
+
+#define FRAMES_TO_TIME(port,f) ((port->current_format.info.raw.framerate.denom * (f) * SPA_NSEC_PER_SEC) / \
+                                (port->current_format.info.raw.framerate.num))
+
+#define DEFAULT_LIVE true
+
+struct props {
+	bool live;
+};
+
+static void reset_props(struct props *props)
+{
+	props->live = DEFAULT_LIVE;
+}
+
+#define MAX_PORTS 1
+
+struct buffer {
+	uint32_t id;
+#define BUFFER_FLAG_OUT (1<<0)
+	uint32_t flags;
+	struct spa_buffer *outbuf;
+	struct spa_meta_header *h;
+	struct spa_list link;
+};
+
+struct port {
+	uint64_t info_all;
+	struct spa_port_info info;
+	struct spa_param_info params[5];
+
+	struct spa_io_buffers *io;
+
+	bool have_format;
+	struct spa_video_info current_format;
+	size_t bpp;
+	int stride;
+
+	struct buffer buffers[MAX_BUFFERS];
+	uint32_t n_buffers;
+
+	struct spa_list empty;
+	struct spa_list ready;
+};
+
+struct impl {
+	struct spa_handle handle;
+	struct spa_node node;
+
+	struct spa_log *log;
+	struct spa_loop *data_loop;
+
+	uint64_t info_all;
+	struct spa_node_info info;
+	struct spa_param_info params[2];
+	struct props props;
+
+	struct spa_hook_list hooks;
+	struct spa_callbacks callbacks;
+
+	bool async;
+	struct spa_source timer_source;
+	struct itimerspec timerspec;
+
+	bool started;
+	uint64_t start_time;
+	uint64_t elapsed_time;
+
+	uint64_t frame_count;
+
+	struct vulkan_state state;
+	struct port port;
+};
+
+#define CHECK_PORT(this,d,p)  ((d) == SPA_DIRECTION_OUTPUT && (p) < MAX_PORTS)
+
+static int impl_node_enum_params(void *object, int seq,
+				 uint32_t id, uint32_t start, uint32_t num,
+				 const struct spa_pod *filter)
+{
+	struct impl *this = object;
+	struct spa_pod *param;
+	struct spa_pod_builder b = { 0 };
+	uint8_t buffer[1024];
+	struct spa_result_node_params result;
+	uint32_t count = 0;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+	spa_return_val_if_fail(num != 0, -EINVAL);
+
+	result.id = id;
+	result.next = start;
+      next:
+	result.index = result.next++;
+
+	spa_pod_builder_init(&b, buffer, sizeof(buffer));
+
+	switch (id) {
+	case SPA_PARAM_PropInfo:
+	{
+		struct props *p = &this->props;
+
+		switch (result.index) {
+		case 0:
+			param = spa_pod_builder_add_object(&b,
+				SPA_TYPE_OBJECT_PropInfo, id,
+				SPA_PROP_INFO_id,   SPA_POD_Id(SPA_PROP_live),
+				SPA_PROP_INFO_name, SPA_POD_String("Configure live mode of the source"),
+				SPA_PROP_INFO_type, SPA_POD_Bool(p->live));
+			break;
+		default:
+			return 0;
+		}
+		break;
+	}
+	case SPA_PARAM_Props:
+	{
+		struct props *p = &this->props;
+
+		switch (result.index) {
+		case 0:
+			param = spa_pod_builder_add_object(&b,
+				SPA_TYPE_OBJECT_Props, id,
+				SPA_PROP_live,        SPA_POD_Bool(p->live));
+			break;
+		default:
+			return 0;
+		}
+		break;
+	}
+	default:
+		return -ENOENT;
+	}
+
+	if (spa_pod_filter(&b, &result.param, param, filter) < 0)
+		goto next;
+
+	spa_node_emit_result(&this->hooks, seq, 0, SPA_RESULT_TYPE_NODE_PARAMS, &result);
+
+	if (++count != num)
+		goto next;
+
+	return 0;
+}
+
+static int impl_node_set_io(void *object, uint32_t id, void *data, size_t size)
+{
+	return -ENOTSUP;
+}
+
+static int impl_node_set_param(void *object, uint32_t id, uint32_t flags,
+			       const struct spa_pod *param)
+{
+	struct impl *this = object;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+
+	switch (id) {
+	case SPA_PARAM_Props:
+	{
+		struct props *p = &this->props;
+
+		if (param == NULL) {
+			reset_props(p);
+			return 0;
+		}
+		spa_pod_parse_object(param,
+			SPA_TYPE_OBJECT_Props, NULL,
+			SPA_PROP_live,        SPA_POD_OPT_Bool(&p->live));
+
+		if (p->live)
+			this->info.flags |= SPA_PORT_FLAG_LIVE;
+		else
+			this->info.flags &= ~SPA_PORT_FLAG_LIVE;
+		break;
+	}
+	default:
+		return -ENOENT;
+	}
+	return 0;
+}
+
+
+static void set_timer(struct impl *this, bool enabled)
+{
+	if (this->async || this->props.live) {
+		if (enabled) {
+			if (this->props.live) {
+				uint64_t next_time = this->start_time + this->elapsed_time;
+				this->timerspec.it_value.tv_sec = next_time / SPA_NSEC_PER_SEC;
+				this->timerspec.it_value.tv_nsec = next_time % SPA_NSEC_PER_SEC;
+			} else {
+				this->timerspec.it_value.tv_sec = 0;
+				this->timerspec.it_value.tv_nsec = 1;
+			}
+		} else {
+			this->timerspec.it_value.tv_sec = 0;
+			this->timerspec.it_value.tv_nsec = 0;
+		}
+		timerfd_settime(this->timer_source.fd, TFD_TIMER_ABSTIME, &this->timerspec, NULL);
+	}
+}
+
+static void read_timer(struct impl *this)
+{
+	uint64_t expirations;
+
+	if (this->async || this->props.live) {
+		if (read(this->timer_source.fd, &expirations, sizeof(uint64_t)) != sizeof(uint64_t))
+			perror("read timerfd");
+	}
+}
+
+static int make_buffer(struct impl *this)
+{
+	struct buffer *b;
+	struct port *port = &this->port;
+	uint32_t n_bytes;
+	int res;
+
+	read_timer(this);
+
+	if ((res = spa_vulkan_ready(&this->state)) < 0) {
+		res = SPA_STATUS_OK;
+		goto next;
+	}
+
+	if (spa_list_is_empty(&port->empty)) {
+		set_timer(this, false);
+		spa_log_error(this->log, NAME " %p: out of buffers", this);
+		return -EPIPE;
+	}
+	b = spa_list_first(&port->empty, struct buffer, link);
+	spa_list_remove(&b->link);
+
+	n_bytes = b->outbuf->datas[0].maxsize;
+
+	spa_log_trace(this->log, NAME " %p: dequeue buffer %d", this, b->id);
+
+	this->state.constants.time = this->elapsed_time / (float) SPA_NSEC_PER_SEC;
+	this->state.constants.frame = this->frame_count;
+
+	spa_vulkan_process(&this->state, b->id);
+
+	b->outbuf->datas[0].chunk->offset = 0;
+	b->outbuf->datas[0].chunk->size = n_bytes;
+	b->outbuf->datas[0].chunk->stride = port->stride;
+
+	if (b->h) {
+		b->h->seq = this->frame_count;
+		b->h->pts = this->start_time + this->elapsed_time;
+		b->h->dts_offset = 0;
+	}
+
+	SPA_FLAG_SET(b->flags, BUFFER_FLAG_OUT);
+	spa_list_append(&port->ready, &b->link);
+
+	res = SPA_STATUS_HAVE_BUFFER;
+next:
+	this->frame_count++;
+	this->elapsed_time = FRAMES_TO_TIME(port, this->frame_count);
+	set_timer(this, true);
+
+	return res;
+}
+
+static inline void reuse_buffer(struct impl *this, struct port *port, uint32_t id)
+{
+	struct buffer *b = &port->buffers[id];
+
+	if (SPA_FLAG_CHECK(b->flags, BUFFER_FLAG_OUT)) {
+		spa_log_trace(this->log, NAME " %p: reuse buffer %d", this, id);
+
+		SPA_FLAG_UNSET(b->flags, BUFFER_FLAG_OUT);
+		spa_list_append(&port->empty, &b->link);
+
+		if (!this->props.live)
+			set_timer(this, true);
+	}
+}
+
+static void on_output(struct spa_source *source)
+{
+	struct impl *this = source->data;
+	struct port *port = &this->port;
+	struct spa_io_buffers *io = port->io;
+	int res;
+
+	if (io == NULL)
+		return;
+
+	if (io->status == SPA_STATUS_HAVE_BUFFER)
+		return;
+
+	if (io->buffer_id < port->n_buffers) {
+		reuse_buffer(this, port, io->buffer_id);
+		io->buffer_id = SPA_ID_INVALID;
+	}
+
+	res = make_buffer(this);
+
+	if (!spa_list_is_empty(&port->ready)) {
+		struct buffer *b = spa_list_first(&port->ready, struct buffer, link);
+		spa_list_remove(&b->link);
+
+		io->buffer_id = b->id;
+		io->status = SPA_STATUS_HAVE_BUFFER;
+        }
+	spa_node_call_ready(&this->callbacks, res);
+}
+
+static int impl_node_send_command(void *object, const struct spa_command *command)
+{
+	struct impl *this = object;
+	struct port *port;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+	spa_return_val_if_fail(command != NULL, -EINVAL);
+
+	port = &this->port;
+
+	switch (SPA_NODE_COMMAND_ID(command)) {
+	case SPA_NODE_COMMAND_Start:
+	{
+		struct timespec now;
+
+		if (!port->have_format)
+			return -EIO;
+		if (port->n_buffers == 0)
+			return -EIO;
+
+		if (this->started)
+			return 0;
+
+		clock_gettime(CLOCK_MONOTONIC, &now);
+		if (this->props.live)
+			this->start_time = SPA_TIMESPEC_TO_NSEC(&now);
+		else
+			this->start_time = 0;
+		this->frame_count = 0;
+		this->elapsed_time = 0;
+
+		this->started = true;
+		set_timer(this, true);
+		spa_vulkan_start(&this->state);
+		break;
+	}
+	case SPA_NODE_COMMAND_Pause:
+		if (!port->have_format)
+			return -EIO;
+		if (port->n_buffers == 0)
+			return -EIO;
+
+		if (!this->started)
+			return 0;
+
+		this->started = false;
+		set_timer(this, false);
+		spa_vulkan_stop(&this->state);
+		break;
+	default:
+		return -ENOTSUP;
+	}
+	return 0;
+}
+
+static const struct spa_dict_item node_info_items[] = {
+	{ SPA_KEY_MEDIA_CLASS, "Video/Source" },
+	{ SPA_KEY_NODE_DRIVER, "true" },
+};
+
+static void emit_node_info(struct impl *this, bool full)
+{
+	if (full)
+		this->info.change_mask = this->info_all;
+	if (this->info.change_mask) {
+		this->info.props = &SPA_DICT_INIT_ARRAY(node_info_items);
+		spa_node_emit_info(&this->hooks, &this->info);
+		this->info.change_mask = 0;
+	}
+}
+
+static void emit_port_info(struct impl *this, struct port *port, bool full)
+{
+	if (full)
+		port->info.change_mask = port->info_all;
+	if (port->info.change_mask) {
+		spa_node_emit_port_info(&this->hooks,
+				SPA_DIRECTION_OUTPUT, 0, &port->info);
+		port->info.change_mask = 0;
+	}
+}
+
+static int
+impl_node_add_listener(void *object,
+		struct spa_hook *listener,
+		const struct spa_node_events *events,
+		void *data)
+{
+	struct impl *this = object;
+	struct spa_hook_list save;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+
+	spa_hook_list_isolate(&this->hooks, &save, listener, events, data);
+
+	emit_node_info(this, true);
+	emit_port_info(this, &this->port, true);
+
+	spa_hook_list_join(&this->hooks, &save);
+
+	return 0;
+}
+
+static int
+impl_node_set_callbacks(void *object,
+			const struct spa_node_callbacks *callbacks,
+			void *data)
+{
+	struct impl *this = object;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+
+	this->callbacks = SPA_CALLBACKS_INIT(callbacks, data);
+
+	return 0;
+}
+
+static int impl_node_add_port(void *object, enum spa_direction direction, uint32_t port_id,
+		const struct spa_dict *props)
+{
+	return -ENOTSUP;
+}
+
+static int
+impl_node_remove_port(void *object, enum spa_direction direction, uint32_t port_id)
+{
+	return -ENOTSUP;
+}
+
+static int port_enum_formats(void *object,
+			     enum spa_direction direction, uint32_t port_id,
+			     uint32_t index,
+			     const struct spa_pod *filter,
+			     struct spa_pod **param,
+			     struct spa_pod_builder *builder)
+{
+	switch (index) {
+	case 0:
+		*param = spa_pod_builder_add_object(builder,
+			SPA_TYPE_OBJECT_Format, SPA_PARAM_EnumFormat,
+			SPA_FORMAT_mediaType,       SPA_POD_Id(SPA_MEDIA_TYPE_video),
+			SPA_FORMAT_mediaSubtype,    SPA_POD_Id(SPA_MEDIA_SUBTYPE_raw),
+			SPA_FORMAT_VIDEO_format,    SPA_POD_Id(SPA_VIDEO_FORMAT_RGBA_F32),
+			SPA_FORMAT_VIDEO_size,      SPA_POD_CHOICE_RANGE_Rectangle(
+							&SPA_RECTANGLE(320, 240),
+							&SPA_RECTANGLE(1, 1),
+							&SPA_RECTANGLE(INT32_MAX, INT32_MAX)),
+			SPA_FORMAT_VIDEO_framerate, SPA_POD_CHOICE_RANGE_Fraction(
+							&SPA_FRACTION(25,1),
+							&SPA_FRACTION(0, 1),
+							&SPA_FRACTION(INT32_MAX, 1)));
+		break;
+	default:
+		return 0;
+	}
+	return 1;
+}
+
+static int
+impl_node_port_enum_params(void *object, int seq,
+			enum spa_direction direction, uint32_t port_id,
+			uint32_t id, uint32_t start, uint32_t num,
+			const struct spa_pod *filter)
+{
+	struct impl *this = object;
+	struct port *port;
+	struct spa_pod_builder b = { 0 };
+	uint8_t buffer[1024];
+	struct spa_pod *param;
+	struct spa_result_node_params result;
+	uint32_t count = 0;
+	int res;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+	spa_return_val_if_fail(num != 0, -EINVAL);
+
+	spa_return_val_if_fail(CHECK_PORT(this, direction, port_id), -EINVAL);
+	port = &this->port;
+
+	result.id = id;
+	result.next = start;
+      next:
+	result.index = result.next++;
+
+	spa_pod_builder_init(&b, buffer, sizeof(buffer));
+
+	switch (id) {
+	case SPA_PARAM_EnumFormat:
+		if ((res = port_enum_formats(this, direction, port_id,
+						result.index, filter, &param, &b)) <= 0)
+			return res;
+		break;
+
+	case SPA_PARAM_Format:
+		if (!port->have_format)
+			return -EIO;
+		if (result.index > 0)
+			return 0;
+
+		param = spa_format_video_raw_build(&b, id, &port->current_format.info.raw);
+		break;
+
+	case SPA_PARAM_Buffers:
+	{
+		struct spa_video_info_raw *raw_info = &port->current_format.info.raw;
+
+		if (!port->have_format)
+			return -EIO;
+		if (result.index > 0)
+			return 0;
+
+		param = spa_pod_builder_add_object(&b,
+			SPA_TYPE_OBJECT_ParamBuffers, id,
+			SPA_PARAM_BUFFERS_buffers, SPA_POD_CHOICE_RANGE_Int(2, 1, MAX_BUFFERS),
+			SPA_PARAM_BUFFERS_blocks,  SPA_POD_Int(1),
+			SPA_PARAM_BUFFERS_size,    SPA_POD_Int(port->stride * raw_info->size.height),
+			SPA_PARAM_BUFFERS_stride,  SPA_POD_Int(port->stride),
+			SPA_PARAM_BUFFERS_align,   SPA_POD_Int(16));
+		break;
+	}
+	case SPA_PARAM_Meta:
+		switch (result.index) {
+		case 0:
+			param = spa_pod_builder_add_object(&b,
+				SPA_TYPE_OBJECT_ParamMeta, id,
+				SPA_PARAM_META_type, SPA_POD_Id(SPA_META_Header),
+				SPA_PARAM_META_size, SPA_POD_Int(sizeof(struct spa_meta_header)));
+			break;
+
+		default:
+			return 0;
+		}
+		break;
+	default:
+		return -ENOENT;
+	}
+
+	if (spa_pod_filter(&b, &result.param, param, filter) < 0)
+		goto next;
+
+	spa_node_emit_result(&this->hooks, seq, 0, SPA_RESULT_TYPE_NODE_PARAMS, &result);
+
+	if (++count != num)
+		goto next;
+
+	return 0;
+}
+
+static int clear_buffers(struct impl *this, struct port *port)
+{
+	if (port->n_buffers > 0) {
+		spa_log_info(this->log, NAME " %p: clear buffers", this);
+		spa_vulkan_use_buffers(&this->state, 0, 0, NULL);
+		port->n_buffers = 0;
+		spa_list_init(&port->empty);
+		spa_list_init(&port->ready);
+		this->started = false;
+		set_timer(this, false);
+	}
+	return 0;
+}
+
+static int port_set_format(struct impl *this, struct port *port,
+			   uint32_t flags,
+			   const struct spa_pod *format)
+{
+	int res;
+
+	if (format == NULL) {
+		port->have_format = false;
+		clear_buffers(this, port);
+		spa_vulkan_unprepare(&this->state);
+	} else {
+		struct spa_video_info info = { 0 };
+
+		if ((res = spa_format_parse(format, &info.media_type, &info.media_subtype)) < 0)
+			return res;
+
+		if (info.media_type != SPA_MEDIA_TYPE_video &&
+		    info.media_subtype != SPA_MEDIA_SUBTYPE_raw)
+			return -EINVAL;
+
+		if (spa_format_video_raw_parse(format, &info.info.raw) < 0)
+			return -EINVAL;
+
+		if (info.info.raw.format == SPA_VIDEO_FORMAT_RGBA_F32)
+			port->bpp = 16;
+		else
+			return -EINVAL;
+
+		this->state.constants.width = info.info.raw.size.width;
+		this->state.constants.height = info.info.raw.size.height;
+
+		port->current_format = info;
+		port->have_format = true;
+		spa_vulkan_prepare(&this->state);
+	}
+
+	port->info.change_mask |= SPA_PORT_CHANGE_MASK_PARAMS;
+	if (port->have_format) {
+		struct spa_video_info_raw *raw_info = &port->current_format.info.raw;
+		port->stride = SPA_ROUND_UP_N(port->bpp * raw_info->size.width, 4);
+		port->params[3] = SPA_PARAM_INFO(SPA_PARAM_Format, SPA_PARAM_INFO_READWRITE);
+		port->params[4] = SPA_PARAM_INFO(SPA_PARAM_Buffers, SPA_PARAM_INFO_READ);
+	} else {
+		port->params[3] = SPA_PARAM_INFO(SPA_PARAM_Format, SPA_PARAM_INFO_WRITE);
+		port->params[4] = SPA_PARAM_INFO(SPA_PARAM_Buffers, 0);
+	}
+	emit_port_info(this, port, false);
+
+	return 0;
+}
+
+static int
+impl_node_port_set_param(void *object,
+			 enum spa_direction direction, uint32_t port_id,
+			 uint32_t id, uint32_t flags,
+			 const struct spa_pod *param)
+{
+	struct impl *this = object;
+	struct port *port;
+	int res;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+	spa_return_val_if_fail(CHECK_PORT(node, direction, port_id), -EINVAL);
+	port = &this->port;
+
+	switch (id) {
+	case SPA_PARAM_Format:
+		res = port_set_format(this, port, flags, param);
+		break;
+	default:
+		return -ENOENT;
+	}
+	return res;
+}
+
+static int
+impl_node_port_use_buffers(void *object,
+			   enum spa_direction direction,
+			   uint32_t port_id,
+			   uint32_t flags,
+			   struct spa_buffer **buffers,
+			   uint32_t n_buffers)
+{
+	struct impl *this = object;
+	struct port *port;
+	uint32_t i;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+	spa_return_val_if_fail(CHECK_PORT(this, direction, port_id), -EINVAL);
+	port = &this->port;
+
+	if (!port->have_format)
+		return -EIO;
+
+	clear_buffers(this, port);
+
+	for (i = 0; i < n_buffers; i++) {
+		struct buffer *b;
+
+		b = &port->buffers[i];
+		b->id = i;
+		b->outbuf = buffers[i];
+		b->flags = 0;
+		b->h = spa_buffer_find_meta_data(buffers[i], SPA_META_Header, sizeof(*b->h));
+
+		spa_list_append(&port->empty, &b->link);
+	}
+	spa_vulkan_use_buffers(&this->state, flags, n_buffers, buffers);
+	port->n_buffers = n_buffers;
+
+	return 0;
+}
+
+static int
+impl_node_port_set_io(void *object,
+		      enum spa_direction direction,
+		      uint32_t port_id,
+		      uint32_t id,
+		      void *data, size_t size)
+{
+	struct impl *this = object;
+	struct port *port;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+	spa_return_val_if_fail(CHECK_PORT(this, direction, port_id), -EINVAL);
+	port = &this->port;
+
+	switch (id) {
+	case SPA_IO_Buffers:
+		port->io = data;
+		break;
+	default:
+		return -ENOENT;
+	}
+	return 0;
+}
+
+static int impl_node_port_reuse_buffer(void *object, uint32_t port_id, uint32_t buffer_id)
+{
+	struct impl *this = object;
+	struct port *port;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+	spa_return_val_if_fail(port_id == 0, -EINVAL);
+	port = &this->port;
+	spa_return_val_if_fail(buffer_id < port->n_buffers, -EINVAL);
+
+	reuse_buffer(this, port, buffer_id);
+
+	return 0;
+}
+
+static int impl_node_process(void *object)
+{
+	struct impl *this = object;
+	struct port *port;
+	struct spa_io_buffers *io;
+
+	spa_return_val_if_fail(this != NULL, -EINVAL);
+
+	port = &this->port;
+	io = port->io;
+	spa_return_val_if_fail(io != NULL, -EIO);
+
+	if (io->status == SPA_STATUS_HAVE_BUFFER)
+		return SPA_STATUS_HAVE_BUFFER;
+
+	if (io->buffer_id < port->n_buffers) {
+		reuse_buffer(this, port, io->buffer_id);
+		io->buffer_id = SPA_ID_INVALID;
+	}
+
+	if (!this->props.live && (io->status == SPA_STATUS_NEED_BUFFER))
+		return make_buffer(this);
+	else
+		return SPA_STATUS_OK;
+}
+
+static const struct spa_node_methods impl_node = {
+	SPA_VERSION_NODE_METHODS,
+	.add_listener = impl_node_add_listener,
+	.set_callbacks = impl_node_set_callbacks,
+	.enum_params = impl_node_enum_params,
+	.set_param = impl_node_set_param,
+	.set_io = impl_node_set_io,
+	.send_command = impl_node_send_command,
+	.add_port = impl_node_add_port,
+	.remove_port = impl_node_remove_port,
+	.port_enum_params = impl_node_port_enum_params,
+	.port_set_param = impl_node_port_set_param,
+	.port_use_buffers = impl_node_port_use_buffers,
+	.port_set_io = impl_node_port_set_io,
+	.port_reuse_buffer = impl_node_port_reuse_buffer,
+	.process = impl_node_process,
+};
+
+static int impl_get_interface(struct spa_handle *handle, uint32_t type, void **interface)
+{
+	struct impl *this;
+
+	spa_return_val_if_fail(handle != NULL, -EINVAL);
+	spa_return_val_if_fail(interface != NULL, -EINVAL);
+
+	this = (struct impl *) handle;
+
+	if (type == SPA_TYPE_INTERFACE_Node)
+		*interface = &this->node;
+	else
+		return -ENOENT;
+
+	return 0;
+}
+
+static int impl_clear(struct spa_handle *handle)
+{
+	struct impl *this;
+
+	spa_return_val_if_fail(handle != NULL, -EINVAL);
+
+	this = (struct impl *) handle;
+
+	if (this->data_loop)
+		spa_loop_remove_source(this->data_loop, &this->timer_source);
+	close(this->timer_source.fd);
+
+	return 0;
+}
+
+static size_t
+impl_get_size(const struct spa_handle_factory *factory,
+	      const struct spa_dict *params)
+{
+	return sizeof(struct impl);
+}
+
+static int
+impl_init(const struct spa_handle_factory *factory,
+	  struct spa_handle *handle,
+	  const struct spa_dict *info,
+	  const struct spa_support *support,
+	  uint32_t n_support)
+{
+	struct impl *this;
+	struct port *port;
+	uint32_t i;
+
+	spa_return_val_if_fail(factory != NULL, -EINVAL);
+	spa_return_val_if_fail(handle != NULL, -EINVAL);
+
+	handle->get_interface = impl_get_interface;
+	handle->clear = impl_clear;
+
+	this = (struct impl *) handle;
+
+	for (i = 0; i < n_support; i++) {
+		if (support[i].type == SPA_TYPE_INTERFACE_Log)
+			this->log = support[i].data;
+		else if (support[i].type == SPA_TYPE_INTERFACE_DataLoop)
+			this->data_loop = support[i].data;
+	}
+
+	spa_hook_list_init(&this->hooks);
+
+	this->node.iface = SPA_INTERFACE_INIT(
+			SPA_TYPE_INTERFACE_Node,
+			SPA_VERSION_NODE,
+			&impl_node, this);
+
+	this->info_all = SPA_NODE_CHANGE_MASK_FLAGS |
+			SPA_NODE_CHANGE_MASK_PROPS |
+			SPA_NODE_CHANGE_MASK_PARAMS;
+	this->info = SPA_NODE_INFO_INIT();
+	this->info.max_output_ports = 1;
+	this->info.flags = SPA_NODE_FLAG_RT;
+	this->params[0] = SPA_PARAM_INFO(SPA_PARAM_PropInfo, SPA_PARAM_INFO_READ);
+	this->params[1] = SPA_PARAM_INFO(SPA_PARAM_Props, SPA_PARAM_INFO_READWRITE);
+	this->info.params = this->params;
+	this->info.n_params = 2;
+	reset_props(&this->props);
+
+	this->timer_source.func = on_output;
+	this->timer_source.data = this;
+	this->timer_source.fd = timerfd_create(CLOCK_MONOTONIC, TFD_CLOEXEC);
+	this->timer_source.mask = SPA_IO_IN;
+	this->timer_source.rmask = 0;
+	this->timerspec.it_value.tv_sec = 0;
+	this->timerspec.it_value.tv_nsec = 0;
+	this->timerspec.it_interval.tv_sec = 0;
+	this->timerspec.it_interval.tv_nsec = 0;
+
+	if (this->data_loop)
+		spa_loop_add_source(this->data_loop, &this->timer_source);
+
+	port = &this->port;
+	port->info_all = SPA_PORT_CHANGE_MASK_FLAGS |
+			SPA_PORT_CHANGE_MASK_PARAMS;
+	port->info = SPA_PORT_INFO_INIT();
+	port->info.flags = SPA_PORT_FLAG_NO_REF | SPA_PORT_FLAG_CAN_ALLOC_BUFFERS;
+	if (this->props.live)
+		this->info.flags |= SPA_PORT_FLAG_LIVE;
+	port->params[0] = SPA_PARAM_INFO(SPA_PARAM_EnumFormat, SPA_PARAM_INFO_READ);
+	port->params[1] = SPA_PARAM_INFO(SPA_PARAM_Meta, SPA_PARAM_INFO_READ);
+	port->params[2] = SPA_PARAM_INFO(SPA_PARAM_IO, SPA_PARAM_INFO_READ);
+	port->params[3] = SPA_PARAM_INFO(SPA_PARAM_Format, SPA_PARAM_INFO_WRITE);
+	port->params[4] = SPA_PARAM_INFO(SPA_PARAM_Buffers, 0);
+	port->info.params = port->params;
+	port->info.n_params = 5;
+	spa_list_init(&port->empty);
+	spa_list_init(&port->ready);
+
+	spa_log_info(this->log, NAME " %p: initialized", this);
+
+	return 0;
+}
+
+static const struct spa_interface_info impl_interfaces[] = {
+	{SPA_TYPE_INTERFACE_Node,},
+};
+
+static int
+impl_enum_interface_info(const struct spa_handle_factory *factory,
+			 const struct spa_interface_info **info,
+			 uint32_t *index)
+{
+	spa_return_val_if_fail(factory != NULL, -EINVAL);
+	spa_return_val_if_fail(info != NULL, -EINVAL);
+	spa_return_val_if_fail(index != NULL, -EINVAL);
+
+	switch (*index) {
+	case 0:
+		*info = &impl_interfaces[*index];
+		break;
+	default:
+		return 0;
+	}
+	(*index)++;
+	return 1;
+}
+
+static const struct spa_dict_item info_items[] = {
+	{ SPA_KEY_FACTORY_AUTHOR, "Wim Taymans <wim.taymans@gmail.com>" },
+	{ SPA_KEY_FACTORY_DESCRIPTION, "Generate video frames using a vulkan compute shader" },
+};
+
+static const struct spa_dict info = SPA_DICT_INIT_ARRAY(info_items);
+
+const struct spa_handle_factory spa_vulkan_compute_source_factory = {
+	SPA_VERSION_HANDLE_FACTORY,
+	"api.vulkan.compute.source",
+	&info,
+	impl_get_size,
+	impl_init,
+	impl_enum_interface_info,
+};
diff --git a/spa/plugins/vulkan/vulkan-utils.c b/spa/plugins/vulkan/vulkan-utils.c
new file mode 100644
index 00000000..3d8004b3
--- /dev/null
+++ b/spa/plugins/vulkan/vulkan-utils.c
@@ -0,0 +1,538 @@
+#include <vulkan/vulkan.h>
+
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <sys/mman.h>
+#include <fcntl.h>
+#include <string.h>
+#include <alloca.h>
+#include <errno.h>
+#include <stdio.h>
+#include <assert.h>
+#include <math.h>
+#include <time.h>
+
+#include <spa/debug/mem.h>
+
+#include "vulkan-utils.h"
+
+#define VULKAN_INSTANCE_FUNCTION(name)						\
+	PFN_##name name = (PFN_##name)vkGetInstanceProcAddr(s->instance, #name)
+
+static int vkresult_to_errno(VkResult result)
+{
+	switch (result) {
+	case VK_SUCCESS:
+	case VK_EVENT_SET:
+	case VK_EVENT_RESET:
+		return 0;
+	case VK_NOT_READY:
+	case VK_INCOMPLETE:
+	case VK_ERROR_NATIVE_WINDOW_IN_USE_KHR:
+		return EBUSY;
+	case VK_TIMEOUT:
+		return ETIMEDOUT;
+	case VK_ERROR_OUT_OF_HOST_MEMORY:
+	case VK_ERROR_OUT_OF_DEVICE_MEMORY:
+	case VK_ERROR_MEMORY_MAP_FAILED:
+	case VK_ERROR_OUT_OF_POOL_MEMORY:
+	case VK_ERROR_FRAGMENTATION_EXT:
+	case VK_ERROR_FRAGMENTED_POOL:
+		return ENOMEM;
+	case VK_ERROR_INITIALIZATION_FAILED:
+		return EIO;
+	case VK_ERROR_DEVICE_LOST:
+	case VK_ERROR_SURFACE_LOST_KHR:
+	case VK_ERROR_FULL_SCREEN_EXCLUSIVE_MODE_LOST_EXT:
+		return ENODEV;
+	case VK_ERROR_LAYER_NOT_PRESENT:
+	case VK_ERROR_EXTENSION_NOT_PRESENT:
+	case VK_ERROR_FEATURE_NOT_PRESENT:
+		return ENOENT;
+	case VK_ERROR_INCOMPATIBLE_DRIVER:
+	case VK_ERROR_FORMAT_NOT_SUPPORTED:
+	case VK_ERROR_INCOMPATIBLE_DISPLAY_KHR:
+		return ENOTSUP;
+	case VK_ERROR_TOO_MANY_OBJECTS:
+		return ENFILE;
+	case VK_SUBOPTIMAL_KHR:
+	case VK_ERROR_OUT_OF_DATE_KHR:
+		return EIO;
+	case VK_ERROR_INVALID_EXTERNAL_HANDLE:
+	case VK_ERROR_VALIDATION_FAILED_EXT:
+	case VK_ERROR_INVALID_SHADER_NV:
+	case VK_ERROR_INVALID_DRM_FORMAT_MODIFIER_PLANE_LAYOUT_EXT:
+	case VK_ERROR_INVALID_DEVICE_ADDRESS_EXT:
+		return EINVAL;
+	case VK_ERROR_NOT_PERMITTED_EXT:
+		return EPERM;
+	default:
+		return EIO;
+	}
+}
+
+// Used for validating return values of Vulkan API calls.
+#define VK_CHECK_RESULT(f)								\
+{											\
+    VkResult res = (f);									\
+    if (res != VK_SUCCESS)								\
+    {											\
+        printf("Fatal : VkResult is %d in %s at line %d\n", res,  __FILE__, __LINE__);	\
+        return -vkresult_to_errno(res);							\
+    }											\
+}
+
+static int createInstance(struct vulkan_state *d)
+{
+	const VkApplicationInfo applicationInfo = {
+		.sType = VK_STRUCTURE_TYPE_APPLICATION_INFO,
+		.pApplicationName = "Hello world app",
+		.applicationVersion = 0,
+		.pEngineName = "awesomeengine",
+		.engineVersion = 0,
+		.apiVersion = VK_API_VERSION_1_1
+	};
+	const char *extensions[] = {
+		VK_KHR_EXTERNAL_MEMORY_CAPABILITIES_EXTENSION_NAME
+	};
+        VkInstanceCreateInfo createInfo = {
+		.sType = VK_STRUCTURE_TYPE_INSTANCE_CREATE_INFO,
+		.pApplicationInfo = &applicationInfo,
+		.enabledExtensionCount = 1,
+		.ppEnabledExtensionNames = extensions,
+	};
+
+	VK_CHECK_RESULT(vkCreateInstance(&createInfo, NULL, &d->instance));
+
+	return 0;
+}
+
+static uint32_t getComputeQueueFamilyIndex(struct vulkan_state *d)
+{
+	uint32_t i, queueFamilyCount;
+	VkQueueFamilyProperties *queueFamilies;
+
+	vkGetPhysicalDeviceQueueFamilyProperties(d->physicalDevice, &queueFamilyCount, NULL);
+
+	queueFamilies = alloca(queueFamilyCount * sizeof(VkQueueFamilyProperties));
+	vkGetPhysicalDeviceQueueFamilyProperties(d->physicalDevice, &queueFamilyCount, queueFamilies);
+
+	for (i = 0; i < queueFamilyCount; i++) {
+		VkQueueFamilyProperties props = queueFamilies[i];
+
+		if (props.queueCount > 0 && (props.queueFlags & VK_QUEUE_COMPUTE_BIT))
+			break;
+	}
+	if (i == queueFamilyCount)
+		return -ENODEV;
+
+	return i;
+}
+
+static int findPhysicalDevice(struct vulkan_state *d)
+{
+	uint32_t deviceCount;
+        VkPhysicalDevice *devices;
+
+	vkEnumeratePhysicalDevices(d->instance, &deviceCount, NULL);
+	if (deviceCount == 0)
+		return -ENODEV;
+
+	devices = alloca(deviceCount * sizeof(VkPhysicalDevice));
+        vkEnumeratePhysicalDevices(d->instance, &deviceCount, devices);
+
+	d->physicalDevice = devices[0];
+
+	d->queueFamilyIndex = getComputeQueueFamilyIndex(d);
+
+	return 0;
+}
+
+static int createDevice(struct vulkan_state *d)
+{
+	VkDeviceQueueCreateInfo queueCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_DEVICE_QUEUE_CREATE_INFO,
+		.queueFamilyIndex = d->queueFamilyIndex,
+		.queueCount = 1,
+		.pQueuePriorities = (const float[]) { 1.0f }
+	};
+	const char *extensions[] = {
+		VK_KHR_EXTERNAL_MEMORY_EXTENSION_NAME,
+		VK_KHR_EXTERNAL_MEMORY_FD_EXTENSION_NAME
+	};
+	VkDeviceCreateInfo deviceCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_DEVICE_CREATE_INFO,
+		.queueCreateInfoCount = 1,
+		.pQueueCreateInfos = &queueCreateInfo,
+		.enabledExtensionCount = 2,
+		.ppEnabledExtensionNames = extensions,
+	};
+
+	VK_CHECK_RESULT(vkCreateDevice(d->physicalDevice, &deviceCreateInfo, NULL, &d->device));
+
+	vkGetDeviceQueue(d->device, d->queueFamilyIndex, 0, &d->queue);
+
+	VkFenceCreateInfo fenceCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_FENCE_CREATE_INFO,
+		.flags = 0,
+	};
+	VK_CHECK_RESULT(vkCreateFence(d->device, &fenceCreateInfo, NULL, &d->fence));
+
+	return 0;
+}
+
+static uint32_t findMemoryType(struct vulkan_state *d,
+		uint32_t memoryTypeBits, VkMemoryPropertyFlags properties)
+{
+	uint32_t i;
+	VkPhysicalDeviceMemoryProperties memoryProperties;
+
+	vkGetPhysicalDeviceMemoryProperties(d->physicalDevice, &memoryProperties);
+
+	for (i = 0; i < memoryProperties.memoryTypeCount; i++) {
+		if ((memoryTypeBits & (1 << i)) &&
+		    ((memoryProperties.memoryTypes[i].propertyFlags & properties) == properties))
+			return i;
+	}
+	return -1;
+}
+
+static int createDescriptors(struct vulkan_state *d)
+{
+	VkDescriptorPoolSize descriptorPoolSize = {
+		.type = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
+		.descriptorCount = 1
+	};
+	VkDescriptorPoolCreateInfo descriptorPoolCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_POOL_CREATE_INFO,
+		.maxSets = 1,
+		.poolSizeCount = 1,
+		.pPoolSizes = &descriptorPoolSize,
+	};
+
+        VK_CHECK_RESULT(vkCreateDescriptorPool(d->device,
+				&descriptorPoolCreateInfo, NULL,
+				&d->descriptorPool));
+
+	VkDescriptorSetLayoutBinding descriptorSetLayoutBinding = {
+		.binding = 0,
+		.descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
+		.descriptorCount = 1,
+		.stageFlags = VK_SHADER_STAGE_COMPUTE_BIT
+	};
+	VkDescriptorSetLayoutCreateInfo descriptorSetLayoutCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO,
+		.bindingCount = 1,
+		.pBindings = &descriptorSetLayoutBinding
+	};
+	VK_CHECK_RESULT(vkCreateDescriptorSetLayout(d->device,
+				&descriptorSetLayoutCreateInfo, NULL,
+				&d->descriptorSetLayout));
+
+	VkDescriptorSetAllocateInfo descriptorSetAllocateInfo = {
+		.sType = VK_STRUCTURE_TYPE_DESCRIPTOR_SET_ALLOCATE_INFO,
+		.descriptorPool = d->descriptorPool,
+		.descriptorSetCount = 1,
+		.pSetLayouts = &d->descriptorSetLayout
+	};
+
+	VK_CHECK_RESULT(vkAllocateDescriptorSets(d->device,
+				&descriptorSetAllocateInfo,
+				&d->descriptorSet));
+	return 0;
+}
+
+static int createBuffer(struct vulkan_state *d, uint32_t id)
+{
+	VkBufferCreateInfo bufferCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_BUFFER_CREATE_INFO,
+		.size = d->bufferSize,
+		.usage = VK_BUFFER_USAGE_STORAGE_BUFFER_BIT,
+		.sharingMode = VK_SHARING_MODE_EXCLUSIVE,
+	};
+	VkMemoryRequirements memoryRequirements;
+
+	VK_CHECK_RESULT(vkCreateBuffer(d->device,
+				&bufferCreateInfo, NULL, &d->buffers[id].buffer));
+
+	vkGetBufferMemoryRequirements(d->device,
+			d->buffers[id].buffer, &memoryRequirements);
+
+	VkMemoryAllocateInfo allocateInfo = {
+		.sType = VK_STRUCTURE_TYPE_MEMORY_ALLOCATE_INFO,
+		.allocationSize = memoryRequirements.size
+	};
+	allocateInfo.memoryTypeIndex = findMemoryType(d,
+			memoryRequirements.memoryTypeBits,
+			VK_MEMORY_PROPERTY_HOST_COHERENT_BIT |
+			VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT);
+
+	VK_CHECK_RESULT(vkAllocateMemory(d->device,
+				&allocateInfo, NULL, &d->buffers[id].memory));
+	VK_CHECK_RESULT(vkBindBufferMemory(d->device,
+				d->buffers[id].buffer, d->buffers[id].memory, 0));
+
+	return 0;
+}
+
+static int updateDescriptors(struct vulkan_state *d, uint32_t buffer_id)
+{
+	VkDescriptorBufferInfo descriptorBufferInfo = {
+		.buffer = d->buffers[buffer_id].buffer,
+		.offset = 0,
+		.range = d->bufferSize,
+	};
+	VkWriteDescriptorSet writeDescriptorSet = {
+		.sType = VK_STRUCTURE_TYPE_WRITE_DESCRIPTOR_SET,
+		.dstSet = d->descriptorSet,
+		.dstBinding = 0,
+		.descriptorCount = 1,
+		.descriptorType = VK_DESCRIPTOR_TYPE_STORAGE_BUFFER,
+		.pBufferInfo = &descriptorBufferInfo,
+	};
+	vkUpdateDescriptorSets(d->device, 1, &writeDescriptorSet, 0, NULL);
+
+	d->buffer_id = buffer_id;
+
+	return 0;
+}
+
+static VkShaderModule createShaderModule(struct vulkan_state *d, const char* shaderFile)
+{
+	VkShaderModule shaderModule = VK_NULL_HANDLE;
+	void *data;
+	int fd;
+	struct stat stat;
+
+	if ((fd = open(shaderFile, 0, O_RDONLY)) == -1) {
+		fprintf(stderr, "can't open %s: %m\n", shaderFile);
+		return NULL;
+	}
+	fstat(fd, &stat);
+
+	data = mmap(NULL, stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
+
+	VkShaderModuleCreateInfo shaderModuleCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_SHADER_MODULE_CREATE_INFO,
+		.codeSize = stat.st_size,
+		.pCode = data,
+	};
+	vkCreateShaderModule(d->device, &shaderModuleCreateInfo, 0, &shaderModule);
+
+	munmap(data, stat.st_size);
+	close(fd);
+
+	return shaderModule;
+}
+
+static int createComputePipeline(struct vulkan_state *d, const char *shader_file)
+{
+	const VkPushConstantRange range = {
+		.stageFlags = VK_PIPELINE_STAGE_VERTEX_SHADER_BIT,
+		.offset = 0,
+		.size = sizeof(struct push_constants)
+	};
+
+	VkPipelineLayoutCreateInfo pipelineLayoutCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO,
+		.setLayoutCount = 1,
+		.pSetLayouts = &d->descriptorSetLayout,
+		.pushConstantRangeCount = 1,
+		.pPushConstantRanges = &range,
+	};
+	VK_CHECK_RESULT(vkCreatePipelineLayout(d->device,
+				&pipelineLayoutCreateInfo, NULL,
+				&d->pipelineLayout));
+
+        d->computeShaderModule = createShaderModule(d, shader_file);
+	VkPipelineShaderStageCreateInfo shaderStageCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO,
+		.stage = VK_SHADER_STAGE_COMPUTE_BIT,
+		.module = d->computeShaderModule,
+		.pName = "main",
+	};
+	VkComputePipelineCreateInfo pipelineCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_COMPUTE_PIPELINE_CREATE_INFO,
+		.stage = shaderStageCreateInfo,
+		.layout = d->pipelineLayout,
+	};
+	VK_CHECK_RESULT(vkCreateComputePipelines(d->device, VK_NULL_HANDLE,
+				1, &pipelineCreateInfo, NULL,
+				&d->pipeline));
+	return 0;
+}
+
+static int createCommandBuffer(struct vulkan_state *d)
+{
+	VkCommandPoolCreateInfo commandPoolCreateInfo = {
+		.sType = VK_STRUCTURE_TYPE_COMMAND_POOL_CREATE_INFO,
+		.flags = VK_COMMAND_POOL_CREATE_RESET_COMMAND_BUFFER_BIT,
+		.queueFamilyIndex = d->queueFamilyIndex,
+	};
+        VK_CHECK_RESULT(vkCreateCommandPool(d->device,
+				&commandPoolCreateInfo, NULL,
+				&d->commandPool));
+
+	VkCommandBufferAllocateInfo commandBufferAllocateInfo = {
+		.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_ALLOCATE_INFO,
+		.commandPool = d->commandPool,
+		.level = VK_COMMAND_BUFFER_LEVEL_PRIMARY,
+		.commandBufferCount = 1,
+	};
+        VK_CHECK_RESULT(vkAllocateCommandBuffers(d->device,
+				&commandBufferAllocateInfo,
+				&d->commandBuffer));
+
+	return 0;
+}
+
+static int runCommandBuffer(struct vulkan_state *d)
+{
+	VkCommandBufferBeginInfo beginInfo = {
+		.sType = VK_STRUCTURE_TYPE_COMMAND_BUFFER_BEGIN_INFO,
+		.flags = VK_COMMAND_BUFFER_USAGE_ONE_TIME_SUBMIT_BIT,
+	};
+	VK_CHECK_RESULT(vkBeginCommandBuffer(d->commandBuffer, &beginInfo));
+
+	vkCmdBindPipeline(d->commandBuffer, VK_PIPELINE_BIND_POINT_COMPUTE, d->pipeline);
+	vkCmdPushConstants (d->commandBuffer,
+			d->pipelineLayout, VK_SHADER_STAGE_COMPUTE_BIT,
+			0, sizeof(struct push_constants), (const void *) &d->constants);
+	vkCmdBindDescriptorSets(d->commandBuffer, VK_PIPELINE_BIND_POINT_COMPUTE,
+			d->pipelineLayout, 0, 1, &d->descriptorSet, 0, NULL);
+
+	vkCmdDispatch(d->commandBuffer,
+			(uint32_t)ceil(d->constants.width / (float)WORKGROUP_SIZE),
+			(uint32_t)ceil(d->constants.height / (float)WORKGROUP_SIZE), 1);
+
+	VK_CHECK_RESULT(vkEndCommandBuffer(d->commandBuffer));
+
+	VK_CHECK_RESULT(vkResetFences(d->device, 1, &d->fence));
+
+	VkSubmitInfo submitInfo = {
+		.sType = VK_STRUCTURE_TYPE_SUBMIT_INFO,
+		.commandBufferCount = 1,
+		.pCommandBuffers = &d->commandBuffer,
+	};
+        VK_CHECK_RESULT(vkQueueSubmit(d->queue, 1, &submitInfo, d->fence));
+	d->busy = true;
+
+	return 0;
+}
+
+static void clear_buffers(struct vulkan_state *s)
+{
+	uint32_t i;
+
+	for (i = 0; i < s->n_buffers; i++) {
+		close(s->buffers[i].buf->datas[0].fd);
+		vkFreeMemory(s->device, s->buffers[i].memory, NULL);
+		vkDestroyBuffer(s->device, s->buffers[i].buffer, NULL);
+	}
+	s->n_buffers = 0;
+}
+
+int spa_vulkan_use_buffers(struct vulkan_state *s, uint32_t flags,
+		uint32_t n_buffers, struct spa_buffer **buffers)
+{
+	uint32_t i;
+	VULKAN_INSTANCE_FUNCTION(vkGetMemoryFdKHR);
+
+	clear_buffers(s);
+
+	s->bufferSize = s->constants.width * s->constants.height * sizeof(struct pixel);
+
+	for (i = 0; i < n_buffers; i++) {
+		createBuffer(s, i);
+
+		VkMemoryGetFdInfoKHR getFdInfo = {
+			.sType = VK_STRUCTURE_TYPE_MEMORY_GET_FD_INFO_KHR,
+			.memory = s->buffers[i].memory,
+			.handleType = VK_EXTERNAL_MEMORY_HANDLE_TYPE_DMA_BUF_BIT_EXT
+		};
+		int fd;
+
+		s->buffers[i].buf = buffers[i];
+
+	        VK_CHECK_RESULT(vkGetMemoryFdKHR(s->device, &getFdInfo, &fd));
+
+		buffers[i]->datas[0].type = SPA_DATA_DmaBuf;
+		buffers[i]->datas[0].flags = SPA_DATA_FLAG_READABLE;
+		buffers[i]->datas[0].fd = fd;
+		buffers[i]->datas[0].mapoffset = 0;
+		buffers[i]->datas[0].maxsize = s->bufferSize;
+	}
+	s->n_buffers = n_buffers;
+
+	return 0;
+}
+
+int spa_vulkan_prepare(struct vulkan_state *s)
+{
+	if (!s->prepared) {
+		createInstance(s);
+		findPhysicalDevice(s);
+		createDevice(s);
+		createDescriptors(s);
+		createComputePipeline(s, "spa/plugins/vulkan/shaders/main.spv");
+		createCommandBuffer(s);
+		s->prepared = true;
+	}
+	return 0;
+}
+
+int spa_vulkan_unprepare(struct vulkan_state *s)
+{
+	if (s->prepared) {
+		vkDestroyShaderModule(s->device, s->computeShaderModule, NULL);
+		vkDestroyDescriptorPool(s->device, s->descriptorPool, NULL);
+		vkDestroyDescriptorSetLayout(s->device, s->descriptorSetLayout, NULL);
+		vkDestroyPipelineLayout(s->device, s->pipelineLayout, NULL);
+		vkDestroyPipeline(s->device, s->pipeline, NULL);
+		vkDestroyCommandPool(s->device, s->commandPool, NULL);
+		vkDestroyDevice(s->device, NULL);
+		vkDestroyInstance(s->instance, NULL);
+		s->prepared = false;
+	}
+	return 0;
+}
+
+int spa_vulkan_start(struct vulkan_state *s)
+{
+	s->busy = false;
+	return 0;
+}
+
+int spa_vulkan_stop(struct vulkan_state *s)
+{
+        VK_CHECK_RESULT(vkDeviceWaitIdle(s->device));
+	return 0;
+}
+
+int spa_vulkan_ready(struct vulkan_state *s)
+{
+	VkResult result;
+
+	if (!s->busy)
+		return 0;
+
+	result = vkGetFenceStatus(s->device, s->fence);
+	if (result != VK_SUCCESS)
+		return -vkresult_to_errno(result);
+
+	s->busy = false;
+
+	return 0;
+}
+
+int spa_vulkan_process(struct vulkan_state *s, uint32_t buffer_id)
+{
+	if (buffer_id != s->buffer_id) {
+		updateDescriptors(s, buffer_id);
+		s->buffer_id = buffer_id;
+	}
+	runCommandBuffer(s);
+
+	return 0;
+}
diff --git a/spa/plugins/vulkan/vulkan-utils.h b/spa/plugins/vulkan/vulkan-utils.h
new file mode 100644
index 00000000..4ab95187
--- /dev/null
+++ b/spa/plugins/vulkan/vulkan-utils.h
@@ -0,0 +1,66 @@
+#include <vulkan/vulkan.h>
+
+#include <spa/buffer/buffer.h>
+
+#define MAX_BUFFERS 16
+#define WORKGROUP_SIZE 32
+
+struct pixel {
+	float r, g, b, a;
+};
+
+struct push_constants {
+	float time;
+	int frame;
+	int width;
+	int height;
+};
+
+struct vulkan_buffer {
+	struct spa_buffer *buf;
+	VkBuffer buffer;
+	VkDeviceMemory memory;
+};
+
+struct vulkan_state {
+	struct push_constants constants;
+
+	VkInstance instance;
+
+	VkPhysicalDevice physicalDevice;
+	VkDevice device;
+
+	VkPipeline pipeline;
+	VkPipelineLayout pipelineLayout;
+	VkShaderModule computeShaderModule;
+
+	VkCommandPool commandPool;
+	VkCommandBuffer commandBuffer;
+
+	VkQueue queue;
+	uint32_t queueFamilyIndex;
+	VkFence fence;
+	unsigned int prepared:1;
+	unsigned int busy:1;
+
+	VkDescriptorPool descriptorPool;
+	VkDescriptorSet descriptorSet;
+	VkDescriptorSetLayout descriptorSetLayout;
+
+	struct vulkan_buffer buffers[MAX_BUFFERS];
+	uint32_t n_buffers;
+
+	uint32_t bufferSize;
+	uint32_t buffer_id;
+};
+
+int spa_vulkan_prepare(struct vulkan_state *s);
+int spa_vulkan_use_buffers(struct vulkan_state *s, uint32_t flags,
+		uint32_t n_buffers, struct spa_buffer **buffers);
+int spa_vulkan_unprepare(struct vulkan_state *s);
+
+int spa_vulkan_start(struct vulkan_state *s);
+int spa_vulkan_stop(struct vulkan_state *s);
+int spa_vulkan_ready(struct vulkan_state *s);
+int spa_vulkan_process(struct vulkan_state *s, uint32_t buffer_id);
+int spa_vulkan_cleanup(struct vulkan_state *s);
diff --git a/src/daemon/pipewire.conf.in b/src/daemon/pipewire.conf.in
index 347dccc1..1d6833e4 100644
--- a/src/daemon/pipewire.conf.in
+++ b/src/daemon/pipewire.conf.in
@@ -7,10 +7,12 @@ add-spa-lib audio.convert* audioconvert/libspa-audioconvert
 add-spa-lib api.alsa.* alsa/libspa-alsa
 add-spa-lib api.v4l2.* v4l2/libspa-v4l2
 add-spa-lib api.bluez5.* bluez5/libspa-bluez5
+add-spa-lib api.vulkan.* vulkan/libspa-vulkan
 
 load-module libpipewire-module-rtkit
 load-module libpipewire-module-protocol-native
 load-module libpipewire-module-spa-node-factory
+load-module libpipewire-module-spa-node api.vulkan.compute.source node.name=my-compute-source
 #load-module libpipewire-module-spa-node videotestsrc/libspa-videotestsrc videotestsrc videotestsrc Spa:POD:Object:Props:patternType=Spa:POD:Object:Props:patternType:snow
 load-module libpipewire-module-client-node
 load-module libpipewire-module-client-device