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

Various improvements

Browse Source
This commit is contained in:
Ilya Laktyushin
2024-02-29 01:36:56 +04:00
parent 5ce85e7b75
commit 64eaaae4fd
13 changed files with 474 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
submodules/TelegramUI/Components/MediaEditor/Sources/MediaEditor.swift
View File

@@ -190,6 +190,7 @@ public final class MediaEditor {
public private(set) var canCutout: Bool = false
public var canCutoutUpdated: (Bool) -> Void = { _ in }
public var isCutoutUpdated: (Bool) -> Void = { _ in }
private var textureCache: CVMetalTextureCache!
@@ -1682,6 +1683,51 @@ public final class MediaEditor {
self.renderer.renderFrame()
}
public func getSeparatedImage(point: CGPoint?) -> Signal<UIImage?, NoError> {
guard let textureSource = self.renderer.textureSource as? UniversalTextureSource, let image = textureSource.mainImage else {
return .single(nil)
}
return cutoutImage(from: image, atPoint: point, asImage: true)
|> map { result in
if let result, case let .image(image) = result {
return image
} else {
return nil
}
}
}
public func removeSeparationMask() {
self.isCutoutUpdated(false)
self.renderer.currentMainInputMask = nil
if !self.skipRendering {
self.updateRenderChain()
}
}
public func setSeparationMask(point: CGPoint?) {
guard let renderTarget = self.previewView, let device = renderTarget.mtlDevice else {
return
}
guard let textureSource = self.renderer.textureSource as? UniversalTextureSource, let image = textureSource.mainImage else {
return
}
self.isCutoutUpdated(true)
let _ = (cutoutImage(from: image, atPoint: point, asImage: false)
|> deliverOnMainQueue).start(next: { [weak self] result in
guard let self, let result, case let .image(image) = result else {
return
}
//TODO:replace with pixelbuffer
self.renderer.currentMainInputMask = loadTexture(image: image, device: device)
if !self.skipRendering {
self.updateRenderChain()
}
})
}
private func maybeGeneratePersonSegmentation(_ image: UIImage?) {
if #available(iOS 15.0, *), let cgImage = image?.cgImage {
let faceRequest = VNDetectFaceRectanglesRequest { [weak self] request, _ in