2018-11-09 19:21:17 +08:00
|
|
|
#ifndef RECORDER_H
|
|
|
|
#define RECORDER_H
|
|
|
|
|
2021-01-09 02:24:51 +08:00
|
|
|
#include "common.h"
|
|
|
|
|
2019-03-03 06:52:22 +08:00
|
|
|
#include <stdbool.h>
|
2018-11-09 19:21:17 +08:00
|
|
|
#include <libavformat/avformat.h>
|
|
|
|
|
2021-01-09 02:15:29 +08:00
|
|
|
#include "coords.h"
|
2020-06-20 04:04:06 +08:00
|
|
|
#include "scrcpy.h"
|
2021-04-11 21:01:05 +08:00
|
|
|
#include "trait/packet_sink.h"
|
2019-11-24 18:53:00 +08:00
|
|
|
#include "util/queue.h"
|
2021-02-01 01:24:35 +08:00
|
|
|
#include "util/thread.h"
|
2018-11-09 19:21:17 +08:00
|
|
|
|
2019-07-31 07:55:40 +08:00
|
|
|
struct record_packet {
|
2021-06-14 15:07:49 +08:00
|
|
|
AVPacket *packet;
|
2019-07-31 07:55:40 +08:00
|
|
|
struct record_packet *next;
|
|
|
|
};
|
|
|
|
|
2019-08-02 05:15:47 +08:00
|
|
|
struct recorder_queue QUEUE(struct record_packet);
|
2019-07-31 07:55:40 +08:00
|
|
|
|
2018-11-09 19:21:17 +08:00
|
|
|
struct recorder {
|
2021-04-11 21:01:05 +08:00
|
|
|
struct sc_packet_sink packet_sink; // packet sink trait
|
|
|
|
|
2018-11-09 19:21:17 +08:00
|
|
|
char *filename;
|
2020-06-20 04:04:06 +08:00
|
|
|
enum sc_record_format format;
|
2018-11-09 19:21:17 +08:00
|
|
|
AVFormatContext *ctx;
|
|
|
|
struct size declared_frame_size;
|
2019-03-03 06:52:22 +08:00
|
|
|
bool header_written;
|
2019-07-31 07:55:40 +08:00
|
|
|
|
2021-02-01 01:24:35 +08:00
|
|
|
sc_thread thread;
|
|
|
|
sc_mutex mutex;
|
|
|
|
sc_cond queue_cond;
|
2021-04-11 21:01:05 +08:00
|
|
|
bool stopped; // set on recorder_close()
|
2019-07-31 07:55:40 +08:00
|
|
|
bool failed; // set on packet write failure
|
|
|
|
struct recorder_queue queue;
|
2019-08-09 00:57:20 +08:00
|
|
|
|
|
|
|
// we can write a packet only once we received the next one so that we can
|
|
|
|
// set its duration (next_pts - current_pts)
|
|
|
|
// "previous" is only accessed from the recorder thread, so it does not
|
|
|
|
// need to be protected by the mutex
|
|
|
|
struct record_packet *previous;
|
2018-11-09 19:21:17 +08:00
|
|
|
};
|
|
|
|
|
2019-03-03 06:52:22 +08:00
|
|
|
bool
|
2019-05-31 22:08:30 +08:00
|
|
|
recorder_init(struct recorder *recorder, const char *filename,
|
2020-06-20 04:04:06 +08:00
|
|
|
enum sc_record_format format, struct size declared_frame_size);
|
2019-02-09 22:20:07 +08:00
|
|
|
|
2019-03-03 03:09:56 +08:00
|
|
|
void
|
|
|
|
recorder_destroy(struct recorder *recorder);
|
2018-11-09 19:21:17 +08:00
|
|
|
|
|
|
|
#endif
|