GetDepth.cpp

#include "GetDepth.h"
 
#include "Systems/Core/TransformSystem.h"
 
#include "Components/Data/TrajectoryComponent.h"
#include "Components/Core/TransformComponent.h"
#include "Components/Geometry/ExtrusionComponent.h"
#include "../../Extensions/WellLog/Source/Components/ZipComponent.h"
 
#include "Context.h"
#include "EntityStore.h"
 
#include "Foundation/Geometry/BoundingBox.hpp"
#include "Foundation/Geometry/DepthQuery.hpp"
#include "Foundation/Logging/Logger.h"
 
#include <vector>
 
namespace {
  float returnFail(const char* message)
  {
    const Cogs::Logging::Log logger = Cogs::Logging::getLogger("GetDepth");
    LOG_ERROR(logger, "%s", message);
    return 0.0f;
  }
}
 
float Cogs::Core::getDepth(Context * context, const EntityId id, const glm::vec3 position)
{
  const Entity* entity = context->store->getEntityPtr(id);
  if (!entity) {
    return returnFail("Invalid entity id");
  }
 
  const TransformComponent* transform = entity->getComponent<TransformComponent>();
  if (!transform) {
    return returnFail("Missing TransformComponent");
  }
 
  glm::mat4 world2Object = glm::inverse(context->transformSystem->getLocalToWorld(transform));
  glm::vec3 localPosition(world2Object * glm::vec4(position, 1));
 
  {
    const TrajectoryComponent* trajectory = entity->getComponent<TrajectoryComponent>();
    if (trajectory) {
      return Cogs::Geometry::getDepthAtPosition(trajectory->indexes, trajectory->positions, localPosition, trajectory->indexes[0], trajectory->indexes[trajectory->indexes.size() - 1]);
    }
  }
 
  const ExtrusionComponent* extrusion = entity->getComponent<ExtrusionComponent>();
  if (extrusion) {
    if (extrusion->trajectory) {
      // Wellbore
      const TrajectoryComponent* trajectory2 = extrusion->trajectory->getComponent<TrajectoryComponent>();
      
      if (trajectory2) {
        if (trajectory2->indexes.empty()) {
          return returnFail("Empty Extrusion trajectory indexes field");
        }
        else if (trajectory2->indexes.size() != trajectory2->positions.size()) {
          return returnFail("Extrusion trajectory fields indexes and positions size mismatch");
        }
 
        return Cogs::Geometry::getDepthAtPosition(trajectory2->indexes, trajectory2->positions, localPosition, trajectory2->indexes.front(), trajectory2->indexes.back());
      }
    }
  }
 
  const ZipComponent* zip = entity->getComponent<ZipComponent>();
  if (zip) {
    if (zip->extrusion) {
      const ExtrusionComponent* extrusionComponent = zip->extrusion->getComponent<ExtrusionComponent>();
      if (extrusionComponent) {
        // Wellbore
        const TrajectoryComponent* trajectory2 = extrusionComponent->trajectory->getComponent<TrajectoryComponent>();
 
        if (trajectory2) {
          if (trajectory2->indexes.empty()) {
            return returnFail("Empty Zip Extrusion trajectory indexes field");
          }
          else if (trajectory2->indexes.size() != trajectory2->positions.size()) {
            return returnFail("Zip Extrusion trajectory fields indexes and positions size mismatch");
          }
 
          return Cogs::Geometry::getDepthAtPosition(trajectory2->indexes, trajectory2->positions, localPosition, trajectory2->indexes.front(), trajectory2->indexes.back());
        }
      }
    }
  }
 
  return returnFail("Entity does not contain component setup required for Depth calculation");
}