PerspectiveTransform/Example/Tests/NimbleSpecHelper.swift

//
//  NimbleSpecHelper.swift
//  PerspectiveTransform
//
//  Created by Paul Zabelin on 2/19/16.
//  Copyright © 2016 CocoaPods. All rights reserved.
//

import simd
import GameKit
@testable import Nimble

extension CATransform3D {
    func flattened() -> [CGFloat] {
        return [
            m11, m12, m13, m14,
            m21, m22, m23, m24,
            m31, m32, m33, m34,
            m41, m42, m43, m44
        ]
    }
}

public func beCloseTo(_ expectedValue: CATransform3D, within delta: CGFloat = CGFloat(DefaultDelta)) -> Predicate<CATransform3D> {
    let errorMessage = "be close to <\(stringify(expectedValue))> (each within \(stringify(delta)))"
    return Predicate.simple(errorMessage) { actualExpression in
        if let actual = try actualExpression.evaluate() {
            if CATransform3DEqualToTransform(actual, expectedValue) {
                return .matches
            } else {
                let expected = expectedValue.flattened()
                for (index, m) in actual.flattened().enumerated() {
                    if abs(m - expected[index]) > delta {
                        return .doesNotMatch
                    }
                }
                return .matches
            }
        }
        return .doesNotMatch
    }
}

public func beCloseTo(_ expectedValue: CGPoint, within delta: CGFloat = CGFloat(DefaultDelta)) -> Predicate<CGPoint> {
    let errorMessage = "be close to <\(stringify(expectedValue))> (each within \(stringify(delta)))"
    return Predicate.simple(errorMessage) { actualExpression in
        if let actual = try actualExpression.evaluate() {
            if actual == expectedValue {
                return .matches
            } else {
                let eachCoordinate = beCloseTo([expectedValue.x, expectedValue.y].map {Double($0)})
                return try! eachCoordinate.satisfies(actualExpression.cast {[$0!.x, $0!.y].map {Double($0)}}).status
            }
        }
        return .doesNotMatch
    }
}
public func ≈(lhs: Expectation<CATransform3D>, rhs: CATransform3D) {
    lhs.to(beCloseTo(rhs))
}

public func ≈(lhs: Expectation<CGPoint>, rhs: CGPoint) {
    lhs.to(beCloseTo(rhs))
}

extension CATransform3D: Equatable {
    public static func ==(lhs: CATransform3D, rhs: CATransform3D) -> Bool {
        return CATransform3DEqualToTransform(lhs, rhs)
    }
}

extension Int {
    func times(block: () -> Void) {
        for _ in 1...self {
            block()
        }
    }
}

extension GKRandomSource {
    func nextDouble() -> Double {
        return Double(nextUniform())
    }

    func nextPoint() -> CGPoint {
        return CGPoint(x: nextDouble(), y: nextDouble())
    }

    func nextFourPoints() -> [CGPoint] {
        return Array(0...3).map {_ in nextPoint()}
    }
}

extension CGRect {
    func corners() -> [CGPoint] {
        return [
            CGPoint(x: minX, y: minY),  // topLeft
            CGPoint(x: maxX, y: minY),  // topRight
            CGPoint(x: minX, y: maxY),  // bottomLeft
            CGPoint(x: maxX, y: maxY)   // bottomRight
        ]
    }
}