rest of Termux related mods from Andrew
[goodguy/cinelerra.git] / cinelerra-5.1 / cinelerra / ffmpeg.C
index 97b6698acbb2ffc342f93049dbd54537d0a39235..3655d643fb097f4f954c768aeabf26bef44e6e60 100644 (file)
@@ -29,7 +29,9 @@
 #include "ffmpeg.h"
 #include "indexfile.h"
 #include "interlacemodes.h"
+#ifdef HAVE_DV
 #include "libdv.h"
+#endif
 #include "libmjpeg.h"
 #include "mainerror.h"
 #include "mwindow.h"
@@ -603,6 +605,7 @@ int FFStream::encode_frame(AVFrame *frame)
                        ret = avcodec_send_frame(avctx, frame);
                if( !ret && frame ) return pkts;
                if( ret < 0 && ret != AVERROR(EAGAIN) ) break;
+               if ( ret == AVERROR(EAGAIN) && !frame ) continue;
                FFPacket opkt;
                ret = avcodec_receive_packet(avctx, opkt);
                if( !frame && ret == AVERROR_EOF ) return pkts;
@@ -1213,6 +1216,40 @@ int FFVideoStream::decode_frame(AVFrame *frame)
        return 1;
 }
 
+int FFVideoStream::probe(int64_t pos)
+{
+       int ret = video_seek(pos);
+       if( ret < 0 ) return -1;
+       if( !frame && !(frame=av_frame_alloc()) ) {
+               fprintf(stderr, "FFVideoStream::probe: av_frame_alloc failed\n");
+               return -1;
+       }
+               
+       if (ffmpeg->interlace_from_codec) return 1;
+
+               ret = read_frame(frame);
+               if( ret > 0 ) {
+                       //printf("codec interlace: %i \n",frame->interlaced_frame);
+                       //printf("codec tff: %i \n",frame->top_field_first);
+
+                       if (!frame->interlaced_frame)
+                               ffmpeg->interlace_from_codec = AV_FIELD_PROGRESSIVE;
+                       if ((frame->interlaced_frame) && (frame->top_field_first))
+                               ffmpeg->interlace_from_codec = AV_FIELD_TT;
+                       if ((frame->interlaced_frame) && (!frame->top_field_first))
+                               ffmpeg->interlace_from_codec = AV_FIELD_BB;
+                       //printf("Interlace mode from codec: %i\n", ffmpeg->interlace_from_codec);
+
+       }
+
+       if( frame->format == AV_PIX_FMT_NONE || frame->width <= 0 || frame->height <= 0 )
+               ret = -1;
+
+       ret = ret > 0 ? 1 : ret < 0 ? -1 : 0;
+       av_frame_free(&frame);
+       return ret;
+}
+
 int FFVideoStream::load(VFrame *vframe, int64_t pos)
 {
        int ret = video_seek(pos);
@@ -1221,6 +1258,8 @@ int FFVideoStream::load(VFrame *vframe, int64_t pos)
                fprintf(stderr, "FFVideoStream::load: av_frame_alloc failed\n");
                return -1;
        }
+       
+
        int i = MAX_RETRY + pos - curr_pos;
        int64_t cache_start = 0;
        while( ret>=0 && !flushed && curr_pos<=pos && --i>=0 ) {
@@ -1391,6 +1430,7 @@ int FFVideoStream::encode(VFrame *vframe)
 
 int FFVideoStream::drain()
 {
+
        return 0;
 }
 
@@ -1553,9 +1593,11 @@ int FFVideoConvert::convert_picture_vframe(VFrame *frame, AVFrame *ip, AVFrame *
        }
        int color_space = SWS_CS_ITU601;
        switch( preferences->yuv_color_space ) {
-       case BC_COLORS_BT601:  color_space = SWS_CS_ITU601;  break;
+       case BC_COLORS_BT601_PAL:  color_space = SWS_CS_ITU601;  break;
+       case BC_COLORS_BT601_NTSC: color_space = SWS_CS_SMPTE170M; break;
        case BC_COLORS_BT709:  color_space = SWS_CS_ITU709;  break;
-       case BC_COLORS_BT2020: color_space = SWS_CS_BT2020;  break;
+       case BC_COLORS_BT2020_NCL: 
+       case BC_COLORS_BT2020_CL: color_space = SWS_CS_BT2020;  break;
        }
        const int *color_table = sws_getCoefficients(color_space);
 
@@ -1682,9 +1724,11 @@ int FFVideoConvert::convert_vframe_picture(VFrame *frame, AVFrame *op, AVFrame *
        }
        int color_space = SWS_CS_ITU601;
        switch( preferences->yuv_color_space ) {
-       case BC_COLORS_BT601:  color_space = SWS_CS_ITU601;  break;
+       case BC_COLORS_BT601_PAL:  color_space = SWS_CS_ITU601;  break;
+       case BC_COLORS_BT601_NTSC: color_space = SWS_CS_SMPTE170M; break;
        case BC_COLORS_BT709:  color_space = SWS_CS_ITU709;  break;
-       case BC_COLORS_BT2020: color_space = SWS_CS_BT2020;  break;
+       case BC_COLORS_BT2020_NCL:
+       case BC_COLORS_BT2020_CL: color_space = SWS_CS_BT2020;  break;
        }
        const int *color_table = sws_getCoefficients(color_space);
 
@@ -1776,6 +1820,7 @@ FFMPEG::FFMPEG(FileBase *file_base)
        flow = 1;
        decoding = encoding = 0;
        has_audio = has_video = 0;
+       interlace_from_codec = 0;
        opts = 0;
        opt_duration = -1;
        opt_video_filter = 0;
@@ -1816,12 +1861,15 @@ int FFMPEG::check_sample_rate(AVCodec *codec, int sample_rate)
        return 0;
 }
 
+// check_frame_rate and std_frame_rate needed for 23.976
+// and 59.94 fps mpeg2
 static inline AVRational std_frame_rate(int i)
 {
        static const int m1 = 1001*12, m2 = 1000*12;
        static const int freqs[] = {
                40*m1, 48*m1, 50*m1, 60*m1, 80*m1,120*m1, 240*m1,
-               24*m2, 30*m2, 60*m2, 12*m2, 15*m2, 48*m2, 0,
+               24*m2, 30*m2, 60*m2, 12*m2, 15*m2, 48*m2, 90*m2,
+               100*m2, 120*m2, 144*m2, 72*m2, 0,
        };
        int freq = i<30*12 ? (i+1)*1001 : freqs[i-30*12];
        return (AVRational) { freq, 1001*12 };
@@ -2411,6 +2459,10 @@ int FFMPEG::info(char *text, int len)
                AVPixelFormat pix_fmt = (AVPixelFormat)st->codecpar->format;
                const char *pfn = av_get_pix_fmt_name(pix_fmt);
                report(" pix %s\n", pfn ? pfn : unkn);
+               int interlace = st->codecpar->field_order;
+               report("  interlace (container level): %i\n", interlace ? interlace : -1);
+               int interlace_codec = interlace_from_codec;
+               report("  interlace (codec level): %i\n", interlace_codec ? interlace_codec : -1);
                enum AVColorSpace space = st->codecpar->color_space;
                const char *nm = av_color_space_name(space);
                report("    color space:%s", nm ? nm : unkn);
@@ -2594,18 +2646,22 @@ int FFMPEG::open_decoder()
                        }
                        switch( avpar->color_space ) {
                        case AVCOL_SPC_BT470BG:
+                               vid->color_space = BC_COLORS_BT601_PAL;
+                               break;
                        case AVCOL_SPC_SMPTE170M:
-                               vid->color_space = BC_COLORS_BT601;
+                               vid->color_space = BC_COLORS_BT601_NTSC;
                                break;
                        case AVCOL_SPC_BT709:
                                vid->color_space = BC_COLORS_BT709;
                                break;
                        case AVCOL_SPC_BT2020_NCL:
+                               vid->color_space = BC_COLORS_BT2020_NCL;
+                               break;
                        case AVCOL_SPC_BT2020_CL:
-                               vid->color_space = BC_COLORS_BT2020;
+                               vid->color_space = BC_COLORS_BT2020_CL;
                                break;
                        default:
-                               vid->color_space = !file_base ? BC_COLORS_BT601 :
+                               vid->color_space = !file_base ? BC_COLORS_BT601_NTSC :
                                        file_base->file->preferences->yuv_color_space;
                                break;
                        }
@@ -2848,9 +2904,11 @@ int FFMPEG::open_encoder(const char *type, const char *spec)
                        if( (vid->color_space = asset->ff_color_space) < 0 )
                                vid->color_space = file_base->file->preferences->yuv_color_space;
                        switch( vid->color_space ) {
-                       case BC_COLORS_BT601:  ctx->colorspace = AVCOL_SPC_SMPTE170M;  break;
+                       case BC_COLORS_BT601_NTSC:  ctx->colorspace = AVCOL_SPC_SMPTE170M;  break;
+                       case BC_COLORS_BT601_PAL: ctx->colorspace = AVCOL_SPC_BT470BG; break;
                        case BC_COLORS_BT709:  ctx->colorspace = AVCOL_SPC_BT709;      break;
-                       case BC_COLORS_BT2020: ctx->colorspace = AVCOL_SPC_BT2020_NCL; break;
+                       case BC_COLORS_BT2020_NCL: ctx->colorspace = AVCOL_SPC_BT2020_NCL; break;
+                       case BC_COLORS_BT2020_CL: ctx->colorspace = AVCOL_SPC_BT2020_CL; break;
                        }
                        AVPixelFormat pix_fmt = av_get_pix_fmt(asset->ff_pixel_format);
                        if( opt_hw_dev != 0 ) {
@@ -2873,7 +2931,12 @@ int FFMPEG::open_encoder(const char *type, const char *spec)
                        int mask_h = (1<<desc->log2_chroma_h)-1;
                        ctx->height = (vid->height+mask_h) & ~mask_h;
                        ctx->sample_aspect_ratio = to_sample_aspect_ratio(asset);
-                       AVRational frame_rate = check_frame_rate(codec->supported_framerates, vid->frame_rate);
+                       AVRational frame_rate;
+                       if (ctx->codec->id == AV_CODEC_ID_MPEG1VIDEO ||
+                           ctx->codec->id == AV_CODEC_ID_MPEG2VIDEO)
+                       frame_rate = check_frame_rate(codec->supported_framerates, vid->frame_rate);
+                       else
+                       frame_rate = av_d2q(vid->frame_rate, INT_MAX);
                        if( !frame_rate.num || !frame_rate.den ) {
                                eprintf(_("check_frame_rate failed %s\n"), filename);
                                ret = 1;
@@ -2889,6 +2952,25 @@ int FFMPEG::open_encoder(const char *type, const char *spec)
                        vid->interlaced = asset->interlace_mode == ILACE_MODE_TOP_FIRST ||
                                asset->interlace_mode == ILACE_MODE_BOTTOM_FIRST ? 1 : 0;
                        vid->top_field_first = asset->interlace_mode == ILACE_MODE_TOP_FIRST ? 1 : 0;
+                       switch (asset->interlace_mode)  {               
+                       case ILACE_MODE_TOP_FIRST: 
+                       if (ctx->codec->id == AV_CODEC_ID_MJPEG)
+                       av_dict_set(&sopts, "field_order", "tt", 0); 
+                       else
+                       av_dict_set(&sopts, "field_order", "tb", 0); 
+                       if (ctx->codec_id != AV_CODEC_ID_MJPEG) 
+                       av_dict_set(&sopts, "flags", "+ilme+ildct", 0);
+                       break;
+                       case ILACE_MODE_BOTTOM_FIRST: 
+                       if (ctx->codec->id == AV_CODEC_ID_MJPEG)
+                       av_dict_set(&sopts, "field_order", "bb", 0); 
+                       else
+                       av_dict_set(&sopts, "field_order", "bt", 0); 
+                       if (ctx->codec_id != AV_CODEC_ID_MJPEG)
+                       av_dict_set(&sopts, "flags", "+ilme+ildct", 0);
+                       break;
+                       case ILACE_MODE_NOTINTERLACED: av_dict_set(&sopts, "field_order", "progressive", 0); break;
+                       }
                        break; }
                default:
                        eprintf(_("not audio/video, %s:%s\n"), codec_name, filename);
@@ -3181,6 +3263,33 @@ int FFMPEG::audio_seek(int stream, int64_t pos)
        return 0;
 }
 
+int FFMPEG::video_probe(int64_t pos)
+{
+       int vidx = vstrm_index[0].st_idx;
+       FFVideoStream *vid = ffvideo[vidx];
+       vid->probe(pos);
+       
+       int interlace1 = interlace_from_codec;
+       //printf("interlace from codec: %i\n", interlace1);
+
+       switch (interlace1)
+       {
+       case AV_FIELD_TT:
+       case AV_FIELD_TB:
+           return ILACE_MODE_TOP_FIRST;
+       case AV_FIELD_BB:
+       case AV_FIELD_BT:
+           return ILACE_MODE_BOTTOM_FIRST;
+       case AV_FIELD_PROGRESSIVE:
+           return ILACE_MODE_NOTINTERLACED;
+       default:
+           return ILACE_MODE_UNDETECTED;
+       }
+
+}
+
+
+
 int FFMPEG::video_seek(int stream, int64_t pos)
 {
        int vidx = vstrm_index[stream].st_idx;
@@ -3468,7 +3577,20 @@ int FFMPEG::ff_coded_height(int stream)
 
 float FFMPEG::ff_aspect_ratio(int stream)
 {
-       return ffvideo[stream]->aspect_ratio;
+       //return ffvideo[stream]->aspect_ratio;
+       AVFormatContext *fmt_ctx = ffvideo[stream]->fmt_ctx;
+       AVStream *strm = ffvideo[stream]->st;
+       AVCodecParameters *par = ffvideo[stream]->st->codecpar;
+       AVRational dar;
+       AVRational sar = av_guess_sample_aspect_ratio(fmt_ctx, strm, NULL);
+        if (sar.num) {
+            av_reduce(&dar.num, &dar.den,
+                      par->width  * sar.num,
+                      par->height * sar.den,
+                      1024*1024);
+                      return av_q2d(dar);
+                      }
+        return ffvideo[stream]->aspect_ratio;
 }
 
 const char* FFMPEG::ff_video_codec(int stream)
@@ -3509,6 +3631,30 @@ int FFMPEG::ff_video_mpeg_color_range(int stream)
        return ffvideo[stream]->st->codecpar->color_range == AVCOL_RANGE_MPEG ? 1 : 0;
 }
 
+int FFMPEG::ff_interlace(int stream)
+{
+// https://ffmpeg.org/doxygen/trunk/structAVCodecParserContext.html
+/* reads from demuxer because codec frame not ready */
+       int interlace0 = ffvideo[stream]->st->codecpar->field_order;
+
+       switch (interlace0)
+       {
+       case AV_FIELD_TT:
+       case AV_FIELD_TB:
+           return ILACE_MODE_TOP_FIRST;
+       case AV_FIELD_BB:
+       case AV_FIELD_BT:
+           return ILACE_MODE_BOTTOM_FIRST;
+       case AV_FIELD_PROGRESSIVE:
+           return ILACE_MODE_NOTINTERLACED;
+       default:
+           return ILACE_MODE_UNDETECTED;
+       }
+       
+}
+
+
+
 int FFMPEG::ff_cpus()
 {
        return !file_base ? 1 : file_base->file->cpus;