guernikacore / schedulers Goto Github PK

	// https://github.com/GuernikaCore/Schedulers/blob/1f517514d679e38bb9915c3a74bf04f75d5b5875/Sources/Schedulers/DPMSolverMultistepScheduler.swift#L96
        // sd1.5  [999, 850, 736, 645, 545, 455, 343, 233, 124, 24]
        let alignSteps = [999, 845, 730, 587, 443, 310, 193, 116, 53, 13]
        timeSteps = DPMSolverMultistepScheduler.logInterpolation(steps: alignSteps, desiredCount: stepCount).map { Double($0) }

    // https://github.com/GuernikaCore/Schedulers/blob/1f517514d679e38bb9915c3a74bf04f75d5b5875/Sources/Schedulers/Base/Scheduler.swift#L133
    static func logInterpolation(steps: [Int], desiredCount: Int) -> [Int] {
        var interpolatedSteps = [Int]()
        let logSteps = steps.map { log(Double($0)) }
        let stepSize = Double(logSteps.count - 1) / Double(desiredCount - 1)
        for i in 0..<desiredCount {
            let index = Double(i) * stepSize
            let lowerIndex = Int(floor(index))
            let upperIndex = Int(ceil(index))
            
            if lowerIndex == upperIndex {
                interpolatedSteps.append(steps[lowerIndex])
            } else {
                let lowerValue = logSteps[lowerIndex]
                let upperValue = logSteps[upperIndex]
                let fraction = index - Double(lowerIndex)
                let interpolatedValue = exp(lowerValue + fraction * (upperValue - lowerValue))
                interpolatedSteps.append(Int(interpolatedValue))
            }
        }
        return interpolatedSteps
    }