sjs/SJSAssetExportSession
1
0
Fork 0
mirror of https://github.com/samsonjs/SJSAssetExportSession.git synced 2026-04-27 14:57:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Make the test more thorough and move color stuff to video composition

Browse source
This commit is contained in:
Sami Samhuri 2024-07-06 22:30:12 -07:00
parent 750b77210c
commit 6319aaaad1
No known key found for this signature in database
1 changed files with 22 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
SJSAssetExportSessionTests/SJSAssetExportSessionTests.swift
View file

@ -17,13 +17,16 @@ final class ExportSessionTests {
let filename = "ExportSessionTests_testEncode_\(timestamp).mp4" let filename = "ExportSessionTests_testEncode_\(timestamp).mp4"
let destinationURL = URL.temporaryDirectory.appending(component: filename) let destinationURL = URL.temporaryDirectory.appending(component: filename)
defer { _ = try? FileManager.default.removeItem(at: destinationURL) } defer { _ = try? FileManager.default.removeItem(at: destinationURL) }
let size = CGSize(width: 1280, height: 720) let size = CGSize(width: 1280, height: 720)
let duration = CMTime(seconds: 1, preferredTimescale: 600) let duration = CMTime(seconds: 1, preferredTimescale: 600)
let videoComposition = try await AVMutableVideoComposition.videoComposition(withPropertiesOf: sourceAsset) let videoComposition = try await AVMutableVideoComposition.videoComposition(withPropertiesOf: sourceAsset)
videoComposition.renderSize = size videoComposition.renderSize = size
videoComposition.renderScale = 1 videoComposition.renderScale = 1
videoComposition.frameDuration = CMTime(value: 1, timescale: 30) videoComposition.frameDuration = CMTime(value: 1, timescale: 30)
videoComposition.colorPrimaries = AVVideoColorPrimaries_ITU_R_709_2
videoComposition.colorTransferFunction = AVVideoTransferFunction_ITU_R_709_2
videoComposition.colorYCbCrMatrix = AVVideoYCbCrMatrix_ITU_R_709_2
try await ExportSession.export( try await ExportSession.export(
asset: sourceAsset, asset: sourceAsset,
audioMix: nil, audioMix: nil,
@ -41,11 +44,6 @@ final class ExportSessionTests {
AVVideoAverageBitRateKey: NSNumber(value: 1_000_000), AVVideoAverageBitRateKey: NSNumber(value: 1_000_000),
AVVideoProfileLevelKey: AVVideoProfileLevelH264HighAutoLevel as String, AVVideoProfileLevelKey: AVVideoProfileLevelH264HighAutoLevel as String,
] as [String: any Sendable], ] as [String: any Sendable],
AVVideoColorPropertiesKey: [
AVVideoColorPrimariesKey: AVVideoColorPrimaries_ITU_R_709_2,
AVVideoTransferFunctionKey: AVVideoTransferFunction_ITU_R_709_2,
AVVideoYCbCrMatrixKey: AVVideoYCbCrMatrix_ITU_R_709_2,
],
], ],
timeRange: CMTimeRange(start: .zero, duration: duration), timeRange: CMTimeRange(start: .zero, duration: duration),
to: destinationURL, to: destinationURL,
@ -54,9 +52,24 @@ final class ExportSessionTests {
let asset = AVURLAsset(url: destinationURL) let asset = AVURLAsset(url: destinationURL)
#expect(try await asset.load(.duration) == duration) #expect(try await asset.load(.duration) == duration)
try #require(await asset.loadTracks(withMediaType: .video).count == 1) // Audio
let videoTrack = try #require(try await asset.loadTracks(withMediaType: .video).first)
#expect(try await videoTrack.load(.naturalSize) == size)
try #require(try await asset.loadTracks(withMediaType: .audio).count == 1) try #require(try await asset.loadTracks(withMediaType: .audio).count == 1)
let audioTrack = try #require(await asset.loadTracks(withMediaType: .audio).first)
let audioFormat = try #require(await audioTrack.load(.formatDescriptions).first)
#expect(audioFormat.mediaType == .audio)
#expect(audioFormat.mediaSubType == .mpeg4AAC)
#expect(audioFormat.audioChannelLayout?.numberOfChannels == 2)
#expect(audioFormat.audioStreamBasicDescription?.mSampleRate == 44_100)
// Video
try #require(await asset.loadTracks(withMediaType: .video).count == 1)
let videoTrack = try #require(await asset.loadTracks(withMediaType: .video).first)
#expect(try await videoTrack.load(.naturalSize) == size)
print(try await videoTrack.load(.formatDescriptions))
let videoFormat = try #require(await videoTrack.load(.formatDescriptions).first)
#expect(videoFormat.mediaType == .video)
#expect(videoFormat.mediaSubType == .h264)
#expect(videoFormat.extensions[.colorPrimaries] == .colorPrimaries(.itu_R_709_2))
#expect(videoFormat.extensions[.transferFunction] == .transferFunction(.itu_R_709_2))
#expect(videoFormat.extensions[.yCbCrMatrix] == .yCbCrMatrix(.itu_R_709_2))
} }
} }