1 /*
2  * Copyright (c) 2023 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_PATTERNS_BADGE_BADGE_MODEL_NG_H
17 #define FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERNS_BADGE_BADGE_MODEL_NG_H
18 
19 #include "core/components_ng/pattern/badge/badge_model.h"
20 
21 namespace OHOS::Ace::NG {
22 class ACE_EXPORT BadgeModelNG : public OHOS::Ace::BadgeModel {
23 public:
24     void Create(BadgeParameters& badgeParameters) override;
SetIsDefault(bool isDefaultFontSize,bool isDefaultBadgeSize)25     void SetIsDefault(bool isDefaultFontSize, bool isDefaultBadgeSize) override
26     {
27         isDefaultFontSize_ = isDefaultFontSize;
28         isDefaultBadgeSize_ = isDefaultBadgeSize;
29     };
30 
31     static RefPtr<FrameNode> CreateFrameNode(int32_t nodeId);
32     static void SetBadgeParam(
33         FrameNode* frameNode, BadgeParameters& badgeParameters, bool isDefaultFontSize, bool isDefaultBadgeSize);
34 
35 private:
36     static void UpdateBadgeStyle(BadgeParameters& badgeParameters, const RefPtr<FrameNode>& frameNode);
37     bool isDefaultFontSize_ = true;
38     bool isDefaultBadgeSize_ = true;
39 };
40 } // namespace OHOS::Ace::NG
41 #endif // FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_PATTERNS_BADGE_BADGE_MODEL_NG_H
42