/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/

#include <AzFramework/Components/TransformComponent.h>

#include <RigidBodyComponent.h>
#include <StaticRigidBodyComponent.h>

namespace PhysX
{
    namespace TestUtils
    {
        template<typename ColliderType>
        EntityPtr AddUnitTestObject(const AZ::Vector3& position, const char* name)
        {
            EntityPtr entity = AZStd::make_shared<AZ::Entity>(name);

            AZ::TransformConfig transformConfig;
            transformConfig.m_worldTransform = AZ::Transform::CreateTranslation(position);
            entity->CreateComponent<AzFramework::TransformComponent>()->SetConfiguration(transformConfig);
            auto colliderComponent = entity->CreateComponent<ColliderType>();
            auto colliderConfig = AZStd::make_shared<Physics::ColliderConfiguration>();
            auto shapeConfig = AZStd::make_shared<typename ColliderType::Configuration>();
            colliderComponent->SetShapeConfigurationList({ AZStd::make_pair(colliderConfig, shapeConfig) });
            Physics::RigidBodyConfiguration rigidBodyConfig;
            rigidBodyConfig.m_computeMass = false;
            entity->CreateComponent<RigidBodyComponent>(rigidBodyConfig);
            entity->Init();
            entity->Activate();
            return entity;
        }

        template<typename ColliderType>
        EntityPtr AddStaticUnitTestObject(const AZ::Vector3& position, const char* name)
        {
            EntityPtr entity = AZStd::make_shared<AZ::Entity>(name);

            AZ::TransformConfig transformConfig;
            transformConfig.m_worldTransform = AZ::Transform::CreateTranslation(position);
            entity->CreateComponent<AzFramework::TransformComponent>()->SetConfiguration(transformConfig);
            auto colliderComponent = entity->CreateComponent<ColliderType>();
            auto colliderConfig = AZStd::make_shared<Physics::ColliderConfiguration>();
            auto shapeConfig = AZStd::make_shared<typename ColliderType::Configuration>();
            colliderComponent->SetShapeConfigurationList({ AZStd::make_pair(colliderConfig, shapeConfig) });

            entity->CreateComponent<StaticRigidBodyComponent>();
            
            entity->Init();
            entity->Activate();
            return entity;
        }

        template<typename ColliderT>
        EntityPtr CreateTriggerAtPosition(const AZ::Vector3& position)
        {
            EntityPtr triggerEntity = AZStd::make_shared<AZ::Entity>("TriggerEntity");

            AZ::TransformConfig transformConfig;
            transformConfig.m_worldTransform = AZ::Transform::CreateTranslation(position);
            triggerEntity->CreateComponent<AzFramework::TransformComponent>()->SetConfiguration(transformConfig);

            auto colliderConfiguration = AZStd::make_shared<Physics::ColliderConfiguration>();
            colliderConfiguration->m_isTrigger = true;
            auto shapeConfiguration = AZStd::make_shared<typename ColliderT::Configuration>();
            auto colliderComponent = triggerEntity->CreateComponent<ColliderT>();
            colliderComponent->SetShapeConfigurationList({ AZStd::make_pair(colliderConfiguration, shapeConfiguration) });

            triggerEntity->CreateComponent<StaticRigidBodyComponent>();

            triggerEntity->Init();
            triggerEntity->Activate();
            
            return triggerEntity;
        }

        template<typename ColliderT>
        EntityPtr CreateDynamicTriggerAtPosition(const AZ::Vector3& position)
        {
            EntityPtr triggerEntity = AZStd::make_shared<AZ::Entity>("DynamicTriggerEntity");

            AZ::TransformConfig transformConfig;
            transformConfig.m_worldTransform = AZ::Transform::CreateTranslation(position);
            triggerEntity->CreateComponent<AzFramework::TransformComponent>()->SetConfiguration(transformConfig);

            auto colliderConfiguration = AZStd::make_shared<Physics::ColliderConfiguration>();
            colliderConfiguration->m_isTrigger = true;
            auto shapeConfiguration = AZStd::make_shared<typename ColliderT::Configuration>();
            auto colliderComponent = triggerEntity->CreateComponent<ColliderT>();
            colliderComponent->SetShapeConfigurationList({ AZStd::make_pair(colliderConfiguration, shapeConfiguration) });

            triggerEntity->CreateComponent<RigidBodyComponent>();

            triggerEntity->Init();
            triggerEntity->Activate();

            return triggerEntity;
        }
    }
}