FISLib_Old/Vinno.vCloud.FIS.CrossPlatform.Windows/LiveVideo/FormatConvertUtil.cs

using FFmpeg.AutoGen;
using System;
using Vinno.vCloud.FIS.CrossPlatform.Common;

namespace Vinno.vCloud.FIS.CrossPlatform.Windows.LiveVideo
{
    public unsafe class FormatConvertUtil : IDisposable
    {
        private SwsContext* _swsContext;

        /// <summary>
        /// 转换图像格式
        /// </summary>
        /// <param name="src">原图像</param>
        /// <param name="dest">目标图像</param>
        public void Convert(AVFrame* src, AVFrame* dest)
        {
            if (_swsContext == null)
            {
                CrossPlatformHelper.Instance.LogWriter?.WriteLineInfo($"Create SwsContext,SRC:{src->width},{src->height},{(AVPixelFormat)src->format},DEST:{dest->width},{dest->height},{(AVPixelFormat)dest->format}");
                _swsContext = ffmpeg.sws_getContext(src->width, src->height, ConvertDeprecatedFormat((AVPixelFormat)src->format), dest->width, dest->height, (AVPixelFormat)dest->format, ffmpeg.SWS_BILINEAR, null, null, null);
            }
            var ret = ffmpeg.sws_scale(_swsContext, src->data, src->linesize, 0, src->height, dest->data, dest->linesize);
            if (ret != dest->height)
            {
                CrossPlatformHelper.Instance.LogWriter?.WriteLineError($"sws_scale error:{ret}");
            }
        }

        private AVPixelFormat ConvertDeprecatedFormat(AVPixelFormat format)
        {
            switch (format)
            {
                case AVPixelFormat.AV_PIX_FMT_YUVJ420P:
                    return AVPixelFormat.AV_PIX_FMT_YUV420P;

                case AVPixelFormat.AV_PIX_FMT_YUVJ422P:
                    return AVPixelFormat.AV_PIX_FMT_YUV422P;

                case AVPixelFormat.AV_PIX_FMT_YUVJ444P:
                    return AVPixelFormat.AV_PIX_FMT_YUV444P;

                case AVPixelFormat.AV_PIX_FMT_YUVJ440P:
                    return AVPixelFormat.AV_PIX_FMT_YUV440P;

                default:
                    return format;
            }
        }

        public void Dispose()
        {
            if (_swsContext != null)
            {
                CrossPlatformHelper.Instance.LogWriter?.WriteLineInfo("Free SwsContext!");
                ffmpeg.sws_freeContext(_swsContext);
                _swsContext = null;
            }
        }
    }
}