/*
   Copyright (C) 2001-2006, William Joseph.
   All Rights Reserved.

   This file is part of GtkRadiant.

   GtkRadiant is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.

   GtkRadiant is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with GtkRadiant; if not, write to the Free Software
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#if !defined( INCLUDED_BRUSHXML_H )
#define INCLUDED_BRUSHXML_H

#include "stream/stringstream.h"
#include "xml/xmlelement.h"

#include "brush.h"

inline void FaceTexdef_BP_importXML(FaceTexdef &texdef, const char *xmlContent)
{
    StringTokeniser content(xmlContent);

    texdef.m_projection.m_brushprimit_texdef.coords[0][0] = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_brushprimit_texdef.coords[0][1] = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_brushprimit_texdef.coords[0][2] = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_brushprimit_texdef.coords[1][0] = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_brushprimit_texdef.coords[1][1] = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_brushprimit_texdef.coords[1][2] = static_cast<float>( atof(content.getToken()));
}

inline void FaceTexdef_importXML(FaceTexdef &texdef, const char *xmlContent)
{
    StringTokeniser content(xmlContent);

    texdef.m_projection.m_texdef.shift[0] = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_texdef.shift[1] = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_texdef.rotate = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_texdef.scale[0] = static_cast<float>( atof(content.getToken()));
    texdef.m_projection.m_texdef.scale[1] = static_cast<float>( atof(content.getToken()));

    ASSERT_MESSAGE(texdef_sane(texdef.m_projection.m_texdef), "FaceTexdef_importXML: bad texdef");
}

inline void FacePlane_importXML(FacePlane &facePlane, const char *xmlContent)
{
    StringTokeniser content(xmlContent);

    for (int i = 0; i < 3; ++i) {
        for (int j = 0; j < 3; ++j) {
            facePlane.planePoints()[i][j] = atof(content.getToken());
        }
    }
    facePlane.MakePlane();
}


class FaceXMLImporter {
    struct xml_state_t {
        enum EState {
            eDefault,
            ePlanePts,
            eTexdef,
            eBPMatrix,
            eFlags,
            eShader,
        };

        EState m_state;
        StringOutputStream m_content;

        xml_state_t(EState state)
                : m_state(state)
        {}

        EState state() const
        {
            return m_state;
        }

        const char *content() const
        {
            return m_content.c_str();
        }

        std::size_t write(const char *buffer, std::size_t length)
        {
            return m_content.write(buffer, length);
        }
    };

    std::vector<xml_state_t> m_xml_state;
    Face &m_face;
public:
    FaceXMLImporter(Face &face) : m_face(face)
    {
        m_xml_state.push_back(xml_state_t::eDefault);
    }

    ~FaceXMLImporter()
    {
        m_face.planeChanged();
    }

    void pushElement(const XMLElement &element)
    {
        ASSERT_MESSAGE(m_xml_state.back().state() == xml_state_t::eDefault, "parse error");

        if (strcmp(element.name(), "planepts") == 0) {
            m_xml_state.push_back(xml_state_t::ePlanePts);
        } else if (strcmp(element.name(), "texdef") == 0) {
            m_xml_state.push_back(xml_state_t::eTexdef);
        } else if (strcmp(element.name(), "bpmatrix") == 0) {
            m_xml_state.push_back(xml_state_t::eBPMatrix);
        } else if (strcmp(element.name(), "flags") == 0) {
            m_xml_state.push_back(xml_state_t::eFlags);
        } else if (strcmp(element.name(), "shader") == 0) {
            m_xml_state.push_back(xml_state_t::eShader);
        }
    }

    void popElement(const char *name)
    {
        ASSERT_MESSAGE(m_xml_state.back().state() != xml_state_t::eDefault, "parse error");

        switch (m_xml_state.back().state()) {
            case xml_state_t::ePlanePts: {
                FacePlane_importXML(m_face.getPlane(), m_xml_state.back().content());
            }
                break;
            case xml_state_t::eTexdef: {
                FaceTexdef_importXML(m_face.getTexdef(), m_xml_state.back().content());
            }
                break;
            case xml_state_t::eBPMatrix: {
                FaceTexdef_BP_importXML(m_face.getTexdef(), m_xml_state.back().content());
            }
                break;
            case xml_state_t::eFlags: {
                StringTokeniser content(m_xml_state.back().content());

                m_face.getShader().m_flags.m_contentFlags = atoi(content.getToken());
                m_face.getShader().m_flags.m_surfaceFlags = atoi(content.getToken());
                m_face.getShader().m_flags.m_value = atoi(content.getToken());
            }
                break;
            case xml_state_t::eShader: {
                m_face.getShader().setShader(m_xml_state.back().content());
            }
                break;
            default:
                break;
        }

        m_xml_state.pop_back();
    }

    std::size_t write(const char *data, std::size_t length)
    {
        ASSERT_MESSAGE(!m_xml_state.empty(), "parse error");
        return m_xml_state.back().write(data, length);
    }
};


inline void FaceTexdef_exportXML(const FaceTexdef &texdef, XMLImporter &importer)
{
    StaticElement element("texdef");
    importer.pushElement(element);

    ASSERT_MESSAGE(texdef_sane(texdef.m_projection.m_texdef), "FaceTexdef_exportXML: bad texdef");

    importer << texdef.m_projection.m_texdef.shift[0]
             << ' ' << texdef.m_projection.m_texdef.shift[1]
             << ' ' << texdef.m_projection.m_texdef.rotate
             << ' ' << texdef.m_projection.m_texdef.scale[0]
             << ' ' << texdef.m_projection.m_texdef.scale[1];

    importer.popElement(element.name());
}

inline void FaceTexdef_BP_exportXML(const FaceTexdef &texdef, XMLImporter &importer)
{
    StaticElement element("texdef");
    importer.pushElement(element);

    for (int i = 0; i < 2; ++i) {
        for (int j = 0; j < 3; ++j) {
            importer << texdef.m_projection.m_brushprimit_texdef.coords[i][j] << ' ';
        }
    }

    importer.popElement(element.name());
}

inline void FaceShader_ContentsFlagsValue_exportXML(const FaceShader &faceShader, XMLImporter &importer)
{
    StaticElement element("flags");
    importer.pushElement(element);

    {
        importer << faceShader.m_flags.m_contentFlags
                 << ' ' << faceShader.m_flags.m_surfaceFlags
                 << ' ' << faceShader.m_flags.m_value;
    }

    importer.popElement(element.name());
}

inline void FacePlane_exportXML(const FacePlane &facePlane, XMLImporter &importer)
{
    StaticElement element("planepts");
    importer.pushElement(element);

    {
        // write planepts
        for (int i = 0; i < 3; i++) {
            for (int j = 0; j < 3; j++) {
                importer << Face::m_quantise(facePlane.planePoints()[i][j]) << ' ';
            }
        }
    }

    importer.popElement(element.name());
}

inline void FacePolygon_exportXML(const Winding &w, const BasicVector3<double> &normal, XMLImporter &importer)
{
    DynamicElement element("polygon");

    char tmp[32];

    sprintf(tmp, "%f", normal.x());
    element.insertAttribute("nx", tmp);

    sprintf(tmp, "%f", normal.y());
    element.insertAttribute("ny", tmp);

    sprintf(tmp, "%f", normal.z());
    element.insertAttribute("nz", tmp);

    importer.pushElement(element);

    for (unsigned int i = 0; i < w.numpoints; ++i) {
        DynamicElement c("vertex");

        sprintf(tmp, "%f", w.points[i].vertex.x());
        c.insertAttribute("x", tmp);

        sprintf(tmp, "%f", w.points[i].vertex.y());
        c.insertAttribute("y", tmp);

        sprintf(tmp, "%f", w.points[i].vertex.z());
        c.insertAttribute("z", tmp);

        sprintf(tmp, "%f", w.points[i].texcoord.x());
        c.insertAttribute("s", tmp);

        sprintf(tmp, "%f", w.points[i].texcoord.y());
        c.insertAttribute("t", tmp);

        importer.pushElement(c);
        importer.popElement(c.name());
    }

    importer.popElement(element.name());
}

class FaceXMLExporter {
    const Face &m_face;
public:
    FaceXMLExporter(const Face &face) : m_face(face)
    {
    }

    void exportXML(XMLImporter &importer)
    {
        bool bAlternateTexdef = (Face::m_type == eBrushTypeQuake3BP || Face::m_type == eBrushTypeDoom3 ||
                                 Face::m_type == eBrushTypeQuake4);

        // write shader
        {
            StaticElement element("shader");
            importer.pushElement(element);
            importer << m_face.getShader().getShader();
            importer.popElement(element.name());
        }

        FacePolygon_exportXML(m_face.getWinding(), m_face.getPlane().plane3().normal(), importer);
        FacePlane_exportXML(m_face.getPlane(), importer);

        if (!bAlternateTexdef) {
            FaceTexdef_exportXML(m_face.getTexdef(), importer);
        } else {
            FaceTexdef_BP_exportXML(m_face.getTexdef(), importer);
        }

        FaceShader_ContentsFlagsValue_exportXML(m_face.getShader(), importer);
    }
};


class BrushXMLImporter : public XMLImporter {
    class xml_state_t {
    public:
        enum EState {
            eDefault,
            eBrush,
            eFace,
        };

    private:
        EState m_state;

    public:
        xml_state_t(EState state)
                : m_state(state)
        {
        }

        EState state() const
        {
            return m_state;
        }
    };

    std::vector<xml_state_t> m_xml_state;
    char m_faceImporter[sizeof(FaceXMLImporter)];
    Brush &m_brush;

    FaceXMLImporter &faceImporter()
    {
        return *reinterpret_cast<FaceXMLImporter *>( m_faceImporter );
    }

public:
    BrushXMLImporter(Brush &brush) : m_brush(brush)
    {
        m_xml_state.push_back(xml_state_t::eDefault);
    }

    void pushElement(const XMLElement &element)
    {
        switch (m_xml_state.back().state()) {
            case xml_state_t::eDefault:
                ASSERT_MESSAGE(strcmp(element.name(), "brush") == 0, "parse error");
                m_xml_state.push_back(xml_state_t::eBrush);
                break;
            case xml_state_t::eBrush:
                ASSERT_MESSAGE(strcmp(element.name(), "plane") == 0, "parse error");
                m_xml_state.push_back(xml_state_t::eFace);
                m_brush.push_back(FaceSmartPointer(new Face(&m_brush)));
                constructor(faceImporter(), makeReference(*m_brush.back()));
                m_brush.planeChanged();
                m_brush.shaderChanged();
                break;
            case xml_state_t::eFace:
                m_xml_state.push_back(xml_state_t::eFace);
                faceImporter().pushElement(element);
                break;
        }
    }

    void popElement(const char *name)
    {
        ASSERT_MESSAGE(!m_xml_state.empty(), "parse error");
        m_xml_state.pop_back();

        switch (m_xml_state.back().state()) {
            case xml_state_t::eDefault:
                break;
            case xml_state_t::eBrush:
                destructor(faceImporter());
                break;
            case xml_state_t::eFace:
                faceImporter().popElement(name);
                break;
        }
    }

    std::size_t write(const char *data, std::size_t length)
    {
        switch (m_xml_state.back().state()) {
            case xml_state_t::eFace:
                return faceImporter().write(data, length);
                break;
            default:
                break;
        }
        return length;
    }
};

class BrushXMLExporter : public XMLExporter {
    const Brush &m_brush;

public:
    BrushXMLExporter(const Brush &brush) : m_brush(brush)
    {
    }

    void exportXML(XMLImporter &importer)
    {
        m_brush.evaluateBRep(); // ensure b-rep is up-to-date, so that non-contributing faces can be identified.
        ASSERT_MESSAGE(m_brush.hasContributingFaces(), "exporting an empty brush");

        const StaticElement brushElement("brush");
        importer.pushElement(brushElement);

        for (Brush::const_iterator i = m_brush.begin(); i != m_brush.end(); ++i) {
            if ((*i)->contributes()) {
                const StaticElement element("plane");
                importer.pushElement(element);
                FaceXMLExporter(*(*i)).exportXML(importer);
                importer.popElement(element.name());
            }
        }

        importer.popElement(brushElement.name());
    }
};


#endif