/******************************************************************/
/* This file is part of the homework assignments for CSCI-427/527 */
/* at The College of William & Mary and authored by Pieter Peers. */
/* No part of this file, whether altered or in original form, can */
/* be distributed or used outside the context of CSCI-427/527 */
/* without consent of either the College of William & Mary or */
/* Pieter Peers. */
/******************************************************************/
#include <memory>
#include <string>
#include <cassert>
#include <fstream>
#include "vec3d.h"
#include "vec2d.h"
#include "importOBJ.h"
#include "errorMessage.h"
void skipLine(std::ifstream& ifs)
{
unsigned char c = ifs.get();
while(ifs.good() && c != '\r' && c != '\n') c = ifs.get();
while(ifs.good() && (c == '\r' || c == '\n')) c = ifs.get();
ifs.unget();
}
void importOBJ(const std::string filename, std::vector<triangle>& triangle_list)
{
// define buffers
auto vertex_list = std::make_shared<std::vector<vec3d>>();
auto normal_list = std::make_shared<std::vector<vec3d>>();
auto textureCoord_list = std::make_shared<std::vector<vec2d>>();
// open obj file
std::ifstream ifs(filename, std::ifstream::in | std::ifstream::binary);
if(!ifs.is_open())
errorMessage("Unable to import OBJ (%s).", filename.c_str());
// read line by line
unsigned char key;
while(ifs.good())
{
// get key
key = ifs.get();
// VERTEX DATA:
if(key == 'v')
{
// get vertex data
key = ifs.get();
// Vertex Coordinate
if(key == ' ')
{
vec3d v;
ifs >> v.x >> v.y >> v.z;
vertex_list->push_back(v);
}
// Texture Coordinate
else if(key == 't')
{
vec2d t;
ifs >> t.u >> t.v;
textureCoord_list->push_back(t);
}
// Normal
else if(key == 'n')
{
vec3d n;
ifs >> n.x >> n.y >> n.z;
normal_list->push_back(n);
}
// Unknown => ERROR
else errorMessage("Unknown OBJ vertex-key: v%c.", key);
}
// POLYGON:
else if(key == 'f')
{
// temp data structures
std::vector<unsigned int> vidx;
std::vector<unsigned int> nidx;
std::vector<unsigned int> tidx;
// get polygon data
// can be either:
// 1) v0 v1 .. vn
// 2) v0//n0 v1//n1 .. vn//nn
// 3) v0/t0 v1/t1 ... vn/tn
// 4) v0/t0/n0 ... vn/tn/nn
bool done=false;
while(!done)
{
// read entry
signed int vi, vt=-1, vn=-1;
ifs >> vi;
unsigned char c = ifs.get();
if(c == '/')
{
c = ifs.get();
if(c != '/') { ifs.unget(); ifs >> vt; c = ifs.get(); }
if(c == '/') { ifs >> vn; c = ifs.get(); }
}
// sanity check
assert(vt != 0 && vn != 0 && vi != 0);
// store in polygon
if(vt != -1) { assert(vt-1 < textureCoord_list->size()); tidx.push_back(vt-1); }
if(vn != -1) { assert(vn-1 < normal_list->size()); nidx.push_back(vn-1); }
if(vi != -1) { assert(vi-1 < vertex_list->size()); vidx.push_back(vi-1); }
// eat spaces and end-of-lines
ifs.unget();
c = ifs.get();
while(ifs.good() && (c == ' ' ||c == '\r' || c == '\n'))
{
if(c == '\r' || c == '\n') done = true;
c = ifs.get();
}
ifs.unget();
// sanity check
assert(tidx.empty() || tidx.size() == vidx.size());
assert(nidx.empty() || nidx.size() == vidx.size());
// store triangle
if(vidx.size() >= 3)
{
size_t m=vidx.size() - 2;
size_t l=vidx.size() - 1;
if(tidx.empty() && nidx.empty()) triangle_list.push_back( triangle(vidx[0], vidx[m], vidx[l], vertex_list) );
if(tidx.empty() && !nidx.empty()) triangle_list.push_back( triangle(vidx[0], vidx[m], vidx[l], vertex_list,
nidx[0], nidx[m], nidx[l], normal_list) );
if(!tidx.empty() && nidx.empty()) triangle_list.push_back( triangle(vidx[0], vidx[m], vidx[l], vertex_list,
tidx[0], tidx[m], tidx[l], textureCoord_list) );
if(!tidx.empty() && !nidx.empty()) triangle_list.push_back( triangle(vidx[0], vidx[m], vidx[l], vertex_list,
nidx[0], nidx[m], nidx[l], normal_list,
tidx[0], tidx[m], tidx[l], textureCoord_list) );
}
}
}
else { ifs.unget(); skipLine(ifs); }
}
// Done.
}
