PrimeDecomp
/
prime
mirror of https://github.com/PrimeDecomp/prime.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
prime/src/MetroidPrime/ScriptObjects/CScriptMazeNode.cpp

523 lines
18 KiB
C++
Raw Blame History

#include "MetroidPrime/ScriptObjects/CScriptMazeNode.hpp"
#include "Kyoto/CResFactory.hpp"
#include "MetroidPrime/CActorParameters.hpp"
#include "MetroidPrime/CObjectList.hpp"
#include "rstl/algorithm.hpp"
uint CScriptMazeNode::sMazeSeeds[300];
CMazeState::CMazeState(int enterCol, int enterRow, int targetCol, int targetRow)
: x0_rand(0)
, x84_enterCol(enterCol)
, x88_enterRow(enterRow)
, x8c_targetCol(targetCol)
, x90_targetRow(targetRow)
, x94_24_initialized(false) {}
void CMazeState::Reset(int seed) {
int numCells = NUM_MAZE_CELLS - 1;
int cellIdx = 0;
x0_rand = CRandom16(seed);
x94_24_initialized = false;
for (int i = 0; i < NUM_MAZE_CELLS; i++) {
x4_cells[i] = SMazeCell();
}
ESide sides[4];
while (numCells != 0) {
int acc = 0;
if (cellIdx - skMazeCols > 0 && GetCell(cellIdx - skMazeCols).IsClosed()) {
sides[acc++] = kS_Top;
}
if (cellIdx < NUM_MAZE_CELLS - 2 && (cellIdx + 1) % skMazeCols != 0 && GetCell(cellIdx + 1).IsClosed()) {
sides[acc++] = kS_Right;
}
if (cellIdx + skMazeCols <= NUM_MAZE_CELLS - 1 && GetCell(cellIdx + skMazeCols).IsClosed()) {
sides[acc++] = kS_Bottom;
}
if (cellIdx > 0 && cellIdx % skMazeCols != 0 && GetCell(cellIdx - 1).IsClosed()) {
sides[acc++] = kS_Left;
}
if (acc != 0) {
numCells--;
ESide side = sides[x0_rand.Next() % static_cast< uint >(acc)];
switch (side) {
case kS_Top:
GetCell(cellIdx).x0_24_openTop = true;
GetCell(cellIdx - skMazeCols).x0_26_openBottom = true;
cellIdx -= skMazeCols;
break;
case kS_Right:
GetCell(cellIdx).x0_25_openRight = true;
GetCell(cellIdx + 1).x0_27_openLeft = true;
cellIdx++;
break;
case kS_Bottom:
GetCell(cellIdx).x0_26_openBottom = true;
GetCell(cellIdx + skMazeCols).x0_24_openTop = true;
cellIdx += skMazeCols;
break;
case kS_Left:
GetCell(cellIdx).x0_27_openLeft = true;
GetCell(cellIdx - 1).x0_25_openRight = true;
cellIdx--;
break;
}
continue;
}
do {
cellIdx++;
if (cellIdx > NUM_MAZE_CELLS - 1) {
cellIdx = 0;
}
} while (GetCell(cellIdx).IsClosed());
}
}
SMazeCell& CMazeState::GetCell(uint col, uint row) { return x4_cells[col + row * skMazeCols]; }
const SMazeCell& CMazeState::GetCell(uint col, uint row) const { return x4_cells[col + row * skMazeCols]; }
static inline int GetRandom(CRandom16& rand, int offset) {
int tmp = rand.Next();
return tmp - ((tmp / 5) * 5) + offset;
}
void CMazeState::GenerateObstacles() {
if (!x94_24_initialized) {
Initialize();
}
int gate1Idx = GetRandom(x0_rand, 9);
int gate2Idx = GetRandom(x0_rand, 21);
int gate3Idx = GetRandom(x0_rand, 33);
int puddle1Idx = GetRandom(x0_rand, 13);
int puddle2Idx = GetRandom(x0_rand, 29);
int prevCol = x84_enterCol;
int prevRow = x88_enterRow;
ESide side = kS_Invalid;
int idx = 0;
int col = prevCol;
int row = prevRow;
while (col != x8c_targetCol || row != x90_targetRow) {
if (idx == gate1Idx || idx == gate2Idx || idx == gate3Idx) {
switch (side) {
case kS_Top:
GetCellInline(col, row).x0_30_gateBottom = true;
GetCellInline(prevCol, prevRow).x0_28_gateTop = true;
break;
case kS_Right:
GetCellInline(col, row).x0_31_gateLeft = true;
GetCellInline(prevCol, prevRow).x0_29_gateRight = true;
break;
case kS_Bottom:
GetCellInline(col, row).x0_28_gateTop = true;
GetCellInline(prevCol, prevRow).x0_30_gateBottom = true;
break;
case kS_Left:
GetCellInline(col, row).x0_29_gateRight = true;
GetCellInline(prevCol, prevRow).x0_31_gateLeft = true;
break;
}
}
int curCol = col;
int curRow = row;
if (row > 0 && side != kS_Bottom && GetCellInline(col, row).x0_24_openTop && GetCellInline(col, row - 1).x1_25_onPath) {
side = kS_Top;
row--;
} else if (row < skMazeRows - 1 && side != kS_Top && GetCellInline(col, row).x0_26_openBottom &&
GetCellInline(col, row + 1).x1_25_onPath) {
side = kS_Bottom;
row++;
} else if (col > 0 && side != kS_Right && GetCellInline(col, row).x0_27_openLeft && GetCellInline(col - 1, row).x1_25_onPath) {
side = kS_Left;
col--;
} else if (col < skMazeCols - 1 && side != kS_Left && GetCellInline(col, row).x0_25_openRight &&
GetCellInline(col + 1, row).x1_25_onPath) {
side = kS_Right;
col++;
} else {
return;
}
#define GetCellInline(c, r) x4_cells[c + r * skMazeCols]
if (idx == puddle1Idx || idx == puddle2Idx) {
if (curCol == 0 || curRow == 0 || curCol == skMazeCols - 1 || curRow == skMazeRows - 1) {
if (idx == puddle1Idx) {
puddle1Idx++;
} else {
puddle2Idx++;
}
} else {
GetCellInline(curCol, curRow).x1_24_puddle = true;
switch (side) {
case kS_Top:
GetCellInline(col, row).x0_26_openBottom = false;
GetCellInline(curCol, curRow).x0_24_openTop = false;
break;
case kS_Right:
GetCellInline(col, row).x0_27_openLeft = false;
GetCellInline(curCol, curRow).x0_25_openRight = false;
break;
case kS_Bottom:
GetCellInline(col, row).x0_24_openTop = false;
GetCellInline(curCol, curRow).x0_26_openBottom = false;
break;
case kS_Left:
GetCellInline(col, row).x0_25_openRight = false;
GetCellInline(curCol, curRow).x0_27_openLeft = false;
break;
}
}
}
#undef GetCellInline
idx++;
prevCol = curCol;
prevRow = curRow;
}
}
void CMazeState::Initialize() {
int path[NUM_MAZE_CELLS];
int pathIdx = 0;
int targetRow = x90_targetRow;
int targetCol = x8c_targetCol;
int cellIdx = x84_enterCol + x88_enterRow * skMazeCols;
int max = targetCol + targetRow * skMazeCols;
path[pathIdx++] = cellIdx;
x4_cells[cellIdx].x1_26_checked = true;
while (cellIdx != max) {
if (x4_cells[cellIdx].x0_24_openTop && !x4_cells[cellIdx - skMazeCols].x1_26_checked) {
path[pathIdx++] = cellIdx - skMazeCols;
}
if (x4_cells[cellIdx].x0_25_openRight && !x4_cells[cellIdx + 1].x1_26_checked) {
path[pathIdx++] = cellIdx + 1;
}
if (x4_cells[cellIdx].x0_26_openBottom && !x4_cells[cellIdx + skMazeCols].x1_26_checked) {
path[pathIdx++] = cellIdx + skMazeCols;
}
if (x4_cells[cellIdx].x0_27_openLeft && !x4_cells[cellIdx - 1].x1_26_checked) {
path[pathIdx++] = cellIdx - 1;
}
if (cellIdx == path[pathIdx - 1]) {
pathIdx--;
}
cellIdx = path[pathIdx - 1];
x4_cells[cellIdx].x1_26_checked = true;
}
while (pathIdx--) {
cellIdx = path[pathIdx];
if (x4_cells[cellIdx].x1_26_checked) {
x4_cells[cellIdx].x1_25_onPath = true;
}
}
x94_24_initialized = true;
}
CScriptMazeNode::CScriptMazeNode(TUniqueId uid, const rstl::string& name, const CEntityInfo& info, const CTransform4f& xf, bool active,
int col, int row, int side, const CVector3f& actorPos, const CVector3f& triggerPos,
const CVector3f& effectPos)
: CActor(uid, active, name, info, xf, CModelData::CModelDataNull(), CMaterialList(), CActorParameters::None(), kInvalidUniqueId)
, xe8_col(col)
, xec_row(row)
, xf0_side(static_cast< ESide >(side))
, xf4_gateEffectId(kInvalidUniqueId)
, xf8_msgTimer(1.f)
, xfc_actorId(kInvalidUniqueId)
, x100_actorPos(actorPos)
, x10c_triggerId(kInvalidUniqueId)
, x110_triggerPos(triggerPos)
, x11c_effectId(kInvalidUniqueId)
, x120_effectPos(effectPos)
, x13c_24_hasPuddle(false)
, x13c_25_hasGate(false)
, x13c_26_gateActive(true) {}
void CScriptMazeNode::Accept(IVisitor& visitor) { visitor.Visit(*this); }
static inline TUniqueId GenerateObject(CStateManager& mgr, const TEditorId& eid) {
bool wasGeneratingObject = mgr.IsGeneratingObject();
mgr.SetIsGeneratingObject(true);
TUniqueId objUid = mgr.GenerateObject(eid).second;
mgr.SetIsGeneratingObject(wasGeneratingObject);
return objUid;
}
// TODO non-matching
// https://decomp.me/scratch/IvHBz
void CScriptMazeNode::GenerateObjects(CStateManager& mgr) {
rstl::vector< SConnection >::const_iterator conn = GetConnectionList().begin();
for (; conn != GetConnectionList().end(); ++conn) {
if (conn->x0_state != kSS_MaxReached || conn->x4_msg != kSM_Activate) {
continue;
}
CEntity* ent = mgr.ObjectById(mgr.GetIdForScript(conn->x8_objId));
CScriptEffect* scriptEffect = TCastToPtr< CScriptEffect >(ent);
CScriptActor* scriptActor = TCastToPtr< CScriptActor >(ent);
CScriptTrigger* scriptTrigger = TCastToPtr< CScriptTrigger >(ent);
if (!scriptEffect && !scriptActor && !scriptTrigger) {
continue;
}
if (scriptEffect && x13c_25_hasGate) {
continue;
}
// TUniqueId objUid = GenerateObject(mgr, conn->x8_objId);
TUniqueId objUid = GenerateObject(mgr, conn->x8_objId);
// bool wasGeneratingObject = mgr.IsGeneratingObject();
// mgr.SetIsGeneratingObject(true);
// TUniqueId objUid = mgr.GenerateObject(conn->x8_objId).second;
// mgr.SetIsGeneratingObject(wasGeneratingObject);
if (CActor* actor = static_cast< CActor* >(mgr.ObjectById(objUid))) {
mgr.SendScriptMsg(actor, GetUniqueId(), kSM_Activate);
if (scriptEffect) {
actor->SetTranslation(GetTranslation() + x120_effectPos);
x11c_effectId = objUid;
}
if (scriptActor) {
actor->SetTranslation(GetTranslation() + x100_actorPos);
xfc_actorId = objUid;
}
if (scriptTrigger) {
actor->SetTranslation(GetTranslation() + x110_triggerPos);
x10c_triggerId = objUid;
}
}
}
}
void CScriptMazeNode::Reset(CStateManager& mgr) {
mgr.FreeScriptObject(x11c_effectId);
mgr.FreeScriptObject(xfc_actorId);
mgr.FreeScriptObject(x10c_triggerId);
mgr.FreeScriptObject(xf4_gateEffectId);
x11c_effectId = xfc_actorId = x10c_triggerId = xf4_gateEffectId = kInvalidUniqueId;
}
void CScriptMazeNode::SendScriptMsgs(CStateManager& mgr, EScriptObjectMessage msg) {
SendScriptMsg(mgr, mgr.ObjectById(x11c_effectId), GetUniqueId(), msg);
SendScriptMsg(mgr, mgr.ObjectById(xfc_actorId), GetUniqueId(), msg);
SendScriptMsg(mgr, mgr.ObjectById(x10c_triggerId), GetUniqueId(), msg);
SendScriptMsg(mgr, mgr.ObjectById(xf4_gateEffectId), GetUniqueId(), msg);
}
template < typename Iter, typename T >
static inline Iter contains(Iter it, Iter end, const T& value) {
for (; it != end && *it != value; ++it) {
}
return it;
}
template < typename T >
static inline bool contains(const rstl::vector< T >& vec, typename rstl::vector< T >::const_iterator end, const T& value) {
return rstl::find< rstl::vector< T >::const_iterator, T >(vec.begin(), end, value) != end;
}
void CScriptMazeNode::AcceptScriptMsg(EScriptObjectMessage msg, TUniqueId uid, CStateManager& mgr) {
if (GetActive()) {
switch (msg) {
case kSM_InitializedInArea: {
if (mgr.CurrentMaze() == nullptr) {
rstl::single_ptr< CMazeState > maze = new CMazeState(skEnterCol, skEnterRow, skTargetCol, skTargetRow);
maze->Reset(sMazeSeeds[mgr.GetActiveRandom()->Next() % 300]);
maze->Initialize();
maze->GenerateObstacles();
mgr.SetCurrentMaze(maze.release());
}
break;
}
case kSM_Action: {
bool shouldGenObjs = false;
if (const CMazeState* maze = mgr.CurrentMaze()) {
const SMazeCell& cell = maze->GetCell(xe8_col, xec_row);
if (xf0_side == CMazeState::kS_Top && cell.x0_24_openTop) {
if (cell.x0_28_gateTop) {
shouldGenObjs = true;
x13c_25_hasGate = true;
}
} else if (xf0_side == CMazeState::kS_Right && cell.x0_25_openRight) {
if (cell.x0_29_gateRight) {
shouldGenObjs = true;
x13c_25_hasGate = true;
}
} else {
shouldGenObjs = true;
}
if (shouldGenObjs) {
GenerateObjects(mgr);
}
if (xf0_side == CMazeState::kS_Right && cell.x1_24_puddle) {
x13c_24_hasPuddle = true;
}
if (x13c_25_hasGate) {
#ifndef NON_MATCHING
// Unused
CTransform4f xf = GetTransform();
#endif
rstl::vector< SConnection >::const_iterator conn = GetConnectionList().begin();
for (; conn != GetConnectionList().end(); ++conn) {
if (conn->x0_state != kSS_Modify || conn->x4_msg != kSM_Activate) {
continue;
}
// TUniqueId genObj = GenerateObject(mgr, conn->x8_objId);
bool wasGeneratingObject = mgr.IsGeneratingObject();
mgr.SetIsGeneratingObject(true);
TUniqueId genObj = mgr.GenerateObject(conn->x8_objId).second;
mgr.SetIsGeneratingObject(wasGeneratingObject);
xf4_gateEffectId = genObj;
if (CActor* actor = TCastToPtr< CActor >(mgr.ObjectById(genObj))) {
actor->SetTranslation(GetTranslation() + x120_effectPos);
mgr.SendScriptMsg(actor, GetUniqueId(), kSM_Activate);
}
break;
}
}
if (x13c_24_hasPuddle) {
int count = 0;
rstl::vector< SConnection >::const_iterator conn = GetConnectionList().begin();
for (; conn != GetConnectionList().end(); ++conn) {
if ((conn->x0_state == kSS_Closed || conn->x0_state == kSS_DeactivateState) && conn->x4_msg == kSM_Activate) {
count++;
}
}
x12c_puddleObjectIds.reserve(count);
conn = GetConnectionList().begin();
for (; conn != GetConnectionList().end(); ++conn) {
if ((conn->x0_state == kSS_Closed || conn->x0_state == kSS_DeactivateState) && conn->x4_msg == kSM_Activate) {
// TUniqueId genObj = GenerateObject(mgr, conn->x8_objId);
bool wasGeneratingObject = mgr.IsGeneratingObject();
mgr.SetIsGeneratingObject(true);
TUniqueId genObj = mgr.GenerateObject(conn->x8_objId).second;
mgr.SetIsGeneratingObject(wasGeneratingObject);
x12c_puddleObjectIds.push_back(genObj);
if (CActor* actor = TCastToPtr< CActor >(mgr.ObjectById(genObj))) {
actor->SetTransform(GetTransform());
if (conn->x0_state == kSS_Closed) {
mgr.SendScriptMsg(actor, GetUniqueId(), kSM_Activate);
}
}
}
}
}
}
break;
}
case kSM_SetToZero: {
CMazeState* maze = mgr.CurrentMaze();
if (x13c_24_hasPuddle && maze != nullptr) {
rstl::vector< TUniqueId >::const_iterator pend = x12c_puddleObjectIds.end();
rstl::vector< TUniqueId >::const_iterator pit =
rstl::find< rstl::vector< TUniqueId >::const_iterator, TUniqueId >(x12c_puddleObjectIds.begin(), pend, uid);
if (pit != pend) {
// if (contains(x12c_puddleObjectIds, x12c_puddleObjectIds.end(), uid)) {
rstl::vector< TUniqueId >::const_iterator it = x12c_puddleObjectIds.begin();
for (; it != x12c_puddleObjectIds.end(); ++it) {
if (CEntity* ent = mgr.ObjectById(*it)) {
if (!ent->GetActive()) {
mgr.SendScriptMsg(ent, GetUniqueId(), kSM_Activate);
} else {
mgr.FreeScriptObject(ent->GetUniqueId());
}
}
}
CObjectList& list = mgr.GetObjectListById(kOL_All);
int objIdx = list.GetFirstObjectIndex();
while (objIdx != -1) {
if (CScriptMazeNode* node = TCastToPtr< CScriptMazeNode >(list[objIdx])) {
if (node->xe8_col == xe8_col - 1 && node->xec_row == xec_row && node->xf0_side == CMazeState::kS_Right) {
SMazeCell& cell = maze->GetCell(xe8_col - 1, xec_row);
if (!cell.x0_25_openRight) {
cell.x0_25_openRight = true;
node->Reset(mgr);
node->x13c_25_hasGate = false;
}
}
if (node->xe8_col == xe8_col && node->xec_row == xec_row && node->xf0_side == CMazeState::kS_Right) {
SMazeCell& cell = maze->GetCell(xe8_col, xec_row);
if (!cell.x0_25_openRight) {
cell.x0_25_openRight = true;
node->Reset(mgr);
node->x13c_25_hasGate = false;
}
}
if (node->xe8_col == xe8_col && node->xec_row == xec_row && node->xf0_side == CMazeState::kS_Top) {
SMazeCell& cell = maze->GetCell(xe8_col, xec_row);
if (!cell.x0_24_openTop) {
cell.x0_24_openTop = true;
node->Reset(mgr);
node->x13c_25_hasGate = false;
}
}
if (node->xe8_col == xe8_col && node->xec_row == xec_row + 1 && node->xf0_side == CMazeState::kS_Top) {
SMazeCell& cell = maze->GetCell(xe8_col, xec_row + 1);
if (!cell.x0_24_openTop) {
cell.x0_24_openTop = true;
node->Reset(mgr);
node->x13c_25_hasGate = false;
}
}
}
objIdx = list.GetNextObjectIndex(objIdx);
}
}
}
break;
}
case kSM_Deleted: {
if (mgr.GetCurrentMaze()) {
mgr.SetCurrentMaze(nullptr);
}
Reset(mgr);
break;
}
case kSM_Deactivate: {
Reset(mgr);
break;
}
}
}
CEntity::AcceptScriptMsg(msg, uid, mgr);
}
void CScriptMazeNode::Think(float dt, CStateManager& mgr) {
if (GetActive() && x13c_25_hasGate) {
xf8_msgTimer -= dt;
if (xf8_msgTimer <= 0.f) {
xf8_msgTimer = 1.f;
if (x13c_26_gateActive) {
x13c_26_gateActive = false;
SendScriptMsgs(mgr, kSM_Deactivate);
} else {
x13c_26_gateActive = true;
SendScriptMsgs(mgr, kSM_Activate);
}
}
}
}
void CScriptMazeNode::LoadMazeSeeds() {
const SObjectTag* tag = gpResourceFactory->GetResourceIdByName("DUMB_MazeSeeds");
rstl::auto_ptr< CInputStream > stream = gpResourceFactory->GetResLoader().LoadNewResourceSync(*tag, nullptr);
for (int i = 0; i < 300; i++) {
sMazeSeeds[i] = stream->ReadLong();
}
}
CScriptMazeNode::~CScriptMazeNode() {}
Reference in New Issue View Git Blame Copy Permalink