renamed SharedThumbnailUtils to SharedPreviewUtils

Signed-off-by: Guthrie Adams <guthadam@amazon.com>
2021-10-10 02:10:15 -05:00
parent 9f92bd2d33
commit 8aea92af29
7 changed files with 10 additions and 17 deletions
@@ -0,0 +1,116 @@
+/*
+ * Copyright (c) Contributors to the Open 3D Engine Project.
+ * For complete copyright and license terms please see the LICENSE at the root of this distribution.
+ *
+ * SPDX-License-Identifier: Apache-2.0 OR MIT
+ *
+ */
+
+#include <API/EditorAssetSystemAPI.h>
+#include <AssetBrowser/Thumbnails/ProductThumbnail.h>
+#include <AssetBrowser/Thumbnails/SourceThumbnail.h>
+#include <Atom/RPI.Reflect/Material/MaterialAsset.h>
+#include <Atom/RPI.Reflect/Model/ModelAsset.h>
+#include <Atom/RPI.Reflect/System/AnyAsset.h>
+#include <SharedPreview/SharedPreviewUtils.h>
+
+namespace AZ
+{
+    namespace LyIntegration
+    {
+        namespace SharedPreviewUtils
+        {
+            Data::AssetId GetAssetId(
+                AzToolsFramework::Thumbnailer::SharedThumbnailKey key,
+                const Data::AssetType& assetType,
+                const Data::AssetId& defaultAssetId)
+            {
+                // if it's a source thumbnail key, find first product with a matching asset type
+                auto sourceKey = azrtti_cast<const AzToolsFramework::AssetBrowser::SourceThumbnailKey*>(key.data());
+                if (sourceKey)
+                {
+                    bool foundIt = false;
+                    AZStd::vector<Data::AssetInfo> productsAssetInfo;
+                    AzToolsFramework::AssetSystemRequestBus::BroadcastResult(
+                        foundIt, &AzToolsFramework::AssetSystemRequestBus::Events::GetAssetsProducedBySourceUUID,
+                        sourceKey->GetSourceUuid(), productsAssetInfo);
+                    if (!foundIt)
+                    {
+                        return defaultAssetId;
+                    }
+                    auto assetInfoIt = AZStd::find_if(
+                        productsAssetInfo.begin(), productsAssetInfo.end(),
+                        [&assetType](const Data::AssetInfo& assetInfo)
+                        {
+                            return assetInfo.m_assetType == assetType;
+                        });
+                    if (assetInfoIt == productsAssetInfo.end())
+                    {
+                        return defaultAssetId;
+                    }
+
+                    return assetInfoIt->m_assetId;
+                }
+
+                // if it's a product thumbnail key just return its assetId
+                auto productKey = azrtti_cast<const AzToolsFramework::AssetBrowser::ProductThumbnailKey*>(key.data());
+                if (productKey && productKey->GetAssetType() == assetType)
+                {
+                    return productKey->GetAssetId();
+                }
+                return defaultAssetId;
+            }
+
+            QString WordWrap(const QString& string, int maxLength)
+            {
+                QString result;
+                int length = 0;
+
+                for (const QChar& c : string)
+                {
+                    if (c == '\n')
+                    {
+                        length = 0;
+                    }
+                    else if (length > maxLength)
+                    {
+                        result.append('\n');
+                        length = 0;
+                    }
+                    else
+                    {
+                        length++;
+                    }
+                    result.append(c);
+                }
+                return result;
+            }
+
+            AZStd::unordered_set<AZ::Uuid> GetSupportedAssetTypes()
+            {
+                return { RPI::AnyAsset::RTTI_Type(), RPI::MaterialAsset::RTTI_Type(), RPI::ModelAsset::RTTI_Type() };
+            }
+
+            bool IsSupportedAssetType(AzToolsFramework::Thumbnailer::SharedThumbnailKey key)
+            {
+                for (const AZ::Uuid& typeId : SharedPreviewUtils::GetSupportedAssetTypes())
+                {
+                    const AZ::Data::AssetId& assetId = SharedPreviewUtils::GetAssetId(key, typeId);
+                    if (assetId.IsValid())
+                    {
+                        if (typeId == RPI::AnyAsset::RTTI_Type())
+                        {
+                            AZ::Data::AssetInfo assetInfo;
+                            AZ::Data::AssetCatalogRequestBus::BroadcastResult(
+                                assetInfo, &AZ::Data::AssetCatalogRequestBus::Events::GetAssetInfoById, assetId);
+                            return AzFramework::StringFunc::EndsWith(assetInfo.m_relativePath.c_str(), "lightingpreset.azasset");
+                        }
+                        return true;
+                    }
+                }
+
+                return false;
+            }
+        } // namespace SharedPreviewUtils
+    } // namespace LyIntegration
+} // namespace AZ