MetricCapture

@ExperimentalBenchmarkConfigApi
abstract class MetricCapture

import androidx.benchmark.MetricCapture

/**
 * Sample shows how to collect multiple continuously running counters as metrics which measure
 * difference.
 *
 * This is similar to how [androidx.benchmark.TimeCapture] produces `timeNs`.
 */
class MyMetricCapture : MetricCapture(listOf("foo", "bar")) {
    private var currentStartedFoo = 0L
    private var currentPausedStartedFoo = 0L
    private var currentTotalPausedFoo = 0L

    private var currentStartedBar = 0L
    private var currentPausedStartedBar = 0L
    private var currentTotalPausedBar = 0L

    override fun captureStart(timeNs: Long) {
        // reset paused state, capture current
        currentTotalPausedBar = 0
        currentTotalPausedFoo = 0

        currentStartedFoo = getCurrentFoo()
        currentStartedBar = getCurrentBar()
    }

    override fun captureStop(timeNs: Long, output: LongArray, offset: Int) {
        output[offset + 0] = getCurrentFoo() - currentStartedFoo - currentTotalPausedFoo
        output[offset + 1] = getCurrentBar() - currentStartedBar - currentTotalPausedBar
    }

    override fun capturePaused() {
        currentPausedStartedFoo = getCurrentFoo()
        currentPausedStartedBar = getCurrentBar()
    }

    override fun captureResumed() {
        currentTotalPausedFoo += getCurrentFoo() - currentPausedStartedFoo
        currentTotalPausedBar += getCurrentBar() - currentPausedStartedBar
    }
}