/* +---------------------------------------------------------------------------+
   |                     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 details in https://www.mrpt.org/License   |
   +---------------------------------------------------------------------------+
   */
#pragma once
#include <QGLWidget>

#include "CRobotPose.h"
#include "mrpt/gui/CQtGlCanvasBase.h"
#include "mrpt/maps/CMultiMetricMap.h"
#include "mrpt/maps/CSimpleMap.h"
#include "mrpt/opengl/CGridPlaneXY.h"
#include "mrpt/opengl/CPlanarLaserScan.h"
#include "mrpt/opengl/CPointCloud.h"
#include "mrpt/opengl/CSetOfLines.h"
#include "mrpt/opengl/CSetOfObjects.h"

class CDocument;

/** This class implements the work with map*/
class CGlWidget : public mrpt::gui::CQtGlCanvasBase
{
	Q_OBJECT
   public:
	CGlWidget(bool is2D, QWidget* parent = nullptr);
	~CGlWidget() override;

	void fillMap(const mrpt::opengl::CSetOfObjects::Ptr& renderizableMap);
	void setDocument(CDocument* doc);

	void updateObservations();

	void setSelected(const mrpt::math::TPose3D& pose);
	void setSelectedObservation(bool is);
	void setLaserScan(mrpt::opengl::CPlanarLaserScan::Ptr laserScan);

	void setZoom(float zoom);
	float getZoom() const;

	void setCameraParams(const CamaraParams& params) override;

	void setAzimuthDegrees(float ang) override;
	void setElevationDegrees(float ang) override;

	void setBackgroundColor(float r, float g, float b, float a);
	void setGridColor(double r, double g, double b, double a);

	void setVisibleGrid(bool is);

	bool setBot(int value);

	bool setObservationSize(double s);
	bool setObservationColor(int type);

	bool setSelectedObservationSize(double s);
	bool setSelectedObservationColor(int type);

   public slots:
	void updateSelectionWithoutSignals(const std::vector<size_t>& idx);

   signals:
	void selectedChanged(const std::vector<CRobotPose::Ptr>& idx);
	void zoomChanged(float zoom);
	void mousePosChanged(double x, double y);
	void azimuthChanged(float ang);
	void elevationChanged(float ang);
	void deleteRobotPoses(const std::vector<size_t>& idx);
	void moveRobotPoses(const std::vector<size_t>& idx, const QPointF& dist);
	void showPoseDirection(size_t idx, double yaw, double pitch, double roll);

   protected:
	void resizeGL(int width, int height) override;
	void updateCamerasParams() override;
	void insertToMap(
		const mrpt::opengl::CRenderizable::Ptr& newObject) override;
	void removeFromMap(
		const mrpt::opengl::CRenderizable::Ptr& newObject) override;
	void mouseMoveEvent(QMouseEvent* event) override;
	void mousePressEvent(QMouseEvent* event) override;
	void mouseReleaseEvent(QMouseEvent* event) override;
	void keyPressEvent(QKeyEvent* event) override;

   private:
	mrpt::img::TColorf typeToColor(int type) const;
	std::pair<bool, mrpt::math::TPoint3D> sceneToWorld(const QPoint& pos) const;
	double maximumSizeObservation(const QPoint& pos) const;

	bool selectPoint(float x, float y, double maxDist);
	void selectPoint(size_t index);

	template <class Container>
	int searchPoseFromList(
		float x, float y, double maxDist, Container list) const;
	CRobotPose::Ptr removePoseFromPointsCloud(
		mrpt::opengl::CSetOfObjects::Ptr points, int index) const;

	void removeRobotDirection();
	void updateMinimapPos();
	int searchSelectedPose(float x, float y, double maxDist);
	int searchPose(float x, float y, double maxDist);
	void removePoseFromSelected(int index);
	void selectPose(CRobotPose::Ptr robotPose);
	bool deselectAll();
	CRobotPose::Ptr getRobotPose(size_t index);

	mrpt::opengl::Viewport::Ptr m_miniMapViewport;
	mrpt::opengl::CSetOfObjects::Ptr m_map;
	mrpt::opengl::CGridPlaneXY::Ptr m_groundPlane;

	CDocument* m_doc;
	float m_miniMapSize;
	const float m_minimapPercentSize;
	double m_observationSize;
	mrpt::img::TColorf m_observationColor;
	double m_selectedObsSize;
	mrpt::img::TColorf m_selectedColor;

	bool m_isShowObs;
	mrpt::opengl::CSetOfObjects::Ptr m_visiblePoints;
	std::vector<CRobotPose::Ptr> m_selectedPoints;
	mrpt::opengl::CPlanarLaserScan::Ptr m_currentLaserScan;
	mrpt::opengl::CRenderizable::Ptr m_currentObs;
	mrpt::opengl::CSetOfLines::Ptr m_line;

	bool m_is2D;
	bool m_showRobot;
	bool m_moveSelected;

	QPoint m_pressedPos;
	QPoint m_lastPos;
};
