ErsatzTV/ErsatzTV.FFmpeg/Decoder/Cuvid/DecoderMpeg2Cuvid.cs

namespace ErsatzTV.FFmpeg.Decoder.Cuvid;

public class DecoderMpeg2Cuvid : DecoderBase
{
    private readonly bool _contentIsInterlaced;
    private readonly FFmpegState _ffmpegState;

    public DecoderMpeg2Cuvid(FFmpegState ffmpegState, bool contentIsInterlaced)
    {
        _ffmpegState = ffmpegState;
        _contentIsInterlaced = contentIsInterlaced;
    }

    public override string Name => "mpeg2_cuvid";

    protected override FrameDataLocation OutputFrameDataLocation =>
        _contentIsInterlaced || _ffmpegState.EncoderHardwareAccelerationMode == HardwareAccelerationMode.None
            ? FrameDataLocation.Software
            : FrameDataLocation.Hardware;

    public override IList<string> InputOptions(InputFile inputFile)
    {
        IList<string> result = base.InputOptions(inputFile);

        if (_ffmpegState.EncoderHardwareAccelerationMode != HardwareAccelerationMode.None)
        {
            result.Add("-hwaccel_output_format");
            result.Add("cuda");
        }
        else
        {
            result.Add("-hwaccel_output_format");
            result.Add(InputBitDepth(inputFile) == 10 ? "p010le" : "nv12");
        }

        return result;
    }
}