PipeWire  0.3.51
Tutorial - Part 5: Capturing Video Frames

Tutorial - Part 4: Playing A Tone | Index | Tutorial - Part 6: Binding Objects

In this tutorial we show how to use a stream to capture a stream of video frames.

Even though we are now working with a different media type and we are capturing instead of playback, you will see that this example is very similar to Tutorial - Part 4: Playing A Tone.

Let's take a look at the code before we break it down:

struct data {
struct pw_main_loop *loop;
struct pw_stream *stream;
struct spa_video_info format;
};
/* [on_process] */
static void on_process(void *userdata)
{
struct data *data = userdata;
struct pw_buffer *b;
struct spa_buffer *buf;
if ((b = pw_stream_dequeue_buffer(data->stream)) == NULL) {
pw_log_warn("out of buffers: %m");
return;
}
buf = b->buffer;
if (buf->datas[0].data == NULL)
return;
printf("got a frame of size %d\n", buf->datas[0].chunk->size);
pw_stream_queue_buffer(data->stream, b);
}
/* [on_process] */
static void on_param_changed(void *userdata, uint32_t id, const struct spa_pod *param)
{
struct data *data = userdata;
if (param == NULL || id != SPA_PARAM_Format)
return;
if (spa_format_parse(param,
&data->format.media_type,
&data->format.media_subtype) < 0)
return;
if (data->format.media_type != SPA_MEDIA_TYPE_video ||
data->format.media_subtype != SPA_MEDIA_SUBTYPE_raw)
return;
if (spa_format_video_raw_parse(param, &data->format.info.raw) < 0)
return;
printf("got video format:\n");
printf(" format: %d (%s)\n", data->format.info.raw.format,
data->format.info.raw.format));
printf(" size: %dx%d\n", data->format.info.raw.size.width,
data->format.info.raw.size.height);
printf(" framerate: %d/%d\n", data->format.info.raw.framerate.num,
data->format.info.raw.framerate.denom);
}
static const struct pw_stream_events stream_events = {
.param_changed = on_param_changed,
.process = on_process,
};
int main(int argc, char *argv[])
{
struct data data = { 0, };
const struct spa_pod *params[1];
uint8_t buffer[1024];
struct spa_pod_builder b = SPA_POD_BUILDER_INIT(buffer, sizeof(buffer));
pw_init(&argc, &argv);
data.loop = pw_main_loop_new(NULL);
"video-capture",
PW_KEY_MEDIA_ROLE, "Camera",
NULL),
&stream_events,
&data);
&SPA_RECTANGLE(320, 240),
&SPA_RECTANGLE(1, 1),
&SPA_RECTANGLE(4096, 4096)),
&SPA_FRACTION(25, 1),
&SPA_FRACTION(0, 1),
&SPA_FRACTION(1000, 1)));
argc > 1 ? (uint32_t)atoi(argv[1]) : PW_ID_ANY,
params, 1);
return 0;
}
#define PW_ID_ANY
Definition: core.h:83
#define PW_KEY_MEDIA_TYPE
Media.
Definition: keys.h:429
#define PW_KEY_MEDIA_ROLE
Role: Movie, Music, Camera, Screen, Communication, Game, Notification, DSP, Production,...
Definition: keys.h:435
#define PW_KEY_MEDIA_CATEGORY
Media Category: Playback, Capture, Duplex, Monitor, Manager.
Definition: keys.h:432
#define pw_log_warn(...)
Definition: log.h:163
struct pw_main_loop * pw_main_loop_new(const struct spa_dict *props)
Create a new main loop.
Definition: main-loop.c:80
void pw_main_loop_destroy(struct pw_main_loop *loop)
Destroy a loop.
Definition: main-loop.c:90
int pw_main_loop_run(struct pw_main_loop *loop)
Run a main loop.
Definition: main-loop.c:139
struct pw_loop * pw_main_loop_get_loop(struct pw_main_loop *loop)
Get the loop implementation.
Definition: main-loop.c:113
void pw_init(int *argc, char **argv[])
Initialize PipeWire.
Definition: pipewire.c:580
#define PW_DIRECTION_INPUT
Definition: port.h:65
struct pw_properties * pw_properties_new(const char *key,...) 1
Make a new properties object.
Definition: properties.c:102
int pw_stream_connect(struct pw_stream *stream, enum pw_direction direction, uint32_t target_id, enum pw_stream_flags flags, const struct spa_pod **params, uint32_t n_params)
Connect a stream for input or output on port_path.
Definition: stream.c:1767
struct pw_stream * pw_stream_new_simple(struct pw_loop *loop, const char *name, struct pw_properties *props, const struct pw_stream_events *events, void *data)
Definition: stream.c:1514
struct pw_buffer * pw_stream_dequeue_buffer(struct pw_stream *stream)
Get a buffer that can be filled for playback streams or consumed for capture streams.
Definition: stream.c:2220
int pw_stream_queue_buffer(struct pw_stream *stream, struct pw_buffer *buffer)
Submit a buffer for playback or recycle a buffer for capture.
Definition: stream.c:2247
#define PW_VERSION_STREAM_EVENTS
Definition: stream.h:310
void pw_stream_destroy(struct pw_stream *stream)
Destroy a stream.
Definition: stream.c:1576
@ PW_STREAM_FLAG_MAP_BUFFERS
mmap the buffers except DmaBuf
Definition: stream.h:358
@ PW_STREAM_FLAG_AUTOCONNECT
try to automatically connect this stream
Definition: stream.h:353
static const char * spa_debug_type_find_name(const struct spa_type_info *info, uint32_t type)
Definition: types.h:73
static int spa_format_parse(const struct spa_pod *format, uint32_t *media_type, uint32_t *media_subtype)
Definition: format-utils.h:47
static const struct spa_type_info spa_type_video_format[]
Definition: type-info.h:48
static int spa_format_video_raw_parse(const struct spa_pod *format, struct spa_video_info_raw *info)
Definition: format-utils.h:47
@ SPA_MEDIA_TYPE_video
Definition: format.h:48
@ SPA_PARAM_Format
configured format as SPA_TYPE_OBJECT_Format
Definition: param.h:54
@ SPA_PARAM_EnumFormat
available formats as SPA_TYPE_OBJECT_Format
Definition: param.h:53
@ SPA_FORMAT_VIDEO_framerate
frame rate (Fraction)
Definition: format.h:131
@ SPA_FORMAT_mediaType
media type (Id enum spa_media_type)
Definition: format.h:109
@ SPA_FORMAT_VIDEO_size
size (Rectangle)
Definition: format.h:130
@ SPA_FORMAT_VIDEO_format
video format (Id enum spa_video_format)
Definition: format.h:127
@ SPA_FORMAT_mediaSubtype
media subtype (Id enum spa_media_subtype)
Definition: format.h:110
@ SPA_MEDIA_SUBTYPE_raw
Definition: format.h:58
@ SPA_VIDEO_FORMAT_YUY2
Definition: raw.h:69
@ SPA_VIDEO_FORMAT_RGBA
Definition: raw.h:76
@ SPA_VIDEO_FORMAT_RGBx
Definition: raw.h:72
@ SPA_VIDEO_FORMAT_BGRx
Definition: raw.h:73
@ SPA_VIDEO_FORMAT_I420
Definition: raw.h:67
@ SPA_VIDEO_FORMAT_RGB
Definition: raw.h:80
#define SPA_POD_CHOICE_ENUM_Id(n_vals,...)
Definition: vararg.h:71
#define SPA_POD_CHOICE_RANGE_Fraction(def, min, max)
Definition: vararg.h:135
#define SPA_POD_Id(val)
Definition: vararg.h:69
#define SPA_POD_BUILDER_INIT(buffer, size)
Definition: builder.h:82
#define SPA_POD_CHOICE_RANGE_Rectangle(def, min, max)
Definition: vararg.h:126
#define spa_pod_builder_add_object(b, type, id,...)
Definition: builder.h:677
@ SPA_TYPE_OBJECT_Format
Definition: type.h:96
#define SPA_FRACTION(num, denom)
Definition: defs.h:120
#define SPA_RECTANGLE(width, height)
Definition: defs.h:99
spa/param/video/type-info.h
pipewire/pipewire.h
a buffer structure obtained from pw_stream_dequeue_buffer().
Definition: stream.h:183
struct spa_buffer * buffer
the spa buffer
Definition: stream.h:184
A main loop object.
Events for a stream.
Definition: stream.h:308
A Buffer.
Definition: buffer.h:105
struct spa_data * datas
array of data members
Definition: buffer.h:109
uint32_t size
size of valid data.
Definition: buffer.h:66
struct spa_chunk * chunk
valid chunk of memory
Definition: buffer.h:101
void * data
optional data pointer
Definition: buffer.h:100
Definition: builder.h:73
void * data
Definition: builder.h:74
Definition: pod.h:63
Definition: format.h:45
spa/debug/types.h

Save as tutorial5.c and compile with:

gcc -Wall tutorial5.c -o tutorial5 -lm $(pkg-config --cflags --libs libpipewire-0.3)

Most of the application is structured like Tutorial - Part 4: Playing A Tone.

We create a stream object with different properties to make it a Camera Video Capture stream.

"video-capture",
PW_KEY_MEDIA_ROLE, "Camera",
NULL),
&stream_events,
&data);

In addition to the process event, we are also going to listen to a new event, param_changed:

static const struct pw_stream_events stream_events = {
.param_changed = on_param_changed,
.process = on_process,
};

Because we capture a stream of a wide range of different video formats and resolutions, we have to describe our accepted formats in a different way:

This is using a struct spa_pod_builder to make a struct spa_pod * object in the buffer array on the stack. The parameter is of type SPA_PARAM_EnumFormat which means that it enumerates the possible formats for this stream.

In this example we use the builder to create some CHOICE entries for the format properties.

We have an enumeration of formats, we need to first give the amount of enumerations that follow, then the default (preferred) value, followed by alternatives in order of preference:

We also have a RANGE of values for the size. We need to give a default (preferred) size and then a min and max value:

&SPA_RECTANGLE(320, 240), /* default */
&SPA_RECTANGLE(1, 1), /* min */
&SPA_RECTANGLE(4096, 4096)), /* max */

We have something similar for the framerate.

Note that there are other video parameters that we don't specify here. This means that we don't have any restrictions for their values.

See SPA POD for more information about how to make these POD objects.

Now we're ready to connect the stream and run the main loop:

To connect we specify that we have a PW_DIRECTION_INPUT stream. PW_ID_ANY means that we are ok with connecting to any producer. We also allow the user to pass an optional target id.

We're setting the PW_STREAM_FLAG_AUTOCONNECT flag to make an automatic connection to a suitable camera and PW_STREAM_FLAG_MAP_BUFFERS to let the stream mmap the data for us.

And last we pass the extra parameters for our stream. Here we only have the allowed formats (SPA_PARAM_EnumFormat).

Running the mainloop will start the connection and negotiation process. First our param_changed event will be called with the format that was negotiated between our stream and the camera. This is always something that is compatible with what we enumerated in the EnumFormat param when we connected.

Let's take a look at how we can parse the format in the param_changed event:

static void on_param_changed(void *userdata, uint32_t id, const struct spa_pod *param)
{
struct data *data = userdata;
if (param == NULL || id != SPA_PARAM_Format)
return;

First check if there is a param. A NULL param means that it is cleared. The ID of the param tells you what param it is. We are only interested in Format param (SPA_PARAM_Format).

We can parse the media type and subtype as below and ensure that it is of the right type. In our example this will always be true but when your EnumFormat contains different media types or subtypes, this is how you can parse them:

if (spa_format_parse(param,
&data->format.media_type,
&data->format.media_subtype) < 0)
return;
if (data->format.media_type != SPA_MEDIA_TYPE_video ||
data->format.media_subtype != SPA_MEDIA_SUBTYPE_raw)
return;

For the video/raw media type/subtype there is a utility function to parse out the values into a struct spa_video_info. This makes it easier to deal with.

if (spa_format_video_raw_parse(param, &data->format.info.raw) < 0)
return;
printf("got video format:\n");
printf(" format: %d (%s)\n", data->format.info.raw.format,
data->format.info.raw.format));
printf(" size: %dx%d\n", data->format.info.raw.size.width,
data->format.info.raw.size.height);
printf(" framerate: %d/%d\n", data->format.info.raw.framerate.num,
data->format.info.raw.framerate.denom);
}

In this example we dump the video size and parameters but in a real playback or capture application you might want to set up the screen or encoder to deal with the format.

After negotiation, the process function is called for each new frame. Check out Tutorial - Part 4: Playing A Tone for another example.

static void on_process(void *userdata)
{
struct data *data = userdata;
struct pw_buffer *b;
struct spa_buffer *buf;
if ((b = pw_stream_dequeue_buffer(data->stream)) == NULL) {
pw_log_warn("out of buffers: %m");
return;
}
buf = b->buffer;
if (buf->datas[0].data == NULL)
return;
printf("got a frame of size %d\n", buf->datas[0].chunk->size);
pw_stream_queue_buffer(data->stream, b);
}

In a real playback application, one would do something with the data, like copy it to the screen or encode it into a file.

Tutorial - Part 4: Playing A Tone | Index | Tutorial - Part 6: Binding Objects