import TestsUtils

public var ChainedFilterMap = [
  BenchmarkInfo(name: "ChainedFilterMap", runFunction: run_ChainedFilterMap, tags: [.algorithm]),
  BenchmarkInfo(name: "FatCompactMap", runFunction: run_FatCompactMap, tags: [.algorithm])
]

public let first100k = Array(0...100_000-1)

@inline(never)
public func run_ChainedFilterMap(_ N: Int) {
  var result = 0
  for _ in 1...10*N {
    let numbers = first100k.lazy
      .filter { $0 % 3 == 0 }
      .map { $0 * 2 }
      .filter { $0 % 8 == 0 }
      .map { $0 / 2 + 1}
    result = numbers.reduce(into: 0) { $0 += $1 }
  }

  CheckResults(result == 416691666)
}

@inline(never)
public func run_FatCompactMap(_ N: Int) {
  var result = 0
  for _ in 1...10*N {
    let numbers = first100k.lazy
      .compactMap { (x: Int) -> Int? in
        if x % 3 != 0 { return nil }
        let y = x * 2
        if y % 8 != 0 { return nil }
        let z = y / 2 + 1
        return z
      }
    result = numbers.reduce(into: 0) { $0 += $1 }
  }
  CheckResults(result == 416691666)
}