forked from shadowfacts/Tusker
40 lines
960 B
Swift
40 lines
960 B
Swift
//
|
|
// Array+Uniques.swift
|
|
// Tusker
|
|
//
|
|
// Created by Shadowfacts on 1/6/19.
|
|
// Copyright © 2019 Shadowfacts. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
extension Array {
|
|
func uniques<ID: Hashable>(by identify: (Element) -> ID) -> [Element] {
|
|
var uniques = Set<Hashed<Element, ID>>()
|
|
for (index, elem) in self.enumerated() {
|
|
uniques.insert(Hashed(element: elem, id: identify(elem), origIndex: index))
|
|
}
|
|
return uniques.sorted(by: { $0.origIndex < $1.origIndex }).map(\.element)
|
|
}
|
|
}
|
|
|
|
extension Array where Element: Hashable {
|
|
func uniques() -> [Element] {
|
|
return uniques(by: { $0 })
|
|
}
|
|
}
|
|
|
|
fileprivate struct Hashed<Element, ID: Hashable>: Hashable {
|
|
let element: Element
|
|
let id: ID
|
|
let origIndex: Int
|
|
|
|
static func ==(lhs: Self, rhs: Self) -> Bool {
|
|
return lhs.id == rhs.id
|
|
}
|
|
|
|
func hash(into hasher: inout Hasher) {
|
|
hasher.combine(id)
|
|
}
|
|
}
|