//
// Source
// ------------------------------------------

import androidx.compose.runtime.Composable
import androidx.compose.runtime.NonRestartableComposable
import androidx.compose.runtime.ReadOnlyComposable


@Composable fun TestMemoizedFun(compute: TestFunInterface) {}
@Composable fun Test() {
    val capture = 0
    TestMemoizedFun {
        // no captures
        use(it)
    }
    TestMemoizedFun {
        // stable captures
        use(capture)
    }
}

//
// Transformed IR
// ------------------------------------------

@Composable
@FunctionKeyMeta(key = 1461782433, startOffset = 226, endOffset = 275)
fun TestMemoizedFun(compute: TestFunInterface, %composer: Composer?, %changed: Int) {
  %composer = %composer.startRestartGroup(<>)
  if (%composer.shouldExecute(%changed and 0b0001 != 0, %changed and 0b0001)) {
    if (isTraceInProgress()) {
      traceEventStart(<>, %changed, -1, <>)
    }
    if (isTraceInProgress()) {
      traceEventEnd()
    }
  } else {
    %composer.skipToGroupEnd()
  }
  %composer.endRestartGroup()?.updateScope { %composer: Composer?, %force: Int ->
    TestMemoizedFun(compute, %composer, updateChangedFlags(%changed or 0b0001))
  }
}
@Composable
@FunctionKeyMeta(key = -1794342280, startOffset = 300, endOffset = 597)
fun Test(%composer: Composer?, %changed: Int) {
  %composer = %composer.startRestartGroup(<>)
  if (%composer.shouldExecute(%changed != 0, %changed and 0b0001)) {
    if (isTraceInProgress()) {
      traceEventStart(<>, %changed, -1, <>)
    }
    val capture = 0
    TestMemoizedFun(<block>{
      %composer.cache(false) {
        TestFunInterface { it: Int ->
          use(it)
        }
      }
    }, %composer, 0)
    TestMemoizedFun(<block>{
      %composer.cache(false) {
        TestFunInterface { it: Int ->
          use(capture)
        }
      }
    }, %composer, 0)
    if (isTraceInProgress()) {
      traceEventEnd()
    }
  } else {
    %composer.skipToGroupEnd()
  }
  %composer.endRestartGroup()?.updateScope { %composer: Composer?, %force: Int ->
    Test(%composer, updateChangedFlags(%changed or 0b0001))
  }
}
