exploring jetpack compose custom Layout

Stack.kt

sealed class StackOrientation {
    object Horizontal : StackOrientation()
    object Vertical : StackOrientation()
}

@Composable
fun Stack(
    modifier: Modifier = Modifier,
    orientation: StackOrientation = StackOrientation.Vertical,
    gap: Dp = 0.dp,
    content: @Composable () -> Unit
) {

    Layout(
        content = content,
        modifier = modifier,
    ) { measurables, constraints ->

        var width = 0
        var height = 0
        val isHorizontal = orientation == StackOrientation.Horizontal

        val parcelables = measurables.map { measurable ->
            val placeable = measurable.measure(constraints)
            width += placeable.width + gap.roundToPx()
            height += placeable.height + gap.roundToPx()
            placeable
        }

        layout(
            width = if (isHorizontal) width else constraints.minWidth,
            height = if (!isHorizontal) height else constraints.minHeight
        ) {
            var cursor = 0
            parcelables.forEach { placeable ->
                cursor += if (isHorizontal) {
                    placeable.place(cursor, 0)
                    placeable.width + gap.roundToPx()
                } else {
                    placeable.place(0, cursor)
                    placeable.height + gap.roundToPx()
                }
            }
        }
    }
}