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

/*
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: Dim Light with Rain, Finish preCaching, check out if multiple rain emitters work,  Think about what happens with building poss. to hang movewithcam off, benchmark, possible to activate lightening, turn off visibility when in a building, variable emitter size depending on playable, also rain velocity

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

        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.1 - progress, 1.1 - progress, 1.1 - progress);

                // 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 && this->localTimer < this->rainFadeOutDuration) {
                this->localTimer += dt;
                float progress = 1 - (this->localTimer / this->rainFadeOutDuration);

                // Dim Light
                lightMan->setAmbientColor(1.1 - progress, 1.1 - progress, 1.1 - progress);

                // 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);
        }
        
}

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

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

        this->localTimer = 0;
        this->activate();

}