/* +------------------------------------------------------------------------+
   |                     Mobile Robot Programming Toolkit (MRPT)            |
   |                          https://www.mrpt.org/                         |
   |                                                                        |
   | Copyright (c) 2005-2023, Individual contributors, see AUTHORS file     |
   | See: https://www.mrpt.org/Authors - All rights reserved.               |
   | Released under BSD License. See: https://www.mrpt.org/License          |
   +------------------------------------------------------------------------+ */

#include "opengl-precomp.h"	 // Precompiled header
//
#include <mrpt/math/geometry.h>
#include <mrpt/opengl/CArrow.h>
#include <mrpt/serialization/CArchive.h>
#include <mrpt/serialization/CSchemeArchiveBase.h>

#include <memory>

using namespace mrpt;
using namespace mrpt::opengl;
using namespace mrpt::math;
using namespace std;

IMPLEMENTS_SERIALIZABLE(CArrow, CRenderizableShaderTriangles, mrpt::opengl)

void CArrow::onUpdateBuffers_Triangles()
{
	using P3f = mrpt::math::TPoint3Df;
	using V3f = mrpt::math::TVector3Df;

	std::unique_lock<std::shared_mutex> trisWriteLock(
		CRenderizableShaderTriangles::m_trianglesMtx.data);
	auto& tris = CRenderizableShaderTriangles::m_triangles;

	tris.clear();

	// Compute the XYZ local frame of reference for the arrow:
	// XY radial, Z from Point #0 -> point #1:
	const P3f p0(m_x0, m_y0, m_z0), p1(m_x1, m_y1, m_z1);
	auto p = p1 - p0;
	const float P10_norm = p.norm();
	ASSERT_GT_(P10_norm, .0f);
	// Unit vector:
	p *= (1.0f / P10_norm);

	setLocalRepresentativePoint((p0 + p1) * 0.5f);

	// each column is a unit vector:
	const CMatrixDouble44 HM = mrpt::math::generateAxisBaseFromDirectionAndAxis(
		p, 2 /* provided vector is "z"*/);

	// Transformation:
	const mrpt::poses::CPose3D T(
		HM.blockCopy<3, 3>(0, 0), mrpt::math::TPoint3D(m_x0, m_y0, m_z0));

	// precomputed table:
	ASSERT_GT_(m_slices, 2);

	const float dAng = 2 * M_PIf / m_slices;
	float a = 0;
	// unit cc points: cos(ang),sin(ang)
	std::vector<mrpt::math::TPoint2Df> cc(m_slices);
	for (unsigned int i = 0; i < m_slices; i++, a += dAng)
	{
		cc[i].x = cos(a);
		cc[i].y = sin(a);
	}

	ASSERT_GE_(m_headRatio, .0f);
	ASSERT_LE_(m_headRatio, 1.0f);

	const float r0 = m_smallRadius, r1 = m_largeRadius,
				h0 = P10_norm * (1.0f - m_headRatio), h1 = P10_norm;

	const float wall_tilt = 0;
	const float coswt = std::cos(wall_tilt), sinwt = std::sin(wall_tilt);

	const float head_tilt = std::atan2(r1, P10_norm * m_headRatio);
	const float cosht = std::cos(head_tilt), sinht = std::sin(head_tilt);

	// cylinder walls:
	for (unsigned int i = 0; i < m_slices; i++)
	{
		const auto ip = (i + 1) % m_slices;

		tris.emplace_back(
			// Points:
			T.composePoint(P3f(r0 * cc[i].x, r0 * cc[i].y, .0f)),
			T.composePoint(P3f(r0 * cc[ip].x, r0 * cc[ip].y, .0f)),
			T.composePoint(P3f(r0 * cc[i].x, r0 * cc[i].y, h0)),
			// Normals:
			T.rotateVector(V3f(coswt * cc[i].x, coswt * cc[i].y, sinwt)),
			T.rotateVector(V3f(coswt * cc[ip].x, coswt * cc[ip].y, sinwt)),
			T.rotateVector(V3f(coswt * cc[i].x, coswt * cc[i].y, sinwt)));

		tris.emplace_back(
			// Points:
			T.composePoint(P3f(r0 * cc[ip].x, r0 * cc[ip].y, .0f)),
			T.composePoint(P3f(r0 * cc[ip].x, r0 * cc[ip].y, h0)),
			T.composePoint(P3f(r0 * cc[i].x, r0 * cc[i].y, h0)),
			// Normals:
			T.rotateVector(V3f(coswt * cc[ip].x, coswt * cc[ip].y, sinwt)),
			T.rotateVector(V3f(coswt * cc[ip].x, coswt * cc[ip].y, sinwt)),
			T.rotateVector(V3f(coswt * cc[i].x, coswt * cc[i].y, sinwt)));
	}

	// top cone:
	if (m_headRatio > 0)
	{
		for (unsigned int i = 0; i < m_slices; i++)
		{
			const auto ip = (i + 1) % m_slices;
			tris.emplace_back(
				// Points:
				T.composePoint(P3f(r1 * cc[i].x, r1 * cc[i].y, h0)),
				T.composePoint(P3f(r1 * cc[ip].x, r1 * cc[ip].y, h0)),
				T.composePoint(P3f(.0f, .0f, h1)),
				// Normals:
				T.rotateVector(V3f(cosht * cc[i].x, cosht * cc[i].y, sinht)),
				T.rotateVector(V3f(cosht * cc[ip].x, cosht * cc[ip].y, sinht)),
				T.rotateVector(V3f(0, 0, 1)));
		}
	}

	// All faces, same color:
	for (auto& t : tris)
		t.setColor(m_color);
}

uint8_t CArrow::serializeGetVersion() const { return 3; }
void CArrow::serializeTo(mrpt::serialization::CArchive& out) const
{
	writeToStreamRender(out);
	out << m_x0 << m_y0 << m_z0;
	out << m_x1 << m_y1 << m_z1;
	out << m_headRatio << m_smallRadius << m_largeRadius;
	out << m_slices;
	CRenderizableShaderTriangles::params_serialize(out);  // v3
}

void CArrow::serializeFrom(mrpt::serialization::CArchive& in, uint8_t version)
{
	switch (version)
	{
		case 0:
		case 1:
		case 2:
		case 3:
		{
			readFromStreamRender(in);
			in >> m_x0 >> m_y0 >> m_z0;
			in >> m_x1 >> m_y1 >> m_z1;
			in >> m_headRatio >> m_smallRadius >> m_largeRadius;
			if (version == 1)
			{
				float arrow_roll, arrow_pitch, arrow_yaw;
				in >> arrow_roll >> arrow_pitch >> arrow_yaw;
			}
			if (version >= 2) in >> m_slices;
			if (version >= 3)
				CRenderizableShaderTriangles::params_deserialize(in);
		}
		break;
		default: MRPT_THROW_UNKNOWN_SERIALIZATION_VERSION(version);
	};
	CRenderizable::notifyChange();
}

void CArrow::serializeTo(mrpt::serialization::CSchemeArchiveBase& out) const
{
	SCHEMA_SERIALIZE_DATATYPE_VERSION(1);
	out["x0"] = m_x0;
	out["y0"] = m_y0;
	out["z0"] = m_z0;
	out["x1"] = m_x1;
	out["y1"] = m_y1;
	out["z1"] = m_z1;
	out["headRatio"] = m_headRatio;
	out["smallRadius"] = m_smallRadius;
	out["largeRadius"] = m_largeRadius;
	out["slices"] = m_slices;
}

void CArrow::serializeFrom(mrpt::serialization::CSchemeArchiveBase& in)
{
	uint8_t version;
	SCHEMA_DESERIALIZE_DATATYPE_VERSION();
	switch (version)
	{
		case 1:
		{
			m_x0 = static_cast<float>(in["x0"]);
			m_y0 = static_cast<float>(in["y0"]);
			m_z0 = static_cast<float>(in["z0"]);
			m_x1 = static_cast<float>(in["x1"]);
			m_y1 = static_cast<float>(in["y1"]);
			m_z1 = static_cast<float>(in["z1"]);
			m_headRatio = static_cast<float>(in["headRatio"]);
			m_smallRadius = static_cast<float>(in["smallRadius"]);
			m_largeRadius = static_cast<float>(in["largeRadius"]);
			m_slices = static_cast<unsigned int>(in["slices"]);
		}
		break;
		default: MRPT_THROW_UNKNOWN_SERIALIZATION_VERSION(version);
	}
}
auto CArrow::internalBoundingBoxLocal() const -> mrpt::math::TBoundingBoxf
{
	return mrpt::math::TBoundingBoxf::FromUnsortedPoints(
		{std::min(m_x0, m_x1), std::min(m_y0, m_y1), std::min(m_z0, m_z1)},
		{std::max(m_x0, m_x1), std::max(m_y0, m_y1), std::max(m_z0, m_z1)});
}
