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

import androidx.compose.runtime.*

@Composable fun Test(test: Boolean) {
    Column {
       if (!test) {
           Text("Say")
           return@Column
       }
       Text("Hello")
    }

    NonComposable {
        if (!test) {
           Text("Say")
           return@NonComposable
       }
       Text("Hello")
    }
}

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

@Composable
@FunctionKeyMeta(key = -378182080, startOffset = 72, endOffset = 529)
fun Test(test: Boolean, %composer: Composer?, %changed: Int) {
  %composer = %composer.startRestartGroup(<>)
  sourceInformation(%composer, "C(Test)N(test)<Column>,*<Text("...>:Test.kt")
  val %dirty = %changed
  if (%changed and 0b0110 == 0) {
    %dirty = %dirty or if (%composer.changed(test)) 0b0100 else 0b0010
  }
  if (%composer.shouldExecute(%dirty and 0b0011 != 0b0010, %dirty and 0b0001)) {
    if (isTraceInProgress()) {
      traceEventStart(<>, %dirty, -1, <>)
    }
    Column({ %composer: Composer?, %changed: Int ->
      %composer.startReplaceGroup(<>)
      sourceInformation(%composer, "C<Text("...>:Test.kt")
      if (!test) {
        %composer.startReplaceGroup(<>)
        sourceInformation(%composer, "<Text("...>")
        Text("Say", %composer, 0b0110)
        %composer.endReplaceGroup()
        %composer.endReplaceGroup()
        return@Column
      } else {
        %composer.startReplaceGroup(<>)
        %composer.endReplaceGroup()
      }
      Text("Hello", %composer, 0b0110)
      %composer.endReplaceGroup()
    }, %composer, 0)
    NonComposable {
      if (!test) {
        %composer.startReplaceGroup(<>)
        sourceInformation(%composer, "<Text("...>")
        Text("Say", %composer, 0b0110)
        %composer.endReplaceGroup()
        return@NonComposable
      } else {
        %composer.startReplaceGroup(<>)
        %composer.endReplaceGroup()
      }
      Text("Hello", %composer, 0b0110)
    }
    if (isTraceInProgress()) {
      traceEventEnd()
    }
  } else {
    %composer.skipToGroupEnd()
  }
  %composer.endRestartGroup()?.updateScope { %composer: Composer?, %force: Int ->
    Test(test, %composer, updateChangedFlags(%changed or 0b0001))
  }
}
