PapaTutuWawa/gpu-screen-recorder
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix timestamp and video duration: Add header and trailer to output file

Browse Source
This commit is contained in:
dec05eba 2020-03-31 19:02:52 +02:00
parent ac7eec12f0
commit 9635df6287
1 changed files with 29 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/main.cpp
View File

@ -224,7 +224,8 @@ static void receive_frames(AVCodecContext *av_codec_context, AVStream *stream,
av_packet.stream_index = stream->index; av_packet.stream_index = stream->index;
// Write the encoded video frame to disk // Write the encoded video frame to disk
// av_write_frame(av_format_context, &av_packet) // av_write_frame(av_format_context, &av_packet)
if (write(STDOUT_FILENO, av_packet.data, av_packet.size) < 0) { // write(STDOUT_FILENO, av_packet.data, av_packet.size)
if (av_write_frame(av_format_context, &av_packet) < 0) {
fprintf(stderr, "Error: Failed to write frame to muxer\n"); fprintf(stderr, "Error: Failed to write frame to muxer\n");
} }
// av_packet_unref(&av_packet); // av_packet_unref(&av_packet);
@ -417,6 +418,8 @@ int main(int argc, char **argv) {
return 1; return 1;
} }
const char *filename = "/dev/stdout";
const float target_fps = 1.0f / (float)fps; const float target_fps = 1.0f / (float)fps;
Display *dpy = XOpenDisplay(nullptr); Display *dpy = XOpenDisplay(nullptr);
@ -513,7 +516,23 @@ int main(int argc, char **argv) {
open_video(video_codec, video_stream, window_pixmap, &device_ctx, open_video(video_codec, video_stream, window_pixmap, &device_ctx,
&cuda_graphics_resource); &cuda_graphics_resource);
output_format->flags = AVFMT_NOFILE; av_dump_format(av_format_context, 0, filename, 1);
if (!(output_format->flags & AVFMT_NOFILE)) {
int ret = avio_open(&av_format_context->pb, filename, AVIO_FLAG_WRITE);
if (ret < 0) {
fprintf(stderr, "Error: Could not open '%s': %s\n", filename,
"blabla"); // av_err2str(ret));
return 1;
}
}
int ret = avformat_write_header(av_format_context, nullptr);
if (ret < 0) {
fprintf(stderr, "Error occurred when opening output file: %s\n",
"blabla"); // av_err2str(ret));
return 1;
}
AVHWDeviceContext *hw_device_context = AVHWDeviceContext *hw_device_context =
(AVHWDeviceContext *)device_ctx->data; (AVHWDeviceContext *)device_ctx->data;
@ -666,15 +685,21 @@ int main(int argc, char **argv) {
usleep(5000); usleep(5000);
} }
if (av_write_trailer(av_format_context) != 0) {
fprintf(stderr, "Failed to write trailer\n");
}
/* add sequence end code to have a real MPEG file */ /* add sequence end code to have a real MPEG file */
/*
const uint8_t endcode[] = { 0, 0, 1, 0xb7 }; const uint8_t endcode[] = { 0, 0, 1, 0xb7 };
if (video_codec->id == AV_CODEC_ID_MPEG1VIDEO || video_codec->id == AV_CODEC_ID_MPEG2VIDEO) if (video_codec->id == AV_CODEC_ID_MPEG1VIDEO || video_codec->id == AV_CODEC_ID_MPEG2VIDEO)
write(STDOUT_FILENO, endcode, sizeof(endcode)); write(STDOUT_FILENO, endcode, sizeof(endcode));
*/
// close_video(video_stream, NULL); // close_video(video_stream, NULL);
// if(!(output_format->flags & AVFMT_NOFILE)) if(!(output_format->flags & AVFMT_NOFILE))
// avio_close(av_format_context->pb); avio_close(av_format_context->pb);
// avformat_free_context(av_format_context); // avformat_free_context(av_format_context);
// XDamageDestroy(dpy, xdamage); // XDamageDestroy(dpy, xdamage);