import * as neataptic from "neataptic"

export const allRatePolicyNames = [
    "FIXED",
    "STEP",
    "EXP",
    "INV",
] as const

export type RatePolicyName = typeof allRatePolicyNames[number]

export type RateFn = (baseRate: number, iteration: number) => number

export type RatePolicy = (gamma: number, power: number) => RateFn

export const FIXED: RatePolicy = neataptic.methods.rate.FIXED

export const STEP: RatePolicy = neataptic.methods.rate.STEP

export const EXP: RatePolicy = neataptic.methods.rate.EXP

export const INV: RatePolicy = neataptic.methods.rate.INV

export function getRatePolicy(name: RatePolicyName): RatePolicy {
    return neataptic.methods.rate[name]
}