NANDImporter: Use a indeterminate progress meter

3 files changed, 13 insertions, 35 deletions

diff --git a/Source/Core/DiscIO/NANDImporter.cpp b/Source/Core/DiscIO/NANDImporter.cpp
index 7b523dbca9..512d3f6264 100644
--- a/Source/Core/DiscIO/NANDImporter.cpp
+++ b/Source/Core/DiscIO/NANDImporter.cpp
@@ -23,7 +23,7 @@ NANDImporter::NANDImporter() = default;
 NANDImporter::~NANDImporter() = default;
 
 void NANDImporter::ImportNANDBin(const std::string& path_to_bin,
-                                 std::function<void(size_t, size_t)> update_callback)
+                                 std::function<void()> update_callback)
 {
   m_update_callback = std::move(update_callback);
 
@@ -32,7 +32,6 @@ void NANDImporter::ImportNANDBin(const std::string& path_to_bin,
 
   const std::string nand_root = File::GetUserPath(D_WIIROOT_IDX);
   FindSuperblock();
-  CountEntries(0);
   ProcessEntry(0, nand_root);
   ExportKeys(nand_root);
   ExtractCertificates(nand_root);
@@ -62,6 +61,11 @@ bool NANDImporter::ReadNANDBin(const std::string& path_to_bin)
 
   for (size_t i = 0; i < NAND_TOTAL_BLOCKS; i++)
   {
+    // Instead of updating on every cycle, we only update every 1000 cycles for a balance between
+    // not updating fast enough vs updating too fast
+    if (i % 1000 == 0)
+      m_update_callback();
+
     file.ReadBytes(&m_nand[i * NAND_BLOCK_SIZE], NAND_BLOCK_SIZE);
     file.Seek(NAND_ECC_BLOCK_SIZE, SEEK_CUR);  // We don't care about the ECC blocks
   }
@@ -113,7 +117,6 @@ void NANDImporter::ProcessEntry(u16 entry_number, const std::string& parent_path
   NANDFSTEntry entry;
   memcpy(&entry, &m_nand[m_nand_fst_offset + sizeof(NANDFSTEntry) * Common::swap16(entry_number)],
          sizeof(NANDFSTEntry));
-  UpdateStatus();
 
   if (entry.sib != 0xffff)
     ProcessEntry(entry.sib, parent_path);
@@ -126,6 +129,8 @@ void NANDImporter::ProcessEntry(u16 entry_number, const std::string& parent_path
 
 void NANDImporter::ProcessDirectory(const NANDFSTEntry& entry, const std::string& parent_path)
 {
+  m_update_callback();
+
   const std::string path = GetPath(entry, parent_path);
   File::CreateDir(path);
 
@@ -138,6 +143,8 @@ void NANDImporter::ProcessFile(const NANDFSTEntry& entry, const std::string& par
   constexpr size_t NAND_AES_KEY_OFFSET = 0x158;
   constexpr size_t NAND_FAT_BLOCK_SIZE = 0x4000;
 
+  m_update_callback();
+
   const std::string path = GetPath(entry, parent_path);
   File::IOFile file(path, "wb");
   std::array<u8, 16> key{};
@@ -201,24 +208,4 @@ void NANDImporter::ExportKeys(const std::string& nand_root)
   if (!file.WriteBytes(m_nand_keys.data(), NAND_KEYS_SIZE))
     PanicAlertT("Unable to write to file %s", file_path.c_str());
 }
-
-void NANDImporter::CountEntries(u16 entry_number)
-{
-  NANDFSTEntry entry;
-  memcpy(&entry, &m_nand[m_nand_fst_offset + sizeof(NANDFSTEntry) * Common::swap16(entry_number)],
-         sizeof(NANDFSTEntry));
-
-  m_total_entries++;
-
-  if (entry.sib != 0xffff)
-    CountEntries(entry.sib);
-
-  if ((entry.mode & 3) == 2 && entry.sub != 0xffff)
-    CountEntries(entry.sub);
-}
-
-void NANDImporter::UpdateStatus()
-{
-  m_update_callback(m_current_entry++, m_total_entries);
-}
 }
diff --git a/Source/Core/DiscIO/NANDImporter.h b/Source/Core/DiscIO/NANDImporter.h
index d79f733284..f46d8ff855 100644
--- a/Source/Core/DiscIO/NANDImporter.h
+++ b/Source/Core/DiscIO/NANDImporter.h
@@ -18,8 +18,7 @@ public:
   NANDImporter();
   ~NANDImporter();
 
-  void ImportNANDBin(const std::string& path_to_bin,
-                     std::function<void(size_t, size_t)> update_callback);
+  void ImportNANDBin(const std::string& path_to_bin, std::function<void()> update_callback);
   void ExtractCertificates(const std::string& nand_root);
 
 private:
@@ -46,15 +45,11 @@ private:
   void ProcessFile(const NANDFSTEntry& entry, const std::string& parent_path);
   void ProcessDirectory(const NANDFSTEntry& entry, const std::string& parent_path);
   void ExportKeys(const std::string& nand_root);
-  void CountEntries(u16 entry_number);
-  void UpdateStatus();
 
   std::vector<u8> m_nand;
   std::vector<u8> m_nand_keys;
   size_t m_nand_fat_offset = 0;
   size_t m_nand_fst_offset = 0;
-  std::function<void(size_t, size_t)> m_update_callback;
-  size_t m_total_entries = 0;
-  size_t m_current_entry = 0;
+  std::function<void()> m_update_callback;
 };
 }
diff --git a/Source/Core/DolphinWX/FrameTools.cpp b/Source/Core/DolphinWX/FrameTools.cpp
index c294efefbd..c7682c9fac 100644
--- a/Source/Core/DolphinWX/FrameTools.cpp
+++ b/Source/Core/DolphinWX/FrameTools.cpp
@@ -1279,11 +1279,7 @@ void CFrame::OnImportBootMiiBackup(wxCommandEvent& WXUNUSED(event))
 
   wxProgressDialog dialog(_("Importing NAND backup"), _("Working..."), 100, this,
                           wxPD_APP_MODAL | wxPD_ELAPSED_TIME | wxPD_SMOOTH);
-  DiscIO::NANDImporter().ImportNANDBin(file_name,
-                                       [&dialog](size_t current_entry, size_t total_entries) {
-                                         dialog.SetRange(total_entries);
-                                         dialog.Update(current_entry);
-                                       });
+  DiscIO::NANDImporter().ImportNANDBin(file_name, [&dialog] { dialog.Pulse(); });
   UpdateLoadWiiMenuItem();
 }