// /*****************************************************************************************
//  *                                                                                       *
//  * 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/iswa/rendering/dataplane.h>
#include <ghoul/filesystem/filesystem>
#include <ghoul/io/texture/texturereader.h>
#include <ghoul/opengl/programobject.h>
#include <ghoul/opengl/textureunit.h>
#include <modules/kameleon/include/kameleonwrapper.h>
#include <openspace/scene/scene.h>
#include <openspace/scene/scenegraphnode.h>
#include <openspace/engine/openspaceengine.h>
#include <openspace/rendering/renderengine.h>
#include <openspace/util/spicemanager.h>
#include <openspace/util/time.h>

namespace {
	const std::string _loggerCat = "DataPlane";
}

namespace openspace {

DataPlane::DataPlane(const ghoul::Dictionary& dictionary)
	:CygnetPlane(dictionary)
	,_topColor("topColor", "Top Color", glm::vec4(1,0,0,1), glm::vec4(0), glm::vec4(1))
	,_midColor("midColor", "Mid Color", glm::vec4(0,0,0,0), glm::vec4(0), glm::vec4(1))
	,_botColor("botColor", "Bot Color", glm::vec4(0,0,1,1), glm::vec4(0), glm::vec4(1))
	,_tfValues("tfValues", "TF Values", glm::vec2(0.5,0.1), glm::vec2(0), glm::vec2(1))
	,_colorbar(nullptr)
{	
	_id = id();
	setName("DataPlane" + std::to_string(_id));

	addProperty(_topColor);
	addProperty(_midColor);
	addProperty(_botColor);
	addProperty(_tfValues);

	registerProperties();
	OsEng.gui()._iSWAproperty.registerProperty(&_topColor);
	OsEng.gui()._iSWAproperty.registerProperty(&_midColor);
	OsEng.gui()._iSWAproperty.registerProperty(&_botColor);
	OsEng.gui()._iSWAproperty.registerProperty(&_tfValues);

	dictionary.getValue("kwPath", _kwPath);
}

DataPlane::~DataPlane(){}


bool DataPlane::initialize(){
	// std::string kwPath;
	_kw = std::make_shared<KameleonWrapper>(_kwPath);
	// dictionary.getValue("KW", _kw);

	KameleonWrapper::Model model = _kw->model();
	if(	model == KameleonWrapper::Model::BATSRUS)
		_var = "p";
	else
		_var = "rho";


    createPlane();
    
    if (_shader == nullptr) {
    // DatePlane Program
    RenderEngine& renderEngine = OsEng.renderEngine();
    _shader = renderEngine.buildRenderProgram("PlaneProgram",
        "${MODULE_ISWA}/shaders/dataplane_vs.glsl",
        "${MODULE_ISWA}/shaders/dataplane_fs.glsl"
        );
    if (!_shader)
        return false;
    }

	_dimensions = glm::size3_t(500,500,1);
	float zSlice = 0.5f;
	_dataSlice = _kw->getUniformSliceValues(std::string(_var), _dimensions, zSlice);

    loadTexture();

    std::cout << "Creating Colorbar" << std::endl;
    _colorbar = std::make_shared<ColorBar>();
    if(_colorbar){
    	_colorbar->initialize(); 
    }

    return isReady();
}

bool DataPlane::deinitialize(){
    unregisterProperties();
    destroyPlane();
    destroyShader();
	
	_kw = nullptr;
	_memorybuffer = "";
	
	_colorbar->deinitialize();
	_colorbar = nullptr;

	return true;
}



void DataPlane::render(const RenderData& data){
	if(_texture){
		psc position = data.position;
		glm::mat4 transform = glm::mat4(1.0);

		glm::mat4 rotx = glm::rotate(transform, static_cast<float>(M_PI_2), glm::vec3(1, 0, 0));
		glm::mat4 roty = glm::rotate(transform, static_cast<float>(M_PI_2), glm::vec3(0, -1, 0));
		glm::mat4 rotz = glm::rotate(transform, static_cast<float>(M_PI_2), glm::vec3(0, 0, 1));

		glm::mat4 rot = glm::mat4(1.0);
		for (int i = 0; i < 3; i++){
			for (int j = 0; j < 3; j++){
				transform[i][j] = static_cast<float>(_stateMatrix[i][j]);
			}
		}

		transform = transform * rotz * roty; //BATSRUS

		if(_data->frame == "GSM"){
			glm::vec4 v(1,0,0,1);
			glm::vec3 xVec = glm::vec3(transform*v);
			xVec = glm::normalize(xVec);

			double  lt;
		    glm::vec3 sunVec =
		    SpiceManager::ref().targetPosition("SUN", "Earth", "GALACTIC", {}, _time, lt);
		    sunVec = glm::normalize(sunVec);

		    float angle = acos(glm::dot(xVec, sunVec));
		    glm::vec3 ref =  glm::cross(xVec, sunVec);

		    glm::mat4 rotation = glm::rotate(glm::mat4(1.0f), angle, ref); 
		    transform = rotation * transform;
		}

		position += transform*glm::vec4(_data->offset.x, _data->offset.z, _data->offset.y, _data->offset.w);

		// Activate shader
		_shader->activate();
		glEnable(GL_ALPHA_TEST);
		glDisable(GL_CULL_FACE);

		_shader->setUniform("ViewProjection", data.camera.viewProjectionMatrix());
		_shader->setUniform("ModelTransform", transform);

		_shader->setUniform("top", _topColor.value());
		_shader->setUniform("mid", _midColor.value());
		_shader->setUniform("bot", _botColor.value());
		_shader->setUniform("tfValues", _tfValues.value());

		setPscUniforms(*_shader.get(), data.camera, position);

		ghoul::opengl::TextureUnit unit;
		unit.activate();
		_texture->bind();
		_shader->setUniform("texture1", unit);

		glBindVertexArray(_quad);
		glDrawArrays(GL_TRIANGLES, 0, 6);
		glEnable(GL_CULL_FACE);
		_shader->deactivate();

		position += transform*(glm::vec4(0.5f*_data->scale.x+100.0f ,-0.5f*_data->scale.y, 0.0f, _data->scale.w));
    	// RenderData data = { *_camera, psc(), doPerformanceMeasurements };
		ColorBarData cbdata = {	data.camera, 
								position,
								transform,
								_topColor.value(),
								_midColor.value(),
								_botColor.value(),
								_tfValues.value()
								// transform
							  };	
		_colorbar->render(cbdata);
	}
}

void DataPlane::update(const UpdateData& data){
	if(_planeIsDirty)
		createPlane();

	_time = Time::ref().currentTime();
	_stateMatrix = SpiceManager::ref().positionTransformMatrix("GALACTIC", _data->frame, _time);
    
    float openSpaceUpdateInterval = abs(Time::ref().deltaTime()*_updateInterval);
    if(openSpaceUpdateInterval){
    	if(abs(_time-_lastUpdateTime) >= openSpaceUpdateInterval){
    		updateTexture();
    		_lastUpdateTime = _time;
    	}
    }

}

void DataPlane::loadTexture() {
		ghoul::opengl::Texture::FilterMode filtermode = ghoul::opengl::Texture::FilterMode::Linear;
		ghoul::opengl::Texture::WrappingMode wrappingmode = ghoul::opengl::Texture::WrappingMode::ClampToEdge;
		std::unique_ptr<ghoul::opengl::Texture> texture = 
			std::make_unique<ghoul::opengl::Texture>(_dataSlice, _dimensions, ghoul::opengl::Texture::Format::Red, GL_RED, GL_FLOAT, filtermode, wrappingmode);

		if (texture) {
			// LDEBUG("Loaded texture from '" << absPath(_path) << "'");

			texture->uploadTexture();

			// Textures of planets looks much smoother with AnisotropicMipMap rather than linear
			texture->setFilter(ghoul::opengl::Texture::FilterMode::Linear);

            _texture = std::move(texture);
		}
}

void DataPlane::updateTexture(){}

int DataPlane::id(){
	static int id = 0;
	return id++;
}
}// namespace openspace