mirror of https://github.com/DanilaFe/abacus
32 lines
1.3 KiB
Kotlin
32 lines
1.3 KiB
Kotlin
package org.nwapw.abacus.plugin.standard
|
|
|
|
import org.nwapw.abacus.context.MutableEvaluationContext
|
|
import org.nwapw.abacus.function.NumberOperator
|
|
import org.nwapw.abacus.function.OperatorAssociativity
|
|
import org.nwapw.abacus.function.OperatorType
|
|
import org.nwapw.abacus.number.NumberInterface
|
|
|
|
class OperatorFactorial: NumberOperator(OperatorAssociativity.LEFT, OperatorType.UNARY_POSTFIX, 0) {
|
|
|
|
override fun matchesParams(context: MutableEvaluationContext, params: Array<out NumberInterface>) =
|
|
params.size == 1
|
|
&& params[0].fractionalPart().compareTo(context.inheritedNumberImplementation.instanceForString("0")) == 0
|
|
&& params[0].signum() >= 0
|
|
|
|
override fun applyInternal(context: MutableEvaluationContext, params: Array<out NumberInterface>): NumberInterface {
|
|
val implementation = context.inheritedNumberImplementation
|
|
val one = implementation.instanceForString("1")
|
|
if (params[0].signum() == 0) {
|
|
return one
|
|
}
|
|
var factorial = params[0]
|
|
var multiplier = params[0] - one
|
|
//It is necessary to later prevent calls of factorial on anything but non-negative integers.
|
|
while (multiplier.signum() == 1) {
|
|
factorial *= multiplier
|
|
multiplier -= one
|
|
}
|
|
return factorial
|
|
}
|
|
|
|
} |