1v1/MSYOUPAI/3rd/WMZBanner/WMZBannerControl.m

//
//  WMZBannerControl.m
//  WMZBanner
//
//  Created by wmz on 2019/9/6.
//  Copyright © 2019 wmz. All rights reserved.
//

#import "WMZBannerControl.h"
@implementation WMZBannerControl

- (instancetype)initWithFrame:(CGRect)frame WithModel:(WMZBannerParam *)param{
    if (self = [super initWithFrame:frame]) {
        self.param = param;
        self.userInteractionEnabled = NO;
        self.hidesForSinglePage = YES;
        self.currentPageIndicatorTintColor = param.wBannerControlSelectColor;
        self.pageIndicatorTintColor = param.wBannerControlColor;
        if (param.wBannerControlImage) {
            self.inactiveImage = [UIImage imageNamed:param.wBannerControlImage];
            self.inactiveImageSize = param.wBannerControlImageSize;
            self.pageIndicatorTintColor = [UIColor clearColor];
        }
        if (param.wBannerControlSelectImage) {
            self.currentImage = [UIImage imageNamed:param.wBannerControlSelectImage];
            self.currentImageSize = param.wBannerControlSelectImageSize;
            self.currentPageIndicatorTintColor = [UIColor clearColor];
        }
        
        [self resetFrame];

    }
    return self;
}

- (void)setCurrentPage:(NSInteger)currentPage{
    [super setCurrentPage:currentPage];
    [self updateDots];
}


- (void)updateDots{
    for (int i = 0; i < [self.subviews count]; i++) {
        UIImageView *dot = [self imageViewForSubview:[self.subviews objectAtIndex:i] currPage:i];
        if (i == self.currentPage){
            dot.image = self.currentImage;
            CGRect rect = dot.frame;
            rect.size = self.currentImageSize;
            dot.frame = rect;
            dot.layer.masksToBounds = YES;
            dot.layer.cornerRadius =  self.param.wBannerControlImageRadius?:self. self.currentImageSize.height/2;
        }else{
            dot.image = self.inactiveImage;
            CGRect rect = dot.frame;
            rect.size = self.inactiveImageSize;
            dot.frame = rect;
            dot.layer.masksToBounds = YES;
            dot.layer.cornerRadius = self.param.wBannerControlImageRadius?:self. self.inactiveImageSize.height/2;
        }
    }
}


- (UIImageView *)imageViewForSubview:(UIView *)view currPage:(int)currPage{
    UIImageView *dot = nil;
    if ([view isKindOfClass:[UIView class]]) {
        for (UIView *subview in view.subviews) {
            if ([subview isKindOfClass:[UIImageView class]]) {
                dot = (UIImageView *)subview;
                break;
            }
        }
        if (dot == nil) {
            dot = [[UIImageView alloc] initWithFrame:CGRectMake(0.0f, 0.0f, view.frame.size.width, view.frame.size.height)];
            [view addSubview:dot];
        }
    }else {
        dot = (UIImageView *)view;
    }
    
    return dot;
}

- (void)layoutSubviews
{
    [super layoutSubviews];
    if (self.param.wBannerControlImage&&self.param.wBannerControlSelectImage){
        UIImageView *tmp = nil;
        for (int i=0; i<[self.subviews count]; i++) {
            UIImageView* dot = [self.subviews objectAtIndex:i];
            CGFloat x = (tmp?CGRectGetMaxX(tmp.frame):0)+self.param.wBannerControlSelectMargin;
            CGFloat y = 0;
            if (i == self.currentPage) {
                y = (self.bounds.size.height - self.currentImageSize.height)/2;
                [dot setFrame:CGRectMake(x, y, self.currentImageSize.width, self.currentImageSize.height)];
            }else {
                y = (self.bounds.size.height - self.inactiveImageSize.height)/2;
                [dot setFrame:CGRectMake(x, y, self.inactiveImageSize.width, self.inactiveImageSize.height)];
            }
            tmp = dot;
            if (i == [self.subviews count]-1) {
                CGRect rect = self.frame;
                rect.size.width = CGRectGetMaxX(dot.frame);
                rect.origin.x = (self.param.wFrame.size.width - rect.size.width)/2;
                self.frame = rect;
            }
        }
        [self resetFrame];
    }
}

- (void)resetFrame{
    for (int i=0; i<[self.subviews count]; i++) {
        UIImageView* dot = [self.subviews objectAtIndex:i];
        if (i == [self.subviews count]-1) {
            CGRect rect = self.frame;
            rect.size.width = CGRectGetMaxX(dot.frame);
            rect.origin.x = (self.param.wFrame.size.width - rect.size.width)/2;
            self.frame = rect;
        }
    }
    if (self.param.wBannerControlPosition == BannerControlLeft) {
          CGRect rect = self.frame;
          rect.origin.x = 30;
          self.frame = rect;
      }
      if (self.param.wBannerControlPosition == BannerControlRight) {
          CGRect rect = self.frame;
          rect.origin.x = self.superview.frame.size.width - rect.size.width  - 30;
          self.frame = rect;
      }
      if (self.param.wCustomControl) {
          self.param.wCustomControl(self);
      }
}

@end