Files
UnrealEngine/Engine/Source/Editor/Sequencer/Public/MVVM/ViewModels/TrackModel.h
2025-05-18 13:04:45 +08:00

179 lines
5.4 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "MVVM/ViewModels/ViewModel.h"
#include "MVVM/ViewModels/OutlinerItemModel.h"
#include "MVVM/Extensions/IGeometryExtension.h"
#include "MVVM/Extensions/IGroupableExtension.h"
#include "MVVM/Extensions/IRenameableExtension.h"
#include "MVVM/Extensions/IResizableExtension.h"
#include "MVVM/Extensions/ISortableExtension.h"
#include "MVVM/Extensions/ITrackAreaExtension.h"
#include "MVVM/Extensions/ILockableExtension.h"
#include "MVVM/Extensions/ITrackExtension.h"
#include "MVVM/Extensions/IDeletableExtension.h"
#include "MVVM/Extensions/IDraggableOutlinerExtension.h"
#include "MVVM/Extensions/IDeletableExtension.h"
#include "MVVM/Extensions/IConditionableExtension.h"
#include "MVVM/ViewModels/ViewModelHierarchy.h"
#include "EventHandlers/ISignedObjectEventHandler.h"
#include "MovieSceneSignedObject.h"
#include "UObject/WeakObjectPtr.h"
#include "UObject/WeakObjectPtrTemplates.h"
class ISequencer;
class ISequencerSection;
class ISequencerTrackEditor;
class UMovieSceneTrack;
namespace UE
{
namespace Sequencer
{
class FSectionModel;
class SEQUENCER_API FTrackModel
: public FEvaluableOutlinerItemModel
, public IRenameableExtension
, public IResizableExtension
, public ITrackExtension
, public ITrackAreaExtension
, public ILockableExtension
, public IGroupableExtension
, public ISortableExtension
, public IDraggableOutlinerExtension
, public IDeletableExtension
, public IConditionableExtension
, public UE::MovieScene::TIntrusiveEventHandler<UE::MovieScene::ISignedObjectEventHandler>
, public UE::MovieScene::IDeferredSignedObjectFlushSignal
{
public:
UE_SEQUENCER_DECLARE_CASTABLE(FTrackModel
, FEvaluableOutlinerItemModel
, IRenameableExtension
, IResizableExtension
, ITrackExtension
, ITrackAreaExtension
, ILockableExtension
, IGroupableExtension
, ISortableExtension
, IDraggableOutlinerExtension
, IDeletableExtension
, IConditionableExtension);
explicit FTrackModel(UMovieSceneTrack* Track);
~FTrackModel();
FViewModelChildren GetTopLevelChannels();
public:
static EViewModelListType GetTopLevelChannelType();
static EViewModelListType GetTopLevelChannelGroupType();
/*~ ITrackExtension */
UMovieSceneTrack* GetTrack() const override;
int32 GetRowIndex() const override;
FViewModelChildren GetSectionModels() override;
TSharedPtr<ISequencerTrackEditor> GetTrackEditor() const override;
/*~ ISignedObjectEventHandler */
void OnModifiedDirectly(UMovieSceneSignedObject*) override;
void OnModifiedIndirectly(UMovieSceneSignedObject*) override;
/* IDeferredSignedObjectFlushSignal */
virtual void OnDeferredModifyFlush() override;
/*~ IOutlinerExtension */
FOutlinerSizing GetOutlinerSizing() const override;
TSharedPtr<SWidget> CreateOutlinerViewForColumn(const FCreateOutlinerViewParams& InParams, const FName& InColumnName) override;
FSlateFontInfo GetLabelFont() const override;
const FSlateBrush* GetIconBrush() const override;
FText GetLabel() const override;
FSlateColor GetLabelColor() const override;
FText GetLabelToolTipText() const override;
/*~ IDimmableExtension */
bool IsDimmed() const override;
/*~ IResizableExtension */
bool IsResizable() const override;
void Resize(float NewSize) override;
/*~ ILockableExtension Interface */
ELockableLockState GetLockState() const override;
void SetIsLocked(bool bIsLocked) override;
/*~ IConditionableExtension Interface */
const UMovieSceneCondition* GetCondition() const override;
EConditionableConditionState GetConditionState() const override;
void SetConditionEditorForceTrue(bool bEditorForceTrue) override;
/*~ ITrackAreaExtension */
FTrackAreaParameters GetTrackAreaParameters() const override;
FViewModelVariantIterator GetTrackAreaModelList() const override;
FViewModelVariantIterator GetTopLevelChildTrackAreaModels() const override;
/*~ ICurveEditorTreeItem */
void CreateCurveModels(TArray<TUniquePtr<FCurveModel>>& OutCurveModels) override;
/*~ IGroupableIdentifier */
void GetIdentifierForGrouping(TStringBuilder<128>& OutString) const override;
/*~ IRenameableExtension */
bool CanRename() const override;
void Rename(const FText& NewName) override;
bool IsRenameValidImpl(const FText& NewName, FText& OutErrorMessage) const override;
/*~ ISortableExtension */
void SortChildren() override;
FSortingKey GetSortingKey() const override;
void SetCustomOrder(int32 InCustomOrder) override;
/*~ IDraggableOutlinerExtension */
bool CanDrag() const override;
/*~ FOutlinerItemModel */
bool HasCurves() const override;
void BuildContextMenu(FMenuBuilder& MenuBuilder) override;
bool GetDefaultExpansionState() const override;
void BuildSidebarMenu(FMenuBuilder& MenuBuilder) override;
/*~ IDeletableExtension */
bool CanDelete(FText* OutErrorMessage) const override;
void Delete() override;
/*~ FViewModel interface */
virtual void OnConstruct() override;
private:
void ForceUpdate();
bool FindBoundObjects(TArray<UObject*>& OutBoundObjects) const;
/** A second children list for the sections inside this track */
FViewModelListHead SectionList;
FViewModelListHead TopLevelChannelList;
/** The actual track wrapped by this data model */
TWeakObjectPtr<UMovieSceneTrack> WeakTrack;
// @todo_sequencer_mvvm: move all the track editor behavior into the view model
TSharedPtr<ISequencerTrackEditor> TrackEditor;
bool bNeedsUpdate;
int32 PreviousLayoutNumRows = -1;
};
} // namespace Sequencer
} // namespace UE