YYXC_Student/jiaPei/Cluster/BMKClusterAlgorithm.m

//
//  BMKClusterAlgorithm.m
//  IphoneMapSdkDemo
//
//  Created by wzy on 15/9/15.
//  Copyright © 2015年 Baidu. All rights reserved.
//

#import "BMKClusterAlgorithm.h"

#define MAX_DISTANCE_IN_DP    200 //300dp

@implementation BMKClusterAlgorithm

@synthesize quadItems = _quadItems;
@synthesize quadtree = _quadtree;

- (id)init {
    self = [super init];
    if (self) {
        _quadtree = [[BMKClusterQuadtree alloc] initWithRect:CGRectMake(0, 0, 1, 1)];
        _quadItems = [[NSMutableArray alloc] init];
    }
    return self;
}

///添加item
- (void)addItem:(BMKClusterItem*)clusterItem {
    BMKQuadItem *quadItem = [[BMKQuadItem alloc] init];
    quadItem.clusterItem = clusterItem;
    @synchronized(_quadtree) {
        [_quadItems addObject:quadItem];
        [_quadtree addItem:quadItem];
//        NSLog(@"%@", _quadtree.quadItems);
    }
}

///清除items
- (void)clearItems {
    @synchronized(_quadtree) {
        [_quadItems removeAllObjects];
        [_quadtree clearItems];
    }
    
}

/**
 *  cluster算法核心
 * @param zoom map的级别
 * @return BMKCluster数组
 */
- (NSArray*)getClusters:(CGFloat) zoomLevel {
    if (zoomLevel < 3 || zoomLevel > 22) {
        return nil;
    }

    NSMutableArray *results = [NSMutableArray array];
    
    NSUInteger zoom = (NSUInteger)zoomLevel;
    CGFloat zoomSpecificSpan = MAX_DISTANCE_IN_DP / pow(2, zoom) / 256;
    NSMutableSet *visitedCandidates = [NSMutableSet set];
    NSMutableDictionary *distanceToCluster = [NSMutableDictionary dictionary];
    NSMutableDictionary *itemToCluster = [NSMutableDictionary dictionary];
    
    @synchronized(_quadtree) {
        
        for (BMKQuadItem *candidate in _quadItems) {
            //candidate已经添加到另一cluster中
            if ([visitedCandidates containsObject:candidate]) {
                continue;
            }
            BMKCluster *cluster = [[BMKCluster alloc] init];
            cluster.coordinate = candidate.clusterItem.coor;
            cluster.dataDic = candidate.clusterItem.dataDic;
            
            CGRect searchRect = [self getRectWithPt:candidate.pt Span:zoomSpecificSpan];
            NSMutableArray *items = (NSMutableArray*)[_quadtree searchInRect:searchRect];
            if (items.count == 1) {
                [cluster.clusterItems addObject:candidate.clusterItem];
                cluster.dataArr = [NSMutableArray arrayWithArray:@[candidate.clusterItem.dataDic]];
                [results addObject:cluster];
                [visitedCandidates addObject:candidate];
                [distanceToCluster setObject:[NSNumber numberWithDouble:0] forKey:[NSNumber numberWithLongLong:candidate.hash]];
                continue;
            }
            
            NSMutableArray *mArr = [NSMutableArray new];
            for (BMKQuadItem *quadItem in items) {
                NSNumber *existDistache = [distanceToCluster objectForKey:[NSNumber numberWithLongLong:quadItem.hash]];
                CGFloat distance = [self getDistanceSquared:candidate.pt point:quadItem.pt];
                if (existDistache != nil) {
                    if (existDistache.doubleValue < distance) {
                        continue;
                    }
                    BMKCluster *existCluster = [itemToCluster objectForKey:[NSNumber numberWithLongLong:quadItem.hash]];
                    [existCluster.clusterItems removeObject:quadItem.clusterItem];
                    [existCluster.dataArr removeObject:quadItem.clusterItem.dataDic];
                }
                [distanceToCluster setObject:[NSNumber numberWithDouble:distance] forKey:[NSNumber numberWithLongLong:quadItem.hash]];
                [cluster.clusterItems addObject:quadItem.clusterItem];
                [mArr addObject:quadItem.clusterItem.dataDic];
                [itemToCluster setObject:cluster forKey:[NSNumber numberWithLongLong:quadItem.hash]];
            }
            [visitedCandidates addObjectsFromArray:items];
            cluster.dataArr = mArr;
            [results addObject:cluster];
            
        }
    }
    
    return results;
}

- (CGRect)getRectWithPt:(CGPoint) pt  Span:(CGFloat) span {
    CGFloat half = span / 2.f;
    return CGRectMake(pt.x - half, pt.y - half, span, span);
}

- (CGFloat)getDistanceSquared:(CGPoint) pt1 point:(CGPoint) pt2 {
    return (pt1.x - pt2.x) * (pt1.x - pt2.x) + (pt1.y - pt2.y) * (pt1.y - pt2.y);
}

@end