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 SCENE_SRC_ENTITY_CONVERTING_VALUE_H 17 #define SCENE_SRC_ENTITY_CONVERTING_VALUE_H 18 19 #include <scene/ext/intf_ecs_context.h> 20 #include <scene/ext/intf_render_resource.h> 21 22 #include <3d/ecs/components/render_handle_component.h> 23 24 #include "converting_value.h" 25 #include "util.h" 26 27 SCENE_BEGIN_NAMESPACE() 28 29 // Interface pointer to entity converter 30 template<typename Interface> 31 struct InterfacePtrEntityConverter { 32 using SourceType = typename Interface::Ptr; 33 using TargetType = CORE_NS::Entity; 34 InterfacePtrEntityConverterInterfacePtrEntityConverter35 InterfacePtrEntityConverter(IInternalScene::Ptr scene, META_NS::ObjectId id) : scene_(scene), id_(id) {} 36 ConvertToSourceInterfacePtrEntityConverter37 SourceType ConvertToSource(META_NS::IAny& any, const TargetType& v) const 38 { 39 auto p = META_NS::GetPointer<Interface>(any); 40 if (auto scene = scene_.lock()) { 41 if (CORE_NS::EntityUtil::IsValid(v)) { 42 auto f = scene->AddTask([&] { return scene->GetEcsContext().GetEcsObject(v); }); 43 auto obj = f.GetResult(); 44 if (p) { 45 if (auto i = interface_cast<IEcsObjectAccess>(p)) { 46 if (i->GetEcsObject() != obj) { 47 p = nullptr; 48 } 49 } 50 } 51 if (!p) { 52 p = META_NS::GetObjectRegistry().Create<Interface>(id_); 53 if (auto i = interface_cast<IEcsObjectAccess>(p)) { 54 i->SetEcsObject(obj); 55 } 56 } 57 } else { 58 // need to release the entity? 59 p = nullptr; 60 } 61 } 62 return p; 63 } ConvertToTargetInterfacePtrEntityConverter64 TargetType ConvertToTarget(const SourceType& v) const 65 { 66 CORE_NS::Entity ent; 67 if (auto scene = scene_.lock()) { 68 if (v) { 69 if (auto i = interface_cast<IEcsObjectAccess>(v)) { 70 ent = i->GetEcsObject()->GetEntity(); 71 } 72 } 73 } 74 return ent; 75 } 76 77 IInternalScene::WeakPtr scene_; 78 META_NS::ObjectId id_; 79 }; 80 81 template<typename Interface> 82 using InterfacePtrEntityValue = ConvertingValue<InterfacePtrEntityConverter<Interface>>; 83 84 // render resource to entity converter 85 template<typename Interface> 86 struct RenderResourceConverter { 87 using SourceType = typename Interface::Ptr; 88 using TargetType = CORE_NS::EntityReference; 89 RenderResourceConverterRenderResourceConverter90 RenderResourceConverter(IInternalScene::Ptr scene, META_NS::ObjectId id) : scene_(scene), id_(id) {} 91 ConvertToSourceRenderResourceConverter92 SourceType ConvertToSource(META_NS::IAny& any, const TargetType& v) const 93 { 94 auto p = META_NS::GetPointer<Interface>(any); 95 if (auto scene = scene_.lock()) { 96 if (v) { 97 p = scene->AddTask([&] { return ObjectWithRenderHandle<Interface>(scene, v, p, id_); }).GetResult(); 98 } else { 99 p = nullptr; 100 } 101 } 102 return p; 103 } 104 ConvertToTargetRenderResourceConverter105 TargetType ConvertToTarget(const SourceType& v) const 106 { 107 CORE_NS::EntityReference ent; 108 if (auto scene = scene_.lock()) { 109 if (auto i = interface_cast<IEcsResource>(v)) { 110 scene->AddTask([&] { ent = scene->GetEcsContext().GetEntityReference(i->GetEntity()); }).Wait(); 111 } 112 } 113 return ent; 114 } 115 116 IInternalScene::WeakPtr scene_; 117 META_NS::ObjectId id_; 118 }; 119 120 template<typename Interface> 121 using RenderResourceValue = ConvertingValue<RenderResourceConverter<Interface>>; 122 123 SCENE_END_NAMESPACE() 124 125 #endif 126