EM_Task/TraceInsights/Private/Insights/Table/Widgets/STableTreeViewRow.cpp
Boshuang Zhao 5144a49c9b add
2026-02-13 16:18:33 +08:00

202 lines
7.0 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#include "STableTreeViewRow.h"
#include "EditorStyleSet.h"
#include "SlateOptMacros.h"
#include "Widgets/SOverlay.h"
#include "Widgets/Images/SImage.h"
// Insights
#include "Insights/Common/TimeUtils.h"
#include "Insights/Table/ViewModels/Table.h"
#include "Insights/Table/ViewModels/TableColumn.h"
#include "Insights/Table/Widgets/STableTreeViewCell.h"
#include "Insights/Table/Widgets/STableTreeViewTooltip.h"
#define LOCTEXT_NAMESPACE "STableTreeView"
namespace Insights
{
////////////////////////////////////////////////////////////////////////////////////////////////////
BEGIN_SLATE_FUNCTION_BUILD_OPTIMIZATION
void STableTreeViewRow::Construct(const FArguments& InArgs, const TSharedRef<STableViewBase>& InOwnerTableView)
{
OnShouldBeEnabled = InArgs._OnShouldBeEnabled;
IsColumnVisibleDelegate = InArgs._OnIsColumnVisible;
GetColumnOutlineHAlignmentDelegate = InArgs._OnGetColumnOutlineHAlignmentDelegate;
SetHoveredCellDelegate = InArgs._OnSetHoveredCell;
HighlightText = InArgs._HighlightText;
HighlightedNodeName = InArgs._HighlightedNodeName;
TablePtr = InArgs._TablePtr;
TableTreeNodePtr = InArgs._TableTreeNodePtr;
RowToolTip = MakeShared<STableTreeRowToolTip>(TableTreeNodePtr);
SetEnabled(TAttribute<bool>(this, &STableTreeViewRow::HandleShouldBeEnabled));
SMultiColumnTableRow<FTableTreeNodePtr>::Construct(SMultiColumnTableRow<FTableTreeNodePtr>::FArguments(), InOwnerTableView);
}
////////////////////////////////////////////////////////////////////////////////////////////////////
TSharedRef<SWidget> STableTreeViewRow::GenerateWidgetForColumn(const FName& ColumnId)
{
TSharedPtr<FTableColumn> ColumnPtr = TablePtr->FindColumnChecked(ColumnId);
return SNew(SOverlay)
.Visibility(EVisibility::SelfHitTestInvisible)
+ SOverlay::Slot()
.Padding(0.0f)
[SNew(SImage)
.Image(FEditorStyle::GetBrush("Profiler.LineGraphArea"))
.ColorAndOpacity(this, &STableTreeViewRow::GetBackgroundColorAndOpacity)]
+ SOverlay::Slot()
.Padding(0.0f)
[SNew(SImage)
.Image(this, &STableTreeViewRow::GetOutlineBrush, ColumnId)
.ColorAndOpacity(this, &STableTreeViewRow::GetOutlineColorAndOpacity)]
+ SOverlay::Slot()
[SNew(STableTreeViewCell, SharedThis(this))
.Visibility(this, &STableTreeViewRow::IsColumnVisible, ColumnId)
.TablePtr(TablePtr)
.ColumnPtr(ColumnPtr)
.TableTreeNodePtr(TableTreeNodePtr)
.HighlightText(HighlightText)
.IsNameColumn(ColumnPtr->IsHierarchy())
.OnSetHoveredCell(this, &STableTreeViewRow::OnSetHoveredCell)];
}
END_SLATE_FUNCTION_BUILD_OPTIMIZATION
////////////////////////////////////////////////////////////////////////////////////////////////////
FReply STableTreeViewRow::OnDragDetected(const FGeometry& MyGeometry, const FPointerEvent& MouseEvent)
{
return SMultiColumnTableRow<FTableTreeNodePtr>::OnDragDetected(MyGeometry, MouseEvent);
}
////////////////////////////////////////////////////////////////////////////////////////////////////
TSharedRef<IToolTip> STableTreeViewRow::GetRowToolTip() const
{
return RowToolTip.ToSharedRef();
}
////////////////////////////////////////////////////////////////////////////////////////////////////
void STableTreeViewRow::InvalidateContent()
{
RowToolTip->InvalidateWidget();
}
////////////////////////////////////////////////////////////////////////////////////////////////////
FSlateColor STableTreeViewRow::GetBackgroundColorAndOpacity() const
{
return FLinearColor(0.0f, 0.0f, 0.0f, 1.0f);
}
////////////////////////////////////////////////////////////////////////////////////////////////////
FSlateColor STableTreeViewRow::GetBackgroundColorAndOpacity(double Time) const
{
const FLinearColor Color = Time > TimeUtils::Second ? FLinearColor(0.3f, 0.0f, 0.0f, 1.0f) :
Time > TimeUtils::Milisecond ? FLinearColor(0.3f, 0.1f, 0.0f, 1.0f) :
Time > TimeUtils::Microsecond ? FLinearColor(0.0f, 0.1f, 0.0f, 1.0f) :
FLinearColor(0.0f, 0.0f, 0.0f, 1.0f);
return Color;
}
////////////////////////////////////////////////////////////////////////////////////////////////////
FSlateColor STableTreeViewRow::GetOutlineColorAndOpacity() const
{
const FLinearColor NoColor(0.0f, 0.0f, 0.0f, 0.0f);
const bool bShouldBeHighlighted = TableTreeNodePtr->GetName() == HighlightedNodeName.Get();
const FLinearColor OutlineColorAndOpacity = bShouldBeHighlighted ? FLinearColor(FColorList::SlateBlue) : NoColor;
return OutlineColorAndOpacity;
}
////////////////////////////////////////////////////////////////////////////////////////////////////
const FSlateBrush* STableTreeViewRow::GetOutlineBrush(const FName ColumnId) const
{
EHorizontalAlignment Result = HAlign_Center;
if (IsColumnVisibleDelegate.IsBound())
{
Result = GetColumnOutlineHAlignmentDelegate.Execute(ColumnId);
}
const FSlateBrush* Brush = nullptr;
if (Result == HAlign_Left)
{
Brush = FEditorStyle::GetBrush("Profiler.EventGraph.Border.L");
}
else if (Result == HAlign_Right)
{
Brush = FEditorStyle::GetBrush("Profiler.EventGraph.Border.R");
}
else
{
Brush = FEditorStyle::GetBrush("Profiler.EventGraph.Border.TB");
}
return Brush;
}
////////////////////////////////////////////////////////////////////////////////////////////////////
bool STableTreeViewRow::HandleShouldBeEnabled() const
{
bool bResult = false;
if (TableTreeNodePtr->IsGroup())
{
bResult = true;
}
else
{
if (OnShouldBeEnabled.IsBound())
{
bResult = OnShouldBeEnabled.Execute(TableTreeNodePtr);
}
}
return bResult;
}
////////////////////////////////////////////////////////////////////////////////////////////////////
EVisibility STableTreeViewRow::IsColumnVisible(const FName ColumnId) const
{
EVisibility Result = EVisibility::Collapsed;
if (IsColumnVisibleDelegate.IsBound())
{
Result = IsColumnVisibleDelegate.Execute(ColumnId) ? EVisibility::Visible : EVisibility::Collapsed;
}
return Result;
}
////////////////////////////////////////////////////////////////////////////////////////////////////
void STableTreeViewRow::OnSetHoveredCell(TSharedPtr<FTable> InTablePtr, TSharedPtr<FTableColumn> InColumnPtr, FTableTreeNodePtr InTreeNodePtr)
{
SetHoveredCellDelegate.ExecuteIfBound(InTablePtr, InColumnPtr, InTreeNodePtr);
}
////////////////////////////////////////////////////////////////////////////////////////////////////
} // namespace Insights
#undef LOCTEXT_NAMESPACE