/*
* 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 "stdafx.h"

#include "EditorCommon.h"

#define UICANVASEDITOR_HIERARCHY_HEADER_ICON_EYE        ":/Icons/Eye.svg"
#define UICANVASEDITOR_HIERARCHY_HEADER_ICON_PADLOCK    ":/Icons/Padlock.svg"

HierarchyHeader::HierarchyHeader(HierarchyWidget* parent)
    : QHeaderView(Qt::Horizontal, parent)
    , m_hierarchy(parent)
    , m_visibleIcon(UICANVASEDITOR_HIERARCHY_HEADER_ICON_EYE)
    , m_selectableIcon(UICANVASEDITOR_HIERARCHY_HEADER_ICON_PADLOCK)
{
    setMouseTracking(true);
    setSectionsMovable(false);
    setStretchLastSection(false);

    QObject::connect(this,
        &QHeaderView::sectionClicked,
        this,
        [parent](int logicalIndex)
        {
            if (logicalIndex == kHierarchyColumnName)
            {
                // Nothing to do.
                return;
            }

            HierarchyItemRawPtrList items = SelectionHelpers::GetSelectedHierarchyItems(parent, parent->selectedItems());
            if (items.empty())
            {
                // If nothing is selected, then act on all existing items.
                HierarchyHelpers::AppendAllChildrenToEndOfList(parent->invisibleRootItem(), items);

                if (items.empty())
                {
                    // Nothing to do.
                    return;
                }
            }

            if (logicalIndex == kHierarchyColumnIsVisible)
            {
                CommandHierarchyItemToggleIsVisible::Push(parent->GetEditorWindow()->GetActiveStack(),
                    parent,
                    items);
            }
            else if (logicalIndex == kHierarchyColumnIsSelectable)
            {
                CommandHierarchyItemToggleIsSelectable::Push(parent->GetEditorWindow()->GetActiveStack(),
                    parent,
                    items);
            }
            else
            {
                // This should NEVER happen.
                AZ_Assert(0, "Unxpected value for logicalIndex");
            }
        });
}

QSize HierarchyHeader::sizeHint() const
{
    // This controls the height of the header.
    return QSize(UICANVASEDITOR_HIERARCHY_HEADER_ICON_SIZE, UICANVASEDITOR_HIERARCHY_HEADER_ICON_SIZE);
}

void HierarchyHeader::paintSection(QPainter* painter, const QRect& rect, int logicalIndex) const
{
    if (logicalIndex == kHierarchyColumnIsVisible)
    {
        m_visibleIcon.paint(painter, rect);
    }
    else if (logicalIndex == kHierarchyColumnIsSelectable)
    {
        m_selectableIcon.paint(painter, rect);
    }

    // IMPORTANT: We DON'T want to call QHeaderView::paintSection() here.
    // Otherwise it will draw over our icons.
}

void HierarchyHeader::enterEvent(QEvent* ev)
{
    m_hierarchy->ClearItemBeingHovered();

    QHeaderView::enterEvent(ev);
}

#include <HierarchyHeader.moc>