64 lines
1.5 KiB
C++
64 lines
1.5 KiB
C++
#include "precompiled.h"
|
|
|
|
#include "Unit.h"
|
|
#include "Model.h"
|
|
#include "ObjectEntry.h"
|
|
|
|
CUnit::~CUnit() {
|
|
delete m_Model;
|
|
}
|
|
|
|
void CUnit::ShowAmmunition()
|
|
{
|
|
if( !m_Object->m_AmmunitionModel || !m_Object->m_AmmunitionPoint )
|
|
return;
|
|
m_Model->AddProp( m_Object->m_AmmunitionPoint, m_Object->m_AmmunitionModel->Clone() );
|
|
}
|
|
|
|
void CUnit::HideAmmunition()
|
|
{
|
|
if( !m_Object->m_AmmunitionModel || !m_Object->m_AmmunitionPoint )
|
|
return;
|
|
// Find out what the usual prop is:
|
|
std::vector<CModel::Prop>& props = m_Object->m_Model->GetProps();
|
|
std::vector<CModel::Prop>::iterator it;
|
|
for( it = props.begin(); it != props.end(); ++it )
|
|
if( it->m_Point == m_Object->m_AmmunitionPoint )
|
|
{
|
|
m_Model->AddProp( m_Object->m_AmmunitionPoint, it->m_Model->Clone() );
|
|
return;
|
|
}
|
|
// No usual prop.
|
|
m_Model->RemoveProp( m_Object->m_AmmunitionPoint );
|
|
}
|
|
|
|
bool CUnit::SetRandomAnimation(const CStr& name, bool once)
|
|
{
|
|
CSkeletonAnim* anim = GetRandomAnimation(name);
|
|
if (anim)
|
|
{
|
|
m_Model->SetAnimation(anim, once);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
// TODO - report an error?
|
|
return false;
|
|
}
|
|
}
|
|
|
|
CSkeletonAnim* CUnit::GetRandomAnimation(const CStr& name)
|
|
{
|
|
CSkeletonAnim* anim = m_Object->GetRandomAnimation(name);
|
|
// Fall back to 'idle', if no matching animation is found
|
|
if (anim == NULL && name != "idle")
|
|
anim = m_Object->GetRandomAnimation("idle");
|
|
|
|
return anim;
|
|
}
|
|
|
|
bool CUnit::IsPlayingAnimation(const CStr& name)
|
|
{
|
|
return (m_Model->GetAnimation()->m_Name == name);
|
|
}
|