WeatherEffect.h

#pragma once
#include "BaseObject.h"
#include "util/FastRandom.h"
#include "util/Tick.h"
#include "util/unordered_array.hpp"
#include "WeatherParticle.h"

namespace ParaEngine
{
    class WeatherParticleSpawner;

    class CWeatherEffect : public CBaseObject
    {
    public:
        CWeatherEffect();
        virtual ~CWeatherEffect();
        
        ATTRIBUTE_DEFINE_CLASS(CWeatherEffect);
        ATTRIBUTE_SUPPORT_CREATE_FACTORY(CWeatherEffect);

        virtual int InstallFields(CAttributeClass* pClass, bool bOverride);

        ATTRIBUTE_METHOD1(CWeatherEffect, EnableRain_s, bool)   { cls->EnableRain(p1); return S_OK; }
        ATTRIBUTE_METHOD1(CWeatherEffect, EnableSnow_s, bool)   { cls->EnableSnow(p1); return S_OK; }
        
        ATTRIBUTE_METHOD1(CWeatherEffect, SetRainStrength_s, float) { cls->SetRainStrength(p1); return S_OK; }
        ATTRIBUTE_METHOD1(CWeatherEffect, SetSnowStrength_s, float) { cls->SetSnowStrength(p1); return S_OK; }

    public:
        WeatherParticle* AddParticle(float x, float y, float z, WeatherType weather_type);

        void SpawnAllParticles();

        /*spawn particles for the given weather item
         @param weather_item : can be snow or rain item info.
         */
        void SpawnParticles(WeatherParticleSpawner& weather_type, float strength, float spawn_tick_fps);

        virtual void Cleanup();

        FastRandom& GetSpawnRand() { return m_spawnRand; }
        FastRandom& GetSpeedRand() { return m_speedRand; }

        bool HasActiveSpawners();

        WeatherParticleSpawner& GetParticleSpawner(WeatherType weather_type);

        void EnableWeatherItem(WeatherType weather_type, bool bEnabled);
        bool IsWeatherItemEnabled(WeatherType weather_type);
        
        int32 GetTerrainHighestBlock(int32 x, int32 z, int32& outY);

        int PrepareRender(CBaseCamera* pCamera, SceneState* pSceneState);

        virtual void Animate(double dTimeDelta, int nRenderNumber = 0);

        virtual HRESULT Draw(SceneState * sceneState);

        void EnableSnow(bool bEnabled);
        void EnableRain(bool bEnabled);
        void SetRainStrength(float fStrength);
        void SetSnowStrength(float fStrength);

    protected:
        // keep a weak reference for fast access
        WeatherParticleSpawner* m_particle_spawners[WeatherType_Last];
    
        float m_default_strength;
        FastRandom m_spawnRand;
        FastRandom m_speedRand;
        
        unordered_array<WeatherParticle*> m_particles;
        Tick m_updateTick;
    };
}