2019-03-02 22:16:55 +08:00
|
|
|
#ifndef VIDEO_BUFFER_H
|
|
|
|
#define VIDEO_BUFFER_H
|
2017-12-12 22:12:07 +08:00
|
|
|
|
2021-01-09 02:24:51 +08:00
|
|
|
#include "common.h"
|
|
|
|
|
2019-03-03 06:52:22 +08:00
|
|
|
#include <stdbool.h>
|
2017-12-12 22:12:07 +08:00
|
|
|
|
2018-08-15 23:01:54 +08:00
|
|
|
#include "fps_counter.h"
|
2021-02-01 01:24:35 +08:00
|
|
|
#include "util/thread.h"
|
2018-02-07 19:25:52 +08:00
|
|
|
|
2017-12-12 22:12:07 +08:00
|
|
|
// forward declarations
|
|
|
|
typedef struct AVFrame AVFrame;
|
|
|
|
|
2021-02-01 16:38:11 +08:00
|
|
|
/**
|
|
|
|
* There are 3 frames in memory:
|
2021-02-20 04:16:57 +08:00
|
|
|
* - one frame is held by the producer (producer_frame)
|
|
|
|
* - one frame is held by the consumer (consumer_frame)
|
|
|
|
* - one frame is shared between the producer and the consumer (pending_frame)
|
2021-02-01 16:38:11 +08:00
|
|
|
*
|
2021-02-20 04:16:57 +08:00
|
|
|
* The producer generates a frame into the producer_frame (it may takes time).
|
2021-02-01 16:38:11 +08:00
|
|
|
*
|
2021-02-20 04:16:57 +08:00
|
|
|
* Once the frame is produced, it calls video_buffer_producer_offer_frame(),
|
|
|
|
* which swaps the producer and pending frames.
|
2021-02-01 16:38:11 +08:00
|
|
|
*
|
2021-02-20 04:16:57 +08:00
|
|
|
* When the consumer is notified that a new frame is available, it calls
|
|
|
|
* video_buffer_consumer_take_frame() to retrieve it, which swaps the pending
|
|
|
|
* and consumer frames. The frame is valid until the next call, without
|
|
|
|
* blocking the producer.
|
2021-02-01 16:38:11 +08:00
|
|
|
*/
|
|
|
|
|
2019-03-02 22:16:55 +08:00
|
|
|
struct video_buffer {
|
2021-02-20 04:16:57 +08:00
|
|
|
AVFrame *producer_frame;
|
2021-02-01 16:38:11 +08:00
|
|
|
AVFrame *pending_frame;
|
2021-02-20 04:16:57 +08:00
|
|
|
AVFrame *consumer_frame;
|
2021-02-01 16:38:11 +08:00
|
|
|
|
2021-02-01 01:24:35 +08:00
|
|
|
sc_mutex mutex;
|
2021-02-20 04:16:57 +08:00
|
|
|
bool wait_consumer; // never overwrite a pending frame if it is not consumed
|
2019-03-03 06:52:22 +08:00
|
|
|
bool interrupted;
|
2021-02-01 16:38:11 +08:00
|
|
|
|
|
|
|
sc_cond pending_frame_consumed_cond;
|
|
|
|
bool pending_frame_consumed;
|
2017-12-12 22:12:07 +08:00
|
|
|
};
|
|
|
|
|
2019-03-03 06:52:22 +08:00
|
|
|
bool
|
2021-02-20 04:16:57 +08:00
|
|
|
video_buffer_init(struct video_buffer *vb, bool wait_consumer);
|
2019-03-03 03:09:56 +08:00
|
|
|
|
|
|
|
void
|
|
|
|
video_buffer_destroy(struct video_buffer *vb);
|
2017-12-12 22:12:07 +08:00
|
|
|
|
2021-02-20 04:16:57 +08:00
|
|
|
// set the producer frame as ready for consuming
|
2019-03-03 07:26:48 +08:00
|
|
|
// the output flag is set to report whether the previous frame has been skipped
|
|
|
|
void
|
2021-02-20 04:16:57 +08:00
|
|
|
video_buffer_producer_offer_frame(struct video_buffer *vb,
|
|
|
|
bool *previous_frame_skipped);
|
2018-02-09 02:23:24 +08:00
|
|
|
|
2021-02-20 04:16:57 +08:00
|
|
|
// mark the consumer frame as consumed and return it
|
2021-02-01 16:38:11 +08:00
|
|
|
// the frame is valid until the next call to this function
|
2019-03-03 03:09:56 +08:00
|
|
|
const AVFrame *
|
2021-02-20 04:16:57 +08:00
|
|
|
video_buffer_consumer_take_frame(struct video_buffer *vb);
|
2017-12-12 22:12:07 +08:00
|
|
|
|
2018-02-09 15:42:39 +08:00
|
|
|
// wake up and avoid any blocking call
|
2019-03-03 03:09:56 +08:00
|
|
|
void
|
|
|
|
video_buffer_interrupt(struct video_buffer *vb);
|
2018-02-09 15:42:39 +08:00
|
|
|
|
2017-12-12 22:12:07 +08:00
|
|
|
#endif
|