teldufalsari/kmath
2
0
Fork 0
forked from kscience/kmath
Code Pull Requests 1 Activity

fixes to power

Browse Source
This commit is contained in:
Alexander Nozik 2021-05-14 09:10:05 +03:00 committed by GitHub
parent a299003236
commit bdb9ce6a59
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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 {
this == zero && power > 0U -> zero arg == zero && power > 0U -> zero
this == one -> arg arg == one -> arg
this == -one -> powWithoutOptimization(arg, power % 2U) arg == -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: Int): T = when { public fun <T> Field<T>.power(arg: T, power: UInt): T = when {
power < 0 -> one / (this as Ring<T>).power(arg, if (power == Int.MIN_VALUE) Int.MAX_VALUE.toUInt().inc() else (-power).toUInt()) power < 0 -> one / (this as Ring<T>).power(arg, power)
else -> (this as Ring<T>).power(arg, power.toUInt()) else -> (this as Ring<T>).power(arg, power)
} }