csgo/cstrike15_src/utils/hlmv/attachments_window.cpp

//========= Copyright � 1996-2005, Valve Corporation, All rights reserved. ============//
//
// Purpose: 
//
//=============================================================================//

#include "attachments_window.h"
#include "ControlPanel.h"
#include "ViewerSettings.h"
#include "StudioModel.h"
#include "MatSysWin.h"


#define IDC_ATTACHMENT_LIST			(IDC_ATTACHMENT_WINDOW_FIRST+0)
#define IDC_ATTACHMENT_LIST_BONES	(IDC_ATTACHMENT_WINDOW_FIRST+1)
#define IDC_ATTACHMENT_TRANSLATION	(IDC_ATTACHMENT_WINDOW_FIRST+2)
#define IDC_ATTACHMENT_ROTATION		(IDC_ATTACHMENT_WINDOW_FIRST+3)
#define IDC_ATTACHMENT_QC_STRING	(IDC_ATTACHMENT_WINDOW_FIRST+4)

#define IDC_ATTACHMENT_NUDGE (IDC_ATTACHMENT_WINDOW_FIRST+5)


CAttachmentsWindow::CAttachmentsWindow( ControlPanel* pParent )	: mxWindow( pParent, 0, 0, 0, 0 )
{
	m_pControlPanel = pParent;
	g_viewerSettings.m_iEditAttachment = -1;
}


void CAttachmentsWindow::Init( )
{
	int left, top;
	left = 5;
	top = 0;

	// Attachment selection list
	new mxLabel( this, left + 3, top + 4, 60, 18, "Attachment" );
	m_cAttachmentList = new mxListBox( this, left, top + 20, 260, 100, IDC_ATTACHMENT_LIST );
	m_cAttachmentList->add ("None");
	m_cAttachmentList->select (0);
	mxToolTip::add (m_cAttachmentList, "Select an attachment to modify");

	left = 280;
	new mxLabel( this, left + 3, top + 4, 60, 18, "Attach To Bone" );
	m_cBoneList = new mxListBox( this, left, top + 20, 260, 100, IDC_ATTACHMENT_LIST_BONES );
	m_cBoneList->add ("None");
	m_cBoneList->select( 0 );
	mxToolTip::add( m_cBoneList, "Select a bone to attach to" );

	
	left = 5;
	top = 120;
	new mxLabel( this, left + 3, top + 4, 60, 18, "Translation" );
	m_cTranslation = new mxLineEdit2( this, left + 70, top, 90, 25, "10 20 30", IDC_ATTACHMENT_TRANSLATION );

	
	left = 170;
	top = 120;
	new mxLabel( this, left + 3, top + 4, 60, 18, "Rotation" );
	m_cRotation = new mxLineEdit2( this, left + 70, top, 90, 25, "0 90 180", IDC_ATTACHMENT_ROTATION );

	
	top = 145;
	left = 5;
	new mxLabel( this, left, top, 60, 18, "QC String" );
	m_cQCString = new mxLineEdit2( this, left + 70, top, 400, 25, "$attachment \"controlpanel0_ur\" \"Vgui\" -22 -15 4 rotate 0 0 0", IDC_ATTACHMENT_QC_STRING );

	top = 5;
	left = 550;

	m_bTranslateXLargeMinus = new mxButton( this, left, top, 30, 20, "-X", IDC_ATTACHMENT_NUDGE );
	m_bTranslateXSmallMinus = new mxButton( this, left+30, top, 30, 20, "-x", IDC_ATTACHMENT_NUDGE );
	m_bTranslateXSmallPlus  = new mxButton( this, left+70, top, 30, 20, "+x", IDC_ATTACHMENT_NUDGE );
	m_bTranslateXLargePlus  = new mxButton( this, left+100, top, 30, 20, "+X", IDC_ATTACHMENT_NUDGE );

	top += 20;

	m_bTranslateYLargeMinus = new mxButton( this, left, top, 30, 20, "-Y", IDC_ATTACHMENT_NUDGE );
	m_bTranslateYSmallMinus = new mxButton( this, left+30, top, 30, 20, "-y", IDC_ATTACHMENT_NUDGE );
	m_bTranslateYSmallPlus  = new mxButton( this, left+70, top, 30, 20, "+y", IDC_ATTACHMENT_NUDGE );
	m_bTranslateYLargePlus  = new mxButton( this, left+100, top, 30, 20, "+Y", IDC_ATTACHMENT_NUDGE );

	top += 20;

	m_bTranslateZLargeMinus = new mxButton( this, left, top, 30, 20, "-Z", IDC_ATTACHMENT_NUDGE );
	m_bTranslateZSmallMinus = new mxButton( this, left+30, top, 30, 20, "-z", IDC_ATTACHMENT_NUDGE );
	m_bTranslateZSmallPlus  = new mxButton( this, left+70, top, 30, 20, "+z", IDC_ATTACHMENT_NUDGE );
	m_bTranslateZLargePlus  = new mxButton( this, left+100, top, 30, 20, "+Z", IDC_ATTACHMENT_NUDGE );

	top += 30;

	m_bRotateXLargeMinus = new mxButton( this, left, top, 30, 20, "-XR", IDC_ATTACHMENT_NUDGE );
	m_bRotateXSmallMinus = new mxButton( this, left+30, top, 30, 20, "-xr", IDC_ATTACHMENT_NUDGE );
	m_bRotateXSmallPlus  = new mxButton( this, left+70, top, 30, 20, "+xr", IDC_ATTACHMENT_NUDGE );
	m_bRotateXLargePlus  = new mxButton( this, left+100, top, 30, 20, "+XR", IDC_ATTACHMENT_NUDGE );

	top += 20;

	m_bRotateYLargeMinus = new mxButton( this, left, top, 30, 20, "-YR", IDC_ATTACHMENT_NUDGE );
	m_bRotateYSmallMinus = new mxButton( this, left+30, top, 30, 20, "-yr", IDC_ATTACHMENT_NUDGE );
	m_bRotateYSmallPlus  = new mxButton( this, left+70, top, 30, 20, "+yr", IDC_ATTACHMENT_NUDGE );
	m_bRotateYLargePlus  = new mxButton( this, left+100, top, 30, 20, "+YR", IDC_ATTACHMENT_NUDGE );

	top += 20;

	m_bRotateZLargeMinus = new mxButton( this, left, top, 30, 20, "-ZR", IDC_ATTACHMENT_NUDGE );
	m_bRotateZSmallMinus = new mxButton( this, left+30, top, 30, 20, "-zr", IDC_ATTACHMENT_NUDGE );
	m_bRotateZSmallPlus  = new mxButton( this, left+70, top, 30, 20, "+zr", IDC_ATTACHMENT_NUDGE );
	m_bRotateZLargePlus  = new mxButton( this, left+100, top, 30, 20, "+ZR", IDC_ATTACHMENT_NUDGE );

}


void CAttachmentsWindow::OnLoadModel()
{
	int iPrevEdit = g_viewerSettings.m_iEditAttachment;
	PopulateBoneList();
	PopulateAttachmentsList();

	if ( iPrevEdit >= 0 && iPrevEdit < m_cAttachmentList->getItemCount())
	{
		m_cAttachmentList->select( iPrevEdit + 1 );
	}
	g_viewerSettings.m_iEditAttachment = iPrevEdit;
	UpdateStrings();
}


void CAttachmentsWindow::OnTabSelected()
{
	// for now, keep selection
	// g_viewerSettings.m_iEditAttachment = m_cAttachmentList->getSelectedIndex() - 1;
}


void CAttachmentsWindow::OnTabUnselected()
{
	// for now, keep selection
	// g_viewerSettings.m_iEditAttachment = -1;
}


void CAttachmentsWindow::PopulateAttachmentsList()
{
	m_cAttachmentList->removeAll();

	m_cAttachmentList->add( "(none)" );

	if ( g_pStudioModel )
	{
		CStudioHdr* pHdr = g_pStudioModel->GetStudioHdr();
		if (pHdr->GetNumAttachments())
		{
			for ( int i = 0; i < pHdr->GetNumAttachments(); i++ )
			{
				m_cAttachmentList->add ( pHdr->pAttachment(i).pszName() );
			}

			m_cAttachmentList->select (0);
			OnSelChangeAttachmentList();
			return;
		}
	}

	m_cAttachmentList->select (0);
}


void CAttachmentsWindow::PopulateBoneList()
{
	m_cBoneList->removeAll();

	if ( g_pStudioModel )
	{
		CStudioHdr* pHdr = g_pStudioModel->GetStudioHdr();
		if (pHdr->numbones())
		{
			for ( int i = 0; i < pHdr->numbones(); i++ )
			{
				m_cBoneList->add ( pHdr->pBone(i)->pszName() );
			}

			m_cBoneList->select (0);
			return;
		}
	}

	m_cBoneList->add( "None" );
	m_cBoneList->select (0);
}


int CAttachmentsWindow::handleEvent (mxEvent *event)
{
	MDLCACHE_CRITICAL_SECTION_( g_pMDLCache );

	if ( !g_pStudioModel )
		return 0;

	CStudioHdr* pHdr = g_pStudioModel->GetStudioHdr();
	switch( event->action )
	{
		case IDC_ATTACHMENT_LIST:
		{
			OnSelChangeAttachmentList();
		}
		break;

		case IDC_ATTACHMENT_LIST_BONES:
		{
			int iAttachment = g_viewerSettings.m_iEditAttachment;
			int iBone = m_cBoneList->getSelectedIndex();

			if ( iAttachment >= 0 && 
				iAttachment < pHdr->GetNumAttachments() && 
				iBone >= 0 && 
				iBone < pHdr->numbones() )
			{
				pHdr->SetAttachmentBone( iAttachment, iBone );
				UpdateStrings();
			} 
		}
		break;

		case IDC_ATTACHMENT_TRANSLATION:
		{
			int iAttachment = g_viewerSettings.m_iEditAttachment;

			if ( iAttachment >= 0 && 
				iAttachment < pHdr->GetNumAttachments() )
			{
				mstudioattachment_t &pAttachment = (mstudioattachment_t &)pHdr->pAttachment( iAttachment );
				
				Vector vTrans( 0, 0, 0 );
				char curText[512];
				m_cTranslation->getText( curText, sizeof( curText ) );
				sscanf( curText, "%f %f %f", &vTrans.x, &vTrans.y, &vTrans.z );
				
				pAttachment.local[0][3] = vTrans.x;
				pAttachment.local[1][3] = vTrans.y;
				pAttachment.local[2][3] = vTrans.z;

				UpdateStrings( true, false, false );
			}
		}
		break;
	
		case IDC_ATTACHMENT_ROTATION:
		{
			int iAttachment = g_viewerSettings.m_iEditAttachment;

			if ( iAttachment >= 0 && 
				iAttachment < pHdr->GetNumAttachments() )
			{
				mstudioattachment_t &pAttachment = (mstudioattachment_t &)pHdr->pAttachment( iAttachment );
				
				QAngle vRotation( 0, 0, 0 );
				char curText[512];
				m_cRotation->getText( curText, sizeof( curText ) );
				sscanf( curText, "%f %f %f", &vRotation.x, &vRotation.y, &vRotation.z );
				
				Vector vTrans = GetCurrentTranslation();
				AngleMatrix( vRotation, vTrans, pAttachment.local );
				
				UpdateStrings( true, false, false );
			}
		}
		break;
	
		case IDC_ATTACHMENT_NUDGE:
		{
			int iAttachment = g_viewerSettings.m_iEditAttachment;
			if ( iAttachment >= 0 && iAttachment < pHdr->GetNumAttachments() )
			{
				mstudioattachment_t &pAttachment = (mstudioattachment_t &)pHdr->pAttachment( iAttachment );
				
				QAngle vRotation( 0, 0, 0 );
				char curText[512];
				m_cRotation->getText( curText, sizeof( curText ) );
				sscanf( curText, "%f %f %f", &vRotation.x, &vRotation.y, &vRotation.z );
				
				Vector vTrans( 0, 0, 0 );
				m_cTranslation->getText( curText, sizeof( curText ) );
				sscanf( curText, "%f %f %f", &vTrans.x, &vTrans.y, &vTrans.z );

				if ( Q_strcmp( event->widget->getLabel(), "+x" ) == 0 )
					vTrans.x += 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "-x" ) == 0 )
					vTrans.x -= 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "+X" ) == 0 )
					vTrans.x += 1;
				if ( Q_strcmp( event->widget->getLabel(), "-X" ) == 0 )
					vTrans.x -= 1;

				if ( Q_strcmp( event->widget->getLabel(), "+y" ) == 0 )
					vTrans.y += 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "-y" ) == 0 )
					vTrans.y -= 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "+Y" ) == 0 )
					vTrans.y += 1;
				if ( Q_strcmp( event->widget->getLabel(), "-Y" ) == 0 )
					vTrans.y -= 1;

				if ( Q_strcmp( event->widget->getLabel(), "+z" ) == 0 )
					vTrans.z += 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "-z" ) == 0 )
					vTrans.z -= 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "+Z" ) == 0 )
					vTrans.z += 1;
				if ( Q_strcmp( event->widget->getLabel(), "-Z" ) == 0 )
					vTrans.z -= 1;

				if ( Q_strcmp( event->widget->getLabel(), "+xr" ) == 0 )
					vRotation.x += 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "-xr" ) == 0 )
					vRotation.x -= 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "+XR" ) == 0 )
					vRotation.x += 1;
				if ( Q_strcmp( event->widget->getLabel(), "-XR" ) == 0 )
					vRotation.x -= 1;

				if ( Q_strcmp( event->widget->getLabel(), "+yr" ) == 0 )
					vRotation.y += 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "-yr" ) == 0 )
					vRotation.y -= 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "+YR" ) == 0 )
					vRotation.y += 1;
				if ( Q_strcmp( event->widget->getLabel(), "-YR" ) == 0 )
					vRotation.y -= 1;

				if ( Q_strcmp( event->widget->getLabel(), "+zr" ) == 0 )
					vRotation.z += 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "-zr" ) == 0 )
					vRotation.z -= 0.1f;
				if ( Q_strcmp( event->widget->getLabel(), "+ZR" ) == 0 )
					vRotation.z += 1;
				if ( Q_strcmp( event->widget->getLabel(), "-ZR" ) == 0 )
					vRotation.z -= 1;

				AngleMatrix( vRotation, vTrans, pAttachment.local );
				UpdateStrings( true, true, true );
			}
		}
		break;

		default:
			return 0;
	}

	return 1;
}


void CAttachmentsWindow::OnSelChangeAttachmentList()
{
	CStudioHdr *pStudioHdr = g_pStudioModel ? g_pStudioModel->GetStudioHdr() : NULL;
	
	if ( !pStudioHdr )
		return;

	int iAttachment = m_cAttachmentList->getSelectedIndex() - 1;
	if ( iAttachment >= 0 && iAttachment < pStudioHdr->GetNumAttachments() )
	{
		g_viewerSettings.m_iEditAttachment = iAttachment;

		// Init the bone list index.
		int iBone = g_pStudioModel->GetStudioHdr()->GetAttachmentBone( iAttachment );
		m_cBoneList->select( iBone );
	}
	else
	{
		g_viewerSettings.m_iEditAttachment = -1;
	}

	UpdateStrings();
}


Vector CAttachmentsWindow::GetCurrentTranslation()
{
	CStudioHdr *pStudioHdr = g_pStudioModel ? g_pStudioModel->GetStudioHdr() : NULL;
	
	int iAttachment = m_cAttachmentList->getSelectedIndex() - 1;
	if ( pStudioHdr && iAttachment >= 0 && iAttachment < pStudioHdr->GetNumAttachments() )
	{
		mstudioattachment_t &pAttachment = (mstudioattachment_t &)pStudioHdr->pAttachment( iAttachment );

		return Vector( pAttachment.local[0][3],
			pAttachment.local[1][3],
			pAttachment.local[2][3] );
	}
	else
	{
		return vec3_origin;
	}
}


Vector CAttachmentsWindow::GetCurrentRotation()
{
	CStudioHdr *pStudioHdr = g_pStudioModel ? g_pStudioModel->GetStudioHdr() : NULL;
	
	int iAttachment = m_cAttachmentList->getSelectedIndex() - 1;
	if ( pStudioHdr && iAttachment >= 0 && iAttachment < pStudioHdr->GetNumAttachments() )
	{
		mstudioattachment_t &pAttachment = (mstudioattachment_t &)pStudioHdr->pAttachment( iAttachment );

		float angles[3];
		MatrixAngles( pAttachment.local, angles );
		return Vector( angles[0], angles[1], angles[2] );
	}
	else
	{
		return vec3_origin;
	}
}


void CAttachmentsWindow::UpdateStrings( bool bUpdateQC, bool bUpdateTranslation, bool bUpdateRotation )
{
	char str[1024];

	int iAttachment = -1;
	CStudioHdr* pHdr = NULL;
	if ( g_pStudioModel )
	{
		pHdr = g_pStudioModel->GetStudioHdr();
		iAttachment = m_cAttachmentList->getSelectedIndex() - 1;
		if ( iAttachment < 0 || iAttachment >= pHdr->GetNumAttachments() )
			iAttachment = -1;
	}

	if ( iAttachment == -1 )
	{
		m_cTranslation->setText( "(none)" );
		m_cRotation->setText( "(none)" );
		m_cQCString->setText( "(none)" );
	}
	else
	{
		mstudioattachment_t &pAttachment = (mstudioattachment_t &)pHdr->pAttachment( iAttachment );
		int iBone= pHdr->GetAttachmentBone( iAttachment );
		Vector vTranslation = GetCurrentTranslation();
		Vector vRotation = GetCurrentRotation();

		if ( bUpdateQC )
		{
			sprintf( str, "$attachment \"%s\" \"%s\" %.2f %.2f %.2f rotate %.0f %.0f %.0f", 
				pAttachment.pszName(),
				pHdr->pBone( iBone )->pszName(),
				VectorExpand( vTranslation ),
				VectorExpand( vRotation ) );

			m_cQCString->setText( str );
		}

		if ( bUpdateTranslation )
		{
			sprintf( str, "%.2f %.2f %.2f", VectorExpand( vTranslation ) );
			m_cTranslation->setText( str );
		}

		if ( bUpdateRotation )
		{
			sprintf( str, "%.0f %.0f %.0f", VectorExpand( vRotation ) );
			m_cRotation->setText( str );
		}
	}
}