colobot/src/graphics/engine/oldmodelmanager.cpp

218 lines
6.5 KiB
C++

/*
* This file is part of the Colobot: Gold Edition source code
* Copyright (C) 2001-2020, Daniel Roux, EPSITEC SA & TerranovaTeam
* http://epsitec.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/oldmodelmanager.h"
#include "app/app.h"
#include "common/logger.h"
#include "common/stringutils.h"
#include "common/resources/inputstream.h"
#include "graphics/engine/engine.h"
#include "graphics/model/model_input.h"
#include "graphics/model/model_io_exception.h"
#include <cstdio>
namespace Gfx
{
COldModelManager::COldModelManager(CEngine* engine)
{
m_engine = engine;
}
COldModelManager::~COldModelManager()
{
}
bool COldModelManager::LoadModel(const std::string& fileName, bool mirrored, int variant)
{
GetLogger()->Debug("Loading model '%s'\n", fileName.c_str());
CModel model;
try
{
CInputStream stream;
stream.open("models/" + fileName);
if (!stream.is_open())
throw CModelIOException(std::string("Could not open file '") + fileName + "'");
std::string::size_type extension_index = fileName.find_last_of('.');
if (extension_index == std::string::npos)
throw CModelIOException(std::string("Filename '") + fileName + "' has no extension");
std::string extension = fileName.substr(extension_index + 1);
if (extension == "mod")
model = ModelInput::Read(stream, ModelFormat::Old);
else if (extension == "txt")
model = ModelInput::Read(stream, ModelFormat::Text);
else
throw CModelIOException(std::string("Filename '") + fileName + "' has unknown extension");
}
catch (const CModelIOException& e)
{
GetLogger()->Error("Loading model '%s' failed: %s\n", fileName.c_str(), e.what());
return false;
}
CModelMesh* mesh = model.GetMesh("main");
assert(mesh != nullptr);
ModelInfo modelInfo;
modelInfo.baseObjRank = m_engine->CreateBaseObject();
modelInfo.triangles = mesh->GetTriangles();
if (mirrored)
Mirror(modelInfo.triangles);
if (variant != 0)
ChangeVariant(modelInfo.triangles, variant);
FileInfo fileInfo(fileName, mirrored, variant);
m_models[fileInfo] = modelInfo;
m_engine->AddBaseObjTriangles(modelInfo.baseObjRank, modelInfo.triangles);
return true;
}
bool COldModelManager::AddModelReference(const std::string& fileName, bool mirrored, int objRank, int variant)
{
auto it = m_models.find(FileInfo(fileName, mirrored, variant));
if (it == m_models.end())
{
if (!LoadModel(fileName, mirrored, variant))
return false;
it = m_models.find(FileInfo(fileName, mirrored, variant));
}
m_engine->SetObjectBaseRank(objRank, (*it).second.baseObjRank);
return true;
}
bool COldModelManager::AddModelCopy(const std::string& fileName, bool mirrored, int objRank, int variant)
{
auto it = m_models.find(FileInfo(fileName, mirrored, variant));
if (it == m_models.end())
{
if (!LoadModel(fileName, mirrored, variant))
return false;
it = m_models.find(FileInfo(fileName, mirrored, variant));
}
int copyBaseObjRank = m_engine->CreateBaseObject();
m_engine->CopyBaseObject((*it).second.baseObjRank, copyBaseObjRank);
m_engine->SetObjectBaseRank(objRank, copyBaseObjRank);
m_copiesBaseRanks.push_back(copyBaseObjRank);
return true;
}
bool COldModelManager::IsModelLoaded(const std::string& fileName, bool mirrored, int variant)
{
return m_models.count(FileInfo(fileName, mirrored, variant)) > 0;
}
int COldModelManager::GetModelBaseObjRank(const std::string& fileName, bool mirrored, int variant)
{
auto it = m_models.find(FileInfo(fileName, mirrored, variant));
if (it == m_models.end())
return -1;
return (*it).second.baseObjRank;
}
void COldModelManager::DeleteAllModelCopies()
{
for (int baseObjRank : m_copiesBaseRanks)
{
m_engine->DeleteBaseObject(baseObjRank);
}
m_copiesBaseRanks.clear();
}
void COldModelManager::UnloadModel(const std::string& fileName, bool mirrored, int variant)
{
auto it = m_models.find(FileInfo(fileName, mirrored, variant));
if (it == m_models.end())
return;
m_engine->DeleteBaseObject((*it).second.baseObjRank);
m_models.erase(it);
}
void COldModelManager::UnloadAllModels()
{
for (auto& mf : m_models)
m_engine->DeleteBaseObject(mf.second.baseObjRank);
m_models.clear();
}
void COldModelManager::Mirror(std::vector<ModelTriangle>& triangles)
{
for (int i = 0; i < static_cast<int>( triangles.size() ); i++)
{
VertexTex2 t = triangles[i].p1;
triangles[i].p1 = triangles[i].p2;
triangles[i].p2 = t;
triangles[i].p1.position.z = -triangles[i].p1.position.z;
triangles[i].p2.position.z = -triangles[i].p2.position.z;
triangles[i].p3.position.z = -triangles[i].p3.position.z;
triangles[i].p1.normal.z = -triangles[i].p1.normal.z;
triangles[i].p2.normal.z = -triangles[i].p2.normal.z;
triangles[i].p3.normal.z = -triangles[i].p3.normal.z;
}
}
void COldModelManager::ChangeVariant(std::vector<ModelTriangle>& triangles, int variant)
{
for (int i = 0; i < static_cast<int>( triangles.size() ); i++)
{
if (triangles[i].tex1Name == "base1.png" ||
triangles[i].tex1Name == "convert.png" ||
triangles[i].tex1Name == "derrick.png" ||
triangles[i].tex1Name == "factory.png" ||
triangles[i].tex1Name == "lemt.png" ||
triangles[i].tex1Name == "roller.png" ||
triangles[i].tex1Name == "rollert.png" ||
triangles[i].tex1Name == "search.png" ||
triangles[i].tex1Name == "drawer.png" ||
triangles[i].tex1Name == "subm.png" )
{
triangles[i].tex1Name += StrUtils::ToString<int>(variant);
}
}
}
}