52 lines
1.2 KiB
C++
52 lines
1.2 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "Containers/Map.h"
|
|
#include "HAL/Platform.h"
|
|
#include "MVVM/Extensions/DynamicExtensionContainer.h"
|
|
#include "MVVM/ViewModelTypeID.h"
|
|
#include "Templates/SharedPointer.h"
|
|
#include "Templates/Tuple.h"
|
|
#include "UObject/ObjectKey.h"
|
|
|
|
class UMovieSceneTrack;
|
|
|
|
namespace UE
|
|
{
|
|
namespace Sequencer
|
|
{
|
|
|
|
class FSequenceModel;
|
|
class FTrackRowModel;
|
|
class FViewModel;
|
|
|
|
class SEQUENCER_API FTrackRowModelStorageExtension
|
|
: public IDynamicExtension
|
|
{
|
|
public:
|
|
|
|
UE_SEQUENCER_DECLARE_VIEW_MODEL_TYPE_ID(FTrackRowModelStorageExtension)
|
|
|
|
FTrackRowModelStorageExtension();
|
|
|
|
virtual void OnCreated(TSharedRef<FViewModel> InWeakOwner) override;
|
|
virtual void OnReinitialize() override;
|
|
|
|
TSharedPtr<FTrackRowModel> CreateModelForTrackRow(UMovieSceneTrack* InTrack, int32 InRowIndex, TSharedPtr<FViewModel> DesiredParent = nullptr);
|
|
|
|
TSharedPtr<FTrackRowModel> FindModelForTrackRow(UMovieSceneTrack* InTrack, int32 InRowIndex) const;
|
|
|
|
private:
|
|
|
|
using KeyType = TTuple<FObjectKey, int32>;
|
|
|
|
TMap<KeyType, TWeakPtr<FTrackRowModel>> TrackToModel;
|
|
|
|
FSequenceModel* OwnerModel;
|
|
};
|
|
|
|
} // namespace Sequencer
|
|
} // namespace UE
|
|
|