254 lines
8.3 KiB
C++
254 lines
8.3 KiB
C++
/*
|
||
* This file is part of the Colobot: Gold Edition source code
|
||
* Copyright (C) 2001-2014, Daniel Roux, EPSITEC SA & TerranovaTeam
|
||
* http://epsiteс.ch; http://colobot.info; http://github.com/colobot
|
||
*
|
||
* This program is free software: you can redistribute it and/or modify
|
||
* it under the terms of the GNU General Public License as published by
|
||
* the Free Software Foundation, either version 3 of the License, or
|
||
* (at your option) any later version.
|
||
*
|
||
* This program is distributed in the hope that it will be useful,
|
||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
||
* See the GNU General Public License for more details.
|
||
*
|
||
* You should have received a copy of the GNU General Public License
|
||
* along with this program. If not, see http://gnu.org/licenses
|
||
*/
|
||
#include "graphics/engine/modelfile.h"
|
||
|
||
#include "math/func.h"
|
||
|
||
#include <gtest/gtest.h>
|
||
|
||
#include <cassert>
|
||
#include <sstream>
|
||
|
||
/* Test model file (new text format) */
|
||
const char* const TEXT_MODEL =
|
||
"# Colobot text model\n"
|
||
"\n"
|
||
"### HEAD\n"
|
||
"version 1\n"
|
||
"total_triangles 2\n"
|
||
"\n"
|
||
"### TRIANGLES\n"
|
||
"p1 c -12.4099 10.0016 -2.54558 n 1 -0 1.87319e-07 t1 0.970703 0.751953 t2 0 0\n"
|
||
"p2 c -12.4099 10.0016 2.54558 n 1 -0 1.87319e-07 t1 0.998047 0.751953 t2 0 0\n"
|
||
"p3 c -12.4099 4.00165 -2.54558 n 1 -0 1.87319e-07 t1 0.970703 0.998047 t2 0 0\n"
|
||
"mat dif 1 1 1 0 amb 0.5 0.5 0.5 0 spc 0 0 0 0\n"
|
||
"tex1 lemt.png\n"
|
||
"tex2\n"
|
||
"var_tex2 N\n"
|
||
"lod_level 0\n"
|
||
"state 1024\n"
|
||
"\n"
|
||
"p1 c -19 -1 4 n -1 0 0 t1 0.248047 0.123047 t2 0.905224 0.52067\n"
|
||
"p2 c -19 4 4 n -1 0 0 t1 0.248047 0.00195312 t2 0.905224 0.614223\n"
|
||
"p3 c -19 4 -4 n -1 0 0 t1 0.00195312 0.00195312 t2 0.0947756 0.614223\n"
|
||
"mat dif 1 1 1 0 amb 0.5 0.5 0.5 0 spc 0 0 0 0\n"
|
||
"tex1 derrick.png\n"
|
||
"tex2\n"
|
||
"var_tex2 Y\n"
|
||
"lod_level 1\n"
|
||
"state 0\n"
|
||
"";
|
||
|
||
// Triangles as defined in model file
|
||
Gfx::ModelTriangle TRIANGLE_1;
|
||
Gfx::ModelTriangle TRIANGLE_2;
|
||
|
||
// Sets triangle data
|
||
void Init()
|
||
{
|
||
|
||
TRIANGLE_1.p1 = Gfx::VertexTex2(Math::Vector(-12.4099, 10.0016, -2.54558),
|
||
Math::Vector(1, 0, 1.87319e-07),
|
||
Math::Point(0.970703, 0.751953),
|
||
Math::Point(0, 0));
|
||
TRIANGLE_1.p2 = Gfx::VertexTex2(Math::Vector(-12.4099, 10.0016, 2.54558),
|
||
Math::Vector(1, 0, 1.87319e-07),
|
||
Math::Point(0.998047, 0.751953),
|
||
Math::Point(0, 0));
|
||
TRIANGLE_1.p3 = Gfx::VertexTex2(Math::Vector(-12.4099, 4.00165, -2.54558),
|
||
Math::Vector(1, 0, 1.87319e-07),
|
||
Math::Point(0.970703, 0.998047),
|
||
Math::Point(0, 0));
|
||
TRIANGLE_1.material.diffuse = Gfx::Color(1, 1, 1, 0);
|
||
TRIANGLE_1.material.ambient = Gfx::Color(0.5, 0.5, 0.5, 0);
|
||
TRIANGLE_1.material.specular = Gfx::Color(0, 0, 0, 0);
|
||
TRIANGLE_1.tex1Name = "lemt.png";
|
||
TRIANGLE_1.variableTex2 = false;
|
||
TRIANGLE_1.lodLevel = Gfx::LOD_Constant;
|
||
TRIANGLE_1.state = 1024;
|
||
|
||
TRIANGLE_2.p1 = Gfx::VertexTex2(Math::Vector(-19, -1, 4),
|
||
Math::Vector(-1, 0, 0),
|
||
Math::Point(0.248047, 0.123047),
|
||
Math::Point(0.905224, 0.52067));
|
||
TRIANGLE_2.p2 = Gfx::VertexTex2(Math::Vector(-19, 4, 4),
|
||
Math::Vector(-1, 0, 0),
|
||
Math::Point(0.248047, 0.00195312),
|
||
Math::Point(0.905224, 0.614223));
|
||
TRIANGLE_2.p3 = Gfx::VertexTex2(Math::Vector(-19, 4, -4),
|
||
Math::Vector(-1, 0, 0),
|
||
Math::Point(0.00195312, 0.00195312),
|
||
Math::Point(0.0947756, 0.614223));
|
||
TRIANGLE_2.material.diffuse = Gfx::Color(1, 1, 1, 0);
|
||
TRIANGLE_2.material.ambient = Gfx::Color(0.5, 0.5, 0.5, 0);
|
||
TRIANGLE_2.material.specular = Gfx::Color(0, 0, 0, 0);
|
||
TRIANGLE_2.tex1Name = "derrick.png";
|
||
TRIANGLE_2.variableTex2 = true;
|
||
TRIANGLE_2.lodLevel = Gfx::LOD_Low;
|
||
TRIANGLE_2.state = 0;
|
||
}
|
||
|
||
|
||
// Compares vertices (within Math::TOLERANCE)
|
||
bool CompareVertices(const Gfx::VertexTex2& v1, const Gfx::VertexTex2& v2)
|
||
{
|
||
if ( !( Math::IsEqual(v1.coord.x, v2.coord.x) &&
|
||
Math::IsEqual(v1.coord.y, v2.coord.y) &&
|
||
Math::IsEqual(v1.coord.z, v2.coord.z) ) )
|
||
return false;
|
||
|
||
if ( !( Math::IsEqual(v1.normal.x, v2.normal.x) &&
|
||
Math::IsEqual(v1.normal.y, v2.normal.y) &&
|
||
Math::IsEqual(v1.normal.z, v2.normal.z) ) )
|
||
return false;
|
||
|
||
if ( !( Math::IsEqual(v1.texCoord.x, v2.texCoord.x) &&
|
||
Math::IsEqual(v1.texCoord.y, v2.texCoord.y) ) )
|
||
return false;
|
||
|
||
if ( !( Math::IsEqual(v1.texCoord2.x, v2.texCoord2.x) &&
|
||
Math::IsEqual(v1.texCoord2.y, v2.texCoord2.y) ) )
|
||
return false;
|
||
|
||
return true;
|
||
}
|
||
|
||
// Compares colors (within Math::TOLERANCE)
|
||
bool CompareColors(const Gfx::Color& c1, const Gfx::Color& c2)
|
||
{
|
||
return Math::IsEqual(c1.r, c2.r) &&
|
||
Math::IsEqual(c1.g, c2.g) &&
|
||
Math::IsEqual(c1.b, c2.b) &&
|
||
Math::IsEqual(c1.a, c2.a);
|
||
}
|
||
|
||
// Compares model triangles (within Math::TOLERANCE)
|
||
bool CompareTriangles(const Gfx::ModelTriangle& t1, const Gfx::ModelTriangle& t2)
|
||
{
|
||
if (! CompareVertices(t1.p1, t2.p1))
|
||
return false;
|
||
|
||
if (! CompareVertices(t1.p2, t2.p2))
|
||
return false;
|
||
|
||
if (! CompareVertices(t1.p3, t2.p3))
|
||
return false;
|
||
|
||
if (! CompareColors(t1.material.diffuse, t2.material.diffuse))
|
||
return false;
|
||
|
||
if (! CompareColors(t1.material.ambient, t2.material.ambient))
|
||
return false;
|
||
|
||
if (! CompareColors(t1.material.specular, t2.material.specular))
|
||
return false;
|
||
|
||
if (t1.tex1Name != t2.tex1Name)
|
||
return false;
|
||
|
||
if (t1.tex2Name != t2.tex2Name)
|
||
return false;
|
||
|
||
if (t1.variableTex2 != t2.variableTex2)
|
||
return false;
|
||
|
||
if (t1.lodLevel != t2.lodLevel)
|
||
return false;
|
||
|
||
if (t1.state != t2.state)
|
||
return false;
|
||
|
||
return true;
|
||
}
|
||
|
||
// Tests reading/writing new text model file
|
||
TEST(ModelFileTest, RWTxtModel)
|
||
{
|
||
std::stringstream str;
|
||
str.str(TEXT_MODEL);
|
||
|
||
Gfx::CModelFile modelFile;
|
||
|
||
EXPECT_TRUE(modelFile.ReadTextModel(str));
|
||
|
||
EXPECT_EQ(modelFile.GetTriangleCount(), 2);
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[0], TRIANGLE_1));
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[1], TRIANGLE_2));
|
||
|
||
str.str("");
|
||
|
||
EXPECT_TRUE(modelFile.WriteTextModel(str));
|
||
str.seekg(0);
|
||
EXPECT_TRUE(modelFile.ReadTextModel(str));
|
||
|
||
EXPECT_EQ(modelFile.GetTriangleCount(), 2);
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[0], TRIANGLE_1));
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[1], TRIANGLE_2));
|
||
}
|
||
|
||
// Tests reading/writing new binary model
|
||
TEST(ModelFileTest, RWBinModel)
|
||
{
|
||
std::stringstream str;
|
||
str.str(TEXT_MODEL);
|
||
|
||
Gfx::CModelFile modelFile;
|
||
|
||
EXPECT_TRUE(modelFile.ReadTextModel(str));
|
||
|
||
EXPECT_EQ(modelFile.GetTriangleCount(), 2);
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[0], TRIANGLE_1));
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[1], TRIANGLE_2));
|
||
|
||
str.str("");
|
||
|
||
EXPECT_TRUE(modelFile.WriteBinaryModel(str));
|
||
str.seekg(0);
|
||
EXPECT_TRUE(modelFile.ReadBinaryModel(str));
|
||
|
||
EXPECT_EQ(modelFile.GetTriangleCount(), 2);
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[0], TRIANGLE_1));
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[1], TRIANGLE_2));
|
||
}
|
||
|
||
// Tests reading/writing old model file
|
||
TEST(ModelFileTest, RWOldModel)
|
||
{
|
||
std::stringstream str;
|
||
str.str(TEXT_MODEL);
|
||
|
||
Gfx::CModelFile modelFile;
|
||
|
||
EXPECT_TRUE(modelFile.ReadTextModel(str));
|
||
|
||
EXPECT_EQ(modelFile.GetTriangleCount(), 2);
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[0], TRIANGLE_1));
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[1], TRIANGLE_2));
|
||
|
||
str.str("");
|
||
|
||
EXPECT_TRUE(modelFile.WriteModel(str));
|
||
str.seekg(0);
|
||
EXPECT_TRUE(modelFile.ReadModel(str));
|
||
|
||
EXPECT_EQ(modelFile.GetTriangleCount(), 2);
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[0], TRIANGLE_1));
|
||
EXPECT_TRUE(CompareTriangles(modelFile.GetTriangles()[1], TRIANGLE_2));
|
||
}
|