1 /*
2  * Copyright (c) 2022-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_TEXT_PICKER_TEXT_PICKER_PAINT_METHOD_H
17 #define FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERN_TEXT_PICKER_TEXT_PICKER_PAINT_METHOD_H
18 
19 #include "base/geometry/ng/size_t.h"
20 #include "base/memory/ace_type.h"
21 #include "base/memory/referenced.h"
22 #include "base/utils/macros.h"
23 #include "base/utils/utils.h"
24 #include "core/components/common/properties/color.h"
25 #include "core/components_ng/pattern/pattern.h"
26 #include "core/components_ng/pattern/text_picker/textpicker_properties.h"
27 #include "core/components_ng/render/node_paint_method.h"
28 namespace OHOS::Ace::NG {
29 
30 class ACE_EXPORT TextPickerPaintMethod : public NodePaintMethod {
31     DECLARE_ACE_TYPE(TextPickerPaintMethod, NodePaintMethod)
32 public:
33     TextPickerPaintMethod() = default;
34     ~TextPickerPaintMethod() override = default;
35 
TextPickerPaintMethod(const WeakPtr<Pattern> & pattern)36     TextPickerPaintMethod(const WeakPtr<Pattern>& pattern)
37     {
38         pattern_ = pattern;
39     }
40 
GetDefaultPickerItemHeight()41     double GetDefaultPickerItemHeight() const
42     {
43         return defaultPickerItemHeight_;
44     }
45 
SetDefaultPickerItemHeight(double defaultPickerItemHeight)46     void SetDefaultPickerItemHeight(double defaultPickerItemHeight)
47     {
48         defaultPickerItemHeight_ = defaultPickerItemHeight;
49     }
50 
SetEnabled(bool enabled)51     void SetEnabled(bool enabled)
52     {
53         enabled_ = enabled;
54     }
55 
56     CanvasDrawFunction GetForegroundDrawFunction(PaintWrapper* paintWrapper) override;
57     void PaintDividerLines(RSCanvas& canvas, const RectF& contentRect, const DividerInfo &info,
58         bool isDefaultLine = true);
59 
60 private:
61     double defaultPickerItemHeight_ = 0.0;
62     bool enabled_ = true;
63 
64     WeakPtr<Pattern> pattern_;
65 
66     bool NeedPaintDividerLines(const RectF &contentRect, const ItemDivider &divider, double dividerHeight,
67         DividerInfo& info);
68     void PaintDefaultDividerLines(RSCanvas& canvas, const RectF &contentRect, double dividerHeight);
69     void PaintCustomDividerLines(RSCanvas& canvas, const RectF &contentRect, const RectF &frameRect,
70         const ItemDivider &divider, double dividerHeight);
71     void PaintLine(const OffsetF& offset, const DividerInfo &info, RSCanvas& canvas);
72     bool SetStrokeWidth(const ItemDivider &divider, double dividerHeight, DividerInfo& info);
73 };
74 } // namespace OHOS::Ace::NG
75 #endif // FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERN_TEXT_PICKER_TEXT_PICKER_PAINT_METHOD_H
76