/*
orxonox - the future of 3D-vertical-scrollers

Copyright (C) 2004 orx

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2, or (at your option)
any later version.

### File Specific:
main-programmer: hdavid, amaechler
*/

#include "rain_effect.h"

#include "util/loading/load_param.h"
#include "util/loading/factory.h"
#include "util/loading/resource_manager.h"

#include "glincl.h"
#include "p_node.h"
#include "state.h"
#include "spark_particles.h"
#include "plane_emitter.h"
#include "shell_command.h"
#include "light.h"

#include "parser/tinyxml/tinyxml.h"

SHELL_COMMAND(activate, RainEffect, activateRain);
SHELL_COMMAND(deactivate, RainEffect, deactivateRain);
SHELL_COMMAND(startraining, RainEffect, startRaining);
SHELL_COMMAND(stopraining, RainEffect, stopRaining);

using namespace std;

CREATE_FACTORY(RainEffect, CL_RAIN_EFFECT);

/* TODO:
  - test multiple rain emitters
  - Think about what happens with building poss. to hang movewithcam off
  - Possible to activate lightening
  - turn off visibility when in a building
  - variable emitter size depending on playable
*/

RainEffect::RainEffect(const TiXmlElement* root) {
    this->setClassID(CL_RAIN_EFFECT, "RainEffect");

    this->init();

    if (root != NULL)
        this->loadParams(root);

    //load rain sound
    if (this->rainBuffer != NULL)
        ResourceManager::getInstance()->unload(this->rainBuffer);
    this->rainBuffer = (OrxSound::SoundBuffer*)ResourceManager::getInstance()->load("sound/atmosphere/rain.wav", WAV);

    //load wind sound
    if (this->rainWindForce != 0) {
        if (this->windBuffer != NULL)
            ResourceManager::getInstance()->unload(this->windBuffer);
        this->windBuffer = (OrxSound::SoundBuffer*)ResourceManager::getInstance()->load("sound/atmosphere/wind.wav", WAV);
    }

    if(rainActivate) {
        this->activate();
        RainEffect::rainParticles->precache((int)this->rainLife * 5); // TODO: Figure out the correct value
    }
}

RainEffect::~RainEffect() {
    this->deactivate();

    if (this->rainBuffer != NULL)
        ResourceManager::getInstance()->unload(this->rainBuffer);

    if (this->windBuffer != NULL)
        ResourceManager::getInstance()->unload(this->windBuffer);
}

void RainEffect::loadParams(const TiXmlElement* root) {
    WeatherEffect::loadParams(root);

    LoadParam(root, "coord", this, RainEffect, setRainCoord);
    LoadParam(root, "size", this, RainEffect, setRainSize);
    LoadParam(root, "rate", this, RainEffect, setRainRate);
    LoadParam(root, "velocity", this, RainEffect, setRainVelocity);
    LoadParam(root, "life", this, RainEffect, setRainLife);
    LoadParam(root, "wind", this, RainEffect, setRainWind);
    LoadParam(root, "fadeinduration", this, RainEffect, setRainFadeIn);
    LoadParam(root, "fadeoutduration", this, RainEffect, setRainFadeOut);

    LOAD_PARAM_START_CYCLE(root, element);
    {
        LoadParam_CYCLE(element, "option", this, RainEffect, setRainOption);
    }
    LOAD_PARAM_END_CYCLE(element);

}


void RainEffect::init() {
    //Default values
    this->rainActivate = false;
    this->rainMove = false;
    this->rainCoord = Vector(500, 500, 500);
    this->rainSize = Vector2D(1000, 1000);
    this->rainRate = 4000;
    this->rainVelocity = -300;
    this->rainLife = 4;
    this->rainMaxParticles = this->rainRate * this->rainLife;
    this->rainWindForce  = 0;
    this->rainFadeInDuration = 0;
    this->rainFadeOutDuration = 0;
    this->localTimer = 0;
    this->soundRainVolume = 0.8f;

    this->emitter = new PlaneEmitter(this->rainSize);

    lightMan = LightManager::getInstance();
    this->rainAmbient = lightMan->getAmbientColor();
}


SparkParticles* RainEffect::rainParticles = NULL;

void RainEffect::activate() {
    PRINTF(0)( "Activating RainEffect, coord: %f, %f, %f, size: %f, %f, rate: %f, velocity: %f, moveRain: %s\n", this->rainCoord.x, this->rainCoord.y, this->rainCoord.z, this->rainSize.x, this-> rainSize.y, this->rainRate, this->rainVelocity, this->rainMove ? "true" : "false" );

    this->rainActivate = true;

    if (unlikely(RainEffect::rainParticles == NULL)) {
        RainEffect::rainParticles = new SparkParticles((int) this->rainMaxParticles);
        RainEffect::rainParticles->setName("RainParticles");
        RainEffect::rainParticles->setLifeSpan(this->rainLife, 2);
        RainEffect::rainParticles->setRadius(0, 0.03);
        RainEffect::rainParticles->setRadius(0.2, 0.02);
        RainEffect::rainParticles->setRadius(1, 0.01);
        RainEffect::rainParticles->setColor(0, 0.3, 0.3, 0.5, 0.2); // grey blue 1
        RainEffect::rainParticles->setColor(0.5, 0.4, 0.4, 0.5, 0.2); // grey blue 2
        RainEffect::rainParticles->setColor(1, 0.7, 0.7, 0.7, 0.2); // light grey
    }

    this->emitter->setSystem(RainEffect::rainParticles);

    this->emitter->setRelCoor(this->rainCoord);

    this->emitter->setEmissionRate(this->rainRate);
    this->emitter->setEmissionVelocity(this->rainVelocity);

    this->emitter->setSpread(this->rainWindForce / 50, 0.2);

    this->soundSource.play(this->rainBuffer, this->soundRainVolume, true);

    if (this->rainWindForce != 0)
        this->soundSource.play(this->windBuffer, 0.1f * this->rainWindForce, true);

    if (this->rainFadeInDuration == 0)
        lightMan->setAmbientColor(.1,.1,.1);

}


void RainEffect::deactivate() {
    PRINTF(0)("Deactivating RainEffect\n");

    this->rainActivate = false;
    this->emitter->setSystem(NULL);

    // Stop Sound
    this->soundSource.stop();

    // Restore Light Ambient
    lightMan->setAmbientColor(this->rainAmbient, this->rainAmbient, this->rainAmbient);

}

void RainEffect::tick (float dt) {
    if (!this->rainActivate)
        return;

    if (this->rainMove) {
        this->rainCoord = State::getCameraNode()->getAbsCoor();
        this->emitter->setRelCoor(this->rainCoord.x , this->rainCoord.y+800, this->rainCoord.z);
    }

    if (this->rainFadeInDuration != 0 && this->localTimer < this->rainFadeInDuration) {
        this->localTimer += dt;
        float progress = this->localTimer / this->rainFadeInDuration;

        // Dim Light
        lightMan->setAmbientColor(1 - progress * 0.9, 1 - progress * 0.9, 1 - progress * 0.9);

        // use alpha in color to fade in
        RainEffect::rainParticles->setColor(0,   0.3, 0.3, 0.5, 0.2 * progress); // grey blue 1
        RainEffect::rainParticles->setColor(0.5, 0.4, 0.4, 0.5, 0.2 * progress); // grey blue 2
        RainEffect::rainParticles->setColor(1,   0.7, 0.7, 0.7, 0.2 * progress); // light grey

        // increase radius for more "heavy" rain
        RainEffect::rainParticles->setRadius(0, 0.03 * progress);
        RainEffect::rainParticles->setRadius(0.2, 0.02 * progress);
        RainEffect::rainParticles->setRadius(1, 0.01 * progress);

        // increase sound volume
        // this->soundSource.fadein(this->rainBuffer, 10);
    } else if ( this->rainFadeOutDuration != 0 ) {
        if ( this->localTimer < this->rainFadeOutDuration ) {
            this->localTimer += dt;
            float progress = 1 - (this->localTimer / this->rainFadeOutDuration);

            // Fade In Light
            lightMan->setAmbientColor(1 - progress * 0.9, 1 - progress * 0.9, 1 - progress * 0.9);

            // use alpha in color to fade out
            RainEffect::rainParticles->setColor(0,   0.3, 0.3, 0.5, 0.2 * progress); // grey blue 1
            RainEffect::rainParticles->setColor(0.5, 0.4, 0.4, 0.5, 0.2 * progress); // grey blue 2
            RainEffect::rainParticles->setColor(1,   0.7, 0.7, 0.7, 0.2 * progress); // light grey

            // decrease radius
            RainEffect::rainParticles->setRadius(0, 0.03 * progress);
            RainEffect::rainParticles->setRadius(0.2, 0.02 * progress);
            RainEffect::rainParticles->setRadius(1, 0.01 * progress);

            // decrease sound volume
            // this->soundSource.fadeout(this->rainBuffer, 10);
        } else
            this->deactivate();
    }

}

void RainEffect::startRaining() {

    if (this->rainActivate)
        this->deactivate();

    if (!this->rainFadeInDuration > 0)
        this->rainFadeInDuration = 20;

    this->localTimer = 0;

    this->activate();

}

void RainEffect::stopRaining() {

    if (!this->rainActivate)
        this->activate();

    if (!this->rainFadeOutDuration > 0)
        this->rainFadeOutDuration = 20;

    this->localTimer = 0;

}