LenzSDK_iOS/LenzCameraNativeModuleForRN/Classes/inner/UI/PanoramaGuideView/PanoramaGuideView.m

//
//  PanoramaGuideView.m
//  LenzCameraNativeModuleForRN
//
//  Created by 王昭威 on 2023/1/25.
//

#import "PanoramaGuideView.h"
#import <YYText/NSAttributedString+YYText.h>
#import <AVFoundation/AVFoundation.h>
#import <YYText/YYLabel.h>
#import <MetalKit/MetalKit.h>
#import "Renderer.h"
#import "LenzHeader.h"


CGFloat const PanoramaGuideViewWOrH = 120;

@interface PanoramaGuideViewPreviewContainer : MTKView

@end

@implementation PanoramaGuideViewPreviewContainer

@end

@interface PanoramaGuideView() <MTKViewDelegate>

@property (nonatomic, strong) PanoramaGuideViewPreviewContainer* container;

@property (nonatomic, assign) CGFloat aspect;

@property (nonatomic, assign) CGFloat offset;

@property (nonatomic, strong) NSTimer* timer;


@property (nonatomic, strong) UIImageView *whiteArrow;

@property (nonatomic, strong) UIView* line;
@property (nonatomic, weak) UIView* superV;
@property (nonatomic, strong)YYLabel *label;


@property (nonatomic) UIImageView *showImageView;
@property (nonatomic) PanoramaArrowDir lastArrowDir;

@property (nonatomic) NSNumber *offSetTop;
@property (nonatomic) NSNumber *offSetLeft;

@end

@implementation PanoramaGuideView

- (instancetype)init{
    self = [super init];
    self.backgroundColor = [UIColor jk_colorWithHexString:@"#000000" andAlpha:0.5];
    [self addSubview:self.showImageView];
    self.offSetTop = @((120-38)/2);
    self.offSetLeft = @((120-38)/2);
    return self;
}

- (void)willMoveToSuperview:(nullable UIView *)newSuperview;{
    [super willMoveToSuperview:newSuperview];
    self.superV = newSuperview;
}

- (void)updatePreviewAspect:(CGFloat)aspect{
    if(aspect > 0.01){
        self.aspect = 1.0 / aspect;
    }
    else{
        self.aspect = 1;
    }
    
    [self.container mas_remakeConstraints:^(MASConstraintMaker *make) {
        make.left.top.equalTo(self).offset(self.offset);
        make.height.equalTo(self);
        //  it means aspect > 0
        make.width.equalTo(self.container.mas_height).multipliedBy(self.aspect);
    }];
}

- (void)updateOffset:(CGFloat)offset{
    self.offset = offset;
    [self.container mas_remakeConstraints:^(MASConstraintMaker *make) {
        make.left.equalTo(self).offset(self.offset);
        make.top.equalTo(self);
        make.height.equalTo(self);
        make.width.equalTo(self.container.mas_height).multipliedBy(self.aspect);
    }];
}

- (void)direction:(PanoramaOrientationViewDirectionEnum)dir{

    [self insertSubview:self.line atIndex:0];
    self.line.hidden = YES;
    [self.superV addSubview:self.label];
    [self addSubview:self.whiteArrow];
    [self.line mas_remakeConstraints:^(MASConstraintMaker *make) {
        switch (dir) {
            case PanoramaOrientationViewLeft:
            case PanoramaOrientationViewRight:
            {
                make.left.right.offset(0);
                make.height.offset(0.5);
                make.centerY.mas_equalTo(self);
            }
                break;
            case PanoramaOrientationViewUp:
            case PanoramaOrientationViewDown:
            {
                make.top.bottom.offset(0);
                make.width.offset(0.5);
                make.centerX.mas_equalTo(self);
            }
                break;
                
            default:
                break;
        }
    }];
    
    
    [self.label mas_remakeConstraints:^(MASConstraintMaker *make) {
        switch (dir) {
            case PanoramaOrientationViewLeft:
            case PanoramaOrientationViewRight:  // 横
            {
                make.top.mas_equalTo(self.mas_bottom).offset(5);
                make.centerX.mas_equalTo(self.superV);
                make.height.offset(20);
                make.width.offset(160);
                self.label.verticalForm = NO;
            }
                break;
            case PanoramaOrientationViewUp:
            case PanoramaOrientationViewDown: // 上下
            {
                make.right.mas_equalTo(self.mas_left).offset(-5);
                make.centerY.mas_equalTo(self.superV);
                make.width.offset(20);
                make.height.offset(160);
                self.label.verticalForm = YES;
            }
                break;
                
            default:
                break;
        }
    }];

    [self.showImageView mas_remakeConstraints:^(MASConstraintMaker *make) {
        switch (dir) {
            case PanoramaOrientationViewUp: {
                make.left.right.top.mas_equalTo(0);
                make.height.mas_equalTo(0);
                make.height.mas_lessThanOrEqualTo(self);

            }
                break;
            case PanoramaOrientationViewDown: {
                make.left.right.bottom.mas_equalTo(0);
                make.height.mas_equalTo(0);


            }
                break;
            case PanoramaOrientationViewLeft: {
                make.left.top.bottom.mas_equalTo(0);
                make.width.mas_equalTo(0);
                make.width.mas_lessThanOrEqualTo(self);

            }
                break;
            case PanoramaOrientationViewRight: {
                make.top.right.bottom.mas_equalTo(0);
                make.width.mas_equalTo(0);
                make.width.mas_lessThanOrEqualTo(self);
                

            }
                break;
            default:
                break;
        }
    }];
    
    [UIView animateWithDuration:0.2 animations:^{
        __block UIImage *arrow = nil;
        /// 白色的箭头
        [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
            switch (dir) {
                case PanoramaOrientationViewLeft: //左
                {
                    
                    make.centerY.mas_equalTo(self);
                    make.left.offset(0);
                    arrow = [UIImage loadNamed:@"ai_right_arrow"];
                }
                    break;
                case PanoramaOrientationViewRight:  // 右
                {
                    make.centerY.mas_equalTo(self);
                    make.right.offset(0);
                    arrow = [UIImage loadNamed:@"ai_left_arrow"];

                }
                    break;
                case PanoramaOrientationViewUp: //上
                {
                    
                    make.centerX.mas_equalTo(self);
                    make.top.offset(0);
                    arrow = [UIImage loadNamed:@"ai_down_arrow"];
                }
                    break;
                case PanoramaOrientationViewDown: // 下
                {
                    make.centerX.mas_equalTo(self);
                    make.bottom.offset(0);
                    arrow = [UIImage loadNamed:@"ai_up_arrow"];

                }
                    break;
                    
                default:
                    break;
            }
        }];
        self.whiteArrow.image = arrow;
        [self setNeedsLayout];
        [self layoutIfNeeded];
    } completion:^(BOOL finished) {
        self.line.hidden = NO;
    }];
}

- (void)showImageWith:(UIImage * _Nullable)image dir:(PanoramaOrientationViewDirectionEnum)dir {
//    if (!image) {
//        dispatch_async(dispatch_get_main_queue(), ^{
//            self.showImageView.image = nil;
//        });
//        return;
//    }
    dispatch_async(dispatch_get_main_queue(), ^{
        CGFloat imageWH = 0;
        if (image) {
            if (dir == PanoramaOrientationViewDown || dir == PanoramaOrientationViewUp) {
                imageWH = (PanoramaGuideViewWOrH * image.size.height)/image.size.width;
            } else {
                imageWH = (PanoramaGuideViewWOrH * image.size.width)/image.size.height;
            }
        } else {
            self.offSetTop = @((120-38)/2);
            self.offSetLeft = @((120-38)/2);
        }

    //    UIImage *resultImage = [self showDir:dir image:image];
        self.showImageView.image = image;
        [UIView animateWithDuration:.15 animations:^{
            switch (dir) {
                case PanoramaOrientationViewDown: {
                    [self.showImageView mas_remakeConstraints:^(MASConstraintMaker *make) {
                        make.height.mas_equalTo(imageWH);
                        make.height.mas_lessThanOrEqualTo(self);
                        make.left.right.bottom.mas_equalTo(0);
                    }];
                    [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
                        make.bottom.offset(-imageWH);
                        make.left.mas_equalTo(self.offSetLeft.doubleValue);
//                        make.centerX.mas_equalTo(self);
                    }];
                }
                    break;
                case PanoramaOrientationViewUp:{
                    [self.showImageView mas_remakeConstraints:^(MASConstraintMaker *make) {
                        make.left.right.top.mas_equalTo(0);
                        make.height.mas_equalTo(imageWH);
                        make.height.mas_lessThanOrEqualTo(self);
                    }];
                    [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
                        make.top.offset(imageWH);
//                        make.centerX.mas_equalTo(self);
                        make.left.mas_equalTo(self.offSetLeft.doubleValue);

                    }];
                            
                }
                    break;
                case PanoramaOrientationViewRight: {
                    [self.showImageView mas_remakeConstraints:^(MASConstraintMaker *make) {
                        make.width.mas_equalTo(imageWH);
                        make.width.mas_lessThanOrEqualTo(self);
                        make.top.right.bottom.mas_equalTo(0);
                    }];
                    
                    [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
                        make.right.offset(-imageWH);
//                        make.centerY.mas_equalTo(self);
                        make.top.mas_equalTo(self.offSetTop.doubleValue);

                    }];
                }
                    break;
                    
                case PanoramaOrientationViewLeft:{
                    [self.showImageView mas_remakeConstraints:^(MASConstraintMaker *make) {
                        make.width.mas_equalTo(imageWH);
                        make.width.mas_lessThanOrEqualTo(self);
                        make.left.top.bottom.mas_equalTo(0);
                    }];
                    
                    [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
                        make.left.offset(imageWH);
//                        make.centerY.mas_equalTo(self);
                        make.top.mas_equalTo(self.offSetTop.doubleValue);

                    }];
                }
                    
                    break;
                    
                default:
                    break;
            }
            [self setNeedsLayout];
            [self layoutIfNeeded];
        }];
    });
    
    
    
}

- (void)updateArrowWith:(PanoramaArrowDir)arrowDir dir:(PanoramaOrientationViewDirectionEnum)dir {

//    self.lastArrowDir = arrowDir;
    dispatch_async(dispatch_get_main_queue(), ^{
        if (!self.showImageView.image) {
            return;
        }
        CGFloat imageWidth = self.showImageView.image.size.width;
        CGFloat imageHight = self.showImageView.image.size.height;
        if (dir == PanoramaOrientationViewDown || dir == PanoramaOrientationViewUp) {
            if (arrowDir == PanoramaArrowDirCenter) {
                self.offSetLeft = @((120-38)/2);
//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.bottom.offset(-(PanoramaGuideViewWOrH * imageWidth)/imageHight);
////                    make.centerX.mas_equalTo(self);
//                    make.left.mas_offset(self.offSetLeft.doubleValue);
//                }];
            } else if (arrowDir == PanoramaArrowDirLeft) {
                self.offSetLeft = @(0);
//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.bottom.offset(-(PanoramaGuideViewWOrH * imageWidth)/imageHight);
////                    make.left.mas_equalTo(self).mas_offset(10);
//                    make.left.mas_offset(self.offSetLeft.doubleValue);
//
//                }];
            } else if (arrowDir == PanoramaArrowDirRight) {
                self.offSetLeft = @(120-38);

//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.bottom.offset(-(PanoramaGuideViewWOrH * imageWidth)/imageHight);
////                    make.right.mas_equalTo(self).mas_offset(-10);
//                    make.left.mas_offset(self.offSetLeft.doubleValue);
//
//                }];
            }
        }
//        else if (dir == PanoramaOrientationViewUp) {
//            if (arrowDir == PanoramaArrowDirCenter) {
//                self.offSetTop = @((120-38)/2);
//
////                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
////                    make.top.offset((PanoramaGuideViewWOrH * imageWidth)/imageHight);
//////                    make.centerX.mas_equalTo(self);
////                    make.top.mas_offset(self.offSetTop.doubleValue);
////                }];
//            } else if (arrowDir == PanoramaArrowDirLeft) {
//                self.offSetTop = @((120-38)/2);
//
//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.top.offset((PanoramaGuideViewWOrH * imageWidth)/imageHight);
//                    make.left.mas_equalTo(self).mas_offset(10);
//                }];
//            } else if (arrowDir == PanoramaArrowDirRight) {
//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.top.offset((PanoramaGuideViewWOrH * imageWidth)/imageHight);
//                    make.right.mas_equalTo(self).mas_offset(-10);
//                }];
//            }
//
//        }
        else if (dir == PanoramaOrientationViewLeft || dir == PanoramaOrientationViewRight) {
            if (arrowDir == PanoramaArrowDirCenter) {
                self.offSetTop = @((120-38)/2);
//
//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.left.offset((PanoramaGuideViewWOrH * imageWidth)/imageHight);
////                    make.centerY.mas_equalTo(self);
//                    make.top.mas_offset(self.offSetTop.doubleValue);
//
//                }];
            } else if (arrowDir == PanoramaArrowDirUp) {
                self.offSetTop = @(0);

//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.left.offset((PanoramaGuideViewWOrH * imageWidth)/imageHight);
////                    make.top.mas_equalTo(self).mas_offset(10);
//                    make.top.mas_offset(self.offSetTop.doubleValue);
//
//                }];
            } else if (arrowDir == PanoramaArrowDirDown) {
                self.offSetTop = @(120-38);

//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.left.offset((PanoramaGuideViewWOrH * imageWidth)/imageHight);
////                    make.bottom.mas_equalTo(self).mas_offset(-10);
//                    make.top.mas_offset(self.offSetTop.doubleValue);
//
//                }];
            }
            
        }
//        else if (dir == PanoramaOrientationViewRight) {
//            if (arrowDir == PanoramaArrowDirCenter) {
//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.left.offset((PanoramaGuideViewWOrH * imageWidth)/imageHight);
//                    make.centerY.mas_equalTo(self);
//                }];
//            } else if (arrowDir == PanoramaArrowDirUp) {
//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.left.offset((PanoramaGuideViewWOrH * imageWidth)/imageHight);
//                    make.top.mas_equalTo(self).mas_offset(10);
//                }];
//            } else if (arrowDir == PanoramaArrowDirDown) {
//                [self.whiteArrow mas_remakeConstraints:^(MASConstraintMaker *make) {
//                    make.right.offset(-(PanoramaGuideViewWOrH * imageWidth)/imageHight);
//                    make.bottom.mas_equalTo(self).mas_offset(-10);
//                }];
//            }
//        }
        
    });
}

- (void)clearImage {
    dispatch_async(dispatch_get_main_queue(), ^{
        self.showImageView.image = nil;
    });
}
- (UIImage *)showDir:(PanoramaOrientationViewDirectionEnum)dir image:(UIImage *)image {
    switch (dir) {
        case PanoramaOrientationViewUp: {
            UIImage *flipImage = [UIImage imageWithCGImage:image.CGImage scale:image.scale orientation:UIImageOrientationRight];

            return flipImage;
        }
            break;
        case PanoramaOrientationViewDown: {
            UIImage *flipImage = [UIImage imageWithCGImage:image.CGImage scale:image.scale orientation:UIImageOrientationLeft];

            return flipImage;
        }
            
            
            break;
        case PanoramaOrientationViewLeft: {
            return image;
        }
            
            break;
        case PanoramaOrientationViewRight: {
            UIImage *flipImage = [UIImage imageWithCGImage:image.CGImage scale:image.scale orientation:UIImageOrientationUpMirrored];
            return flipImage;
        }
            break;
            
        default:
            break;
    }
    
    return nil;
}

- (void)setHidden:(BOOL)hidden{
    [super setHidden:hidden];
    self.label.hidden = hidden;
}

- (void)awakeFromNib{
    [super awakeFromNib];
    
    self.container = [[PanoramaGuideViewPreviewContainer alloc] initWithFrame:CGRectMake(0, 0, 50, 50) device:[Renderer shared].device];
    self.container.clearColor = MTLClearColorMake(0, 0, 0, 1);
    self.container.backgroundColor = [UIColor redColor];
    [self addSubview:self.container];
    self.container.delegate = self;
    self.container.sampleCount = 1;
    self.container.colorPixelFormat = MTLPixelFormatBGRA8Unorm;
    [self.container mas_makeConstraints:^(MASConstraintMaker *make) {
        make.left.top.equalTo(self);
        make.height.equalTo(self);
        make.width.equalTo(@(50));
    }];
    
    [[Renderer shared] setup];
    
}

- (void)mtkView:(MTKView *)view drawableSizeWillChange:(CGSize)size{
    [[Renderer shared] mtkView:view drawableSizeWillChange:size];
}

- (void)drawInMTKView:(MTKView *)view{
    [[Renderer shared] drawInMTKView:view];
}

- (YYLabel *)label{
    if (_label == nil) {
        NSAttributedString *att = [[NSAttributedString alloc]initWithString:@"请沿箭头方向匀速拍摄"];
        _label = [[YYLabel alloc]init];
        _label.numberOfLines = 0;
        _label.backgroundColor = [UIColor jk_colorWithHexString:@"000000" andAlpha:0.35];
        _label.attributedText = att;
        _label.textAlignment = NSTextAlignmentCenter;
        _label.layer.cornerRadius = 3;
        _label.layer.masksToBounds = YES;
        _label.textVerticalAlignment = YYTextVerticalAlignmentCenter; // 文字置顶显示
        _label.textColor = UIColor.whiteColor;
    }
    return _label;
}

- (UIImageView *)showImageView {
    if (!_showImageView) {
        _showImageView = [[UIImageView alloc]init];
        _showImageView.contentMode = UIViewContentModeScaleAspectFill;
        _showImageView.clipsToBounds = YES;
    }
    return _showImageView;
}
- (UIView *)line{
    if (_line == nil) {
        _line = [[UIView alloc]init];
        _line.backgroundColor = UIColor.yellowColor;
    }
    return _line;
}

- (UIImageView *)whiteArrow{
    if (_whiteArrow == nil) {
        _whiteArrow = [[UIImageView alloc]initWithImage:[UIImage loadNamed:@"ai_right_arrow"]];
    }
    return _whiteArrow;
}

/*
// Only override drawRect: if you perform custom drawing.
// An empty implementation adversely affects performance during animation.
- (void)drawRect:(CGRect)rect {
    // Drawing code
}
*/

@end