/*
Copyright (C) 1999-2007 id Software, Inc. and contributors.
For a list of contributors, see the accompanying CONTRIBUTORS file.

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 <stdlib.h>

#include "entitymodel.h"

extern CModelManager g_model_cache;

//
// CEntityMiscModel implementation
//

CEntityMiscModel::CEntityMiscModel ()
{
  refCount = 1;
  m_name = NULL;
  m_model = NULL;
	m_entity = NULL;
  m_frame = 0;
  m_remaps = g_ptr_array_new ();
  m_shaders = g_ptr_array_new ();
  VectorSet(m_translate, 0,0,0);
  VectorSet(m_euler, 0,0,0);
  VectorSet(m_scale, 1,1,1);
  VectorSet(m_pivot, 0,0,0);
  m4x4_identity(m_transform);
  m4x4_identity(m_inverse_transform);
}

typedef struct remap_s {
  char m_key[64];
  char m_remapbuff[64+1024];
  char *m_remap[2];
} remap_t;

CEntityMiscModel::~CEntityMiscModel ()
{
  unsigned int i;

  if(m_name && *m_name != '\0') {
    if( !g_model_cache.DeleteByNameAndFrame(m_name,m_frame) && m_model )
      m_model->RemoveParent( this );
    m_model = NULL;
    delete [] m_name;
  }

  for( i = 0; i < m_remaps->len; i++ )
    delete (remap_t*)m_remaps->pdata[i];
  g_ptr_array_free(m_remaps, FALSE);

  for( i = 0; i < m_shaders->len; i++ )
  {
    (*(IShader**)m_shaders->pdata[i])->DecRef();
    delete (IShader**)m_shaders->pdata[i];
  }
  g_ptr_array_free(m_shaders, FALSE);

  if(m_entity) {
    // This might be just an evasion of the actual problem
    m_entity->model.pRender = NULL;
    m_entity->model.pSelect = NULL;
    m_entity->model.pEdit = NULL;
  }
}

// IRender

void CEntityMiscModel::Draw(int state, int rflags) const
{
  m4x4_t matrix;
  vec3_t pivot;
 
  memcpy(matrix, m_transform, sizeof(m4x4_t));
  m4x4_transpose(matrix);

  VectorAdd(m_pivot, m_translate, pivot);
  pivot_draw(pivot);
  
  // push the current modelview matrix
  // FIXME: put in a check for stack recursion depth..
  // or avoid recursion of opengl matrix stack
  g_QglTable.m_pfn_qglPushMatrix();
  // apply the parent-to-local transform
  g_QglTable.m_pfn_qglMultMatrixf(matrix);

  // draw children
  if(m_model)
    m_model->Draw(state, m_shaders, rflags);
 
  g_QglTable.m_pfn_qglPopMatrix();
}

// ISelect

bool CEntityMiscModel::TestRay(const ray_t *ray, vec_t *dist) const
{
  vec_t dist_start = *dist;
  vec_t dist_local = *dist;
	ray_t ray_local = *ray;

  if (!aabb_intersect_ray(&m_BBox, &ray_local, &dist_local))
    return false;

  if(m_model){
    ray_transform(&ray_local, m_inverse_transform);
    dist_local = dist_start;
    if(m_model->TestRay(&ray_local, &dist_local))
        *dist = dist_local;
  } else *dist = dist_local;

  return *dist < dist_start;
}


//IEdit

void CEntityMiscModel::Translate(const vec3_t translation)
{
  VectorIncrement(translation, m_translate);
  UpdateCachedData();
}

void CEntityMiscModel::Rotate(const vec3_t pivot, const vec3_t rotation)
{
  m4x4_t rotation_matrix;

  m4x4_identity(rotation_matrix);
  m4x4_pivoted_rotate_by_vec3(rotation_matrix, rotation, pivot);
  m4x4_transform_point(rotation_matrix, m_translate);

  VectorIncrement(rotation, m_euler);

  UpdateCachedData();
}

void CEntityMiscModel::OnKeyChanged(entity_t *e, const char *key)
{
  const char *value;

  // FIXME: keys are case-sensitive?
	
	m_entity = e;

  if(strcmp(key,"model") == 0)
    SetName(ValueForKey(e,"model"));
  else if(strcmp(key,"_frame") == 0)
    SetFrame(IntForKey(e,"_frame"));
  else if(strcmp(key,"angle") == 0 || strcmp(key,"angles") == 0)
  {
    VectorSet(m_euler, 0.f, 0.f, 0.f);
    m_euler[2] = FloatForKey(e,"angle");
    value = ValueForKey(e,"angles");
    if (value[0] != '\0')
      sscanf (value, "%f %f %f", &m_euler[0], &m_euler[2], &m_euler[1]);
    UpdateCachedData();
  }
  else if(strcmp(key,"modelscale") == 0 || strcmp(key,"modelscale_vec") == 0)
  {
    VectorSet(m_scale, 1.f, 1.f, 1.f);
    value = ValueForKey(e,"modelscale");
    if (value[0] != '\0')
    {
      float f = atof(value);
      if( f != 0 )
        VectorSet(m_scale, f, f, f);
      else
        Sys_FPrintf(SYS_WRN, "WARNING: ignoring 0 modelscale key\n");
    }
    value = ValueForKey(e,"modelscale_vec");
    if (value[0] != '\0')
    {
      sscanf (value, "%f %f %f", &m_scale[0], &m_scale[1], &m_scale[2]);
      if (m_scale[0] == 0.0 && m_scale[1] == 0.0 && m_scale[2] == 0.0)
      {
        VectorSet(m_scale, 1,1,1);
        Sys_FPrintf(SYS_WRN, "WARNING: ignoring 0 0 0 modelscale_vec key\n");
      }
    }
    UpdateCachedData();
  }
  else if(strcmp(key,"origin") == 0)
  {
    value = ValueForKey(e,"origin");
    sscanf(value, "%f %f %f", &m_translate[0], &m_translate[1], &m_translate[2]); 
    UpdateCachedData();
  }
  else if(strncmp(key,"_remap",6) == 0)
  {
    unsigned int i;
    remap_t *pRemap;
    char *ch;

    value = ValueForKey(e,key);

    for(i=0; i<m_remaps->len; i++)
    {
      pRemap = (remap_t*)m_remaps->pdata[i];
      if(strcmp(key,pRemap->m_key) == 0)
        break;
    }

    if( i == m_remaps->len )
    {
      if( value[0] == '\0' )
        return;

      pRemap = new remap_t;
      g_ptr_array_add(m_remaps, pRemap);
    }
    else if( value[0] == '\0' )
    {
      g_ptr_array_remove_index_fast(m_remaps, i);
      delete pRemap;

      UpdateShaders();
      return;
    }

    strncpy(pRemap->m_remapbuff,value,sizeof(pRemap->m_remapbuff));
    strncpy(pRemap->m_key,key,sizeof(pRemap->m_key));

    pRemap->m_remap[0] = ch = pRemap->m_remapbuff;

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

    if( *ch == '\0' )
    {    
      // bad remap
      Sys_FPrintf(SYS_WRN, "WARNING: Shader _remap key found in misc_model without a ; character\n" );
      g_ptr_array_remove_index_fast(m_remaps, i);
      delete pRemap;
      return;
    }
    else
    {
      *ch = '\0';
      pRemap->m_remap[1] = ch + 1;
    }

    UpdateShaders();
  }
}

//
// CEntityMiscModel
//

// private:

void CEntityMiscModel::SetName(const char *name)
{
  if(m_name && *m_name != '\0') {
    if(strcmp(m_name, name) == 0)
      return;
    if( !g_model_cache.DeleteByNameAndFrame(m_name,m_frame) && m_model )
      m_model->RemoveParent( this );
    delete [] m_name;
  }

  m_model = NULL;
  m_name = new char[strlen(name)+1];
  strcpy(m_name,name);

  if(*m_name != '\0') {
    m_model = g_model_cache.GetByNameAndFrame(m_name, m_frame);
    m_model->AddParent( this );
  }

  UpdateCachedData();
  UpdateShaders();
}

void CEntityMiscModel::SetFrame(const int frame)
{
  if( m_frame == frame )
    return;

  if(m_name && *m_name != '\0') {
    if( !g_model_cache.DeleteByNameAndFrame(m_name,m_frame) && m_model )
      m_model->RemoveParent( this );
  }

  m_model = NULL;

  m_frame = frame;

  if(*m_name != '\0') {
    m_model = g_model_cache.GetByNameAndFrame(m_name, m_frame);
    m_model->AddParent( this );
  }

  UpdateCachedData();
}

void CEntityMiscModel::UpdateCachedData()
{
  aabb_t aabb_temp;
  bbox_t bbox_temp;

  m4x4_identity(m_transform);
  m4x4_pivoted_transform_by_vec3(m_transform, m_translate, m_euler, m_scale, m_pivot);
  memcpy(m_inverse_transform, m_transform, sizeof(m4x4_t));
  if(m4x4_invert(m_inverse_transform) == 1) {
    Sys_Printf("ERROR: Singular Matrix, cannot invert");
  }

  aabb_clear(&aabb_temp);

  if(m_model)
    aabb_extend_by_aabb(&aabb_temp, m_model->GetAABB());
  else
	{
		if (m_entity->eclass)
			VectorSet(aabb_temp.extents, m_entity->eclass->maxs[0], m_entity->eclass->maxs[1], m_entity->eclass->maxs[2]);
		else
			VectorSet(aabb_temp.extents, 8, 8, 8);
	}

  // create an oriented BBox in world-space
  bbox_for_oriented_aabb(&bbox_temp, &aabb_temp, m_transform, m_euler, m_scale);
  // create an axis aligned bbox in world-space
  aabb_for_bbox(&m_BBox, &bbox_temp);

  aabb_update_radius(&m_BBox);
}

void CEntityMiscModel::UpdateShaders()
{
  unsigned int i, j, numSurfaces;
  remap_t *pRemap, *pGlobRemap = NULL;
  char *surfShaderName;
  IShader **pShader;

  if( !m_model )
  {
    if( m_shaders->len )
    {
      // free our shaders
      for( i = 0; i < m_shaders->len; i++ )
      {
        g_ptr_array_remove_index_fast(m_shaders, i);
        (*(IShader**)m_shaders->pdata[i])->DecRef();
        delete (IShader**)m_shaders->pdata[i];
      }
    }
    return;
  }
  
  numSurfaces = m_model->GetNumSurfaces();

  if( numSurfaces < m_shaders->len )
  {
    // free unneeded shader pointers
    for( i = m_shaders->len - 1; i >= numSurfaces; i-- )
    {
      g_ptr_array_remove_index_fast(m_shaders, i);
      (*(IShader**)m_shaders->pdata[i])->DecRef();
      delete (IShader**)m_shaders->pdata[i];
    }
  }

  // now go through our surface and find our shaders, remap if needed
  for( j = 0; j < numSurfaces; j++ )
  {
    surfShaderName = m_model->GetShaderNameForSurface(j);

    if( j < m_shaders->len )
    {
      pShader = (IShader **)m_shaders->pdata[j];
    }
    else
    {
      pShader = new (IShader *);
      *pShader = NULL;
      g_ptr_array_add(m_shaders, pShader);
    }

    if( m_remaps->len )
    {
      for( i = 0; i < m_remaps->len; i++ )
      {
        pRemap = (remap_t*)m_remaps->pdata[i];
        if( stricmp(pRemap->m_remap[0],surfShaderName) == 0 )
        {
          // only do the shader lookups if really needed
          if( !(*pShader) || stricmp(pRemap->m_remap[1],(*pShader)->getName()) )
          {
            if( *pShader )
              (*pShader)->DecRef();
            *pShader = QERApp_Shader_ForName(pRemap->m_remap[1]);
          }

          pGlobRemap = NULL;
          break;
        }
        else if( pRemap->m_remap[0][0] == '*' && pRemap->m_remap[0][1] == '\0' )
          pGlobRemap = pRemap;
      }

      if( pGlobRemap )
      {
        if( !(*pShader) || stricmp(pGlobRemap->m_remap[1],(*pShader)->getName()) )
        {
          if( *pShader )
            (*pShader)->DecRef();
          *pShader = QERApp_Shader_ForName(pGlobRemap->m_remap[1]);
        }
      }
      else if( i == m_remaps->len )
      {
        // Back to the default one, if needed
        if( !(*pShader) || (stricmp(surfShaderName,(*pShader)->getName()) && !(surfShaderName[0] == '\0')) )
        {
          if( *pShader )
            (*pShader)->DecRef();
          *pShader = QERApp_Shader_ForName(surfShaderName);
        }
      }
    }
    else
    {
      // Model specified shader, if needed
      if( !(*pShader) || (stricmp(surfShaderName,(*pShader)->getName()) && !(surfShaderName[0] == '\0')) )
      {
        if( *pShader )
          (*pShader)->DecRef();
        *pShader = QERApp_Shader_ForName(surfShaderName);
      }
    }
  }
}