Generate contains function for all combinations of primitive numeric ranges and arguments.

#KT-6361
This commit is contained in:
Ilya Gorbunov
2015-07-11 00:46:52 +03:00
parent 8892192e9c
commit b7fbb60db4
4 changed files with 274 additions and 5 deletions
@@ -50,8 +50,6 @@ enum class PrimitiveType {
val descendingByDomainCapacity = listOf(Double, Float, Long, Int, Short, Char, Byte)
fun maxByCapacity(fromType: PrimitiveType, toType: PrimitiveType): PrimitiveType = descendingByDomainCapacity.first { it == fromType || it == toType }
val primitivePermutations = numericPrimitives.flatMap { fromType -> numericPrimitives map { toType -> fromType to toType } } + (Char to Char)
}
}
@@ -72,7 +72,11 @@ fun ranges(): List<GenericFunction> {
body { "return $progressionType($fromExpr, $toExpr, $incrementExpr)" }
}
templates addAll PrimitiveType.primitivePermutations.map { downTo(it.first, it.second) }
val numericPrimitives = PrimitiveType.numericPrimitives
val numericPermutations = numericPrimitives flatMap { fromType -> numericPrimitives map { toType -> fromType to toType }}
val primitivePermutations = numericPermutations + (PrimitiveType.Char to PrimitiveType.Char)
templates addAll primitivePermutations.map { downTo(it.first, it.second) }
fun until(fromType: PrimitiveType, toType: PrimitiveType) = f("until(to: $toType)") {
only(Primitives)
@@ -109,9 +113,19 @@ fun ranges(): List<GenericFunction> {
}
}
templates addAll PrimitiveType.primitivePermutations
templates addAll primitivePermutations
.filter { it.first.isIntegral() && it.second.isIntegral() }
.map { until(it.first, it.second) }
fun contains(rangeType: PrimitiveType, itemType: PrimitiveType) = f("contains(item: $itemType)") {
only(RangesOfPrimitives)
only(rangeType)
returns("Boolean")
doc { "Checks if the specified [item] belongs to this range." }
body { "return start <= item && item <= end" }
}
templates addAll numericPermutations.filter { it.first != it.second }.map { contains(it.first, it.second) }
return templates
}