67 lines
2.6 KiB
Swift
67 lines
2.6 KiB
Swift
//
|
|
// https://mczachurski.dev
|
|
// Copyright © 2023 Marcin Czachurski and the repository contributors.
|
|
// Licensed under the Apache License 2.0.
|
|
//
|
|
|
|
import SwiftUI
|
|
|
|
extension UIImage {
|
|
public func getJpegData() -> Data? {
|
|
#if targetEnvironment(simulator)
|
|
// For testing purposes.
|
|
let converted = self.convertToExtendedSRGBJpeg()
|
|
let filePath = URL.temporaryDirectory.appending(path: "\(UUID().uuidString).jpg")
|
|
try? converted?.write(to: filePath)
|
|
print(filePath.string)
|
|
#endif
|
|
|
|
// API don't support images over 5K.
|
|
if self.size.height > 10_000 || self.size.width > 10_000 {
|
|
return self
|
|
.resized(to: .init(width: self.size.width / 4, height: self.size.height / 4))
|
|
.convertToExtendedSRGBJpeg()
|
|
} else if self.size.height > 5000 || self.size.width > 5000 {
|
|
return self
|
|
.resized(to: .init(width: self.size.width / 2, height: self.size.height / 2))
|
|
.convertToExtendedSRGBJpeg()
|
|
} else {
|
|
return self
|
|
.convertToExtendedSRGBJpeg()
|
|
}
|
|
}
|
|
|
|
public func convertToExtendedSRGBJpeg() -> Data? {
|
|
guard let sourceImage = CIImage(image: self, options: [.applyOrientationProperty: true]) else {
|
|
return self.jpegData(compressionQuality: 0.9)
|
|
}
|
|
|
|
// We have to store correct image orientation.
|
|
let orientedImage = sourceImage.oriented(forExifOrientation: self.imageOrientation.exifOrientation)
|
|
|
|
// We dont have to convert images which already are in sRGB color space.
|
|
if orientedImage.colorSpace?.name == CGColorSpace.sRGB || orientedImage.colorSpace?.name == CGColorSpace.extendedSRGB {
|
|
return self.jpegData(compressionQuality: 0.9)
|
|
}
|
|
|
|
guard let colorSpace = CGColorSpace(name: CGColorSpace.extendedSRGB) else {
|
|
return self.jpegData(compressionQuality: 0.9)
|
|
}
|
|
|
|
guard let displayP3 = CGColorSpace(name: CGColorSpace.displayP3) else {
|
|
return self.jpegData(compressionQuality: 0.9)
|
|
}
|
|
|
|
// Create Core Image context (with working color space).
|
|
let ciContext = CIContext(options: [CIContextOption.workingColorSpace: orientedImage.colorSpace ?? displayP3])
|
|
|
|
// Creating image with new color space (and preserving colors).
|
|
guard let converted = ciContext.jpegRepresentation(of: orientedImage, colorSpace: colorSpace) else {
|
|
return self.jpegData(compressionQuality: 0.9)
|
|
}
|
|
|
|
// Returning successfully converted image.
|
|
return converted
|
|
}
|
|
}
|