// 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