/*
  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 "cloud_effect.h"

#include "parser/tinyxml/tinyxml.h"

// Define shell commands
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);

/**
 * @brief standard constructor
 */
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 * 2);
  }
}

/**
 * @brief standard deconstructor
 */
RainEffect::~RainEffect()
{
  this->deactivate();

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

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

/**
 * @brief initalizes the rain effect with default values
 */
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->rainWindForce  = 0;
  this->rainFadeInDuration = 0;
  this->rainFadeOutDuration = 0;
  
  this->cloudColor = Vector(0.8f, 0.8f, 0.8f);
  this->skyColor = Vector(0.0f, 0.0f, 0.0f);

  this->rainMaxParticles = this->rainRate * this->rainLife;
  this->localTimer = 0;
  this->soundRainVolume = 0.3f;
  this->emitter = new PlaneEmitter(this->rainSize);

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

/**
 * @brief loads the rain effect parameters.
 * @param root: the XML-Element to load the data from
 */
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);
  LoadParam(root, "cloudcolor", this, RainEffect, setCloudColor);
  LoadParam(root, "skycolor", this, RainEffect, setSkyColor);

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

SparkParticles* RainEffect::rainParticles = NULL;

/**
 * @brief activates the rain effect
 */
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);

  // plays the rain sound and loops it 
  this->soundSource.play(this->rainBuffer, this->soundRainVolume, true);

  // if there is wind, play the wind sound
  if (this->rainWindForce != 0)
    this->soundSource.play(this->windBuffer, 0.1f * this->rainWindForce, true);

  if (this->rainFadeInDuration == 0)
    lightMan->setAmbientColor(.1,.1,.1);
  
  // Change the cloudcolor,skycolor
  this->oldCloudColor = CloudEffect::cloudColor;
  this->oldSkyColor = CloudEffect::skyColor;
  CloudEffect::changeCloudColor(this->cloudColor);
  CloudEffect::changeSkyColor(this->skyColor);
}

/**
 * @brief deactivates the rain effect
 */
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);
  
  CloudEffect::changeCloudColor(this->oldCloudColor);
  CloudEffect::changeSkyColor(this->oldSkyColor);
}

/**
 * @brief ticks the rain effect
 * @param dt: tick float
 */
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
    if (!this->soundSource.isPlaying())
      this->soundSource.play(this->rainBuffer, this->soundRainVolume, true);

    this->soundSource.gain(this->rainBuffer, this->soundRainVolume * progress);
  }
  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
      if (!this->soundSource.isPlaying())
        this->soundSource.play(this->rainBuffer, this->soundRainVolume, true);

      this->soundSource.gain(this->rainBuffer, this->soundRainVolume * progress);
    }
    else
      this->deactivate();
  }
}

/**
 * @brief starts raining slowly
*/
void RainEffect::startRaining()
{

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

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

  this->localTimer = 0;

  this->activate();
}

/**
 * @brief stops raining slowly
 */
void RainEffect::stopRaining()
{

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

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

  this->localTimer = 0;
}

/**
 * @brief hides the rain
 */
void RainEffect::hideRain()
{
  RainEffect::rainParticles->setColor(0, 0,0,0, 0);
  RainEffect::rainParticles->setColor(0, 0,0,0, 0);
}