/* -*-c++-*- */
/* osgEarth - Geospatial SDK for OpenSceneGraph
* Copyright 2008-2012 Pelican Mapping
* http://osgearth.org
*
* osgEarth is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
* IN THE SOFTWARE.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program.  If not, see <http://www.gnu.org/licenses/>
*/
#pragma once

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

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;

    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;

        void checkForLayerError(Layer*);
    };

} } // namespace osgEarth::Procedural
