1 /*
2  * Copyright (c) 2024 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_BAR_SCROLL_BAR_PAINT_METHOD_H
17 #define FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERN_SCROLL_BAR_SCROLL_BAR_PAINT_METHOD_H
18 
19 #include "core/components_ng/pattern/scroll/inner/scroll_bar.h"
20 #include "core/components_ng/render/node_paint_method.h"
21 
22 namespace OHOS::Ace::NG {
23 
24 class ScrollBarPaintMethod : public NodePaintMethod {
DECLARE_ACE_TYPE(ScrollBarPaintMethod,NodePaintMethod)25     DECLARE_ACE_TYPE(ScrollBarPaintMethod, NodePaintMethod)
26 
27 public:
28     explicit ScrollBarPaintMethod(bool hasChild) : hasChild_(hasChild) {};
29     ~ScrollBarPaintMethod() override = default;
30 
SetScrollBar(WeakPtr<ScrollBar> && scrollBar)31     void SetScrollBar(WeakPtr<ScrollBar>&& scrollBar)
32     {
33         scrollBar_ = scrollBar;
34     }
35 
SetScrollBarOverlayModifier(WeakPtr<ScrollBarOverlayModifier> && scrollBarOverlayModifier)36     void SetScrollBarOverlayModifier(WeakPtr<ScrollBarOverlayModifier>&& scrollBarOverlayModifier)
37     {
38         scrollBarOverlayModifier_ = scrollBarOverlayModifier;
39     }
40 
GetOverlayModifier(PaintWrapper * paintWrapper)41     RefPtr<Modifier> GetOverlayModifier(PaintWrapper* paintWrapper) override
42     {
43         return scrollBarOverlayModifier_.Upgrade();
44     }
45 
46     void UpdateOverlayModifier(PaintWrapper* paintWrapper) override;
47 
48 private:
49     bool hasChild_ = false;
50     WeakPtr<ScrollBar> scrollBar_;
51     WeakPtr<ScrollBarOverlayModifier> scrollBarOverlayModifier_;
52 };
53 } // namespace OHOS::Ace::NG
54 
55 #endif // FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERN_SCROLL_BAR_SCROLL_BAR_PAINT_METHOD_H
56