non906 / fftools_lib Goto Github PK

using FfmpegUnity;
using System;
using UnityEngine;
using System.IO;
using OwnRequestStruct;

public class FFmpegManager : MonoBehaviour
{
    GameObject captureObject;
    FfmpegCaptureCommand capture;
    FfplayCommand videoPlayer;
    FfmpegCommand videoConverter;


    private void Awake()
    {

        // 初始化Ffmpeg for unity，用于音视频录制
        captureObject = new GameObject("FFmpeg");

        // 使用Ffmpeg for unity播放视频
        videoPlayer = captureObject.AddComponent<FfplayCommand>();
        videoPlayer.ExecuteOnStart = false;
        videoPlayer.DefaultPath = FfmpegPath.DefaultPath.NONE;
        var videoTexture = gameObject.AddComponent<FfmpegPlayerVideoTexture>();
        videoTexture.VideoTexture = GetComponent<InitBaseScense>().imageShow.texture;
        videoPlayer.VideoTexture = videoTexture;
        var audioSourceVideo = captureObject.AddComponent<AudioSource>();
        videoPlayer.AudioSourceComponent = audioSourceVideo;

        // 使用Ffmpeg for unity调整音轨时间
        videoConverter = captureObject.AddComponent<FfmpegCommand>();
        videoConverter.ExecuteOnStart = false;
    }

    #region FFmpeg相关函数
    /// <summary>
    /// 开始录制，在这里初始化录制脚本，避免修改分辨率以后画面错位问题。
    /// 以“.mp4”结尾录制为文件，否则推流，流地址为rtmp://ip:1935/live/{savePath}
    /// </summary>
    /// <param name="savePath"></param>
    public void StartRecord(string savePath, RequestStruct req)
    {
        // 设置相关参数
        capture = captureObject.AddComponent<FfmpegCaptureCommand>();   // 视频录制脚本
        capture.ExecuteOnStart = false;
        capture.req = req;
        capture.callback = GetComponent<Callback>();
        var video = capture.CaptureSources[0];
        video.Type = FfmpegCaptureCommand.CaptureSource.SourceType.Video_Camera;
        video.SourceCamera = Camera.main;
        video.Width = Screen.width;
        video.Height = Screen.height;
        video.FrameRate = 30;

        // 录制为文件或推流
        string recordArgs;
        string msg;

        var ext = Path.GetExtension(savePath);
        if (ext == "")
        {
            string rtmp = $"rtmp://127.0.0.1:1935/live/{savePath}";
            recordArgs = $"-c:v libopenh264 -c:a aac -map 0:v:0 -map 1:a:0 -g 30 -vsync 1 -async 96000 -f flv {rtmp}";
            msg = $"Vidoe will stream to: {rtmp}";
        }
        else
        {
            if (ext == ".webm") recordArgs = $"-c:v libvpx-vp9 -c:a libvorbis -map 0:v:0 -map 1:a:0 -crf 30 -b:v 0 -deadline realtime -cpu-used 8 {savePath}";
            else if (ext == ".mp4") recordArgs = $"-c:v libopenh264 -c:a aac -map 0:v:0 -map 1:a:0 {savePath}";
            else throw new NotImplementedException($"Recording video to {ext} is unsupport");
            msg = $"Vidoe will save to: {savePath}";
        }

        // 开始FFmpeg录制线程
        capture.CaptureOptions = recordArgs;
        capture.StartFfmpeg();
        Debug.Log($" [ - FFmpeg - ] Start recording. Res: {video.Width} x {video.Height}. {msg}");
    }


    public void StopRecord()
    {
        // 停止FFmpeg线程
        if (capture.IsRunning)
        {
            capture.StopFfmpeg();
            Destroy(capture);
            Destroy(Camera.main.GetComponent<FfmpegCaptureCamera>());
            Debug.Log(" [ - FFmpeg - ] Stop recording.");
        }
    }


    public enum VideoExt
    {
        mp4,
        avi,
        flv,
        m4v,
        mkv,
        mov,
        webm,
        wmv
    }
    
    /// <summary>
    /// 播放视频文件，经过测试，支持的后缀为：mp4,avi,flv,m4v,mkv,mov,webm,wmv
    /// </summary>
    /// <param name="videoPath"></param>
    /// <param name="volume"></param>
    /// <exception cref="Exception"></exception>
    public void PlayVideoFile(string videoPath, float volume=1.0f)
    {
        if (!File.Exists(videoPath))
            throw new Exception("Video file not exist: " + videoPath);
        GetComponent<AudioPlayer>().StopPlay(); // 关闭音频
        if (videoPlayer.IsRunning)
        {
            videoPlayer.Dispose();
        }
        videoPlayer.InputPath = videoPath;
        videoPlayer.Volume = volume;
        videoPlayer.Play();
        GetComponent<InitBaseScense>().canvasShow.enabled = true;
        Debug.Log($" [ - FFmpeg - ] Start playing: {videoPath}. volume={volume}");
    }

    public void StopPlay()
    {
        if (videoPlayer.IsRunning)
        {
            videoPlayer.Dispose();
            GetComponent<InitBaseScense>().canvasShow.enabled = false;
            Debug.Log(" [ - FFmpeg - ] Stop playing");
        }
    }

    private void Update()
    {
        if (!videoPlayer.IsRunning)
            return;

        // 视频播放完毕时IsRunning=true,Paused=true
        if (videoPlayer.Paused)
        {
            StopPlay();
        }
    }

    public void Convert(string inputPath, string outputPath, float audioDelaySec)
    {
        videoConverter.Options = string.Join(
          "\n",
          "-y",
          $"-i {inputPath}",
          $"-itsoffset {audioDelaySec}",
          $"-i {inputPath}",
          "-map 0:v",
          "-map 1:a",
          "-c copy",
          outputPath
          );
        videoConverter.ExecuteFfmpeg();
        Debug.Log($"Convert {inputPath} to {outputPath} by delaying audio {audioDelaySec} s");
    }
#endregion
}