ModelHandle/source/import/importerobj.js

OV.ImporterObj = class extends OV.ImporterBase
{
    constructor ()
    {
        super ();

        this.globalVertices = null;
        this.globalNormals = null;
        this.globalUvs = null;
    
        this.currentMesh = null;
        this.currentMeshData = null;
        this.currentMaterial = null;
        this.currentMaterialIndex = null;
    
        this.meshNameToMeshData = null;
        this.materialNameToIndex = null;
    }
    
    ResetState ()
    {
        this.globalVertices = [];
        this.globalNormals = [];
        this.globalUvs = [];

        this.currentMesh = null;
        this.currentMeshData = null;
        this.currentMaterial = null;
        this.currentMaterialIndex = null;        

        this.meshNameToMeshData = {};
        this.materialNameToIndex = {};
    }

    CanImportExtension (extension)
    {
        return extension === 'obj';
    }
    
    GetKnownFileFormats ()
    {
        return {
            'obj' : OV.FileFormat.Text,
            'mtl' : OV.FileFormat.Text
        };
    }

    GetUpDirection ()
    {
        return OV.Direction.Y;
    }
    
    ImportContent (fileContent, onFinish)
    {
        let obj = this;
        OV.ReadLines (fileContent, function (line) {
            if (!obj.IsError ()) {
                obj.ProcessLine (line);
            }
        });
        onFinish ();
    }
    
    ProcessLine (line)
    {
        if (line[0] === '#') {
            return;
        }

        let parameters = OV.ParametersFromLine (line, '#');
        if (parameters.length === 0) {
            return;
        }
        
        let keyword = parameters[0].toLowerCase ();
        parameters.shift ();

        if (this.ProcessMeshParameter (keyword, parameters, line)) {
            return;
        }

        if (this.ProcessMaterialParameter (keyword, parameters, line)) {
            return;
        }
    }
    
    AddNewMesh (name)
    {
        let meshData = this.meshNameToMeshData[name];
        if (meshData === undefined) {
            let mesh = new OV.Mesh ();
            if (name !== null) {
                mesh.SetName (name);
            }
            this.model.AddMesh (mesh);
            meshData = {
                mesh : mesh,
                data : {
                    globalToCurrentVertices : {},
                    globalToCurrentNormals : {},
                    globalToCurrentUvs : {}
                }
            };
            this.meshNameToMeshData[name] = meshData;
        }
        this.currentMesh = meshData.mesh;
        this.currentMeshData = meshData.data;
    }

    ProcessMeshParameter (keyword, parameters, line)
    {
        if (keyword === 'g' || keyword === 'o') {
            if (parameters.length === 0) {
                return true;
            }
            let name = OV.NameFromLine (line, keyword.length, '#');
            this.AddNewMesh (name);
            return true;
        } else if (keyword === 'v') {
            if (parameters.length < 3) {
                return true;
            }
            this.globalVertices.push (new OV.Coord3D (
                parseFloat (parameters[0]),
                parseFloat (parameters[1]),
                parseFloat (parameters[2])
            ));
            return true;
        } else if (keyword === 'vn') {
            if (parameters.length < 3) {
                return true;
            }
            this.globalNormals.push (new OV.Coord3D (
                parseFloat (parameters[0]),
                parseFloat (parameters[1]),
                parseFloat (parameters[2])
            ));
            return true;
        } else if (keyword === 'vt') {
            if (parameters.length < 2) {
                return true;
            }
            this.globalUvs.push (new OV.Coord2D (
                parseFloat (parameters[0]),
                parseFloat (parameters[1])
            ));
            return true;
        } else if (keyword === 'f') {
            if (parameters.length < 3) {
                return true;
            }
            this.ProcessFace (parameters);
            return true;
        }

        return false;
    }

    ProcessMaterialParameter (keyword, parameters, line)
    {
        function CreateColor (parameters)
        {
            return new OV.Color (
                parseInt (parseFloat (parameters[0] * 255.0), 10),
                parseInt (parseFloat (parameters[1] * 255.0), 10),
                parseInt (parseFloat (parameters[2] * 255.0), 10)
            );
        }

        function CreateTexture (keyword, line, callbacks)
        {
            let texture = new OV.TextureMap ();
            let textureName = OV.NameFromLine (line, keyword.length, '#');
            let textureBuffer = callbacks.getTextureBuffer (textureName);
            texture.name = textureName;
            if (textureBuffer !== null) {
                texture.url = textureBuffer.url;
                texture.buffer = textureBuffer.buffer;
            }
            return texture;
        }

        let obj = this;
        if (keyword === 'newmtl') {
            if (parameters.length === 0) {
                return true;
            }
            
            let material = new OV.Material ();
            let materialName = OV.NameFromLine (line, keyword.length, '#');
            let materialIndex = this.model.AddMaterial (material);
            material.name = materialName;
            this.currentMaterial = material;
            this.materialNameToIndex[materialName] = materialIndex;
            return true;
        } else if (keyword === 'usemtl') {
            if (parameters.length === 0) {
                return true;
            }
            
            let materialName = OV.NameFromLine (line, keyword.length, '#');
            let materialIndex = this.materialNameToIndex[materialName];
            if (materialIndex !== undefined) {
                this.currentMaterialIndex = materialIndex;
            }
            return true;
        } else if (keyword === 'mtllib') {
            if (parameters.length === 0) {
                return true;
            }
            let fileName = OV.NameFromLine (line, keyword.length, '#');
            let fileBuffer = this.callbacks.getFileBuffer (fileName);
            if (fileBuffer !== null) {
                OV.ReadLines (fileBuffer, function (line) {
                    if (!obj.IsError ()) {
                        obj.ProcessLine (line);
                    }
                });
            }
            return true;
        } else if (keyword === 'map_kd') {
            if (this.currentMaterial === null || parameters.length === 0) {
                return true;
            }
            this.currentMaterial.diffuseMap = CreateTexture (keyword, line, this.callbacks);
            return true;
        } else if (keyword === 'map_ks') {
            if (this.currentMaterial === null || parameters.length === 0) {
                return true;
            }
            this.currentMaterial.specularMap = CreateTexture (keyword, line, this.callbacks);
            return true;
        } else if (keyword === 'map_bump' || keyword === 'bump') {
            if (this.currentMaterial === null || parameters.length === 0) {
                return true;
            }
            this.currentMaterial.bumpMap = CreateTexture (keyword, line, this.callbacks);
            return true;
        } else if (keyword === 'ka') {
            if (this.currentMaterial === null || parameters.length < 3) {
                return true;
            }
            this.currentMaterial.ambient = CreateColor (parameters);
            return true;
        } else if (keyword === 'kd') {
            if (this.currentMaterial === null || parameters.length < 3) {
                return true;
            }
            this.currentMaterial.diffuse = CreateColor (parameters);
            return true;
        } else if (keyword === 'ks') {
            if (this.currentMaterial === null || parameters.length < 3) {
                return true;
            }
            this.currentMaterial.specular = CreateColor (parameters);
            return true;
        } else if (keyword === 'ns') {
            if (this.currentMaterial === null || parameters.length < 1) {
                return true;
            }
            this.currentMaterial.shininess = parseFloat (parameters[0]) / 100.0;
            return true;
        } else if (keyword === 'tr') {
            if (this.currentMaterial === null || parameters.length < 1) {
                return true;
            }
            this.currentMaterial.opacity = 1.0 - parseFloat (parameters[0]);
            this.currentMaterial.transparent = OV.IsLower (this.currentMaterial.opacity, 1.0);
            return true;
        } else if (keyword === 'd') {
            if (this.currentMaterial === null || parameters.length < 1) {
                return true;
            }
            this.currentMaterial.opacity = parseFloat (parameters[0]);
            this.currentMaterial.transparent = OV.IsLower (this.currentMaterial.opacity, 1.0);
            return true;
        }

        return false;    
    }

    ProcessFace (parameters)
    {
        function GetRelativeIndex (index, count)
        {
            if (index > 0) {
                return index - 1;
            } else {
                return count + index;
            }
        }
    
        function GetLocalIndex (globalValueArray, globalToCurrentIndices, globalIndex, valueAdderFunc)
        {
            if (isNaN (globalIndex) || globalIndex < 0 || globalIndex >= globalValueArray.length) {
                return null;
            }
            let result = globalToCurrentIndices[globalIndex];
            if (result === undefined) {
                let globalValue = globalValueArray[globalIndex];
                if (globalValue === undefined) {
                    return null;
                }
                result = valueAdderFunc (globalValue);
                globalToCurrentIndices[globalIndex] = result;
            }
            return result;
        }
        
        function GetLocalVertexIndex (obj, mesh, globalIndex)
        {
            return GetLocalIndex (obj.globalVertices, obj.currentMeshData.globalToCurrentVertices, globalIndex, function (val) {
                return mesh.AddVertex (new OV.Coord3D (val.x, val.y, val.z));
            });
        }
        
        function GetLocalNormalIndex (obj, mesh, globalIndex)
        {
            return GetLocalIndex (obj.globalNormals, obj.currentMeshData.globalToCurrentNormals, globalIndex, function (val) {
                return mesh.AddNormal (new OV.Coord3D (val.x, val.y, val.z));
            });
        }
        
        function GetLocalUVIndex (obj, mesh, globalIndex)
        {
            return GetLocalIndex (obj.globalUvs, obj.currentMeshData.globalToCurrentUvs, globalIndex, function (val) {
                return mesh.AddTextureUV (new OV.Coord2D (val.x, val.y));
            });
        }        
        
        let vertices = [];
        let normals = [];
        let uvs = [];
        
        for (let i = 0; i < parameters.length; i++) {
            let vertexParams = parameters[i].split ('/');
            vertices.push (GetRelativeIndex (parseInt (vertexParams[0], 10), this.globalVertices.length));
            if (vertexParams.length > 1 && vertexParams[1].length > 0) {
                uvs.push (GetRelativeIndex (parseInt (vertexParams[1], 10), this.globalUvs.length));
            }
            if (vertexParams.length > 2 && vertexParams[2].length > 0) {
                normals.push (GetRelativeIndex (parseInt (vertexParams[2], 10), this.globalNormals.length));
            }
        }
        
        if (this.currentMesh === null) {
            this.AddNewMesh ('');
        }
        
        for (let i = 0; i < vertices.length - 2; i++) {
            let v0 = GetLocalVertexIndex (this, this.currentMesh, vertices[0]);
            let v1 = GetLocalVertexIndex (this, this.currentMesh, vertices[i + 1]);
            let v2 = GetLocalVertexIndex (this, this.currentMesh, vertices[i + 2]);
            if (v0 === null || v1 === null || v2 === null) {
                this.SetError ();
                this.SetMessage ('Invalid vertex index.');
                break;
            }
            let triangle = new OV.Triangle (v0, v1, v2);
            if (normals.length === vertices.length) {
                let n0 = GetLocalNormalIndex (this, this.currentMesh, normals[0]);
                let n1 = GetLocalNormalIndex (this, this.currentMesh, normals[i + 1]);
                let n2 = GetLocalNormalIndex (this, this.currentMesh, normals[i + 2]);
                if (n0 === null || n1 === null || n2 === null) {
                    this.SetError ();
                    this.SetMessage ('Invalid normal index.');
                    break;
                }
                triangle.SetNormals (n0, n1, n2);
            }
            if (uvs.length === vertices.length) {
                let u0 = GetLocalUVIndex (this, this.currentMesh, uvs[0]);
                let u1 = GetLocalUVIndex (this, this.currentMesh, uvs[i + 1]);
                let u2 = GetLocalUVIndex (this, this.currentMesh, uvs[i + 2]);
                if (u0 === null || u1 === null || u2 === null) {
                    this.SetError ();
                    this.SetMessage ('Invalid uv index.');
                    break;
                }
                triangle.SetTextureUVs (u0, u1, u2);
            }
            if (this.currentMaterialIndex !== null) {
                triangle.mat = this.currentMaterialIndex;
            }
            this.currentMesh.AddTriangle (triangle);
        }
    }
};