/* 
 * esmini - Environment Simulator Minimalistic 
 * https://github.com/esmini/esmini
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 * 
 * Copyright (c) partners of Simulation Scenarios
 * https://sites.google.com/view/simulationscenarios
 */

#ifndef VIEWER_HPP_
#define VIEWER_HPP_

#include <osg/PositionAttitudeTransform>
#include <osgViewer/Viewer>
#include <osgGA/NodeTrackerManipulator>
#include <osg/MatrixTransform>
#include <osg/Material>
#include <osgText/Text>
#include <osgAnimation/EaseMotion>
#include <osg/BlendColor>
#include <string>

#include "RubberbandManipulator.hpp"
#include "IdealSensor.hpp"
#include "RoadManager.hpp"
#include "CommonMini.hpp"
#include "roadgeom.hpp"

#define TRAIL_DOT_FADE_DURATION 3.0  // seconds
#define TRAIL_DOTS_DT 0.5
#define TRAIL_MAX_DOTS 50
#define TRAIL_DOT_LIFE_SPAN (0.5 * TRAIL_MAX_DOTS * TRAIL_DOTS_DT) // Start fade when half of the dots have been launched (seconds)

extern double color_green[3];
extern double color_gray[3];
extern double color_dark_gray[3];
extern double color_red[3];
extern double color_blue[3];
extern double color_yellow[3];
extern double color_white[3];

using namespace scenarioengine;

namespace viewer
{
	typedef enum
	{
		NODE_MASK_NONE =           (0),
		NODE_MASK_OBJECT_SENSORS = (1 << 0),
		NODE_MASK_TRAILS =         (1 << 1),
		NODE_MASK_ODR_FEATURES =   (1 << 2),
		NODE_MASK_OSI_POINTS =     (1 << 3),
		NODE_MASK_OSI_LINES =      (1 << 4),
		NODE_MASK_ENV_MODEL =      (1 << 5),
		NODE_MASK_ENTITY_MODEL =   (1 << 6),
		NODE_MASK_ENTITY_BB =      (1 << 7),
		NODE_MASK_INFO =           (1 << 8),
		NODE_MASK_ROAD_SENSORS =   (1 << 9),
	} NodeMask;

	class Line
	{
	public:
		osg::ref_ptr<osg::Geometry> line_;
		osg::ref_ptr<osg::Vec3Array> line_vertex_data_;

		Line(double x0, double y0, double z0, double x1, double y1, double z1, double r, double g, double b);
		void SetPoints(double x0, double y0, double z0, double x1, double y1, double z1);
	};

	class SensorViewFrustum
	{
	public:
		osg::ref_ptr<osg::PositionAttitudeTransform> txNode_;
		osg::ref_ptr<osg::Group> line_group_;
		std::vector<Line*> lines_;
		ObjectSensor *sensor_;

		SensorViewFrustum(ObjectSensor *sensor, osg::Group *parent);
		~SensorViewFrustum();
		void Update();
	};

	class AlphaFadingCallback : public osg::StateAttributeCallback
	{
	public:
		AlphaFadingCallback(osgViewer::Viewer *viewer, osg::Vec4 color)
		{
			_motion = new osgAnimation::InCubicMotion(0.0f, TRAIL_DOT_FADE_DURATION);
			color_ = color;
			viewer_ = viewer;
			Reset();
		}
		virtual void operator()(osg::StateAttribute*, osg::NodeVisitor*);
		void Reset() 
		{ 
			born_time_stamp_ = viewer_->elapsedTime();
			time_stamp_ = born_time_stamp_;
			_motion->reset(); 
		}

	protected:
		osg::ref_ptr<osgAnimation::InCubicMotion> _motion;

	private:
		osg::Vec4 color_;
		double time_stamp_;
		double born_time_stamp_;
		osgViewer::Viewer *viewer_;
	};

	class TrailDot
	{
	public:
		osg::ref_ptr<osg::PositionAttitudeTransform> dot_;
		osg::ref_ptr<osg::Material> material_;

		TrailDot(double x, double y, double z, double heading,
			osgViewer::Viewer *viewer, osg::Group *parent, osg::ref_ptr<osg::Node> dot_node, osg::Vec4 trail_color);
		void Reset(double x, double y, double z, double heading);

	private:
		osg::ref_ptr<AlphaFadingCallback> fade_callback_;
	};

	class Trail
	{
	public:
		TrailDot* dot_[TRAIL_MAX_DOTS];
		int n_dots_;
		int current_;
		osg::Group *parent_;
		osg::Node *dot_node_;
		void AddDot(double x, double y, double z, double heading);

		Trail(osg::Group *parent, osgViewer::Viewer *viewer, osg::ref_ptr<osg::Node> dot_node, osg::Vec3 color) :
			parent_(parent), 
			viewer_(viewer),
			n_dots_(0), 
			current_(0),
			dot_node_(dot_node)
		{
			color_[0] = color[0];
			color_[1] = color[1];
			color_[2] = color[2];
		}
		~Trail();

	private:
		osg::Vec4 color_;
		osgViewer::Viewer *viewer_;
	};

	class PointSensor
	{
	public:
		osg::ref_ptr<osg::Group> group_;
		osg::ref_ptr<osg::PositionAttitudeTransform> ball_;
		double ball_radius_;
		osg::ref_ptr<osg::Geometry> line_;
		osg::ref_ptr<osg::Vec3Array> line_vertex_data_;

		osg::Vec3 pivot_pos;
		osg::Vec3 target_pos;

		PointSensor() : line_(0), line_vertex_data_(0), ball_(0) {};
		void Show() { group_->setNodeMask(NODE_MASK_ROAD_SENSORS); }
		void Hide() { group_->setNodeMask(0x0); };
	};

	class EntityModel
	{
	public:

		typedef enum {
			ENTITY_TYPE_VEHICLE,
			ENTITY_TYPE_OTHER
		} EntityType;

		osg::ref_ptr<osg::Group> group_;
		osg::ref_ptr<osg::LOD> lod_;
		osg::ref_ptr<osg::PositionAttitudeTransform> txNode_;
		osg::ref_ptr<osg::Group> bb_;
		osg::Quat quat_;
		double size_x;
		double size_y;
		double center_x;
		double center_y;
		static const int entity_type_ = ENTITY_TYPE_OTHER;
		virtual int GetType() { return entity_type_; }

		std::string name_;
		std::string filename_;
		osg::ref_ptr<osg::BlendColor> blend_color_;
		osg::ref_ptr<osg::StateSet> state_set_;

		EntityModel(osgViewer::Viewer* viewer, osg::ref_ptr<osg::Group> group, osg::ref_ptr<osg::Group> parent, osg::ref_ptr<osg::Group> 
			trail_parent, osg::ref_ptr<osg::Node> dot_node, osg::Vec3 trail_color, std::string name);
		void SetPosition(double x, double y, double z);
		void SetRotation(double h, double p, double r);

		void SetTransparency(double factor);


		Trail* trail_;
		osgViewer::Viewer* viewer_;
	};

	class CarModel : public EntityModel
	{
	public:
		std::vector<osg::ref_ptr<osg::PositionAttitudeTransform>> wheel_;
		double wheel_angle_;
		double wheel_rot_;
		PointSensor* road_sensor_;
		PointSensor* lane_sensor_;
		PointSensor* trail_sensor_;
		PointSensor* steering_sensor_;
		static const int entity_type_ = ENTITY_TYPE_VEHICLE;
		virtual int GetType() { return entity_type_; }

		CarModel(osgViewer::Viewer* viewer, osg::ref_ptr<osg::Group> group, osg::ref_ptr<osg::Group> parent, osg::ref_ptr<osg::Group>
			trail_parent, osg::ref_ptr<osg::Node> dot_node, osg::Vec3 trail_color, std::string name);
		~CarModel();
		osg::ref_ptr<osg::PositionAttitudeTransform>  AddWheel(osg::ref_ptr<osg::Node> carNode, const char* wheelName);
		void UpdateWheels(double wheel_angle, double wheel_rotation);
		void UpdateWheelsDelta(double wheel_angle, double wheel_rotation_delta);
	};

	class VisibilityCallback : public osg::NodeCallback
	{
	public:
		VisibilityCallback(osg::Node* node, scenarioengine::Object* object, EntityModel* entity)
		{
			node_ = (osg::LOD*)node;
			object_ = object;
			entity_ = entity;
		}
		virtual void operator()(osg::Node*, osg::NodeVisitor*);

	protected:
		osg::ref_ptr<osgAnimation::InCubicMotion> _motion;

	private:
		scenarioengine::Object* object_;
		EntityModel* entity_;
		osg::LOD* node_;
	};

	typedef struct
	{
		int key_;
		int modKeyMask_;
		bool down_;
	} KeyEvent;

	typedef enum
	{
		ENTITY_HIDE,
		ENTITY_3D_MODEL,
		ENTITY_BOUNDINGBOX,
		ENTITY_BOTH
	} ENTITY_SHOW_MODE;

	typedef void (*KeyEventCallbackFunc)(KeyEvent*, void*);

	typedef struct
	{
		KeyEventCallbackFunc func;
		void* data;
	} KeyEventCallback;

	class Viewer
	{
	public:
		int currentCarInFocus_;
		int camMode_;
		osg::ref_ptr<osg::Group> line_node_;

		// Vehicle position debug visualization
		osg::ref_ptr<osg::Node> shadow_node_;

		// Trail dot model
		osg::ref_ptr<osg::Node> dot_node_;

		// Road debug visualization
		osg::ref_ptr<osg::Group> odrLines_;
		osg::ref_ptr<osg::Group> osiLines_;
		osg::ref_ptr<osg::Group> osiPoints_;
		osg::ref_ptr<osg::PositionAttitudeTransform> envTx_;
		osg::ref_ptr<osg::Node> environment_;
		osg::ref_ptr<osgGA::RubberbandManipulator> rubberbandManipulator_;
		osg::ref_ptr<osgGA::NodeTrackerManipulator> nodeTrackerManipulator_;
		std::vector<EntityModel*> entities_;
		float lodScale_;
		osgViewer::Viewer *osgViewer_;
		osg::MatrixTransform* rootnode_;
		osg::ref_ptr<osg::Group> roadSensors_;
		osg::ref_ptr<osg::Group> trails_;
		roadmanager::OpenDrive *odrManager_;
		bool showInfoText;
		RoadGeom* roadGeom;

		std::string exe_path_;
		std::vector<KeyEventCallback> callback_;

		osg::ref_ptr<osg::Camera> infoTextCamera;
		osg::ref_ptr<osgText::Text> infoText;

		Viewer(roadmanager::OpenDrive *odrManager, const char* modelFilename, const char* scenarioFilename, const char* exe_path, osg::ArgumentParser arguments, SE_Options* opt = 0);
		~Viewer();
		void SetCameraMode(int mode);
		void SetVehicleInFocus(int idx);
		int GetEntityInFocus() { return currentCarInFocus_; }
		EntityModel* AddEntityModel(std::string modelFilepath, osg::Vec3 trail_color, EntityModel::EntityType type, 
			bool road_sensor, std::string name, OSCBoundingBox *boundingBox);
                EntityModel* AddRadarModel(osg::Vec3 trail_color,std::string name);
		void RemoveCar(std::string name);
		int LoadShadowfile(std::string vehicleModelFilename);
		int AddEnvironment(const char* filename);
		osg::ref_ptr<osg::Group> LoadEntityModel(const char *filename);
		void UpdateSensor(PointSensor *sensor);
		void SensorSetPivotPos(PointSensor *sensor, double x, double y, double z);
		void SensorSetTargetPos(PointSensor *sensor, double x, double y, double z);
		void UpdateRoadSensors(PointSensor *road_sensor, PointSensor *lane_sensor, roadmanager::Position *pos);
		void setKeyUp(bool pressed) { keyUp_ = pressed; }
		void setKeyDown(bool pressed) { keyDown_ = pressed; }
		void setKeyLeft(bool pressed) { keyLeft_ = pressed; }
		void setKeyRight(bool pressed) { keyRight_ = pressed; }
		bool getKeyUp() { return keyUp_; }
		bool getKeyDown() { return keyDown_; }
		bool getKeyLeft() { return keyLeft_; }
		bool getKeyRight() { return keyRight_; }
		void SetQuitRequest(bool value) { quit_request_ = value; }
		bool GetQuitRequest() { return quit_request_;  }
		void SetInfoTextProjection(int width, int height);
		void SetInfoText(const char* text);
		void ShowInfoText(bool show);
		void SetNodeMaskBits(int bits);
		void SetNodeMaskBits(int mask, int bits);
		void ClearNodeMaskBits(int bits);
		void ToggleNodeMaskBits(int bits);
		int GetNodeMaskBit(int mask);
		PointSensor* CreateSensor(double color[], bool create_ball, bool create_line, double ball_radius, double line_width);
		bool CreateRoadSensors(CarModel *vehicle_model);
		void SetWindowTitle(std::string title);
		void SetWindowTitleFromArgs(std::vector<std::string> &arg);
		void SetWindowTitleFromArgs(int argc, char* argv[]);
		void RegisterKeyEventCallback(KeyEventCallbackFunc func, void* data);

	private:

		bool CreateRoadLines(roadmanager::OpenDrive* od);
		bool CreateRoadMarkLines(roadmanager::OpenDrive* od);
		int CreateOutlineObject(roadmanager::Outline* outline);
		int LoadRoadFeature(roadmanager::Road* road, std::string filename, double s, double t,
			double z_offset, double scale_x, double scale_y, double scale_z, double heading_offset);
		int CreateRoadSignsAndObjects(roadmanager::OpenDrive* od);
		bool keyUp_;
		bool keyDown_;
		bool keyLeft_;
		bool keyRight_;
		bool quit_request_;
	};

	class ViewerEventHandler : public osgGA::GUIEventHandler
	{
	public:
		ViewerEventHandler(Viewer *viewer) : viewer_(viewer) {}
		bool handle(const osgGA::GUIEventAdapter& ea, osgGA::GUIActionAdapter&);

	private:
		Viewer* viewer_;
	};
}



#endif  // VIEWER_HPP_