#include "logvisor/logvisor.hpp"
#include "specter/SplitView.hpp"
#include "specter/RootView.hpp"
#include "specter/ViewResources.hpp"
#include "specter/Space.hpp"

namespace specter
{
static logvisor::Module Log("specter::SplitView");

static const zeus::RGBA32 Tex[3] =
{
    {{0,0,0,64}},
    {{0,0,0,255}},
    {{255,255,255,64}}
};

void SplitView::Resources::init(boo::IGraphicsDataFactory::Context& ctx, const IThemeData& theme)
{
    m_shadingTex = ctx.newStaticTexture(3, 1, 1, boo::TextureFormat::RGBA8, boo::TextureClampMode::Repeat, Tex, 12);
}

SplitView::SplitView(ViewResources& res, View& parentView, ISplitSpaceController* controller,
                     Axis axis, float split, int clearanceA, int clearanceB)
: View(res, parentView), m_controller(controller), m_axis(axis), m_slide(split),
  m_clearanceA(clearanceA), m_clearanceB(clearanceB)
{
    commitResources(res, [&](boo::IGraphicsDataFactory::Context& ctx) -> bool
    {
        buildResources(ctx, res);
        m_splitBlockBuf = res.m_viewRes.m_bufPool.allocateBlock(res.m_factory);
        m_splitVertsBinding.init(ctx, res, 4, m_splitBlockBuf, res.m_splitRes.m_shadingTex.get());
        return true;
    });
}

View* SplitView::setContentView(int slot, View* view)
{
    if (slot < 0 || slot > 1)
        Log.report(logvisor::Fatal, "out-of-range slot to RootView::SplitView::setContentView");
    View* ret = m_views[slot].m_view;
    m_views[slot].m_view = view;
    m_views[slot].m_mouseDown = 0;
    m_views[slot].m_mouseIn = false;
    updateSize();
    return ret;
}

void SplitView::_setSplit(float slide)
{
    m_slide = std::min(std::max(slide, 0.0f), 1.0f);
    const boo::SWindowRect& rect = subRect();
    if (rect.size[0] && rect.size[1] &&
        (m_clearanceA >= 0 || m_clearanceB >= 0))
    {
        if (m_axis == Axis::Horizontal)
        {
            int slidePx = rect.size[1] * m_slide;
            if (m_clearanceA >= 0 && slidePx < m_clearanceA)
                m_slide = m_clearanceA / float(rect.size[1]);
            if (m_clearanceB >= 0 && (rect.size[1] - slidePx) < m_clearanceB)
                m_slide = 1.0 - m_clearanceB / float(rect.size[1]);
        }
        else if (m_axis == Axis::Vertical)
        {
            int slidePx = rect.size[0] * m_slide;
            if (m_clearanceA >= 0 && slidePx < m_clearanceA)
                m_slide = m_clearanceA / float(rect.size[0]);
            if (m_clearanceB >= 0 && (rect.size[0] - slidePx) < m_clearanceB)
                m_slide = 1.0 - m_clearanceB / float(rect.size[0]);
        }
        m_slide = std::min(std::max(m_slide, 0.0f), 1.0f);
    }
    if (m_controller)
        m_controller->updateSplit(m_slide);
}

void SplitView::setSplit(float slide)
{
    _setSplit(slide);
    updateSize();
}

void SplitView::setAxis(Axis axis)
{
    m_axis = axis;
    setSplit(m_slide);
}

bool SplitView::testSplitHover(const boo::SWindowCoord& coord)
{
    if (m_axis == Axis::Horizontal)
    {
        int slidePx = subRect().size[1] * m_slide;
        if (abs(int(coord.pixel[1] - subRect().location[1]) - slidePx) < 4)
            return true;
    }
    else if (m_axis == Axis::Vertical)
    {
        int slidePx = subRect().size[0] * m_slide;
        if (abs(int(coord.pixel[0] - subRect().location[0]) - slidePx) < 4)
            return true;
    }
    return false;
}

bool SplitView::testJoinArrowHover(const boo::SWindowCoord& coord, int& origSlotOut,
                                   SplitView*& splitOut, int& slotOut,
                                   boo::SWindowRect& rectOut, ArrowDir& dirOut, int forceSlot)
{
    if (!subRect().coordInRect(coord))
        return false;

    int origDummy;
    ArrowDir dirDummy;

    if (m_axis == Axis::Horizontal)
    {
        int slidePx = subRect().size[1] * m_slide;
        if ((forceSlot == -1 && coord.pixel[1] - subRect().location[1] - slidePx >= 0) || forceSlot == 1)
        {
            origSlotOut = 0;
            dirOut = ArrowDir::Up;
            if (m_views[1].m_view)
            {
                SplitView* chSplit = m_views[1].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Horizontal)
                    return chSplit->testJoinArrowHover(coord, origDummy, splitOut, slotOut, rectOut, dirDummy, 0);
            }
            splitOut = this;
            slotOut = 1;
            rectOut = subRect();
            rectOut.location[1] += slidePx;
            rectOut.size[1] -= slidePx;
            return true;
        }
        else
        {
            origSlotOut = 1;
            dirOut = ArrowDir::Down;
            if (m_views[0].m_view)
            {
                SplitView* chSplit = m_views[0].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Horizontal)
                    return chSplit->testJoinArrowHover(coord, origDummy, splitOut, slotOut, rectOut, dirDummy, 1);
            }
            splitOut = this;
            slotOut = 0;
            rectOut = subRect();
            rectOut.size[1] = slidePx;
            return true;
        }
    }
    else
    {
        int slidePx = subRect().size[0] * m_slide;
        if ((forceSlot == -1 && coord.pixel[0] - subRect().location[0] - slidePx >= 0) || forceSlot == 1)
        {
            origSlotOut = 0;
            dirOut = ArrowDir::Right;
            if (m_views[1].m_view)
            {
                SplitView* chSplit = m_views[1].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Vertical)
                    return chSplit->testJoinArrowHover(coord, origDummy, splitOut, slotOut, rectOut, dirDummy, 0);
            }
            splitOut = this;
            slotOut = 1;
            rectOut = subRect();
            rectOut.location[0] += slidePx;
            rectOut.size[0] -= slidePx;
            return true;
        }
        else
        {
            origSlotOut = 1;
            dirOut = ArrowDir::Left;
            if (m_views[0].m_view)
            {
                SplitView* chSplit = m_views[0].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Vertical)
                    return chSplit->testJoinArrowHover(coord, origDummy, splitOut, slotOut, rectOut, dirDummy, 1);
            }
            splitOut = this;
            slotOut = 0;
            rectOut = subRect();
            rectOut.size[0] = slidePx;
            return true;
        }
    }
}

void SplitView::getJoinArrowHover(int slot, boo::SWindowRect& rectOut, ArrowDir& dirOut)
{
    if (m_axis == Axis::Horizontal)
    {
        int slidePx = subRect().size[1] * m_slide;
        if (slot == 1)
        {
            if (m_views[1].m_view)
            {
                SplitView* chSplit = m_views[1].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Horizontal)
                    return chSplit->getJoinArrowHover(0, rectOut, dirOut);
            }
            rectOut = subRect();
            rectOut.location[1] += slidePx;
            rectOut.size[1] -= slidePx;
            dirOut = ArrowDir::Up;
        }
        else
        {
            if (m_views[0].m_view)
            {
                SplitView* chSplit = m_views[0].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Horizontal)
                    return chSplit->getJoinArrowHover(1, rectOut, dirOut);
            }
            rectOut = subRect();
            rectOut.size[1] = slidePx;
            dirOut = ArrowDir::Down;
        }
    }
    else
    {
        int slidePx = subRect().size[0] * m_slide;
        if (slot == 1)
        {
            if (m_views[1].m_view)
            {
                SplitView* chSplit = m_views[1].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Vertical)
                    return chSplit->getJoinArrowHover(0, rectOut, dirOut);
            }
            rectOut = subRect();
            rectOut.location[0] += slidePx;
            rectOut.size[0] -= slidePx;
            dirOut = ArrowDir::Right;
        }
        else
        {
            if (m_views[0].m_view)
            {
                SplitView* chSplit = m_views[0].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Vertical)
                    return chSplit->getJoinArrowHover(1, rectOut, dirOut);
            }
            rectOut = subRect();
            rectOut.size[0] = slidePx;
            dirOut = ArrowDir::Left;
        }
    }
}

bool SplitView::testSplitLineHover(const boo::SWindowCoord& coord, int& slotOut,
                                   boo::SWindowRect& rectOut, float& splitOut, Axis& axisOut)
{
    if (!subRect().coordInRect(coord))
        return false;

    if (m_axis == Axis::Horizontal)
    {
        int slidePx = subRect().size[1] * m_slide;
        if (coord.pixel[1] - subRect().location[1] - slidePx >= 0)
        {
            if (m_views[1].m_view)
            {
                SplitView* chSplit = m_views[1].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Horizontal)
                    return chSplit->testSplitLineHover(coord, slotOut, rectOut, splitOut, axisOut);
            }
            slotOut = 1;
            rectOut = subRect();
            rectOut.location[1] += slidePx;
            rectOut.size[1] -= slidePx;
            splitOut = (coord.pixel[0] - rectOut.location[0]) / float(rectOut.size[0]);
            axisOut = Axis::Vertical;
            return true;
        }
        else
        {
            if (m_views[0].m_view)
            {
                SplitView* chSplit = m_views[0].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Horizontal)
                    return chSplit->testSplitLineHover(coord, slotOut, rectOut, splitOut, axisOut);
            }
            slotOut = 0;
            rectOut = subRect();
            rectOut.size[1] = slidePx;
            splitOut = (coord.pixel[0] - rectOut.location[0]) / float(rectOut.size[0]);
            axisOut = Axis::Vertical;
            return true;
        }
    }
    else
    {
        int slidePx = subRect().size[0] * m_slide;
        if (coord.pixel[0] - subRect().location[0] - slidePx >= 0)
        {
            if (m_views[1].m_view)
            {
                SplitView* chSplit = m_views[1].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Vertical)
                    return chSplit->testSplitLineHover(coord, slotOut, rectOut, splitOut, axisOut);
            }
            slotOut = 1;
            rectOut = subRect();
            rectOut.location[0] += slidePx;
            rectOut.size[0] -= slidePx;
            splitOut = (coord.pixel[1] - rectOut.location[1]) / float(rectOut.size[1]);
            axisOut = Axis::Horizontal;
            return true;
        }
        else
        {
            if (m_views[0].m_view)
            {
                SplitView* chSplit = m_views[0].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Vertical)
                    return chSplit->testSplitLineHover(coord, slotOut, rectOut, splitOut, axisOut);
            }
            slotOut = 0;
            rectOut = subRect();
            rectOut.size[0] = slidePx;
            splitOut = (coord.pixel[1] - rectOut.location[1]) / float(rectOut.size[1]);
            axisOut = Axis::Horizontal;
            return true;
        }
    }
}

void SplitView::getSplitLineHover(int slot, boo::SWindowRect& rectOut, Axis& axisOut)
{
    if (m_axis == Axis::Horizontal)
    {
        int slidePx = subRect().size[1] * m_slide;
        if (slot == 1)
        {
            if (m_views[1].m_view)
            {
                SplitView* chSplit = m_views[1].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Horizontal)
                    return chSplit->getSplitLineHover(0, rectOut, axisOut);
            }
            rectOut = subRect();
            rectOut.location[1] += slidePx;
            rectOut.size[1] -= slidePx;
            axisOut = Axis::Vertical;
        }
        else
        {
            if (m_views[0].m_view)
            {
                SplitView* chSplit = m_views[0].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Horizontal)
                    return chSplit->getSplitLineHover(1, rectOut, axisOut);
            }
            rectOut = subRect();
            rectOut.size[1] = slidePx;
            axisOut = Axis::Vertical;
        }
    }
    else
    {
        int slidePx = subRect().size[0] * m_slide;
        if (slot == 1)
        {
            if (m_views[1].m_view)
            {
                SplitView* chSplit = m_views[1].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Vertical)
                    return chSplit->getSplitLineHover(0, rectOut, axisOut);
            }
            rectOut = subRect();
            rectOut.location[0] += slidePx;
            rectOut.size[0] -= slidePx;
            axisOut = Axis::Horizontal;
        }
        else
        {
            if (m_views[0].m_view)
            {
                SplitView* chSplit = m_views[0].m_view->castToSplitView();
                if (chSplit && chSplit->m_axis == Axis::Vertical)
                    return chSplit->getSplitLineHover(1, rectOut, axisOut);
            }
            rectOut = subRect();
            rectOut.size[0] = slidePx;
            axisOut = Axis::Horizontal;
        }
    }
}

void SplitView::startDragSplit(const boo::SWindowCoord& coord)
{
    m_dragging = true;
    if (m_axis == Axis::Horizontal)
        setSplit((coord.pixel[1] - subRect().location[1]) / float(subRect().size[1]));
    else if (m_axis == Axis::Vertical)
        setSplit((coord.pixel[0] - subRect().location[0]) / float(subRect().size[0]));
}

void SplitView::endDragSplit()
{
    m_dragging = false;
}

void SplitView::moveDragSplit(const boo::SWindowCoord& coord)
{
    if (m_axis == Axis::Horizontal)
        setSplit((coord.pixel[1] - subRect().location[1]) / float(subRect().size[1]));
    else if (m_axis == Axis::Vertical)
        setSplit((coord.pixel[0] - subRect().location[0]) / float(subRect().size[0]));
}

void SplitView::mouseDown(const boo::SWindowCoord& coord, boo::EMouseButton button, boo::EModifierKey mod)
{
    m_views[0].mouseDown(coord, button, mod);
    m_views[1].mouseDown(coord, button, mod);
}

void SplitView::mouseUp(const boo::SWindowCoord& coord, boo::EMouseButton button, boo::EModifierKey mod)
{
    m_views[0].mouseUp(coord, button, mod);
    m_views[1].mouseUp(coord, button, mod);
}

void SplitView::mouseMove(const boo::SWindowCoord& coord)
{
    m_views[0].mouseMove(coord);
    m_views[1].mouseMove(coord);
}

void SplitView::mouseEnter(const boo::SWindowCoord& coord)
{
    m_views[0].mouseEnter(coord);
    m_views[1].mouseEnter(coord);
}

void SplitView::mouseLeave(const boo::SWindowCoord& coord)
{
    m_views[0].mouseLeave(coord);
    m_views[1].mouseLeave(coord);
}

void SplitView::resized(const boo::SWindowRect& root, const boo::SWindowRect& sub)
{
    View::resized(root, sub);
    _setSplit(m_slide);
    if (m_axis == Axis::Horizontal)
    {
        boo::SWindowRect ssub = sub;
        ssub.size[1] *= m_slide;
        if (m_views[0].m_view)
            m_views[0].m_view->resized(root, ssub);
        ssub.location[1] += ssub.size[1];
        ssub.size[1] = sub.size[1] - ssub.size[1];
        if (m_views[1].m_view)
            m_views[1].m_view->resized(root, ssub);
        ssub.location[1] -= 1;
        m_splitBlock.setViewRect(root, ssub);
        setHorizontalVerts(ssub.size[0]);
    }
    else if (m_axis == Axis::Vertical)
    {
        boo::SWindowRect ssub = sub;
        ssub.size[0] *= m_slide;
        if (m_views[0].m_view)
            m_views[0].m_view->resized(root, ssub);
        ssub.location[0] += ssub.size[0];
        ssub.size[0] = sub.size[0] - ssub.size[0];
        if (m_views[1].m_view)
            m_views[1].m_view->resized(root, ssub);
        ssub.location[0] -= 1;
        m_splitBlock.setViewRect(root, ssub);
        setVerticalVerts(ssub.size[1]);
    }
    m_splitBlockBuf.access().finalAssign(m_splitBlock);
    m_splitVertsBinding.load<decltype(m_splitVerts)>(m_splitVerts);
}

void SplitView::draw(boo::IGraphicsCommandQueue* gfxQ)
{
    View::draw(gfxQ);
    if (m_views[0].m_view)
        m_views[0].m_view->draw(gfxQ);
    if (m_views[1].m_view)
        m_views[1].m_view->draw(gfxQ);
    gfxQ->setShaderDataBinding(m_splitVertsBinding);
    gfxQ->draw(0, 4);
}

}