Disable YUV->RGB conversion controls with non-YUV sources (#649).

[dcpomatic.git] / src / lib / ffmpeg_examiner.cc

diff --git a/src/lib/ffmpeg_examiner.cc b/src/lib/ffmpeg_examiner.cc
index 40fe0d28ec5cae2e6bc89d800dd9ef57bb195402..719e7a2b0863e085fcb6a0308d73e345c8df1bcb 100644 (file)
--- a/src/lib/ffmpeg_examiner.cc
+++ b/src/lib/ffmpeg_examiner.cc
@@ -76,13 +76,14 @@ FFmpegExaminer::FFmpegExaminer (shared_ptr<const FFmpegContent> c, shared_ptr<Jo
        _need_video_length = _format_context->duration == AV_NOPTS_VALUE;
        if (!_need_video_length) {
                _video_length = (double (_format_context->duration) / AV_TIME_BASE) * video_frame_rate().get ();
-       } else if (job) {
-               job->sub (_("Finding length"));
-               job->set_progress_unknown ();
        }
 
        if (job) {
-               job->sub (_("Finding subtitles"));
+               if (_need_video_length) {
+                       job->sub (_("Finding length and subtitles"));
+               } else {
+                       job->sub (_("Finding subtitles"));
+               }
        }
 
        /* Run through until we find:
@@ -95,6 +96,7 @@ FFmpegExaminer::FFmpegExaminer (shared_ptr<const FFmpegContent> c, shared_ptr<Jo
         * so they are ok).
         */
 
+       int64_t const len = _file_group.length ();
        while (true) {
                int r = av_read_frame (_format_context, &_packet);
                if (r < 0) {
@@ -102,7 +104,11 @@ FFmpegExaminer::FFmpegExaminer (shared_ptr<const FFmpegContent> c, shared_ptr<Jo
                }
 
                if (job) {
-                       job->set_progress_unknown ();
+                       if (len > 0) {
+                               job->set_progress (float (_format_context->pb->pos) / len);
+                       } else {
+                               job->set_progress_unknown ();
+                       }
                }
 
                AVCodecContext* context = _format_context->streams[_packet.stream_index]->codec;
@@ -326,3 +332,86 @@ FFmpegExaminer::bits_per_pixel () const
 {
        return av_get_bits_per_pixel (av_pix_fmt_desc_get (video_codec_context()->pix_fmt));
 }
+
+bool
+FFmpegExaminer::yuv () const
+{
+       switch (video_codec_context()->pix_fmt) {
+       case AV_PIX_FMT_YUV420P:
+       case AV_PIX_FMT_YUYV422:
+       case AV_PIX_FMT_YUV422P:
+       case AV_PIX_FMT_YUV444P:
+       case AV_PIX_FMT_YUV410P:
+       case AV_PIX_FMT_YUV411P:
+       case AV_PIX_FMT_YUVJ420P:
+       case AV_PIX_FMT_YUVJ422P:
+       case AV_PIX_FMT_YUVJ444P:
+       case AV_PIX_FMT_UYVY422:
+       case AV_PIX_FMT_UYYVYY411:
+       case AV_PIX_FMT_NV12:
+       case AV_PIX_FMT_NV21:
+       case AV_PIX_FMT_YUV440P:
+       case AV_PIX_FMT_YUVJ440P:
+       case AV_PIX_FMT_YUVA420P:
+       case AV_PIX_FMT_YUV420P16LE:
+       case AV_PIX_FMT_YUV420P16BE:
+       case AV_PIX_FMT_YUV422P16LE:
+       case AV_PIX_FMT_YUV422P16BE:
+       case AV_PIX_FMT_YUV444P16LE:
+       case AV_PIX_FMT_YUV444P16BE:
+       case AV_PIX_FMT_YUV420P9BE:
+       case AV_PIX_FMT_YUV420P9LE:
+       case AV_PIX_FMT_YUV420P10BE:
+       case AV_PIX_FMT_YUV420P10LE:
+       case AV_PIX_FMT_YUV422P10BE:
+       case AV_PIX_FMT_YUV422P10LE:
+       case AV_PIX_FMT_YUV444P9BE:
+       case AV_PIX_FMT_YUV444P9LE:
+       case AV_PIX_FMT_YUV444P10BE:
+       case AV_PIX_FMT_YUV444P10LE:
+       case AV_PIX_FMT_YUV422P9BE:
+       case AV_PIX_FMT_YUV422P9LE:
+       case AV_PIX_FMT_YUVA422P_LIBAV:
+       case AV_PIX_FMT_YUVA444P_LIBAV:
+       case AV_PIX_FMT_YUVA420P9BE:
+       case AV_PIX_FMT_YUVA420P9LE:
+       case AV_PIX_FMT_YUVA422P9BE:
+       case AV_PIX_FMT_YUVA422P9LE:
+       case AV_PIX_FMT_YUVA444P9BE:
+       case AV_PIX_FMT_YUVA444P9LE:
+       case AV_PIX_FMT_YUVA420P10BE:
+       case AV_PIX_FMT_YUVA420P10LE:
+       case AV_PIX_FMT_YUVA422P10BE:
+       case AV_PIX_FMT_YUVA422P10LE:
+       case AV_PIX_FMT_YUVA444P10BE:
+       case AV_PIX_FMT_YUVA444P10LE:
+       case AV_PIX_FMT_YUVA420P16BE:
+       case AV_PIX_FMT_YUVA420P16LE:
+       case AV_PIX_FMT_YUVA422P16BE:
+       case AV_PIX_FMT_YUVA422P16LE:
+       case AV_PIX_FMT_YUVA444P16BE:
+       case AV_PIX_FMT_YUVA444P16LE:
+       case AV_PIX_FMT_NV16:
+       case AV_PIX_FMT_NV20LE:
+       case AV_PIX_FMT_NV20BE:
+       case AV_PIX_FMT_YVYU422:
+       case AV_PIX_FMT_YUVA444P:
+       case AV_PIX_FMT_YUVA422P:
+       case AV_PIX_FMT_YUV420P12BE:
+       case AV_PIX_FMT_YUV420P12LE:
+       case AV_PIX_FMT_YUV420P14BE:
+       case AV_PIX_FMT_YUV420P14LE:
+       case AV_PIX_FMT_YUV422P12BE:
+       case AV_PIX_FMT_YUV422P12LE:
+       case AV_PIX_FMT_YUV422P14BE:
+       case AV_PIX_FMT_YUV422P14LE:
+       case AV_PIX_FMT_YUV444P12BE:
+       case AV_PIX_FMT_YUV444P12LE:
+       case AV_PIX_FMT_YUV444P14BE:
+       case AV_PIX_FMT_YUV444P14LE:
+       case AV_PIX_FMT_YUVJ411P:
+               return true;
+       default:
+               return false;
+       }
+}