source: orxonox.OLD/branches/atmospheric_engine/src/lib/graphics/effects/rain_effect.cc @ 8434

/*
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:
        - preCaching
        - 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::~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);

}


bool 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();

        return 0;
}


SparkParticles* RainEffect::rainParticles = NULL;

bool 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->precache((int)this->rainLife);
                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.loop(this->rainBuffer, this->soundRainVolume);
        if (this->rainWindForce != 0) this->soundSource.loop(this->windBuffer, 0.1f * this->rainWindForce);

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

        return 0;
}


bool 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);

        return 0;
}

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;

}