This repository has been archived on 2025-02-27. You can view files and clone it, but cannot push or open issues or pull requests.
CnC_Renegade/Code/Tools/Blender2/Blender2.cpp

772 lines
19 KiB
C++

/*
** Command & Conquer Renegade(tm)
** Copyright 2025 Electronic Arts Inc.
**
** 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/>.
*/
#include "blender2.h"
#include <istdplug.h>
#include "utilapi.h"
#include "appdata.h"
//----------------------------------------------------------------------------
// Blender_Class
//----------------------------------------------------------------------------
class Blender_Class : public UtilityObj
{
public:
Blender_Class();
void BeginEditParams(Interface *ip,IUtil *iu);
void EndEditParams(Interface *ip,IUtil *iu);
void SelectionSetChanged ( Interface * ip, IUtil * ui );
void DeleteThis() {}
void Init(HWND hWnd);
void Destroy(HWND hWnd);
void Close () { iu->CloseUtility (); }
void Get_Active_Time_Range ();
void Blend_Keys ();
void Loop_Controllers ();
void OutputObject(INode *node,TCHAR *fname);
private:
BOOL Is_Root ( INode * node );
float Heading_Delta_From_Quat ( Quat q );
void Set_Data_Chunk ( INode * node, const Blender_Data_Chunk & new_data );
void Remove_Data_Chunk ( INode * node );
int first_frame;
int first_match;
int last_frame;
int last_match;
ISpinnerControl * first_frame_spin;
ISpinnerControl * first_match_spin;
ISpinnerControl * last_frame_spin;
ISpinnerControl * last_match_spin;
IUtil * iu;
Interface *ip;
HWND hPanel;
};
//----------------------------------------------------------------------------
// the_blender
//----------------------------------------------------------------------------
static Blender_Class the_blender;
//----------------------------------------------------------------------------
// Blender_Desc_Class
//----------------------------------------------------------------------------
class Blender_Desc_Class:public ClassDesc
{
public:
int IsPublic() {return 1;}
void * Create(BOOL) {return &the_blender;}
const TCHAR * ClassName() {return _T("Key Blender");}
SClass_ID SuperClassID() {return UTILITY_CLASS_ID;}
Class_ID ClassID() {return Blender_Class_ID;}
const TCHAR* Category() {return _T("Westwood Tools");}
};
//----------------------------------------------------------------------------
// blender_desc
//----------------------------------------------------------------------------
static Blender_Desc_Class blender_desc;
//----------------------------------------------------------------------------
// BlenderDesc
//----------------------------------------------------------------------------
ClassDesc* BlenderDesc() {return &blender_desc;}
//----------------------------------------------------------------------------
// BlenderDlgProc
//----------------------------------------------------------------------------
static BOOL CALLBACK BlenderDlgProc
(
HWND hWnd,
UINT msg,
WPARAM wParam,
LPARAM lParam
)
{
switch (msg)
{
case WM_INITDIALOG:
the_blender.Init(hWnd);
break;
case WM_DESTROY:
the_blender.Destroy(hWnd);
break;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case ID_CLOSE:
the_blender.Close ();
break;
case ID_GET_ACTIVE_TIME_RANGE:
the_blender.Get_Active_Time_Range ();
break;
case ID_APPLY:
the_blender.Blend_Keys ();
break;
}
break;
default:
return FALSE;
}
return TRUE;
}
//----------------------------------------------------------------------------
// Blender_Class::Blender_Class
//----------------------------------------------------------------------------
Blender_Class::Blender_Class()
{
iu = NULL;
ip = NULL;
hPanel = NULL;
first_frame = 0;
first_match = 0;
last_frame = 0;
last_match = 0;
}
//----------------------------------------------------------------------------
// Blender_Class::BeginEditParams
//----------------------------------------------------------------------------
void Blender_Class::BeginEditParams(Interface *ip,IUtil *iu)
{
this->iu = iu;
this->ip = ip;
hPanel = ip->AddRollupPage
(
hInstance,
MAKEINTRESOURCE(IDD_ASCIIOUT_PANEL),
BlenderDlgProc,
_T("Key Blender"),
0
);
}
//----------------------------------------------------------------------------
// Blender_Class::EndEditParams
//----------------------------------------------------------------------------
void Blender_Class::EndEditParams ( Interface *ip, IUtil *iu )
{
this->iu = NULL;
this->ip = NULL;
ip->DeleteRollupPage(hPanel);
hPanel = NULL;
}
//----------------------------------------------------------------------------
// Blender_Class::SelectionSetChanged
//----------------------------------------------------------------------------
void Blender_Class::SelectionSetChanged ( Interface * ip, IUtil * iu )
{
if ( ip->GetSelNodeCount () == 0 )
{
EnableWindow ( GetDlgItem ( hPanel, ID_APPLY ), FALSE );
}
else
{
EnableWindow ( GetDlgItem ( hPanel, ID_APPLY ), TRUE );
}
}
//----------------------------------------------------------------------------
// Blender_Class::Init
//----------------------------------------------------------------------------
void Blender_Class::Init ( HWND hWnd )
{
if ( ip->GetSelNodeCount () == 0 )
{
EnableWindow ( GetDlgItem ( hWnd, ID_APPLY ), FALSE );
}
else
{
EnableWindow ( GetDlgItem ( hWnd, ID_APPLY ), TRUE );
}
int ticks_per_frame = GetTicksPerFrame ();
int start_frame = ip->GetAnimRange ().Start () / ticks_per_frame;
int end_frame = ip->GetAnimRange ().End () / ticks_per_frame;
first_frame = start_frame;
first_match = start_frame;
last_frame = end_frame;
last_match = start_frame;
first_frame_spin = SetupIntSpinner
(
hWnd,
IDC_FIRST_SPIN,
IDC_FIRST_EDIT,
-65536,
65535,
first_frame
);
first_frame_spin->SetResetValue ( first_frame );
first_match_spin = SetupIntSpinner
(
hWnd,
IDC_FIRST_MATCH_SPIN,
IDC_FIRST_MATCH_EDIT,
-65536,
65535,
first_match
);
first_match_spin->SetResetValue ( first_match );
last_frame_spin = SetupIntSpinner
(
hWnd,
IDC_LAST_SPIN,
IDC_LAST_EDIT,
-65536,
65535,
last_frame
);
last_frame_spin->SetResetValue ( last_frame );
last_match_spin = SetupIntSpinner
(
hWnd,
IDC_LAST_MATCH_SPIN,
IDC_LAST_MATCH_EDIT,
-65536,
65535,
last_match
);
last_match_spin->SetResetValue ( last_match );
}
//----------------------------------------------------------------------------
// Blender_Class::Destroy
//----------------------------------------------------------------------------
void Blender_Class::Destroy(HWND hWnd)
{
}
//----------------------------------------------------------------------------
// Blender_Class::Get_Active_Time_Range
//----------------------------------------------------------------------------
void Blender_Class::Get_Active_Time_Range ()
{
int ticks_per_frame = GetTicksPerFrame ();
int start_frame = ip->GetAnimRange ().Start () / ticks_per_frame;
int end_frame = ip->GetAnimRange ().End () / ticks_per_frame;
first_frame_spin->SetValue ( start_frame, FALSE );
last_frame_spin->SetValue ( end_frame, FALSE );
first_match_spin->SetValue ( start_frame, FALSE );
last_match_spin->SetValue ( start_frame, FALSE );
}
//----------------------------------------------------------------------------
// Blender_Class::Blend_Keys
//----------------------------------------------------------------------------
static const Quat zero (0.0,0.0,0.0,1.0);
void Blender_Class::Blend_Keys ()
{
Interval interval; // Used for returned validity intervals (which are not used).
int ticks_per_frame = GetTicksPerFrame ();
TimeValue start_time = first_frame_spin->GetIVal () * ticks_per_frame;
TimeValue end_time = last_frame_spin->GetIVal () * ticks_per_frame;
TimeValue start_match = first_match_spin->GetIVal () * ticks_per_frame;
TimeValue end_match = last_match_spin->GetIVal () * ticks_per_frame;
float t_scale = 1.0f / (float) (end_time - start_time);
BOOL bad_controller_found = FALSE;
theHold.Begin ();
SetCursor ( LoadCursor (NULL, IDC_WAIT) );
SuspendAnimate ();
AnimateOn ();
int number_of_nodes = ip->GetSelNodeCount ();
for ( int i = 0; i < number_of_nodes; ++ i )
{
// Get the inode.
INode * inode = ip->GetSelNode ( i );
Control * tm_controller = inode->GetTMController ();
if ( tm_controller == NULL )
continue;
Quat rot_start_delta = zero;
Quat rot_end_delta = zero;
Blender_Data_Chunk data;
data.position_delta = Point3 (0.0f, 0.0f, 0.0f);
data.heading_delta = 0.0f;
Control * c;
//--------------------------------------------------------------------
// Get the rotation controller and change its keys.
c = tm_controller->GetRotationController ();
if ( c != NULL )
{
if ( c->ClassID () != Class_ID (TCBINTERP_ROTATION_CLASS_ID, 0) )
{
#if 0
char m [ 256 ];
sprintf ( m, "Node %s has rotation controller ID %u.",
inode->GetName (), c->ClassID ().PartA () );
MessageBox ( GetActiveWindow (), m, "Debug", MB_OK );
#endif
bad_controller_found = TRUE;
}
else
{
Quat actual_start_orientation;
Quat desired_start_orientation;
Quat actual_end_orientation;
Quat desired_end_orientation;
c->GetValue ( start_time, & actual_start_orientation, interval );
c->GetValue ( start_match, & desired_start_orientation, interval );
c->GetValue ( end_time, & actual_end_orientation, interval );
c->GetValue ( end_match, & desired_end_orientation, interval );
// Ensure there are keys at the beginning and end of the blend interval.
c->SetValue ( start_time, & actual_start_orientation );
c->SetValue ( end_time, & actual_end_orientation );
actual_end_orientation.MakeClosest ( actual_start_orientation );
desired_start_orientation.MakeClosest ( actual_start_orientation );
desired_end_orientation.MakeClosest ( actual_end_orientation );
rot_start_delta = desired_start_orientation / actual_start_orientation;
rot_end_delta = desired_end_orientation / actual_end_orientation;
data.heading_delta = Heading_Delta_From_Quat
( actual_end_orientation / actual_start_orientation );
#if 1
int number_of_keys = c->NumKeys ();
for ( int j = 0; j < number_of_keys; ++ j )
{
TimeValue key_time = c->GetKeyTime ( j );
if ( key_time >= start_time && key_time <= end_time )
{
Quat orientation;
c->GetValue ( key_time, & orientation, interval );
float t = (float) (key_time - start_time) * t_scale;
Quat delta = Slerp ( rot_start_delta, zero, t ) *
Slerp ( zero, rot_end_delta, t );
orientation = orientation * delta;
c->SetValue ( key_time, & orientation );
}
}
#else
IKeyControl * keys = GetKeyControlInterface ( c );
if ( keys != NULL )
{
int number_of_keys = keys->GetNumKeys ();
Quat prev_key_absolute (0.0,0.0,0.0,1.0);
Quat new_prev_key_absolute (0.0,0.0,0.0,1.0);
for ( int j = 0; j < number_of_keys; ++ j )
{
ITCBRotKey key;
keys->GetKey ( j, & key );
Quat this_key_absolute = prev_key_absolute * (Quat) key.val;
Quat new_key_absolute = this_key_absolute;
if ( key.time >= start_time && key.time <= end_time )
{
float t = (float) (key.time - start_time) * t_scale;
Quat delta = Slerp ( rot_start_delta, zero, t ) *
Slerp ( zero, rot_end_delta, t );
new_key_absolute = this_key_absolute * delta;
key.val = new_key_absolute / new_prev_key_absolute;
keys->SetKey ( j, & key );
}
prev_key_absolute = this_key_absolute;
new_prev_key_absolute = new_key_absolute;
}
}
#endif
}
}
//--------------------------------------------------------------------
// Get the position controller and change its keys.
c = tm_controller->GetPositionController ();
if ( c != NULL )
{
if ( c->ClassID () != Class_ID (TCBINTERP_POSITION_CLASS_ID, 0) )
{
bad_controller_found = TRUE;
}
else
{
Point3 actual_start_position;
Point3 desired_start_position;
Point3 actual_end_position;
Point3 desired_end_position;
c->GetValue ( start_time, & actual_start_position, interval );
c->GetValue ( start_match, & desired_start_position, interval );
c->GetValue ( end_time, & actual_end_position, interval );
c->GetValue ( end_match, & desired_end_position, interval );
// Ensure there are keys at the beginning and end of the blend interval.
c->SetValue ( start_time, & actual_start_position );
c->SetValue ( end_time, & actual_end_position );
data.position_delta = actual_end_position - actual_start_position;
#if 1
int number_of_keys = c->NumKeys ();
for ( int j = 0; j < number_of_keys; ++ j )
{
TimeValue key_time = c->GetKeyTime ( j );
if ( key_time >= start_time && key_time <= end_time )
{
Point3 position;
c->GetValue ( key_time, & position, interval );
float t = (float) (key_time - start_time) * t_scale;
// Calculate the position the node would be in if it
// traveled in a straight line.
Point3 actual_position = actual_start_position * (1.0f - t) +
actual_end_position * t;
// Find the delta between the straight-line position and
// the real position.
Point3 delta = position - actual_position;
// Rotate the delta by the change in orientation
// at this time.
Quat rot_delta = Slerp ( rot_start_delta, zero, t ) *
Slerp ( zero, rot_end_delta, t );
Matrix3 rot_matrix;
rot_delta.MakeMatrix ( rot_matrix );
delta = delta * rot_matrix;
// Add the delta to the straight-line position on the
// desired line.
Point3 desired_position = desired_start_position * (1.0f - t) +
desired_end_position * t;
position = desired_position + delta;
// Store the new key value.
c->SetValue ( key_time, & position );
}
}
#else
IKeyControl * keys = GetKeyControlInterface ( c );
if ( keys != NULL )
{
int number_of_keys = keys->GetNumKeys ();
for ( int j = 0; j < number_of_keys; ++ j )
{
ITCBPoint3Key key;
keys->GetKey ( j, & key );
if ( key.time >= start_time && key.time <= end_time )
{
float t = (float) (key.time - start_time) * t_scale;
// Calculate the position the node would be in if it
// traveled in a straight line.
Point3 actual_position = actual_start_position * (1.0f - t) +
actual_end_position * t;
// Find the delta between the straight-line position and
// the real position.
Point3 delta = key.val - actual_position;
// Rotate the delta by the change in orientation
// at this time.
Quat rot_delta = Slerp ( rot_start_delta, zero, t ) *
Slerp ( zero, rot_end_delta, t );
Matrix3 rot_matrix;
rot_delta.MakeMatrix ( rot_matrix );
delta = delta * rot_matrix;
// Add the delta to the straight-line position on the
// desired line.
Point3 desired_position = desired_start_position * (1.0f - t) +
desired_end_position * t;
key.val = desired_position + delta;
// Store the new key value.
keys->SetKey ( j, & key );
}
}
}
#endif
}
}
// If this node is a root node (its parent is not selected), attach
// a data chunk that describes its motion during the blended section.
if ( Is_Root ( inode ) )
Set_Data_Chunk ( inode, data );
else
Remove_Data_Chunk ( inode );
inode->InvalidateTM ();
}
ResumeAnimate ();
TSTR undostr;
undostr.printf ( "Blend Keys" );
theHold.Accept ( undostr );
SetCursor ( LoadCursor (NULL, IDC_ARROW) );
ip->RedrawViews ( ip->GetTime () );
if ( bad_controller_found )
{
MessageBox ( GetActiveWindow (),
"Warning: Some controllers were not blended\n"
"because they were not TCB controllers.", "Warning", MB_OK );
}
}
//----------------------------------------------------------------------------
// Blender_Class::Is_Root
//----------------------------------------------------------------------------
BOOL Blender_Class::Is_Root ( INode * node )
{
node = node->GetParentNode ();
int number_of_nodes = ip->GetSelNodeCount ();
for ( int i = 0; i < number_of_nodes; ++ i )
{
if ( ip->GetSelNode ( i ) == node )
return FALSE;
}
return TRUE;
}
//----------------------------------------------------------------------------
// Blender_Class::Heading_Delta_From_Quat
//----------------------------------------------------------------------------
float Blender_Class::Heading_Delta_From_Quat ( Quat q )
{
Matrix3 rot_matrix;
q.MakeMatrix ( rot_matrix );
Point3 row = rot_matrix.GetRow ( 0 );
return (float) atan2 ( row.y, row.x );
}
//----------------------------------------------------------------------------
// Blender_Class::Set_Data_Chunk
//----------------------------------------------------------------------------
void Blender_Class::Set_Data_Chunk
(
INode * node,
const Blender_Data_Chunk & new_data
)
{
AppDataChunk * chunk = node->GetAppDataChunk
(
Blender_Class_ID,
UTILITY_CLASS_ID,
1
);
Blender_Data_Chunk * data;
if ( chunk != NULL )
{
data = (Blender_Data_Chunk *) chunk->data;
}
else
{
data = (Blender_Data_Chunk *) malloc ( sizeof (Blender_Data_Chunk) );
node->AddAppDataChunk
(
Blender_Class_ID,
UTILITY_CLASS_ID,
1,
sizeof (Blender_Data_Chunk),
data
);
}
*data = new_data;
#if 0
char m [ 256 ];
sprintf ( m, "Setting data chunk for %s:\n"
"Position delta = (%.1f, %.1f, %.1f)\n"
"Heading delta = %.1f",
node->GetName (),
data->position_delta.x,
data->position_delta.y,
data->position_delta.z,
data->heading_delta * 180 / PI );
MessageBox ( GetActiveWindow (), m, "Debug", MB_OK );
#endif
}
//----------------------------------------------------------------------------
// Blender_Class::Remove_Data_Chunk
//----------------------------------------------------------------------------
void Blender_Class::Remove_Data_Chunk ( INode * node )
{
node->RemoveAppDataChunk
(
Blender_Class_ID,
UTILITY_CLASS_ID,
1
);
}
//----------------------------------------------------------------------------
// Blender_Class::Loop_Controllers
//----------------------------------------------------------------------------
void Blender_Class::Loop_Controllers ()
{
int number_of_nodes = ip->GetSelNodeCount ();
for ( int i = 0; i < number_of_nodes; ++ i )
{
// Get the inode.
INode * inode = ip->GetSelNode ( i );
Control * tm_controller = inode->GetTMController ();
if ( tm_controller == NULL )
continue;
Control * c;
c = tm_controller->GetRotationController ();
if ( c )
{
c->SetORT ( ORT_LOOP, ORT_BEFORE );
c->SetORT ( ORT_LOOP, ORT_AFTER );
}
c = tm_controller->GetPositionController ();
if ( c )
{
c->SetORT ( ORT_LOOP, ORT_BEFORE );
c->SetORT ( ORT_LOOP, ORT_AFTER );
}
c = tm_controller->GetScaleController ();
if ( c )
{
c->SetORT ( ORT_LOOP, ORT_BEFORE );
c->SetORT ( ORT_LOOP, ORT_AFTER );
}
}
}