surface_deformation.h 2.82 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
#ifndef _SURFACE_DEFORMATION_PLUGIN_H_
#define _SURFACE_DEFORMATION_PLUGIN_H_

#include "plugin_interaction.h"
#include "surface_deformation_dockTab.h"

#include "mapHandler.h"

#include "Container/fakeAttribute.h"

#include "NL/nl.h"
#include "Algo/LinearSolving/basic.h"
#include "Eigen/Dense"

namespace CGoGN
{

namespace SCHNApps
{

typedef NoTypeNameAttribute<Eigen::Matrix3f> Eigen_Matrix3f;

struct MapParameters
{
	MapParameters();
	~MapParameters();

	void start(MapHandlerGen* mh);
	void stop(MapHandlerGen* mh);

	VertexAttribute<PFP2::VEC3> positionAttribute;

	CellSelector<VERTEX>* lockedSelector;
	CellSelector<VERTEX>* handleSelector;

	bool initialized;

	VertexAttribute<PFP2::VEC3> positionInit;
	VertexAttribute<PFP2::VEC3> diffCoord;
	VertexAttribute<Eigen_Matrix3f> vertexRotationMatrix;
	VertexAttribute<PFP2::VEC3> rotatedDiffCoord;

	VertexAttribute<unsigned int> vIndex;
	unsigned int nb_vertices;

	NLContext nlContext;
};

class Surface_Deformation_Plugin : public PluginInteraction
{
	Q_OBJECT
	Q_INTERFACES(CGoGN::SCHNApps::Plugin)

	friend class Surface_Deformation_DockTab;

public:
	Surface_Deformation_Plugin() :
		m_dragging(false)
	{}

	~Surface_Deformation_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 selectedMapChanged(MapHandlerGen* prev, MapHandlerGen* cur);
	void mapAdded(MapHandlerGen* map);
	void mapRemoved(MapHandlerGen* map);

	// slots called from MapHandler signals
	void attributeAdded(unsigned int orbit, const QString& name);
	void cellSelectorAdded(unsigned int orbit, const QString& name);
	void cellSelectorRemoved(unsigned int orbit, const QString& name);
	void selectedCellsChanged();

public slots:
	// slots for Python calls
	void changePositionAttribute(const QString& map, const QString& name);
	void changeLockedSelector(const QString& map, const QString& name);
	void changeHandleSelector(const QString& map, const QString& name);

protected:
	void toggleMapDeformation(MapHandlerGen* map);
	void matchDiffCoord(MapHandlerGen* map);
	void asRigidAsPossible(MapHandlerGen* map);

protected:
	Surface_Deformation_DockTab* m_dockTab;
	QHash<MapHandlerGen*, MapParameters> h_parameterSet;

	bool m_dragging;
	PFP2::REAL m_dragZ;
	qglviewer::Vec m_dragPrevious;
};

} // namespace SCHNApps

} // namespace CGoGN

#endif