/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
#include "LyShine_precompiled.h"
#include "UiLayoutManager.h"

#include <LyShine/Bus/UiLayoutBus.h>
#include <LyShine/Bus/UiElementBus.h>
#include <LyShine/Bus/UiLayoutControllerBus.h>

////////////////////////////////////////////////////////////////////////////////////////////////////
// PUBLIC MEMBER FUNCTIONS
////////////////////////////////////////////////////////////////////////////////////////////////////

////////////////////////////////////////////////////////////////////////////////////////////////////
UiLayoutManager::UiLayoutManager(AZ::EntityId canvasEntityId)
{
    UiLayoutManagerBus::Handler::BusConnect(canvasEntityId);
}

////////////////////////////////////////////////////////////////////////////////////////////////////
UiLayoutManager::~UiLayoutManager()
{
    UiLayoutManagerBus::Handler::BusDisconnect();
}

////////////////////////////////////////////////////////////////////////////////////////////////////
void UiLayoutManager::MarkToRecomputeLayout(AZ::EntityId entityId)
{
    if (UiLayoutControllerBus::FindFirstHandler(entityId))
    {
        AddToRecomputeLayoutList(entityId);
    }
}

////////////////////////////////////////////////////////////////////////////////////////////////////
void UiLayoutManager::MarkToRecomputeLayoutsAffectedByLayoutCellChange(AZ::EntityId entityId, bool isDefaultLayoutCell)
{
    AZ::EntityId topParent;
    AZ::EntityId parent;
    EBUS_EVENT_ID_RESULT(parent, entityId, UiElementBus, GetParentEntityId);
    while (parent.IsValid())
    {
        bool usesLayoutCells = false;
        EBUS_EVENT_ID_RESULT(usesLayoutCells, parent, UiLayoutBus, IsUsingLayoutCellsToCalculateLayout);
        if (usesLayoutCells && isDefaultLayoutCell)
        {
            bool ignoreDefaultLayoutCells = true;
            EBUS_EVENT_ID_RESULT(ignoreDefaultLayoutCells, parent, UiLayoutBus, GetIgnoreDefaultLayoutCells);
            usesLayoutCells = !ignoreDefaultLayoutCells;
        }

        if (usesLayoutCells)
        {
            topParent = parent;
            parent.SetInvalid();
            EBUS_EVENT_ID_RESULT(parent, topParent, UiElementBus, GetParentEntityId);
        }
        else
        {
            break;
        }
    }

    if (topParent.IsValid())
    {
        AddToRecomputeLayoutList(topParent);
    }
}

////////////////////////////////////////////////////////////////////////////////////////////////////
void UiLayoutManager::UnmarkAllLayouts()
{
    m_elementsToRecomputeLayout.clear();
}

////////////////////////////////////////////////////////////////////////////////////////////////////
void UiLayoutManager::RecomputeMarkedLayouts()
{
    for (auto element : m_elementsToRecomputeLayout)
    {
        ComputeLayoutForElementAndDescendants(element);
    }

    UnmarkAllLayouts();
}

////////////////////////////////////////////////////////////////////////////////////////////////////
void UiLayoutManager::ComputeLayoutForElementAndDescendants(AZ::EntityId entityId)
{
    // Get a list of layout children
    auto FindLayoutChildren = [](const AZ::Entity* entity)
        {
            return (UiLayoutControllerBus::FindFirstHandler(entity->GetId())) ? true : false;
        };

    LyShine::EntityArray layoutChildren;
    EBUS_EVENT_ID(entityId, UiElementBus, FindDescendantElements, FindLayoutChildren, layoutChildren);

    EBUS_EVENT_ID(entityId, UiLayoutControllerBus, ApplyLayoutWidth);
    for (auto layoutChild : layoutChildren)
    {
        EBUS_EVENT_ID(layoutChild->GetId(), UiLayoutControllerBus, ApplyLayoutWidth);
    }

    EBUS_EVENT_ID(entityId, UiLayoutControllerBus, ApplyLayoutHeight);
    for (auto layoutChild : layoutChildren)
    {
        EBUS_EVENT_ID(layoutChild->GetId(), UiLayoutControllerBus, ApplyLayoutHeight);
    }
}

////////////////////////////////////////////////////////////////////////////////////////////////////
// PRIVATE MEMBER FUNCTIONS
////////////////////////////////////////////////////////////////////////////////////////////////////

////////////////////////////////////////////////////////////////////////////////////////////////////
void UiLayoutManager::AddToRecomputeLayoutList(AZ::EntityId entityId)
{
    // Check if element is already in the list
    auto iter = AZStd::find(m_elementsToRecomputeLayout.begin(), m_elementsToRecomputeLayout.end(), entityId);
    if (iter == m_elementsToRecomputeLayout.end())
    {
        // Check if element's parent is already in the list
        for (iter = m_elementsToRecomputeLayout.begin(); iter != m_elementsToRecomputeLayout.end(); ++iter)
        {
            if (IsParentOfElement(*iter, entityId))
            {
                // Don't need to add this element
                return;
            }
        }

        // Remove element's children from the list
        LyShine::EntityArray descendants;
        EBUS_EVENT_ID(entityId, UiElementBus, FindDescendantElements,
            [](const AZ::Entity* entity) { return true; },
            descendants);

        m_elementsToRecomputeLayout.remove_if(
            [descendants](const AZ::EntityId& e)
            {
                for (auto descendant : descendants)
                {
                    if (descendant->GetId() == e)
                    {
                        return true;
                    }
                }
                return false;
            }
            );

        // Add element to list
        m_elementsToRecomputeLayout.push_back(entityId);
    }
}

////////////////////////////////////////////////////////////////////////////////////////////////////
bool UiLayoutManager::IsParentOfElement(AZ::EntityId checkParentEntity, AZ::EntityId checkChildEntity)
{
    AZ::EntityId parent;
    EBUS_EVENT_ID_RESULT(parent, checkChildEntity, UiElementBus, GetParentEntityId);
    while (parent.IsValid())
    {
        if (parent == checkParentEntity)
        {
            return true;
        }

        AZ::EntityId newParent = parent;
        parent.SetInvalid();
        EBUS_EVENT_ID_RESULT(parent, newParent, UiElementBus, GetParentEntityId);
    }

    return false;
}