/*****************************************************************************************
*                                                                                       *
* OpenSpace                                                                             *
*                                                                                       *
* Copyright (c) 2014-2016                                                               *
*                                                                                       *
* Permission is hereby granted, free of charge, to any person obtaining a copy of this  *
* software and associated documentation files (the "Software"), to deal in the Software *
* without restriction, including without limitation the rights to use, copy, modify,    *
* merge, publish, distribute, sublicense, and/or sell copies of the Software, and to    *
* permit persons to whom the Software is furnished to do so, subject to the following   *
* conditions:                                                                           *
*                                                                                       *
* The above copyright notice and this permission notice shall be included in all copies *
* or substantial portions of the Software.                                              *
*                                                                                       *
* 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.                                         *
****************************************************************************************/

#include <modules/globebrowsing/globes/renderableglobe.h>
#include <modules/globebrowsing/globes/globemesh.h>
#include <modules/globebrowsing/other/threadpool.h>
#include <modules/globebrowsing/other/temporaltileprovider.h>

// open space includes
#include <openspace/engine/openspaceengine.h>
#include <openspace/rendering/renderengine.h>
#include <openspace/util/spicemanager.h>
#include <openspace/scene/scenegraphnode.h>

// ghoul includes
#include <ghoul/misc/assert.h>

namespace {
    const std::string _loggerCat = "RenderableGlobe";

    // Keys for the dictionary
    const std::string keyFrame = "Frame";
    const std::string keyRadii = "Radii";
    const std::string keySegmentsPerPatch = "SegmentsPerPatch";
    const std::string keyTextures = "Textures";
    const std::string keyColorTextures = "ColorTextures";
    const std::string keyHeightMaps = "HeightMaps";
}



namespace openspace {


    RenderableGlobe::RenderableGlobe(const ghoul::Dictionary& dictionary)
        : _saveOrThrowCamera(properties::BoolProperty("saveOrThrowCamera", "saveOrThrowCamera"))
        , doFrustumCulling(properties::BoolProperty("doFrustumCulling", "doFrustumCulling"))
        , doHorizonCulling(properties::BoolProperty("doHorizonCulling", "doHorizonCulling"))
        , mergeInvisible(properties::BoolProperty("mergeInvisible", "mergeInvisible", true))
        , lodScaleFactor(properties::FloatProperty("lodScaleFactor", "lodScaleFactor", 5.0f, 1.0f, 20.0f))
        , initChunkVisible(properties::BoolProperty("initChunkVisible", "initChunkVisible", true))
        , renderSmallChunksFirst(properties::BoolProperty("renderSmallChunksFirst", "renderSmallChunksFirst", true))
        , chunkHeight(properties::FloatProperty("chunkHeight", "chunkHeight", 8700.0f, 0.0f, 8700.0f))
        , blendHeightMap(properties::BoolProperty("blendHeightMap", "blendHeightMap", true))
        , blendColorMap(properties::BoolProperty("blendColorMap", "blendColorMap", true))
        , blendNightTexture(properties::BoolProperty("blendNightTexture", "blendNightTexture", true))
        , blendOverlay(properties::BoolProperty("blendOverlay", "blendOverlay", true))
        , blendWaterMask(properties::BoolProperty("blendWaterMask", "blendWaterMask", true))
        , atmosphereEnabled(properties::BoolProperty("atmosphereEnabled", "atmosphereEnabled", false))
        , showChunkEdges(properties::BoolProperty("showChunkEdges", "showChunkEdges", false))
    {
        setName("RenderableGlobe");
        
        addProperty(_saveOrThrowCamera);
        addProperty(doFrustumCulling);
        addProperty(doHorizonCulling);
        addProperty(mergeInvisible);
        addProperty(lodScaleFactor);
        addProperty(initChunkVisible);
        addProperty(renderSmallChunksFirst);
        addProperty(chunkHeight);

        addProperty(blendHeightMap);
        addProperty(blendColorMap);
        addProperty(blendNightTexture);
        addProperty(blendOverlay);
        addProperty(blendWaterMask);
        addProperty(atmosphereEnabled);
        addProperty(showChunkEdges);

        doFrustumCulling.setValue(true);
        doHorizonCulling.setValue(true);
        renderSmallChunksFirst.setValue(true);

        dictionary.getValue(keyFrame, _frame);

        // Read the radii in to its own dictionary
        Vec3 radii;
        dictionary.getValue(keyRadii, radii);
        _ellipsoid = Ellipsoid(radii);
        setBoundingSphere(pss(_ellipsoid.averageRadius(), 0.0));

        // Ghoul can't read ints from lua dictionaries
        double patchSegmentsd;
        dictionary.getValue(keySegmentsPerPatch, patchSegmentsd);
        int patchSegments = patchSegmentsd;
        
        // Init tile provider manager
        ghoul::Dictionary texturesDictionary;
        dictionary.getValue(keyTextures, texturesDictionary);
        _tileProviderManager = std::shared_ptr<TileProviderManager>(
            new TileProviderManager(texturesDictionary));

        auto colorProviders = _tileProviderManager->colorTextureProviders();
        auto nightProviders = _tileProviderManager->nightTextureProviders();
        auto overlayProviders = _tileProviderManager->overlayProviders();
        auto heightProviders = _tileProviderManager->heightMapProviders();
        auto waterProviders = _tileProviderManager->waterMaskProviders();

        addToggleLayerProperties(colorProviders, _activeColorLayers);
        addToggleLayerProperties(nightProviders, _activeNightLayers);
        addToggleLayerProperties(overlayProviders, _activeOverlays);
        addToggleLayerProperties(heightProviders, _activeHeightMapLayers);
        addToggleLayerProperties(waterProviders, _activeWaterMaskLayers);

        _chunkedLodGlobe = std::shared_ptr<ChunkedLodGlobe>(
            new ChunkedLodGlobe(_ellipsoid, patchSegments, _tileProviderManager));

        _distanceSwitch.addSwitchValue(_chunkedLodGlobe, 1e12);
    }

    RenderableGlobe::~RenderableGlobe() {

    }

    void RenderableGlobe::addToggleLayerProperties(
        std::vector<TileProviderManager::TileProviderWithName>& tileProviders,
        std::vector<properties::BoolProperty>& dest)
    {
        for (size_t i = 0; i < tileProviders.size(); i++) {
            bool enabled = i == 0; // Only enable first layer
            std::string name = tileProviders[i].name;
            dest.push_back(properties::BoolProperty(name, name, enabled));
        }
        auto it = dest.begin();
        auto end = dest.end();
        while (it != end) {
            addProperty(*(it++));
        }
    }

    bool RenderableGlobe::initialize() {
        return _distanceSwitch.initialize();
    }

    bool RenderableGlobe::deinitialize() {
        return _distanceSwitch.deinitialize();
    }

    bool RenderableGlobe::isReady() const {
        return _distanceSwitch.isReady();
    }

    void RenderableGlobe::render(const RenderData& data) {
        if (_saveOrThrowCamera.value()) {
            _saveOrThrowCamera.setValue(false);

            if (_chunkedLodGlobe->getSavedCamera() == nullptr) { // save camera
                LDEBUG("Saving snapshot of camera!");
                _chunkedLodGlobe->setSaveCamera(new Camera(data.camera));
            }
            else { // throw camera
                LDEBUG("Throwing away saved camera!");
                _chunkedLodGlobe->setSaveCamera(nullptr);
            }
        }

        _distanceSwitch.render(data);
    }

    void RenderableGlobe::update(const UpdateData& data) {
        // set spice-orientation in accordance to timestamp
        //_chunkedLodGlobe->setStateMatrix(
        //    SpiceManager::ref().positionTransformMatrix(_frame, "GALACTIC", data.time));
        // We currently do not consider rotation anywhere in the rendering.
        // @TODO Consider rotation everywhere in the rendering (culling, splitting, camera, etc)
        _chunkedLodGlobe->setStateMatrix(glm::dmat3(1.0));
        _time = data.time;
        _distanceSwitch.update(data);

        _chunkedLodGlobe->doFrustumCulling = doFrustumCulling.value();
        _chunkedLodGlobe->doHorizonCulling = doHorizonCulling.value();
        _chunkedLodGlobe->mergeInvisible = mergeInvisible.value();
        _chunkedLodGlobe->lodScaleFactor = lodScaleFactor.value();
        _chunkedLodGlobe->initChunkVisible = initChunkVisible.value();
        _chunkedLodGlobe->chunkHeight = chunkHeight.value();

        _chunkedLodGlobe->blendHeightMap = blendHeightMap.value();
        _chunkedLodGlobe->blendColorMap = blendColorMap.value();
        _chunkedLodGlobe->blendNightTexture = blendNightTexture.value();
        _chunkedLodGlobe->blendOverlay = blendOverlay.value();
        _chunkedLodGlobe->blendWaterMask = blendWaterMask.value();
        _chunkedLodGlobe->atmosphereEnabled = atmosphereEnabled.value();
        _chunkedLodGlobe->showChunkEdges = showChunkEdges.value();

        std::vector<TileProviderManager::TileProviderWithName>& colorTextureProviders =
            _tileProviderManager->colorTextureProviders();
        std::vector<TileProviderManager::TileProviderWithName>& nightTextureProviders =
            _tileProviderManager->nightTextureProviders();
        std::vector<TileProviderManager::TileProviderWithName>& overlayProviders =
            _tileProviderManager->overlayProviders();
        std::vector<TileProviderManager::TileProviderWithName>& heightMapProviders =
            _tileProviderManager->heightMapProviders();
        std::vector<TileProviderManager::TileProviderWithName>& waterMaskProviders =
            _tileProviderManager->waterMaskProviders();
        
        for (size_t i = 0; i < colorTextureProviders.size(); i++) {
            colorTextureProviders[i].isActive = _activeColorLayers[i].value();
        }
        for (size_t i = 0; i < nightTextureProviders.size(); i++) {
            nightTextureProviders[i].isActive = _activeNightLayers[i].value();
        }
        for (size_t i = 0; i < overlayProviders.size(); i++) {
            overlayProviders[i].isActive = _activeOverlays[i].value();
        }
        for (size_t i = 0; i < heightMapProviders.size(); i++) {
            heightMapProviders[i].isActive = _activeHeightMapLayers[i].value();
        }
        for (size_t i = 0; i < waterMaskProviders.size(); i++) {
            waterMaskProviders[i].isActive = _activeWaterMaskLayers[i].value();
        }

        // Update this after active layers have been updated
        _tileProviderManager->prerender();
    }

    glm::dvec3 RenderableGlobe::geodeticSurfaceProjection(glm::dvec3 position) {
        return _ellipsoid.geodeticSurfaceProjection(position);
    }

    std::shared_ptr<ChunkedLodGlobe> RenderableGlobe::chunkedLodGlobe() {
        return _chunkedLodGlobe;
    }


}  // namespace openspace