UIImage_extension.swift
2.76 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
//
// UIImage_extension.swift
// VisionDetection
//
// Created by zx on 2024/5/11.
// Copyright © 2024 Willjay. All rights reserved.
//
import UIKit
enum ImageCopmpressSize {
//压缩到100KB(1024*100字节)以内
case one
//压缩到200KB(1024*100字节)以内
case two
//压缩到300KB(1024*100字节)以内
case three
}
extension UIImage {
//不大于100KB
func compressImage()->Data{
compressImage(size: .one)
}
func compressImage(size:ImageCopmpressSize) -> Data{
switch size {
case .one:
return compressImage(maxLength: 1024*100)
case .two:
return compressImage(maxLength: 1024*200)
case .three:
return compressImage(maxLength: 1024*300)
default:
break
}
}
// 图片压缩 byte
func compressImage(maxLength: Int) -> Data {
// let tempMaxLength: Int = maxLength / 8
let tempMaxLength: Int = maxLength
var compression: CGFloat = 1
guard var data = self.jpegData(compressionQuality: compression), data.count > tempMaxLength else { return self.jpegData(compressionQuality: compression)! }
// 压缩大小
var max: CGFloat = 1
var min: CGFloat = 0
for _ in 0..<6 {
compression = (max + min) / 2
data = self.jpegData(compressionQuality: compression)!
if CGFloat(data.count) < CGFloat(tempMaxLength) * 0.9 {
min = compression
} else if data.count > tempMaxLength {
max = compression
} else {
break
}
}
var resultImage: UIImage = UIImage(data: data)!
if data.count < tempMaxLength { return data }
// 压缩大小
var lastDataLength: Int = 0
while data.count > tempMaxLength && data.count != lastDataLength {
lastDataLength = data.count
let ratio: CGFloat = CGFloat(tempMaxLength) / CGFloat(data.count)
print("Ratio =", ratio)
let size: CGSize = CGSize(width: Int(resultImage.size.width * sqrt(ratio)),
height: Int(resultImage.size.height * sqrt(ratio)))
UIGraphicsBeginImageContext(size)
resultImage.draw(in: CGRect(x: 0, y: 0, width: size.width, height: size.height))
resultImage = UIGraphicsGetImageFromCurrentImageContext()!
UIGraphicsEndImageContext()
data = resultImage.jpegData(compressionQuality: compression)!
}
return data
}
}