#include "LogVisualizer.h"
#include "STimeline.h"
#include "STimelineBar.h"
#include "TimeSliderController.h"
#include "STimelinesContainer.h"

FReply STimeline::OnMouseButtonDown(const FGeometry& MyGeometry, const FPointerEvent& MouseEvent)
{
	Owner->ChangeSelection(SharedThis(this), MouseEvent);
	return FReply::Unhandled();
}

FReply STimeline::OnMouseButtonUp(const FGeometry& MyGeometry, const FPointerEvent& MouseEvent)
{
	return FReply::Unhandled();
}

bool STimeline::IsSelected() const
{
	// Ask the tree if we are selected
	return Owner->IsNodeSelected(SharedThis(const_cast<STimeline*>(this)));
}

void STimeline::OnSelect()
{
	if (TimelineBar.IsValid())
	{
		TimelineBar->OnSelect();
	}
}

void STimeline::OnDeselect()
{
	if (TimelineBar.IsValid())
	{
		TimelineBar->OnDeselect();
	}
}

void STimeline::UpdateVisibility()
{
	SetVisibility(HiddenEntries.Num() == Entries.Num() || (SearchFilter.Len() > 0 && Name.ToString().Find(SearchFilter) == INDEX_NONE) ? EVisibility::Collapsed : EVisibility::Visible);
	if (GetVisibility() != EVisibility::Visible)
	{
		Owner->SetSelectionState(SharedThis(this), false, false);
	}
}

void STimeline::OnFiltersChanged()
{
	HiddenEntries.Reset();

	for (auto& CurrentEntry : Entries)
	{
		TArray<FVisualLoggerCategoryVerbosityPair> OutCategories;
		FVisualLoggerHelpers::GetCategories(CurrentEntry.Entry, OutCategories);
		if (VisualLoggerInterface->HasValidCategories(OutCategories) == false)
		{
			HiddenEntries.AddUnique(&CurrentEntry);
		}
	}

	UpdateVisibility();
}

void STimeline::OnSearchChanged(const FText& Filter)
{
	SearchFilter = Filter.ToString();
	UpdateVisibility();
}

bool STimeline::IsEntryHidden(const FVisualLogDevice::FVisualLogEntryItem& EntryItem) const
{
	return HiddenEntries.Contains(&EntryItem);
}

void STimeline::Construct(const FArguments& InArgs, TSharedPtr<SVisualLoggerView> VisualLoggerView, TSharedPtr<FSequencerTimeSliderController> TimeSliderController, TSharedPtr<STimelinesContainer> InContainer, const FVisualLogDevice::FVisualLogEntryItem& Entry)
{
	VisualLoggerInterface = InArgs._VisualLoggerInterface.Get();
	Owner = InContainer;
	Name = Entry.OwnerName;

	Entries.Add(Entry);
	OnFiltersChanged();

	ChildSlot
		[
			SNew(SHorizontalBox)
			+ SHorizontalBox::Slot()
			.Padding(FMargin(0, 4, 0, 0))
			.HAlign(HAlign_Fill)
			.VAlign(VAlign_Fill)
			.FillWidth(TAttribute<float>(VisualLoggerView.Get(), &SVisualLoggerView::GetAnimationOutlinerFillPercentage))
			[
				SNew(SBorder)
				.HAlign(HAlign_Fill)
				.Padding(FMargin(0, 0, 4, 0))
				.BorderImage(FCoreStyle::Get().GetBrush("NoBorder"))
				[
					SNew(SBorder)
					.VAlign(VAlign_Center)
					.BorderImage(this, &STimeline::GetBorder)
					.Padding(FMargin(4, 0, 2, 0))
					[
						// Search box for searching through the outliner
						SNew(STextBlock)
						.Text(FString(Name.ToString()))
						.ShadowOffset(FVector2D(1.f, 1.f))
					]
				]
			]
			+ SHorizontalBox::Slot()
			.Padding(FMargin(0, 4, 0, 0))
			.HAlign(HAlign_Left)
			[
				SNew(SBox)
				.Padding(FMargin(0, 0, 0, 0))
				.HAlign(HAlign_Left)
				[
					// Search box for searching through the outliner
					SAssignNew(TimelineBar, STimelineBar, TimeSliderController, SharedThis(this))
					.VisualLoggerInterface(InArgs._VisualLoggerInterface)
				]
			]
		];
}

const FSlateBrush* STimeline::GetBorder() const
{
	if (IsSelected())
	{
		return FLogVisualizerStyle::Get().GetBrush("ToolBar.Button.Hovered");
	}
	else
	{
		return FLogVisualizerStyle::Get().GetBrush("ToolBar.Button.Normal");
	}
}