WebSVN – gelsvn – Diff – /trunk/src/Geometry/obj_load.cpp

 			if (i<0) {
 				return mesh->texcoords.no_vertices()+i;
 			} else
 				return i-1;
+		}
+    void read_material_library(const string& filename, vector<Material>& materials);
 	public:
 		TriMeshObjLoader(TriMesh *_mesh): mesh(_mesh) {}
 		void load(const std::string& filename);
+    void load_material_library(const string& filename, vector<Material>& materials)
+    {
+      pathname = get_path(filename);
-		void read_material_library(const string& filename);
+      read_material_library(filename, materials);
+    }
 	};
-	void TriMeshObjLoader::read_material_library(const string& filename)
+	void TriMeshObjLoader::read_material_library(const string& filename, vector<Material>& materials)
+	{
-		string fn = pathname + filename;
+    string fn = pathname + filename;
-        ifstream material_file(fn.data());
+    ifstream material_file(fn.data());
+    if(!material_file)
+    {
+      cerr << "Could not open " << filename << endl;
+      return;
+    }
+    string buf;
+    unsigned int nummaterials=0;
-		if(material_file)
+    while(material_file >> buf)
+    {
+      switch(buf[0])
+      {
+      case 'n':
+        ++nummaterials;
+        mesh->materials.push_back(Material());
+        material_file >> materials[nummaterials].name;
+        break;
+      case 'N':
+        switch(buf[1])
+        {
-            string buf;
-            unsigned int nummaterials=0;
-            while(material_file >> buf)
-            {
-                switch(buf[0])
-                {
+        case 's':
-                    case 'n':
-                        ++nummaterials;
-                        mesh->materials.push_back(Material());
-                        material_file >> mesh->materials[nummaterials].name;
-                        break;
-                    case 'N':
-                        switch(buf[1])
-                    {
-                        case 's':
-                            material_file >> mesh->materials[nummaterials].shininess;
+          material_file >> materials[nummaterials].shininess;
-                            mesh->materials[nummaterials].shininess *= 128.0f/1000.0f;
+          materials[nummaterials].shininess *= 128.0f/1000.0f;
-                            break;
+          break;
-                        case 'i':
+        case 'i':
-                            material_file >> mesh->materials[nummaterials].ior_in;
+          material_file >> materials[nummaterials].ior;
-                            break;
+          break;
-                    }
+        }
-                        break;
+        break;
-                    case 'K':
+      case 'K':
-                        switch(buf[1])
+        switch(buf[1])
-					{
+        {
-                        case 'd':
+        case 'd':
-                            material_file >> mesh->materials[nummaterials].diffuse;
+          material_file >> materials[nummaterials].diffuse[0]
-                            break;
+                        >> materials[nummaterials].diffuse[1]
-                        case 's':
-                            material_file >> mesh->materials[nummaterials].specular;
+                        >> materials[nummaterials].diffuse[2];
-                            break;
+          break;
-                        case 'a':
+        case 's':
-                            material_file >> mesh->materials[nummaterials].ambient;
+          material_file >> materials[nummaterials].specular[0]
-                            break;
+                        >> materials[nummaterials].specular[1]
-					}
-                        break;
-                    case 'T':
-                        material_file >> mesh->materials[nummaterials].transmission;
+                        >> materials[nummaterials].specular[2];
-                        break;
+          break;
-                    case 'i':
+        case 'a':
-                        material_file >> mesh->materials[nummaterials].illum;
-                        break;
-                    case 'm': // Map ... all maps are treated equally.
-                        material_file >> mesh->materials[nummaterials].tex_name;
+          material_file >> materials[nummaterials].ambient[0]
-                        mesh->materials[nummaterials].has_texture = true;
+                        >> materials[nummaterials].ambient[1]
-                        mesh->materials[nummaterials].tex_path = pathname;
+                        >> materials[nummaterials].ambient[2];
-                        break;
-                    case '#':
-                    default:
-                        material_file.ignore(1024, '\n');
-                        break;
-                }
-            }
+          break;
+        }
+        break;
+      case 'T':
+        material_file >> materials[nummaterials].transmission[0]
+                      >> materials[nummaterials].transmission[1]
+                      >> materials[nummaterials].transmission[2];
+        break;
+      case 'i':
+        material_file >> materials[nummaterials].illum;
+        break;
+      case 'm': // Map ... all maps are treated equally.
+        material_file >> materials[nummaterials].tex_name;
+        materials[nummaterials].has_texture = true;
+        materials[nummaterials].tex_path = pathname;
+        break;
+      case '#':
+      default:
+        material_file.ignore(1024, '\n');
+        break;
+      }
+    }
+  }
-	void TriMeshObjLoader::load(const std::string& filename)
+  void TriMeshObjLoader::load(const std::string& filename)
-	{
+  {
-		pathname = get_path(filename);
+    pathname = get_path(filename);
-        ifstream obj_file(filename.data());
+    ifstream obj_file(filename.data());
-        if(obj_file)
+    if(!obj_file)
+    {
+			cerr << "File " << filename << " does not exist" << endl;
+      exit(0);
+    }
+    mesh->materials.resize(1);
+    string buf;
+    int current_material=0;
+    while(obj_file >> buf)
+    {
+      switch(buf[0])
+      {
+      case 'v': // v, vn, vt
+        {
-            mesh->materials.resize(1);
-            string buf;
+          if(buf == "v")
-            int current_material=0;
-            while(obj_file >> buf)
-            {
+          {
-                cout << buf[0] << endl;
-                switch(buf[0])
-                {
-                    case 'v': // v, vn, vt
-                    {
+            Vec3f v_geo;
-                        if(buf == "v")
-                        {
-                            Vec3f v_geo;
-                            obj_file >> v_geo;
+            obj_file >> v_geo;
-                            mesh->geometry.add_vertex(v_geo);
+            mesh->geometry.add_vertex(v_geo);
-                        }
+          }
-                        else if(buf == "vn")
+          else if(buf == "vn")
-                        {
+          {
-                            Vec3f v_norm;
+            Vec3f v_norm;
-                            obj_file >> v_norm;
+            obj_file >> v_norm;
-                            mesh->normals.add_vertex(v_norm);
+            mesh->normals.add_vertex(v_norm);
-                        }
+          }
-                        else if(buf == "vt")
+          else if(buf == "vt")
-                        {
+          {
-                            Vec3f v_tex;
+            Vec3f v_tex;
-                            obj_file >> v_tex[0] >> v_tex[1];
+            obj_file >> v_tex[0] >> v_tex[1];
-                            v_tex[2]=1;
+            v_tex[2]=1;
-                            mesh->texcoords.add_vertex(v_tex);
+            mesh->texcoords.add_vertex(v_tex);
-                        }
+          }
-                    }
+        }
-                        break;
+        break;
-                    case 'f':
+      case 'f':
-                    {
+        {
-                        char line[1024];
+          char line[1024];
-                        vector<int> v_indices;
-                        vector<int> t_indices;
-                        vector<int> n_indices;
-                        obj_file.getline(line, 1022);
-                        char* pch = strtok(line, " \t");
-                        while(pch != 0)
-                        {
-                            int v,t,n;
-                            if(sscanf(pch, "%d/%d/%d", &v, &t, &n)==3)
-                            {
-                                v_indices.push_back(get_vert(v));
-                                t_indices.push_back(get_texcoord(t));
-                                n_indices.push_back(get_normal(n));
-                            }
-                            else if(sscanf(pch, "%d//%d", &v, &n)==2)
-                            {
-                                v_indices.push_back(get_vert(v));
-                                n_indices.push_back(get_normal(n));
-                            }
-                            else if(sscanf(pch, "%d/%d", &v, &t)==2)
-                            {
-                                v_indices.push_back(get_vert(v));
-                                t_indices.push_back(get_texcoord(t));
-                            }
-                            else if(sscanf(pch, "%d", &v)==1)
-                                v_indices.push_back(get_vert(v));
-                            pch = strtok(0, " \t");
-                        }
-                        if(v_indices.size()>=3)
-                            for(size_t i=0;i<=(v_indices.size()-3);++i)
-                            {
-                                int idx = mesh->geometry.add_face(Vec3i(v_indices[0], v_indices[i+1], v_indices[i+2]));
-                                if(t_indices.size())
-                                    mesh->texcoords.add_face(Vec3i(t_indices[0], t_indices[i+1], t_indices[i+2]), idx);
-                                if(n_indices.size())
-                                    mesh->normals.add_face(Vec3i(n_indices[0], n_indices[i+1], n_indices[i+2]), idx);
-                                mesh->mat_idx.push_back(current_material);
-                            }
-                    }
-                        break;
-                    case 'm':
-                        obj_file >> buf;
-                        read_material_library(buf);
-                        break;
+          vector<int> v_indices;
-                    case 'u':
-                        obj_file >> buf;
-                        current_material = mesh->find_material(buf);
-                        break;
+          vector<int> t_indices;
-                    case '#':
-                    default:
-                        obj_file.ignore(1024, '\n');
-                        break;
+          vector<int> n_indices;
+          obj_file.getline(line, 1022);
+          char* pch = strtok(line, " \t");
+          while(pch != 0)
+          {
+            int v,t,n;
+            if(sscanf(pch, "%d/%d/%d", &v, &t, &n)==3)
+            {
+              v_indices.push_back(get_vert(v));
+              t_indices.push_back(get_texcoord(t));
+              n_indices.push_back(get_normal(n));
-                }
+            }
+            else if(sscanf(pch, "%d//%d", &v, &n)==2)
+            {
+              v_indices.push_back(get_vert(v));
+              n_indices.push_back(get_normal(n));
+            }
+            else if(sscanf(pch, "%d/%d", &v, &t)==2)
+            {
+              v_indices.push_back(get_vert(v));
+              t_indices.push_back(get_texcoord(t));
+            }
+            else if(sscanf(pch, "%d", &v)==1)
+              v_indices.push_back(get_vert(v));
+            pch = strtok(0, " \t");
+          }
+          if(v_indices.size()>=3)
+            for(size_t i=0;i<=(v_indices.size()-3);++i)
+            {
+              int idx = mesh->geometry.add_face(Vec3i(v_indices[0], v_indices[i+1], v_indices[i+2]));
+              if(t_indices.size())
+              mesh->texcoords.add_face(Vec3i(t_indices[0], t_indices[i+1], t_indices[i+2]), idx);
+              if(n_indices.size())
+              mesh->normals.add_face(Vec3i(n_indices[0], n_indices[i+1], n_indices[i+2]), idx);
+              mesh->mat_idx.push_back(current_material);
+            }
+        }
+        break;
+      case 'm':
+        obj_file >> buf;
+        read_material_library(buf, mesh->materials);
+        break;
+      case 'u':
+        obj_file >> buf;
+        current_material = mesh->find_material(buf);
+        break;
+      case '#':
+      default:
+        obj_file.ignore(1024, '\n');
+        break;
+      }
+    }
+  }
 	void obj_load(const string& filename, TriMesh &mesh)
+	{
 		TriMeshObjLoader loader(&mesh);
 		loader.load(filename);
+	}
-    /// Load materials from an MRL file
+  /// Load materials from an MTL file
-    void mtl_load(const std::string& filename, std::vector<Material>& materials)
+  void mtl_load(const std::string& filename, std::vector<Material>& materials)
-    {
+  {
-        TriMesh m;
-		TriMeshObjLoader loader(&m);
+    TriMeshObjLoader loader(0);
-		loader.read_material_library(filename);
+    loader.load_material_library(filename, materials);
-        materials = m.materials;
-    }
+  }
+}

Subversion Repositories gelsvn

(root)/trunk/src/Geometry/obj_load.cpp – Rev 605 → 606