#include "CharacterGame.h"

// Declare our game instance
CharacterGame game;

// Input flags
#define NORTH 1
#define SOUTH 2
#define EAST 4
#define WEST 8
#define RUNNING 16

// Character defines
#define WALK_SPEED  5.0f
#define STRAFE_SPEED 1.5f
#define RUN_SPEED 15.0f
#define CAMERA_FOCUS_DISTANCE 16.0f

#define BUTTON_1 0
#define BUTTON_2 1

CharacterGame::CharacterGame()
    : _font(NULL), _scene(NULL), _character(NULL), _characterNode(NULL), _characterMeshNode(NULL), _characterShadowNode(NULL), _basketballNode(NULL),
      _animation(NULL), _currentClip(NULL), _jumpClip(NULL), _kickClip(NULL), _rotateX(0), _materialParameterAlpha(NULL),
      _keyFlags(0), _physicsDebug(false), _wireframe(false), _hasBall(false), _applyKick(false), _gamepad(NULL)
{
    _buttonPressed = new bool[2];
}

void CharacterGame::initialize()
{
    // Enable multi-touch (only affects devices that support multi-touch).
    setMultiTouch(true);

    // Display the gameplay splash screen for at least 1 second.
    displayScreen(this, &CharacterGame::drawSplash, NULL, 1000L);

    // Load the font.
    _font = Font::create("res/ui/arial.gpb");

    // Load scene.
    _scene = Scene::load("res/common/sample.scene");

    // Update the aspect ratio for our scene's camera to match the current device resolution.
    _scene->getActiveCamera()->setAspectRatio(getAspectRatio());
    
    // Initialize the physics character.
    initializeCharacter();

    // Create a collision object for the ceiling.
    Node* ceiling = _scene->addNode("ceiling");
    ceiling->setTranslationY(14.5f);
    PhysicsRigidBody::Parameters rbParams;
    rbParams.mass = 0.0f;
    rbParams.friction = 0.5f;
    rbParams.restitution = 0.75f;
    rbParams.linearDamping = 0.025f;
    rbParams.angularDamping = 0.16f;
    ceiling->setCollisionObject(PhysicsCollisionObject::RIGID_BODY, PhysicsCollisionShape::box(Vector3(49.5f, 1.0f, 49.5f)), &rbParams);

    // Initialize scene.
    _scene->visit(this, &CharacterGame::initializeScene);

    _gamepad = getGamepad(0);
}

bool CharacterGame::initializeScene(Node* node)
{
    Model* model = node->getModel();
    if (model && model->getMaterial())
    {
        initializeMaterial(_scene, node, model->getMaterial());
    }

    return true;
}

void CharacterGame::initializeMaterial(Scene* scene, Node* node, Material* material)
{
    // Bind light shader parameters to dynamic objects only
    if (node->hasTag("dynamic"))
    {
        material->getParameter("u_ambientColor")->bindValue(scene, &Scene::getAmbientColor);
        Node* lightNode = scene->findNode("sun");
        if (lightNode)
        {
            material->getParameter("u_directionalLightColor[0]")->bindValue(lightNode->getLight(), &Light::getColor);
            material->getParameter("u_directionalLightDirection[0]")->bindValue(lightNode, &Node::getForwardVectorView);
        }
    }
}

void CharacterGame::initializeCharacter()
{
    Node* node = _scene->findNode("boycharacter");
    
    // Store the physics character object.
    _character = static_cast<PhysicsCharacter*>(node->getCollisionObject());
    
    // Store character nodes.
    _characterNode = node->findNode("boyScale");
    _characterMeshNode = _scene->findNode("boymesh");
    _characterShadowNode = _scene->findNode("boyshadow");

    // Get the basketball node.
    _basketballNode = _scene->findNode("basketball");
    _basketballNode->getCollisionObject()->addCollisionListener(this);

    _floorLevel = _basketballNode->getTranslationY();

    // Store the alpha material parameter from the character's model.
    _materialParameterAlpha = _characterMeshNode->getModel()->getMaterial()->getTechniqueByIndex(0)->getPassByIndex(0)->getParameter("u_modulateAlpha");

    // Load character animations.
    _animation = node->getAnimation("animations");
    _animation->createClips("res/common/boy.animation");
    _jumpClip = _animation->getClip("jump");
    _jumpClip->addListener(this, _jumpClip->getDuration() - 250);
    _kickClip = _animation->getClip("kick");
    // when to cross fade
    _kickClip->addListener(this, _kickClip->getDuration() - 250); 
    // when to turn on _isKicking.
    _kickClip->addListener(this, 416);  

    // Start playing the idle animation when we load.
    play("idle", true);
}

void CharacterGame::finalize()
{
    SAFE_RELEASE(_scene);
    SAFE_RELEASE(_font);
    SAFE_DELETE_ARRAY(_buttonPressed);
}

void CharacterGame::drawSplash(void* param)
{
    clear(CLEAR_COLOR_DEPTH, Vector4(0, 0, 0, 1), 1.0f, 0);
    SpriteBatch* batch = SpriteBatch::create("res/logo_powered_white.png");
    batch->start();
    batch->draw(getWidth() * 0.5f, getHeight() * 0.5f, 0.0f, 512.0f, 512.0f, 0.0f, 1.0f, 1.0f, 0.0f, Vector4::one(), true);
    batch->finish();
    SAFE_DELETE(batch);
}

bool CharacterGame::drawScene(Node* node, bool transparent)
{
    if (node->getModel() && (transparent == node->hasTag("transparent")))
        node->getModel()->draw(_wireframe);

    return true;
}

void CharacterGame::play(const char* id, bool repeat, float speed)
{
    AnimationClip* clip = _animation->getClip(id);

    // Set clip properties
    clip->setSpeed(speed);
    clip->setRepeatCount(repeat ? AnimationClip::REPEAT_INDEFINITE : 1);

    // Is the clip already playing?
    if (clip == _currentClip && clip->isPlaying())
        return;

    if (_jumpClip->isPlaying() || _kickClip->isPlaying())
    {
        _currentClip = clip;
        return;
    }

    // If a current clip is playing, crossfade into the new one
    if (_currentClip && _currentClip->isPlaying())
    {
        _currentClip->crossFade(clip, 150);
    }
    else
    {
        clip->play();
    }
    _currentClip = clip;
}

void CharacterGame::jump()
{
    if (isOnFloor() && !_kickClip->isPlaying())
    {
        play("jump", false, 0.55f);
        _character->jump(3.0f);
    }
}

void CharacterGame::kick()
{
    if (!_jumpClip->isPlaying())
        play("kick", false, 1.75f);
    _kicking = true;
}

bool CharacterGame::isOnFloor() const
{
    return (std::fabs(_character->getCurrentVelocity().y) < MATH_EPSILON);
}

void CharacterGame::update(float elapsedTime)
{
    if (_applyKick)
    {
        // apply impulse from kick.
        Vector3 impulse(-_characterNode->getForwardVectorWorld());
        impulse.normalize();
        // add some lift to kick
        impulse.y = 1.0f; 
        //scale the impulse.
        impulse.scale(16.6f);
        ((PhysicsRigidBody*)_basketballNode->getCollisionObject())->applyImpulse(impulse);
        _hasBall = false;
        _applyKick = false;
    }
    if (!_kickClip->isPlaying())
        _kicking = false;

    if (_gamepad->isButtonDown(Gamepad::BUTTON_A))
    {
        if (_buttonPressed[BUTTON_1])
        {
            _buttonPressed[BUTTON_1] = false;
            // Jump while the gamepad button is being pressed
            jump();
        }
    }
    else
    {
        _buttonPressed[BUTTON_1] = true;
    }

    if (_gamepad->isButtonDown(Gamepad::BUTTON_B))
    {
        if (_buttonPressed[BUTTON_2])
        {
            _buttonPressed[BUTTON_2] = false;
            kick();
        }
    }
    else
    {
        _buttonPressed[BUTTON_2] = true;
    }

    _currentDirection.set(Vector2::zero());

    if (!_kicking)
    {
        if (_gamepad->getJoystickCount() > 0)
        {
            _gamepad->getJoystickValues(0, &_currentDirection);
        }
    }
    if (_gamepad->getJoystickCount() > 1)
    {
        Vector2 out;
        _gamepad->getJoystickValues(1, &out);
       _character->getNode()->rotateY(-MATH_DEG_TO_RAD(out.x * 2.0f));
    }
    
    if (_currentDirection.isZero())
    {
        // Construct direction vector from keyboard input
        if (_keyFlags & NORTH)
            _currentDirection.y = 1;
        else if (_keyFlags & SOUTH)
            _currentDirection.y = -1;
        else
            _currentDirection.y = 0;

        if (_keyFlags & EAST)
            _currentDirection.x = 1;
        else if (_keyFlags & WEST)
            _currentDirection.x = -1;
        else 
            _currentDirection.x = 0;

        _currentDirection.normalize();
        if ((_keyFlags & RUNNING) == 0)
            _currentDirection *= 0.5f;
    }

    // Update character animation and velocity
    if (_currentDirection.isZero())
    {
        play("idle", true);
        _character->setVelocity(Vector3::zero());
    }
    else
    {
        bool running = (_currentDirection.lengthSquared() > 0.75f);
        float speed = running ? RUN_SPEED : WALK_SPEED;

        play(running ? "running" : "walking", true, 1.0f);

        // Orient the character relative to the camera so he faces the direction we want to move.
        const Matrix& cameraMatrix = _scene->getActiveCamera()->getNode()->getWorldMatrix();
        Vector3 cameraRight, cameraForward;
        cameraMatrix.getRightVector(&cameraRight);
        cameraMatrix.getForwardVector(&cameraForward);

        // Get the current forward vector for the mesh node (negate it since the character was modelled facing +z)
        Vector3 currentHeading(-_characterNode->getForwardVectorWorld());

        // Construct a new forward vector for the mesh node
        Vector3 newHeading(cameraForward * _currentDirection.y + cameraRight * _currentDirection.x);

        // Compute the rotation amount based on the difference between the current and new vectors
        float angle = atan2f(newHeading.x, newHeading.z) - atan2f(currentHeading.x, currentHeading.z);
        if (angle > MATH_PI)
            angle -= MATH_PIX2;
        else if (angle < -MATH_PI)
            angle += MATH_PIX2;
        angle *= (float)elapsedTime * 0.001f * MATH_PIX2;
        _characterNode->rotate(Vector3::unitY(), angle);

        // Update the character's velocity
        Vector3 velocity = -_characterNode->getForwardVectorWorld();
        velocity.normalize();
        velocity *= speed;
        _character->setVelocity(velocity);
    }

    // Adjust camera to avoid it from being obstructed by walls and objects in the scene.
    adjustCamera(elapsedTime);

    // Project the character's shadow node onto the surface directly below him.
    PhysicsController::HitResult hitResult;
    Vector3 v = _character->getNode()->getTranslationWorld();
    if (getPhysicsController()->rayTest(Ray(Vector3(v.x, v.y + 1.0f, v.z), Vector3(0, -1, 0)), 100.0f, &hitResult, NULL))
    {
        _characterShadowNode->setTranslation(Vector3(hitResult.point.x, hitResult.point.y + 0.1f, hitResult.point.z));
    }

    if (_hasBall)
    {
        // This is the first time entering this block of code if the basketball is still enabled.
        // Disable physics on basketball, and create a bigger collision object around the boy to include the ball.
        // This will ensure the boy cannot walk through walls/objects with the basketball.
        PhysicsRigidBody* basketball = (PhysicsRigidBody*)_basketballNode->getCollisionObject();
        if (basketball->isEnabled())
        {
            grabBall();
        }

        // Capture the basketball's old position, and then calculate the basketball's new position in front of the character
        _oldBallPosition = _basketballNode->getTranslationWorld();
        Vector3 characterForwardVector(_characterNode->getForwardVectorWorld());
        Vector3 translation(_characterNode->getTranslationWorld() + characterForwardVector.normalize() * -2.2f);
        translation.y = _floorLevel;

        // Calculates rotation to be applied to the basketball.
        Vector3 rotationVector(0.0f, -_basketballNode->getBoundingSphere().radius, 0.0f);
        Vector3::cross(rotationVector, _oldBallPosition - translation, &rotationVector);
        if (!rotationVector.isZero())
        {
            Matrix m;
            _basketballNode->getWorldMatrix().transpose(&m);

            Vector3 rotNorm;
            m.transformVector(rotationVector, &rotNorm);
            rotNorm.normalize();
            _basketballNode->rotate(rotNorm, rotationVector.length());
        }
        _basketballNode->setTranslation(translation.x, _floorLevel, translation.z);
    }
}

void CharacterGame::render(float elapsedTime)
{
    // Clear the color and depth buffers.
    clear(CLEAR_COLOR_DEPTH, Vector4(0.41f, 0.48f, 0.54f, 1.0f), 1.0f, 0);

    // Draw our scene, with separate passes for opaque and transparent objects.
    _scene->visit(this, &CharacterGame::drawScene, false);
    _scene->visit(this, &CharacterGame::drawScene, true);

    // Draw physics debug
    if (_physicsDebug)
        getPhysicsController()->drawDebug(_scene->getActiveCamera()->getViewProjectionMatrix());

    _gamepad->draw();

    // Draw FPS
    _font->start();
    char fps[32];
    sprintf(fps, "%d", getFrameRate());
    _font->drawText(fps, 5, 5, Vector4(1,1,0,1), 20);
    _font->finish();
}

void CharacterGame::keyEvent(Keyboard::KeyEvent evt, int key)
{
    if (evt == Keyboard::KEY_PRESS)
    {
        switch (key)
        {
        case Keyboard::KEY_ESCAPE:
            exit();
            break;
        case Keyboard::KEY_W:
        case Keyboard::KEY_CAPITAL_W:
            _keyFlags |= NORTH;
            _keyFlags &= ~SOUTH;
            break;
        case Keyboard::KEY_S:
        case Keyboard::KEY_CAPITAL_S:
            _keyFlags |= SOUTH;
            _keyFlags &= ~NORTH;
            break;
        case Keyboard::KEY_A:
        case Keyboard::KEY_CAPITAL_A:
            _keyFlags |= WEST;
            _keyFlags &= ~EAST;
            break;
        case Keyboard::KEY_D:
        case Keyboard::KEY_CAPITAL_D:
            _keyFlags |= EAST;
            _keyFlags &= ~WEST;
            break;
        case Keyboard::KEY_B:
            _physicsDebug = !_physicsDebug;
            break;
        case Keyboard::KEY_SPACE:
            jump();
            break;
        case Keyboard::KEY_SHIFT:
            _keyFlags |= RUNNING;
            break;
        case Keyboard::KEY_M:
        case Keyboard::KEY_CAPITAL_M:
            _wireframe = !_wireframe;
            break;
        case Keyboard::KEY_C:
        case Keyboard::KEY_CAPITAL_C:
            clone();
            break;
        }
    }
    else if (evt == Keyboard::KEY_RELEASE)
    {
        switch (key)
        {
        case Keyboard::KEY_W:
        case Keyboard::KEY_CAPITAL_W:
            _keyFlags &= ~NORTH;
            break;
        case Keyboard::KEY_S:
        case Keyboard::KEY_CAPITAL_S:
            _keyFlags &= ~SOUTH;
            break;
        case Keyboard::KEY_A:
        case Keyboard::KEY_CAPITAL_A:
            _keyFlags &= ~WEST;
            break;
        case Keyboard::KEY_D:
        case Keyboard::KEY_CAPITAL_D:
            _keyFlags &= ~EAST;
            break;
        case Keyboard::KEY_SHIFT:
            _keyFlags &= ~RUNNING;
            break;
        }
    }
}

void CharacterGame::touchEvent(Touch::TouchEvent evt, int x, int y, unsigned int contactIndex)
{
    // This should only be called if the gamepad did not handle the touch event.
    switch (evt)
    {
    case Touch::TOUCH_PRESS:
        {
            _rotateX = x;
        }
        break;
    case Touch::TOUCH_RELEASE:
        {
            _rotateX = 0;
        }
        break;
    case Touch::TOUCH_MOVE:
        {
            int deltaX = x - _rotateX;
            _rotateX = x;
            _character->getNode()->rotateY(-MATH_DEG_TO_RAD(deltaX * 0.5f));
        }
        break;
    default:
        break;
    }
}

bool CharacterGame::mouseEvent(Mouse::MouseEvent evt, int x, int y, int wheelDelta)
{
    if (evt == Mouse::MOUSE_PRESS_RIGHT_BUTTON)
    {
        kick();
        return true;
    }
    return false;
}

void CharacterGame::gamepadEvent(Gamepad::GamepadEvent evt, Gamepad* gamepad)
{
    switch(evt)
    {
    case Gamepad::CONNECTED_EVENT:
    case Gamepad::DISCONNECTED_EVENT:
        _gamepad = getGamepad(0);
        break;
    }
}

void CharacterGame::adjustCamera(float elapsedTime)
{
    static float cameraOffset = 0.0f;

    PhysicsController* physics = getPhysicsController();
    Node* cameraNode = _scene->getActiveCamera()->getNode();

    // Reset camera
    if (cameraOffset != 0.0f)
    {
        cameraNode->translateForward(-cameraOffset);
        cameraOffset = 0.0f;
    }

    Vector3 cameraPosition = cameraNode->getTranslationWorld();
    Vector3 cameraDirection = cameraNode->getForwardVectorWorld();
    cameraDirection.normalize();

    // Get focal point of camera (use the resolved world location of the head joint as a focal point)
    Vector3 focalPoint(cameraPosition + (cameraDirection * CAMERA_FOCUS_DISTANCE));

    Vector3 oldPosition = cameraNode->getTranslationWorld();

    PhysicsController::HitResult result;
    PhysicsCollisionObject* occlusion = NULL;
    do
    {
        // Perform a ray test to check for camera collisions
        if (!physics->sweepTest(cameraNode->getCollisionObject(), focalPoint, &result) || result.object == _character)
            break;

        occlusion = result.object;

        // Step the camera closer to the focal point to resolve the occlusion
        float d = cameraNode->getTranslationWorld().distance(result.point);
        cameraNode->translateForward(d);
        cameraOffset += d;
        while (physics->sweepTest(cameraNode->getCollisionObject(), focalPoint, &result) && result.object == occlusion)
        {
            // Prevent the camera from getting too close to the character.
            // Without this check, it's possible for the camera to fly past the character
            // and essentially end up in an infinite loop here.
            if (cameraNode->getTranslationWorld().distanceSquared(focalPoint) <= 2.0f)
                return;

            cameraNode->translateForward(0.1f);
            cameraOffset += 0.1f;
        }
    } while (true);

    // If the character is closer than 10 world units to the camera, apply transparency to the character so he does not obstruct the view.
    if (occlusion)
    {
        float d = _scene->getActiveCamera()->getNode()->getTranslationWorld().distance(_characterNode->getTranslationWorld());
        float alpha = d < 10 ? (d * 0.1f) : 1.0f;
        _characterMeshNode->setTag("transparent", alpha < 1.0f ? "true" : NULL);
        _materialParameterAlpha->setValue(alpha);
    }
    else
    {
        _characterMeshNode->setTag("transparent", NULL);
        _materialParameterAlpha->setValue(1.0f);
    }
}

void CharacterGame::animationEvent(AnimationClip* clip, AnimationClip::Listener::EventType type)
{
    if (clip == _kickClip && !_applyKick)
    {
        _keyFlags = 0;
        if (_hasBall)
        {
            _applyKick = true;
            releaseBall();
        }
    }
    else
    {
        clip->crossFade(_currentClip, 100);
    }
}

void CharacterGame::clone()
{
    Node* clone = _scene->findNode("boycharacter")->clone();
    Animation* cloneAnimation = clone->getAnimation();

    // Find the current clip and have the clone play that clip repeatedly.
    const char* clipId = _currentClip->getId();
    if (_jumpClip->isPlaying())
        clipId = _jumpClip->getId();
    AnimationClip* clip = cloneAnimation->getClip(clipId);
    clip->setRepeatCount(AnimationClip::REPEAT_INDEFINITE);
    clip->play();

    _scene->addNode(clone);
    clone->release();
}

void CharacterGame::collisionEvent(PhysicsCollisionObject::CollisionListener::EventType type,
                                    const PhysicsCollisionObject::CollisionPair& collisionPair,
                                    const Vector3& contactPointA,
                                    const Vector3& contactPointB)
{
    // objectA -> basketball, only care about collisions between the physics character and the basketball.
    if (type == PhysicsCollisionObject::CollisionListener::COLLIDING && collisionPair.objectB == _character)
        _hasBall = true;
}

void CharacterGame::grabBall()
{
    // Disables physics on the basketball, and increases the size of the character's collison object to include the basketball.
    _basketballNode->getCollisionObject()->setEnabled(false);
    PhysicsRigidBody::Parameters rbParams;
    rbParams.mass = 20.0f;
    Vector3 currentVelocity = _character->getCurrentVelocity();
    Node* boy = _character->getNode();
    boy->setCollisionObject(PhysicsCollisionObject::CHARACTER, PhysicsCollisionShape::capsule(2.9f, 6.0f, Vector3(0.0f, 3.0f, 0.0f), true), &rbParams);
    _character = static_cast<PhysicsCharacter*>(boy->getCollisionObject());
    _character->setMaxSlopeAngle(0.0f);
    _character->setMaxStepHeight(0.0f);
    _character->setVelocity(currentVelocity);
}

void CharacterGame::releaseBall()
{
    // Decreases the size of the character's collision object and re-enables physics simulation on the basketball.
    PhysicsRigidBody::Parameters rbParams;
    rbParams.mass = 20.0f;
    Vector3 velocity = _character->getCurrentVelocity();
    Node* boy = _character->getNode();
    boy->setCollisionObject(PhysicsCollisionObject::CHARACTER, PhysicsCollisionShape::capsule(1.2f, 6.0f, Vector3(0.0f, 3.0f, 0.0f), true), &rbParams);
    _character = static_cast<PhysicsCharacter*>(boy->getCollisionObject());
    _character->setVelocity(velocity);
    _character->setMaxSlopeAngle(0.0f);
    _character->setMaxStepHeight(0.0f);
    
    PhysicsRigidBody* basketball = (PhysicsRigidBody*) _basketballNode->getCollisionObject();
    basketball->setEnabled(true);
}