metaforce/Runtime/Character/CPASAnimState.cpp

250 lines
8.7 KiB
C++

#include "CPASAnimState.hpp"
#include "CPASAnimParmData.hpp"
#include "CRandom16.hpp"
#include "zeus/Math.hpp"
#include <math.h>
#include <float.h>
namespace urde
{
CPASAnimState::CPASAnimState(CInputStream& in)
{
x0_id = in.readUint32Big();
u32 parmCount = in.readUint32Big();
u32 animCount = in.readUint32Big();
x4_parms.reserve(parmCount);
x14_anims.reserve(animCount);
x24_selectionCache.reserve(animCount);
for (u32 i=0 ; i<parmCount ; ++i)
x4_parms.emplace_back(in);
for (u32 i=0 ; i<animCount ; ++i)
{
s32 id = in.readUint32Big();
rstl::reserved_vector<CPASAnimParm::UParmValue, 8> parms;
for (const CPASParmInfo& parm : x4_parms)
{
CPASAnimParm::UParmValue val = {};
switch (parm.GetParameterType())
{
case CPASAnimParm::EParmType::Int32:
val.m_int = in.readInt32Big();
break;
case CPASAnimParm::EParmType::UInt32:
val.m_uint = in.readUint32Big();
break;
case CPASAnimParm::EParmType::Float:
val.m_float = in.readFloatBig();
break;
case CPASAnimParm::EParmType::Bool:
val.m_bool = in.readBool();
break;
case CPASAnimParm::EParmType::Enum:
val.m_int = in.readInt32Big();
break;
default: break;
}
parms.push_back(val);
}
auto search = std::lower_bound(x14_anims.begin(), x14_anims.end(), id,
[](const CPASAnimInfo& item, const u32& testId) -> bool {return item.GetAnimId() < testId;});
x14_anims.emplace(search, id, std::move(parms));
}
}
CPASAnimParm CPASAnimState::GetAnimParmData(s32 idx, u32 id) const
{
auto search = std::lower_bound(x14_anims.begin(), x14_anims.end(), id,
[](const CPASAnimInfo& item, const u32& testId) ->
bool {return item.GetAnimId() < testId;});
if (search == x14_anims.end())
return CPASAnimParm::NoParameter();
CPASParmInfo parm = x4_parms.at(idx);
return (*search).GetAnimParmData(idx, parm.GetParameterType());
}
s32 CPASAnimState::PickRandomAnimation(CRandom16& rand) const
{
if (x24_selectionCache.size() == 1)
return x24_selectionCache[0];
if (x24_selectionCache.size() > 1)
{
u32 idx = u32(floor(rand.Float()));
if (idx == x24_selectionCache.size())
idx--;
return x24_selectionCache[idx];
}
return -1;
}
std::pair<float, s32> CPASAnimState::FindBestAnimation(const rstl::reserved_vector<CPASAnimParm, 8>& parms,
CRandom16& rand, s32 anim) const
{
const_cast<std::vector<s32>*>(&x24_selectionCache)->clear();
float weight = -1.f;
if (x14_anims.size() != 0)
{
for (const CPASAnimInfo& info : x14_anims)
{
float calcWeight = 1.f;
if (x4_parms.size() > 0)
calcWeight = 0.f;
u32 r23 = 0 ;
u32 i = 0;
for (; i<x4_parms.size() ; ++i)
{
CPASAnimParm::UParmValue val = info.GetAnimParmValue(i);
const CPASParmInfo& parmInfo = x4_parms[i];
float parmWeight = parmInfo.GetParameterWeight();
float computedWeight;
if (parmInfo.GetWeightFunction() == CPASParmInfo::EWeightFunction::AngularPercent)
computedWeight = ComputeAngularPercentErrorWeight(i, parms[i], val);
else if (parmInfo.GetWeightFunction() == CPASParmInfo::EWeightFunction::ExactMatch)
computedWeight = ComputeExactMatchWeight(i, parms[i], val);
else if (parmInfo.GetWeightFunction() == CPASParmInfo::EWeightFunction::PercentError)
computedWeight = ComputePercentErrorWeight(i, parms[i], val);
else if (parmInfo.GetWeightFunction() == CPASParmInfo::EWeightFunction::Three)
r23++;
calcWeight = parmWeight * calcWeight + computedWeight;
}
if (r23 == x4_parms.size())
calcWeight = 1.0f;
if (calcWeight < weight)
continue;
auto search = std::find(x24_selectionCache.cbegin(), x24_selectionCache.cend(),
info.GetAnimId());
if (search == x24_selectionCache.cend())
const_cast<std::vector<s32>*>(&x24_selectionCache)->push_back(info.GetAnimId());
}
}
return {weight, PickRandomAnimation(rand)};
}
float CPASAnimState::ComputeExactMatchWeight(u32, const CPASAnimParm& parm, CPASAnimParm::UParmValue parmVal) const
{
if (parm.GetParameterType() == CPASAnimParm::EParmType::Int32)
{
return (parm.GetInt32Value() == parmVal.m_int ? 1.f : 0.f);
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::UInt32)
{
return (parm.GetUint32Value() == parmVal.m_uint ? 1.f : 0.f);
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Float)
{
return ((parmVal.m_float - parm.GetReal32Value()) < FLT_EPSILON ? 1.f : 0.f);
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Bool)
{
return (parm.GetBoolValue() == parmVal.m_bool ? 1.f : 0.f);
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Enum)
{
return (parm.GetEnumValue() == parmVal.m_int ? 1.f : 0.f);
}
return 0.f;
}
float CPASAnimState::ComputePercentErrorWeight(u32 idx, const CPASAnimParm& parm, CPASAnimParm::UParmValue parmVal) const
{
float range = 0.f;
float val = 0.f;
if (parm.GetParameterType() == CPASAnimParm::EParmType::Int32)
{
const CPASParmInfo& info = x4_parms[idx];
range = float(info.GetWeightMinValue().m_int - info.GetWeightMaxValue().m_int);
val = float(fabs(parmVal.m_int - parm.GetInt32Value()));
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::UInt32)
{
const CPASParmInfo& info = x4_parms[idx];
range = float(info.GetWeightMinValue().m_uint - info.GetWeightMaxValue().m_uint);
val = float(fabs(parmVal.m_uint - parm.GetUint32Value()));
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Float)
{
const CPASParmInfo& info = x4_parms[idx];
range = float(info.GetWeightMinValue().m_float - info.GetWeightMaxValue().m_float);
val = float(fabs(parmVal.m_float - parm.GetReal32Value()));
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Bool)
{
val = parm.GetBoolValue() ? 1.f : 0.f;
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Enum)
{
const CPASParmInfo& info = x4_parms[idx];
range = float(info.GetWeightMinValue().m_float - info.GetWeightMaxValue().m_float);
val = float(fabs(parmVal.m_float - parm.GetReal32Value()));
}
if (range > FLT_EPSILON)
return (val / range) - 1.0f;
return (val < FLT_EPSILON ? 1.f : 0.f);
}
float CPASAnimState::ComputeAngularPercentErrorWeight(u32 idx, const CPASAnimParm& parm, CPASAnimParm::UParmValue parmVal) const
{
float range = 0.f;
float val = 0.f;
if (parm.GetParameterType() == CPASAnimParm::EParmType::Int32)
{
const CPASParmInfo& info = x4_parms[idx];
range = float(info.GetWeightMinValue().m_int - info.GetWeightMaxValue().m_int);
val = float(fabs(parmVal.m_int - parm.GetInt32Value()));
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::UInt32)
{
const CPASParmInfo& info = x4_parms[idx];
range = float(info.GetWeightMinValue().m_uint - info.GetWeightMaxValue().m_uint);
val = float(fabs(parmVal.m_uint - parm.GetUint32Value()));
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Float)
{
const CPASParmInfo& info = x4_parms[idx];
range = float(info.GetWeightMinValue().m_float - info.GetWeightMaxValue().m_float);
val = float(fabs(parmVal.m_float - parm.GetReal32Value()));
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Bool)
{
val = parm.GetBoolValue() ? 1.f : 0.f;
}
else if (parm.GetParameterType() == CPASAnimParm::EParmType::Enum)
{
const CPASParmInfo& info = x4_parms[idx];
range = float(info.GetWeightMinValue().m_float - info.GetWeightMaxValue().m_float);
val = float(fabs(parmVal.m_float - parm.GetReal32Value()));
}
if (range > FLT_EPSILON)
{
float mid = 0.5f * range;
float offset = 1.f - ((val > mid ? range - val : val) / mid);
return zeus::clamp(0.f, offset, 1.f);
}
return (val >= FLT_EPSILON ? 0.f : 1.f);
}
}