Merge pull request #8900 from JosJuice/disc-deduplicate

DiscIO: Deduplicate some VolumeGC/VolumeWii code
dolphin-emu · Jun 25, 2020 · 98de22a · 98de22a
2 parents 4ddd4ff + d17b801
commit 98de22a
Show file tree

Hide file tree

Showing 19 changed files with 139 additions and 167 deletions.
diff --git a/Source/Core/Core/Boot/Boot.cpp b/Source/Core/Core/Boot/Boot.cpp
@@ -58,7 +58,7 @@ namespace fs = std::filesystem;
 #include "Core/PowerPC/PowerPC.h"
 
 #include "DiscIO/Enums.h"
-#include "DiscIO/Volume.h"
+#include "DiscIO/VolumeDisc.h"
 #include "DiscIO/VolumeWad.h"
 
 static std::vector<std::string> ReadM3UFile(const std::string& m3u_path,

diff --git a/Source/Core/Core/Boot/Boot.h b/Source/Core/Core/Boot/Boot.h
@@ -16,7 +16,7 @@
 #include "Core/IOS/IOSC.h"
 #include "DiscIO/Blob.h"
 #include "DiscIO/Enums.h"
-#include "DiscIO/Volume.h"
+#include "DiscIO/VolumeDisc.h"
 #include "DiscIO/VolumeWad.h"
 
 namespace File

diff --git a/Source/Core/Core/Boot/Boot_BS2Emu.cpp b/Source/Core/Core/Boot/Boot_BS2Emu.cpp
@@ -33,7 +33,7 @@
 #include "Core/PowerPC/PowerPC.h"
 
 #include "DiscIO/Enums.h"
-#include "DiscIO/Volume.h"
+#include "DiscIO/VolumeDisc.h"
 
 namespace
 {

diff --git a/Source/Core/Core/HW/DVD/DVDInterface.cpp b/Source/Core/Core/HW/DVD/DVDInterface.cpp
@@ -38,7 +38,7 @@
 
 #include "DiscIO/Blob.h"
 #include "DiscIO/Enums.h"
-#include "DiscIO/Volume.h"
+#include "DiscIO/VolumeDisc.h"
 #include "DiscIO/VolumeWii.h"
 
 #include "VideoCommon/OnScreenDisplay.h"

diff --git a/Source/Core/Core/WiiUtils.cpp b/Source/Core/Core/WiiUtils.cpp
@@ -40,7 +40,7 @@
 #include "DiscIO/DiscExtractor.h"
 #include "DiscIO/Enums.h"
 #include "DiscIO/Filesystem.h"
-#include "DiscIO/Volume.h"
+#include "DiscIO/VolumeDisc.h"
 #include "DiscIO/VolumeFileBlobReader.h"
 #include "DiscIO/VolumeWad.h"
 

diff --git a/Source/Core/DiscIO/CMakeLists.txt b/Source/Core/DiscIO/CMakeLists.txt
@@ -32,6 +32,8 @@ add_library(discio
   TGCBlob.h
   Volume.cpp
   Volume.h
+  VolumeDisc.cpp
+  VolumeDisc.h
   VolumeFileBlobReader.cpp
   VolumeFileBlobReader.h
   VolumeGC.cpp

diff --git a/Source/Core/DiscIO/DiscIO.vcxproj b/Source/Core/DiscIO/DiscIO.vcxproj
@@ -60,6 +60,7 @@
     <ClCompile Include="ScrubbedBlob.cpp" />
     <ClCompile Include="TGCBlob.cpp" />
     <ClCompile Include="Volume.cpp" />
+    <ClCompile Include="VolumeDisc.cpp" />
     <ClCompile Include="VolumeFileBlobReader.cpp" />
     <ClCompile Include="VolumeGC.cpp" />
     <ClCompile Include="VolumeVerifier.cpp" />
@@ -89,6 +90,7 @@
     <ClInclude Include="ScrubbedBlob.h" />
     <ClInclude Include="TGCBlob.h" />
     <ClInclude Include="Volume.h" />
+    <ClInclude Include="VolumeDisc.h" />
     <ClInclude Include="VolumeFileBlobReader.h" />
     <ClInclude Include="VolumeGC.h" />
     <ClInclude Include="VolumeVerifier.h" />

diff --git a/Source/Core/DiscIO/DiscIO.vcxproj.filters b/Source/Core/DiscIO/DiscIO.vcxproj.filters
@@ -99,6 +99,7 @@
     <ClCompile Include="WIACompression.cpp">
       <Filter>Volume\Blob</Filter>
     </ClCompile>
+    <ClCompile Include="VolumeDisc.cpp" />
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="DiscScrubber.h">
@@ -182,6 +183,7 @@
     <ClInclude Include="WIACompression.h">
       <Filter>Volume\Blob</Filter>
     </ClInclude>
+    <ClInclude Include="VolumeDisc.h" />
   </ItemGroup>
   <ItemGroup>
     <Text Include="CMakeLists.txt" />

diff --git a/Source/Core/DiscIO/FileSystemGCWii.cpp b/Source/Core/DiscIO/FileSystemGCWii.cpp
@@ -23,7 +23,7 @@
 #include "DiscIO/DiscExtractor.h"
 #include "DiscIO/FileSystemGCWii.h"
 #include "DiscIO/Filesystem.h"
-#include "DiscIO/Volume.h"
+#include "DiscIO/VolumeDisc.h"
 
 namespace DiscIO
 {

diff --git a/Source/Core/DiscIO/ScrubbedBlob.cpp b/Source/Core/DiscIO/ScrubbedBlob.cpp
@@ -12,7 +12,7 @@
 #include "Common/Align.h"
 #include "DiscIO/Blob.h"
 #include "DiscIO/DiscScrubber.h"
-#include "DiscIO/Volume.h"
+#include "DiscIO/VolumeDisc.h"
 
 namespace DiscIO
 {

diff --git a/Source/Core/DiscIO/Volume.cpp b/Source/Core/DiscIO/Volume.cpp
@@ -17,6 +17,7 @@
 
 #include "DiscIO/Blob.h"
 #include "DiscIO/Enums.h"
+#include "DiscIO/VolumeDisc.h"
 #include "DiscIO/VolumeGC.h"
 #include "DiscIO/VolumeWad.h"
 #include "DiscIO/VolumeWii.h"

diff --git a/Source/Core/DiscIO/Volume.h b/Source/Core/DiscIO/Volume.h
@@ -23,6 +23,7 @@ namespace DiscIO
 class BlobReader;
 enum class BlobType;
 class FileSystem;
+class VolumeDisc;
 class VolumeWAD;
 
 struct Partition final
@@ -162,10 +163,6 @@ class Volume
   static const std::vector<u8> INVALID_CERT_CHAIN;
 };
 
-class VolumeDisc : public Volume
-{
-};
-
 std::unique_ptr<VolumeDisc> CreateDisc(const std::string& path);
 std::unique_ptr<VolumeWAD> CreateWAD(const std::string& path);
 std::unique_ptr<Volume> CreateVolume(const std::string& path);

diff --git a/Source/Core/DiscIO/VolumeDisc.cpp b/Source/Core/DiscIO/VolumeDisc.cpp
@@ -0,0 +1,87 @@
+// Copyright 2020 Dolphin Emulator Project
+// Licensed under GPLv2+
+// Refer to the license.txt file included.
+
+#include "DiscIO/VolumeDisc.h"
+
+#include <optional>
+#include <string>
+
+#include "Common/CommonTypes.h"
+#include "DiscIO/Enums.h"
+
+namespace DiscIO
+{
+std::string VolumeDisc::GetGameID(const Partition& partition) const
+{
+  char id[6];
+
+  if (!Read(0, sizeof(id), reinterpret_cast<u8*>(id), partition))
+    return std::string();
+
+  return DecodeString(id);
+}
+
+Country VolumeDisc::GetCountry(const Partition& partition) const
+{
+  // The 0 that we use as a default value is mapped to Country::Unknown and Region::Unknown
+  const u8 country_byte = ReadSwapped<u8>(3, partition).value_or(0);
+  const Region region = GetRegion();
+  const std::optional<u16> revision = GetRevision();
+
+  if (CountryCodeToRegion(country_byte, GetVolumeType(), region, revision) != region)
+    return TypicalCountryForRegion(region);
+
+  return CountryCodeToCountry(country_byte, GetVolumeType(), region, revision);
+}
+
+Region VolumeDisc::RegionCodeToRegion(std::optional<u32> region_code) const
+{
+  if (!region_code)
+    return Region::Unknown;
+
+  const Region region = static_cast<Region>(*region_code);
+  return region <= Region::NTSC_K ? region : Region::Unknown;
+}
+
+std::string VolumeDisc::GetMakerID(const Partition& partition) const
+{
+  char maker_id[2];
+
+  if (!Read(0x4, sizeof(maker_id), reinterpret_cast<u8*>(&maker_id), partition))
+    return std::string();
+
+  return DecodeString(maker_id);
+}
+
+std::optional<u16> VolumeDisc::GetRevision(const Partition& partition) const
+{
+  std::optional<u8> revision = ReadSwapped<u8>(7, partition);
+  return revision ? *revision : std::optional<u16>();
+}
+
+std::string VolumeDisc::GetInternalName(const Partition& partition) const
+{
+  char name[0x60];
+  if (!Read(0x20, sizeof(name), reinterpret_cast<u8*>(&name), partition))
+    return std::string();
+
+  return DecodeString(name);
+}
+
+std::string VolumeDisc::GetApploaderDate(const Partition& partition) const
+{
+  char date[16];
+
+  if (!Read(0x2440, sizeof(date), reinterpret_cast<u8*>(&date), partition))
+    return std::string();
+
+  return DecodeString(date);
+}
+
+std::optional<u8> VolumeDisc::GetDiscNumber(const Partition& partition) const
+{
+  return ReadSwapped<u8>(6, partition);
+}
+
+}  // namespace DiscIO
diff --git a/Source/Core/DiscIO/VolumeDisc.h b/Source/Core/DiscIO/VolumeDisc.h
@@ -0,0 +1,30 @@
+// Copyright 2020 Dolphin Emulator Project
+// Licensed under GPLv2+
+// Refer to the license.txt file included.
+
+#pragma once
+
+#include <optional>
+#include <string>
+
+#include "Common/CommonTypes.h"
+#include "DiscIO/Volume.h"
+
+namespace DiscIO
+{
+class VolumeDisc : public Volume
+{
+public:
+  std::string GetGameID(const Partition& partition = PARTITION_NONE) const override;
+  Country GetCountry(const Partition& partition = PARTITION_NONE) const override;
+  std::string GetMakerID(const Partition& partition = PARTITION_NONE) const override;
+  std::optional<u16> GetRevision(const Partition& partition = PARTITION_NONE) const override;
+  std::string GetInternalName(const Partition& partition = PARTITION_NONE) const override;
+  std::string GetApploaderDate(const Partition& partition) const override;
+  std::optional<u8> GetDiscNumber(const Partition& partition = PARTITION_NONE) const override;
+
+protected:
+  Region RegionCodeToRegion(std::optional<u32> region_code) const;
+};
+
+}  // namespace DiscIO
diff --git a/Source/Core/DiscIO/VolumeGC.cpp b/Source/Core/DiscIO/VolumeGC.cpp
@@ -57,21 +57,6 @@ const FileSystem* VolumeGC::GetFileSystem(const Partition& partition) const
   return m_file_system->get();
 }
 
-std::string VolumeGC::GetGameID(const Partition& partition) const
-{
-  static const std::string NO_UID("NO_UID");
-
-  char id[6];
-
-  if (!Read(0, sizeof(id), reinterpret_cast<u8*>(id), partition))
-  {
-    PanicAlertT("Failed to read unique ID from disc image");
-    return NO_UID;
-  }
-
-  return DecodeString(id);
-}
-
 std::string VolumeGC::GetGameTDBID(const Partition& partition) const
 {
   // Don't return an ID for Datel discs
@@ -83,48 +68,7 @@ std::string VolumeGC::GetGameTDBID(const Partition& partition) const
 
 Region VolumeGC::GetRegion() const
 {
-  const std::optional<u32> region_code = ReadSwapped<u32>(0x458, PARTITION_NONE);
-  if (!region_code)
-    return Region::Unknown;
-  const Region region = static_cast<Region>(*region_code);
-  return region <= Region::PAL ? region : Region::Unknown;
-}
-
-Country VolumeGC::GetCountry(const Partition& partition) const
-{
-  // The 0 that we use as a default value is mapped to Country::Unknown and Region::Unknown
-  const u8 country = ReadSwapped<u8>(3, partition).value_or(0);
-  const Region region = GetRegion();
-  const std::optional<u16> revision = GetRevision();
-
-  if (CountryCodeToRegion(country, Platform::GameCubeDisc, region, revision) != region)
-    return TypicalCountryForRegion(region);
-
-  return CountryCodeToCountry(country, Platform::GameCubeDisc, region, revision);
-}
-
-std::string VolumeGC::GetMakerID(const Partition& partition) const
-{
-  char maker_id[2];
-  if (!Read(0x4, sizeof(maker_id), reinterpret_cast<u8*>(&maker_id), partition))
-    return std::string();
-
-  return DecodeString(maker_id);
-}
-
-std::optional<u16> VolumeGC::GetRevision(const Partition& partition) const
-{
-  std::optional<u8> revision = ReadSwapped<u8>(7, partition);
-  return revision ? *revision : std::optional<u16>();
-}
-
-std::string VolumeGC::GetInternalName(const Partition& partition) const
-{
-  char name[0x60];
-  if (Read(0x20, sizeof(name), reinterpret_cast<u8*>(name), partition))
-    return DecodeString(name);
-
-  return "";
+  return RegionCodeToRegion(m_reader->ReadSwapped<u32>(0x458));
 }
 
 std::map<Language, std::string> VolumeGC::GetShortNames() const
@@ -159,15 +103,6 @@ std::vector<u32> VolumeGC::GetBanner(u32* width, u32* height) const
   return m_converted_banner->image_buffer;
 }
 
-std::string VolumeGC::GetApploaderDate(const Partition& partition) const
-{
-  char date[16];
-  if (!Read(0x2440, sizeof(date), reinterpret_cast<u8*>(&date), partition))
-    return std::string();
-
-  return DecodeString(date);
-}
-
 BlobType VolumeGC::GetBlobType() const
 {
   return m_reader->GetBlobType();
@@ -193,11 +128,6 @@ const BlobReader& VolumeGC::GetBlobReader() const
   return *m_reader;
 }
 
-std::optional<u8> VolumeGC::GetDiscNumber(const Partition& partition) const
-{
-  return ReadSwapped<u8>(6, partition);
-}
-
 Platform VolumeGC::GetVolumeType() const
 {
   return Platform::GameCubeDisc;

diff --git a/Source/Core/DiscIO/VolumeGC.h b/Source/Core/DiscIO/VolumeGC.h
@@ -14,6 +14,7 @@
 #include "Common/Lazy.h"
 #include "DiscIO/Filesystem.h"
 #include "DiscIO/Volume.h"
+#include "DiscIO/VolumeDisc.h"
 
 namespace DiscIO
 {
@@ -33,28 +34,21 @@ class VolumeGC : public VolumeDisc
   bool Read(u64 offset, u64 length, u8* buffer,
             const Partition& partition = PARTITION_NONE) const override;
   const FileSystem* GetFileSystem(const Partition& partition = PARTITION_NONE) const override;
-  std::string GetGameID(const Partition& partition = PARTITION_NONE) const override;
   std::string GetGameTDBID(const Partition& partition = PARTITION_NONE) const override;
-  std::string GetMakerID(const Partition& partition = PARTITION_NONE) const override;
-  std::optional<u16> GetRevision(const Partition& partition = PARTITION_NONE) const override;
-  std::string GetInternalName(const Partition& partition = PARTITION_NONE) const override;
   std::map<Language, std::string> GetShortNames() const override;
   std::map<Language, std::string> GetLongNames() const override;
   std::map<Language, std::string> GetShortMakers() const override;
   std::map<Language, std::string> GetLongMakers() const override;
   std::map<Language, std::string> GetDescriptions() const override;
   std::vector<u32> GetBanner(u32* width, u32* height) const override;
-  std::string GetApploaderDate(const Partition& partition = PARTITION_NONE) const override;
-  std::optional<u8> GetDiscNumber(const Partition& partition = PARTITION_NONE) const override;
 
   Platform GetVolumeType() const override;
   Region GetRegion() const override;
-  Country GetCountry(const Partition& partition = PARTITION_NONE) const override;
   BlobType GetBlobType() const override;
   u64 GetSize() const override;
   bool IsSizeAccurate() const override;
   u64 GetRawSize() const override;
-  const BlobReader& GetBlobReader() const;
+  const BlobReader& GetBlobReader() const override;
 
 private:
   static const u32 GC_BANNER_WIDTH = 96;

diff --git a/Source/Core/DiscIO/VolumeWad.h b/Source/Core/DiscIO/VolumeWad.h
@@ -66,7 +66,7 @@ class VolumeWAD : public Volume
   u64 GetSize() const override;
   bool IsSizeAccurate() const override;
   u64 GetRawSize() const override;
-  const BlobReader& GetBlobReader() const;
+  const BlobReader& GetBlobReader() const override;
 
 private:
   std::unique_ptr<BlobReader> m_reader;