/*
* Copyright 2022 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.tv.foundation.lazy.grid
import androidx.compose.animation.core.FiniteAnimationSpec
import androidx.compose.foundation.ExperimentalFoundationApi
import androidx.compose.ui.layout.Placeable
import androidx.compose.ui.unit.IntOffset
import androidx.compose.ui.unit.IntSize
import androidx.compose.ui.unit.LayoutDirection
import androidx.compose.ui.util.fastForEach
/**
* Represents one measured item of the lazy grid. It can in fact consist of multiple placeables
* if the user emit multiple layout nodes in the item callback.
*/
@OptIn(ExperimentalFoundationApi::class)
internal class LazyMeasuredItem(
val index: ItemIndex,
val key: Any,
private val isVertical: Boolean,
/**
* Cross axis size is the same for all [placeables]. Take it as parameter for the case when
* [placeables] is empty.
*/
val crossAxisSize: Int,
val mainAxisSpacing: Int,
private val reverseLayout: Boolean,
private val layoutDirection: LayoutDirection,
private val beforeContentPadding: Int,
private val afterContentPadding: Int,
val placeables: List<Placeable>,
private val placementAnimator: LazyGridItemPlacementAnimator,
/**
* The offset which shouldn't affect any calculations but needs to be applied for the final
* value passed into the place() call.
*/
private val visualOffset: IntOffset
) {
/**
* Main axis size of the item - the max main axis size of the placeables.
*/
val mainAxisSize: Int
/**
* The max main axis size of the placeables plus mainAxisSpacing.
*/
val mainAxisSizeWithSpacings: Int
init {
var maxMainAxis = 0
placeables.fastForEach {
maxMainAxis = maxOf(maxMainAxis, if (isVertical) it.height else it.width)
}
mainAxisSize = maxMainAxis
mainAxisSizeWithSpacings = (maxMainAxis + mainAxisSpacing).coerceAtLeast(0)
}
/**
* Calculates positions for the inner placeables at [rawCrossAxisOffset], [rawCrossAxisOffset].
* [layoutWidth] and [layoutHeight] should be provided to not place placeables which are ended
* up outside of the viewport (for example one item consist of 2 placeables, and the first one
* is not going to be visible, so we don't place it as an optimization, but place the second
* one). If [reverseOrder] is true the inner placeables would be placed in the inverted order.
*/
fun position(
rawMainAxisOffset: Int,
rawCrossAxisOffset: Int,
layoutWidth: Int,
layoutHeight: Int,
row: Int,
column: Int,
lineMainAxisSize: Int
): TvLazyGridPositionedItem {
val wrappers = mutableListOf<LazyGridPlaceableWrapper>()
val mainAxisLayoutSize = if (isVertical) layoutHeight else layoutWidth
val mainAxisOffset = if (reverseLayout) {
mainAxisLayoutSize - rawMainAxisOffset - mainAxisSize
} else {
rawMainAxisOffset
}
val crossAxisLayoutSize = if (isVertical) layoutWidth else layoutHeight
val crossAxisOffset = if (isVertical && layoutDirection == LayoutDirection.Rtl) {
crossAxisLayoutSize - rawCrossAxisOffset - crossAxisSize
} else {
rawCrossAxisOffset
}
val placeableOffset = if (isVertical) {
IntOffset(crossAxisOffset, mainAxisOffset)
} else {
IntOffset(mainAxisOffset, crossAxisOffset)
}
var placeableIndex = if (reverseLayout) placeables.lastIndex else 0
while (if (reverseLayout) placeableIndex >= 0 else placeableIndex < placeables.size) {
val it = placeables[placeableIndex]
val addIndex = if (reverseLayout) 0 else wrappers.size
wrappers.add(
addIndex,
LazyGridPlaceableWrapper(placeableOffset, it, placeables[placeableIndex].parentData)
)
if (reverseLayout) placeableIndex-- else placeableIndex++
}
return TvLazyGridPositionedItem(
offset = if (isVertical) {
IntOffset(rawCrossAxisOffset, rawMainAxisOffset)
} else {
IntOffset(rawMainAxisOffset, rawCrossAxisOffset)
},
placeableOffset = placeableOffset,
index = index.value,
key = key,
row = row,
column = column,
size = if (isVertical) {
IntSize(crossAxisSize, mainAxisSize)
} else {
IntSize(mainAxisSize, crossAxisSize)
},
lineMainAxisSize = lineMainAxisSize,
mainAxisSpacing = mainAxisSpacing,
minMainAxisOffset = -if (!reverseLayout) {
beforeContentPadding
} else {
afterContentPadding
},
maxMainAxisOffset = mainAxisLayoutSize +
if (!reverseLayout) afterContentPadding else beforeContentPadding,
isVertical = isVertical,
wrappers = wrappers,
placementAnimator = placementAnimator,
visualOffset = visualOffset
)
}
}
@OptIn(ExperimentalFoundationApi::class)
internal class TvLazyGridPositionedItem(
override val offset: IntOffset,
val placeableOffset: IntOffset,
override val index: Int,
override val key: Any,
override val row: Int,
override val column: Int,
override val size: IntSize,
val lineMainAxisSize: Int,
private val mainAxisSpacing: Int,
private val minMainAxisOffset: Int,
private val maxMainAxisOffset: Int,
private val isVertical: Boolean,
private val wrappers: List<LazyGridPlaceableWrapper>,
private val placementAnimator: LazyGridItemPlacementAnimator,
private val visualOffset: IntOffset
) : TvLazyGridItemInfo {
val placeablesCount: Int get() = wrappers.size
val mainAxisSizeWithSpacings: Int get() =
mainAxisSpacing + if (isVertical) size.height else size.width
val lineMainAxisSizeWithSpacings: Int get() = mainAxisSpacing + lineMainAxisSize
fun getMainAxisSize(index: Int) = wrappers[index].placeable.mainAxisSize
fun getCrossAxisSize() = if (isVertical) size.width else size.height
fun getCrossAxisOffset() = if (isVertical) offset.x else offset.y
@Suppress("UNCHECKED_CAST")
fun getAnimationSpec(index: Int) =
wrappers[index].parentData as? FiniteAnimationSpec<IntOffset>?
val hasAnimations = run {
repeat(placeablesCount) { index ->
if (getAnimationSpec(index) != null) {
return@run true
}
}
false
}
fun place(
scope: Placeable.PlacementScope,
) = with(scope) {
repeat(placeablesCount) { index ->
val placeable = wrappers[index].placeable
val minOffset = minMainAxisOffset - placeable.mainAxisSize
val maxOffset = maxMainAxisOffset
val offset = if (getAnimationSpec(index) != null) {
placementAnimator.getAnimatedOffset(
key, index, minOffset, maxOffset, placeableOffset
)
} else {
placeableOffset
}
if (offset.mainAxis > minOffset && offset.mainAxis < maxOffset) {
if (isVertical) {
placeable.placeWithLayer(offset + visualOffset)
} else {
placeable.placeRelativeWithLayer(offset + visualOffset)
}
}
}
}
private val IntOffset.mainAxis get() = if (isVertical) y else x
private val Placeable.mainAxisSize get() = if (isVertical) height else width
}
internal class LazyGridPlaceableWrapper(
val offset: IntOffset,
val placeable: Placeable,
val parentData: Any?
)