Camera and editor improvements

2025-12-23 22:55:00 +00:00 · 2023-05-15 15:12:51 +04:00
parent 37e7a74e4e
commit a2d7cfba4f
39 changed files with 1877 additions and 1338 deletions
--- a/submodules/TelegramUI/Components/MediaEditor/Sources/Drawing/DrawingStickerEntity.swift
+++ b/submodules/TelegramUI/Components/MediaEditor/Sources/Drawing/DrawingStickerEntity.swift
@@ -0,0 +1,113 @@
+import Foundation
+import UIKit
+import Display
+import AccountContext
+import TelegramCore
+
+public final class DrawingStickerEntity: DrawingEntity, Codable {
+    public enum Content {
+        case file(TelegramMediaFile)
+        case image(UIImage)
+    }
+    private enum CodingKeys: String, CodingKey {
+        case uuid
+        case file
+        case image
+        case referenceDrawingSize
+        case position
+        case scale
+        case rotation
+        case mirrored
+    }
+    
+    public let uuid: UUID
+    public let content: Content
+    
+    public var referenceDrawingSize: CGSize
+    public var position: CGPoint
+    public var scale: CGFloat
+    public var rotation: CGFloat
+    public var mirrored: Bool
+    
+    public var color: DrawingColor = DrawingColor.clear
+    public var lineWidth: CGFloat = 0.0
+    
+    public var center: CGPoint {
+        return self.position
+    }
+    
+    public var baseSize: CGSize {
+        let size = max(10.0, min(self.referenceDrawingSize.width, self.referenceDrawingSize.height) * 0.2)
+        return CGSize(width: size, height: size)
+    }
+    
+    public var isAnimated: Bool {
+        switch self.content {
+        case let .file(file):
+            return file.isAnimatedSticker || file.isVideoSticker || file.mimeType == "video/webm"
+        case .image:
+            return false
+        }
+    }
+    
+    public var isMedia: Bool {
+        return false
+    }
+    
+    public var renderImage: UIImage?
+    public var renderSubEntities: [DrawingEntity]?
+    
+    public init(content: Content) {
+        self.uuid = UUID()
+        self.content = content
+        
+        self.referenceDrawingSize = .zero
+        self.position = CGPoint()
+        self.scale = 1.0
+        self.rotation = 0.0
+        self.mirrored = false
+    }
+    
+    public init(from decoder: Decoder) throws {
+        let container = try decoder.container(keyedBy: CodingKeys.self)
+        self.uuid = try container.decode(UUID.self, forKey: .uuid)
+        if let file = try container.decodeIfPresent(TelegramMediaFile.self, forKey: .file) {
+            self.content = .file(file)
+        } else if let imageData = try container.decodeIfPresent(Data.self, forKey: .image), let image = UIImage(data: imageData) {
+            self.content = .image(image)
+        } else {
+            fatalError()
+        }
+        self.referenceDrawingSize = try container.decode(CGSize.self, forKey: .referenceDrawingSize)
+        self.position = try container.decode(CGPoint.self, forKey: .position)
+        self.scale = try container.decode(CGFloat.self, forKey: .scale)
+        self.rotation = try container.decode(CGFloat.self, forKey: .rotation)
+        self.mirrored = try container.decode(Bool.self, forKey: .mirrored)
+    }
+    
+    public func encode(to encoder: Encoder) throws {
+        var container = encoder.container(keyedBy: CodingKeys.self)
+        try container.encode(self.uuid, forKey: .uuid)
+        switch self.content {
+        case let .file(file):
+            try container.encode(file, forKey: .file)
+        case let .image(image):
+            try container.encodeIfPresent(image.pngData(), forKey: .image)
+        }
+        try container.encode(self.referenceDrawingSize, forKey: .referenceDrawingSize)
+        try container.encode(self.position, forKey: .position)
+        try container.encode(self.scale, forKey: .scale)
+        try container.encode(self.rotation, forKey: .rotation)
+        try container.encode(self.mirrored, forKey: .mirrored)
+    }
+        
+    public func duplicate() -> DrawingEntity {
+        let newEntity = DrawingStickerEntity(content: self.content)
+        newEntity.referenceDrawingSize = self.referenceDrawingSize
+        newEntity.position = self.position
+        newEntity.scale = self.scale
+        newEntity.rotation = self.rotation
+        newEntity.mirrored = self.mirrored
+        return newEntity
+    }
+}