#include <LogVisor/LogVisor.hpp>
#include "Specter/Button.hpp"
#include "Specter/ViewResources.hpp"
#include "Specter/RootView.hpp"

namespace Specter
{
static LogVisor::LogModule Log("Specter::Button");

void Button::Resources::init(boo::IGraphicsDataFactory* factory, const ThemeData& theme)
{
}

Button::Button(ViewResources& res, View& parentView, const std::string& text)
: View(res, parentView), m_textStr(text)
{
    m_bBlockBuf = res.m_factory->newDynamicBuffer(boo::BufferUse::Uniform, sizeof(ViewBlock), 1);
    m_bVertsBuf = res.m_factory->newDynamicBuffer(boo::BufferUse::Vertex, sizeof(SolidShaderVert), 28);

    if (!res.m_viewRes.m_texVtxFmt)
    {
        boo::VertexElementDescriptor vdescs[] =
        {
            {m_bVertsBuf, nullptr, boo::VertexSemantic::Position4},
            {m_bVertsBuf, nullptr, boo::VertexSemantic::UV4}
        };
        m_bVtxFmt = res.m_factory->newVertexFormat(2, vdescs);
        boo::IGraphicsBuffer* bufs[] = {m_bBlockBuf};
        m_bShaderBinding = res.m_factory->newShaderDataBinding(res.m_viewRes.m_solidShader,
                                                               m_bVtxFmt, m_bVertsBuf, nullptr,
                                                               nullptr, 1, bufs, 0, nullptr);
    }
    else
    {
        boo::IGraphicsBuffer* bufs[] = {m_bBlockBuf};
        m_bShaderBinding = res.m_factory->newShaderDataBinding(res.m_viewRes.m_solidShader,
                                                               res.m_viewRes.m_texVtxFmt,
                                                               m_bVertsBuf, nullptr,
                                                               nullptr, 1, bufs, 0, nullptr);
    }

    commitResources(res);
    resetResources(res);
}

void Button::setText(const std::string& text)
{
    m_textStr = text;

    m_text->typesetGlyphs(text, rootView().themeData().uiText());
    float pf = rootView().window()->getVirtualPixelFactor();
    float width = m_text->nominalWidth() + 10 * pf;
    float height = 20 * pf;
    m_verts[0].m_pos.assign(1, height+1, 0);
    m_verts[1].m_pos.assign(1, 1, 0);
    m_verts[2].m_pos.assign(width+1, height+1, 0);
    m_verts[3].m_pos.assign(width+1, 1, 0);
    m_verts[4].m_pos.assign(width+1, 1, 0);

    m_verts[5].m_pos.assign(1, height+1, 0);
    m_verts[6].m_pos.assign(1, height+1, 0);
    m_verts[7].m_pos.assign(0, height+1, 0);
    m_verts[8].m_pos.assign(1, 1, 0);
    m_verts[9].m_pos.assign(0, 1, 0);
    m_verts[10].m_pos.assign(0, 1, 0);

    m_verts[11].m_pos.assign(width+2, height+1, 0);
    m_verts[12].m_pos.assign(width+2, height+1, 0);
    m_verts[13].m_pos.assign(width+1, height+1, 0);
    m_verts[14].m_pos.assign(width+2, 1, 0);
    m_verts[15].m_pos.assign(width+1, 1, 0);
    m_verts[16].m_pos.assign(width+1, 1, 0);

    m_verts[17].m_pos.assign(1, height+2, 0);
    m_verts[18].m_pos.assign(1, height+2, 0);
    m_verts[19].m_pos.assign(1, height+1, 0);
    m_verts[20].m_pos.assign(width+1, height+2, 0);
    m_verts[21].m_pos.assign(width+1, height+1, 0);
    m_verts[22].m_pos.assign(width+1, height+1, 0);

    m_verts[23].m_pos.assign(1, 1, 0);
    m_verts[24].m_pos.assign(1, 1, 0);
    m_verts[25].m_pos.assign(1, 0, 0);
    m_verts[26].m_pos.assign(width+1, 1, 0);
    m_verts[27].m_pos.assign(width+1, 0, 0);

    m_bVertsBuf->load(m_verts, sizeof(SolidShaderVert) * 28);
    m_nomWidth = width + 2;
    m_nomHeight = height;
}

void Button::setInactive()
{
    m_verts[0].m_color = rootView().themeData().button1Inactive();
    m_verts[1].m_color = rootView().themeData().button2Inactive();
    m_verts[2].m_color = rootView().themeData().button1Inactive();
    m_verts[3].m_color = rootView().themeData().button2Inactive();
    m_verts[4].m_color = rootView().themeData().button2Inactive();
    m_bVertsBuf->load(m_verts, sizeof(SolidShaderVert) * 28);
}

void Button::setHover()
{
    m_verts[0].m_color = rootView().themeData().button1Hover();
    m_verts[1].m_color = rootView().themeData().button2Hover();
    m_verts[2].m_color = rootView().themeData().button1Hover();
    m_verts[3].m_color = rootView().themeData().button2Hover();
    m_verts[4].m_color = rootView().themeData().button2Hover();
    m_bVertsBuf->load(m_verts, sizeof(SolidShaderVert) * 28);
}

void Button::setPressed()
{
    m_verts[0].m_color = rootView().themeData().button1Press();
    m_verts[1].m_color = rootView().themeData().button2Press();
    m_verts[2].m_color = rootView().themeData().button1Press();
    m_verts[3].m_color = rootView().themeData().button2Press();
    m_verts[4].m_color = rootView().themeData().button2Press();
    m_bVertsBuf->load(m_verts, sizeof(SolidShaderVert) * 28);
}

void Button::setDisabled()
{
    m_verts[0].m_color = rootView().themeData().button1Disabled();
    m_verts[1].m_color = rootView().themeData().button2Disabled();
    m_verts[2].m_color = rootView().themeData().button1Disabled();
    m_verts[3].m_color = rootView().themeData().button2Disabled();
    m_verts[4].m_color = rootView().themeData().button2Disabled();
    m_bVertsBuf->load(m_verts, sizeof(SolidShaderVert) * 28);
}

void Button::mouseDown(const boo::SWindowCoord& coord, boo::EMouseButton button, boo::EModifierKey mod)
{
    m_pressed = true;
    setPressed();
}

void Button::mouseUp(const boo::SWindowCoord& coord, boo::EMouseButton button, boo::EModifierKey mod)
{
    m_pressed = false;
    if (m_hovered)
        setHover();
    else
        setInactive();
}

void Button::mouseEnter(const boo::SWindowCoord& coord)
{
    m_hovered = true;
    if (m_pressed)
        setPressed();
    else
        setHover();
}

void Button::mouseLeave(const boo::SWindowCoord& coord)
{
    m_hovered = false;
    setInactive();
}

void Button::resetResources(ViewResources& res)
{
    m_text.reset(new TextView(res, *this, res.m_mainFont));
    setText(m_textStr);
    m_verts[0].m_color = res.themeData().button1Inactive();
    m_verts[1].m_color = res.themeData().button2Inactive();
    m_verts[2].m_color = res.themeData().button1Inactive();
    m_verts[3].m_color = res.themeData().button2Inactive();
    m_verts[4].m_color = res.themeData().button2Inactive();
    for (int i=5 ; i<28 ; ++i)
        m_verts[i].m_color = res.themeData().button2Inactive();
    m_bVertsBuf->load(m_verts, sizeof(SolidShaderVert) * 28);
}

void Button::resized(const boo::SWindowRect& root, const boo::SWindowRect& sub)
{
    View::resized(root, sub);
    boo::SWindowRect textRect = sub;
    m_bBlock.setViewRect(root, sub);
    m_bBlockBuf->load(&m_bBlock, sizeof(ViewBlock));
    float pf = rootView().window()->getVirtualPixelFactor();
    textRect.location[0] += 5 * pf;
    textRect.location[1] = 11 * pf;
    m_text->resized(root, textRect);
}

void Button::draw(boo::IGraphicsCommandQueue* gfxQ)
{
    View::draw(gfxQ);
    gfxQ->setShaderDataBinding(m_bShaderBinding);
    gfxQ->setDrawPrimitive(boo::Primitive::TriStrips);
    gfxQ->draw(0, 28);
    m_text->draw(gfxQ);
}

}