// Copyright (c) 2018-2019 Kiwano - Nomango
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#include <kiwano-physics/Body.h>
#include <kiwano-physics/World.h>

namespace kiwano
{
namespace physics
{

BodyPtr Body::Create(World* world, Actor* actor, Type type)
{
    BodyPtr ptr = new (std::nothrow) Body;
    if (ptr)
    {
        if (!ptr->InitBody(world, actor))
            return nullptr;

        ptr->SetType(type);
    }
    return ptr;
}

Body::Body()
    : body_(nullptr)
    , actor_(nullptr)
    , world_(nullptr)
    , category_bits_(0x0001)
    , mask_bits_(0xFFFF)
    , group_index_(0)
{
}

Body::~Body()
{
    Destroy();
}

bool Body::InitBody(World* world, Actor* actor)
{
    KGE_ASSERT(world);

    Destroy();

    world_ = world;
    b2BodyDef def;
    b2Body*   b2body = world->GetB2World()->CreateBody(&def);

    if (b2body)
    {
        SetB2Body(b2body);
        SetActor(actor);
        UpdateFromActor();
        return true;
    }
    return false;
}

void Body::AddFixture(FixturePtr fixture)
{
    fixtures_.push_back(fixture);
}

Fixture* Body::AddCircleShape(float radius, float density, float friction, float restitution, bool is_sensor)
{
    Fixture::Param param(density, friction, restitution, is_sensor);
    FixturePtr     fixture = Fixture::CreateCircle(this, param, radius);
    AddFixture(fixture);
    return fixture.get();
}

Fixture* Body::AddRectShape(Vec2 const& size, float density, float friction, float restitution, bool is_sensor)
{
    Fixture::Param param(density, friction, restitution, is_sensor);
    FixturePtr     fixture = Fixture::CreateRect(this, param, size);
    AddFixture(fixture);
    return fixture.get();
}

Fixture* Body::AddPolygonShape(Vector<Point> const& vertexs, float density, float friction, float restitution,
                               bool is_sensor)
{
    Fixture::Param param(density, friction, restitution, is_sensor);
    FixturePtr     fixture = Fixture::CreatePolygon(this, param, vertexs);
    AddFixture(fixture);
    return fixture.get();
}

Fixture* Body::AddEdgeShape(Point const& p1, Point const& p2, float density, float friction, float restitution,
                            bool is_sensor)
{
    Fixture::Param param(density, friction, restitution, is_sensor);
    FixturePtr     fixture = Fixture::CreateEdge(this, param, p1, p2);
    AddFixture(fixture);
    return fixture.get();
}

Fixture* Body::AddChainShape(Vector<Point> const& vertexs, bool loop, float density, float friction, float restitution,
                             bool is_sensor)
{
    Fixture::Param param(density, friction, restitution, is_sensor);
    FixturePtr     fixture = Fixture::CreateChain(this, param, vertexs, loop);
    AddFixture(fixture);
    return fixture.get();
}

void Body::RemoveFixture(FixturePtr fixture)
{
    if (fixture)
    {
        body_->DestroyFixture(fixture->GetB2Fixture());

        for (auto iter = fixtures_.begin(); iter != fixtures_.end();)
        {
            if (*iter == fixture)
                iter = fixtures_.erase(iter);
            else
                ++iter;
        }
    }
}

void Body::SetCategoryBits(uint16_t category_bits)
{
    KGE_ASSERT(body_);

    if (category_bits != category_bits_)
    {
        category_bits_ = category_bits;

        b2Fixture* fixture = body_->GetFixtureList();
        while (fixture)
        {
            UpdateFixtureFilter(fixture);
            fixture = fixture->GetNext();
        }
    }
}

void Body::SetMaskBits(uint16_t mask_bits)
{
    KGE_ASSERT(body_);

    if (mask_bits != mask_bits_)
    {
        mask_bits_ = mask_bits;

        b2Fixture* fixture = body_->GetFixtureList();
        while (fixture)
        {
            UpdateFixtureFilter(fixture);
            fixture = fixture->GetNext();
        }
    }
}

void Body::SetGroupIndex(int16_t index)
{
    KGE_ASSERT(body_);

    if (index != group_index_)
    {
        group_index_ = index;

        b2Fixture* fixture = body_->GetFixtureList();
        while (fixture)
        {
            UpdateFixtureFilter(fixture);
            fixture = fixture->GetNext();
        }
    }
}

void Body::GetMassData(float* mass, Point* center, float* inertia) const
{
    KGE_ASSERT(body_ && world_);

    b2MassData data;
    body_->GetMassData(&data);

    if (mass)
        *mass = data.mass;
    if (center)
        *center = world_->World2Stage(data.center);
    if (inertia)
        *inertia = data.I;
}

void Body::SetMassData(float mass, Point const& center, float inertia)
{
    KGE_ASSERT(body_ && world_);

    b2MassData data;
    data.mass   = mass;
    data.center = world_->Stage2World(center);
    data.I      = inertia;
    body_->SetMassData(&data);
}

void Body::ResetMassData()
{
    KGE_ASSERT(body_);
    body_->ResetMassData();
}

Point Body::GetBodyPosition() const
{
    KGE_ASSERT(body_ && world_);
    return world_->World2Stage(body_->GetPosition());
}

void Body::SetBodyTransform(Point const& pos, float angle)
{
    KGE_ASSERT(body_ && world_);
    body_->SetTransform(world_->Stage2World(pos), math::Degree2Radian(angle));
}

Point Body::GetLocalPoint(Point const& world) const
{
    KGE_ASSERT(body_ && world_);
    return world_->World2Stage(body_->GetLocalPoint(world_->Stage2World(world)));
}

Point Body::GetWorldPoint(Point const& local) const
{
    KGE_ASSERT(body_ && world_);
    return world_->World2Stage(body_->GetWorldPoint(world_->Stage2World(local)));
}

Point Body::GetLocalCenter() const
{
    KGE_ASSERT(body_ && world_);
    return world_->World2Stage(body_->GetLocalCenter());
}

Point Body::GetWorldCenter() const
{
    KGE_ASSERT(body_ && world_);
    return world_->World2Stage(body_->GetWorldCenter());
}

void Body::ApplyForce(Vec2 const& force, Point const& point, bool wake)
{
    KGE_ASSERT(body_ && world_);
    body_->ApplyForce(b2Vec2(force.x, force.y), world_->Stage2World(point), wake);
}

void Body::ApplyForceToCenter(Vec2 const& force, bool wake)
{
    KGE_ASSERT(body_ && world_);
    body_->ApplyForceToCenter(b2Vec2(force.x, force.y), wake);
}

void Body::ApplyTorque(float torque, bool wake)
{
    KGE_ASSERT(body_ && world_);
    body_->ApplyTorque(torque, wake);
}

void Body::SetB2Body(b2Body* body)
{
    body_ = body;
    if (body_)
    {
        body_->SetUserData(this);
    }
}

void Body::Destroy()
{
    fixtures_.clear();

    if (world_)
    {
        world_->RemoveBody(this);
    }

    body_  = nullptr;
    world_ = nullptr;
    actor_ = nullptr;
}

void Body::UpdateActor()
{
    if (actor_ && body_)
    {
        if (world_)
        {
            actor_->SetPosition(world_->World2Stage(body_->GetPosition()));
        }
        else
        {
            actor_->SetPosition(World2Stage(body_->GetPosition()));
        }
        actor_->SetRotation(math::Radian2Degree(body_->GetAngle()));
    }
}

void Body::UpdateFromActor()
{
    if (actor_ && body_)
    {
        if (world_)
        {
            body_->SetTransform(world_->Stage2World(actor_->GetPosition()), math::Degree2Radian(actor_->GetRotation()));
        }
        else
        {
            body_->SetTransform(Stage2World(actor_->GetPosition()), math::Degree2Radian(actor_->GetRotation()));
        }
    }
}

void Body::UpdateFixtureFilter(b2Fixture* fixture)
{
    b2Filter filter;
    filter.categoryBits = category_bits_;
    filter.maskBits     = mask_bits_;
    filter.groupIndex   = group_index_;
    fixture->SetFilterData(filter);
}

}  // namespace physics
}  // namespace kiwano