mirror of https://github.com/AxioDL/metaforce.git
63 lines
2.2 KiB
C++
63 lines
2.2 KiB
C++
#include "CScriptRandomRelay.hpp"
|
|
#include "CStateManager.hpp"
|
|
#include "TCastTo.hpp"
|
|
|
|
namespace urde {
|
|
CScriptRandomRelay::CScriptRandomRelay(TUniqueId uid, std::string_view name, const CEntityInfo& info, s32 sendSetSize,
|
|
s32 sendSetVariance, bool percentSize, bool active)
|
|
: CEntity(uid, info, active, name)
|
|
, x34_sendSetSize((percentSize && sendSetSize > 100) ? 100 : sendSetSize)
|
|
, x38_sendSetVariance(sendSetVariance)
|
|
, x3c_percentSize(percentSize) {}
|
|
|
|
void CScriptRandomRelay::Accept(IVisitor& visitor) { visitor.Visit(this); }
|
|
|
|
void CScriptRandomRelay::AcceptScriptMsg(EScriptObjectMessage msg, TUniqueId objId, CStateManager& stateMgr) {
|
|
CEntity::AcceptScriptMsg(msg, objId, stateMgr);
|
|
if (msg == EScriptObjectMessage::SetToZero) {
|
|
if (!x30_24_active)
|
|
return;
|
|
SendLocalScriptMsgs(EScriptObjectState::Zero, stateMgr);
|
|
}
|
|
}
|
|
|
|
void CScriptRandomRelay::SendLocalScriptMsgs(EScriptObjectState state, CStateManager& stateMgr) {
|
|
if (state != EScriptObjectState::Zero) {
|
|
SendScriptMsgs(state, stateMgr, EScriptObjectMessage::None);
|
|
return;
|
|
}
|
|
|
|
std::vector<std::pair<CEntity*, EScriptObjectMessage>> objs;
|
|
objs.reserve(10);
|
|
for (const SConnection& conn : x20_conns) {
|
|
auto list = stateMgr.GetIdListForScript(conn.x8_objId);
|
|
for (auto it = list.first; it != list.second; ++it) {
|
|
CEntity* ent = stateMgr.ObjectById(it->second);
|
|
if (ent && ent->GetActive())
|
|
objs.emplace_back(ent, conn.x4_msg);
|
|
}
|
|
}
|
|
|
|
s32 targetSetSize = x34_sendSetSize;
|
|
if (x3c_percentSize)
|
|
targetSetSize = s32(0.5f + (float(x34_sendSetSize * objs.size()) / 100.f));
|
|
targetSetSize += s32(float(x38_sendSetVariance) * 2.f * stateMgr.GetActiveRandom()->Float()) - x38_sendSetVariance;
|
|
targetSetSize = zeus::clamp(0, targetSetSize, 64);
|
|
|
|
while (objs.size() > targetSetSize) {
|
|
s32 randomRemoveIdx = s32(stateMgr.GetActiveRandom()->Float() * float(objs.size()) * 0.99f);
|
|
auto it = objs.begin();
|
|
for (s32 i = 0; i < randomRemoveIdx; ++i) {
|
|
++it;
|
|
if (it == objs.end())
|
|
break;
|
|
}
|
|
if (it != objs.end())
|
|
objs.erase(it);
|
|
}
|
|
|
|
for (const auto& o : objs)
|
|
stateMgr.SendScriptMsg(o.first, GetUniqueId(), o.second);
|
|
}
|
|
} // namespace urde
|