1 /*
2  * Copyright (c) 2021 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_FLEX_FLEX_ELEMENT_H
17 #define FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_FLEX_FLEX_ELEMENT_H
18 
19 #include "base/utils/macros.h"
20 #include "core/common/container.h"
21 #include "core/components/common/layout/constants.h"
22 #include "core/pipeline/base/component_group.h"
23 #include "core/focus/focus_node.h"
24 #include "core/pipeline/base/component_group_element.h"
25 
26 namespace OHOS::Ace {
27 
28 class ACE_EXPORT FlexElement : public ComponentGroupElement, public FocusGroup {
29     DECLARE_ACE_TYPE(FlexElement, ComponentGroupElement, FocusGroup);
30 
31 public:
32     void Update() override;
33     bool RequestNextFocus(bool vertical, bool reverse, const Rect& rect) override;
34     bool AcceptFocusByRectOfLastFocus(const Rect& rect) override;
CanUpdate(const RefPtr<Component> & newComponent)35     bool CanUpdate(const RefPtr<Component>& newComponent) override
36     {
37         // The raw ptr is persistent during app process.
38         auto flexComponent = AceType::DynamicCast<ComponentGroup>(newComponent);
39         if (!flexComponent) {
40             return false;
41         }
42 
43         // partial update does not produce children
44         auto pipelineContext = context_.Upgrade();
45         if (Container::IsCurrentUsePartialUpdate()) {
46             return true;
47         }
48         return GetChildren().size() == flexComponent->GetSizeOfChildren();
49     }
50 
51 protected:
GetCachedRenderNode()52     RefPtr<RenderNode> GetCachedRenderNode() override
53     {
54         auto context = GetContext().Upgrade();
55         if (context) {
56             return context->GetRenderFactory()->GetRenderFlexFactory()->Get();
57         }
58         return nullptr;
59     }
60 
61     FlexDirection direction_ { FlexDirection::ROW };
62 };
63 
64 class RowElement : public FlexElement {
65     DECLARE_ACE_TYPE(RowElement, FlexElement);
66 };
67 
68 class ColumnElement : public FlexElement {
69     DECLARE_ACE_TYPE(ColumnElement, FlexElement);
70 };
71 
72 } // namespace OHOS::Ace
73 
74 #endif // FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_FLEX_FLEX_ELEMENT_H
75