kscience/kmath
9
0
Fork 1
Code Issues Pull Requests 6 Packages Projects Releases 16 Wiki Activity

Revert "fixes to power"

Browse Source 
This reverts commit bdb9ce6a
This commit is contained in:
Alexander Nozik 2021-05-14 09:14:48 +03:00
parent c1b94ff0bc
commit 1b60f71ed0
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
kmath-core/src/commonMain/kotlin/space/kscience/kmath/operations/algebraExtensions.kt
View File

@ -108,9 +108,9 @@ public fun <T, S> Sequence<T>.averageWith(space: S): T where S : Ring<T>, S : Sc
* @author Evgeniy Zhelenskiy * @author Evgeniy Zhelenskiy
*/ */
public fun <T> Ring<T>.power(arg: T, power: UInt): T = when { public fun <T> Ring<T>.power(arg: T, power: UInt): T = when {
arg == zero && power > 0U -> zero this == zero && power > 0U -> zero
arg == one -> arg this == one -> arg
arg == -one -> powWithoutOptimization(arg, power % 2U) this == -one -> powWithoutOptimization(arg, power % 2U)
else -> powWithoutOptimization(arg, power) else -> powWithoutOptimization(arg, power)
} }
@ -135,7 +135,7 @@ private fun <T> Ring<T>.powWithoutOptimization(base: T, exponent: UInt): T = whe
* @return the base raised to the power. * @return the base raised to the power.
* @author Iaroslav Postovalov, Evgeniy Zhelenskiy * @author Iaroslav Postovalov, Evgeniy Zhelenskiy
*/ */
public fun <T> Field<T>.power(arg: T, power: UInt): T = when { public fun <T> Field<T>.power(arg: T, power: Int): T = when {
power < 0 -> one / (this as Ring<T>).power(arg, power) power < 0 -> one / (this as Ring<T>).power(arg, if (power == Int.MIN_VALUE) Int.MAX_VALUE.toUInt().inc() else (-power).toUInt())
else -> (this as Ring<T>).power(arg, power) else -> (this as Ring<T>).power(arg, power.toUInt())
} }