AnnotationView.swift

//
// AnnotationView.swift
// Example
//
// Created by Lasha Efremidze on 10/9/18.
// Copyright © 2018 efremidze. All rights reserved.
//

import UIKit
import MapKit
import Cluster

class CountClusterAnnotationView: ClusterAnnotationView {
 override func configure() {
 super.configure()
 
 guard let annotation = annotation as? ClusterAnnotation else { return }
 let count = annotation.annotations.count
 let diameter = radius(for: count) * 2
 self.frame.size = CGSize(width: diameter, height: diameter)
 self.layer.cornerRadius = self.frame.width / 2
 self.layer.masksToBounds = true
 self.layer.borderColor = UIColor.white.cgColor
 self.layer.borderWidth = 1.5
 }
 
 func radius(for count: Int) -> CGFloat {
 if count < 5 {
 return 12
 } else if count < 10 {
 return 16
 } else {
 return 20
 }
 }
}

class ImageCountClusterAnnotationView: ClusterAnnotationView {
 lazy var once: Void = { [unowned self] in
 self.countLabel.frame.size.width -= 6
 self.countLabel.frame.origin.x += 3
 self.countLabel.frame.origin.y -= 6
 }()
 override func layoutSubviews() {
 super.layoutSubviews()
 
 _ = once
 }
}