| /* |
| * Copyright 2019 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package androidx.ui.core.test |
| |
| import android.app.Activity |
| import android.graphics.Bitmap |
| import android.os.Build |
| import android.os.Handler |
| import android.view.PixelCopy |
| import android.view.View |
| import android.view.ViewGroup |
| import android.view.ViewTreeObserver |
| import android.widget.FrameLayout |
| import androidx.annotation.RequiresApi |
| import androidx.compose.Composable |
| import androidx.compose.Compose |
| import androidx.compose.Model |
| import androidx.compose.state |
| import androidx.compose.unaryPlus |
| import androidx.test.filters.SdkSuppress |
| import androidx.test.filters.SmallTest |
| import androidx.test.rule.ActivityTestRule |
| import androidx.ui.core.AlignmentLine |
| import androidx.ui.core.AndroidComposeView |
| import androidx.ui.core.Constraints |
| import androidx.ui.core.ContextAmbient |
| import androidx.ui.core.Density |
| import androidx.ui.core.DensityAmbient |
| import androidx.ui.core.DensityScope |
| import androidx.ui.core.Draw |
| import androidx.ui.core.HorizontalAlignmentLine |
| import androidx.ui.core.IntPx |
| import androidx.ui.core.IntPxPosition |
| import androidx.ui.core.IntPxSize |
| import androidx.ui.core.Layout |
| import androidx.ui.core.LayoutModifier |
| import androidx.ui.core.Measurable |
| import androidx.ui.core.Modifier |
| import androidx.ui.core.ParentData |
| import androidx.ui.core.Ref |
| import androidx.ui.core.RepaintBoundary |
| import androidx.ui.core.VerticalAlignmentLine |
| import androidx.ui.core.WithConstraints |
| import androidx.ui.core.coerceAtLeast |
| import androidx.ui.core.coerceIn |
| import androidx.ui.core.ipx |
| import androidx.ui.core.max |
| import androidx.ui.core.min |
| import androidx.ui.core.offset |
| import androidx.ui.core.px |
| import androidx.ui.core.setContent |
| import androidx.ui.core.toRect |
| import androidx.ui.engine.geometry.Rect |
| import androidx.ui.framework.test.TestActivity |
| import androidx.ui.graphics.Color |
| import androidx.ui.graphics.Paint |
| import org.junit.Assert.assertEquals |
| import org.junit.Assert.assertFalse |
| import org.junit.Assert.assertNull |
| import org.junit.Assert.assertSame |
| import org.junit.Assert.assertTrue |
| import org.junit.Assert.fail |
| import org.junit.Before |
| import org.junit.Rule |
| import org.junit.Test |
| import org.junit.runner.RunWith |
| import org.junit.runners.JUnit4 |
| import java.lang.Math.abs |
| import java.util.concurrent.CountDownLatch |
| import java.util.concurrent.TimeUnit |
| |
| /** |
| * Corresponds to ContainingViewTest, but tests single composition measure, layout and draw. |
| * It also tests that layouts with both Layout and MeasureBox work. |
| */ |
| @SmallTest |
| @RunWith(JUnit4::class) |
| class AndroidLayoutDrawTest { |
| @get:Rule |
| val activityTestRule = ActivityTestRule<TestActivity>( |
| TestActivity::class.java |
| ) |
| private lateinit var activity: TestActivity |
| private lateinit var drawLatch: CountDownLatch |
| |
| @Before |
| fun setup() { |
| activity = activityTestRule.activity |
| activity.hasFocusLatch.await(5, TimeUnit.SECONDS) |
| drawLatch = CountDownLatch(1) |
| } |
| |
| // Tests that simple drawing works with layered squares |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun simpleDrawTest() { |
| val yellow = Color(0xFFFFFF00) |
| val red = Color(0xFF800000) |
| val model = SquareModel(outerColor = yellow, innerColor = red, size = 10.ipx) |
| composeSquares(model) |
| |
| validateSquareColors(outerColor = yellow, innerColor = red, size = 10) |
| } |
| |
| // Tests that simple drawing works with draw with nested children |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun nestedDrawTest() { |
| val yellow = Color(0xFFFFFF00) |
| val red = Color(0xFF800000) |
| val model = SquareModel(outerColor = yellow, innerColor = red, size = 10.ipx) |
| composeNestedSquares(model) |
| |
| validateSquareColors(outerColor = yellow, innerColor = red, size = 10) |
| } |
| |
| // Tests that recomposition works with models used within Draw components |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun recomposeDrawTest() { |
| val white = Color(0xFFFFFFFF) |
| val blue = Color(0xFF000080) |
| val model = SquareModel(outerColor = blue, innerColor = white) |
| composeSquares(model) |
| validateSquareColors(outerColor = blue, innerColor = white, size = 10) |
| |
| drawLatch = CountDownLatch(1) |
| val red = Color(0xFF800000) |
| val yellow = Color(0xFFFFFF00) |
| activityTestRule.runOnUiThreadIR { |
| model.outerColor = red |
| model.innerColor = yellow |
| } |
| |
| validateSquareColors(outerColor = red, innerColor = yellow, size = 10) |
| } |
| |
| // Tests that recomposition of nested repaint boundaries work |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun recomposeNestedRepaintBoundariesColorChange() { |
| val white = Color(0xFFFFFFFF) |
| val blue = Color(0xFF000080) |
| val model = SquareModel(outerColor = blue, innerColor = white) |
| composeSquaresWithNestedRepaintBoundaries(model) |
| validateSquareColors(outerColor = blue, innerColor = white, size = 10) |
| |
| drawLatch = CountDownLatch(1) |
| val yellow = Color(0xFFFFFF00) |
| activityTestRule.runOnUiThreadIR { |
| model.innerColor = yellow |
| } |
| |
| validateSquareColors(outerColor = blue, innerColor = yellow, size = 10) |
| } |
| |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun recomposeNestedRepaintBoundariesSizeChange() { |
| val white = Color(0xFFFFFFFF) |
| val blue = Color(0xFF000080) |
| val model = SquareModel(outerColor = blue, innerColor = white) |
| composeSquaresWithNestedRepaintBoundaries(model) |
| validateSquareColors(outerColor = blue, innerColor = white, size = 10) |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.size = 20.ipx |
| } |
| |
| validateSquareColors(outerColor = blue, innerColor = white, size = 20) |
| } |
| |
| // When there is a repaint boundary around a moving child, the child move |
| // should be reflected in the repainted bitmap |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun recomposeRepaintBoundariesMove() { |
| val white = Color(0xFFFFFFFF) |
| val blue = Color(0xFF000080) |
| val model = SquareModel(outerColor = blue, innerColor = white) |
| var offset = OffsetModel(10.ipx) |
| composeMovingSquaresWithRepaintBoundary(model, offset) |
| validateSquareColors(outerColor = blue, innerColor = white, size = 10) |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| // there isn't going to be a normal draw because we are just moving the repaint |
| // boundary, but we should have a draw cycle |
| activityTestRule.findAndroidComposeView().viewTreeObserver.addOnDrawListener(object : |
| ViewTreeObserver.OnDrawListener { |
| override fun onDraw() { |
| drawLatch.countDown() |
| } |
| }) |
| offset.offset = 20.ipx |
| } |
| |
| validateSquareColors(outerColor = blue, innerColor = white, offset = 10, size = 10) |
| } |
| |
| // When there is no repaint boundary around a moving child, the child move |
| // should be reflected in the repainted bitmap |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun recomposeMove() { |
| val white = Color(0xFFFFFFFF) |
| val blue = Color(0xFF000080) |
| val model = SquareModel(outerColor = blue, innerColor = white) |
| var offset = OffsetModel(10.ipx) |
| composeMovingSquares(model, offset) |
| validateSquareColors(outerColor = blue, innerColor = white, size = 10) |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| // there isn't going to be a normal draw because we are just moving the repaint |
| // boundary, but we should have a draw cycle |
| activityTestRule.findAndroidComposeView().viewTreeObserver.addOnDrawListener(object : |
| ViewTreeObserver.OnDrawListener { |
| override fun onDraw() { |
| drawLatch.countDown() |
| } |
| }) |
| offset.offset = 20.ipx |
| } |
| |
| validateSquareColors(outerColor = blue, innerColor = white, offset = 10, size = 10) |
| } |
| |
| // Tests that recomposition works with models used within Layout components |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun recomposeSizeTest() { |
| val white = Color(0xFFFFFFFF) |
| val blue = Color(0xFF000080) |
| val model = SquareModel(outerColor = blue, innerColor = white) |
| composeSquares(model) |
| validateSquareColors(outerColor = blue, innerColor = white, size = 10) |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.size = 20.ipx } |
| validateSquareColors(outerColor = blue, innerColor = white, size = 20) |
| } |
| |
| // The size and color are both changed in a simpler single-color square. |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun simpleSquareColorAndSizeTest() { |
| val green = Color(0xFF00FF00) |
| val model = SquareModel(size = 20.ipx, outerColor = green, innerColor = green) |
| |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Padding(size = (model.size * 3)) { |
| FillColor(model, isInner = false) |
| } |
| } |
| } |
| validateSquareColors(outerColor = green, innerColor = green, size = 20) |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.size = 30.ipx |
| } |
| validateSquareColors(outerColor = green, innerColor = green, size = 30) |
| |
| drawLatch = CountDownLatch(1) |
| val blue = Color(0xFF0000FF) |
| |
| activityTestRule.runOnUiThreadIR { |
| model.innerColor = blue |
| model.outerColor = blue |
| } |
| validateSquareColors(outerColor = blue, innerColor = blue, size = 30) |
| } |
| |
| // Components that aren't placed shouldn't be drawn. |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun noPlaceNoDraw() { |
| val green = Color(0xFF00FF00) |
| val white = Color(0xFFFFFFFF) |
| val model = SquareModel(size = 20.ipx, outerColor = green, innerColor = white) |
| |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Layout(children = { |
| Padding(size = (model.size * 3)) { |
| FillColor(model, isInner = false) |
| } |
| Padding(size = model.size) { |
| FillColor(model, isInner = true) |
| } |
| }, measureBlock = { measurables, constraints -> |
| val placeables = measurables.map { it.measure(constraints) } |
| layout(placeables[0].width, placeables[0].height) { |
| placeables[0].place(0.ipx, 0.ipx) |
| } |
| }) |
| } |
| } |
| validateSquareColors(outerColor = green, innerColor = green, size = 20) |
| } |
| |
| // Make sure that draws intersperse properly with sub-layouts |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun drawOrderWithChildren() { |
| val green = Color(0xFF00FF00) |
| val white = Color(0xFFFFFFFF) |
| val model = SquareModel(size = 20.ipx, outerColor = green, innerColor = white) |
| |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Draw { canvas, parentSize -> |
| // Fill the space with the outerColor |
| val paint = Paint() |
| paint.color = model.outerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| canvas.nativeCanvas.save() |
| val offset = parentSize.width.value / 3 |
| // clip drawing to the inner rectangle |
| canvas.clipRect(Rect(offset, offset, offset * 2, offset * 2)) |
| } |
| Padding(size = (model.size * 3)) { |
| Draw { canvas, parentSize -> |
| // Fill top half with innerColor -- should be clipped |
| drawLatch.countDown() |
| val paint = Paint() |
| paint.color = model.innerColor |
| val paintRect = Rect( |
| 0f, 0f, parentSize.width.value, |
| parentSize.height.value / 2f |
| ) |
| canvas.drawRect(paintRect, paint) |
| } |
| } |
| Draw { canvas, parentSize -> |
| // Fill bottom half with innerColor -- should be clipped |
| val paint = Paint() |
| paint.color = model.innerColor |
| val paintRect = Rect( |
| 0f, parentSize.height.value / 2f, |
| parentSize.width.value, parentSize.height.value |
| ) |
| canvas.drawRect(paintRect, paint) |
| // restore the canvas |
| canvas.nativeCanvas.restore() |
| } |
| } |
| } |
| validateSquareColors(outerColor = green, innerColor = white, size = 20) |
| } |
| |
| @Test |
| fun withConstraintsTest() { |
| val size = 20.ipx |
| |
| val countDownLatch = CountDownLatch(1) |
| val topConstraints = Ref<Constraints>() |
| val paddedConstraints = Ref<Constraints>() |
| val firstChildConstraints = Ref<Constraints>() |
| val secondChildConstraints = Ref<Constraints>() |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| WithConstraints { constraints -> |
| topConstraints.value = constraints |
| Padding(size = size) { |
| WithConstraints { constraints -> |
| paddedConstraints.value = constraints |
| Layout(measureBlock = { _, childConstraints -> |
| firstChildConstraints.value = childConstraints |
| layout(size, size) { } |
| }, children = { }) |
| Layout(measureBlock = { _, chilConstraints -> |
| secondChildConstraints.value = chilConstraints |
| layout(size, size) { } |
| }, children = { }) |
| Draw { _, _ -> |
| countDownLatch.countDown() |
| } |
| } |
| } |
| } |
| } |
| } |
| assertTrue(countDownLatch.await(1, TimeUnit.SECONDS)) |
| |
| val expectedPaddedConstraints = Constraints( |
| 0.ipx, |
| topConstraints.value!!.maxWidth - size * 2, |
| 0.ipx, |
| topConstraints.value!!.maxHeight - size * 2 |
| ) |
| assertEquals(expectedPaddedConstraints, paddedConstraints.value) |
| assertEquals(paddedConstraints.value, firstChildConstraints.value) |
| assertEquals(paddedConstraints.value, secondChildConstraints.value) |
| } |
| |
| @Suppress("UNUSED_ANONYMOUS_PARAMETER") |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun withConstraints_layoutListener() { |
| val green = Color.Green |
| val white = Color.White |
| val model = SquareModel(size = 20.ipx, outerColor = green, innerColor = white) |
| |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| WithConstraints { constraints -> |
| Layout(children = { |
| Draw { canvas, parentSize -> |
| val paint = Paint() |
| paint.color = model.outerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| Layout(children = { |
| Draw { canvas, parentSize -> |
| drawLatch.countDown() |
| val paint = Paint() |
| paint.color = model.innerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| }) { measurables, constraints2 -> |
| layout(model.size, model.size) {} |
| } |
| }) { measurables, constraints3 -> |
| val placeable = measurables[0].measure( |
| Constraints.tightConstraints( |
| model.size, |
| model.size |
| ) |
| ) |
| layout(model.size * 3, model.size * 3) { |
| placeable.place(model.size, model.size) |
| } |
| } |
| } |
| } |
| } |
| validateSquareColors(outerColor = green, innerColor = white, size = 20) |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.size = 10.ipx |
| } |
| |
| validateSquareColors(outerColor = green, innerColor = white, size = 10) |
| } |
| |
| // Tests that calling measure multiple times on the same Measurable causes an exception |
| @Test |
| fun multipleMeasureCall() { |
| val latch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| TwoMeasureLayout(50.ipx, latch) { |
| AtLeastSize(50.ipx) { |
| } |
| } |
| } |
| } |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun multiChildLayoutTest() { |
| val childrenCount = 3 |
| val childConstraints = arrayOf( |
| Constraints(), |
| Constraints.tightConstraintsForWidth(50.ipx), |
| Constraints.tightConstraintsForHeight(50.ipx) |
| ) |
| val headerChildrenCount = 1 |
| val footerChildrenCount = 2 |
| |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val header = @Composable { |
| Layout(measureBlock = { _, constraints -> |
| assertEquals(childConstraints[0], constraints) |
| layout(0.ipx, 0.ipx) {} |
| }, children = {}) |
| } |
| val footer = @Composable { |
| Layout(measureBlock = { _, constraints -> |
| assertEquals(childConstraints[1], constraints) |
| layout(0.ipx, 0.ipx) {} |
| }, children = {}) |
| Layout(measureBlock = { _, constraints -> |
| assertEquals(childConstraints[2], constraints) |
| layout(0.ipx, 0.ipx) {} |
| }, children = {}) |
| } |
| |
| Layout(header, footer) { measurables, _ -> |
| assertEquals(childrenCount, measurables.size) |
| measurables.forEachIndexed { index, measurable -> |
| measurable.measure(childConstraints[index]) |
| } |
| assertEquals(headerChildrenCount, measurables[header].size) |
| assertSame(measurables[0], measurables[header][0]) |
| assertEquals(footerChildrenCount, measurables[footer].size) |
| assertSame(measurables[1], measurables[footer][0]) |
| assertSame(measurables[2], measurables[footer][1]) |
| layout(0.ipx, 0.ipx) {} |
| } |
| } |
| } |
| } |
| |
| @Test |
| fun multiChildLayoutTest_doesNotOverrideChildrenParentData() { |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val header = @Composable { |
| ParentData(data = 0) { |
| Layout(children = {}) { _, _ -> layout(0.ipx, 0.ipx) {} } |
| } |
| } |
| val footer = @Composable { |
| ParentData(data = 1) { |
| Layout(children = {}) { _, _ -> layout(0.ipx, 0.ipx) {} } |
| } |
| } |
| |
| Layout(header, footer) { measurables, _ -> |
| assertEquals(0, measurables[0].parentData) |
| assertEquals(1, measurables[1].parentData) |
| layout(0.ipx, 0.ipx) { } |
| } |
| } |
| } |
| } |
| |
| // TODO(lmr): refactor to use the globally provided one when it lands |
| private fun Activity.compose(composable: @Composable() () -> Unit) { |
| val root = AndroidComposeView(this) |
| |
| setContentView(root) |
| Compose.composeInto(root.root, context = this) { |
| ContextAmbient.Provider(value = this) { |
| DensityAmbient.Provider(value = Density(this), children = composable) |
| } |
| } |
| } |
| |
| // When a child's measure() is done within the layout, it should not affect the parent's |
| // size. The parent's layout shouldn't be called when the child's size changes |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun measureInLayoutDoesNotAffectParentSize() { |
| val white = Color(0xFFFFFFFF) |
| val blue = Color(0xFF000080) |
| val model = SquareModel(outerColor = blue, innerColor = white) |
| var measureCalls = 0 |
| var layoutCalls = 0 |
| |
| val layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.compose { |
| Draw { canvas, parentSize -> |
| val paint = Paint() |
| paint.color = model.outerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| Layout(children = { |
| AtLeastSize(size = model.size) { |
| Draw { canvas, parentSize -> |
| drawLatch.countDown() |
| val paint = Paint() |
| paint.color = model.innerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| } |
| }, measureBlock = { measurables, constraints -> |
| measureCalls++ |
| layout(30.ipx, 30.ipx) { |
| layoutCalls++ |
| layoutLatch.countDown() |
| val placeable = measurables[0].measure(constraints) |
| placeable.place( |
| (30.ipx - placeable.width) / 2, |
| (30.ipx - placeable.height) / 2 |
| ) |
| } |
| }) |
| } |
| } |
| layoutLatch.await(1, TimeUnit.SECONDS) |
| |
| validateSquareColors(outerColor = blue, innerColor = white, size = 10) |
| |
| layoutCalls = 0 |
| measureCalls = 0 |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.size = 20.ipx |
| } |
| |
| validateSquareColors(outerColor = blue, innerColor = white, size = 20, totalSize = 30) |
| assertEquals(0, measureCalls) |
| assertEquals(1, layoutCalls) |
| } |
| |
| @Test |
| fun testLayout_whenMeasuringIsDoneDuringPlacing() { |
| @Composable |
| fun FixedSizeRow( |
| width: IntPx, |
| height: IntPx, |
| children: @Composable() () -> Unit |
| ) { |
| Layout(children = children, measureBlock = { measurables, constraints -> |
| val resolvedWidth = width.coerceIn(constraints.minWidth, constraints.maxWidth) |
| val resolvedHeight = height.coerceIn(constraints.minHeight, constraints.maxHeight) |
| layout(resolvedWidth, resolvedHeight) { |
| val childConstraints = Constraints( |
| IntPx.Zero, |
| IntPx.Infinity, |
| resolvedHeight, |
| resolvedHeight |
| ) |
| var left = IntPx.Zero |
| for (measurable in measurables) { |
| val placeable = measurable.measure(childConstraints) |
| if (left + placeable.width > width) { |
| break |
| } |
| placeable.place(left, IntPx.Zero) |
| left += placeable.width |
| } |
| } |
| }) |
| } |
| |
| @Composable |
| fun FixedWidthBox( |
| width: IntPx, |
| measured: Ref<Boolean?>, |
| laidOut: Ref<Boolean?>, |
| drawn: Ref<Boolean?>, |
| latch: CountDownLatch |
| ) { |
| Layout(children = { |
| Draw { _, _ -> |
| drawn.value = true |
| latch.countDown() |
| } |
| }, measureBlock = { _, constraints -> |
| measured.value = true |
| val resolvedWidth = width.coerceIn(constraints.minWidth, constraints.maxWidth) |
| val resolvedHeight = constraints.minHeight |
| layout(resolvedWidth, resolvedHeight) { laidOut.value = true } |
| }) |
| } |
| |
| val childrenCount = 5 |
| val measured = Array(childrenCount) { Ref<Boolean?>() } |
| val laidOut = Array(childrenCount) { Ref<Boolean?>() } |
| val drawn = Array(childrenCount) { Ref<Boolean?>() } |
| val latch = CountDownLatch(3) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Align { |
| FixedSizeRow(width = 90.ipx, height = 40.ipx) { |
| for (i in 0 until childrenCount) { |
| FixedWidthBox( |
| width = 30.ipx, |
| measured = measured[i], |
| laidOut = laidOut[i], |
| drawn = drawn[i], |
| latch = latch |
| ) |
| } |
| } |
| } |
| } |
| } |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| |
| for (i in 0 until childrenCount) { |
| assertEquals(i <= 3, measured[i].value ?: false) |
| assertEquals(i <= 2, laidOut[i].value ?: false) |
| assertEquals(i <= 2, drawn[i].value ?: false) |
| } |
| } |
| |
| // When a new child is added, the parent must be remeasured because we don't know |
| // if it affects the size and the child's measure() must be called as well. |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun testRelayoutOnNewChild() { |
| val drawChild = DoDraw() |
| |
| val outerColor = Color(0xFF000080) |
| val innerColor = Color(0xFFFFFFFF) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| AtLeastSize(size = 30.ipx) { |
| FillColor(outerColor) |
| if (drawChild.value) { |
| Padding(size = 20.ipx) { |
| AtLeastSize(size = 20.ipx) { |
| FillColor(innerColor) |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| // The padded area doesn't draw |
| validateSquareColors(outerColor = outerColor, innerColor = outerColor, size = 10) |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { drawChild.value = true } |
| |
| validateSquareColors(outerColor = outerColor, innerColor = innerColor, size = 20) |
| } |
| |
| // When we change a position of one LayoutNode up the tree it automatically |
| // changes the position of all the children. RepaintBoundary with few intermediate |
| // LayoutNode parents should be drawn on a correct position |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun moveRootLayoutRedrawsLeafRepaintBoundary() { |
| val offset = OffsetModel(0.ipx) |
| drawLatch = CountDownLatch(2) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| FillColor(Color.Green) |
| Layout( |
| children = { |
| AtLeastSize(size = 10.ipx) { |
| AtLeastSize(size = 10.ipx) { |
| RepaintBoundary { |
| FillColor(Color.Cyan) |
| } |
| } |
| } |
| } |
| ) { measurables, constraints -> |
| layout(width = 20.ipx, height = 20.ipx) { |
| measurables.first().measure(constraints) |
| .place(offset.offset, offset.offset) |
| } |
| } |
| } |
| } |
| |
| assertTrue(drawLatch.await(1, TimeUnit.SECONDS)) |
| activityTestRule.waitAndScreenShot().apply { |
| assertRect(Color.Cyan, size = 10, centerX = 5, centerY = 5) |
| assertRect(Color.Green, size = 10, centerX = 15, centerY = 15) |
| } |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { offset.offset = 10.ipx } |
| |
| assertTrue(drawLatch.await(1, TimeUnit.SECONDS)) |
| activityTestRule.waitAndScreenShot().apply { |
| assertRect(Color.Green, size = 10, centerX = 5, centerY = 5) |
| assertRect(Color.Cyan, size = 10, centerX = 15, centerY = 15) |
| } |
| } |
| |
| // When a child is removed, the parent must be remeasured and redrawn. |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun testRedrawOnRemovedChild() { |
| val drawChild = DoDraw(true) |
| |
| val outerColor = Color(0xFF000080) |
| val innerColor = Color(0xFFFFFFFF) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| AtLeastSize(size = 30.ipx) { |
| Draw { canvas, parentSize -> |
| drawLatch.countDown() |
| val paint = Paint() |
| paint.color = outerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| AtLeastSize(size = 30.ipx) { |
| if (drawChild.value) { |
| Padding(size = 10.ipx) { |
| AtLeastSize(size = 10.ipx) { |
| Draw { canvas, parentSize -> |
| drawLatch.countDown() |
| val paint = Paint() |
| paint.color = innerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| } |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| validateSquareColors(outerColor = outerColor, innerColor = innerColor, size = 10) |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { drawChild.value = false } |
| |
| // The padded area doesn't draw |
| validateSquareColors(outerColor = outerColor, innerColor = outerColor, size = 10) |
| } |
| |
| // When a child is removed, the parent must be remeasured. |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun testRelayoutOnRemovedChild() { |
| val drawChild = DoDraw(true) |
| |
| val outerColor = Color(0xFF000080) |
| val innerColor = Color(0xFFFFFFFF) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| AtLeastSize(size = 30.ipx) { |
| Draw { canvas, parentSize -> |
| drawLatch.countDown() |
| val paint = Paint() |
| paint.color = outerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| Padding(size = 20.ipx) { |
| if (drawChild.value) { |
| AtLeastSize(size = 20.ipx) { |
| Draw { canvas, parentSize -> |
| drawLatch.countDown() |
| val paint = Paint() |
| paint.color = innerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| validateSquareColors(outerColor = outerColor, innerColor = innerColor, size = 20) |
| |
| drawLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { drawChild.value = false } |
| |
| // The padded area doesn't draw |
| validateSquareColors(outerColor = outerColor, innerColor = outerColor, size = 10) |
| } |
| |
| @Test |
| fun testAlignmentLines() { |
| val TestVerticalLine = VerticalAlignmentLine(::min) |
| val TestHorizontalLine = HorizontalAlignmentLine(::max) |
| val layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child1 = @Composable { |
| Wrap { |
| Layout({ }) { _, _ -> |
| layout( |
| 0.ipx, |
| 0.ipx, |
| mapOf( |
| TestVerticalLine to 10.ipx, |
| TestHorizontalLine to 20.ipx |
| ) |
| ) { } |
| } |
| } |
| } |
| val child2 = @Composable { |
| Wrap { |
| Layout({ }) { _, _ -> |
| layout( |
| 0.ipx, |
| 0.ipx, |
| mapOf( |
| TestVerticalLine to 20.ipx, |
| TestHorizontalLine to 10.ipx |
| ) |
| ) { } |
| } |
| } |
| } |
| val inner = @Composable { |
| Layout({ child1(); child2() }) { measurables, constraints -> |
| val placeable1 = measurables[0].measure(constraints) |
| val placeable2 = measurables[1].measure(constraints) |
| assertEquals(10.ipx, placeable1[TestVerticalLine]) |
| assertEquals(20.ipx, placeable1[TestHorizontalLine]) |
| assertEquals(20.ipx, placeable2[TestVerticalLine]) |
| assertEquals(10.ipx, placeable2[TestHorizontalLine]) |
| layout(0.ipx, 0.ipx) { |
| placeable1.place(0.ipx, 0.ipx) |
| placeable2.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| assertEquals(10.ipx, placeable[TestVerticalLine]) |
| assertEquals(20.ipx, placeable[TestHorizontalLine]) |
| layout(placeable.width, placeable.height) { |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun testAlignmentLines_areNotInheritedFromInvisibleChildren() { |
| val TestLine1 = VerticalAlignmentLine(::min) |
| val TestLine2 = VerticalAlignmentLine(::min) |
| val layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child1 = @Composable { |
| Layout(children = { }) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine1 to 10.ipx)) {} |
| } |
| } |
| val child2 = @Composable { |
| Layout({ }) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine2 to 20.ipx)) { } |
| } |
| } |
| val inner = @Composable { |
| Layout({ child1(); child2() }) { measurables, constraints -> |
| val placeable1 = measurables[0].measure(constraints) |
| measurables[1].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| // Only place the first child. |
| placeable1.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| assertEquals(10.ipx, placeable[TestLine1]) |
| assertNull(placeable[TestLine2]) |
| layout(placeable.width, placeable.height) { |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun testAlignmentLines_doNotCauseMultipleMeasuresOrLayouts() { |
| val TestLine1 = VerticalAlignmentLine(::min) |
| val TestLine2 = VerticalAlignmentLine(::min) |
| var child1Measures = 0 |
| var child2Measures = 0 |
| var child1Layouts = 0 |
| var child2Layouts = 0 |
| val layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child1 = @Composable { |
| Layout({ }) { _, _ -> |
| ++child1Measures |
| layout(0.ipx, 0.ipx, mapOf(TestLine1 to 10.ipx)) { |
| ++child1Layouts |
| } |
| } |
| } |
| val child2 = @Composable { |
| Layout({ }) { _, _ -> |
| ++child2Measures |
| layout(0.ipx, 0.ipx, mapOf(TestLine2 to 20.ipx)) { |
| ++child2Layouts |
| } |
| } |
| } |
| val inner = @Composable { |
| Layout({ child1(); child2() }) { measurables, constraints -> |
| val placeable1 = measurables[0].measure(constraints) |
| val placeable2 = measurables[1].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| placeable1.place(0.ipx, 0.ipx) |
| placeable2.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| assertEquals(10.ipx, placeable[TestLine1]) |
| assertEquals(20.ipx, placeable[TestLine2]) |
| layout(placeable.width, placeable.height) { |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, child1Measures) |
| assertEquals(1, child2Measures) |
| assertEquals(1, child1Layouts) |
| assertEquals(1, child2Layouts) |
| } |
| |
| @Test |
| fun testAlignmentLines_onlyLayoutEarlyWhenNeeded() { |
| val TestLine1 = VerticalAlignmentLine(::min) |
| val TestLine2 = VerticalAlignmentLine(::min) |
| var child1Measures = 0 |
| var child2Measures = 0 |
| var child1Layouts = 0 |
| var child2Layouts = 0 |
| val layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child1 = @Composable { |
| Layout({ }) { _, _ -> |
| ++child1Measures |
| layout(0.ipx, 0.ipx, mapOf(TestLine1 to 10.ipx)) { |
| ++child1Layouts |
| } |
| } |
| } |
| val child2 = @Composable { |
| Layout({ }) { _, _ -> |
| ++child2Measures |
| layout(0.ipx, 0.ipx, mapOf(TestLine2 to 20.ipx)) { |
| ++child2Layouts |
| } |
| } |
| } |
| val inner = @Composable { |
| Layout({ child1(); child2() }) { measurables, constraints -> |
| val placeable1 = measurables[0].measure(constraints) |
| assertEquals(10.ipx, placeable1[TestLine1]) |
| val placeable2 = measurables[1].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| placeable1.place(0.ipx, 0.ipx) |
| placeable2.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| layout(placeable.width, placeable.height) { |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, child1Measures) |
| assertEquals(1, child2Measures) |
| assertEquals(1, child1Layouts) |
| assertEquals(0, child2Layouts) |
| } |
| |
| @Test |
| fun testAlignmentLines_canBeQueriedInThePositioningBlock() { |
| val TestLine = VerticalAlignmentLine(::min) |
| val layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child1 = @Composable { |
| Layout(children = { }) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine to 10.ipx)) { } |
| } |
| } |
| val child2 = @Composable { |
| Layout({ }) { _, _ -> layout(0.ipx, 0.ipx, mapOf(TestLine to 20.ipx)) { } } |
| } |
| val inner = @Composable { |
| Layout({ child1(); child2() }) { measurables, constraints -> |
| val placeable1 = measurables[0].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| assertEquals(10.ipx, placeable1[TestLine]) |
| val placeable2 = measurables[1].measure(constraints) |
| assertEquals(20.ipx, placeable2[TestLine]) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| layout(placeable.width, placeable.height) { |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun testAlignmentLines_doNotCauseExtraLayout_whenQueriedAfterPositioning() { |
| val TestLine = VerticalAlignmentLine(::min) |
| val layoutLatch = CountDownLatch(1) |
| var childLayouts = 0 |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child = @Composable { |
| Layout(children = { }) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine to 10.ipx)) { |
| ++childLayouts |
| } |
| } |
| } |
| val inner = @Composable { |
| Layout({ child() }) { measurables, constraints -> |
| val placeable = measurables[0].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| assertEquals(10.ipx, placeable[TestLine]) |
| placeable.place(0.ipx, 0.ipx) |
| assertEquals(10.ipx, placeable[TestLine]) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| layout(placeable.width, placeable.height) { |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, childLayouts) |
| } |
| |
| @Test |
| fun testAlignmentLines_recomposeCorrectly() { |
| val TestLine = VerticalAlignmentLine(::min) |
| var layoutLatch = CountDownLatch(1) |
| val model = OffsetModel(10.ipx) |
| var measure = 0 |
| var layout = 0 |
| var linePosition: IntPx? = null |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child = @Composable { |
| Layout({ }) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine to model.offset)) {} } |
| } |
| Layout(child) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| linePosition = placeable[TestLine] |
| ++measure |
| layout(placeable.width, placeable.height) { |
| ++layout |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, measure) |
| assertEquals(1, layout) |
| assertEquals(10.ipx, linePosition) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.offset = 20.ipx |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(2, measure) |
| assertEquals(2, layout) |
| assertEquals(20.ipx, linePosition) |
| } |
| |
| @Test |
| fun testAlignmentLines_recomposeCorrectly_whenQueriedInLayout() { |
| val TestLine = VerticalAlignmentLine(::min) |
| var layoutLatch = CountDownLatch(1) |
| val model = OffsetModel(10.ipx) |
| var measure = 0 |
| var layout = 0 |
| var linePosition: IntPx? = null |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child = @Composable { |
| Layout({ }) { _, _ -> layout(0.ipx, 0.ipx, mapOf(TestLine to model.offset)) {} } |
| } |
| Layout(child) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| ++measure |
| layout(placeable.width, placeable.height) { |
| linePosition = placeable[TestLine] |
| ++layout |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, measure) |
| assertEquals(1, layout) |
| assertEquals(10.ipx, linePosition) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 20.ipx } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(2, measure) |
| assertEquals(2, layout) |
| assertEquals(20.ipx, linePosition) |
| } |
| |
| @Test |
| fun testAlignmentLines_recomposeCorrectly_whenMeasuredAndQueriedInLayout() { |
| val TestLine = VerticalAlignmentLine(::min) |
| var layoutLatch = CountDownLatch(1) |
| val model = OffsetModel(10.ipx) |
| var measure = 0 |
| var layout = 0 |
| var linePosition: IntPx? = null |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child = @Composable { |
| Layout({ }) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine to model.offset)) { } |
| } |
| } |
| Layout(child) { measurables, constraints -> |
| ++measure |
| layout(1.ipx, 1.ipx) { |
| val placeable = measurables.first().measure(constraints) |
| linePosition = placeable[TestLine] |
| ++layout |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, measure) |
| assertEquals(1, layout) |
| assertEquals(10.ipx, linePosition) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 20.ipx } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, measure) |
| assertEquals(2, layout) |
| assertEquals(20.ipx, linePosition) |
| } |
| |
| @Test |
| fun testAlignmentLines_onlyComputesAlignmentLinesWhenNeeded() { |
| var layoutLatch = CountDownLatch(1) |
| val model = OffsetModel(10.ipx) |
| var alignmentLinesCalculations = 0 |
| val TestLine = VerticalAlignmentLine { _, _ -> |
| ++alignmentLinesCalculations |
| 0.ipx |
| } |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val innerChild = @Composable { |
| model.offset // Artificial remeasure. |
| Layout({ }) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine to 10.ipx)) { } |
| } |
| } |
| val child = @Composable { |
| Layout({ innerChild(); innerChild() }) { measurables, constraints -> |
| model.offset // Artificial remeasure. |
| val placeable1 = measurables[0].measure(constraints) |
| val placeable2 = measurables[1].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| placeable1.place(0.ipx, 0.ipx) |
| placeable2.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| Layout(child) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| if (model.offset < 15.ipx) { |
| placeable[TestLine] |
| } |
| layout(0.ipx, 0.ipx) { |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, alignmentLinesCalculations) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 20.ipx } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, alignmentLinesCalculations) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 10.ipx } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(2, alignmentLinesCalculations) |
| } |
| |
| @Test |
| fun testAlignmentLines_providedLinesOverrideInherited() { |
| val layoutLatch = CountDownLatch(1) |
| val TestLine = VerticalAlignmentLine(::min) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val innerChild = @Composable { |
| Layout({}) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine to 10.ipx)) { } |
| } |
| } |
| val child = @Composable { |
| Layout({ innerChild() }) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| layout(0.ipx, 0.ipx, mapOf(TestLine to 20.ipx)) { |
| placeable.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| Layout(child) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| assertEquals(20.ipx, placeable[TestLine]) |
| layout(0.ipx, 0.ipx) { |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun testAlignmentLines_areRecalculatedCorrectlyOnRelayout_withNoRemeasure() { |
| val TestLine = VerticalAlignmentLine(::min) |
| var layoutLatch = CountDownLatch(1) |
| var innerChildMeasures = 0 |
| var innerChildLayouts = 0 |
| var outerChildMeasures = 0 |
| var outerChildLayouts = 0 |
| val model = OffsetModel(0.ipx) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child = @Composable { |
| Layout({}) { _, _ -> |
| ++innerChildMeasures |
| layout(0.ipx, 0.ipx, mapOf(TestLine to 10.ipx)) { ++innerChildLayouts } |
| } |
| } |
| val inner = @Composable { |
| Layout({ Wrap { Wrap { child() } } }) { measurables, constraints -> |
| ++outerChildMeasures |
| val placeable = measurables[0].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| ++outerChildLayouts |
| placeable.place(model.offset, 0.ipx) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| val width = placeable.width.coerceAtLeast(10.ipx) |
| val height = placeable.height.coerceAtLeast(10.ipx) |
| layout(width, height) { |
| assertEquals(model.offset + 10.ipx, placeable[TestLine]) |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, innerChildMeasures) |
| assertEquals(1, innerChildLayouts) |
| assertEquals(1, outerChildMeasures) |
| assertEquals(1, outerChildLayouts) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.offset = 10.ipx |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(1, innerChildMeasures) |
| assertEquals(1, innerChildLayouts) |
| assertEquals(1, outerChildMeasures) |
| assertEquals(2, outerChildLayouts) |
| } |
| |
| @Test |
| fun testAlignmentLines_whenQueriedAfterPlacing() { |
| val TestLine = VerticalAlignmentLine(::min) |
| val layoutLatch = CountDownLatch(1) |
| var childLayouts = 0 |
| activityTestRule.runOnUiThreadIR { |
| activity.setContent { |
| val child = @Composable { |
| Layout({}) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine to 10.ipx)) { ++childLayouts } |
| } |
| } |
| val inner = @Composable { |
| Layout({ Wrap { Wrap { child() } } }) { measurables, constraints -> |
| val placeable = measurables[0].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| placeable.place(0.ipx, 0.ipx) |
| assertEquals(10.ipx, placeable[TestLine]) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| layout(placeable.width, placeable.height) { |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| // Two layouts as the alignment line was only queried after the child was placed. |
| assertEquals(2, childLayouts) |
| } |
| |
| @Test |
| fun testAlignmentLines_whenQueriedAfterPlacing_haveCorrectNumberOfLayouts() { |
| val TestLine = VerticalAlignmentLine(::min) |
| var layoutLatch = CountDownLatch(1) |
| var childLayouts = 0 |
| val model = OffsetModel(10.ipx) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContent { |
| val child = @Composable { |
| Layout({}) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(TestLine to 10.ipx)) { |
| model.offset // To ensure relayout. |
| ++childLayouts |
| } |
| } |
| } |
| val inner = @Composable { |
| Layout({ WrapForceRelayout(model) { child() } }) { measurables, constraints -> |
| val placeable = measurables[0].measure(constraints) |
| layout(0.ipx, 0.ipx) { |
| if (model.offset > 15.ipx) assertEquals(10.ipx, placeable[TestLine]) |
| placeable.place(0.ipx, 0.ipx) |
| if (model.offset > 5.ipx) assertEquals(10.ipx, placeable[TestLine]) |
| } |
| } |
| } |
| Layout(inner) { measurables, constraints -> |
| val placeable = measurables.first().measure(constraints) |
| val width = placeable.width.coerceAtLeast(10.ipx) |
| val height = placeable.height.coerceAtLeast(10.ipx) |
| layout(width, height) { |
| model.offset // To ensure relayout. |
| placeable.place(0.ipx, 0.ipx) |
| layoutLatch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| // Two layouts as the alignment line was only queried after the child was placed. |
| assertEquals(2, childLayouts) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 12.ipx } |
| assertTrue(layoutLatch.await(5, TimeUnit.SECONDS)) |
| // Just one more layout as the alignment lines were speculatively calculated this time. |
| assertEquals(3, childLayouts) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 17.ipx } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| // One layout as the alignment lines are queried before. |
| assertEquals(4, childLayouts) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 12.ipx } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| // One layout as the alignment lines are still calculated speculatively. |
| assertEquals(5, childLayouts) |
| |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 1.ipx } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| assertEquals(6, childLayouts) |
| layoutLatch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { model.offset = 10.ipx } |
| assertTrue(layoutLatch.await(1, TimeUnit.SECONDS)) |
| // Two layouts again, since alignment lines were not queried during last layout, |
| // so we did not calculate them speculatively anymore. |
| assertEquals(8, childLayouts) |
| } |
| |
| /** |
| * WithConstraints will cause a requestLayout during layout in some circumstances. |
| * The test here is the minimal example from a bug. |
| */ |
| @SdkSuppress(minSdkVersion = Build.VERSION_CODES.O) |
| @Test |
| fun requestLayoutDuringLayout() { |
| composeSquaresWithConstraints() |
| |
| validateSquareColors(outerColor = Color.Yellow, innerColor = Color.Red, size = 10) |
| } |
| |
| @Test |
| fun testLayoutBeforeDraw_forRecomposingNodesNotAffectingRootSize() { |
| val model = OffsetModel(0.ipx) |
| var latch = CountDownLatch(1) |
| var laidOut = false |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val container = @Composable { children: @Composable() () -> Unit -> |
| // This simulates a Container optimisation, when the child does not |
| // affect parent size. |
| Layout(children) { measurables, constraints -> |
| layout(30.ipx, 30.ipx) { |
| measurables[0].measure(constraints).place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| val recomposingChild = @Composable { children: @Composable() (IntPx) -> Unit -> |
| // This simulates a child that recomposes, for example due to a transition. |
| children(model.offset) |
| } |
| val assumeLayoutBeforeDraw = @Composable { |
| // This assumes a layout was done before the draw pass. |
| Layout({ |
| Draw { _, _ -> |
| assertTrue(laidOut) |
| latch.countDown() |
| } |
| }) { _, _ -> |
| laidOut = true |
| layout(0.ipx, 0.ipx) {} |
| } |
| } |
| |
| container { |
| recomposingChild { |
| assumeLayoutBeforeDraw() |
| } |
| } |
| } |
| } |
| |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| latch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.offset = 10.ipx |
| } |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun testDrawWithLayoutNotPlaced() { |
| var latch = CountDownLatch(1) |
| var drawn = false |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Layout(children = { |
| AtLeastSize(30.ipx) { |
| Draw { _, _ -> |
| drawn = true |
| } |
| } |
| Draw { _, _ -> |
| drawLatch.countDown() |
| } |
| }) { _, _ -> |
| // don't measure or place the AtLeastSize |
| latch.countDown() |
| layout(20.ipx, 20.ipx) {} |
| } |
| } |
| } |
| |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| assertTrue(drawLatch.await(1, TimeUnit.SECONDS)) |
| activityTestRule.runOnUiThreadIR { |
| assertFalse(drawn) |
| } |
| } |
| |
| /** |
| * Because we use invalidate() to cause relayout when children |
| * are laid out, we want to ensure that when the View is 0-sized |
| * that it gets a relayout when it needs to change to non-0 |
| */ |
| @Test |
| fun testZeroSizeCanRelayout() { |
| var latch = CountDownLatch(1) |
| val model = SquareModel(size = 0.ipx) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Layout(children = { }) { _, _ -> |
| latch.countDown() |
| layout(model.size, model.size) {} |
| } |
| } |
| } |
| |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| latch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.size = 10.ipx |
| } |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun testZeroSizeCanRelayout_child() { |
| var latch = CountDownLatch(1) |
| val model = SquareModel(size = 0.ipx) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Layout(children = { |
| Layout(children = {}) { _, _ -> |
| latch.countDown() |
| layout(model.size, model.size) {} |
| } |
| }) { measurables, constraints -> |
| val placeable = measurables[0].measure(constraints) |
| layout(placeable.width, placeable.height) { |
| placeable.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| } |
| |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| latch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.size = 10.ipx |
| } |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun testZeroSizeCanRelayout_childRepaintBoundary() { |
| var latch = CountDownLatch(1) |
| val model = SquareModel(size = 0.ipx) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Layout(children = { |
| RepaintBoundary { |
| Layout(children = {}) { _, _ -> |
| latch.countDown() |
| layout(model.size, model.size) {} |
| } |
| } |
| }) { measurables, constraints -> |
| val placeable = measurables[0].measure(constraints) |
| layout(placeable.width, placeable.height) { |
| placeable.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| } |
| |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| latch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| model.size = 10.ipx |
| } |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun parentSizeForDrawIsProvidedWithoutPadding() { |
| val latch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| AtLeastSize(100.ipx, PaddingModifier(10.ipx)) { |
| Draw { _, parentSize -> |
| assertEquals(100.px, parentSize.width) |
| assertEquals(100.px, parentSize.height) |
| latch.countDown() |
| } |
| } |
| } |
| } |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun parentSizeForDrawInsideRepaintBoundaryIsProvidedWithoutPadding() { |
| val latch = CountDownLatch(1) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| AtLeastSize(100.ipx, PaddingModifier(10.ipx)) { |
| RepaintBoundary { |
| Draw { _, parentSize -> |
| assertEquals(100.px, parentSize.width) |
| assertEquals(100.px, parentSize.height) |
| latch.countDown() |
| } |
| } |
| } |
| } |
| } |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| } |
| |
| @Test |
| fun testParentData_noModifier() { |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Layout({ |
| ParentData(data = 123) { |
| Layout(children = { }) { _, _ -> |
| layout(0.ipx, 0.ipx) { } |
| } |
| } |
| }) { measurables, _ -> |
| assertEquals(123, measurables.first().parentData) |
| layout(0.ipx, 0.ipx) { } |
| } |
| } |
| } |
| } |
| |
| @Test |
| fun testParentData_withModifier() { |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Layout({ |
| ParentData(data = 456) { |
| Layout(children = { }, modifier = PaddingModifier(10.ipx)) { _, _ -> |
| layout(0.ipx, 0.ipx) { } |
| } |
| } |
| }) { measurables, _ -> |
| assertEquals(456, measurables.first().parentData) |
| layout(0.ipx, 0.ipx) { } |
| } |
| } |
| } |
| } |
| |
| @Test |
| fun alignmentLinesInheritedCorrectlyByParents_withModifiedPosition() { |
| val testLine = HorizontalAlignmentLine(::min) |
| val latch = CountDownLatch(1) |
| val alignmentLinePosition = 10.ipx |
| val padding = 20.ipx |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| val child = @Composable { |
| Wrap { |
| Layout({}, modifier = PaddingModifier(padding)) { _, _ -> |
| layout(0.ipx, 0.ipx, mapOf(testLine to alignmentLinePosition)) { } |
| } |
| } |
| } |
| |
| Layout(child) { measurables, constraints -> |
| assertEquals( |
| padding + alignmentLinePosition, |
| measurables[0].measure(constraints)[testLine] |
| ) |
| latch.countDown() |
| layout(0.ipx, 0.ipx) { } |
| } |
| } |
| } |
| } |
| |
| private fun composeSquares(model: SquareModel) { |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Draw { canvas, parentSize -> |
| val paint = Paint() |
| paint.color = model.outerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| Padding(size = model.size) { |
| AtLeastSize(size = model.size) { |
| Draw { canvas, parentSize -> |
| drawLatch.countDown() |
| val paint = Paint() |
| paint.color = model.innerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| private fun composeSquaresWithNestedRepaintBoundaries(model: SquareModel) { |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| FillColor(model, isInner = false, doCountDown = false) |
| Padding(size = model.size) { |
| RepaintBoundary { |
| RepaintBoundary { |
| AtLeastSize(size = model.size) { |
| FillColor(model, isInner = true) |
| } |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| private fun composeMovingSquaresWithRepaintBoundary(model: SquareModel, offset: OffsetModel) { |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| FillColor(model, isInner = false, doCountDown = false) |
| Position(size = model.size * 3, offset = offset) { |
| RepaintBoundary { |
| AtLeastSize(size = model.size) { |
| FillColor(model, isInner = true) |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| private fun composeMovingSquares(model: SquareModel, offset: OffsetModel) { |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| FillColor(model, isInner = false, doCountDown = false) |
| Position(size = model.size * 3, offset = offset) { |
| AtLeastSize(size = model.size) { |
| FillColor(model, isInner = true) |
| } |
| } |
| } |
| } |
| } |
| |
| private fun composeNestedSquares(model: SquareModel) { |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Draw(children = { |
| AtLeastSize(size = (model.size * 3)) { |
| Draw(children = { |
| FillColor(model, isInner = true) |
| }, onPaint = { canvas, parentSize -> |
| val paint = Paint() |
| paint.color = model.outerColor |
| canvas.drawRect(parentSize.toRect(), paint) |
| val start = model.size.value.toFloat() |
| val end = start * 2 |
| canvas.nativeCanvas.save() |
| canvas.clipRect(Rect(start, start, end, end)) |
| drawChildren() |
| canvas.nativeCanvas.restore() |
| }) |
| } |
| }, onPaint = { canvas, parentSize -> |
| val paint = Paint() |
| paint.color = Color(0xFF000000) |
| canvas.drawRect(parentSize.toRect(), paint) |
| }) |
| } |
| } |
| } |
| |
| private fun composeSquaresWithConstraints() { |
| val offset = OffsetModel(0.ipx) |
| activityTestRule.runOnUiThreadIR { |
| activity.setContentInFrameLayout { |
| Draw { canvas, parentSize -> |
| val paint = Paint() |
| paint.color = Color.Yellow |
| canvas.drawRect(parentSize.toRect(), paint) |
| drawLatch.countDown() |
| } |
| Scroller( |
| onScrollPositionChanged = { position, _ -> |
| offset.offset = position |
| }, |
| offset = offset |
| ) { |
| // Need to pass some param here to a separate function or else it works fine |
| TestLayout(5) |
| } |
| } |
| } |
| } |
| |
| private fun validateSquareColors( |
| outerColor: Color, |
| innerColor: Color, |
| size: Int, |
| offset: Int = 0, |
| totalSize: Int = size * 3 |
| ) { |
| assertTrue(drawLatch.await(1, TimeUnit.SECONDS)) |
| val bitmap = activityTestRule.waitAndScreenShot() |
| assertEquals(totalSize, bitmap.width) |
| assertEquals(totalSize, bitmap.height) |
| val squareStart = (totalSize - size) / 2 + offset |
| val squareEnd = totalSize - ((totalSize - size) / 2) + offset |
| for (x in 0 until totalSize) { |
| for (y in 0 until totalSize) { |
| val pixel = Color(bitmap.getPixel(x, y)) |
| val expected = |
| if (!(x < squareStart || x >= squareEnd || y < squareStart || y >= squareEnd)) { |
| innerColor |
| } else { |
| outerColor |
| } |
| assertColorsEqual(expected, pixel) { |
| "Pixel within drawn rect[$x, $y] is $expected, but was $pixel" |
| } |
| } |
| } |
| } |
| |
| @Composable |
| private fun FillColor(color: Color, doCountDown: Boolean = true) { |
| Draw { canvas, parentSize -> |
| canvas.drawRect(parentSize.toRect(), Paint().apply { |
| this.color = color |
| }) |
| if (doCountDown) { |
| drawLatch.countDown() |
| } |
| } |
| } |
| |
| @Composable |
| private fun FillColor(squareModel: SquareModel, isInner: Boolean, doCountDown: Boolean = true) { |
| Draw { canvas, parentSize -> |
| canvas.drawRect(parentSize.toRect(), Paint().apply { |
| this.color = if (isInner) squareModel.innerColor else squareModel.outerColor |
| }) |
| if (doCountDown) { |
| drawLatch.countDown() |
| } |
| } |
| } |
| } |
| |
| fun Bitmap.assertRect( |
| color: Color, |
| holeSize: Int = 0, |
| size: Int = width, |
| centerX: Int = width / 2, |
| centerY: Int = height / 2 |
| ) { |
| assertTrue(centerX + size / 2 <= width) |
| assertTrue(centerX - size / 2 >= 0) |
| assertTrue(centerY + size / 2 <= height) |
| assertTrue(centerY - size / 2 >= 0) |
| val halfHoleSize = holeSize / 2 |
| for (x in centerX - size / 2 until centerX + size / 2) { |
| for (y in centerY - size / 2 until centerY + size / 2) { |
| if (abs(x - centerX) > halfHoleSize && |
| abs(y - centerY) > halfHoleSize |
| ) { |
| val currentColor = Color(getPixel(x, y)) |
| assertColorsEqual(color, currentColor) |
| } |
| } |
| } |
| } |
| |
| fun assertColorsEqual( |
| expected: Color, |
| color: Color, |
| error: () -> String = { "$expected and $color are not similar!" } |
| ) { |
| val errorString = error() |
| assertEquals(errorString, expected.red, color.red, 0.01f) |
| assertEquals(errorString, expected.green, color.green, 0.01f) |
| assertEquals(errorString, expected.blue, color.blue, 0.01f) |
| assertEquals(errorString, expected.alpha, color.alpha, 0.01f) |
| } |
| |
| @Composable |
| fun AtLeastSize( |
| size: IntPx, |
| modifier: Modifier = Modifier.None, |
| children: @Composable() () -> Unit |
| ) { |
| Layout( |
| measureBlock = { measurables, constraints -> |
| val newConstraints = Constraints( |
| minWidth = max(size, constraints.minWidth), |
| maxWidth = max(size, constraints.maxWidth), |
| minHeight = max(size, constraints.minHeight), |
| maxHeight = max(size, constraints.maxHeight) |
| ) |
| val placeables = measurables.map { m -> |
| m.measure(newConstraints) |
| } |
| var maxWidth = size |
| var maxHeight = size |
| placeables.forEach { child -> |
| maxHeight = max(child.height, maxHeight) |
| maxWidth = max(child.width, maxWidth) |
| } |
| layout(maxWidth, maxHeight) { |
| placeables.forEach { child -> |
| child.place(0.ipx, 0.ipx) |
| } |
| } |
| }, |
| modifier = modifier, |
| children = children |
| ) |
| } |
| |
| @Composable |
| fun Align(children: @Composable() () -> Unit) { |
| Layout( |
| measureBlock = { measurables, constraints -> |
| val newConstraints = Constraints( |
| minWidth = IntPx.Zero, |
| maxWidth = constraints.maxWidth, |
| minHeight = IntPx.Zero, |
| maxHeight = constraints.maxHeight |
| ) |
| val placeables = measurables.map { m -> |
| m.measure(newConstraints) |
| } |
| var maxWidth = constraints.minWidth |
| var maxHeight = constraints.minHeight |
| placeables.forEach { child -> |
| maxHeight = max(child.height, maxHeight) |
| maxWidth = max(child.width, maxWidth) |
| } |
| layout(maxWidth, maxHeight) { |
| placeables.forEach { child -> |
| child.place(0.ipx, 0.ipx) |
| } |
| } |
| }, children = children |
| ) |
| } |
| |
| @Composable |
| fun Padding(size: IntPx, children: @Composable() () -> Unit) { |
| Layout( |
| measureBlock = { measurables, constraints -> |
| val totalDiff = size * 2 |
| val newConstraints = Constraints( |
| minWidth = (constraints.minWidth - totalDiff).coerceAtLeast(0.ipx), |
| maxWidth = (constraints.maxWidth - totalDiff).coerceAtLeast(0.ipx), |
| minHeight = (constraints.minHeight - totalDiff).coerceAtLeast(0.ipx), |
| maxHeight = (constraints.maxHeight - totalDiff).coerceAtLeast(0.ipx) |
| ) |
| val placeables = measurables.map { m -> |
| m.measure(newConstraints) |
| } |
| var maxWidth = size |
| var maxHeight = size |
| placeables.forEach { child -> |
| maxHeight = max(child.height + totalDiff, maxHeight) |
| maxWidth = max(child.width + totalDiff, maxWidth) |
| } |
| layout(maxWidth, maxHeight) { |
| placeables.forEach { child -> |
| child.place(size, size) |
| } |
| } |
| }, children = children |
| ) |
| } |
| |
| @Composable |
| fun TwoMeasureLayout( |
| size: IntPx, |
| latch: CountDownLatch, |
| children: @Composable() () -> Unit |
| ) { |
| Layout(children = children) { measurables, _ -> |
| val testConstraints = Constraints() |
| measurables.forEach { it.measure(testConstraints) } |
| val childConstraints = Constraints.tightConstraints(size, size) |
| try { |
| val placeables2 = measurables.map { it.measure(childConstraints) } |
| fail("Measuring twice on the same Measurable should throw an exception") |
| layout(size, size) { |
| placeables2.forEach { child -> |
| child.place(0.ipx, 0.ipx) |
| } |
| } |
| } catch (_: IllegalStateException) { |
| // expected |
| latch.countDown() |
| } |
| layout(0.ipx, 0.ipx) { } |
| } |
| } |
| |
| @Composable |
| fun Position(size: IntPx, offset: OffsetModel, children: @Composable() () -> Unit) { |
| Layout(children) { measurables, constraints -> |
| val placeables = measurables.map { m -> |
| m.measure(constraints) |
| } |
| layout(size, size) { |
| placeables.forEach { child -> |
| child.place(offset.offset, offset.offset) |
| } |
| } |
| } |
| } |
| |
| @Composable |
| fun Wrap(children: @Composable() () -> Unit) { |
| Layout(children) { measurables, constraints -> |
| val placeables = measurables.map { it.measure(constraints) } |
| val width = placeables.maxBy { it.width.value }?.width ?: 0.ipx |
| val height = placeables.maxBy { it.height.value }?.height ?: 0.ipx |
| layout(width, height) { |
| placeables.forEach { it.place(0.ipx, 0.ipx) } |
| } |
| } |
| } |
| |
| @Composable |
| private fun TestLayout(@Suppress("UNUSED_PARAMETER") someInput: Int) { |
| Layout(children = { |
| WithConstraints { |
| NeedsOtherMeasurementComposable(10.ipx) |
| } |
| }) { measurables, constraints -> |
| val withConstraintsPlaceable = measurables[0].measure(constraints) |
| |
| layout(30.ipx, 30.ipx) { |
| withConstraintsPlaceable.place(10.ipx, 10.ipx) |
| } |
| } |
| } |
| |
| @Composable |
| private fun NeedsOtherMeasurementComposable(foo: IntPx) { |
| Layout(children = { |
| Draw { canvas, parentSize -> |
| val paint = Paint() |
| paint.color = Color.Red |
| canvas.drawRect(parentSize.toRect(), paint) |
| } |
| }) { _, _ -> |
| layout(foo, foo) { } |
| } |
| } |
| |
| @Composable |
| private fun Scroller( |
| onScrollPositionChanged: (position: IntPx, maxPosition: IntPx) -> Unit, |
| offset: OffsetModel, |
| child: @Composable() () -> Unit |
| ) { |
| val maxPosition = +state { IntPx.Infinity } |
| ScrollerLayout( |
| maxPosition = maxPosition.value, |
| onMaxPositionChanged = { |
| maxPosition.value = 0.ipx |
| onScrollPositionChanged(offset.offset, 0.ipx) |
| }, |
| child = child |
| ) |
| } |
| |
| @Composable |
| private fun ScrollerLayout( |
| @Suppress("UNUSED_PARAMETER") maxPosition: IntPx, |
| onMaxPositionChanged: () -> Unit, |
| child: @Composable() () -> Unit |
| ) { |
| Layout(child) { measurables, constraints -> |
| val childConstraints = constraints.copy( |
| maxHeight = constraints.maxHeight, |
| maxWidth = IntPx.Infinity |
| ) |
| val childMeasurable = measurables.first() |
| val placeable = childMeasurable.measure(childConstraints) |
| val width = min(placeable.width, constraints.maxWidth) |
| layout(width, placeable.height) { |
| onMaxPositionChanged() |
| placeable.place(0.ipx, 0.ipx) |
| } |
| } |
| } |
| |
| @Composable |
| fun WrapForceRelayout(model: OffsetModel, children: @Composable() () -> Unit) { |
| Layout(children) { measurables, constraints -> |
| val placeables = measurables.map { it.measure(constraints) } |
| val width = placeables.maxBy { it.width.value }?.width ?: 0.ipx |
| val height = placeables.maxBy { it.height.value }?.height ?: 0.ipx |
| layout(width, height) { |
| model.offset |
| placeables.forEach { it.place(0.ipx, 0.ipx) } |
| } |
| } |
| } |
| |
| class DrawCounterListener(private val view: View) : |
| ViewTreeObserver.OnPreDrawListener { |
| val latch = CountDownLatch(5) |
| |
| override fun onPreDraw(): Boolean { |
| latch.countDown() |
| if (latch.count > 0) { |
| view.postInvalidate() |
| } else { |
| view.viewTreeObserver.removeOnPreDrawListener(this) |
| } |
| return true |
| } |
| } |
| |
| fun PaddingModifier(padding: IntPx) = PaddingModifier(padding, padding, padding, padding) |
| |
| data class PaddingModifier( |
| val left: IntPx = 0.ipx, |
| val top: IntPx = 0.ipx, |
| val right: IntPx = 0.ipx, |
| val bottom: IntPx = 0.ipx |
| ) : LayoutModifier { |
| |
| override fun DensityScope.minIntrinsicWidthOf(measurable: Measurable, height: IntPx): IntPx = |
| measurable.minIntrinsicWidth((height - (top + bottom)).coerceAtLeast(0.ipx)) + |
| (left + right) |
| |
| override fun DensityScope.maxIntrinsicWidthOf(measurable: Measurable, height: IntPx): IntPx = |
| measurable.maxIntrinsicWidth((height - (top + bottom)).coerceAtLeast(0.ipx)) + |
| (left + right) |
| |
| override fun DensityScope.minIntrinsicHeightOf(measurable: Measurable, width: IntPx): IntPx = |
| measurable.minIntrinsicHeight((width - (left + right)).coerceAtLeast(0.ipx)) + |
| (top + bottom) |
| |
| override fun DensityScope.maxIntrinsicHeightOf(measurable: Measurable, width: IntPx): IntPx = |
| measurable.maxIntrinsicHeight((width - (left + right)).coerceAtLeast(0.ipx)) + |
| (top + bottom) |
| |
| override fun DensityScope.modifyConstraints( |
| constraints: Constraints |
| ) = constraints.offset( |
| horizontal = -left - right, |
| vertical = -top - bottom |
| ) |
| |
| override fun DensityScope.modifySize( |
| constraints: Constraints, |
| childSize: IntPxSize |
| ) = IntPxSize( |
| (left + childSize.width + right) |
| .coerceIn(constraints.minWidth, constraints.maxWidth), |
| (top + childSize.height + bottom) |
| .coerceIn(constraints.minHeight, constraints.maxHeight) |
| ) |
| |
| override fun DensityScope.modifyPosition( |
| childPosition: IntPxPosition, |
| childSize: IntPxSize, |
| containerSize: IntPxSize |
| ) = IntPxPosition(left + childPosition.x, top + childPosition.y) |
| |
| override fun DensityScope.modifyAlignmentLine(line: AlignmentLine, value: IntPx?): IntPx? { |
| if (value == null) return null |
| return if (line.horizontal) value + left else value + top |
| } |
| |
| override fun DensityScope.modifyParentData(parentData: Any?): Any? = parentData |
| } |
| |
| @Model |
| class SquareModel( |
| var size: IntPx = 10.ipx, |
| var outerColor: Color = Color(0xFF000080), |
| var innerColor: Color = Color(0xFFFFFFFF) |
| ) |
| |
| @Model |
| class OffsetModel(var offset: IntPx) |
| |
| @Model |
| class DoDraw(var value: Boolean = false) |
| |
| // We only need this because IR compiler doesn't like converting lambdas to Runnables |
| fun ActivityTestRule<*>.runOnUiThreadIR(block: () -> Unit) { |
| val runnable: Runnable = object : Runnable { |
| override fun run() { |
| block() |
| } |
| } |
| runOnUiThread(runnable) |
| } |
| |
| fun ActivityTestRule<*>.findAndroidComposeView(): AndroidComposeView { |
| val contentViewGroup = activity.findViewById<ViewGroup>(android.R.id.content) |
| return findAndroidComposeView(contentViewGroup)!! |
| } |
| |
| fun findAndroidComposeView(parent: ViewGroup): AndroidComposeView? { |
| for (index in 0 until parent.childCount) { |
| val child = parent.getChildAt(index) |
| if (child is AndroidComposeView) { |
| return child |
| } else if (child is ViewGroup) { |
| val composeView = findAndroidComposeView(child) |
| if (composeView != null) { |
| return composeView |
| } |
| } |
| } |
| return null |
| } |
| |
| @RequiresApi(Build.VERSION_CODES.O) |
| fun ActivityTestRule<*>.waitAndScreenShot(): Bitmap { |
| val view = findAndroidComposeView() |
| val flushListener = DrawCounterListener(view) |
| val offset = intArrayOf(0, 0) |
| var handler: Handler? = null |
| runOnUiThreadIR { |
| view.getLocationInWindow(offset) |
| view.viewTreeObserver.addOnPreDrawListener(flushListener) |
| view.invalidate() |
| handler = Handler() |
| } |
| |
| assertTrue(flushListener.latch.await(1, TimeUnit.SECONDS)) |
| val width = view.width |
| val height = view.height |
| |
| val dest = |
| Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888) |
| val srcRect = android.graphics.Rect(0, 0, width, height) |
| srcRect.offset(offset[0], offset[1]) |
| val latch = CountDownLatch(1) |
| var copyResult = 0 |
| val onCopyFinished = object : PixelCopy.OnPixelCopyFinishedListener { |
| override fun onPixelCopyFinished(result: Int) { |
| copyResult = result |
| latch.countDown() |
| } |
| } |
| PixelCopy.request(activity.window, srcRect, dest, onCopyFinished, handler!!) |
| assertTrue(latch.await(1, TimeUnit.SECONDS)) |
| assertEquals(PixelCopy.SUCCESS, copyResult) |
| return dest |
| } |
| |
| fun Activity.setContentInFrameLayout(children: @Composable() () -> Unit) { |
| val content = findViewById<ViewGroup>(android.R.id.content) |
| val frameLayout = FrameLayout(this) |
| content.addView(frameLayout, |
| ViewGroup.LayoutParams(ViewGroup.LayoutParams.WRAP_CONTENT, |
| ViewGroup.LayoutParams.WRAP_CONTENT)) |
| frameLayout.setContent(children) |
| } |