azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement sum extension function for UArrays (KT-28779)

Browse Source
This commit is contained in:
Abduqodiri Qurbonzoda
2019-01-22 21:19:38 +03:00
committed by Ilya Gorbunov
parent 690e35f11a
commit 7695b5e575
11 changed files with 330 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libraries/tools/binary-compatibility-validator/reference-public-api/kotlin-stdlib-runtime-merged.txt
+15
View File
@@ -2387,6 +2387,10 @@ public final class kotlin/collections/UArraysKt {
public static final fun sliceArray-ojwP5H8 ([SLjava/util/Collection;)[S
public static final fun sliceArray-tAntMlw ([ILkotlin/ranges/IntRange;)[I
public static final fun sliceArray-xo_DsdI ([BLjava/util/Collection;)[B
public static final fun sumOfUByte ([Lkotlin/UByte;)I
public static final fun sumOfUInt ([Lkotlin/UInt;)I
public static final fun sumOfULong ([Lkotlin/ULong;)J
public static final fun sumOfUShort ([Lkotlin/UShort;)I
public static final fun toTypedArray--ajY-9A ([I)[Lkotlin/UInt;
public static final fun toTypedArray-GBYM_sE ([B)[Lkotlin/UByte;
public static final fun toTypedArray-QwZRm1k ([J)[Lkotlin/ULong;
@@ -2406,6 +2410,10 @@ public abstract class kotlin/collections/UByteIterator : java/util/Iterator, kot
}
public final class kotlin/collections/UCollectionsKt {
public static final fun sumOfUByte (Ljava/lang/Iterable;)I
public static final fun sumOfUInt (Ljava/lang/Iterable;)I
public static final fun sumOfULong (Ljava/lang/Iterable;)J
public static final fun sumOfUShort (Ljava/lang/Iterable;)I
public static final fun toUByteArray (Ljava/util/Collection;)[B
public static final fun toUIntArray (Ljava/util/Collection;)[I
public static final fun toULongArray (Ljava/util/Collection;)[J
@@ -4566,6 +4574,13 @@ public final class kotlin/sequences/SequencesKt {
public static final fun zipWithNext (Lkotlin/sequences/Sequence;Lkotlin/jvm/functions/Function2;)Lkotlin/sequences/Sequence;
}
public final class kotlin/sequences/USequencesKt {
public static final fun sumOfUByte (Lkotlin/sequences/Sequence;)I
public static final fun sumOfUInt (Lkotlin/sequences/Sequence;)I
public static final fun sumOfULong (Lkotlin/sequences/Sequence;)J
public static final fun sumOfUShort (Lkotlin/sequences/Sequence;)I
}
public final class kotlin/system/TimingKt {
public static final fun measureNanoTime (Lkotlin/jvm/functions/Function0;)J
public static final fun measureTimeMillis (Lkotlin/jvm/functions/Function0;)J
libraries/tools/kotlin-stdlib-gen/src/templates/Arrays.kt
+1 -1
View File
@@ -968,7 +968,7 @@ object ArrayOps : TemplateGroupBase() {
PrimitiveType.Char ->
body { "return withType(\"CharArray\", fillFrom(this, ${primitive}Array(newSize)))" }
PrimitiveType.Long ->
body { "return withType(\"LongArray\", arrayCopyResize(this, newSize, ZERO))" }
body { "return withType(\"LongArray\", arrayCopyResize(this, newSize, ${primitive!!.zero()}))" }
else ->
body { "return fillFrom(this, ${primitive}Array(newSize))" }
}
libraries/tools/kotlin-stdlib-gen/src/templates/Numeric.kt
+44 -10
View File
@@ -5,31 +5,65 @@
package templates
import templates.Family.*
object Numeric : TemplateGroupBase() {
init {
defaultBuilder {
sequenceClassification(SequenceClass.terminal)
specialFor(ArraysOfUnsigned) {
since("1.3")
annotation("@ExperimentalUnsignedTypes")
}
}
}
private val numericPrimitivesDefaultOrder = PrimitiveType.defaultPrimitives intersect PrimitiveType.numericPrimitives
private val summablePrimitives = numericPrimitivesDefaultOrder + PrimitiveType.unsignedPrimitives
val f_sum = fn("sum()") {
Family.defaultFamilies.forEach { family -> include(family, numericPrimitivesDefaultOrder) }
listOf(Iterables, Sequences, ArraysOfObjects).forEach { include(it, summablePrimitives) }
include(ArraysOfPrimitives, numericPrimitivesDefaultOrder)
include(ArraysOfUnsigned)
} builder {
val p = primitive!!
doc { "Returns the sum of all elements in the ${f.collection}." }
returns("SUM")
platformName("sumOf<T>")
body {
"""
var sum: SUM = ZERO
for (element in this) {
sum += element
returns(p.sumType().name)
specialFor(ArraysOfUnsigned) {
inlineOnly()
body {
if (p == p.sumType())
"return storage.sum().to${p.sumType().name}()"
else
"return sumBy { it.to${p.sumType().name}() }"
}
}
specialFor(Iterables, Sequences, ArraysOfObjects, ArraysOfPrimitives) {
platformName("sumOf<T>")
if (p.isUnsigned()) {
require(f != ArraysOfPrimitives) { "Arrays of unsigneds are separate from arrays of primitives." }
specialFor(Iterables) { sourceFile(SourceFile.UCollections) }
specialFor(Sequences) { sourceFile(SourceFile.USequences) }
specialFor(ArraysOfObjects) { sourceFile(SourceFile.UArrays) }
since("1.3")
annotation("@ExperimentalUnsignedTypes")
}
body {
"""
var sum: ${p.sumType().name} = ${p.sumType().zero()}
for (element in this) {
sum += element
}
return sum
"""
}
return sum
"""
}
}
libraries/tools/kotlin-stdlib-gen/src/templates/dsl/CommonTypes.kt
+16
View File
@@ -75,6 +75,22 @@ enum class PrimitiveType {
fun PrimitiveType.isIntegral(): Boolean = this in PrimitiveType.integralPrimitives
fun PrimitiveType.isNumeric(): Boolean = this in PrimitiveType.numericPrimitives
fun PrimitiveType.isFloatingPoint(): Boolean = this in PrimitiveType.floatingPointPrimitives
fun PrimitiveType.isUnsigned(): Boolean = this in PrimitiveType.unsignedPrimitives
fun PrimitiveType.sumType() = when (this) {
PrimitiveType.Byte, PrimitiveType.Short, PrimitiveType.Char -> PrimitiveType.Int
PrimitiveType.UByte, PrimitiveType.UShort -> PrimitiveType.UInt
else -> this
}
fun PrimitiveType.zero() = when (this) {
PrimitiveType.Double -> "0.0"
PrimitiveType.Float -> "0.0f"
PrimitiveType.Long -> "0L"
PrimitiveType.ULong -> "0uL"
in PrimitiveType.unsignedPrimitives -> "0u"
else -> "0"
}
enum class Inline {
No,
libraries/tools/kotlin-stdlib-gen/src/templates/dsl/MemberBuilder.kt
+3 -12
View File
@@ -194,28 +194,19 @@ class MemberBuilder(
"RECEIVER" -> receiver
"SELF" -> self
"PRIMITIVE" -> primitive?.name ?: token
"SUM" -> {
when (primitive) {
PrimitiveType.Byte, PrimitiveType.Short, PrimitiveType.Char -> "Int"
else -> primitive
}
}
"ZERO" -> when (primitive) {
PrimitiveType.Double -> "0.0"
PrimitiveType.Float -> "0.0f"
PrimitiveType.Long -> "0L"
else -> "0"
}
"ONE" -> when (primitive) {
PrimitiveType.Double -> "1.0"
PrimitiveType.Float -> "1.0f"
PrimitiveType.Long -> "1L"
PrimitiveType.ULong -> "1uL"
in PrimitiveType.unsignedPrimitives -> "1u"
else -> "1"
}
"-ONE" -> when (primitive) {
PrimitiveType.Double -> "-1.0"
PrimitiveType.Float -> "-1.0f"
PrimitiveType.Long -> "-1L"
in PrimitiveType.unsignedPrimitives -> error("-ONE is not in the domain of unsigned primitives")
else -> "-1"
}
"TCollection" -> {
libraries/tools/kotlin-stdlib-gen/src/templates/dsl/SourceFile.kt
+1
View File
@@ -14,6 +14,7 @@ enum class SourceFile(jvmClassName: String? = null, val multifile: Boolean = tru
Sets(packageName = "kotlin.collections"),
Maps(packageName = "kotlin.collections"),
Sequences(packageName = "kotlin.sequences"),
USequences(packageName = "kotlin.sequences"),
Ranges(packageName = "kotlin.ranges"),
URanges(packageName = "kotlin.ranges"),
Comparisons(packageName = "kotlin.comparisons"),