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.
53 lines
2.0 KiB
C++
53 lines
2.0 KiB
C++
/*
|
|
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
|
|
* its licensors.
|
|
*
|
|
* For complete copyright and license terms please see the LICENSE at the root of this
|
|
* distribution (the "License"). All use of this software is governed by the License,
|
|
* or, if provided, by the license below or the license accompanying this file. Do not
|
|
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
*
|
|
*/
|
|
#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;
|
|
};
|
|
}
|