X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?p=laserbrain_demo;a=blobdiff_plain;f=src%2Fsceneload.cc;h=11ce1e1b0208b7773e5fa1afa1f16d90a2191ea0;hp=7fdc0041430505d9178764dabe39c193daa9e633;hb=84f1549d8146ac54574256ae0243747199e51151;hpb=7dc732cd42c88fa16accdbc606f10bcd6815d478

diff --git a/src/sceneload.cc b/src/sceneload.cc
index 7fdc004..11ce1e1 100644
--- a/src/sceneload.cc
+++ b/src/sceneload.cc
@@ -1,5 +1,6 @@
 #include <stdio.h>
 #include <assert.h>
+#include <string>
 #include <vector>
 #include <map>
 #include <gmath/gmath.h>
@@ -35,6 +36,7 @@ static void print_hierarchy(const aiNode *node);
 
 struct LoaderData {
 	const aiScene *aiscn;
+	std::string fname;
 	std::map<std::string, SceneNode*> node_by_name;
 	std::map<aiMesh*, Mesh*> mesh_by_aimesh;
 };
@@ -93,6 +95,7 @@ bool Scene::load(const char *fname, unsigned int flags)
 
 		LoaderData *ldata = new LoaderData;
 		ldata->aiscn = aiscn;
+		ldata->fname = std::string(fname);
 		loader_data = (void*)ldata;
 	}
 
@@ -107,6 +110,7 @@ bool Scene::load(const char *fname, unsigned int flags)
 
 		LoaderData *ldata = (LoaderData*)loader_data;
 		const aiScene *aiscn = ldata->aiscn;
+		fname = ldata->fname.c_str();
 
 		// load all meshes
 		for(unsigned int i=0; i<aiscn->mNumMeshes; i++) {
@@ -135,10 +139,11 @@ bool Scene::load(const char *fname, unsigned int flags)
 			}
 		}
 
+		info_log("loaded scene file: %s, %d meshes\n", fname, (int)meshes.size());
+
 		aiReleaseImport(aiscn);
 		delete ldata;
 		loader_data = 0;
-		info_log("loaded scene file: %s, %d meshes\n", fname, (int)meshes.size());
 		nodes->update(0);
 	}
 	return true;
@@ -355,3 +360,31 @@ static Mat4 assimp_matrix(const aiMatrix4x4 &aim)
 	memcpy(m[0], &aim, 16 * sizeof(float));
 	return transpose(m);
 }
+
+
+// --- SceneSet ---
+
+SceneSet::SceneSet()
+	: DataSet<Scene*>(create_scene, load_scene, done_scene, free_scene)
+{
+}
+
+Scene *SceneSet::create_scene()
+{
+	return new Scene;
+}
+
+bool SceneSet::load_scene(Scene *scn, const char *fname)
+{
+	return scn->load(fname, SCNLOAD_FLIPTEX | SCNLOAD_STAGE_IO);
+}
+
+bool SceneSet::done_scene(Scene *scn)
+{
+	return scn->load(0, SCNLOAD_STAGE_GL);
+}
+
+void SceneSet::free_scene(Scene *scn)
+{
+	delete scn;
+}