fix qsv color normalization (#1046)

ErsatzTV.FFmpeg/Filter/ColorspaceFilter.cs

@@ -6,15 +6,18 @@ public class ColorspaceFilter : BaseFilter
 {
     private readonly VideoStream _videoStream;
     private readonly IPixelFormat _desiredPixelFormat;
+    private readonly bool _forceInputOverrides;
     private readonly FrameDataLocation _nextDataLocation;
 
     public ColorspaceFilter(
         VideoStream videoStream,
         IPixelFormat desiredPixelFormat,
+        bool forceInputOverrides = false,
         FrameDataLocation nextDataLocation = FrameDataLocation.Software)
     {
         _videoStream = videoStream;
         _desiredPixelFormat = desiredPixelFormat;
+        _forceInputOverrides = forceInputOverrides;
         _nextDataLocation = nextDataLocation;
     }
 
@@ -36,10 +39,12 @@ public class ColorspaceFilter : BaseFilter
         {
             string inputOverrides = string.Empty;
             ColorParams cp = _videoStream.ColorParams;
-            if (cp.IsMixed)
+            if (cp.IsMixed || _forceInputOverrides)
             {
+                string range = string.IsNullOrWhiteSpace(cp.ColorRange) ? "tv" : cp.ColorRange;
+
                 inputOverrides =
-                    $"irange={cp.ColorRange}:ispace={cp.ColorSpace}:itrc={cp.ColorTransfer}:iprimaries={cp.ColorPrimaries}:";
+                    $"irange={range}:ispace={cp.ColorSpace}:itrc={cp.ColorTransfer}:iprimaries={cp.ColorPrimaries}:";
             }
 
             string colorspace = _desiredPixelFormat.BitDepth switch

ErsatzTV.FFmpeg/Pipeline/NvidiaPipelineBuilder.cs

@@ -262,7 +262,7 @@ public class NvidiaPipelineBuilder : SoftwarePipelineBuilder
             if (!videoStream.ColorParams.IsBt709)
             {
                 _logger.LogDebug("Adding colorspace filter");
-                var colorspace = new ColorspaceFilter(videoStream, format, currentState.FrameDataLocation);
+                var colorspace = new ColorspaceFilter(videoStream, format, false, currentState.FrameDataLocation);
 
                 currentState = colorspace.NextState(currentState);
                 result.Add(colorspace);

ErsatzTV.FFmpeg/Pipeline/QsvPipelineBuilder.cs

@@ -190,6 +190,7 @@ public class QsvPipelineBuilder : SoftwarePipelineBuilder
         }
         
         List<IPipelineFilterStep> pixelFormatFilterSteps = SetPixelFormat(
+            videoInputFile,
             videoStream,
             desiredState.PixelFormat,
             ffmpegState,
@@ -207,6 +208,7 @@ public class QsvPipelineBuilder : SoftwarePipelineBuilder
     }
     
     private List<IPipelineFilterStep> SetPixelFormat(
+        VideoInputFile videoInputFile,
         VideoStream videoStream,
         Option<IPixelFormat> desiredPixelFormat,
         FFmpegState ffmpegState,
@@ -233,7 +235,15 @@ public class QsvPipelineBuilder : SoftwarePipelineBuilder
             if (!videoStream.ColorParams.IsBt709)
             {
                 _logger.LogDebug("Adding colorspace filter");
-                var colorspace = new ColorspaceFilter(videoStream, format, currentState.FrameDataLocation);
+
+                // vpp_qsv seems to strip color info, so if we use that at all, force overriding input color info
+                bool forceInputOverrides = videoInputFile.FilterSteps.Any(f => f is QsvFormatFilter or ScaleQsvFilter);
+
+                var colorspace = new ColorspaceFilter(
+                    videoStream,
+                    format,
+                    forceInputOverrides,
+                    currentState.FrameDataLocation);
 
                 // force nv12 if we're still in hardware
                 if (currentState.FrameDataLocation == FrameDataLocation.Hardware)

ErsatzTV.FFmpeg/Pipeline/VaapiPipelineBuilder.cs

@@ -240,7 +240,7 @@ public class VaapiPipelineBuilder : SoftwarePipelineBuilder
             if (!videoStream.ColorParams.IsBt709)
             {
                 _logger.LogDebug("Adding colorspace filter");
-                var colorspace = new ColorspaceFilter(videoStream, format, currentState.FrameDataLocation);
+                var colorspace = new ColorspaceFilter(videoStream, format, false, currentState.FrameDataLocation);
                 currentState = colorspace.NextState(currentState);
                 result.Add(colorspace);
             }