Steve Pham
38261d0800
* Updated all copyright headers to split the longer original copyright line into 2 shorter lines Signed-off-by: Steve Pham <spham@amazon.com>
47 lines
1.6 KiB
C++
47 lines
1.6 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"
|
|
#include <AzToolsFramework/API/AssetDatabaseBus.h>
|
|
#include <AzCore/std/containers/unordered_map.h>
|
|
#include <native/utilities/ApplicationManagerAPI.h>
|
|
#include <QDir>
|
|
|
|
namespace AssetProcessor
|
|
{
|
|
class SourceAssetTreeModel : public AssetTreeModel
|
|
{
|
|
public:
|
|
SourceAssetTreeModel(AZStd::shared_ptr<AzToolsFramework::AssetDatabase::AssetDatabaseConnection> sharedDbConnection, QObject *parent = nullptr);
|
|
~SourceAssetTreeModel();
|
|
|
|
// AssetDatabaseNotificationBus::Handler
|
|
void OnSourceFileChanged(const AzToolsFramework::AssetDatabase::SourceDatabaseEntry& entry) override;
|
|
void OnSourceFileRemoved(AZ::s64 sourceId) override;
|
|
|
|
QModelIndex GetIndexForSource(const AZStd::string& source);
|
|
|
|
protected:
|
|
void ResetModel() override;
|
|
|
|
void AddOrUpdateEntry(
|
|
const AzToolsFramework::AssetDatabase::SourceDatabaseEntry& source,
|
|
const AzToolsFramework::AssetDatabase::ScanFolderDatabaseEntry& scanFolder,
|
|
bool modelIsResetting);
|
|
|
|
void RemoveAssetTreeItem(AssetTreeItem* assetToRemove);
|
|
void RemoveFoldersIfEmpty(AssetTreeItem* itemToCheck);
|
|
|
|
AZStd::unordered_map<AZStd::string, AssetTreeItem*> m_sourceToTreeItem;
|
|
AZStd::unordered_map<AZ::s64, AssetTreeItem*> m_sourceIdToTreeItem;
|
|
QDir m_assetRoot;
|
|
bool m_assetRootSet = false;
|
|
};
|
|
}
|