nuclear@25: #include <stdio.h>
nuclear@25: #include "meshgen.h"
nuclear@25: #include "mesh.h"
nuclear@25: 
nuclear@25: // -------- sphere --------
nuclear@25: 
nuclear@25: #define SURAD(u)	((u) * 2.0 * M_PI)
nuclear@25: #define SVRAD(v)	((v) * M_PI)
nuclear@25: 
nuclear@25: static Vector3 sphvec(float theta, float phi)
nuclear@25: {
nuclear@25: 	return Vector3(sin(theta) * sin(phi),
nuclear@25: 			cos(phi),
nuclear@25: 			cos(theta) * sin(phi));
nuclear@25: }
nuclear@25: 
nuclear@25: void gen_sphere(Mesh *mesh, float rad, int usub, int vsub, float urange, float vrange)
nuclear@25: {
nuclear@25: 	if(usub < 4) usub = 4;
nuclear@25: 	if(vsub < 2) vsub = 2;
nuclear@25: 
nuclear@25: 	int uverts = usub + 1;
nuclear@25: 	int vverts = vsub + 1;
nuclear@25: 
nuclear@25: 	int num_verts = uverts * vverts;
nuclear@25: 	int num_quads = usub * vsub;
nuclear@25: 	int num_tri = num_quads * 2;
nuclear@25: 
nuclear@25: 	mesh->clear();
nuclear@25: 	Vector3 *varr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_VERTEX, 3, num_verts, 0);
nuclear@25: 	Vector3 *narr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_NORMAL, 3, num_verts, 0);
nuclear@25: 	Vector3 *tarr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_TANGENT, 3, num_verts, 0);
nuclear@25: 	Vector2 *uvarr = (Vector2*)mesh->set_attrib_data(MESH_ATTR_TEXCOORD, 2, num_verts, 0);
nuclear@25: 	unsigned int *idxarr = mesh->set_index_data(num_tri * 3, 0);
nuclear@25: 
nuclear@25: 	float du = urange / (float)(uverts - 1);
nuclear@25: 	float dv = vrange / (float)(vverts - 1);
nuclear@25: 
nuclear@25: 	float u = 0.0;
nuclear@25: 	for(int i=0; i<uverts; i++) {
nuclear@25: 		float theta = SURAD(u * urange);
nuclear@25: 
nuclear@25: 		float v = 0.0;
nuclear@25: 		for(int j=0; j<vverts; j++) {
nuclear@25: 			float phi = SVRAD(v * vrange);
nuclear@25: 
nuclear@25: 			Vector3 pos = sphvec(theta, phi);
nuclear@25: 
nuclear@25: 			*varr++ = pos * rad;
nuclear@25: 			*narr++ = pos;
nuclear@25: 			*tarr++ = (sphvec(theta + 0.1f, (float)M_PI / 2.0f) - sphvec(theta - 0.1f, (float)M_PI / 2.0f)).normalized();
nuclear@25: 			*uvarr++ = Vector2(u * urange, v * vrange);
nuclear@25: 
nuclear@25: 			if(i < usub && j < vsub) {
nuclear@25: 				int idx = i * vverts + j;
nuclear@25: 				*idxarr++ = idx;
nuclear@25: 				*idxarr++ = idx + 1;
nuclear@25: 				*idxarr++ = idx + vverts + 1;
nuclear@25: 
nuclear@25: 				*idxarr++ = idx;
nuclear@25: 				*idxarr++ = idx + vverts + 1;
nuclear@25: 				*idxarr++ = idx + vverts;
nuclear@25: 			}
nuclear@25: 
nuclear@25: 			v += dv;
nuclear@25: 		}
nuclear@25: 		u += du;
nuclear@25: 	}
nuclear@25: }
nuclear@25: 
nuclear@25: 
nuclear@25: // -------- cylinder --------
nuclear@25: 
nuclear@25: static Vector3 cylvec(float theta, float height)
nuclear@25: {
nuclear@25: 	return Vector3(sin(theta), height, cos(theta));
nuclear@25: }
nuclear@25: 
nuclear@25: void gen_cylinder(Mesh *mesh, float rad, float height, int usub, int vsub, int capsub, float urange, float vrange)
nuclear@25: {
nuclear@25: 	if(usub < 4) usub = 4;
nuclear@25: 	if(vsub < 1) vsub = 1;
nuclear@25: 
nuclear@25: 	int uverts = usub + 1;
nuclear@25: 	int vverts = vsub + 1;
nuclear@25: 
nuclear@25: 	int num_body_verts = uverts * vverts;
nuclear@25: 	int num_body_quads = usub * vsub;
nuclear@25: 	int num_body_tri = num_body_quads * 2;
nuclear@25: 
nuclear@25: 	int capvverts = capsub ? capsub + 1 : 0;
nuclear@25: 	int num_cap_verts = uverts * capvverts;
nuclear@25: 	int num_cap_quads = usub * capsub;
nuclear@25: 	int num_cap_tri = num_cap_quads * 2;
nuclear@25: 
nuclear@25: 	int num_verts = num_body_verts + num_cap_verts * 2;
nuclear@25: 	int num_tri = num_body_tri + num_cap_tri * 2;
nuclear@25: 
nuclear@25: 	mesh->clear();
nuclear@25: 	Vector3 *varr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_VERTEX, 3, num_verts, 0);
nuclear@25: 	Vector3 *narr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_NORMAL, 3, num_verts, 0);
nuclear@25: 	Vector3 *tarr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_TANGENT, 3, num_verts, 0);
nuclear@25: 	Vector2 *uvarr = (Vector2*)mesh->set_attrib_data(MESH_ATTR_TEXCOORD, 2, num_verts, 0);
nuclear@25: 	unsigned int *idxarr = mesh->set_index_data(num_tri * 3, 0);
nuclear@25: 
nuclear@25: 	float du = urange / (float)(uverts - 1);
nuclear@25: 	float dv = vrange / (float)(vverts - 1);
nuclear@25: 
nuclear@25: 	float u = 0.0;
nuclear@25: 	for(int i=0; i<uverts; i++) {
nuclear@25: 		float theta = SURAD(u);
nuclear@25: 
nuclear@25: 		float v = 0.0;
nuclear@25: 		for(int j=0; j<vverts; j++) {
nuclear@25: 			float y = (v - 0.5) * height;
nuclear@25: 			Vector3 pos = cylvec(theta, y);
nuclear@25: 
nuclear@25: 			*varr++ = Vector3(pos.x * rad, pos.y, pos.z * rad);
nuclear@25: 			*narr++ = Vector3(pos.x, 0.0, pos.z);
nuclear@25: 			*tarr++ = (cylvec(theta + 0.1, 0.0) - cylvec(theta - 0.1, 0.0)).normalized();
nuclear@25: 			*uvarr++ = Vector2(u * urange, v * vrange);
nuclear@25: 
nuclear@25: 			if(i < usub && j < vsub) {
nuclear@25: 				int idx = i * vverts + j;
nuclear@25: 
nuclear@25: 				*idxarr++ = idx;
nuclear@25: 				*idxarr++ = idx + vverts + 1;
nuclear@25: 				*idxarr++ = idx + 1;
nuclear@25: 
nuclear@25: 				*idxarr++ = idx;
nuclear@25: 				*idxarr++ = idx + vverts;
nuclear@25: 				*idxarr++ = idx + vverts + 1;
nuclear@25: 			}
nuclear@25: 
nuclear@25: 			v += dv;
nuclear@25: 		}
nuclear@25: 		u += du;
nuclear@25: 	}
nuclear@25: 
nuclear@25: 
nuclear@25: 	// now the cap!
nuclear@25: 	if(!capsub) {
nuclear@25: 		return;
nuclear@25: 	}
nuclear@25: 
nuclear@25: 	dv = 1.0 / (float)(capvverts - 1);
nuclear@25: 
nuclear@25: 	u = 0.0;
nuclear@25: 	for(int i=0; i<uverts; i++) {
nuclear@25: 		float theta = SURAD(u);
nuclear@25: 
nuclear@25: 		float v = 0.0;
nuclear@25: 		for(int j=0; j<capvverts; j++) {
nuclear@25: 			float r = v * rad;
nuclear@25: 
nuclear@25: 			Vector3 pos = cylvec(theta, height / 2.0) * r;
nuclear@25: 			pos.y = height / 2.0;
nuclear@25: 			Vector3 tang = (cylvec(theta + 0.1, 0.0) - cylvec(theta - 0.1, 0.0)).normalized();
nuclear@25: 
nuclear@25: 			*varr++ = pos;
nuclear@25: 			*narr++ = Vector3(0, 1, 0);
nuclear@25: 			*tarr++ = tang;
nuclear@25: 			*uvarr++ = Vector2(u * urange, v);
nuclear@25: 
nuclear@25: 			pos.y = -height / 2.0;
nuclear@25: 			*varr++ = pos;
nuclear@25: 			*narr++ = Vector3(0, -1, 0);
nuclear@25: 			*tarr++ = -tang;
nuclear@25: 			*uvarr++ = Vector2(u * urange, v);
nuclear@25: 
nuclear@25: 			if(i < usub && j < capsub) {
nuclear@25: 				unsigned int idx = num_body_verts + (i * capvverts + j) * 2;
nuclear@25: 
nuclear@25: 				unsigned int vidx[4] = {
nuclear@25: 					idx,
nuclear@25: 					idx + capvverts * 2,
nuclear@25: 					idx + (capvverts + 1) * 2,
nuclear@25: 					idx + 2
nuclear@25: 				};
nuclear@25: 
nuclear@25: 				*idxarr++ = vidx[0];
nuclear@25: 				*idxarr++ = vidx[2];
nuclear@25: 				*idxarr++ = vidx[1];
nuclear@25: 				*idxarr++ = vidx[0];
nuclear@25: 				*idxarr++ = vidx[3];
nuclear@25: 				*idxarr++ = vidx[2];
nuclear@25: 
nuclear@25: 				*idxarr++ = vidx[0] + 1;
nuclear@25: 				*idxarr++ = vidx[1] + 1;
nuclear@25: 				*idxarr++ = vidx[2] + 1;
nuclear@25: 				*idxarr++ = vidx[0] + 1;
nuclear@25: 				*idxarr++ = vidx[2] + 1;
nuclear@25: 				*idxarr++ = vidx[3] + 1;
nuclear@25: 			}
nuclear@25: 
nuclear@25: 			v += dv;
nuclear@25: 		}
nuclear@25: 		u += du;
nuclear@25: 	}
nuclear@25: }
nuclear@25: 
nuclear@25: // -------- cone --------
nuclear@25: 
nuclear@25: static Vector3 conevec(float theta, float y, float height)
nuclear@25: {
nuclear@25: 	float scale = 1.0 - y / height;
nuclear@25: 	return Vector3(sin(theta) * scale, y, cos(theta) * scale);
nuclear@25: }
nuclear@25: 
nuclear@25: void gen_cone(Mesh *mesh, float rad, float height, int usub, int vsub, int capsub, float urange, float vrange)
nuclear@25: {
nuclear@25: 	if(usub < 4) usub = 4;
nuclear@25: 	if(vsub < 1) vsub = 1;
nuclear@25: 
nuclear@25: 	int uverts = usub + 1;
nuclear@25: 	int vverts = vsub + 1;
nuclear@25: 
nuclear@25: 	int num_body_verts = uverts * vverts;
nuclear@25: 	int num_body_quads = usub * vsub;
nuclear@25: 	int num_body_tri = num_body_quads * 2;
nuclear@25: 
nuclear@25: 	int capvverts = capsub ? capsub + 1 : 0;
nuclear@25: 	int num_cap_verts = uverts * capvverts;
nuclear@25: 	int num_cap_quads = usub * capsub;
nuclear@25: 	int num_cap_tri = num_cap_quads * 2;
nuclear@25: 
nuclear@25: 	int num_verts = num_body_verts + num_cap_verts;
nuclear@25: 	int num_tri = num_body_tri + num_cap_tri;
nuclear@25: 
nuclear@25: 	mesh->clear();
nuclear@25: 	Vector3 *varr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_VERTEX, 3, num_verts, 0);
nuclear@25: 	Vector3 *narr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_NORMAL, 3, num_verts, 0);
nuclear@25: 	Vector3 *tarr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_TANGENT, 3, num_verts, 0);
nuclear@25: 	Vector2 *uvarr = (Vector2*)mesh->set_attrib_data(MESH_ATTR_TEXCOORD, 2, num_verts, 0);
nuclear@25: 	unsigned int *idxarr = mesh->set_index_data(num_tri * 3, 0);
nuclear@25: 
nuclear@25: 	float du = urange / (float)(uverts - 1);
nuclear@25: 	float dv = vrange / (float)(vverts - 1);
nuclear@25: 
nuclear@25: 	float u = 0.0;
nuclear@25: 	for(int i=0; i<uverts; i++) {
nuclear@25: 		float theta = SURAD(u);
nuclear@25: 
nuclear@25: 		float v = 0.0;
nuclear@25: 		for(int j=0; j<vverts; j++) {
nuclear@25: 			float y = v * height;
nuclear@25: 			Vector3 pos = conevec(theta, y, height);
nuclear@25: 
nuclear@25: 			Vector3 tang = (conevec(theta + 0.1, 0.0, height) - conevec(theta - 0.1, 0.0, height)).normalized();
nuclear@25: 			Vector3 bitang = (conevec(theta, y + 0.1, height) - pos).normalized();
nuclear@25: 
nuclear@25: 			*varr++ = Vector3(pos.x * rad, pos.y, pos.z * rad);
nuclear@25: 			*narr++ = cross_product(tang, bitang);
nuclear@25: 			*tarr++ = tang;
nuclear@25: 			*uvarr++ = Vector2(u * urange, v * vrange);
nuclear@25: 
nuclear@25: 			if(i < usub && j < vsub) {
nuclear@25: 				int idx = i * vverts + j;
nuclear@25: 
nuclear@25: 				*idxarr++ = idx;
nuclear@25: 				*idxarr++ = idx + vverts + 1;
nuclear@25: 				*idxarr++ = idx + 1;
nuclear@25: 
nuclear@25: 				*idxarr++ = idx;
nuclear@25: 				*idxarr++ = idx + vverts;
nuclear@25: 				*idxarr++ = idx + vverts + 1;
nuclear@25: 			}
nuclear@25: 
nuclear@25: 			v += dv;
nuclear@25: 		}
nuclear@25: 		u += du;
nuclear@25: 	}
nuclear@25: 
nuclear@25: 
nuclear@25: 	// now the bottom cap!
nuclear@25: 	if(!capsub) {
nuclear@25: 		return;
nuclear@25: 	}
nuclear@25: 
nuclear@25: 	dv = 1.0 / (float)(capvverts - 1);
nuclear@25: 
nuclear@25: 	u = 0.0;
nuclear@25: 	for(int i=0; i<uverts; i++) {
nuclear@25: 		float theta = SURAD(u);
nuclear@25: 
nuclear@25: 		float v = 0.0;
nuclear@25: 		for(int j=0; j<capvverts; j++) {
nuclear@25: 			float r = v * rad;
nuclear@25: 
nuclear@25: 			Vector3 pos = conevec(theta, 0.0, height) * r;
nuclear@25: 			Vector3 tang = (cylvec(theta + 0.1, 0.0) - cylvec(theta - 0.1, 0.0)).normalized();
nuclear@25: 
nuclear@25: 			*varr++ = pos;
nuclear@25: 			*narr++ = Vector3(0, -1, 0);
nuclear@25: 			*tarr++ = tang;
nuclear@25: 			*uvarr++ = Vector2(u * urange, v);
nuclear@25: 
nuclear@25: 			if(i < usub && j < capsub) {
nuclear@25: 				unsigned int idx = num_body_verts + i * capvverts + j;
nuclear@25: 
nuclear@25: 				unsigned int vidx[4] = {
nuclear@25: 					idx,
nuclear@25: 					idx + capvverts,
nuclear@25: 					idx + (capvverts + 1),
nuclear@25: 					idx + 1
nuclear@25: 				};
nuclear@25: 
nuclear@25: 				*idxarr++ = vidx[0];
nuclear@25: 				*idxarr++ = vidx[1];
nuclear@25: 				*idxarr++ = vidx[2];
nuclear@25: 				*idxarr++ = vidx[0];
nuclear@25: 				*idxarr++ = vidx[2];
nuclear@25: 				*idxarr++ = vidx[3];
nuclear@25: 			}
nuclear@25: 
nuclear@25: 			v += dv;
nuclear@25: 		}
nuclear@25: 		u += du;
nuclear@25: 	}
nuclear@25: }
nuclear@25: 
nuclear@25: 
nuclear@25: // -------- plane --------
nuclear@25: 
nuclear@25: void gen_plane(Mesh *mesh, float width, float height, int usub, int vsub)
nuclear@25: {
nuclear@25: 	gen_heightmap(mesh, width, height, usub, vsub, 0);
nuclear@25: }
nuclear@25: 
nuclear@25: 
nuclear@25: // ----- heightmap ------
nuclear@25: 
nuclear@25: void gen_heightmap(Mesh *mesh, float width, float height, int usub, int vsub, float (*hf)(float, float, void*), void *hfdata)
nuclear@25: {
nuclear@25: 	if(usub < 1) usub = 1;
nuclear@25: 	if(vsub < 1) vsub = 1;
nuclear@25: 
nuclear@25: 	mesh->clear();
nuclear@25: 
nuclear@25: 	int uverts = usub + 1;
nuclear@25: 	int vverts = vsub + 1;
nuclear@25: 	int num_verts = uverts * vverts;
nuclear@25: 
nuclear@25: 	int num_quads = usub * vsub;
nuclear@25: 	int num_tri = num_quads * 2;
nuclear@25: 
nuclear@25: 	Vector3 *varr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_VERTEX, 3, num_verts, 0);
nuclear@25: 	Vector3 *narr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_NORMAL, 3, num_verts, 0);
nuclear@25: 	Vector3 *tarr = (Vector3*)mesh->set_attrib_data(MESH_ATTR_TANGENT, 3, num_verts, 0);
nuclear@25: 	Vector2 *uvarr = (Vector2*)mesh->set_attrib_data(MESH_ATTR_TEXCOORD, 2, num_verts, 0);
nuclear@25: 	unsigned int *idxarr = mesh->set_index_data(num_tri * 3, 0);
nuclear@25: 
nuclear@25: 	float du = 1.0 / (float)usub;
nuclear@25: 	float dv = 1.0 / (float)vsub;
nuclear@25: 
nuclear@25: 	float u = 0.0;
nuclear@25: 	for(int i=0; i<uverts; i++) {
nuclear@25: 		float v = 0.0;
nuclear@25: 		for(int j=0; j<vverts; j++) {
nuclear@25: 			float x = (u - 0.5) * width;
nuclear@25: 			float y = (v - 0.5) * height;
nuclear@25: 			float z = hf ? hf(u, v, hfdata) : 0.0;
nuclear@25: 
nuclear@25: 			Vector3 normal = Vector3(0, 0, 1);
nuclear@25: 			if(hf) {
nuclear@25: 				float u1z = hf(u + du, v, hfdata);
nuclear@25: 				float v1z = hf(u, v + dv, hfdata);
nuclear@25: 
nuclear@25: 				Vector3 tang = Vector3(du * width, 0, u1z - z);
nuclear@25: 				Vector3 bitan = Vector3(0, dv * height, v1z - z);
nuclear@25: 				normal = cross_product(tang, bitan).normalized();
nuclear@25: 			}
nuclear@25: 
nuclear@25: 			*varr++ = Vector3(x, y, z);
nuclear@25: 			*narr++ = normal;
nuclear@25: 			*tarr++ = Vector3(1, 0, 0);
nuclear@25: 			*uvarr++ = Vector2(u, v);
nuclear@25: 
nuclear@25: 			if(i < usub && j < vsub) {
nuclear@25: 				int idx = i * vverts + j;
nuclear@25: 
nuclear@25: 				*idxarr++ = idx;
nuclear@25: 				*idxarr++ = idx + vverts + 1;
nuclear@25: 				*idxarr++ = idx + 1;
nuclear@25: 
nuclear@25: 				*idxarr++ = idx;
nuclear@25: 				*idxarr++ = idx + vverts;
nuclear@25: 				*idxarr++ = idx + vverts + 1;
nuclear@25: 			}
nuclear@25: 
nuclear@25: 			v += dv;
nuclear@25: 		}
nuclear@25: 		u += du;
nuclear@25: 	}
nuclear@25: }