Simple Obj File Loader

ObjFile.cpp

//
// Created by Morten Nobel-Jørgensen on 06/05/14.
// Copyright (c) 2014 Morten Nobel-Joergensen. All rights reserved.
//

#include "ObjFile.h"

#include <fstream>
#include <sstream>

using namespace std;
using namespace glm;

bool ObjFile::save(std::string filename, std::vector<glm::dvec3> &positions, std::vector<int> &indices) {
    std::ofstream os(filename.c_str());
    if(os.bad())
        return false;


    for(int i=0;i<positions.size();i++){
        glm::dvec3 p = positions[i];
        os << "v "<< p[0] << " " << p[1] << " " << p[2] << "\n";
    }

    for(int i=0;i<indices.size();){
        os << "f ";
        for(size_t j = 0; j < 3 ; j++, i++){
            os << (indices[i]+1) << " ";
        }
        os<<std::endl;
    }
    os.close();
    return true;
}

bool ObjFile::load(std::string filename, std::vector<glm::dvec3> &positions, std::vector<int> &indices) {
    ifstream os(filename.c_str());
    if(os.bad())
        return false;

    positions.clear();
    indices.clear();
    const int c = 256;
    char line[c];
    while (!os.eof()){
        os.getline(line, c);
        if (strlen(line)==0){
            continue;
        }
        stringstream ss{line};
        string token;
        ss >> token;
        if (token=="v"){
            glm::dvec3 v;
            ss >> v.x >> v.y >> v.z;
            positions.push_back(v);
        }
        else if (token=="f"){
            int index;
            for (int i=0;i<3;i++){ // assume only triangles
                ss >> index;
                indices.push_back(index-1);
            }
        }
    }
    os.close();
    return true;
}

ObjFile.h

//
// Created by Morten Nobel-Jørgensen on 06/05/14.
// Copyright (c) 2014 Morten Nobel-Joergensen. All rights reserved.
//


#pragma once

#include <glm/glm.hpp>
#include <vector>
#include <string>

class ObjFile {
public:
    static bool save(std::string filename, std::vector<glm::dvec3> &positions, std::vector<int> &indices);

    static bool load(std::string filename, std::vector<glm::dvec3> &positions, std::vector<int> &indices);
};