/* osgEarth
* Copyright 2008-2012 Pelican Mapping
* MIT License
*/
#pragma once

#include <osgEarthProcedural/Export>
#include <osgEarthProcedural/BiomeLayer>
#include <osgEarth/ImageLayer>
#include <osgEarth/ElevationPool>
#include <osgEarth/LayerReference>
#include <osgEarth/TerrainOptions>

namespace osgEarth { namespace Procedural
{
    using namespace osgEarth;

    enum LifeMap {
        LIFEMAP_RUGGED = 0,
        LIFEMAP_DENSE = 1,
        LIFEMAP_LUSH = 2,
        LIFEMAP_SPECIAL = 3
    };

    class OSGEARTHPROCEDURAL_EXPORT LifeMapLayer : public ImageLayer
    {
    public:
        class OSGEARTHPROCEDURAL_EXPORT Options : public ImageLayer::Options
        {
        public:
            META_LayerOptions(osgEarthProcedural, Options, ImageLayer::Options);
            OE_OPTION_LAYER(BiomeLayer, biomeLayer);
            OE_OPTION_LAYER(ImageLayer, maskLayer);
            OE_OPTION_LAYER(ImageLayer, waterLayer);
            OE_OPTION_LAYER(CoverageLayer, landCoverLayer);
            OE_OPTION_LAYER(ImageLayer, colorLayer);

            OE_OPTION(float, landCoverWeight, 1.0f);
            OE_OPTION(Distance, landCoverBlur, Distance(0.0f, Units::METERS));
            OE_OPTION(float, terrainWeight, 1.0f);
            OE_OPTION(float, slopeIntensity, 1.0f);
            OE_OPTION(float, slopeCutoff, 0.0f);
            OE_OPTION(float, colorWeight, 1.0f);
            OE_OPTION(float, noiseWeight, 0.225f);
            OE_OPTION(float, lushFactor, 1.9f);

            virtual Config getConfig() const;
        private:
            void fromConfig(const Config& conf);
        };

    public:
        META_Layer(osgEarthProcedural, LifeMapLayer, Options, ImageLayer, LifeMap);

        void setBiomeLayer(BiomeLayer*);
        BiomeLayer* getBiomeLayer() const;

        void setMaskLayer(ImageLayer*);
        ImageLayer* getMaskLayer() const;

        void setWaterLayer(ImageLayer*);
        ImageLayer* getWaterLayer() const;

        void setLandCoverLayer(CoverageLayer*);
        CoverageLayer* getLandCoverLayer() const;

        void setColorLayer(ImageLayer*);
        ImageLayer* getColorLayer() const;

        void setLandCoverWeight(float value);
        float getLandCoverWeight() const;
        bool getUseLandCover() const;

        void setTerrainWeight(float value);
        float getTerrainWeight() const;
        bool getUseTerrain() const;

        void setColorWeight(float value);
        float getColorWeight() const;
        bool getUseColor() const;

        void setNoiseWeight(float value);
        float getNoiseWeight() const;
        bool getUseNoise() const;

        //! Access the elevation data working set so we can customize the elevation layer
        //! the LifeMapLayer uses when creating rasters.
        //! Only do this before starting up the terrain engine.
        ElevationPool::WorkingSet& getWorkingSet() { return _workingSet; }

    protected:

        void init() override;

        Status openImplementation() override;

        Status closeImplementation() override;

        GeoImage createImageImplementation(const TileKey&, ProgressCallback*) const override;

        GeoImage applyPostLayer(const GeoImage&, const TileKey&, Layer*, ProgressCallback*) const override;

        void prepareForRendering(TerrainEngine* engine) override;

    public:

        void addedToMap(const Map* map) override;

        void removedFromMap(const Map* map) override;

    protected:

        osg::observer_ptr<const Map> _map;
        mutable ElevationPool::WorkingSet _workingSet;
        osg::ref_ptr<osg::Image> _noiseFunc;
        LandCoverSample::Factory::Ptr _landCoverFactory;
        TerrainOptionsAPI _terrainOptions;

        void checkForLayerError(Layer*);
    };

} } // namespace osgEarth::Procedural
