anki-3d-engine/AnKi/Scene/Components/BodyComponent.cpp

// Copyright (C) 2009-present, Panagiotis Christopoulos Charitos and contributors.
// All rights reserved.
// Code licensed under the BSD License.
// http://www.anki3d.org/LICENSE

#include <AnKi/Scene/Components/BodyComponent.h>
#include <AnKi/Scene/Components/ModelComponent.h>
#include <AnKi/Scene/SceneNode.h>
#include <AnKi/Scene/SceneGraph.h>
#include <AnKi/Resource/CpuMeshResource.h>
#include <AnKi/Resource/ResourceManager.h>
#include <AnKi/Physics2/PhysicsWorld.h>
#include <AnKi/Resource/ModelResource.h>

namespace anki {

BodyComponent::BodyComponent(SceneNode* node)
	: SceneComponent(node, kClassType)
	, m_node(node)
{
	node->setIgnoreParentTransform(true);
}

BodyComponent::~BodyComponent()
{
}

void BodyComponent::teleportTo(const Transform& trf)
{
	m_teleportTrf = trf;
	m_teleported = true;

	m_node->setLocalTransform(trf); // Set that just to be sure
}

Error BodyComponent::update(SceneComponentUpdateInfo& info, Bool& updated)
{
	if(m_shapeType == BodyComponentCollisionShapeType::kCount
	   || (m_shapeType == BodyComponentCollisionShapeType::kFromModelComponent && !m_mesh.m_modelc))
	{
		ANKI_ASSERT(!m_body);
		return Error::kNone;
	}

	const Bool shapeDirty =
		!m_body.isCreated()
		|| (m_shapeType == BodyComponentCollisionShapeType::kFromModelComponent && m_mesh.m_modelc->getUuid() != m_mesh.m_modelcUuid);
	if(shapeDirty)
	{
		updated = true;

		v2::PhysicsBodyInitInfo init;
		init.m_mass = m_mass;
		init.m_transform = (m_teleported) ? m_teleportTrf : m_node->getWorldTransform();

		if(m_mass == 0.0f)
		{
			init.m_layer = v2::PhysicsLayer::kStatic;
		}
		else
		{
			init.m_layer = v2::PhysicsLayer::kMoving;
		}

		if(m_shapeType == BodyComponentCollisionShapeType::kFromModelComponent)
		{
			m_mesh.m_modelcUuid = m_mesh.m_modelc->getUuid();

			ANKI_CHECK(m_mesh.m_modelc->getModelResource()->getModelPatches()[0].getMesh()->getOrCreateCollisionShape(
				m_mass == 0.0f, kMaxLodCount - 1, m_collisionShape));
		}
		else if(m_shapeType == BodyComponentCollisionShapeType::kAabb)
		{
			m_collisionShape = v2::PhysicsWorld::getSingleton().newBoxCollisionShape(m_box.m_extend);
		}
		else
		{
			ANKI_ASSERT(m_shapeType == BodyComponentCollisionShapeType::kSphere);
			m_collisionShape = v2::PhysicsWorld::getSingleton().newSphereCollisionShape(m_sphere.m_radius);
		}

		init.m_shape = m_collisionShape.get();

		m_body = v2::PhysicsWorld::getSingleton().newPhysicsBody(init);
		m_body->setUserData(this);

		m_teleported = false; // Cancel teleportation since the body was re-created
	}

	if(m_teleported)
	{
		updated = true;
		m_teleported = false;
		m_body->setTransform(m_teleportTrf);
	}

	if(m_body->getTransform() != info.m_node->getWorldTransform())
	{
		updated = true;
		info.m_node->setLocalTransform(m_body->getTransform());
	}

	if(m_force.getLengthSquared() > 0.0f)
	{
		m_body->applyForce(m_force, m_forcePosition);
		m_force = Vec3(0.0f);
	}

	return Error::kNone;
}

void BodyComponent::onOtherComponentRemovedOrAdded(SceneComponent* other, Bool added)
{
	if(other->getType() != SceneComponentType::kModel)
	{
		return;
	}

	if(added && m_mesh.m_modelc == nullptr)
	{
		m_mesh.m_modelc = static_cast<ModelComponent*>(other);
	}
	else if(!added && m_mesh.m_modelc == other)
	{
		m_mesh.m_modelc = nullptr;
	}
}

} // end namespace anki