Scene3D.h

 
#ifndef SCENE3D_H
#define SCENE3D_H
 
#include <vector>
#include <map>
#include <fstream>
#include <string>
#include <sstream>
#include <iostream>
#include <math.h>
#define PI 3.14159265
 
using namespace std;
 
class Scene3D {
    private:
        vector<double*> cameras; // xyzr
        vector<double*> lights; // xyz, rgb
        map<string, int> colors; // Index "r_g_b" -> ID
        vector<string> shapesJSON; // JSON strings of shapes
    
        string getColorString(double r, double g, double b) {
            stringstream sstr;
            sstr << r/255 << "," << g/255 << "," << b/255;
            string hash = sstr.str();
            if (colors.find(hash) == colors.end()) {
                colors[hash] = colors.size();
            }
            stringstream ret;
            ret << "color" << colors[hash];
            return ret.str();
        }
 
        string getMaterialsJSON() {
            stringstream json;
            json << "\"materials\":{\n";
            size_t i = 0;
            for (const pair<string, int>& c : colors) {
                json << "\t\"color" << c.second << "\": {";
                json << "\"kd\":[" << c.first << "]}";
                if (i+1 < colors.size()) {
                    json << ",";
                }
                json << "\n";
                i++;
            }
            json << "}";
            return json.str();
        }
        
        string getLightsJSON() {
            stringstream json;
            json << "\"lights\":[\n";
            for (size_t i = 0; i < lights.size(); i++) {
                double* L = lights.at(i);
                json << "{\n";
                json << "\t\"pos\":[" << L[0] << ", " << L[1] << ", " << L[2] << "],\n";
                json << "\t\"color\":[" << L[3] << ", " << L[4] << ", " << L[5] << "]\n";
                json << "}";
                if (i < lights.size() - 1) {
                    json << ",";
                }
                json << "\n";
            }
            json << "]";
            return json.str();
        }
        
        string getCamerasJSON() {
            stringstream json;
            json << "\"cameras\":[\n";
            for (size_t i = 0; i < cameras.size(); i++) {
                double* c = cameras.at(i);
                json << "{\n";
                json << "\t\"pos\":[" << c[0] << ", " << c[1] << ", " << c[2] << "],\n";
                // Compute quaternion for y rotation
                double sh = sin(c[3]*PI/360);
                double ch = cos(c[3]*PI/360);
                json << "\t\"rot\": [0, " << sh << ", 0, " << ch << "]\n";
                json << "}";
                if (i+1 < cameras.size()) {
                    json << ",";
                }
                json << "\n";
            }
            json << "]";
            return json.str();
        }
        
        string getShapesJSON() {
            stringstream json;
            json << "\"children\":[\n";
            for (size_t i = 0; i < shapesJSON.size(); i++) {
                json << shapesJSON.at(i);
                if (i+1 < shapesJSON.size()) {
                    json << ",";
                }
                json << "\n";
            }
            json << "]\n";
            return json.str();
        }
 
 
 
    public:
        Scene3D() {}
        ~Scene3D() {
            for (size_t i = 0; i < cameras.size(); i++) {
                delete[] cameras.at(i);
            }
            for (size_t i = 0; i < lights.size(); i++) {
                delete[] lights.at(i);
            }
        }
        
        string getTransformationHierarchy(double tx, double ty, double tz, 
                                                double rx, double ry, double rz, 
                                                double sx, double sy, double sz, string shape) {
            stringstream json;
            json << "{\n";
            // Translation
            json << "\"transform\":[1, 0, 0, " << tx << ", 0, 1, 0, " << ty << ", 0, 0, 1, " << tz << ", 0, 0, 0, 1],\n";
            json << "\"children\":[{";
            // Rotation about z
            double c = cos(rz*PI/180);
            double s = sin(rz*PI/180);
            json << "\"transform\":[" << c << ", " << -s << ", 0, 0, " << s << ", " << c << ", 0, 0, 0, 0, 1, 0, 0, 0, 0, 1],\n";
            json << "\"children\":[{\n";
            // Rotation about y
            c = cos(ry*PI/180);
            s = sin(ry*PI/180);
            json << "\"transform\":[" << c << ", 0, " << s << ", 0, 0, 1, 0, 0, " << -s << ", 0, " << c << ", 0, 0, 0, 0, 1],\n";
            json << "\"children\":[{\n";
            // Rotation about x
            c = cos(rx*PI/180);
            s = sin(rx*PI/180);
            json << "\"transform\":[1, 0, 0, 0, 0, " << c << ", " << -s << ", 0, 0, " << s << ", " << c << ", 0, 0, 0, 0, 1],\n";
            json << "\"children\":[{\n";
            json << "\"transform\":[" << sx << ", 0, 0, 0, 0, " << sy << ", 0, 0, 0, 0, " << sz << ", 0, 0, 0, 0, 1],\n";
            json << "\"shapes\":[\n";
            json << shape;
            json << "]\n";
            json << "}]\n"; // End rotation about z
            json << "}]\n"; // End rotation about y
            json << "}]\n"; // End rotation about x
            json << "}]\n"; // End translation
            json << "}";
            return json.str();
        }
        
        void addBox(double cx, double cy, double cz, double xlen, 
                        double ylen, double zlen, 
                        double r, double g, double b,
                        double rx, double ry, double rz) {
            stringstream boxJSON;
            boxJSON << "{\"type\":\"box\",\n";
            boxJSON << "\"material\":\"" << getColorString(r, g, b) << "\"}";
            shapesJSON.push_back(getTransformationHierarchy(cx, cy, cz, rx, ry, rz, xlen, ylen, zlen, boxJSON.str()));
        }
        
        void addBox(double cx, double cy, double cz,
                            double xlen, double ylen, double zlen,
                            double r, double g, double b) {
            addBox(cx, cy, cz, xlen, ylen, zlen, r, g, b, 0, 0, 0);
        }
        
        void addCylinder(double cx, double cy, double cz, double radius, 
                                double height, double r, double g, double b,
                                double rx, double ry, double rz,
                                double sx, double sy, double sz) {
            stringstream cylinderJSON;
            cylinderJSON << "{\"type\":\"cylinder\",\n";
            cylinderJSON << "\"radius\":" << radius << ",\n";
            cylinderJSON << "\"height\":" << height << ",\n";
            cylinderJSON << "\"material\":\"" << getColorString(r, g, b) << "\"}";
            shapesJSON.push_back(getTransformationHierarchy(cx, cy, cz, rx, ry, rz, sx, sy, sz, cylinderJSON.str()));
        }
        
        void addCylinder(double cx, double cy, double cz, double radius, 
                                double height, double r, double g, double b) {
            addCylinder(cx, cy, cz, radius, height, r, g, b, 0, 0, 0, 1, 1, 1);
        }
        
        void addCone(double cx, double cy, double cz, double radius, 
                                double height, double r, double g, double b,
                                double rx, double ry, double rz,
                                double sx, double sy, double sz) {
            stringstream coneJSON;
            coneJSON << "{\"type\":\"cone\",\n";
            coneJSON << "\"radius\":" << radius << ",\n";
            coneJSON << "\"height\":" << height << ",\n";
            coneJSON << "\"material\":\"" << getColorString(r, g, b) << "\"}";
            shapesJSON.push_back(getTransformationHierarchy(cx, cy, cz, rx, ry, rz, sx, sy, sz, coneJSON.str()));
        }
        
        void addCone(double cx, double cy, double cz, double radius, 
                                double height, double r, double g, double b) {
            addCone(cx, cy, cz, radius, height, r, g, b, 0, 0, 0, 1, 1, 1);
        }
        
        void addEllipsoid(double cx, double cy, double cz, 
                                double radx, double rady, double radz,
                                double r, double g, double b, 
                                double rx, double ry, double rz) {
            stringstream ellipsoidJSON;
            ellipsoidJSON << "{\"type\":\"sphere\",\n";
            ellipsoidJSON << "\"material\":\"" << getColorString(r, g, b) << "\"}";
            shapesJSON.push_back(getTransformationHierarchy(cx, cy, cz, rx, ry, rz, radx, rady, radz, ellipsoidJSON.str()));
        }
        
        void addEllipsoid(double cx, double cy, double cz, 
                                double radx, double rady, double radz,
                                double r, double g, double b) {
            addEllipsoid(cx, cy, cz, radx, rady, radz, r, g, b, 0, 0, 0);
        }
        
        void addSphere(double cx, double cy, double cz, double radius,
                            double r, double g, double b) {
            addEllipsoid(cx, cy, cz, radius, radius, radius, r, g, b);
        }
        
        void addSpecialMesh(string meshName, 
                                double cx, double cy, double cz, 
                                double rx, double ry, double rz,
                                double sx, double sy, double sz,
                                double r, double g, double b) {
            stringstream meshJSON;
            meshJSON << "{\"type\":\"mesh\",\n";
            meshJSON << "\"material\":\"" << getColorString(r, g, b) << "\",\n";
            meshJSON << "\"filename\":\"../meshes/" << meshName << ".off\"}";
            shapesJSON.push_back(getTransformationHierarchy(cx, cy, cz, rx, ry, rz, sx, sy, sz, meshJSON.str()));
        }
        
        
        void addCamera(double x, double y, double z, double rot) {
            double* xyzr = new double[4];
            xyzr[0] = x;
            xyzr[1] = y;
            xyzr[2] = z;
            xyzr[3] = rot;
            cameras.push_back(xyzr);
        }
        
        void addLight(double x, double y, double z, double r, double g, double b) {
            double* light = new double[6];
            light[0] = x;
            light[1] = y;
            light[2] = z;
            light[3] = r;
            light[4] = g;
            light[5] = b;
            lights.push_back(light);
        }
        
 
        void saveScene(string filename, string sceneName) {
            stringstream json;
            json << "{\n\"name\":\"" << sceneName << "\",\n";
            json << getMaterialsJSON() << ",\n";
            json << getLightsJSON() << ",\n";
            json << getCamerasJSON() << ",\n";
            json << getShapesJSON() << "\n}";
            std::ofstream out(filename.c_str());
            out << json.str();
            out.close();
        }
};
 
#endif