From 4fcad1ada21be34433f676271014f1d93f1cb791 Mon Sep 17 00:00:00 2001
From: Henrique Gemignani Passos Lima <henrique@gemignani.org>
Date: Tue, 4 Oct 2022 21:50:29 +0300
Subject: [PATCH] First pass at CPlayerState::PutTo

---
 include/Kyoto/Streams/COutputStream.hpp  | 11 ++++++++++
 src/MetroidPrime/Player/CPlayerState.cpp | 26 +++++++++++-------------
 2 files changed, 23 insertions(+), 14 deletions(-)
 create mode 100644 include/Kyoto/Streams/COutputStream.hpp

diff --git a/include/Kyoto/Streams/COutputStream.hpp b/include/Kyoto/Streams/COutputStream.hpp
new file mode 100644
index 00000000..871cb99d
--- /dev/null
+++ b/include/Kyoto/Streams/COutputStream.hpp
@@ -0,0 +1,11 @@
+#ifndef _COUTPUTSTREAM_HPP
+#define _COUTPUTSTREAM_HPP
+
+#include "types.h"
+
+class COutputStream {
+public:
+  void WriteBits(int val, int bitCount);
+};
+
+#endif // _COUTPUTSTREAM_HPP
diff --git a/src/MetroidPrime/Player/CPlayerState.cpp b/src/MetroidPrime/Player/CPlayerState.cpp
index 9c4d07e4..8367b768 100644
--- a/src/MetroidPrime/Player/CPlayerState.cpp
+++ b/src/MetroidPrime/Player/CPlayerState.cpp
@@ -9,6 +9,7 @@
 
 #include "Kyoto/Math/CMath.hpp"
 #include "Kyoto/Streams/CInputStream.hpp"
+#include "Kyoto/Streams/COutputStream.hpp"
 
 #include "rstl/math.hpp"
 
@@ -114,32 +115,29 @@ CPlayerState::CPlayerState(CInputStream& stream)
 }
 
 void CPlayerState::PutTo(COutputStream& stream) {
-  /*
   stream.WriteBits(x4_enabledItems, 32);
 
   const float realHP = xc_health.GetHP();
-  u32 integralHP;
-  std::memcpy(&integralHP, &realHP, sizeof(u32));
-
-  stream.WriteBits(integralHP, 32);
-  stream.WriteBits(u32(x8_currentBeam), COutputStream::GetBitCount(5));
-  stream.WriteBits(u32(x20_currentSuit), COutputStream::GetBitCount(4));
+  stream.WriteBits(*(int*)(&realHP), 32);
+  stream.WriteBits(int(x8_currentBeam), GetBitCount(5));
+  stream.WriteBits(int(x20_currentSuit), GetBitCount(4));
   for (size_t i = 0; i < x24_powerups.size(); ++i) {
     const CPowerUp& pup = x24_powerups[i];
-    stream.WriteBits(pup.x0_amount, COutputStream::GetBitCount(PowerUpMaxValues[i]));
-    stream.WriteBits(pup.x4_capacity, COutputStream::GetBitCount(PowerUpMaxValues[i]));
+    stream.WriteBits(pup.x0_amount, GetBitCount(kPowerUpMax[i]));
+    stream.WriteBits(pup.x4_capacity, GetBitCount(kPowerUpMax[i]));
   }
 
-  for (const auto& scanTime : x170_scanTimes) {
-    if (scanTime.second >= 1.f)
+  for (rstl::vector< rstl::pair< CAssetId, float > >::iterator it = x170_scanTimes.begin();
+    it != x170_scanTimes.end(); ++it) {
+    if (it->second >= 1.f)
       stream.WriteBits(true, 1);
     else
       stream.WriteBits(false, 1);
   }
 
-  stream.WriteBits(x180_scanCompletionRate.first, COutputStream::GetBitCount(0x100));
-  stream.WriteBits(x180_scanCompletionRate.second, COutputStream::GetBitCount(0x100));
-  */
+  stream.WriteBits(x180_scanCompletionRateFirst, GetBitCount(0x100));
+  stream.WriteBits(x184_scanCompletionRateSecond, GetBitCount(0x100));
+
 }
 
 void CPlayerState::ReInitializePowerUp(CPlayerState::EItemType type, int capacity) {