1 /* 2 * Copyright (c) 2022 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERN_SCROLL_SCROLL_PAINT_METHOD_H 17 #define FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERN_SCROLL_SCROLL_PAINT_METHOD_H 18 19 #include "core/components_ng/pattern/scroll/inner/scroll_bar.h" 20 #include "core/components_ng/pattern/scroll/inner/scroll_bar_2d.h" 21 #include "core/components_ng/pattern/scroll/inner/scroll_bar_2d_painter.h" 22 #include "core/components_ng/pattern/scroll/scroll_content_modifier.h" 23 #include "core/components_ng/pattern/scroll/scroll_edge_effect.h" 24 #include "core/components_ng/pattern/scrollable/scrollable_paint_method.h" 25 26 namespace OHOS::Ace::NG { 27 28 class ScrollPaintMethod : public ScrollablePaintMethod { 29 DECLARE_ACE_TYPE(ScrollPaintMethod, ScrollablePaintMethod); 30 31 public: 32 ScrollPaintMethod() = default; ScrollPaintMethod(bool vertical,bool isReverse)33 ScrollPaintMethod(bool vertical, bool isReverse) : ScrollablePaintMethod(vertical, isReverse) {} 34 ~ScrollPaintMethod() override = default; 35 36 CanvasDrawFunction GetForegroundDrawFunction(PaintWrapper* paintWrapper) override; 37 SetScrollBar(WeakPtr<ScrollBar> && scrollBar)38 void SetScrollBar(WeakPtr<ScrollBar>&& scrollBar) 39 { 40 scrollBar_ = scrollBar; 41 } 42 GetScrollBar()43 RefPtr<ScrollBar> GetScrollBar() 44 { 45 return scrollBar_.Upgrade(); 46 } 47 SetEdgeEffect(WeakPtr<ScrollEdgeEffect> && edgeEffect)48 void SetEdgeEffect(WeakPtr<ScrollEdgeEffect>&& edgeEffect) 49 { 50 edgeEffect_ = edgeEffect; 51 } 52 SetScrollBarOverlayModifier(WeakPtr<ScrollBarOverlayModifier> && scrollBarOverlayModifier)53 void SetScrollBarOverlayModifier(WeakPtr<ScrollBarOverlayModifier>&& scrollBarOverlayModifier) 54 { 55 scrollBarOverlayModifier_ = scrollBarOverlayModifier; 56 } 57 GetOverlayModifier(PaintWrapper * paintWrapper)58 RefPtr<Modifier> GetOverlayModifier(PaintWrapper* paintWrapper) override 59 { 60 if (auto scrollBar2d = scrollBar2d_.Upgrade()) { 61 return scrollBar2d->GetPainter(); 62 } 63 return scrollBarOverlayModifier_.Upgrade(); 64 } 65 66 void UpdateOverlayModifier(PaintWrapper* paintWrapper) override; 67 GetContentModifier(PaintWrapper * paintWrapper)68 RefPtr<Modifier> GetContentModifier(PaintWrapper* paintWrapper) override 69 { 70 return scrollContentModifier_; 71 } 72 73 void UpdateContentModifier(PaintWrapper* paintWrapper) override; 74 SetContentModifier(const RefPtr<ScrollContentModifier> & modify)75 void SetContentModifier(const RefPtr<ScrollContentModifier>& modify) 76 { 77 scrollContentModifier_ = modify; 78 } 79 80 private: 81 RefPtr<ScrollContentModifier> scrollContentModifier_; 82 void PaintScrollEffect(RSCanvas& canvas, PaintWrapper* paintWrapper) const; 83 84 WeakPtr<ScrollBar> scrollBar_; 85 WeakPtr<ScrollEdgeEffect> edgeEffect_; 86 WeakPtr<ScrollBarOverlayModifier> scrollBarOverlayModifier_; 87 88 /* ============================= Free Scroll Enhancements ============================= */ 89 public: Set2DPainter(const WeakPtr<ScrollBar2D> & scrollBar)90 void Set2DPainter(const WeakPtr<ScrollBar2D>& scrollBar) 91 { 92 scrollBar2d_ = scrollBar; 93 } 94 95 private: 96 void UpdateOverlayModifier2d(PaintWrapper* paintWrapper); 97 98 WeakPtr<ScrollBar2D> scrollBar2d_; 99 /* ============================================================================== */ 100 }; 101 } // namespace OHOS::Ace::NG 102 103 #endif // FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERN_SCROLL_SCROLL_PAINT_METHOD_H 104