All Downloads are FREE. Search and download functionalities are using the official Maven repository.

commonMain.androidx.compose.foundation.gestures.MouseWheelScrollable.kt Maven / Gradle / Ivy

Go to download

Higher level abstractions of the Compose UI primitives. This library is design system agnostic, providing the high-level building blocks for both application and design-system developers

There is a newer version: 1.8.0-alpha01
Show newest version
/*
 * Copyright 2023 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.compose.foundation.gestures

import androidx.compose.animation.core.AnimationState
import androidx.compose.animation.core.LinearEasing
import androidx.compose.animation.core.animateTo
import androidx.compose.animation.core.tween
import androidx.compose.ui.geometry.Offset
import androidx.compose.ui.input.pointer.*
import androidx.compose.ui.node.DelegatingNode
import androidx.compose.ui.node.ModifierNodeElement
import androidx.compose.ui.node.PointerInputModifierNode
import androidx.compose.ui.platform.InspectorInfo
import androidx.compose.ui.unit.IntSize
import androidx.compose.ui.unit.dp
import androidx.compose.ui.util.fastAny
import androidx.compose.ui.util.fastForEach
import kotlin.math.abs
import kotlin.math.roundToInt
import kotlinx.coroutines.CancellationException
import kotlinx.coroutines.channels.Channel
import kotlinx.coroutines.isActive
import kotlinx.coroutines.launch

internal class MouseWheelScrollableElement(
    val scrollingLogic: ScrollingLogic,
    val mouseWheelScrollConfig: ScrollConfig,
    val density: Float,
    val enabled: Boolean,
) : ModifierNodeElement() {
    override fun create(): MouseWheelScrollNode {
        return if (mouseWheelScrollConfig.isSmoothScrollingEnabled) {
            AnimatedMouseWheelScrollNode(scrollingLogic, mouseWheelScrollConfig, density, enabled)
        } else {
            RawMouseWheelScrollNode(scrollingLogic, mouseWheelScrollConfig, enabled)
        }
    }

    override fun update(node: MouseWheelScrollNode) {
        node.scrollingLogic = scrollingLogic
        node.mouseWheelScrollConfig = mouseWheelScrollConfig
        node.enabled = enabled
    }

    override fun hashCode(): Int {
        var result = scrollingLogic.hashCode()
        result = 31 * result + mouseWheelScrollConfig.hashCode()
        result = 31 * result + enabled.hashCode()
        return result
    }

    override fun equals(other: Any?): Boolean {
        if (this === other) return true
        if (other !is MouseWheelScrollableElement) return false

        if (scrollingLogic != other.scrollingLogic) return false
        if (mouseWheelScrollConfig != other.mouseWheelScrollConfig) return false
        if (enabled != other.enabled) return false
        return true
    }

    override fun InspectorInfo.inspectableProperties() = Unit
}

internal abstract class MouseWheelScrollNode(
    var scrollingLogic: ScrollingLogic,
    var mouseWheelScrollConfig: ScrollConfig,
    private var _enabled: Boolean,
) : DelegatingNode(), PointerInputModifierNode {
    private val pointerInputNode = delegate(SuspendingPointerInputModifierNode {
        if (_enabled) {
            mouseWheelInput()
        }
    })

    var enabled
        get() = _enabled
        set(value) {
            if (_enabled != value) {
                _enabled = value
                pointerInputNode.resetPointerInputHandler()
            }
        }

    override fun onPointerEvent(
        pointerEvent: PointerEvent,
        pass: PointerEventPass,
        bounds: IntSize
    ) {
        pointerInputNode.onPointerEvent(pointerEvent, pass, bounds)
    }

    override fun onCancelPointerInput() {
        pointerInputNode.onCancelPointerInput()
    }

    private suspend fun PointerInputScope.mouseWheelInput() = awaitPointerEventScope {
        while (true) {
            val event = awaitScrollEvent()
            if (!event.isConsumed) {
                val consumed = onMouseWheel(event)
                if (consumed) {
                    event.consume()
                }
            }
        }
    }

    protected abstract fun PointerInputScope.onMouseWheel(pointerEvent: PointerEvent): Boolean

    private suspend fun AwaitPointerEventScope.awaitScrollEvent(): PointerEvent {
        var event: PointerEvent
        do {
            event = awaitPointerEvent()
        } while (event.type != PointerEventType.Scroll)
        return event
    }

    private inline val PointerEvent.isConsumed: Boolean get() = changes.fastAny { it.isConsumed }
    private inline fun PointerEvent.consume() = changes.fastForEach { it.consume() }
}

private class RawMouseWheelScrollNode(
    scrollingLogic: ScrollingLogic,
    mouseWheelScrollConfig: ScrollConfig,
    enabled: Boolean,
) : MouseWheelScrollNode(scrollingLogic, mouseWheelScrollConfig, enabled) {
    override fun PointerInputScope.onMouseWheel(pointerEvent: PointerEvent): Boolean {
        val delta = with(mouseWheelScrollConfig) {
            calculateMouseWheelScroll(pointerEvent, size)
        }
        return scrollingLogic.dispatchRawDelta(delta) != Offset.Zero
    }
}

private class AnimatedMouseWheelScrollNode(
    scrollingLogic: ScrollingLogic,
    mouseWheelScrollConfig: ScrollConfig,
    val density: Float,
    enabled: Boolean,
) : MouseWheelScrollNode(scrollingLogic, mouseWheelScrollConfig, enabled) {
    private var isAnimationRunning = false
    private val channel = Channel(capacity = Channel.UNLIMITED)

    override fun onAttach() {
        coroutineScope.launch {
            while (isActive) {
                val event = channel.receive()
                isAnimationRunning = true
                try {
                    scrollingLogic.animatedDispatchScroll(event, speed = 1f * density) {
                        // Sum delta from all pending events to avoid multiple animation restarts.
                        channel.sumOrNull()
                    }
                } finally {
                    isAnimationRunning = false
                }
            }
        }
    }

    override fun PointerInputScope.onMouseWheel(pointerEvent: PointerEvent): Boolean {
        val scrollDelta = with(mouseWheelScrollConfig) {
            calculateMouseWheelScroll(pointerEvent, size)
        }
        return if (mouseWheelScrollConfig.isPreciseWheelScroll(pointerEvent)) {
            // In case of high-resolution wheel, such as a freely rotating wheel with no notches
            // or trackpads, delta should apply directly without any delays.
            scrollingLogic.dispatchRawDelta(scrollDelta) != Offset.Zero

            /*
             * TODO Set isScrollInProgress to true in case of touchpad.
             *  Dispatching raw delta doesn't cause a progress indication even with wrapping in
             *  `scrollableState.scroll` block, since it applies the change within single frame.
             *  Touchpads emit just multiple mouse wheel events, so detecting start and end of this
             *  "gesture" is not straight forward.
             *  Ideally it should be resolved by catching real touches from input device instead of
             *  introducing a timeout (after each event before resetting progress flag).
             */
        } else with(scrollingLogic) {
            val delta = scrollDelta.reverseIfNeeded().toFloat()
            if (isAnimationRunning) {
                channel.trySend(delta).isSuccess
            } else {
                // Try to apply small delta immediately to conditionally consume
                // an input event and to avoid useless animation.
                tryToScrollBySmallDelta(delta, threshold = 4.dp.toPx()) {
                    channel.trySend(it).isSuccess
                }
            }
        }
    }

    private fun Channel.sumOrNull(): Float? {
        val elements = untilNull { tryReceive().getOrNull() }.toList()
        return if (elements.isEmpty()) null else elements.sum()
    }

    private fun  untilNull(builderAction: () -> E?) = sequence {
        do {
            val element = builderAction()?.also {
                yield(it)
            }
        } while (element != null)
    }

    private fun ScrollingLogic.tryToScrollBySmallDelta(
        delta: Float,
        threshold: Float = 4f,
        fallback: (Float) -> Boolean
    ): Boolean {
        return if (abs(delta) > threshold) {
            // Gather possibility to scroll by applying a piece of required delta.
            val testDelta = if (delta > 0f) threshold else -threshold
            val consumedDelta = scrollableState.dispatchRawDelta(testDelta)
            consumedDelta != 0f && fallback(delta - testDelta)
        } else {
            val consumedDelta = scrollableState.dispatchRawDelta(delta)
            consumedDelta != 0f
        }
    }

    private suspend fun ScrollingLogic.animatedDispatchScroll(
        eventDelta: Float,
        speed: Float = 1f,
        maxDurationMillis: Int = 100,
        tryReceiveNext: () -> Float?
    ) {
        var target = eventDelta
        tryReceiveNext()?.let {
            target += it
        }
        if (target.isLowScrollingDelta()) {
            return
        }
        var requiredAnimation = true
        var lastValue = 0f
        val anim = AnimationState(0f)
        while (requiredAnimation) {
            requiredAnimation = false
            val durationMillis = (abs(target - anim.value) / speed)
                .roundToInt()
                .coerceAtMost(maxDurationMillis)
            try {
                scrollableState.scroll {
                    anim.animateTo(
                        target,
                        animationSpec = tween(
                            durationMillis = durationMillis,
                            easing = LinearEasing
                        ),
                        sequentialAnimation = true
                    ) {
                        val delta = value - lastValue
                        if (!delta.isLowScrollingDelta()) {
                            val consumedDelta = scrollBy(delta)
                            if (!(delta - consumedDelta).isLowScrollingDelta()) {
                                cancelAnimation()
                                return@animateTo
                            }
                            lastValue += delta
                        }
                        tryReceiveNext()?.let {
                            target += it
                            requiredAnimation = !(target - lastValue).isLowScrollingDelta()
                            cancelAnimation()
                        }
                    }
                }
            } catch (ignore: CancellationException) {
                requiredAnimation = true
            }
        }
    }
}

/*
 * Returns true, if the value is too low for visible change in scroll (consumed delta, animation-based change, etc),
 * false otherwise
 */
private inline fun Float.isLowScrollingDelta(): Boolean = abs(this) < 0.5f




© 2015 - 2025 Weber Informatics LLC | Privacy Policy