增加换肤功能

TUIKit/TUIMultimediaPlugin/Edit/UI/CommonEditor/Views/TUIMultimediaMosaicDrawView.m

@@ -0,0 +1,172 @@
+// Copyright (c) 2025 Tencent. All rights reserved.
+// Author: eddardliu
+
+#import "TUIMultimediaMosaicDrawView.h"
+#import <CoreImage/CoreImage.h>
+#import <Masonry/Masonry.h>
+
+#import "TUIMultimediaPlugin/TUIMultimediaDrawView.h"
+
+#define MOSAIC_LINE_WIDTH 20
+#define MOSAIC_SCALE_VALUE 50
+
+@interface TUIMultimediaMosaicDrawView () {
+    UIBezierPath *_currentPath;
+    
+    NSMutableArray<CAShapeLayer *> *_layerList;
+    NSMutableArray<CAShapeLayer *> *_redoLayerList;
+    
+    CALayer *_maskParentLayer;
+    CAShapeLayer *_currentShapeLayer;
+    BOOL _isAdddPathFirstPoint;
+}
+@end
+
+@implementation TUIMultimediaMosaicDrawView
+
+- (instancetype)initWithFrame:(CGRect)frame {
+    if (self = [super initWithFrame:frame]) {
+        [self setupView];
+    }
+    return self;
+}
+
+- (void)setupView {
+    NSLog(@"Mosaic setupView");
+    _maskParentLayer = [CALayer layer];
+    _maskParentLayer.frame = self.bounds;
+    self.layer.mask = _maskParentLayer;
+    
+    _layerList = [NSMutableArray array];
+    _redoLayerList = [NSMutableArray array];
+    _isAdddPathFirstPoint = YES;
+}
+
+- (void)setOriginalImage:(UIImage *)originalImage {
+    NSLog(@"Mosaic setOriginalImage");
+    _originalImage = originalImage;
+    self.image = [self generateMosaicImage:originalImage];
+}
+
+-(void)addPathPoint:(CGPoint) point {
+    if (_isAdddPathFirstPoint) {
+        _isAdddPathFirstPoint = NO;
+        _currentShapeLayer = [self createShapeLayer];
+        [_maskParentLayer addSublayer:_currentShapeLayer];
+        [_layerList addObject:_currentShapeLayer];
+        _currentPath = [UIBezierPath bezierPath];
+        [_currentPath moveToPoint:point];
+    } else {
+        [_currentPath addLineToPoint:point];
+        _currentShapeLayer.path = _currentPath.CGPath;
+    }
+}
+
+-(CAShapeLayer*)createShapeLayer{
+    CAShapeLayer* shapeLayer = [CAShapeLayer layer];
+    shapeLayer.frame = self.bounds;
+    shapeLayer.lineCap = kCALineCapRound;
+    shapeLayer.lineJoin = kCALineJoinRound;
+    shapeLayer.lineWidth = MOSAIC_LINE_WIDTH;
+    shapeLayer.strokeColor = [UIColor blackColor].CGColor;
+    shapeLayer.fillColor = nil;
+    return shapeLayer;
+}
+
+-(void)completeAddPoint {
+    _isAdddPathFirstPoint = YES;
+    for (CAShapeLayer * layer in _redoLayerList) {
+        [self safeRemoveShapeLayer:layer];
+    }
+    [_redoLayerList removeAllObjects];
+}
+
+- (NSInteger) pathCount {
+    return _layerList.count;
+}
+
+- (BOOL)canUndo {
+    return _layerList.count != 0;
+}
+
+- (BOOL)canRedo {
+    return _redoLayerList.count != 0;
+}
+
+- (void)clearMosaic {
+    for (CAShapeLayer * layer in _layerList) {
+        [self safeRemoveShapeLayer:layer];
+    }
+    
+    for (CAShapeLayer * layer in _redoLayerList) {
+        [self safeRemoveShapeLayer:layer];
+    }
+    
+    [_currentPath removeAllPoints];
+    [_layerList removeAllObjects];
+    [_redoLayerList removeAllObjects];
+}
+
+- (void)undo {
+    if (_layerList.count == 0) {
+        return;
+    }
+    CAShapeLayer *layer = [_layerList lastObject];
+    [_layerList removeLastObject];
+    layer.hidden = YES;
+    [_redoLayerList addObject:layer];
+}
+
+- (void)redo {
+    if (_redoLayerList.count == 0) {
+        return;
+    }
+    CAShapeLayer *layer = [_redoLayerList lastObject];
+    [_redoLayerList removeLastObject];
+    layer.hidden = NO;
+    [_layerList addObject:layer];
+}
+
+
+- (void)layoutSubviews {
+    [super layoutSubviews];
+    _maskParentLayer.frame = self.bounds;
+    
+    for (CAShapeLayer *layer in _maskParentLayer.sublayers) {
+        if ([layer isKindOfClass:[CAShapeLayer class]]) {
+            float scale = self.bounds.size.width / layer.frame.size.width;
+            layer.frame = self.bounds;
+            CGAffineTransform transform = CGAffineTransformMakeScale(scale, scale);
+            CGPathRef scaledPath = CGPathCreateCopyByTransformingPath(layer.path, &transform);
+            layer.lineWidth = layer.lineWidth * scale;
+            layer.path = scaledPath;
+            CGPathRelease(scaledPath);
+        }
+    }
+}
+
+- (UIImage *)generateMosaicImage:(UIImage *)inputImage {
+    CIImage *ciImage = [[CIImage alloc] initWithImage:inputImage];
+    
+    CIFilter *filter = [CIFilter filterWithName:@"CIPixellate"];
+    [filter setValue:ciImage forKey:kCIInputImageKey];
+    [filter setValue:@(MOSAIC_SCALE_VALUE) forKey:kCIInputScaleKey];
+    
+    CIImage *outputImage = filter.outputImage;
+    CIContext *context = [CIContext contextWithOptions:nil];
+    CGImageRef cgImage = [context createCGImage:outputImage fromRect:outputImage.extent];
+    return [UIImage imageWithCGImage:cgImage];
+}
+
+- (void)safeRemoveShapeLayer:(CAShapeLayer *)layer {
+    if (!layer) return;
+
+    [layer removeAllAnimations];
+    layer.delegate = nil;
+    layer.path = NULL;
+    layer.fillColor = NULL;
+    layer.strokeColor = NULL;
+    [layer removeFromSuperlayer];
+}
+
+@end