SHA256
77 lines
2.5 KiB
C#
77 lines
2.5 KiB
C#
using System.Diagnostics.CodeAnalysis;
|
|
using System.IO;
|
|
using Unity.Collections;
|
|
using UnityEngine;
|
|
|
|
namespace ADOFAIRenderer.Util
|
|
{
|
|
public class VideoFFmpeg : FFmpegProcess
|
|
{
|
|
[SuppressMessage("ReSharper", "InconsistentNaming")]
|
|
public enum Codec
|
|
{
|
|
H264_Default,
|
|
H264_NVIDIA,
|
|
H264_QSV,
|
|
HEVC_Default,
|
|
HEVC_NVIDIA,
|
|
HEVC_QSV,
|
|
AV1_Default,
|
|
AV1_NVIDIA,
|
|
AV1_QSV,
|
|
VP9_Default,
|
|
VP9_NVIDIA
|
|
}
|
|
|
|
public int Width;
|
|
public int Height;
|
|
public int FrameRate;
|
|
public int VideoFPS;
|
|
public Codec VideoCodec;
|
|
|
|
private static string Codec2String(Codec c)
|
|
{
|
|
return c switch
|
|
{
|
|
Codec.H264_Default => "-vf vflip",
|
|
Codec.H264_NVIDIA => "-c:v h264_nvenc -vf vflip",
|
|
Codec.H264_QSV => "-c:v h264_qsv -vf vflip",
|
|
Codec.HEVC_Default => "-c:v libx265 -vf vflip",
|
|
Codec.HEVC_NVIDIA => "-c:v hevc_nvenc -vf vflip",
|
|
Codec.HEVC_QSV => "-c:v hevc_qsv -vf vflip",
|
|
Codec.AV1_Default => "-c:v libaom-av1 -vf vflip",
|
|
Codec.AV1_NVIDIA => "-c:v av1_nvenc -vf vflip",
|
|
Codec.AV1_QSV => "-c:v av1_qsv -vf vflip",
|
|
Codec.VP9_Default => "-c:v libvpx-vp9 -vf vflip",
|
|
Codec.VP9_NVIDIA => "-c:v vp9_nvenc -vf vflip",
|
|
_ => "",
|
|
};
|
|
}
|
|
|
|
public VideoFFmpeg(int w, int h, int frame, int videoFrame, Codec codec)
|
|
{
|
|
Width = w;
|
|
Height = h;
|
|
FrameRate = frame;
|
|
VideoFPS = videoFrame;
|
|
VideoCodec = codec;
|
|
}
|
|
|
|
public void SendData(NativeArray<byte> source)
|
|
{
|
|
var buffer = ToRawBytes(source);
|
|
SendDataFromBuffer(buffer);
|
|
}
|
|
|
|
public void Start(string path, float bitrate)
|
|
{
|
|
if (File.Exists(path))
|
|
{
|
|
File.Delete(path);
|
|
}
|
|
var command = "-y -f rawvideo -vcodec rawvideo -pixel_format rgba -colorspace rgb -color_range 2 -video_size " + Width + "x" + Height + " -framerate " + FrameRate + " -threads 4 -loglevel quiet -i - " + Codec2String(VideoCodec) + " -pix_fmt yuv420p -fps_mode cfr " + " -r " + VideoFPS + $" -b:v {(int)bitrate}k" + " \"" + path + "\"";
|
|
Execute(command);
|
|
}
|
|
}
|
|
}
|