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 META_SRC_MODEL_COMPOSITE_OBJECT_PROVIDER_H
17 #define META_SRC_MODEL_COMPOSITE_OBJECT_PROVIDER_H
18 
19 #include <base/containers/unordered_map.h>
20 
21 #include <meta/base/interface_macros.h>
22 #include <meta/ext/object_container.h>
23 #include <meta/interface/model/intf_object_provider.h>
24 
META_BEGIN_NAMESPACE()25 META_BEGIN_NAMESPACE()
26 
27 class CompositeObjectProvider
28     : public ObjectFlatContainerFwd<CompositeObjectProvider, ClassId::CompositeObjectProvider, IObjectProvider> {
29 public:
30     ~CompositeObjectProvider() override;
31     CompositeObjectProvider();
32     META_NO_COPY_MOVE(CompositeObjectProvider)
33 
34     bool Build(const IMetadata::Ptr&) override;
35 
36     IObject::Ptr CreateObject(const DataModelIndex& index) override;
37     bool DisposeObject(const META_NS::IObject::Ptr& item) override;
38     size_t GetObjectCount(const DataModelIndex& index) const override;
39 
40     META_IMPLEMENT_INTERFACE_EVENT(IObjectProvider, IOnDataAdded, OnDataAdded)
41     META_IMPLEMENT_INTERFACE_EVENT(IObjectProvider, IOnDataRemoved, OnDataRemoved)
42     META_IMPLEMENT_INTERFACE_EVENT(IObjectProvider, IOnDataMoved, OnDataMoved)
43 
44     META_IMPLEMENT_INTERFACE_PROPERTY(IObjectProvider, size_t, CacheHint, 10);
45 
46 private:
47     void OnAddedProviderData(const IObjectProvider::Ptr& provider, DataModelIndex base, size_t count);
48     void OnRemovedProviderData(const IObjectProvider::Ptr& provider, DataModelIndex base, size_t count);
49     void OnMovedProviderData(
50         const IObjectProvider::Ptr& provider, DataModelIndex from, size_t count, DataModelIndex to);
51 
52     IObjectProvider* FindProvider(size_t& index) const;
53     size_t CalculateIndex(const IObjectProvider::Ptr& provider, size_t localIndex) const;
54     size_t CalculateIndexBase(size_t provider) const;
55 
56     void OnProviderAdded(const ChildChangedInfo& info);
57     void OnProviderRemoved(const ChildChangedInfo& info);
58     void OnProviderMoved(const ChildMovedInfo& info);
59 
60 private:
61     BASE_NS::unordered_map<const IObject*, IObjectProvider*> objects_;
62 };
63 
64 META_END_NAMESPACE()
65 
66 #endif
67