52 lines
1.2 KiB
C++
52 lines
1.2 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "UObject/ObjectKey.h"
|
|
#include "MVVM/ViewModels/ViewModel.h"
|
|
#include "MVVM/Extensions/DynamicExtensionContainer.h"
|
|
#include "EventHandlers/ISequenceDataEventHandler.h"
|
|
|
|
class UMovieSceneFolder;
|
|
|
|
namespace UE
|
|
{
|
|
namespace Sequencer
|
|
{
|
|
|
|
class FFolderModel;
|
|
class FSequenceModel;
|
|
|
|
class FFolderModelStorageExtension
|
|
: public IDynamicExtension
|
|
, private UE::MovieScene::TIntrusiveEventHandler<UE::MovieScene::ISequenceDataEventHandler>
|
|
{
|
|
public:
|
|
|
|
UE_SEQUENCER_DECLARE_VIEW_MODEL_TYPE_ID(FFolderModelStorageExtension)
|
|
|
|
FFolderModelStorageExtension();
|
|
|
|
virtual void OnCreated(TSharedRef<FViewModel> InWeakOwner) override;
|
|
virtual void OnReinitialize() override;
|
|
|
|
TSharedPtr<FFolderModel> CreateModelForFolder(UMovieSceneFolder* InFolder, TSharedPtr<FViewModel> DesiredParent = nullptr);
|
|
|
|
TSharedPtr<FFolderModel> FindModelForFolder(UMovieSceneFolder* InFolder) const;
|
|
|
|
private:
|
|
|
|
void OnRootFolderAdded(UMovieSceneFolder*) override;
|
|
void OnRootFolderRemoved(UMovieSceneFolder*) override;
|
|
|
|
private:
|
|
|
|
TMap<FObjectKey, TWeakPtr<FFolderModel>> FolderToModel;
|
|
|
|
FSequenceModel* OwnerModel;
|
|
};
|
|
|
|
} // namespace Sequencer
|
|
} // namespace UE
|
|
|