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

Add medium quality

Browse Source
This commit is contained in:
dec05eba 2020-08-31 23:22:14 +02:00
parent c94ce44dbe
commit 5537ebad6d
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/main.cpp
View File

@ -88,6 +88,7 @@ struct WindowPixmap {
}; };
enum class VideoQuality { enum class VideoQuality {
MEDIUM,
HIGH, HIGH,
ULTRA ULTRA
}; };
@ -449,6 +450,14 @@ static AVStream *add_video_stream(AVFormatContext *av_format_context, AVCodec **
codec_context->pix_fmt = AV_PIX_FMT_CUDA; codec_context->pix_fmt = AV_PIX_FMT_CUDA;
codec_context->color_range = AVCOL_RANGE_JPEG; codec_context->color_range = AVCOL_RANGE_JPEG;
switch(video_quality) { switch(video_quality) {
case VideoQuality::MEDIUM:
codec_context->bit_rate = 5000000 + (codec_context->width * codec_context->height) / 2;
codec_context->qmin = 16;
codec_context->qmax = 23;
//av_opt_set(codec_context->priv_data, "preset", "slow", 0);
//av_opt_set(codec_context->priv_data, "profile", "high", 0);
//codec_context->profile = FF_PROFILE_H264_HIGH;
break;
case VideoQuality::HIGH: case VideoQuality::HIGH:
codec_context->qmin = 12; codec_context->qmin = 12;
codec_context->qmax = 18; codec_context->qmax = 18;
@ -668,12 +677,14 @@ int main(int argc, char **argv) {
quality_str = "high"; quality_str = "high";
VideoQuality quality; VideoQuality quality;
if(strcmp(quality_str, "high") == 0) { if(strcmp(quality_str, "medium") == 0) {
quality = VideoQuality::MEDIUM;
} else if(strcmp(quality_str, "high") == 0) {
quality = VideoQuality::HIGH; quality = VideoQuality::HIGH;
} else if(strcmp(quality_str, "ultra") == 0) { } else if(strcmp(quality_str, "ultra") == 0) {
quality = VideoQuality::ULTRA; quality = VideoQuality::ULTRA;
} else { } else {
fprintf(stderr, "Error: -q should either be 'high' or 'ultra', got: %s\n", quality_str); fprintf(stderr, "Error: -q should either be either 'medium', 'high' or 'ultra', got: '%s'\n", quality_str);
usage(); usage();
} }