gtkradiant/plugins/textool/ControlPointsManager.cpp

/*
   Copyright (C) 1999-2007 id Software, Inc. and contributors.
   For a list of contributors, see the accompanying CONTRIBUTORS file.

   This file is part of GtkRadiant.

   GtkRadiant 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 2 of the License, or
   (at your option) any later version.

   GtkRadiant 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 GtkRadiant; if not, write to the Free Software
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

//-----------------------------------------------------------------------------
//
// DESCRIPTION:
// a class to handle control points in a 2D view
// TODO: this one can be placed under an interface, and provided to the editor as service
//

#include "StdAfx.h"

void CControlPointsManagerBFace::Init( int iPts, CtrlPts_t *Pts, C2DView *p2DView, int TexSize[2],
									   _QERFaceData* pFaceData, _QERQglTable *pQglTable ){
	ManagerState = Idle;
	m_NumPoints = iPts;
	m_pPts = Pts;
	// store the initial config
	memcpy( &m_RefPts, Pts, sizeof( CtrlPts_t ) );
	// init TM
	memset( m_TM, 0, sizeof( float[2][3] ) );
	m_TM[0][0] = 1.0f; m_TM[1][1] = 1.0f;
	m_bGotAnchor = false;
	m_TransOffset[0] = 0.0f; m_TransOffset[1] = 0.0f;
	m_TexSize[0] = TexSize[0];
	m_TexSize[1] = TexSize[1];
	m_pFaceData = pFaceData;

	CControlPointsManager::Init( p2DView, pQglTable );
}

bool CControlPointsManagerBFace::OnLButtonDown( int xPos, int yPos ){
	if ( ManagerState == Idle ) {
		int i;

		// scan the point list to see if we selected something
		for ( i = 0; i < m_NumPoints; i++ )
			if ( m_p2DView->DoesSelect( xPos, yPos, m_pPts->data[i] ) ) {
				m_iDragPoint = i;
				ManagerState = Drag;
				if ( m_bGotAnchor && i == m_iAnchorPoint ) {
					// this means we selected the Anchor, so we'll translate
					m_bGotAnchor = false;
				}
				// perhaps we won't use translation, but we can compute it anyway
				ComputeTransOffset( i );
				if ( m_bGotAnchor ) {
					// we have an Anchor and selected another point
					m_Anchor[0] = m_pPts->data[m_iAnchorPoint][0];
					m_Anchor[1] = m_pPts->data[m_iAnchorPoint][1];
				}
			}
		// send a repaint message
		g_pToolWnd->Redraw();
		return true;
	}
	return false;
}

bool CControlPointsManagerBFace::OnMouseMove( int xPos, int yPos ){
	if ( ManagerState == Drag ) {
		if ( m_bGotAnchor ) {
			// there's an anchor, we are rotating the shape
			// we need to work in XY space for orthonormality
			float Pt[2];
			vec3_t V1,V2;
			vec3_t cross;
			float c,s;
			// used in XY space
			float XYTM[2][3];
			float XYRefAnchor[2];
			float XYAnchor[2];
			m_p2DView->GridForWindow( Pt, xPos, yPos );
			V2[0] = Pt[0] - m_Anchor[0];
			V2[1] = Pt[1] - m_Anchor[1];
			V2[2] = 0.0f;
			V1[0] = m_RefPts.data[m_iDragPoint][0] - m_RefPts.data[m_iAnchorPoint][0];
			V1[1] = m_RefPts.data[m_iDragPoint][1] - m_RefPts.data[m_iAnchorPoint][1];
			V1[2] = 0.0f;
			// compute transformation from V1 to V2
			// we need to work in XY orthonormal space
			XYSpaceForSTSpace( V1, V1 );
			XYSpaceForSTSpace( V2, V2 );
			VectorNormalize( V2, V2 );
			VectorNormalize( V1, V1 );
			c = DotProduct( V1, V2 );
			CrossProduct( V1, V2, cross );
			s = VectorLength( cross );
			// we compute the transformation matrix in XY space
			// reference position of the Anchor in XY space
			XYSpaceForSTSpace( XYRefAnchor, m_RefPts.data[m_iAnchorPoint] );
			// current position of the Anchor in XY space
			XYSpaceForSTSpace( XYAnchor, m_Anchor );
			// compute transformation matrix
			XYTM[0][0] = c; XYTM[1][1] = c;
			if ( cross[2] > 0 ) {
				s *= -1.0f;
			}
			XYTM[0][1] = s; XYTM[1][0] = -s;
			XYTM[0][2] = -c * XYRefAnchor[0] - s * XYRefAnchor[1] + XYAnchor[0];
			XYTM[1][2] = s * XYRefAnchor[0] - c * XYRefAnchor[1] + XYAnchor[1];
			// express this transformation matrix in ST space
			m_TM[0][0] = XYTM[0][0];
			m_TM[1][0] = XYTM[1][0] * (float)m_TexSize[0] / (float)m_TexSize[1];
			m_TM[0][1] = XYTM[0][1] * (float)m_TexSize[1] / (float)m_TexSize[0];
			m_TM[1][1] = XYTM[1][1];
			m_TM[0][2] = XYTM[0][2] / (float)m_TexSize[0];
			m_TM[1][2] = XYTM[1][2] / (float)m_TexSize[1];
			// update all points
			UpdateCtrlPts();
		}
		else
		{
			// no Anchor point is defined, we translate all points
			m_p2DView->GridForWindow( m_pPts->data[m_iDragPoint], xPos, yPos );
			m_TM[0][2] = m_pPts->data[m_iDragPoint][0] + m_TransOffset[0];
			m_TM[1][2] = m_pPts->data[m_iDragPoint][1] + m_TransOffset[1];
			// update all points
			UpdateCtrlPts();
		}
		// send a repaint message
		g_pToolWnd->Redraw();
		return true;
	}

	return false;
}

bool CControlPointsManagerBFace::OnLButtonUp( int x, int y ){
	if ( ManagerState == Drag ) {
		// this button is gonna become our Anchor
		m_bGotAnchor = true;
		m_iAnchorPoint = m_iDragPoint;
		// let's get out of Drag mode
		ManagerState = Idle;
		// send a repaint message
		g_pToolWnd->Redraw();
		return true;
	}
	return false;
}

void CControlPointsManagerBFace::Render(){
	int i;

	m_pQglTable->m_pfn_qglColor3f( 0, 1, 0 );
	m_pQglTable->m_pfn_qglPointSize( 6 );
	m_pQglTable->m_pfn_qglBegin( GL_POINTS );
	for ( i = 0; i < m_NumPoints; i++ )
	{
		if ( ManagerState == Drag && i == m_iDragPoint ) {
			m_pQglTable->m_pfn_qglColor3f( 1, 0, 0 );
		}
		else if ( m_bGotAnchor && i == m_iAnchorPoint ) {
			m_pQglTable->m_pfn_qglColor3f( 0, 0, 1 );
		}
		m_pQglTable->m_pfn_qglVertex2f( m_pPts->data[i][0], m_pPts->data[i][1] );
		m_pQglTable->m_pfn_qglColor3f( 0, 1, 0 );
	}
	m_pQglTable->m_pfn_qglEnd();
}

void CControlPointsManagerBFace::UpdateCtrlPts(){
	int i;

	// update all points
	for ( i = 0; i < m_NumPoints; i++ )
	{
		m_pPts->data[i][0] = m_RefPts.data[i][0] * m_TM[0][0] + m_RefPts.data[i][1] * m_TM[0][1] + m_TM[0][2];
		m_pPts->data[i][1] = m_RefPts.data[i][0] * m_TM[1][0] + m_RefPts.data[i][1] * m_TM[1][1] + m_TM[1][2];
	}

	if ( g_bPrefsUpdateCameraView ) {
		Commit();
		// tell Radiant to update
		// NOTE: little speed optimisation, disable window updates, and only update camera view
		g_FuncTable.m_pfnSetScreenUpdate( false );
		g_SelectedFaceTable.m_pfnSetFaceInfo( 0, m_pFaceData );
		g_FuncTable.m_pfnSetScreenUpdate( true );
		g_FuncTable.m_pfnSysUpdateWindows( W_CAMERA );
	}
}

//++timo FIXME: we are using a global for the reference data, use a m_pCancelFaceData instead
void CControlPointsManagerBFace::Commit(){
	brushprimit_texdef_t aux;
	aux.coords[0][0] = m_TM[0][0] * g_CancelFaceData.brushprimit_texdef.coords[0][0] + m_TM[0][1] * g_CancelFaceData.brushprimit_texdef.coords[1][0];
	aux.coords[0][1] = m_TM[0][0] * g_CancelFaceData.brushprimit_texdef.coords[0][1] + m_TM[0][1] * g_CancelFaceData.brushprimit_texdef.coords[1][1];
	aux.coords[0][2] = m_TM[0][0] * g_CancelFaceData.brushprimit_texdef.coords[0][2] + m_TM[0][1] * g_CancelFaceData.brushprimit_texdef.coords[1][2] + m_TM[0][2];
	aux.coords[1][0] = m_TM[1][0] * g_CancelFaceData.brushprimit_texdef.coords[0][0] + m_TM[1][1] * g_CancelFaceData.brushprimit_texdef.coords[1][0];
	aux.coords[1][1] = m_TM[1][0] * g_CancelFaceData.brushprimit_texdef.coords[0][1] + m_TM[1][1] * g_CancelFaceData.brushprimit_texdef.coords[1][1];
	aux.coords[1][2] = m_TM[1][0] * g_CancelFaceData.brushprimit_texdef.coords[0][2] + m_TM[1][1] * g_CancelFaceData.brushprimit_texdef.coords[1][2] + m_TM[1][2];
	memcpy( &m_pFaceData->brushprimit_texdef, &aux, sizeof( brushprimit_texdef_t ) );
}

void CControlPointsManagerBFace::ComputeTransOffset( int i ){
	// compute the translation offset used to counteract rotation
	m_TransOffset[0] = -m_TM[0][0] * m_RefPts.data[i][0] - m_TM[0][1] * m_RefPts.data[i][1];
	m_TransOffset[1] = -m_TM[1][0] * m_RefPts.data[i][0] - m_TM[1][1] * m_RefPts.data[i][1];
}

void CControlPointsManagerBFace::XYSpaceForSTSpace( float xy[2], const float st[2] ){
	xy[0] = st[0] * (float)m_TexSize[0];
	xy[1] = st[1] * (float)m_TexSize[1];
}

/*
   ======================================================================
   patch manager
   ======================================================================
 */

void CControlPointsManagerPatch::Init( patchMesh_t* pWorkPatch, C2DView *p2DView, _QERQglTable *pQglTable, patchMesh_t* pPatch ){
	CControlPointsManager::Init( p2DView, pQglTable );
	m_pPatch = pPatch;
	m_pWorkPatch = pWorkPatch;
}

bool CControlPointsManagerPatch::OnLButtonDown( int xPos, int yPos ){
	if ( ManagerState == Idle ) {
		int i,j;

		// scan the point list to see if we selected something
		for ( i = 0; i < m_pPatch->width; i++ )
			for ( j = 0; j < m_pPatch->height; j++ )
				if ( m_p2DView->DoesSelect( xPos, yPos, m_pWorkPatch->ctrl[i][j].st ) ) {
					m_iDragPoint[0] = i;
					m_iDragPoint[1] = j;
					ManagerState = Drag;
				}
		// send a repaint message
		g_pToolWnd->Redraw();
		return true;
	}
	return false;
}

bool CControlPointsManagerPatch::OnMouseMove( int xPos, int yPos ){
	if ( ManagerState == Drag ) {
		m_p2DView->GridForWindow( m_pWorkPatch->ctrl[ m_iDragPoint[0] ][ m_iDragPoint[1] ].st, xPos, yPos );
		if ( g_bPrefsUpdateCameraView ) {
			Commit();
			// ask to rebuild the patch display data
			m_pPatch->bDirty = true;
			// send a repaint to the camera window as well
			g_FuncTable.m_pfnSysUpdateWindows( W_CAMERA );
		}
		// send a repaint message
		g_pToolWnd->Redraw();
		return true;
	}
	return false;
}

bool CControlPointsManagerPatch::OnLButtonUp( int x, int y ){
	if ( ManagerState == Drag ) {
		ManagerState = Idle;
		// send a repaint message
		g_pToolWnd->Redraw();
	}
	return false;
}

void CControlPointsManagerPatch::Render(){
	int i,j;

	m_pQglTable->m_pfn_qglColor3f( 0, 1, 0 );
	m_pQglTable->m_pfn_qglPointSize( 6 );
	m_pQglTable->m_pfn_qglBegin( GL_POINTS );
	for ( i = 0; i < m_pPatch->width; i++ )
		for ( j = 0; j < m_pPatch->height; j++ )
		{
			if ( ManagerState == Drag && i == m_iDragPoint[0] && j == m_iDragPoint[1] ) {
				m_pQglTable->m_pfn_qglColor3f( 1, 0, 0 );
			}
			m_pQglTable->m_pfn_qglVertex2f( m_pWorkPatch->ctrl[i][j].st[0], m_pWorkPatch->ctrl[i][j].st[1] );
			m_pQglTable->m_pfn_qglColor3f( 0, 1, 0 );
		}
	m_pQglTable->m_pfn_qglEnd();
}

void CControlPointsManagerPatch::Commit(){
	int i,j;
	for ( i = 0; i < m_pPatch->width; i++ )
		for ( j = 0; j < m_pPatch->height; j++ )
		{
			m_pPatch->ctrl[i][j].st[0] = m_pWorkPatch->ctrl[i][j].st[0];
			m_pPatch->ctrl[i][j].st[1] = m_pWorkPatch->ctrl[i][j].st[1];
		}
}