benchmark/single-source/DictionaryRemove.swift

//===--- DictionaryRemove.swift -------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

// Dictionary element removal benchmark
// rdar://problem/19804127
import TestsUtils

public let DictionaryRemove = [
  BenchmarkInfo(name: "DictionaryRemove", runFunction: run_DictionaryRemove, tags: [.validation, .api, .Dictionary]),
  BenchmarkInfo(name: "DictionaryRemoveOfObjects", runFunction: run_DictionaryRemoveOfObjects, tags: [.validation, .api, .Dictionary]),
]

@inline(never)
public func run_DictionaryRemove(_ N: Int) {
    let size = 100
    var dict = [Int: Int](minimumCapacity: size)

    // Fill dictionary
    for i in 1...size {
        dict[i] = i
    }
    CheckResults(dict.count == size)

    var tmpDict = dict
    for _ in 1...1000*N {
        tmpDict = dict
        // Empty dictionary
        for i in 1...size {
            tmpDict.removeValue(forKey: i)
        }
        if !tmpDict.isEmpty {
            break
        }
    }

    CheckResults(tmpDict.isEmpty)
}

class Box<T : Hashable> : Hashable {
  var value: T

  init(_ v: T) {
    value = v
  }

  func hash(into hasher: inout Hasher) {
    hasher.combine(value)
  }

  static func ==(lhs: Box, rhs: Box) -> Bool {
    return lhs.value == rhs.value
  }
}

@inline(never)
public func run_DictionaryRemoveOfObjects(_ N: Int) {
    let size = 100
    var dict = Dictionary<Box<Int>, Box<Int>>(minimumCapacity: size)

    // Fill dictionary
    for i in 1...size {
        dict[Box(i)] = Box(i)
    }
    CheckResults(dict.count == size)

    var tmpDict = dict
    for _ in 1...1000*N {
        tmpDict = dict
        // Empty dictionary
        for i in 1...size {
            tmpDict.removeValue(forKey: Box(i))
        }
        if !tmpDict.isEmpty {
            break
        }
    }

    CheckResults(tmpDict.isEmpty)
}