|
| 1 | +package dev.silenium.compose.gl.surface |
| 2 | + |
| 3 | +import kotlinx.coroutines.flow.MutableStateFlow |
| 4 | +import kotlinx.coroutines.flow.StateFlow |
| 5 | +import kotlinx.coroutines.flow.asStateFlow |
| 6 | +import kotlin.time.Duration |
| 7 | +import kotlin.time.Duration.Companion.seconds |
| 8 | + |
| 9 | +interface Stats<T : Comparable<T>> { |
| 10 | + val values: List<T> |
| 11 | + val sum: T |
| 12 | + val average: T |
| 13 | + val min: T |
| 14 | + val max: T |
| 15 | + val median: T |
| 16 | + fun percentile(percentile: Double, direction: Percentile = Percentile.UP): T |
| 17 | + |
| 18 | + enum class Percentile { |
| 19 | + UP, LOWEST |
| 20 | + } |
| 21 | +} |
| 22 | + |
| 23 | +data class DurationStats(override val values: List<Duration>) : Stats<Duration> { |
| 24 | + override val sum by lazy { values.fold(Duration.ZERO) { a, it -> a + it } } |
| 25 | + override val average by lazy { sum / values.size } |
| 26 | + override val min by lazy { values.minOrNull() ?: Duration.ZERO } |
| 27 | + override val max by lazy { values.maxOrNull() ?: Duration.ZERO } |
| 28 | + override val median by lazy { |
| 29 | + if (values.isEmpty()) return@lazy Duration.ZERO |
| 30 | + if (values.size == 1) return@lazy values.first() |
| 31 | + val sorted = values.sorted() |
| 32 | + val middle = sorted.size / 2 |
| 33 | + if (sorted.size % 2 == 0) { |
| 34 | + (sorted[middle - 1] + sorted[middle]) / 2 |
| 35 | + } else { |
| 36 | + sorted[middle] |
| 37 | + } |
| 38 | + } |
| 39 | + |
| 40 | + override fun percentile(percentile: Double, direction: Stats.Percentile): Duration { |
| 41 | + if (values.isEmpty()) return Duration.ZERO |
| 42 | + val sorted = values.sorted() |
| 43 | + val index = when(direction) { |
| 44 | + Stats.Percentile.UP -> (percentile * sorted.size).toInt() |
| 45 | + Stats.Percentile.LOWEST -> sorted.size - (percentile * sorted.size).toInt() - 1 |
| 46 | + } |
| 47 | + return sorted[index] |
| 48 | + } |
| 49 | +} |
| 50 | + |
| 51 | +data class DoubleStats(override val values: List<Double>) : Stats<Double> { |
| 52 | + override val sum by lazy { values.fold(0.0) { a, it -> a + it } } |
| 53 | + override val average by lazy { sum / values.size } |
| 54 | + override val min by lazy { values.minOrNull() ?: 0.0 } |
| 55 | + override val max by lazy { values.maxOrNull() ?: 0.0 } |
| 56 | + override val median by lazy { |
| 57 | + if (values.isEmpty()) return@lazy 0.0 |
| 58 | + if (values.size == 1) return@lazy values.first() |
| 59 | + val sorted = values.sorted() |
| 60 | + val middle = sorted.size / 2 |
| 61 | + if (sorted.size % 2 == 0) { |
| 62 | + (sorted[middle - 1] + sorted[middle]) / 2 |
| 63 | + } else { |
| 64 | + sorted[middle] |
| 65 | + } |
| 66 | + } |
| 67 | + |
| 68 | + override fun percentile(percentile: Double, direction: Stats.Percentile): Double { |
| 69 | + if (values.isEmpty()) return 0.0 |
| 70 | + val sorted = values.sorted() |
| 71 | + val index = when(direction) { |
| 72 | + Stats.Percentile.UP -> (percentile * sorted.size).toInt() |
| 73 | + Stats.Percentile.LOWEST -> sorted.size - (percentile * sorted.size).toInt() - 1 |
| 74 | + } |
| 75 | + return sorted[index] |
| 76 | + } |
| 77 | +} |
| 78 | + |
| 79 | +data class RollingWindowStatistics( |
| 80 | + val windowSize: Duration = 5.seconds, |
| 81 | + val values: Map<Long, Duration> = emptyMap(), |
| 82 | +) { |
| 83 | + val frameTimes by lazy { DurationStats(values.values.toList()) } |
| 84 | + val fps by lazy { |
| 85 | + DoubleStats( |
| 86 | + if (values.size < 2) emptyList() |
| 87 | + else values.keys.sorted().zipWithNext().map { (a, b) -> 1_000_000_000.0 / (b - a) } |
| 88 | + ) |
| 89 | + } |
| 90 | + |
| 91 | + fun add(nanos: Long, time: Duration): RollingWindowStatistics { |
| 92 | + val newValues = values.toMutableMap() |
| 93 | + newValues[nanos] = time |
| 94 | + return copy(values = newValues.filter { it.key >= nanos - windowSize.inWholeNanoseconds }) |
| 95 | + } |
| 96 | +} |
| 97 | + |
| 98 | +class GLSurfaceState { |
| 99 | + internal val renderStatistics_ = MutableStateFlow(RollingWindowStatistics()) |
| 100 | + internal val displayStatistics_ = MutableStateFlow(RollingWindowStatistics()) |
| 101 | + |
| 102 | + val renderStatistics: StateFlow<RollingWindowStatistics> get() = renderStatistics_.asStateFlow() |
| 103 | + val displayStatistics: StateFlow<RollingWindowStatistics> get() = displayStatistics_.asStateFlow() |
| 104 | + |
| 105 | + internal fun onDisplay(nanos: Long, frameTime: Duration) { |
| 106 | + displayStatistics_.tryEmit(displayStatistics_.value.add(nanos, frameTime)) |
| 107 | + } |
| 108 | + |
| 109 | + internal fun onRender(nanos: Long, frameTime: Duration) { |
| 110 | + renderStatistics_.tryEmit(renderStatistics_.value.add(nanos, frameTime)) |
| 111 | + } |
| 112 | +} |
0 commit comments