surface_radiance.h

#ifndef _SURFACE_RADIANCE_PLUGIN_H_
#define _SURFACE_RADIANCE_PLUGIN_H_

#include "plugin_interaction.h"
#include "surface_radiance_dockTab.h"
#include "dialog_computeRadianceDistance.h"

#include "Utils/sphericalHarmonics.h"
#include "Utils/bivariatePolynomials.h"
#include "Utils/Shaders/shaderRadiancePerVertex.h"
#include "Utils/Shaders/shaderRadiancePerVertex_P.h"
#include "Utils/drawer.h"

#include "Algo/Decimation/decimation.h"

namespace CGoGN
{

namespace SCHNApps
{

struct MapParameters
{
	MapParameters() :
		positionVBO(NULL),
		tangentVBO(NULL),
		normalVBO(NULL),
		binormalVBO(NULL),
		radiancePerVertexShader(NULL),
		radiancePerVertexPShader(NULL),
		radianceTexture(NULL),
		paramVBO(NULL),
		positionApproximator(NULL),
		normalApproximator(NULL),
		radianceApproximator(NULL),
		selector(NULL)
	{}

	unsigned int nbVertices;

	Utils::VBO* positionVBO;
	Utils::VBO* tangentVBO;
	Utils::VBO* normalVBO;
	Utils::VBO* binormalVBO;

	CGoGN::Utils::ShaderRadiancePerVertex* radiancePerVertexShader;
	CGoGN::Utils::ShaderRadiancePerVertex_P* radiancePerVertexPShader;

	VertexAttribute<Utils::SphericalHarmonics<PFP2::REAL, PFP2::VEC3>, PFP2::MAP> radiance;
	VertexAttribute<Utils::BivariatePolynomials<PFP2::REAL, PFP2::VEC3>, PFP2::MAP> radiance_P;

	Utils::Texture<2, Geom::Vec3f>* radianceTexture;
	VertexAttribute<Geom::Vec2i, PFP2::MAP> param;
	Utils::VBO* paramVBO;

//	DartAttribute<PFP2::REAL, PFP2::MAP> halfedgeError;

	Algo::Surface::Decimation::ApproximatorGen<PFP2>* positionApproximator;
	Algo::Surface::Decimation::ApproximatorGen<PFP2>* normalApproximator;
	Algo::Surface::Decimation::ApproximatorGen<PFP2>* radianceApproximator;

	Algo::Surface::Decimation::Selector<PFP2>* selector;
};

class Surface_Radiance_Plugin : public PluginInteraction
{
	Q_OBJECT
	Q_INTERFACES(CGoGN::SCHNApps::Plugin)
#if CGOGN_QT_DESIRED_VERSION == 5
	Q_PLUGIN_METADATA(IID "CGoGN.SCHNapps.Plugin")
#endif

	friend class Surface_Radiance_DockTab;

public:
	Surface_Radiance_Plugin()
	{}

	~Surface_Radiance_Plugin()
	{}

	virtual bool enable();
	virtual void disable();

	virtual void draw(View *view) {}
	virtual void drawMap(View* view, MapHandlerGen* map);

	virtual void keyPress(View* view, QKeyEvent* event) {}
	virtual void keyRelease(View* view, QKeyEvent* event) {}
	virtual void mousePress(View* view, QMouseEvent* event) {}
	virtual void mouseRelease(View* view, QMouseEvent* event) {}
	virtual void mouseMove(View* view, QMouseEvent* event) {}
	virtual void wheelEvent(View* view, QWheelEvent* event) {}

	virtual void viewLinked(View *view) {}
	virtual void viewUnlinked(View *view) {}

private slots:
	// slots called from SCHNApps signals
	void selectedViewChanged(View* prev, View* cur);
	void selectedMapChanged(MapHandlerGen* prev, MapHandlerGen* cur);
	void mapAdded(MapHandlerGen* map);
	void mapRemoved(MapHandlerGen* map);
	void schnappsClosing();

	// slots called from MapHandler signals
	void vboAdded(Utils::VBO* vbo);
	void vboRemoved(Utils::VBO* vbo);
	void attributeModified(unsigned int orbit, QString nameAttr);

	void importFromFileDialog_SH();
	void importFromFileDialog_P();

	void openComputeRadianceDistanceDialog();
	void computeRadianceDistanceFromDialog();

public slots:
	// slots for Python calls
	void changePositionVBO(const QString& map, const QString& vbo);
	void changeNormalVBO(const QString& map, const QString& vbo);
	MapHandlerGen* importFromFile_SH(const QString& fileName);
	MapHandlerGen* importFromFile_P(const QString& fileName);
	void decimate(
		const QString& mapName,
		const QString& positionAttributeName,
		const QString& normalAttributeName,
		float decimationGoal,
		bool halfCollapse = false,
		bool exportMeshes = false
	);
	void computeRadianceDistance(const QString& mapName1,
		const QString& positionAttributeName1,
		const QString& normalAttributeName1,
		const QString& distanceAttributeName1,
		const QString& mapName2,
		const QString& positionAttributeName2,
		const QString& normalAttributeName2,
		const QString& distanceAttributeName2
	);
	void exportPLY(
		const QString& mapName,
		const QString& positionAttributeName,
		const QString& normalAttributeName,
		const QString& filename
	);

protected:
	MapHandlerGen* currentlyDecimatedMap() { return m_currentlyDecimatedMap; }
	bool currentDecimationHalf() { return m_currentDecimationHalf; }
	static void checkNbVerticesAndExport(Surface_Radiance_Plugin* p, const unsigned int* nbVertices);

	Surface_Radiance_DockTab* m_dockTab;

	Dialog_ComputeRadianceDistance* m_computeRadianceDistanceDialog;
	QAction* m_computeRadianceDistanceAction;

	QHash<MapHandlerGen*, MapParameters> h_mapParameterSet;

	MapHandlerGen* m_currentlyDecimatedMap;
	bool m_currentDecimationHalf;
	std::vector<unsigned int> exportNbVert;
	unsigned int nextExportIndex;

	QAction* m_importSHAction;
	QAction* m_importPAction;

	static bool isInHemisphere(double x, double y, double z, void* u)
	{ // true iff [x,y,z] and u have the same direction
		PFP2::REAL* n = (PFP2::REAL*)(u);
		return x*n[0] + y*n[1] + z*n[2] >= 0.0;
	}

	static double SHEvalCartesian_Error(double x, double y, double z, void* u)
	{
		Utils::SphericalHarmonics<PFP2::REAL, PFP2::VEC3>& e = *(Utils::SphericalHarmonics<PFP2::REAL, PFP2::VEC3>*)(u);
		PFP2::VEC3 c = e.evaluate_at(x, y, z);
		return c.norm2();
	}
};

} // namespace SCHNApps

} // namespace CGoGN

#endif // _SURFACE_RADIANCE_PLUGIN_H_