1 #include "video_encoder.h"
6 #include "quicksync_encoder.h"
12 string generate_local_dump_filename(int frame)
14 time_t now = time(NULL);
16 localtime_r(&now, &now_tm);
19 strftime(timestamp, sizeof(timestamp), "%F-%T%z", &now_tm);
21 // Use the frame number to disambiguate between two cuts starting
22 // on the same second.
24 snprintf(filename, sizeof(filename), "%s%s-f%02d%s",
25 LOCAL_DUMP_PREFIX, timestamp, frame % 100, LOCAL_DUMP_SUFFIX);
31 VideoEncoder::VideoEncoder(QSurface *surface, const std::string &va_display, int width, int height, HTTPD *httpd)
32 : surface(surface), va_display(va_display), width(width), height(height), httpd(httpd)
34 quicksync_encoder.reset(new QuickSyncEncoder(surface, va_display, width, height, httpd));
35 quicksync_encoder->open_output_file(generate_local_dump_filename(/*frame=*/0).c_str());
38 VideoEncoder::~VideoEncoder()
40 quicksync_encoder.reset(nullptr);
43 void VideoEncoder::do_cut(int frame)
45 string filename = generate_local_dump_filename(frame);
46 printf("Starting new recording: %s\n", filename.c_str());
47 quicksync_encoder->close_output_file();
48 quicksync_encoder->shutdown();
49 quicksync_encoder.reset(new QuickSyncEncoder(surface, va_display, width, height, httpd));
50 quicksync_encoder->open_output_file(filename.c_str());
53 void VideoEncoder::add_audio(int64_t pts, std::vector<float> audio)
55 quicksync_encoder->add_audio(pts, audio);
58 bool VideoEncoder::begin_frame(GLuint *y_tex, GLuint *cbcr_tex)
60 return quicksync_encoder->begin_frame(y_tex, cbcr_tex);
63 RefCountedGLsync VideoEncoder::end_frame(int64_t pts, int64_t duration, const std::vector<RefCountedFrame> &input_frames)
65 return quicksync_encoder->end_frame(pts, duration, input_frames);