Mercurial > ldforge / file revision

/*
 *  LDForge: LDraw parts authoring CAD
 *  Copyright (C) 2013 Santeri Piippo
 *  
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef GLDRAW_H
#define GLDRAW_H

#include <QGLWidget>
#include "common.h"
#include "ldtypes.h"

class QDialogButtonBox;
class RadioBox;
class QDoubleSpinBox;
class QSpinBox;
class QLineEdit;
class LDOpenFile;
class QTimer;

enum EditMode {
	Select,
	Draw
};

// Meta for overlays
struct overlayMeta {
	vertex v0, v1;
	ushort ox, oy;
	double lw, lh;
	str fname;
	QImage* img;
};

// =============================================================================
// GLRenderer
// 
// The main renderer object, draws the brick on the screen, manages the camera
// and selection picking. The instance of GLRenderer is accessible as
// g_win->R ()
// =============================================================================
class GLRenderer : public QGLWidget {
	Q_OBJECT
	
	PROPERTY (bool, drawOnly, setDrawOnly)
	PROPERTY (double, zoom, setZoom)
	PROPERTY (LDOpenFile*, file, setFile)
	READ_PROPERTY (bool, picking, setPicking)
	DECLARE_PROPERTY (EditMode, editMode, setEditMode)
	
public:
	enum Camera { Top, Front, Left, Bottom, Back, Right, Free };
	enum ListType { NormalList, PickList, BFCFrontList, BFCBackList };
	
	GLRenderer (QWidget* parent = null);
	~GLRenderer ();
	
	Camera         camera              () const { return m_camera; }
	Axis           cameraAxis          (bool y);
	const char*    cameraName          () const;
	void           clearOverlay        ();
	void           compileObject       (LDObject* obj);
	void           compileAllObjects   ();
	double         depthValue          () const;
	void           drawGLScene         ();
	void           endDraw             (bool accept);
	QColor         getMainColor        ();
	overlayMeta&   getOverlay          (int newcam);
	void           hardRefresh         ();
	void           initGLData          ();
	void           overlaysFromObjects ();
	void           refresh             ();
	void           resetAngles         ();
	uchar*         screencap           (ushort& w, ushort& h);
	void           setBackground       ();
	void           setCamera           (const Camera cam);
	void           setDepthValue       (double depth);
	bool           setupOverlay        ( GLRenderer::Camera cam, str file, int x, int y, int w, int h );
	void           updateOverlayObjects();
	void           zoomNotch           (bool inward);
	void           zoomToFit           ();
	
	static void    deleteLists         (LDObject* obj);

protected:
	void           contextMenuEvent      (QContextMenuEvent* ev);
	void           initializeGL          ();
	void           keyPressEvent         (QKeyEvent* ev);
	void           keyReleaseEvent       (QKeyEvent* ev);
	void           leaveEvent            (QEvent* ev);
	void           mouseDoubleClickEvent (QMouseEvent* ev);
	void           mousePressEvent       (QMouseEvent* ev);
	void           mouseMoveEvent        (QMouseEvent* ev);
	void           mouseReleaseEvent     (QMouseEvent* ev);
	void           paintEvent            (QPaintEvent* ev);
	void           resizeGL              (int w, int h);
	void           wheelEvent            (QWheelEvent* ev);

private:
	// CameraIcon::img is a heap-allocated QPixmap because otherwise it gets
	// initialized before program gets to main() and constructs a QApplication
	// and Qt doesn't like that.
	struct CameraIcon {
		QPixmap* img;
		QRect srcRect, destRect, selRect;
		Camera cam;
	} m_cameraIcons[7];
	
	QTimer* m_toolTipTimer;
	Qt::MouseButtons m_lastButtons;
	Qt::KeyboardModifiers m_keymods;
	ulong m_totalmove;
	vertex m_hoverpos;
	double m_virtWidth, m_virtHeight, m_rotX, m_rotY, m_rotZ, m_panX, m_panY;
	bool m_darkbg, m_rangepick, m_addpick, m_drawToolTip, m_screencap;
	QPoint m_pos, m_rangeStart;
	QPen m_thinBorderPen, m_thickBorderPen;
	Camera m_camera, m_toolTipCamera;
	uint m_axeslist;
	ushort m_width, m_height;
	vector<vertex> m_drawedVerts;
	bool m_rectdraw;
	vertex m_rectverts[4];
	QColor m_bgcolor;
	double m_depthValues[6];
	overlayMeta m_overlays[6];
	vector<vertex> m_knownVerts;
	bool m_panning;
	
	void           addDrawnVertex       (vertex m_hoverpos);
	void           calcCameraIcons      ();                                      // Compute geometry for camera icons
	void           clampAngle           (double& angle) const;                   // Clamps an angle to [0, 360]
	void           compileList          (LDObject* obj, const ListType list);    // Compile one of the lists of an object
	void           compileSubObject     (LDObject* obj, const GLenum gltype);    // Sub-routine for object compiling
	void           compileVertex        (const vertex& vrt);                     // Compile a single vertex to a list
	vertex         coordconv2_3         (const QPoint& pos2d, bool snap) const;  // Convert a 2D point to a 3D point
	QPoint         coordconv3_2         (const vertex& pos3d) const;             // Convert a 3D point to a 2D point
	LDOverlay*     findOverlayObject    ( Camera cam );
	void           updateRectVerts      ();
	void           pick                 (uint mouseX, uint mouseY);              // Perform object selection
	void           setObjectColor       (LDObject* obj, const ListType list);    // Set the color to an object list
	
private slots:
	void           slot_toolTipTimer    ();
};

// Alias for short namespaces
typedef GLRenderer GL;

static const GLRenderer::ListType g_glListTypes[] = {
	GL::NormalList,
	GL::PickList,
	GL::BFCFrontList,
	GL::BFCBackList,
};

extern const GL::Camera g_Cameras[7];
extern const char* g_CameraNames[7];

#endif // GLDRAW_H