ThePain/Swiftgram
1
0
Fork 0
mirror of https://github.com/Swiftgram/Telegram-iOS.git synced 2025-12-22 14:20:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

[WIP] Sticker editor

Browse Source
This commit is contained in:
Ilya Laktyushin
2024-03-09 03:19:31 +04:00
parent c78f407c02
commit 3bdcb7f223
86 changed files with 5498 additions and 1990 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
submodules/TelegramUI/Components/MediaEditorScreen/Sources/MediaCutoutScreen.swift
View File

@@ -47,8 +47,8 @@ private final class MediaCutoutScreenComponent: Component {
private let doneButton = ComponentView<Empty>()
private let fadeView = UIView()
private let separatedImageView = UIImageView()
private var outlineViews: [StickerCutoutOutlineView] = []
private var component: MediaCutoutScreenComponent?
private weak var state: State?
private var environment: ViewControllerComponentContainer.Environment?
@@ -57,9 +57,7 @@ private final class MediaCutoutScreenComponent: Component {
self.buttonsContainerView.clipsToBounds = true
self.fadeView.alpha = 0.0
self.fadeView.backgroundColor = UIColor(rgb: 0x000000, alpha: 0.6)
self.separatedImageView.contentMode = .scaleAspectFit
self.fadeView.backgroundColor = UIColor(rgb: 0x000000, alpha: 0.7)
super.init(frame: frame)
@@ -69,7 +67,6 @@ private final class MediaCutoutScreenComponent: Component {
self.buttonsContainerView.addSubview(self.buttonsBackgroundView)
self.addSubview(self.fadeView)
self.addSubview(self.separatedImageView)
self.addSubview(self.previewContainerView)
self.previewContainerView.addGestureRecognizer(UITapGestureRecognizer(target: self, action: #selector(self.previewTap(_:))))
@@ -89,9 +86,21 @@ private final class MediaCutoutScreenComponent: Component {
x: location.x / self.previewContainerView.frame.width,
y: location.y / self.previewContainerView.frame.height
)
component.mediaEditor.setSeparationMask(point: point)
self.playDissolveAnimation()
component.mediaEditor.processImage { [weak self] originalImage, _ in
cutoutImage(from: originalImage, values: nil, target: .point(point), includeExtracted: false, completion: { [weak self] results in
Queue.mainQueue().async {
if let self, let component = self.component, let result = results.first, let maskImage = result.maskImage {
if case let .image(mask, _) = maskImage {
self.playDissolveAnimation()
component.mediaEditor.setSegmentationMask(mask)
}
}
}
})
}
HapticFeedback().impact(.medium)
}
func animateInFromEditor() {
@@ -106,6 +115,9 @@ private final class MediaCutoutScreenComponent: Component {
self.cancelButton.view?.isHidden = true
self.fadeView.layer.animateAlpha(from: self.fadeView.alpha, to: 0.0, duration: 0.2, removeOnCompletion: false)
for outlineView in self.outlineViews {
outlineView.layer.animateAlpha(from: self.fadeView.alpha, to: 0.0, duration: 0.2, removeOnCompletion: false)
}
self.buttonsBackgroundView.layer.animateAlpha(from: 1.0, to: 0.0, duration: 0.2, removeOnCompletion: false, completion: { _ in
completion()
})
@@ -210,7 +222,7 @@ private final class MediaCutoutScreenComponent: Component {
let labelSize = self.label.update(
transition: transition,
component: AnyComponent(Text(text: "Tap an object to cut it out", font: Font.regular(17.0), color: .white)),
component: AnyComponent(Text(text: "Tap on an object to cut it out", font: Font.regular(17.0), color: .white)),
environment: {},
containerSize: CGSize(width: availableSize.width - 88.0, height: 44.0)
)
@@ -229,7 +241,9 @@ private final class MediaCutoutScreenComponent: Component {
transition.setFrame(view: self.buttonsBackgroundView, frame: CGRect(origin: .zero, size: buttonsContainerFrame.size))
transition.setFrame(view: self.previewContainerView, frame: previewContainerFrame)
transition.setFrame(view: self.separatedImageView, frame: previewContainerFrame)
for view in self.outlineViews {
transition.setFrame(view: view, frame: previewContainerFrame)
}
let frameWidth = floor(previewContainerFrame.width * 0.97)
@@ -237,20 +251,28 @@ private final class MediaCutoutScreenComponent: Component {
self.fadeView.layer.cornerRadius = frameWidth / 8.0
if isFirstTime {
let _ = (component.mediaEditor.getSeparatedImage(point: nil)
|> deliverOnMainQueue).start(next: { [weak self] image in
guard let self else {
return
}
self.separatedImageView.image = image
self.state?.updated(transition: .easeInOut(duration: 0.2))
})
} else {
if let _ = self.separatedImageView.image {
transition.setAlpha(view: self.fadeView, alpha: 1.0)
} else {
transition.setAlpha(view: self.fadeView, alpha: 0.0)
let values = component.mediaEditor.values
component.mediaEditor.processImage { originalImage, editedImage in
cutoutImage(from: originalImage, editedImage: editedImage, values: values, target: .all, completion: { results in
Queue.mainQueue().async {
if !results.isEmpty {
for result in results {
if let extractedImage = result.extractedImage, let maskImage = result.maskImage {
if case let .image(image, _) = extractedImage, case let .image(_, mask) = maskImage {
let outlineView = StickerCutoutOutlineView(frame: self.previewContainerView.frame)
outlineView.update(image: image, maskImage: mask, size: self.previewContainerView.bounds.size, values: values)
self.insertSubview(outlineView, belowSubview: self.previewContainerView)
self.outlineViews.append(outlineView)
}
}
}
self.state?.updated(transition: .easeInOut(duration: 0.4))
}
}
})
}
} else {
transition.setAlpha(view: self.fadeView, alpha: !self.outlineViews.isEmpty ? 1.0 : 0.0)
}
return availableSize
}