Tag color information on asset writer

finnvoor · Dec 30, 2023 · b876f07 · b876f07
1 parent 056353f
commit b876f07
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Sources/Upscaling/UpscalingExportSession.swift b/Sources/Upscaling/UpscalingExportSession.swift
@@ -86,7 +86,12 @@ public class UpscalingExportSession {
                 let videoInput = AVAssetWriterInput(mediaType: .video, outputSettings: [
                     AVVideoWidthKey: outputSize.width,
                     AVVideoHeightKey: outputSize.height,
-                    AVVideoCodecKey: videoCodec
+                    AVVideoCodecKey: videoCodec,
+                    AVVideoColorPropertiesKey: [
+                        AVVideoColorPrimariesKey: formatDescription?.colorPrimaries,
+                        AVVideoTransferFunctionKey: formatDescription?.colorTransferFunction,
+                        AVVideoYCbCrMatrixKey: formatDescription?.colorYCbCrMatrix
+                    ]
                 ])
                 videoInput.expectsMediaDataInRealTime = false
                 if assetWriter.canAdd(videoInput) {