You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
o3de/Code/Tools/AssetProcessor/native/ui/ProductAssetTreeModel.h

49 lines
1.7 KiB
C++

/*
* 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
*
*/
#pragma once
#include "AssetTreeModel.h"
namespace AssetProcessor
{
class ProductAssetTreeItemData;
class ProductAssetTreeModel : public AssetTreeModel
{
public:
ProductAssetTreeModel(AZStd::shared_ptr<AzToolsFramework::AssetDatabase::AssetDatabaseConnection> sharedDbConnection, QObject *parent = nullptr);
virtual ~ProductAssetTreeModel();
// AssetDatabaseNotificationBus::Handler
void OnProductFileChanged(const AzToolsFramework::AssetDatabase::ProductDatabaseEntry& entry) override;
void OnProductFileRemoved(AZ::s64 productId) override;
void OnProductFilesRemoved(const AzToolsFramework::AssetDatabase::ProductDatabaseEntryContainer& products) override;
QModelIndex GetIndexForProduct(const AZStd::string& product);
protected:
void ResetModel() override;
void AddOrUpdateEntry(
const AzToolsFramework::AssetDatabase::ProductDatabaseEntry& product,
bool modelIsResetting);
void RemoveAsset(AZ::s64 productId);
void RemoveAssetTreeItem(AssetTreeItem* assetToRemove);
void RemoveFoldersIfEmpty(AssetTreeItem* folderToCheck);
void CheckForUnresolvedIssues(AZStd::shared_ptr<ProductAssetTreeItemData> productItemData);
AZStd::unordered_map<AZStd::string, AssetTreeItem*> m_productToTreeItem;
// Mapping product Ids to asset tree items makes cleanup easier when files are deleted.
AZStd::unordered_map<AZ::s64, AssetTreeItem*> m_productIdToTreeItem;
};
}