Working with animation frame time in Jetpack Compose

AnimationClocks.kt

/**
 * Returns a [State] holding a local animation time in milliseconds.
 * The value always starts at `0L` and stops updating when
 * the call leaves the composition.
 */
@Composable
fun animationTimeMillis(): State<Long> {
    val millisState = state { 0L }
    val lifecycleOwner = LifecycleOwnerAmbient.current
    launchInComposition {
        val startTime = awaitFrameMillis { it }
        lifecycleOwner.whenStarted {
            while (true) {
                awaitFrameMillis { frameTime ->
                    millisState.value = frameTime - startTime
                }
            }
        }
    }
    return millisState
}