/*
   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
 */

#include "model.h"
#include "globaldefs.h"

#include "picomodel.h"

#include "iarchive.h"
#include "idatastream.h"
#include "imodel.h"
#include "modelskin.h"

#include "cullable.h"
#include "renderable.h"
#include "selectable.h"

#include "math/frustum.h"
#include "string/string.h"
#include "generic/static.h"
#include "shaderlib.h"
#include "scenelib.h"
#include "instancelib.h"
#include "transformlib.h"
#include "traverselib.h"
#include "render.h"

class VectorLightList : public LightList {
    typedef std::vector<const RendererLight *> Lights;
    Lights m_lights;
public:
    void addLight(const RendererLight &light)
    {
        m_lights.push_back(&light);
    }

    void clear()
    {
        m_lights.clear();
    }

    void evaluateLights() const
    {
    }

    void lightsChanged() const
    {
    }

    void forEachLight(const RendererLightCallback &callback) const
    {
        for (Lights::const_iterator i = m_lights.begin(); i != m_lights.end(); ++i) {
            callback(*(*i));
        }
    }
};

class PicoSurface :
        public OpenGLRenderable {
    AABB m_aabb_local;
    CopiedString m_shader;
    Shader *m_state;

    Array<ArbitraryMeshVertex> m_vertices;
    Array<RenderIndex> m_indices;

public:

    PicoSurface()
    {
        constructNull();
        CaptureShader();
    }

    PicoSurface(picoSurface_t *surface)
    {
        CopyPicoSurface(surface);
        CaptureShader();
    }

    ~PicoSurface()
    {
        ReleaseShader();
    }

    void render(RenderStateFlags state) const
    {
        if ((state & RENDER_BUMP) != 0) {
            if (GlobalShaderCache().useShaderLanguage()) {
                glNormalPointer(GL_FLOAT, sizeof(ArbitraryMeshVertex), &m_vertices.data()->normal);
                glVertexAttribPointerARB(c_attr_TexCoord0, 2, GL_FLOAT, 0, sizeof(ArbitraryMeshVertex),
                                         &m_vertices.data()->texcoord);
                glVertexAttribPointerARB(c_attr_Tangent, 3, GL_FLOAT, 0, sizeof(ArbitraryMeshVertex),
                                         &m_vertices.data()->tangent);
                glVertexAttribPointerARB(c_attr_Binormal, 3, GL_FLOAT, 0, sizeof(ArbitraryMeshVertex),
                                         &m_vertices.data()->bitangent);
            } else {
                glVertexAttribPointerARB(11, 3, GL_FLOAT, 0, sizeof(ArbitraryMeshVertex), &m_vertices.data()->normal);
                glVertexAttribPointerARB(8, 2, GL_FLOAT, 0, sizeof(ArbitraryMeshVertex), &m_vertices.data()->texcoord);
                glVertexAttribPointerARB(9, 3, GL_FLOAT, 0, sizeof(ArbitraryMeshVertex), &m_vertices.data()->tangent);
                glVertexAttribPointerARB(10, 3, GL_FLOAT, 0, sizeof(ArbitraryMeshVertex),
                                         &m_vertices.data()->bitangent);
            }
        } else {
            glNormalPointer(GL_FLOAT, sizeof(ArbitraryMeshVertex), &m_vertices.data()->normal);
            glTexCoordPointer(2, GL_FLOAT, sizeof(ArbitraryMeshVertex), &m_vertices.data()->texcoord);
        }
        glVertexPointer(3, GL_FLOAT, sizeof(ArbitraryMeshVertex), &m_vertices.data()->vertex);
        glDrawElements(GL_TRIANGLES, GLsizei(m_indices.size()), RenderIndexTypeID, m_indices.data());

#if 0
        GLfloat modelview[16];
        glGetFloatv(GL_MODELVIEW_MATRIX, modelview); // I know this is slow as hell, but hey - we're in _DEBUG
        Matrix4 modelview_inv(
                modelview[0], modelview[1], modelview[2], modelview[3],
                modelview[4], modelview[5], modelview[6], modelview[7],
                modelview[8], modelview[9], modelview[10], modelview[11],
                modelview[12], modelview[13], modelview[14], modelview[15]);
        matrix4_full_invert(modelview_inv);
        Matrix4 modelview_inv_transposed = matrix4_transposed(modelview_inv);

        glBegin(GL_LINES);

        for (Array<ArbitraryMeshVertex>::const_iterator i = m_vertices.begin(); i != m_vertices.end(); ++i) {
            Vector3 normal = normal3f_to_vector3((*i).normal);
            normal = matrix4_transformed_direction(modelview_inv, vector3_normalised(
                    matrix4_transformed_direction(modelview_inv_transposed, normal))); // do some magic
            Vector3 normalTransformed = vector3_added(vertex3f_to_vector3((*i).vertex), vector3_scaled(normal, 8));
            glVertex3fv(vertex3f_to_array((*i).vertex));
            glVertex3fv(vector3_to_array(normalTransformed));
        }
        glEnd();
#endif
    }

    VolumeIntersectionValue intersectVolume(const VolumeTest &test, const Matrix4 &localToWorld) const
    {
        return test.TestAABB(m_aabb_local, localToWorld);
    }

    const AABB &localAABB() const
    {
        return m_aabb_local;
    }

    void render(Renderer &renderer, const Matrix4 &localToWorld, Shader *state) const
    {
        renderer.SetState(state, Renderer::eFullMaterials);
        renderer.addRenderable(*this, localToWorld);
    }

    void render(Renderer &renderer, const Matrix4 &localToWorld) const
    {
        render(renderer, localToWorld, m_state);
    }

    void testSelect(Selector &selector, SelectionTest &test, const Matrix4 &localToWorld)
    {
        test.BeginMesh(localToWorld);

        SelectionIntersection best;
        testSelect(test, best);
        if (best.valid()) {
            selector.addIntersection(best);
        }
    }

    const char *getShader() const
    {
        return m_shader.c_str();
    }

    Shader *getState() const
    {
        return m_state;
    }

private:

    void CaptureShader()
    {
        m_state = GlobalShaderCache().capture(m_shader.c_str());
    }

    void ReleaseShader()
    {
        GlobalShaderCache().release(m_shader.c_str());
    }

    void UpdateAABB()
    {
        m_aabb_local = AABB();
        for (std::size_t i = 0; i < m_vertices.size(); ++i) {
            aabb_extend_by_point_safe(m_aabb_local, reinterpret_cast<const Vector3 &>( m_vertices[i].vertex ));
        }


        for (Array<RenderIndex>::iterator i = m_indices.begin(); i != m_indices.end(); i += 3) {
            ArbitraryMeshVertex &a = m_vertices[*(i + 0)];
            ArbitraryMeshVertex &b = m_vertices[*(i + 1)];
            ArbitraryMeshVertex &c = m_vertices[*(i + 2)];

            ArbitraryMeshTriangle_sumTangents(a, b, c);
        }

        for (Array<ArbitraryMeshVertex>::iterator i = m_vertices.begin(); i != m_vertices.end(); ++i) {
            vector3_normalise(reinterpret_cast<Vector3 &>((*i).tangent ));
            vector3_normalise(reinterpret_cast<Vector3 &>((*i).bitangent ));
        }
    }

    void testSelect(SelectionTest &test, SelectionIntersection &best)
    {
        test.TestTriangles(
                VertexPointer(VertexPointer::pointer(&m_vertices.data()->vertex), sizeof(ArbitraryMeshVertex)),
                IndexPointer(m_indices.data(), IndexPointer::index_type(m_indices.size())),
                best
        );
    }

    void CopyPicoSurface(picoSurface_t *surface)
    {
        picoShader_t *shader = PicoGetSurfaceShader(surface);
        if (shader == 0) {
            m_shader = "";
        } else {
            m_shader = PicoGetShaderName(shader);
        }

        m_vertices.resize(PicoGetSurfaceNumVertexes(surface));
        m_indices.resize(PicoGetSurfaceNumIndexes(surface));

        for (std::size_t i = 0; i < m_vertices.size(); ++i) {
            picoVec_t *xyz = PicoGetSurfaceXYZ(surface, int(i));
            m_vertices[i].vertex = vertex3f_from_array(xyz);

            picoVec_t *normal = PicoGetSurfaceNormal(surface, int(i));
            m_vertices[i].normal = normal3f_from_array(normal);

            picoVec_t *st = PicoGetSurfaceST(surface, 0, int(i));
            m_vertices[i].texcoord = TexCoord2f(st[0], st[1]);

#if 0
            picoVec_t* color = PicoGetSurfaceColor( surface, 0, int(i) );
            m_vertices[i].colour = Colour4b( color[0], color[1], color[2], color[3] );
#endif
        }

        picoIndex_t *indexes = PicoGetSurfaceIndexes(surface, 0);
        for (std::size_t j = 0; j < m_indices.size(); ++j) {
            m_indices[j] = indexes[j];
        }

        UpdateAABB();
    }

    void constructQuad(std::size_t index, const Vector3 &a, const Vector3 &b, const Vector3 &c, const Vector3 &d,
                       const Vector3 &normal)
    {
        m_vertices[index * 4 + 0] = ArbitraryMeshVertex(
                vertex3f_for_vector3(a),
                normal3f_for_vector3(normal),
                texcoord2f_from_array(aabb_texcoord_topleft)
        );
        m_vertices[index * 4 + 1] = ArbitraryMeshVertex(
                vertex3f_for_vector3(b),
                normal3f_for_vector3(normal),
                texcoord2f_from_array(aabb_texcoord_topright)
        );
        m_vertices[index * 4 + 2] = ArbitraryMeshVertex(
                vertex3f_for_vector3(c),
                normal3f_for_vector3(normal),
                texcoord2f_from_array(aabb_texcoord_botright)
        );
        m_vertices[index * 4 + 3] = ArbitraryMeshVertex(
                vertex3f_for_vector3(d),
                normal3f_for_vector3(normal),
                texcoord2f_from_array(aabb_texcoord_botleft)
        );
    }

    void constructNull()
    {
        AABB aabb(Vector3(0, 0, 0), Vector3(8, 8, 8));

        Vector3 points[8];
        aabb_corners(aabb, points);

        m_vertices.resize(24);

        constructQuad(0, points[2], points[1], points[5], points[6], aabb_normals[0]);
        constructQuad(1, points[1], points[0], points[4], points[5], aabb_normals[1]);
        constructQuad(2, points[0], points[1], points[2], points[3], aabb_normals[2]);
        constructQuad(3, points[0], points[3], points[7], points[4], aabb_normals[3]);
        constructQuad(4, points[3], points[2], points[6], points[7], aabb_normals[4]);
        constructQuad(5, points[7], points[6], points[5], points[4], aabb_normals[5]);

        m_indices.resize(36);

        RenderIndex indices[36] = {
                0, 1, 2, 0, 2, 3,
                4, 5, 6, 4, 6, 7,
                8, 9, 10, 8, 10, 11,
                12, 13, 14, 12, 14, 15,
                16, 17, 18, 16, 18, 19,
                20, 21, 22, 10, 22, 23,
        };


        Array<RenderIndex>::iterator j = m_indices.begin();
        for (RenderIndex *i = indices; i != indices + (sizeof(indices) / sizeof(RenderIndex)); ++i) {
            *j++ = *i;
        }

        m_shader = "";

        UpdateAABB();
    }
};


typedef std::pair<CopiedString, int> PicoModelKey;


class PicoModel :
        public Cullable,
        public Bounded {
    typedef std::vector<PicoSurface *> surfaces_t;
    surfaces_t m_surfaces;

    AABB m_aabb_local;
public:
    Callback<void()> m_lightsChanged;

    PicoModel()
    {
        constructNull();
    }

    PicoModel(picoModel_t *model)
    {
        CopyPicoModel(model);
    }

    ~PicoModel()
    {
        for (surfaces_t::iterator i = m_surfaces.begin(); i != m_surfaces.end(); ++i) {
            delete *i;
        }
    }

    typedef surfaces_t::const_iterator const_iterator;

    const_iterator begin() const
    {
        return m_surfaces.begin();
    }

    const_iterator end() const
    {
        return m_surfaces.end();
    }

    std::size_t size() const
    {
        return m_surfaces.size();
    }

    VolumeIntersectionValue intersectVolume(const VolumeTest &test, const Matrix4 &localToWorld) const
    {
        return test.TestAABB(m_aabb_local, localToWorld);
    }

    virtual const AABB &localAABB() const
    {
        return m_aabb_local;
    }

    void render(Renderer &renderer, const VolumeTest &volume, const Matrix4 &localToWorld,
                std::vector<Shader *> states) const
    {
        for (surfaces_t::const_iterator i = m_surfaces.begin(); i != m_surfaces.end(); ++i) {
            if ((*i)->intersectVolume(volume, localToWorld) != c_volumeOutside) {
                (*i)->render(renderer, localToWorld, states[i - m_surfaces.begin()]);
            }
        }
    }

    void testSelect(Selector &selector, SelectionTest &test, const Matrix4 &localToWorld)
    {
        for (surfaces_t::iterator i = m_surfaces.begin(); i != m_surfaces.end(); ++i) {
            if ((*i)->intersectVolume(test.getVolume(), localToWorld) != c_volumeOutside) {
                (*i)->testSelect(selector, test, localToWorld);
            }
        }
    }

private:
    void CopyPicoModel(picoModel_t *model)
    {
        m_aabb_local = AABB();

        /* each surface on the model will become a new map drawsurface */
        int numSurfaces = PicoGetModelNumSurfaces(model);
        //%  SYs_FPrintf( SYS_VRB, "Model %s has %d surfaces\n", name, numSurfaces );
        for (int s = 0; s < numSurfaces; ++s) {
            /* get surface */
            picoSurface_t *surface = PicoGetModelSurface(model, s);
            if (surface == 0) {
                continue;
            }

            /* only handle triangle surfaces initially (fixme: support patches) */
            if (PicoGetSurfaceType(surface) != PICO_TRIANGLES) {
                continue;
            }

            /* fix the surface's normals */
            PicoFixSurfaceNormals(surface);

            PicoSurface *picosurface = new PicoSurface(surface);
            aabb_extend_by_aabb_safe(m_aabb_local, picosurface->localAABB());
            m_surfaces.push_back(picosurface);
        }
    }

    void constructNull()
    {
        PicoSurface *picosurface = new PicoSurface();
        m_aabb_local = picosurface->localAABB();
        m_surfaces.push_back(picosurface);
    }
};

inline void
Surface_addLight(PicoSurface &surface, VectorLightList &lights, const Matrix4 &localToWorld, const RendererLight &light)
{
    if (light.testAABB(aabb_for_oriented_aabb(surface.localAABB(), localToWorld))) {
        lights.addLight(light);
    }
}

class PicoModelInstance :
        public scene::Instance,
        public Renderable,
        public SelectionTestable,
        public LightCullable,
        public SkinnedModel {
    class TypeCasts {
        InstanceTypeCastTable m_casts;
    public:
        TypeCasts()
        {
            InstanceContainedCast<PicoModelInstance, Bounded>::install(m_casts);
            InstanceContainedCast<PicoModelInstance, Cullable>::install(m_casts);
            InstanceStaticCast<PicoModelInstance, Renderable>::install(m_casts);
            InstanceStaticCast<PicoModelInstance, SelectionTestable>::install(m_casts);
            InstanceStaticCast<PicoModelInstance, SkinnedModel>::install(m_casts);
        }

        InstanceTypeCastTable &get()
        {
            return m_casts;
        }
    };

    PicoModel &m_picomodel;

    const LightList *m_lightList;
    typedef Array<VectorLightList> SurfaceLightLists;
    SurfaceLightLists m_surfaceLightLists;

    class Remap {
    public:
        CopiedString first;
        Shader *second;

        Remap() : second(0)
        {
        }
    };

    typedef Array<Remap> SurfaceRemaps;
    SurfaceRemaps m_skins;

    PicoModelInstance(const PicoModelInstance &);

    PicoModelInstance operator=(const PicoModelInstance &);

public:
    typedef LazyStatic<TypeCasts> StaticTypeCasts;

    void *m_test;

    Bounded &get(NullType<Bounded>)
    {
        return m_picomodel;
    }

    Cullable &get(NullType<Cullable>)
    {
        return m_picomodel;
    }

    void lightsChanged()
    {
        m_lightList->lightsChanged();
    }

    typedef MemberCaller<PicoModelInstance, void(), &PicoModelInstance::lightsChanged> LightsChangedCaller;

    void constructRemaps()
    {
        ASSERT_MESSAGE(m_skins.size() == m_picomodel.size(), "ERROR");
        ModelSkin *skin = NodeTypeCast<ModelSkin>::cast(path().parent());
        if (skin != 0 && skin->realised()) {
            SurfaceRemaps::iterator j = m_skins.begin();
            for (PicoModel::const_iterator i = m_picomodel.begin(); i != m_picomodel.end(); ++i, ++j) {
                const char *remap = skin->getRemap((*i)->getShader());
                if (!string_empty(remap)) {
                    (*j).first = remap;
                    (*j).second = GlobalShaderCache().capture(remap);
                } else {
                    (*j).second = 0;
                }
            }
            SceneChangeNotify();
        }
    }

    void destroyRemaps()
    {
        ASSERT_MESSAGE(m_skins.size() == m_picomodel.size(), "ERROR");
        for (SurfaceRemaps::iterator i = m_skins.begin(); i != m_skins.end(); ++i) {
            if ((*i).second != 0) {
                GlobalShaderCache().release((*i).first.c_str());
                (*i).second = 0;
            }
        }
    }

    void skinChanged()
    {
        destroyRemaps();
        constructRemaps();
    }

    PicoModelInstance(const scene::Path &path, scene::Instance *parent, PicoModel &picomodel) :
            Instance(path, parent, this, StaticTypeCasts::instance().get()),
            m_picomodel(picomodel),
            m_surfaceLightLists(m_picomodel.size()),
            m_skins(m_picomodel.size())
    {
        m_lightList = &GlobalShaderCache().attach(*this);
        m_picomodel.m_lightsChanged = LightsChangedCaller(*this);

        Instance::setTransformChangedCallback(LightsChangedCaller(*this));

        constructRemaps();
    }

    ~PicoModelInstance()
    {
        destroyRemaps();

        Instance::setTransformChangedCallback(Callback<void()>());

        m_picomodel.m_lightsChanged = Callback<void()>();
        GlobalShaderCache().detach(*this);
    }

    void render(Renderer &renderer, const VolumeTest &volume, const Matrix4 &localToWorld) const
    {
        SurfaceLightLists::const_iterator j = m_surfaceLightLists.begin();
        SurfaceRemaps::const_iterator k = m_skins.begin();
        for (PicoModel::const_iterator i = m_picomodel.begin(); i != m_picomodel.end(); ++i, ++j, ++k) {
            if ((*i)->intersectVolume(volume, localToWorld) != c_volumeOutside) {
                renderer.setLights(*j);
                (*i)->render(renderer, localToWorld, (*k).second != 0 ? (*k).second : (*i)->getState());
            }
        }
    }

    void renderSolid(Renderer &renderer, const VolumeTest &volume) const
    {
        m_lightList->evaluateLights();

        render(renderer, volume, Instance::localToWorld());
    }

    void renderWireframe(Renderer &renderer, const VolumeTest &volume) const
    {
        renderSolid(renderer, volume);
    }

    void testSelect(Selector &selector, SelectionTest &test)
    {
        m_picomodel.testSelect(selector, test, Instance::localToWorld());
    }

    bool testLight(const RendererLight &light) const
    {
        return light.testAABB(worldAABB());
    }

    void insertLight(const RendererLight &light)
    {
        const Matrix4 &localToWorld = Instance::localToWorld();
        SurfaceLightLists::iterator j = m_surfaceLightLists.begin();
        for (PicoModel::const_iterator i = m_picomodel.begin(); i != m_picomodel.end(); ++i) {
            Surface_addLight(*(*i), *j++, localToWorld, light);
        }
    }

    void clearLights()
    {
        for (SurfaceLightLists::iterator i = m_surfaceLightLists.begin(); i != m_surfaceLightLists.end(); ++i) {
            (*i).clear();
        }
    }
};

class PicoModelNode : public scene::Node::Symbiot, public scene::Instantiable {
    class TypeCasts {
        NodeTypeCastTable m_casts;
    public:
        TypeCasts()
        {
            NodeStaticCast<PicoModelNode, scene::Instantiable>::install(m_casts);
        }

        NodeTypeCastTable &get()
        {
            return m_casts;
        }
    };


    scene::Node m_node;
    InstanceSet m_instances;
    PicoModel m_picomodel;

public:
    typedef LazyStatic<TypeCasts> StaticTypeCasts;

    PicoModelNode() : m_node(this, this, StaticTypeCasts::instance().get())
    {
    }

    PicoModelNode(picoModel_t *model) : m_node(this, this, StaticTypeCasts::instance().get()), m_picomodel(model)
    {
    }

    void release()
    {
        delete this;
    }

    scene::Node &node()
    {
        return m_node;
    }

    scene::Instance *create(const scene::Path &path, scene::Instance *parent)
    {
        return new PicoModelInstance(path, parent, m_picomodel);
    }

    void forEachInstance(const scene::Instantiable::Visitor &visitor)
    {
        m_instances.forEachInstance(visitor);
    }

    void insert(scene::Instantiable::Observer *observer, const scene::Path &path, scene::Instance *instance)
    {
        m_instances.insert(observer, path, instance);
    }

    scene::Instance *erase(scene::Instantiable::Observer *observer, const scene::Path &path)
    {
        return m_instances.erase(observer, path);
    }
};


#if 0

template<typename Key, typename Type>
class create_new
{
public:
static Type* construct( const Key& key ){
    return new Type( key );
}
static void destroy( Type* value ){
    delete value;
}
};

template<typename Key, typename Type, typename creation_policy = create_new<Key, Type> >
class cache_element : public creation_policy
{
public:
inline cache_element() : m_count( 0 ), m_value( 0 ) {}
inline ~cache_element(){
    ASSERT_MESSAGE( m_count == 0, "destroyed a reference before it was released\n" );
    if ( m_count > 0 ) {
        destroy();
    }
}
inline Type* capture( const Key& key ){
    if ( ++m_count == 1 ) {
        construct( key );
    }
    return m_value;
}
inline void release(){
    ASSERT_MESSAGE( !empty(), "failed to release reference - not found in cache\n" );
    if ( --m_count == 0 ) {
        destroy();
    }
}
inline bool empty(){
    return m_count == 0;
}
inline void refresh( const Key& key ){
    m_value->refresh( key );
}
private:
inline void construct( const Key& key ){
    m_value = creation_policy::construct( key );
}
inline void destroy(){
    creation_policy::destroy( m_value );
}

std::size_t m_count;
Type* m_value;
};

class create_picomodel
{
typedef PicoModelKey key_type;
typedef PicoModel value_type;
public:
static value_type* construct( const key_type& key ){
    picoModel_t* picomodel = PicoLoadModel( const_cast<char*>( key.first.c_str() ), key.second );
    value_type* value = new value_type( picomodel );
    PicoFreeModel( picomodel );
    return value;
}
static void destroy( value_type* value ){
    delete value;
}
};

#include <map>

class ModelCache
{
typedef PicoModel value_type;

public:
typedef PicoModelKey key_type;
typedef cache_element<key_type, value_type, create_picomodel> elem_type;
typedef std::map<key_type, elem_type> cache_type;

value_type* capture( const key_type& key ){
    return m_cache[key].capture( key );
}
void release( const key_type& key ){
    m_cache[key].release();
}

private:
cache_type m_cache;
};

ModelCache g_model_cache;



typedef struct remap_s {
    char m_remapbuff[64 + 1024];
    char *original;
    char *remap;
} remap_t;

class RemapWrapper :
    public Cullable,
    public Bounded
{
public:
RemapWrapper( const char* name ){
    parse_namestr( name );

    m_model = g_model_cache.capture( ModelCache::key_type( m_name, m_frame ) );

    construct_shaders();
}
virtual ~RemapWrapper(){
    g_model_cache.release( ModelCache::key_type( m_name, m_frame ) );

    for ( shaders_t::iterator i = m_shaders.begin(); i != m_shaders.end(); ++i )
    {
        GlobalShaderCache().release( ( *i ).c_str() );
    }

    for ( remaps_t::iterator j = m_remaps.begin(); j != m_remaps.end(); ++j )
    {
        delete ( *j );
    }
}

VolumeIntersectionValue intersectVolume( const VolumeTest& test, const Matrix4& localToWorld ) const {
    return m_model->intersectVolume( test, localToWorld );
}

virtual const AABB& localAABB() const {
    return m_model->localAABB();
}

void render( Renderer& renderer, const VolumeTest& volume, const Matrix4& localToWorld ) const {
    m_model->render( renderer, volume, localToWorld, m_states );
}

void testSelect( Selector& selector, SelectionTest& test, const Matrix4& localToWorld ){
    m_model->testSelect( selector, test, localToWorld );
}

private:
void add_remap( const char *remap ){
    const char *ch;
    remap_t *pRemap;

    ch = remap;

    while ( *ch && *ch != ';' )
        ch++;

    if ( *ch == '\0' ) {
        // bad remap
        globalErrorStream() << "WARNING: Shader _remap key found in a model entity without a ; character\n";
    }
    else {
        pRemap = new remap_t;

        strncpy( pRemap->m_remapbuff, remap, sizeof( pRemap->m_remapbuff ) );

        pRemap->m_remapbuff[ch - remap] = '\0';

        pRemap->original = pRemap->m_remapbuff;
        pRemap->remap = pRemap->m_remapbuff + ( ch - remap ) + 1;

        m_remaps.push_back( pRemap );
    }
}

void parse_namestr( const char *name ){
    const char *ptr, *s;
    bool hasName, hasFrame;

    hasName = hasFrame = false;

    m_frame = 0;

    for ( s = ptr = name; ; ++ptr )
    {
        if ( !hasName && ( *ptr == ':' || *ptr == '\0' ) ) {
            // model name
            hasName = true;
            m_name = CopiedString( s, ptr );
            s = ptr + 1;
        }
        else if ( *ptr == '?' || *ptr == '\0' ) {
            // model frame
            hasFrame = true;
            m_frame = atoi( CopiedString( s, ptr ).c_str() );
            s = ptr + 1;
        }
        else if ( *ptr == '&' || *ptr == '\0' ) {
            // a remap
            add_remap( CopiedString( s, ptr ).c_str() );
            s = ptr + 1;
        }

        if ( *ptr == '\0' ) {
            break;
        }
    }
}

void construct_shaders(){
    const char* global_shader = shader_for_remap( "*" );

    m_shaders.reserve( m_model->size() );
    m_states.reserve( m_model->size() );
    for ( PicoModel::iterator i = m_model->begin(); i != m_model->end(); ++i )
    {
        const char* shader = shader_for_remap( ( *i )->getShader() );
        m_shaders.push_back(
            ( shader[0] != '\0' )
            ? shader
            : ( global_shader[0] != '\0' )
            ? global_shader
            : ( *i )->getShader() );
        m_states.push_back( GlobalShaderCache().capture( m_shaders.back().c_str() ) );
    }
}

inline const char* shader_for_remap( const char* remap ){
    for ( remaps_t::iterator i = m_remaps.begin(); i != m_remaps.end(); ++i )
    {
        if ( shader_equal( remap, ( *i )->original ) ) {
            return ( *i )->remap;
        }
    }
    return "";
}

CopiedString m_name;
int m_frame;
PicoModel* m_model;

typedef std::vector<remap_t*> remaps_t;
remaps_t m_remaps;
typedef std::vector<CopiedString> shaders_t;
shaders_t m_shaders;
typedef std::vector<Shader*> states_t;
states_t m_states;
};

class RemapWrapperInstance : public scene::Instance, public Renderable, public SelectionTestable
{
RemapWrapper& m_remapwrapper;
public:
RemapWrapperInstance( const scene::Path& path, scene::Instance* parent, RemapWrapper& remapwrapper ) : Instance( path, parent ), m_remapwrapper( remapwrapper ){
    scene::Instance::m_cullable = &m_remapwrapper;
    scene::Instance::m_render = this;
    scene::Instance::m_select = this;
}

void renderSolid( Renderer& renderer, const VolumeTest& volume ) const {
    m_remapwrapper.render( renderer, volume, Instance::localToWorld() );
}
void renderWireframe( Renderer& renderer, const VolumeTest& volume ) const {
    renderSolid( renderer, volume );
}

void testSelect( Selector& selector, SelectionTest& test ){
    m_remapwrapper.testSelect( selector, test, Instance::localToWorld() );
}
};

class RemapWrapperNode : public scene::Node::Symbiot, public scene::Instantiable
{
scene::Node m_node;
typedef RemapWrapperInstance instance_type;
InstanceSet m_instances;
RemapWrapper m_remapwrapper;
public:
RemapWrapperNode( const char* name ) : m_node( this ), m_remapwrapper( name ){
    m_node.m_instance = this;
}

void release(){
    delete this;
}
scene::Node& node(){
    return m_node;
}

scene::Instance* create( const scene::Path& path, scene::Instance* parent ){
    return new instance_type( path, parent, m_remapwrapper );
}
void forEachInstance( const scene::Instantiable::Visitor& visitor ){
    m_instances.forEachInstance( visitor );
}
void insert( scene::Instantiable::Observer* observer, const scene::Path& path, scene::Instance* instance ){
    m_instances.insert( observer, path, instance );
}
scene::Instance* erase( scene::Instantiable::Observer* observer, const scene::Path& path ){
    return m_instances.erase( observer, path );
}
};

scene::Node& LoadRemapModel( const char* name ){
    return ( new RemapWrapperNode( name ) )->node();
}

#endif


size_t picoInputStreamReam(void *inputStream, unsigned char *buffer, size_t length)
{
    return reinterpret_cast<InputStream *>( inputStream )->read(buffer, length);
}

scene::Node &loadPicoModel(const picoModule_t *module, ArchiveFile &file)
{
    picoModel_t *model = PicoModuleLoadModelStream(module, &file.getInputStream(), picoInputStreamReam, file.size(), 0,
                                                   file.getName());
    PicoModelNode *modelNode = new PicoModelNode(model);
    PicoFreeModel(model);
    return modelNode->node();
}