#include "IngameState.hpp"
#include "DebugState.hpp"
#include "DrawUI.hpp"
#include "PauseState.hpp"
#include "RWGame.hpp"

#include <ai/PlayerController.hpp>
#include <data/Model.hpp>
#include <data/WeaponData.hpp>
#include <dynamics/CollisionInstance.hpp>
#include <dynamics/RaycastCallbacks.hpp>
#include <engine/GameState.hpp>
#include <engine/GameWorld.hpp>
#include <objects/CharacterObject.hpp>
#include <objects/ItemPickup.hpp>
#include <objects/VehicleObject.hpp>
#include <script/ScriptMachine.hpp>

#include <glm/gtc/constants.hpp>

constexpr float kAutoLookTime = 2.f;
constexpr float kAutolookMinVelocity = 0.2f;
const float kMaxRotationRate = glm::half_pi<float>();
const float kCameraPitchLimit = glm::quarter_pi<float>() * 0.5f;
const float kVehicleCameraPitch =
    glm::half_pi<float>() - glm::quarter_pi<float>() * 0.25f;

IngameState::IngameState(RWGame* game, bool newgame, const std::string& save)
    : State(game)
    , started(false)
    , save(save)
    , newgame(newgame)
    , autolookTimer(0.f)
    , camMode(IngameState::CAMERA_NORMAL)
    , m_cameraAngles{0.f, glm::half_pi<float>()}
    , m_invertedY(game->getConfig().getInputInvertY())
    , m_vehicleFreeLook(true) {
}

void IngameState::startTest() {
    auto playerChar = getWorld()->createPlayer({270.f, -605.f, 40.f});

    getWorld()->state->playerObject = playerChar->getGameObjectID();

    glm::vec3 itemspawn(276.5f, -609.f, 36.5f);
    for (int i = 1; i < getWorld()->data->weaponData.size(); ++i) {
        auto& item = getWorld()->data->weaponData[i];
        getWorld()->createPickup(itemspawn, item->modelID,
                                 PickupObject::OnStreet);
        itemspawn.x += 2.5f;
    }

    auto carPos = glm::vec3(286.f, -591.f, 37.f);
    auto carRot = glm::angleAxis(glm::radians(90.f), glm::vec3(0.f, 0.f, 1.f));
    // Landstalker, Stinger, Linerunner, Trash, Bobcat
    const std::vector<int> kTestVehicles = {90, 92, 93, 98, 111};
    for (auto id : kTestVehicles) {
        getWorld()->createVehicle(id, carPos, carRot);
        carPos += carRot * glm::vec3(5.f, 0.f, 0.f);
    }
}

void IngameState::startGame() {
    game->startScript("data/main.scm");
    game->getScriptVM()->startThread(0);
    getWorld()->sound.playBackground(getWorld()->data->getDataPath() +
                                     "/audio/City.wav");
}

void IngameState::enter() {
    if (!started) {
        if (newgame) {
            if (save.empty()) {
                startGame();
            } else if (save == "test") {
                startTest();
            } else {
                game->loadGame(save);
            }
        }
        started = true;
    }

    getWindow().hideCursor();
}

void IngameState::exit() {
}

void IngameState::tick(float dt) {
    auto world = getWorld();
    autolookTimer = std::max(autolookTimer - dt, 0.f);

    // Update displayed money value
    // @todo the game uses another algorithm which is non-linear
    {
        float moneyFrequency = 1.0f / 30.0f;
        moneyTimer += dt;
        while (moneyTimer >= moneyFrequency) {
            int32_t difference = world->state->playerInfo.money -
                                 world->state->playerInfo.displayedMoney;

            // Generates 0, 1 (difference < 100), 12 (difference < 1000), 123
            // (difference < 10000), .. etc.
            // Negative input will result in negative output
            auto GetIncrement = [](int32_t difference) -> int32_t {
                // @todo is this correct for difference >= 1000000000 ?
                int32_t r = 1;
                int32_t i = 2;
                if (difference == 0) {
                    return 0;
                }
                while (std::abs(difference) >= 100) {
                    difference /= 10;
                    r = r * 10 + i;
                    i++;
                }
                return (difference < 0) ? -r : r;
            };
            world->state->playerInfo.displayedMoney += GetIncrement(difference);

            moneyTimer -= moneyFrequency;
        }
    }

    auto player = game->getPlayer();

    if (player) {
        // Force all input to 0 if player input is disabled
        /// @todo verify 0ing input is the correct behaviour
        const auto inputEnabled = player->isInputEnabled();

        auto input = [&](GameInputState::Control c) {
            return inputEnabled ? world->state->input[0][c] : 0.f;
        };
        auto pressed = [&](GameInputState::Control c) {
            return inputEnabled && world->state->input[0].pressed(c) &&
                   !world->state->input[1].pressed(c);
        };
        auto held = [&](GameInputState::Control c) {
            return inputEnabled && world->state->input[0].pressed(c);
        };

        float viewDistance = 4.f;
        switch (camMode) {
            case IngameState::CAMERA_CLOSE:
                viewDistance = 2.f;
                break;
            case IngameState::CAMERA_NORMAL:
                viewDistance = 4.0f;
                break;
            case IngameState::CAMERA_FAR:
                viewDistance = 6.f;
                break;
            case IngameState::CAMERA_TOPDOWN:
                viewDistance = 15.f;
                break;
            default:
                viewDistance = 4.f;
        }

        auto target = world->pedestrianPool.find(world->state->cameraTarget);

        if (target == nullptr) {
            target = player->getCharacter();
        }

        glm::vec3 targetPosition = target->getPosition();
        glm::vec3 lookTargetPosition = targetPosition;
        targetPosition += glm::vec3(0.f, 0.f, 1.f);
        lookTargetPosition += glm::vec3(0.f, 0.f, 0.5f);

        btCollisionObject* physTarget = player->getCharacter()->physObject;

        auto vehicle =
            (target->type() == GameObject::Character)
                ? static_cast<CharacterObject*>(target)->getCurrentVehicle()
                : nullptr;
        if (vehicle) {
            auto model = vehicle->getModel();
            float maxDist = 0.f;
            for (auto& g : model->geometries) {
                float partSize = glm::length(g->geometryBounds.center) +
                                 g->geometryBounds.radius;
                maxDist = std::max(partSize, maxDist);
            }
            viewDistance = viewDistance + maxDist;
            targetPosition = vehicle->getPosition();
            lookTargetPosition = targetPosition;
            lookTargetPosition.z +=
                (vehicle->info->handling.dimensions.z * 0.5f);
            targetPosition.z += (vehicle->info->handling.dimensions.z * 0.5f);
            physTarget = vehicle->collision->getBulletBody();

            if (!m_vehicleFreeLook) {
                m_cameraAngles.y = kVehicleCameraPitch;
            }

            // Rotate the camera to the ideal angle if the player isn't moving
            // it
            float velocity = vehicle->getVelocity();
            if (autolookTimer <= 0.f &&
                glm::abs(velocity) > kAutolookMinVelocity) {
                auto idealYaw =
                    -glm::roll(vehicle->getRotation()) + glm::half_pi<float>();
                const auto idealPitch = kVehicleCameraPitch;
                if (velocity < 0.f) {
                    idealYaw = glm::mod(idealYaw - glm::pi<float>(),
                                        glm::pi<float>() * 2.f);
                }
                float currentYaw =
                    glm::mod(m_cameraAngles.x, glm::pi<float>() * 2);
                float currentPitch = m_cameraAngles.y;
                float deltaYaw = idealYaw - currentYaw;
                float deltaPitch = idealPitch - currentPitch;
                if (glm::abs(deltaYaw) > glm::pi<float>()) {
                    deltaYaw -= glm::sign(deltaYaw) * glm::pi<float>() * 2.f;
                }
                m_cameraAngles.x +=
                    glm::sign(deltaYaw) *
                    std::min(kMaxRotationRate * dt, glm::abs(deltaYaw));
                m_cameraAngles.y +=
                    glm::sign(deltaPitch) *
                    std::min(kMaxRotationRate * dt, glm::abs(deltaPitch));
            }
        }

        // Non-topdown camera can orbit
        if (camMode != IngameState::CAMERA_TOPDOWN) {
            bool lookleft = held(GameInputState::LookLeft);
            bool lookright = held(GameInputState::LookRight);
            if ((lookleft || lookright) && vehicle != nullptr) {
                auto rotation = vehicle->getRotation();
                if (!lookright) {
                    rotation *= glm::angleAxis(glm::half_pi<float>(),
                                               glm::vec3(0.f, 0.f, -1.f));
                } else if (!lookleft) {
                    rotation *= glm::angleAxis(glm::half_pi<float>(),
                                               glm::vec3(0.f, 0.f, 1.f));
                }
                cameraPosition = targetPosition +
                                 rotation * glm::vec3(0.f, viewDistance, 0.f);
            } else {
                // Determine the "ideal" camera position for the current view
                // angles
                auto yaw =
                    glm::angleAxis(m_cameraAngles.x, glm::vec3(0.f, 0.f, -1.f));
                auto pitch =
                    glm::angleAxis(m_cameraAngles.y, glm::vec3(0.f, 1.f, 0.f));
                auto cameraOffset =
                    yaw * pitch * glm::vec3(0.f, 0.f, viewDistance);
                cameraPosition = targetPosition + cameraOffset;
            }
        } else {
            cameraPosition = targetPosition + glm::vec3(0.f, 0.f, viewDistance);
        }

        glm::quat angle;

        auto camtotarget = targetPosition - cameraPosition;
        auto dir = glm::normalize(camtotarget);
        float correction = glm::length(camtotarget) - viewDistance;
        if (correction < 0.f) {
            float innerDist = viewDistance * 0.1f;
            correction = glm::min(0.f, correction + innerDist);
        }
        cameraPosition += dir * 10.f * correction * dt;

        auto lookdir = glm::normalize(lookTargetPosition - cameraPosition);
        // Calculate the yaw to look at the target.
        float angleYaw = glm::atan(lookdir.y, lookdir.x);
        angle = glm::quat(glm::vec3(0.f, 0.f, angleYaw));
        glm::vec3 movement;

        movement.x = input(GameInputState::GoForward) -
                     input(GameInputState::GoBackwards);
        movement.y =
            input(GameInputState::GoLeft) - input(GameInputState::GoRight);
        /// @todo replace with correct sprint behaviour
        float speed = held(GameInputState::Sprint) ? 2.f : 1.f;

        player->setRunning(!held(GameInputState::Walk));
        /// @todo find the correct behaviour for entering & exiting
        if (pressed(GameInputState::EnterExitVehicle)) {
            /// @todo move me
            if (player->getCharacter()->getCurrentVehicle()) {
                player->exitVehicle();
            } else if (!player->isCurrentActivity(
                           Activities::EnterVehicle::ActivityName)) {
                player->enterNearestVehicle();
            }
        } else if (glm::length2(movement) > 0.001f) {
            if (player->isCurrentActivity(
                    Activities::EnterVehicle::ActivityName)) {
                // Give up entering a vehicle if we're alreadying doing so
                player->skipActivity();
            }
        }

        if (player->getCharacter()->getCurrentVehicle()) {
            auto vehicle = player->getCharacter()->getCurrentVehicle();
            vehicle->setHandbraking(held(GameInputState::Handbrake));
            player->setMoveDirection(movement);
        } else {
            if (pressed(GameInputState::Jump)) {
                player->jump();
            }

            float length = glm::length(movement);
            float movementAngle = angleYaw - glm::half_pi<float>();
            if (length > 0.1f) {
                glm::vec3 direction = glm::normalize(movement);
                movementAngle += atan2(direction.y, direction.x);
                player->setMoveDirection(glm::vec3(speed, 0.f, 0.f));
            } else {
                player->setMoveDirection(glm::vec3(0.f));
            }
            player->setLookDirection({movementAngle, 0.f});
        }

        float len2d = glm::length(glm::vec2(lookdir));
        float anglePitch = glm::atan(lookdir.z, len2d);
        angle *= glm::quat(glm::vec3(0.f, -anglePitch, 0.f));

        // Use rays to ensure target is visible from cameraPosition
        auto rayEnd = cameraPosition;
        auto rayStart = targetPosition;
        auto to = btVector3(rayEnd.x, rayEnd.y, rayEnd.z);
        auto from = btVector3(rayStart.x, rayStart.y, rayStart.z);
        ClosestNotMeRayResultCallback ray(physTarget, from, to);

        world->dynamicsWorld->rayTest(from, to, ray);
        if (ray.hasHit() && ray.m_closestHitFraction < 1.f) {
            cameraPosition =
                glm::vec3(ray.m_hitPointWorld.x(), ray.m_hitPointWorld.y(),
                          ray.m_hitPointWorld.z());
            cameraPosition +=
                glm::vec3(ray.m_hitNormalWorld.x(), ray.m_hitNormalWorld.y(),
                          ray.m_hitNormalWorld.z()) *
                0.1f;
        }

        _look.position = cameraPosition;
        _look.rotation = angle;
    }
}

void IngameState::draw(GameRenderer* r) {
    if (!getWorld()->state->isCinematic && getWorld()->isCutsceneDone()) {
        drawHUD(_look, game->getPlayer(), getWorld(), r);
    }

    State::draw(r);
}

void IngameState::handleEvent(const SDL_Event& event) {
    auto player = game->getPlayer();

    switch (event.type) {
        case SDL_KEYDOWN:
            switch (event.key.keysym.sym) {
                case SDLK_ESCAPE:
                    StateManager::get().enter<PauseState>(game);
                    break;
                case SDLK_m:
                    StateManager::get().enter<DebugState>(game, _look.position,
                                                          _look.rotation);
                    break;
                case SDLK_SPACE:
                    if (getWorld()->state->currentCutscene) {
                        getWorld()->state->skipCutscene = true;
                    }
                    break;
                case SDLK_c:
                    camMode =
                        CameraMode((camMode + (CameraMode)1) % CAMERA_MAX);
                    break;
                default:
                    break;
            }
            break;
        default:
            break;
    }

    if (player && player->isInputEnabled()) {
        handlePlayerInput(event);
    }
    State::handleEvent(event);
}

void IngameState::handlePlayerInput(const SDL_Event& event) {
    auto player = game->getPlayer();
    switch (event.type) {
        case SDL_MOUSEBUTTONDOWN:
            switch (event.button.button) {
                case SDL_BUTTON_LEFT:
                    player->getCharacter()->useItem(true, true);
                    break;
                default:
                    break;
            }
            break;
        case SDL_MOUSEBUTTONUP:
            switch (event.button.button) {
                case SDL_BUTTON_LEFT:
                    player->getCharacter()->useItem(false, true);
                    break;
                default:
                    break;
            }
            break;
        case SDL_MOUSEWHEEL:
            player->getCharacter()->cycleInventory(event.wheel.y > 0);
            break;
        case SDL_MOUSEMOTION:
            if (game->hasFocus()) {
                glm::ivec2 screenSize = getWindow().getSize();
                glm::vec2 mouseMove(
                    event.motion.xrel / static_cast<float>(screenSize.x),
                    event.motion.yrel / static_cast<float>(screenSize.y));

                autolookTimer = kAutoLookTime;
                if (!m_invertedY) {
                    mouseMove.y = -mouseMove.y;
                }
                m_cameraAngles += glm::vec2(mouseMove.x, mouseMove.y);
                m_cameraAngles.y =
                    glm::clamp(m_cameraAngles.y, kCameraPitchLimit,
                               glm::pi<float>() - kCameraPitchLimit);
            }
            break;
        default:
            break;
    }
}

bool IngameState::shouldWorldUpdate() {
    return true;
}

const ViewCamera& IngameState::getCamera() {
    return _look;
}