ExportGltfCommand.cpp

#include "ExportGltfCommand.h"
#include "Serialization/GltfWriter.h"
#include "Components/Core/ModelComponent.h"
#include "Components/Core/MeshComponent.h"
#include "Components/Core/SceneComponent.h"
#include "Components/Core/StaticModelComponent.h"
 
#include "Foundation/Logging/Logger.h"
 
namespace
{
  const Cogs::Logging::Log logger = Cogs::Logging::getLogger("ExportGltfCommand");
 
  void getEntity(Cogs::Core::Context* /*context*/, std::vector<Cogs::ComponentModel::Entity*>& entities, Cogs::ComponentModel::Entity* entity)
  {
    if (entity == nullptr) return;
 
    if (auto result = std::find_if(entities.begin(), entities.end(), [e = entity](Cogs::ComponentModel::Entity*& x) { return x->getId() == e->getId(); });
      result != entities.end()) {
      return;
    }
 
    Cogs::Core::SceneComponent* sceneComp = entity->getComponent<Cogs::Core::SceneComponent>();
    if (!sceneComp) return;
 
    Cogs::Core::MeshComponent* meshComp = entity->getComponent<Cogs::Core::MeshComponent>();
    Cogs::Core::StaticModelComponent* staticModelComp = entity->getComponent<Cogs::Core::StaticModelComponent>();
    if (!meshComp && !staticModelComp) return;
 
    LOG_DEBUG(logger, "export entity id: %zu", entity->getId());
    entities.push_back(entity);
  }
}
 
#ifndef EMSCRIPTEN
void Cogs::Core::ExportGltfCommand::apply()
{
 
  std::vector<ComponentModel::Entity*> entities;
  if (exportScene) {
    const std::unordered_map<EntityId, EntityPtr>& entityMap = context->store->getEntities();
 
    for (const auto& e : entityMap) {
      Cogs::ComponentModel::Entity* entity = context->store->getEntityPtr(static_cast<EntityId>(e.first));
      getEntity(context, entities, entity);
    }
  }
  else {
    for (auto id : state->selected) {
      ComponentModel::Entity* entity = context->store->getEntityPtr(id);
      if (entity) {
        entities.push_back(context->store->getEntityPtr(id));
      }
    }
  }
 
  bool ok;
  if (exportAsGlb == true) {
    ok = GltfWriter::writeGlb(entities, filePath, fileName);
  }
  else {
    ok = GltfWriter::writeGltf(entities, filePath, fileName);
  }
 
  if (!ok) {
    LOG_ERROR(logger, "Save %s failed. Path: %s", (exportAsGlb ? "GLB" : "GLTF"), filePath.c_str());
  }
}
#else
void Cogs::Core::ExportGltfCommand::apply() {}
#endif
 
void Cogs::Core::ExportGltfCommand::undo()
{
}