/*************************************************************************/
/*  grid_map.cpp                                                         */
/*************************************************************************/
/*                       This file is part of:                           */
/*                           GODOT ENGINE                                */
/*                    http://www.godotengine.org                         */
/*************************************************************************/
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur.                 */
/*                                                                       */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the       */
/* "Software"), to deal in the Software without restriction, including   */
/* without limitation the rights to use, copy, modify, merge, publish,   */
/* distribute, sublicense, and/or sell copies of the Software, and to    */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions:                                             */
/*                                                                       */
/* The above copyright notice and this permission notice shall be        */
/* included in all copies or substantial portions of the Software.       */
/*                                                                       */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,       */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF    */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
/*************************************************************************/
#include "grid_map.h"
#include "servers/visual_server.h"
#include "scene/resources/surface_tool.h"
#include "message_queue.h"
#include "scene/3d/light.h"
#include "io/marshalls.h"


bool GridMap::_set(const StringName& p_name, const Variant& p_value) {

	String name=p_name;

	if (name=="theme/theme") {

		set_theme(p_value);
	} else if (name=="cell/size") {
		set_cell_size(p_value);
	} else if (name=="cell/octant_size") {
		set_octant_size(p_value);
	} else if (name=="cell/center_x") {
		set_center_x(p_value);
	} else if (name=="cell/center_y") {
		set_center_y(p_value);
	} else if (name=="cell/center_z") {
		set_center_z(p_value);
	} else if (name=="cell/scale") {
		set_cell_scale(p_value);
	} else if (name=="theme/bake") {
		set_bake(p_value);
/*	} else if (name=="cells") {
		DVector<int> cells = p_value;
		int amount=cells.size();
		DVector<int>::Read r = cells.read();
		ERR_FAIL_COND_V(amount&1,false); // not even
		cell_map.clear();;
		for(int i=0;i<amount/3;i++) {


			IndexKey ik;
			ik.key=decode_uint64(&r[i*3]);
			Cell cell;
			cell.cell=uint32_t(r[i*+1]);
			cell_map[ik]=cell;

		}
		_recreate_octant_data();*/
	} else if (name=="data") {

		Dictionary d = p_value;

		Dictionary baked;
		if (d.has("baked"))
			baked=d["baked"];
		if (d.has("cells")) {

			DVector<int> cells = d["cells"];
			int amount=cells.size();
			DVector<int>::Read r = cells.read();
			ERR_FAIL_COND_V(amount%3,false); // not even
			cell_map.clear();;
			for(int i=0;i<amount/3;i++) {

				IndexKey ik;
				ik.key=decode_uint64((const uint8_t*)&r[i*3]);
				Cell cell;
				cell.cell=decode_uint32((const uint8_t*)&r[i*3+2]);
				cell_map[ik]=cell;

			}
		}
		baked_lock=baked.size()!=0;
		_recreate_octant_data();
		baked_lock=false;
		if (!baked.empty()) {
			List<Variant> kl;
			baked.get_key_list(&kl);
			for (List<Variant>::Element *E=kl.front();E;E=E->next()) {

				Plane ikv = E->get();
				Ref<Mesh> b=baked[ikv];
				ERR_CONTINUE(!b.is_valid());
				OctantKey ok;
				ok.x=ikv.normal.x;
				ok.y=ikv.normal.y;
				ok.z=ikv.normal.z;
				ok.area=ikv.d;

				ERR_CONTINUE(!octant_map.has(ok));

				Octant &g = *octant_map[ok];

				g.baked=b;
				g.bake_instance=VS::get_singleton()->instance_create();;
				VS::get_singleton()->instance_set_base(g.bake_instance,g.baked->get_rid());
			}
		}


	} else if (name.begins_with("areas/")) {
		int which = name.get_slice("/",1).to_int();
		String what=name.get_slice("/",2);
		if (what=="bounds") {
			ERR_FAIL_COND_V(area_map.has(which),false);
			create_area(which,p_value);
			return true;
		}

		ERR_FAIL_COND_V(!area_map.has(which),false);

		if (what=="name")
			area_set_name(which,p_value);
		else if (what=="disable_distance")
			area_set_portal_disable_distance(which,p_value);
		else if (what=="exterior_portal")
			area_set_portal_disable_color(which,p_value);
		else
			return false;
	} else
		return false;

	return true;


}

bool GridMap::_get(const StringName& p_name,Variant &r_ret) const {

	String name=p_name;

	if (name=="theme/theme") {
		r_ret= get_theme();
	} else if (name=="cell/size") {
		r_ret= get_cell_size();
	} else if (name=="cell/octant_size") {
		r_ret= get_octant_size();
	} else if (name=="cell/center_x") {
		r_ret= get_center_x();
	} else if (name=="cell/center_y") {
		r_ret= get_center_y();
	} else if (name=="cell/center_z") {
		r_ret= get_center_z();
	} else if (name=="cell/scale") {
		r_ret= cell_scale;
	} else if (name=="theme/bake") {
		r_ret= bake;
	} else if (name=="data") {

		Dictionary d;

		DVector<int> cells;
		cells.resize(cell_map.size()*3);
		{
			DVector<int>::Write w = cells.write();
			int i=0;
			for (Map<IndexKey,Cell>::Element *E=cell_map.front();E;E=E->next(),i++) {

				encode_uint64(E->key().key,(uint8_t*)&w[i*3]);
				encode_uint32(E->get().cell,(uint8_t*)&w[i*3+2]);
			}
		}

		d["cells"]=cells;

		Dictionary baked;
		for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {

			Octant &g=*E->get();

			if (g.baked.is_valid()) {

				baked[Plane(E->key().x,E->key().y,E->key().z,E->key().area)]=g.baked;
			}


		}

		if (baked.size()) {
			d["baked"]=baked;
		}

		r_ret= d;
	} else if (name.begins_with("areas/")) {
		int which = name.get_slice("/",1).to_int();
		String what=name.get_slice("/",2);
		if (what=="bounds")
			r_ret= area_get_bounds(which);
		else if (what=="name")
			r_ret= area_get_name(which);
		else if (what=="disable_distance")
			r_ret= area_get_portal_disable_distance(which);
		else if (what=="exterior_portal")
			r_ret= area_is_exterior_portal(which);
		else
			return false;
	} else
		return false;

	return true;

}

void GridMap::_get_property_list( List<PropertyInfo> *p_list) const {

	p_list->push_back( PropertyInfo( Variant::OBJECT, "theme/theme", PROPERTY_HINT_RESOURCE_TYPE, "MeshLibrary"));
	p_list->push_back( PropertyInfo( Variant::BOOL, "theme/bake"));
	p_list->push_back( PropertyInfo( Variant::REAL, "cell/size",PROPERTY_HINT_RANGE,"0.01,16384,0.01") );
	p_list->push_back( PropertyInfo( Variant::INT, "cell/octant_size",PROPERTY_HINT_RANGE,"1,1024,1") );
	p_list->push_back( PropertyInfo( Variant::BOOL, "cell/center_x") );
	p_list->push_back( PropertyInfo( Variant::BOOL, "cell/center_y") );
	p_list->push_back( PropertyInfo( Variant::BOOL, "cell/center_z") );
	p_list->push_back( PropertyInfo( Variant::REAL, "cell/scale") );

	p_list->push_back( PropertyInfo( Variant::DICTIONARY, "data", PROPERTY_HINT_NONE,"",PROPERTY_USAGE_STORAGE) );

	for(const Map<int,Area*>::Element *E=area_map.front();E;E=E->next()) {

		String base="areas/"+itos(E->key())+"/";
		p_list->push_back( PropertyInfo( Variant::_AABB, base+"bounds", PROPERTY_HINT_NONE,"",PROPERTY_USAGE_STORAGE) );
		p_list->push_back( PropertyInfo( Variant::STRING, base+"name", PROPERTY_HINT_NONE,"",PROPERTY_USAGE_STORAGE) );
		p_list->push_back( PropertyInfo( Variant::REAL, base+"disable_distance", PROPERTY_HINT_NONE,"",PROPERTY_USAGE_STORAGE) );
		p_list->push_back( PropertyInfo( Variant::COLOR, base+"disable_color", PROPERTY_HINT_NONE,"",PROPERTY_USAGE_STORAGE) );
		p_list->push_back( PropertyInfo( Variant::BOOL, base+"exterior_portal", PROPERTY_HINT_NONE,"",PROPERTY_USAGE_STORAGE) );
	}
}


void GridMap::set_theme(const Ref<MeshLibrary>& p_theme) {

	if (!theme.is_null())
		theme->unregister_owner(this);
	theme=p_theme;
	if (!theme.is_null())
		theme->register_owner(this);

	_recreate_octant_data();
	_change_notify("theme");
}

Ref<MeshLibrary> GridMap::get_theme() const{

	return theme;
}

void GridMap::set_cell_size(float p_size){

	cell_size=p_size;
	_recreate_octant_data();

}
float GridMap::get_cell_size() const{

	return cell_size;
}

void GridMap::set_octant_size(int p_size){

	octant_size=p_size;
	_recreate_octant_data();
}
int GridMap::get_octant_size() const{

	return octant_size;
}

void GridMap::set_center_x(bool p_enable) {

	center_x=p_enable;
	_recreate_octant_data();
}

bool GridMap::get_center_x() const {
	return center_x;
}

void GridMap::set_center_y(bool p_enable) {

	center_y=p_enable;
	_recreate_octant_data();
}

bool GridMap::get_center_y() const {
	return center_y;
}

void GridMap::set_center_z(bool p_enable) {

	center_z=p_enable;
	_recreate_octant_data();
}

bool GridMap::get_center_z() const {
	return center_z;
}


int GridMap::_find_area(const IndexKey& p_pos) const {

	for(const Map<int,Area*>::Element *E=area_map.front();E;E=E->next()) {
	//this should somehow be faster...
		const Area& a=*E->get();
		if (	p_pos.x>=a.from.x && p_pos.x<a.to.x &&
			p_pos.y>=a.from.y && p_pos.y<a.to.y &&
			p_pos.z>=a.from.z && p_pos.z<a.to.z ) {
			return E->key();
		}
	}

	return 0;
}
void GridMap::set_cell_item(int p_x,int p_y,int p_z, int p_item,int p_rot){

	ERR_FAIL_INDEX(ABS(p_x),1<<20);
	ERR_FAIL_INDEX(ABS(p_y),1<<20);
	ERR_FAIL_INDEX(ABS(p_z),1<<20);

	IndexKey key;
	key.x=p_x;
	key.y=p_y;
	key.z=p_z;

	OctantKey ok;
	ok.x=p_x/octant_size;
	ok.y=p_y/octant_size;
	ok.z=p_z/octant_size;
	ok.area = _find_area(key);


	if (cell_map.has(key)) {

		int prev_item=cell_map[key].item;

		OctantKey octantkey=ok;

		ERR_FAIL_COND(!octant_map.has(octantkey));
		Octant& g = *octant_map[octantkey];
		ERR_FAIL_COND(!g.items.has(prev_item));
		ERR_FAIL_COND(!g.items[prev_item].cells.has(key));


		g.items[prev_item].cells.erase(key);
		if (g.items[prev_item].cells.size()==0) {
			VS::get_singleton()->free(g.items[prev_item].multimesh_instance);
			g.items.erase(prev_item);

		}

		if (g.items.empty() || !baked_lock) {
			//unbake just in case
			if (g.baked.is_valid()) {
				VS::get_singleton()->free(g.bake_instance);
				g.bake_instance=RID();
				g.baked=Ref<Mesh>();
			}


		}
		if (g.items.empty()) {

			PhysicsServer::get_singleton()->free(g.static_body);

			memdelete(&g);
			octant_map.erase(octantkey);
		} else {


			g.dirty=true;
		}
		cell_map.erase(key);

		_queue_dirty_map();
	}

	if (p_item<0)
		return;

	OctantKey octantkey=ok;

	//add later
	if (!octant_map.has(octantkey)) {


		Octant *g = memnew( Octant );
		g->dirty=true;
		g->static_body = PhysicsServer::get_singleton()->body_create(PhysicsServer::BODY_MODE_STATIC);
		if (is_inside_world())
			PhysicsServer::get_singleton()->body_set_space(g->static_body,get_world()->get_space());

		octant_map[octantkey]=g;
	}

	Octant& g = *octant_map[octantkey];
	if (!g.items.has(p_item)) {

		Octant::ItemInstances ii;
		if (theme.is_valid() && theme->has_item(p_item)) {
			ii.mesh=theme->get_item_mesh(p_item);
			ii.shape=theme->get_item_shape(p_item);
		}
		ii.multimesh = Ref<MultiMesh>( memnew( MultiMesh ) );
		ii.multimesh->set_mesh(ii.mesh);
		ii.multimesh_instance = VS::get_singleton()->instance_create();
		VS::get_singleton()->instance_set_base(ii.multimesh_instance,ii.multimesh->get_rid());
		if (!baked_lock) {

			//unbake just in case
			if (g.bake_instance.is_valid())
				VS::get_singleton()->free(g.bake_instance);
			g.baked=Ref<Mesh>();
			if (is_inside_world()) {
				VS::get_singleton()->instance_set_scenario(ii.multimesh_instance,get_world()->get_scenario());
				if (ok.area) {
					VS::get_singleton()->instance_set_room( ii.multimesh_instance,area_map[ok.area]->instance);
				}
			}
		}
		g.items[p_item]=ii;
	}

	Octant::ItemInstances &ii = g.items[p_item];
	ii.cells.insert(key);
	g.dirty=true;

	_queue_dirty_map();

	cell_map[key]=Cell();
	Cell &c=cell_map[key];
	c.item=p_item;
	c.rot=p_rot;

}

int GridMap::get_cell_item(int p_x,int p_y,int p_z) const{

	ERR_FAIL_INDEX_V(ABS(p_x),1<<20,INVALID_CELL_ITEM);
	ERR_FAIL_INDEX_V(ABS(p_y),1<<20,INVALID_CELL_ITEM);
	ERR_FAIL_INDEX_V(ABS(p_z),1<<20,INVALID_CELL_ITEM);


	IndexKey key;
	key.x=p_x;
	key.y=p_y;
	key.z=p_z;

	if (!cell_map.has(key))
		return INVALID_CELL_ITEM;
	return cell_map[key].item;

}

int GridMap::get_cell_item_orientation(int p_x,int p_y,int p_z) const{

	ERR_FAIL_INDEX_V(ABS(p_x),1<<20,-1);
	ERR_FAIL_INDEX_V(ABS(p_y),1<<20,-1);
	ERR_FAIL_INDEX_V(ABS(p_z),1<<20,-1);

	IndexKey key;
	key.x=p_x;
	key.y=p_y;
	key.z=p_z;

	if (!cell_map.has(key))
		return -1;
	return cell_map[key].rot;

}

void GridMap::_octant_enter_world(const OctantKey &p_key) {

	ERR_FAIL_COND(!octant_map.has(p_key));
	Octant&g = *octant_map[p_key];
	PhysicsServer::get_singleton()->body_set_state(g.static_body,PhysicsServer::BODY_STATE_TRANSFORM,get_global_transform());
	PhysicsServer::get_singleton()->body_set_space(g.static_body,get_world()->get_space());
	//print_line("BODYPOS: "+get_global_transform());


	if (g.baked.is_valid()) {

		Transform xf = get_global_transform();
		xf.translate(_octant_get_offset(p_key));

		VS::get_singleton()->instance_set_transform(g.bake_instance,xf);
		VS::get_singleton()->instance_set_scenario(g.bake_instance,get_world()->get_scenario());
		if (area_map.has(p_key.area)) {
			VS::get_singleton()->instance_set_room(g.bake_instance,area_map[p_key.area]->instance);

		}
	} else {
		for(Map<int,Octant::ItemInstances>::Element *E=g.items.front();E;E=E->next()) {

			VS::get_singleton()->instance_set_scenario(E->get().multimesh_instance,get_world()->get_scenario());
			VS::get_singleton()->instance_set_transform(E->get().multimesh_instance,get_global_transform());
			//print_line("INSTANCEPOS: "+get_global_transform());

			if (area_map.has(p_key.area)) {
				VS::get_singleton()->instance_set_room(E->get().multimesh_instance,area_map[p_key.area]->instance);
			}
		}
	}

}


void GridMap::_octant_transform(const OctantKey &p_key) {

	ERR_FAIL_COND(!octant_map.has(p_key));
	Octant&g = *octant_map[p_key];
	PhysicsServer::get_singleton()->body_set_state(g.static_body,PhysicsServer::BODY_STATE_TRANSFORM,get_global_transform());

	if (g.baked.is_valid()) {

		Transform xf = get_global_transform();
		xf.origin+=_octant_get_offset(p_key);
		VS::get_singleton()->instance_set_transform(g.bake_instance,xf);
	} else {
		for(Map<int,Octant::ItemInstances>::Element *E=g.items.front();E;E=E->next()) {

			VS::get_singleton()->instance_set_transform(E->get().multimesh_instance,get_global_transform());
			//print_line("UPDATEPOS: "+get_global_transform());
		}
	}

}


void GridMap::_octant_update(const OctantKey &p_key) {

	ERR_FAIL_COND(!octant_map.has(p_key));
	Octant&g = *octant_map[p_key];
	if (!g.dirty)
		return;

	Ref<Mesh> mesh;

	PhysicsServer::get_singleton()->body_clear_shapes(g.static_body);

	for(Map<int,Octant::ItemInstances>::Element *E=g.items.front();E;E=E->next()) {

		Octant::ItemInstances &ii=E->get();
		ii.multimesh->set_instance_count(ii.cells.size());


		AABB aabb;
		AABB mesh_aabb = ii.mesh.is_null()?AABB():ii.mesh->get_aabb();

		Vector3 ofs(cell_size*0.5*int(center_x),cell_size*0.5*int(center_y),cell_size*0.5*int(center_z));


		//print_line("OCTANT, CELLS: "+itos(ii.cells.size()));
		int idx=0;
		for(Set<IndexKey>::Element *F=ii.cells.front();F;F=F->next()) {
			IndexKey ik=F->get();
			Map<IndexKey,Cell>::Element *C=cell_map.find(ik);
			ERR_CONTINUE(!C);

			Vector3 cellpos = Vector3(ik.x,ik.y,ik.z );

			Transform xform;

			if (clip && ( (clip_above && cellpos[clip_axis]>clip_floor) || (!clip_above && cellpos[clip_axis]<clip_floor))) {

				xform.basis.set_zero();

			} else {

				xform.basis.set_orthogonal_index(C->get().rot);
			}


			xform.set_origin( cellpos*cell_size+ofs);
			xform.basis.scale(Vector3(cell_scale,cell_scale,cell_scale));

			ii.multimesh->set_instance_transform(idx,xform);
			ii.multimesh->set_instance_color(idx,Color(1,1,1,1));
			//print_line("MMINST: "+xform);


			if(idx==0) {

				aabb=xform.xform(mesh_aabb);
			} else {

				aabb.merge_with(xform.xform(mesh_aabb));
			}

			if (ii.shape.is_valid()) {

				PhysicsServer::get_singleton()->body_add_shape(g.static_body,ii.shape->get_rid(),xform);
			//	print_line("PHIS x: "+xform);

			}

			idx++;
		}

		ii.multimesh->set_aabb(aabb);


	}

	g.dirty=false;

}


void GridMap::_octant_exit_world(const OctantKey &p_key) {

	ERR_FAIL_COND(!octant_map.has(p_key));
	Octant&g = *octant_map[p_key];
	PhysicsServer::get_singleton()->body_set_state(g.static_body,PhysicsServer::BODY_STATE_TRANSFORM,get_global_transform());
	PhysicsServer::get_singleton()->body_set_space(g.static_body,RID());


	if (g.baked.is_valid()) {

		VS::get_singleton()->instance_set_room(g.bake_instance,RID());
		VS::get_singleton()->instance_set_scenario(g.bake_instance,RID());

	}

	for(Map<int,Octant::ItemInstances>::Element *E=g.items.front();E;E=E->next()) {

		VS::get_singleton()->instance_set_scenario(E->get().multimesh_instance,RID());
	//	VS::get_singleton()->instance_set_transform(E->get().multimesh_instance,get_global_transform());
		VS::get_singleton()->instance_set_room(E->get().multimesh_instance,RID());
	}

}

void GridMap::_octant_clear_baked(const OctantKey &p_key) {


	ERR_FAIL_COND(!octant_map.has(p_key));
	Octant&g = *octant_map[p_key];

	if (!g.baked.is_valid())
		return;

	VS::get_singleton()->free(g.bake_instance);
	g.bake_instance=RID();
	g.baked=Ref<Mesh>();

	if (is_inside_scene())
		_octant_enter_world(p_key);
	g.dirty=true;
	_queue_dirty_map();
}

void GridMap::_octant_bake(const OctantKey &p_key, const Ref<TriangleMesh>& p_tmesh,const Vector<BakeLight> &p_lights,List<Vector3> *p_prebake) {


	ERR_FAIL_COND(!octant_map.has(p_key));
	Octant&g = *octant_map[p_key];

	Ref<TriangleMesh> tm=p_tmesh;
	if (!p_prebake && is_inside_world())
		_octant_exit_world(p_key);

	Map< Ref<Material>, Ref<SurfaceTool> > surfaces;
	Vector3 ofs(cell_size*0.5*int(center_x),cell_size*0.5*int(center_y),cell_size*0.5*int(center_z));
	Vector3 octant_ofs=_octant_get_offset(p_key);

	for(Map<int,Octant::ItemInstances>::Element *E=g.items.front();E;E=E->next()) {

		Octant::ItemInstances &ii=E->get();

		if (ii.mesh.is_null())
			continue;

		for(Set<IndexKey>::Element *F=ii.cells.front();F;F=F->next()) {

			IndexKey ik=F->get();
			Map<IndexKey,Cell>::Element *C=cell_map.find(ik);
			ERR_CONTINUE(!C);
			Vector3 cellpos = Vector3(ik.x,ik.y,ik.z );

			Transform xform;
			xform.basis.set_orthogonal_index(C->get().rot);
			xform.set_origin( cellpos*cell_size+ofs);
			if (!p_prebake)
				xform.origin-=octant_ofs;


			for(int i=0;i<ii.mesh->get_surface_count();i++) {

				if (p_prebake) {

					if (ii.mesh->surface_get_primitive_type(i)!=Mesh::PRIMITIVE_TRIANGLES)
						continue;
					Array a = ii.mesh->surface_get_arrays(i);
					DVector<Vector3> av=a[VS::ARRAY_VERTEX];
					int avs = av.size();
					DVector<Vector3>::Read vr = av.read();

					DVector<int> ai=a[VS::ARRAY_INDEX];
					int ais=ai.size();
					if (ais) {

						DVector<int>::Read ir=ai.read();
						for(int j=0;j<ais;j++) {

							p_prebake->push_back(xform.xform(vr[ir[j]]));
							//print_line("V SET: "+xform.xform(vr[ir[j]]));
						}

					} else {

						for(int j=0;j<avs;j++) {

							p_prebake->push_back(xform.xform(vr[j]));
						}
					}

				} else {

					Ref<Material> m = ii.mesh->surface_get_material(i);

					Map< Ref<Material>, Ref<SurfaceTool> >::Element *S=surfaces.find(m);

					if (!S) {

						S=surfaces.insert(m,Ref<SurfaceTool>( memnew( SurfaceTool )));
					}

					Ref<SurfaceTool> st = S->get();
					List<SurfaceTool::Vertex>::Element *V=st->get_vertex_array().back();
					st->append_from(ii.mesh,i,xform);
					st->set_material(m);


					if (tm.is_valid()) {

						if (V)
							V=V->next();
						else
							V=st->get_vertex_array().front();;
						int lc = p_lights.size();
						const BakeLight* bl = p_lights.ptr();
						float ofs = cell_size*0.02;
						float att = 0.2;


						for(;V;V=V->next()) {

							SurfaceTool::Vertex &v=V->get();

							Vector3 vertex = v.vertex + octant_ofs;
							//print_line("V GET: "+vertex);
							Vector3 normal = tm->get_area_normal( AABB( Vector3(-ofs,-ofs,-ofs)+vertex,Vector3(ofs,ofs,ofs)*2.0));
							if (normal==Vector3()) {
								print_line("couldn't find for vertex: "+vertex);
							}
							ERR_CONTINUE( normal== Vector3());

							float max_l=1.0;
							float max_dist=1.0;

							if (lc) {

								for(int j=0;j<lc;j++) {
									const BakeLight &l=bl[j];
									switch(l.type) {
										case VS::LIGHT_DIRECTIONAL: {

											Vector3 ray_from=vertex + normal *ofs;
											Vector3 ray_to=l.dir*5000;
											Vector3 n;
											Vector3 p;
											if (tm->intersect_segment(ray_from,ray_to,p,n)) {

												float dist = 1.0-l.param[VS::LIGHT_PARAM_SHADOW_DARKENING];
												if (dist<=max_dist) {
													max_dist=dist;
													max_l=1.0-dist;
												}
											}
										} break;
									}

								}
							}

							v.color=Color(max_l,max_l,max_l,1.0);

						}

						st->add_to_format(VS::ARRAY_FORMAT_COLOR);
						if (m.is_valid()) {
							Ref<FixedMaterial> fm = m;
							if (fm.is_valid())
								fm->set_fixed_flag(FixedMaterial::FLAG_USE_COLOR_ARRAY,true);
						}
					}
				}
			}
		}
	}

	if (p_prebake)
		return;

	g.baked = Ref<Mesh>( memnew( Mesh ));

	for(Map< Ref<Material>, Ref<SurfaceTool> >::Element *E=surfaces.front();E;E=E->next()) {

		Ref<SurfaceTool> st = E->get();
		st->commit(g.baked);
	}

	g.bake_instance = VS::get_singleton()->instance_create();
	VS::get_singleton()->instance_set_base(g.bake_instance,g.baked->get_rid());

	if (is_inside_world())
		_octant_enter_world(p_key);

	g.dirty=true;
	_queue_dirty_map();
}

void GridMap::_notification(int p_what) {


	switch(p_what) {

		case NOTIFICATION_ENTER_WORLD: {

			_update_area_instances();

			for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {
//				IndexKey ik;
//				ik.key = E->key().indexkey;
				_octant_enter_world(E->key());
				_octant_update(E->key());
			}

			awaiting_update=false;

			last_transform=get_global_transform();
		} break;
		case NOTIFICATION_TRANSFORM_CHANGED: {

			Transform new_xform = get_global_transform();
			if (new_xform==last_transform)
				break;
			//update run
			for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {
				_octant_transform(E->key());
			}

			last_transform=new_xform;

		} break;
		case NOTIFICATION_EXIT_WORLD: {

			for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {
				_octant_exit_world(E->key());
			}

			//_queue_dirty_map(MAP_DIRTY_INSTANCES|MAP_DIRTY_TRANSFORMS);
			//_update_dirty_map_callback();
			//_update_area_instances();

		} break;
	}
}



void GridMap::_queue_dirty_map() {

	if (awaiting_update)
		return;	

	if (is_inside_world()) {

		MessageQueue::get_singleton()->push_call(this,"_update_dirty_map_callback");
		awaiting_update=true;
	}
}

void GridMap::_recreate_octant_data() {

	Map<IndexKey,Cell> cell_copy=cell_map;
	_clear_internal(true);
	for (Map<IndexKey,Cell>::Element *E=cell_copy.front();E;E=E->next()) {

		set_cell_item(E->key().x,E->key().y,E->key().z,E->get().item,E->get().rot);
	}

}

void GridMap::_clear_internal(bool p_keep_areas) {

	for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {
		if (is_inside_world())
			_octant_exit_world(E->key());

		for (Map<int,Octant::ItemInstances>::Element *F=E->get()->items.front();F;F=F->next()) {

			VS::get_singleton()->free(F->get().multimesh_instance);
		}

		//unbake just in case
		if (E->get()->bake_instance.is_valid())
			VS::get_singleton()->free(E->get()->bake_instance);

		PhysicsServer::get_singleton()->free(E->get()->static_body);
		memdelete(E->get());

	}

	octant_map.clear();
	cell_map.clear();

	if (p_keep_areas)
		return;

	for (Map<int,Area*>::Element *E=area_map.front();E;E=E->next()) {


		VS::get_singleton()->free(E->get()->base_portal);
		VS::get_singleton()->free(E->get()->instance);
		for(int i=0;i<E->get()->portals.size();i++) {
			VS::get_singleton()->free(E->get()->portals[i].instance);
		}

		memdelete(E->get());
	}

}

void GridMap::clear() {

	_clear_internal();

}

void GridMap::resource_changed(const RES& p_res) {

	_recreate_octant_data();
}


void GridMap::_update_dirty_map_callback() {

	if (!awaiting_update)
		return;

	for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {
		_octant_update(E->key());
	}


	awaiting_update=false;

}


void GridMap::_bind_methods() {

	ObjectTypeDB::bind_method(_MD("set_theme","theme:MeshLibrary"),&GridMap::set_theme);
	ObjectTypeDB::bind_method(_MD("get_theme:MeshLibrary"),&GridMap::get_theme);

	ObjectTypeDB::bind_method(_MD("set_bake","enable"),&GridMap::set_bake);
	ObjectTypeDB::bind_method(_MD("is_baking_enabled"),&GridMap::is_baking_enabled);

	ObjectTypeDB::bind_method(_MD("set_cell_size","size"),&GridMap::set_cell_size);
	ObjectTypeDB::bind_method(_MD("get_cell_size"),&GridMap::get_cell_size);

	ObjectTypeDB::bind_method(_MD("set_octant_size","size"),&GridMap::set_octant_size);
	ObjectTypeDB::bind_method(_MD("get_octant_size"),&GridMap::get_octant_size);

	ObjectTypeDB::bind_method(_MD("set_cell_item","x","y","z","item","orientation"),&GridMap::set_cell_item,DEFVAL(0));
	ObjectTypeDB::bind_method(_MD("get_cell_item","x","y","z"),&GridMap::get_cell_item);
	ObjectTypeDB::bind_method(_MD("get_cell_item_orientation","x","y","z"),&GridMap::get_cell_item_orientation);

//	ObjectTypeDB::bind_method(_MD("_recreate_octants"),&GridMap::_recreate_octants);
	ObjectTypeDB::bind_method(_MD("_update_dirty_map_callback"),&GridMap::_update_dirty_map_callback);
	ObjectTypeDB::bind_method(_MD("resource_changed"),&GridMap::resource_changed);

	ObjectTypeDB::bind_method(_MD("set_center_x","enable"),&GridMap::set_center_x);
	ObjectTypeDB::bind_method(_MD("get_center_x"),&GridMap::get_center_x);
	ObjectTypeDB::bind_method(_MD("set_center_y","enable"),&GridMap::set_center_y);
	ObjectTypeDB::bind_method(_MD("get_center_y"),&GridMap::get_center_y);
	ObjectTypeDB::bind_method(_MD("set_center_z","enable"),&GridMap::set_center_z);
	ObjectTypeDB::bind_method(_MD("get_center_z"),&GridMap::get_center_z);

	ObjectTypeDB::bind_method(_MD("set_clip","enabled","clipabove","floor","axis"),&GridMap::set_clip,DEFVAL(true),DEFVAL(0),DEFVAL(Vector3::AXIS_X));

	ObjectTypeDB::bind_method(_MD("crate_area","id","area"),&GridMap::create_area);
	ObjectTypeDB::bind_method(_MD("area_get_bounds","area","bounds"),&GridMap::area_get_bounds);
	ObjectTypeDB::bind_method(_MD("area_set_exterior_portal","area","enable"),&GridMap::area_set_exterior_portal);
	ObjectTypeDB::bind_method(_MD("area_set_name","area","name"),&GridMap::area_set_name);
	ObjectTypeDB::bind_method(_MD("area_get_name","area"),&GridMap::area_get_name);
	ObjectTypeDB::bind_method(_MD("area_is_exterior_portal","area"),&GridMap::area_is_exterior_portal);
	ObjectTypeDB::bind_method(_MD("area_set_portal_disable_distance","area","distance"),&GridMap::area_set_portal_disable_distance);
	ObjectTypeDB::bind_method(_MD("area_get_portal_disable_distance","area"),&GridMap::area_get_portal_disable_distance);
	ObjectTypeDB::bind_method(_MD("area_set_portal_disable_color","area","color"),&GridMap::area_set_portal_disable_color);
	ObjectTypeDB::bind_method(_MD("area_get_portal_disable_color","area"),&GridMap::area_get_portal_disable_color);
	ObjectTypeDB::bind_method(_MD("erase_area","area"),&GridMap::erase_area);
	ObjectTypeDB::bind_method(_MD("get_unused_area_id","area"),&GridMap::get_unused_area_id);
	ObjectTypeDB::bind_method(_MD("bake_geometry"),&GridMap::bake_geometry);

	ObjectTypeDB::set_method_flags("GridMap","bake_geometry",METHOD_FLAGS_DEFAULT|METHOD_FLAG_EDITOR);

	ObjectTypeDB::bind_method(_MD("clear"),&GridMap::clear);

	BIND_CONSTANT( INVALID_CELL_ITEM );

}

void GridMap::set_clip(bool p_enabled, bool p_clip_above, int p_floor, Vector3::Axis p_axis) {

	if (!p_enabled && !clip)
		return;
	if (clip && p_enabled && clip_floor==p_floor && p_clip_above==clip_above && p_axis==clip_axis)
		return;

	clip=p_enabled;
	clip_floor=p_floor;
	clip_axis=p_axis;
	clip_above=p_clip_above;

	//make it all update
	for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {

		Octant *g=E->get();
		g->dirty=true;

	}
	awaiting_update=true;
	_update_dirty_map_callback();
}


void GridMap::_update_areas() {

	//clear the portals
	for(Map<int,Area*>::Element *E=area_map.front();E;E=E->next()) {
		//this should somehow be faster...
		Area& a=*E->get();
		a.portals.clear();
		if (a.instance.is_valid()) {
			VisualServer::get_singleton()->free(a.instance);
			a.instance=RID();
		}
	}

	//test all areas against all areas and create portals - this sucks (slow :( )
	for(Map<int,Area*>::Element *E=area_map.front();E;E=E->next()) {
		Area& a=*E->get();
		if (a.exterior_portal) //that's pretty much all it does... yes it is
			continue;
		Vector3 from_a(a.from.x,a.from.y,a.from.z);
		Vector3 to_a(a.to.x,a.to.y,a.to.z);

		for(Map<int,Area*>::Element *F=area_map.front();F;F=F->next()) {

			Area& b=*F->get();
			Vector3 from_b(b.from.x,b.from.y,b.from.z);
			Vector3 to_b(b.to.x,b.to.y,b.to.z);

			// initially test intersection and discards
			int axis=-1;
			float sign=0;
			bool valid=true;
			Vector3 axmin,axmax;


			for(int i=0;i<3;i++) {

				if (from_a[i]==to_b[i]) {

					if (axis!=-1) {
						valid=false;
						break;
					}

					axis=i;
					sign=-1;
				} else if (from_b[i]==to_a[i]) {

					if (axis!=-1) {
						valid=false;
						break;
					}
					axis=i;
					sign=+1;
				}


				if (from_a[i] > to_b[i] || to_a[i] < from_b[i] ) {
					valid=false;
					break;
				} else {

					axmin[i]= ( from_a[i] > from_b[i] ) ? from_a[i] :from_b[i];
					axmax[i]= ( to_a[i] < to_b[i] ) ? to_a[i] :to_b[i];

				}


			}

			if (axis==-1 || !valid)
				continue;

			Transform xf;


			for(int i=0;i<3;i++) {



				int ax=(axis+i)%3;
				Vector3 axis_vec;
				float scale = (i==0)?sign:((axmax[ax]-axmin[ax])*cell_size);
				axis_vec[ax]=scale;
				xf.basis.set_axis((2+i)%3,axis_vec);
				xf.origin[i]=axmin[i]*cell_size;

			}



			Area::Portal portal;
			portal.xform=xf;
			a.portals.push_back(portal);
		}
	}

	_update_area_instances();

}

void GridMap::_update_area_instances() {

	Transform base_xform;
	if (_in_tree)
		base_xform=get_global_transform();

	for(Map<int,Area*>::Element *E=area_map.front();E;E=E->next()) {
		//this should somehow be faster...
		Area& a=*E->get();
		if (a.instance.is_valid()!=_in_tree) {

			if (!_in_tree) {

				for(int i=0;i<a.portals.size();i++) {

					Area::Portal&p=a.portals[i];
					ERR_CONTINUE(!p.instance.is_valid());
					VisualServer::get_singleton()->free(p.instance);
					p.instance=RID();
				}

				VisualServer::get_singleton()->free(a.instance);
				a.instance=RID();

			} else {

				//a.instance = VisualServer::get_singleton()->instance_create2(base_room,get_world()->get_scenario());
				for(int i=0;i<a.portals.size();i++) {

					Area::Portal&p=a.portals[i];
					ERR_CONTINUE(p.instance.is_valid());
					p.instance=VisualServer::get_singleton()->instance_create2(a.base_portal,get_world()->get_scenario());
					VisualServer::get_singleton()->instance_set_room(p.instance,a.instance);
				}
			}
		}

		if (a.instance.is_valid()) {
			Transform xform;

			Vector3 from_a(a.from.x,a.from.y,a.from.z);
			Vector3 to_a(a.to.x,a.to.y,a.to.z);

			for(int i=0;i<3;i++) {
				xform.origin[i]=from_a[i]*cell_size;
				Vector3 s;
				s[i]=(to_a[i]-from_a[i])*cell_size;
				xform.basis.set_axis(i,s);
			}


			VisualServer::get_singleton()->instance_set_transform(a.instance,base_xform * xform);

			for(int i=0;i<a.portals.size();i++) {

				Area::Portal&p=a.portals[i];
				ERR_CONTINUE(!p.instance.is_valid());

				VisualServer::get_singleton()->instance_set_transform(p.instance,base_xform * xform);

			}

		}
	}

}

Error GridMap::create_area(int p_id,const AABB& p_bounds) {

	ERR_FAIL_COND_V(area_map.has(p_id),ERR_ALREADY_EXISTS);
	ERR_EXPLAIN("ID 0 is taken as global area, start from 1");
	ERR_FAIL_COND_V(p_id==0,ERR_INVALID_PARAMETER);
	ERR_FAIL_COND_V(p_bounds.has_no_area(),ERR_INVALID_PARAMETER);

	// FIRST VALIDATE AREA
	IndexKey from,to;
	from.x=p_bounds.pos.x;
	from.y=p_bounds.pos.y;
	from.z=p_bounds.pos.z;
	to.x=p_bounds.pos.x+p_bounds.size.x;
	to.y=p_bounds.pos.y+p_bounds.size.y;
	to.z=p_bounds.pos.z+p_bounds.size.z;


	for(Map<int,Area*>::Element *E=area_map.front();E;E=E->next()) {
		//this should somehow be faster...
		Area& a=*E->get();

		//does it interset with anything else?

		if (	from.x >= a.to.x ||
			to.x <= a.from.x ||
			from.y >= a.to.y ||
			to.y <= a.from.y ||
			from.z >= a.to.z ||
			to.z <= a.from.z ) {

			// all good
		} else {

			return ERR_INVALID_PARAMETER;
		}
	}


	Area *area = memnew( Area );
	area->from=from;
	area->to=to;
	area->portal_disable_distance=0;
	area->exterior_portal=false;
	area->name="Area "+itos(p_id);
	area_map[p_id]=area;
	_recreate_octant_data();
	return OK;
}

AABB GridMap::area_get_bounds(int p_area) const {

	ERR_FAIL_COND_V(!area_map.has(p_area),AABB());

	const Area *a = area_map[p_area];
	AABB aabb;
	aabb.pos=Vector3(a->from.x,a->from.y,a->from.z);
	aabb.size=Vector3(a->to.x,a->to.y,a->to.z)-aabb.pos;

	return aabb;
}

void GridMap::area_set_name(int p_area,const String& p_name) {

	ERR_FAIL_COND(!area_map.has(p_area));

	Area *a = area_map[p_area];
	a->name=p_name;
}

String GridMap::area_get_name(int p_area) const {

	ERR_FAIL_COND_V(!area_map.has(p_area),"");

	const Area *a = area_map[p_area];
	return a->name;
}


void GridMap::area_set_exterior_portal(int p_area,bool p_enable) {

	ERR_FAIL_COND(!area_map.has(p_area));

	Area *a = area_map[p_area];
	if (a->exterior_portal==p_enable)
		return;
	a->exterior_portal=p_enable;

	_recreate_octant_data();
}

bool GridMap::area_is_exterior_portal(int p_area) const {

	ERR_FAIL_COND_V(!area_map.has(p_area),false);

	const Area *a = area_map[p_area];
	return a->exterior_portal;
}

void GridMap::area_set_portal_disable_distance(int p_area, float p_distance) {

	ERR_FAIL_COND(!area_map.has(p_area));

	Area *a = area_map[p_area];
	a->portal_disable_distance=p_distance;

}

float GridMap::area_get_portal_disable_distance(int p_area) const {

	ERR_FAIL_COND_V(!area_map.has(p_area),0);

	const Area *a = area_map[p_area];
	return a->portal_disable_distance;
}

void GridMap::area_set_portal_disable_color(int p_area, Color p_color) {

	ERR_FAIL_COND(!area_map.has(p_area));

	Area *a = area_map[p_area];
	a->portal_disable_color=p_color;

}

Color GridMap::area_get_portal_disable_color(int p_area) const {

	ERR_FAIL_COND_V(!area_map.has(p_area),Color());

	const Area *a = area_map[p_area];
	return a->portal_disable_color;
}

void GridMap::get_area_list(List<int> *p_areas) const {

	for(const Map<int,Area*>::Element *E=area_map.front();E;E=E->next()) {

		p_areas->push_back(E->key());
	}

}


GridMap::Area::Portal::~Portal() {

	if (instance.is_valid())
		VisualServer::get_singleton()->free(instance);
}


GridMap::Area::Area() {

	base_portal=VisualServer::get_singleton()->portal_create();
	Vector< Point2 > points;
	points.push_back( Point2( 0, 1 ) );
	points.push_back( Point2( 1, 1 ) );
	points.push_back( Point2( 1, 0 ) );
	points.push_back( Point2( 0, 0 ) );
	VisualServer::get_singleton()->portal_set_shape(base_portal,points);

}


GridMap::Area::~Area() {

	if (instance.is_valid())
		VisualServer::get_singleton()->free(instance);
	VisualServer::get_singleton()->free(base_portal);
}

void GridMap::erase_area(int p_area) {

	ERR_FAIL_COND(!area_map.has(p_area));

	Area* a=area_map[p_area];
	memdelete(a);
	area_map.erase(p_area);
	_recreate_octant_data();
}

int GridMap::get_unused_area_id() const {

	if (area_map.empty())
		return 1;
	else
		return area_map.back()->key()+1;
}


void GridMap::set_bake(bool p_bake) {

	bake=p_bake;
	if (bake==false) {
		for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {

			_octant_clear_baked(E->key());
		}
	}
}

bool GridMap::is_baking_enabled() const {

	return bake;
}

void GridMap::set_cell_scale(float p_scale) {

	cell_scale=p_scale;
	_queue_dirty_map();
}

float GridMap::get_cell_scale() const{

	return cell_scale;
}



void GridMap::bake_geometry() {

	//used to compute vertex occlusion
	Ref<TriangleMesh> tmesh;
	Vector<BakeLight> lights;

	if (true) {

		List<Vector3> vertices;

		for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {
			_octant_bake(E->key(),tmesh,lights,&vertices);

		}

		DVector<Vector3> vv;
		vv.fill_with(vertices);
		//print_line("TOTAL VERTICES: "+itos(vv.size()));
		tmesh = Ref<TriangleMesh>( memnew( TriangleMesh ));
		tmesh->create(vv);


		for(int i=0;i<get_child_count();i++) {

			if (get_child(i)->cast_to<Light>()) {
				Light *l = get_child(i)->cast_to<Light>();
				BakeLight bl;
				for(int i=0;i<Light::PARAM_MAX;i++) {
					bl.param[i]=l->get_parameter(Light::Parameter(i));
				}
				Transform t=l->get_global_transform();
				bl.pos=t.origin;
				bl.dir=t.basis.get_axis(2);
				bl.type=l->get_light_type();
				lights.push_back(bl);

			}
		}
	}

	int idx=0;
	for(Map<OctantKey,Octant*>::Element *E=octant_map.front();E;E=E->next()) {
		if (E->get()->baked.is_valid())
			_octant_clear_baked(E->key());

		_octant_bake(E->key(),tmesh,lights);
		print_line("baking "+itos(idx)+"/"+itos(octant_map.size()));
		idx++;
	}

}



GridMap::GridMap() {

	cell_size=2;
	octant_size=4;
	awaiting_update=false;
	_in_tree=false;
	center_x=true;
	center_y=true;
	center_z=true;

	clip=false;
	clip_floor=0;
	clip_axis=Vector3::AXIS_Z;
	clip_above=true;
	baked_lock=false;
	bake=false;
	cell_scale=1.0;




}


GridMap::~GridMap() {

	if (!theme.is_null())
		theme->unregister_owner(this);

	clear();

}
