mirror of https://github.com/AxioDL/metaforce.git
276 lines
8.6 KiB
C++
276 lines
8.6 KiB
C++
#include "CPASAnimState.hpp"
|
|
#include "CPASAnimParmData.hpp"
|
|
#include "CRandom16.hpp"
|
|
#include "zeus/Math.hpp"
|
|
#include <cmath>
|
|
#include <cfloat>
|
|
|
|
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));
|
|
}
|
|
}
|
|
|
|
CPASAnimState::CPASAnimState(int stateId)
|
|
: x0_id(stateId)
|
|
{}
|
|
|
|
CPASAnimParm CPASAnimState::GetAnimParmData(s32 animId, u32 parmIdx) const
|
|
{
|
|
auto search = rstl::binary_find(x14_anims.begin(), x14_anims.end(), animId,
|
|
[](const CPASAnimInfo& item) {return item.GetAnimId();});
|
|
if (search == x14_anims.end())
|
|
return CPASAnimParm::NoParameter();
|
|
|
|
CPASParmInfo parm = x4_parms.at(parmIdx);
|
|
return (*search).GetAnimParmData(parmIdx, 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 ignoreAnim) const
|
|
{
|
|
const_cast<std::vector<s32>*>(&x24_selectionCache)->clear();
|
|
float weight = -1.f;
|
|
|
|
for (const CPASAnimInfo& info : x14_anims)
|
|
{
|
|
if (info.GetAnimId() == ignoreAnim)
|
|
continue;
|
|
|
|
float calcWeight = 1.f;
|
|
if (x4_parms.size() > 0)
|
|
calcWeight = 0.f;
|
|
|
|
u32 unweightedCount = 0;
|
|
|
|
for (u32 i=0 ; i<x4_parms.size() ; ++i)
|
|
{
|
|
CPASAnimParm::UParmValue val = info.GetAnimParmValue(i);
|
|
const CPASParmInfo& parmInfo = x4_parms[i];
|
|
float parmWeight = parmInfo.GetParameterWeight();
|
|
|
|
float computedWeight = 0.f;
|
|
switch (parmInfo.GetWeightFunction())
|
|
{
|
|
case CPASParmInfo::EWeightFunction::AngularPercent:
|
|
computedWeight = ComputeAngularPercentErrorWeight(i, parms[i], val);
|
|
break;
|
|
case CPASParmInfo::EWeightFunction::ExactMatch:
|
|
computedWeight = ComputeExactMatchWeight(i, parms[i], val);
|
|
break;
|
|
case CPASParmInfo::EWeightFunction::PercentError:
|
|
computedWeight = ComputePercentErrorWeight(i, parms[i], val);
|
|
break;
|
|
case CPASParmInfo::EWeightFunction::NoWeight:
|
|
unweightedCount++;
|
|
break;
|
|
default: break;
|
|
}
|
|
|
|
calcWeight += parmWeight * computedWeight;
|
|
}
|
|
|
|
if (unweightedCount == x4_parms.size())
|
|
calcWeight = 1.0f;
|
|
|
|
if (calcWeight > weight)
|
|
{
|
|
const_cast<std::vector<s32>*>(&x24_selectionCache)->clear();
|
|
const_cast<std::vector<s32>*>(&x24_selectionCache)->push_back(info.GetAnimId());
|
|
weight = calcWeight;
|
|
}
|
|
else if (weight == calcWeight)
|
|
{
|
|
const_cast<std::vector<s32>*>(&x24_selectionCache)->push_back(info.GetAnimId());
|
|
weight = calcWeight;
|
|
}
|
|
}
|
|
|
|
return {weight, PickRandomAnimation(rand)};
|
|
}
|
|
|
|
float CPASAnimState::ComputeExactMatchWeight(u32, const CPASAnimParm& parm, CPASAnimParm::UParmValue parmVal) const
|
|
{
|
|
switch (parm.GetParameterType())
|
|
{
|
|
case CPASAnimParm::EParmType::Int32:
|
|
return (parm.GetInt32Value() == parmVal.m_int ? 1.f : 0.f);
|
|
case CPASAnimParm::EParmType::UInt32:
|
|
return (parm.GetUint32Value() == parmVal.m_uint ? 1.f : 0.f);
|
|
case CPASAnimParm::EParmType::Float:
|
|
return ((parmVal.m_float - parm.GetReal32Value()) < FLT_EPSILON ? 1.f : 0.f);
|
|
case CPASAnimParm::EParmType::Bool:
|
|
return (parm.GetBoolValue() == parmVal.m_bool ? 1.f : 0.f);
|
|
case CPASAnimParm::EParmType::Enum:
|
|
return (parm.GetEnumValue() == parmVal.m_int ? 1.f : 0.f);
|
|
default: break;
|
|
}
|
|
|
|
return 0.f;
|
|
}
|
|
|
|
float CPASAnimState::ComputePercentErrorWeight(u32 idx, const CPASAnimParm& parm, CPASAnimParm::UParmValue parmVal) const
|
|
{
|
|
float range = 0.f;
|
|
float val = 0.f;
|
|
|
|
switch (parm.GetParameterType())
|
|
{
|
|
case CPASAnimParm::EParmType::Int32:
|
|
{
|
|
const CPASParmInfo& info = x4_parms[idx];
|
|
range = info.GetWeightMaxValue().m_int - info.GetWeightMinValue().m_int;
|
|
val = std::fabs(parm.GetInt32Value() - parmVal.m_int);
|
|
break;
|
|
}
|
|
case CPASAnimParm::EParmType::UInt32:
|
|
{
|
|
const CPASParmInfo& info = x4_parms[idx];
|
|
range = info.GetWeightMaxValue().m_uint - info.GetWeightMinValue().m_uint;
|
|
val = std::fabs(int(parmVal.m_uint) - int(parm.GetUint32Value()));
|
|
break;
|
|
}
|
|
case CPASAnimParm::EParmType::Float:
|
|
{
|
|
const CPASParmInfo& info = x4_parms[idx];
|
|
range = info.GetWeightMaxValue().m_float - info.GetWeightMinValue().m_float;
|
|
val = std::fabs(parm.GetReal32Value() - parmVal.m_float);
|
|
break;
|
|
}
|
|
case CPASAnimParm::EParmType::Bool:
|
|
{
|
|
val = parm.GetBoolValue() == parmVal.m_bool ? 0.f : 1.f;
|
|
break;
|
|
}
|
|
case CPASAnimParm::EParmType::Enum:
|
|
{
|
|
const CPASParmInfo& info = x4_parms[idx];
|
|
range = info.GetWeightMaxValue().m_int - info.GetWeightMinValue().m_int;
|
|
val = std::fabs(parm.GetEnumValue() - parmVal.m_int);
|
|
break;
|
|
}
|
|
default: break;
|
|
}
|
|
|
|
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;
|
|
|
|
switch (parm.GetParameterType())
|
|
{
|
|
case CPASAnimParm::EParmType::Int32:
|
|
{
|
|
const CPASParmInfo& info = x4_parms[idx];
|
|
range = info.GetWeightMaxValue().m_int - info.GetWeightMinValue().m_int;
|
|
val = std::fabs(parmVal.m_int - parm.GetInt32Value());
|
|
break;
|
|
}
|
|
case CPASAnimParm::EParmType::UInt32:
|
|
{
|
|
const CPASParmInfo& info = x4_parms[idx];
|
|
range = info.GetWeightMaxValue().m_uint - info.GetWeightMinValue().m_uint;
|
|
val = std::fabs(int(parmVal.m_uint) - int(parm.GetUint32Value()));
|
|
break;
|
|
}
|
|
case CPASAnimParm::EParmType::Float:
|
|
{
|
|
const CPASParmInfo& info = x4_parms[idx];
|
|
range = info.GetWeightMaxValue().m_float - info.GetWeightMinValue().m_float;
|
|
val = std::fabs(parm.GetReal32Value() - parmVal.m_float);
|
|
break;
|
|
}
|
|
case CPASAnimParm::EParmType::Bool:
|
|
{
|
|
val = parm.GetBoolValue() == parmVal.m_bool ? 0.f : 1.f;
|
|
break;
|
|
}
|
|
case CPASAnimParm::EParmType::Enum:
|
|
{
|
|
const CPASParmInfo& info = x4_parms[idx];
|
|
range = info.GetWeightMaxValue().m_int - info.GetWeightMinValue().m_int + 1;
|
|
val = std::fabs(parm.GetEnumValue() - parmVal.m_int);
|
|
break;
|
|
}
|
|
default: break;
|
|
}
|
|
|
|
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);
|
|
}
|
|
}
|