Move DiscIO::ConvertToPlain to FileBlob.cpp
There is no longer anything GCZ specific about it.
This commit is contained in:
parent
8a9597e32e
commit
42f6913bcc
|
@ -336,85 +336,6 @@ bool ConvertToGCZ(const std::string& infile_path, const std::string& outfile_pat
|
||||||
return success;
|
return success;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ConvertToPlain(const std::string& infile_path, const std::string& outfile_path,
|
|
||||||
CompressCB callback, void* arg)
|
|
||||||
{
|
|
||||||
std::unique_ptr<BlobReader> reader = CreateBlobReader(infile_path);
|
|
||||||
if (!reader)
|
|
||||||
{
|
|
||||||
PanicAlertT("Failed to open the input file \"%s\".", infile_path.c_str());
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
ASSERT(reader->IsDataSizeAccurate());
|
|
||||||
|
|
||||||
File::IOFile outfile(outfile_path, "wb");
|
|
||||||
if (!outfile)
|
|
||||||
{
|
|
||||||
PanicAlertT("Failed to open the output file \"%s\".\n"
|
|
||||||
"Check that you have permissions to write the target folder and that the media can "
|
|
||||||
"be written.",
|
|
||||||
outfile_path.c_str());
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
constexpr size_t DESIRED_BUFFER_SIZE = 0x80000;
|
|
||||||
u64 buffer_size = reader->GetBlockSize();
|
|
||||||
if (buffer_size == 0)
|
|
||||||
{
|
|
||||||
buffer_size = DESIRED_BUFFER_SIZE;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
while (buffer_size < DESIRED_BUFFER_SIZE)
|
|
||||||
buffer_size *= 2;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<u8> buffer(buffer_size);
|
|
||||||
const u64 num_buffers = (reader->GetDataSize() + buffer_size - 1) / buffer_size;
|
|
||||||
int progress_monitor = std::max<int>(1, num_buffers / 100);
|
|
||||||
bool success = true;
|
|
||||||
|
|
||||||
for (u64 i = 0; i < num_buffers; i++)
|
|
||||||
{
|
|
||||||
if (i % progress_monitor == 0)
|
|
||||||
{
|
|
||||||
const bool was_cancelled =
|
|
||||||
!callback(Common::GetStringT("Unpacking"), (float)i / (float)num_buffers, arg);
|
|
||||||
if (was_cancelled)
|
|
||||||
{
|
|
||||||
success = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
const u64 inpos = i * buffer_size;
|
|
||||||
const u64 sz = std::min(buffer_size, reader->GetDataSize() - inpos);
|
|
||||||
if (!reader->Read(inpos, sz, buffer.data()))
|
|
||||||
{
|
|
||||||
PanicAlertT("Failed to read from the input file \"%s\".", infile_path.c_str());
|
|
||||||
success = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if (!outfile.WriteBytes(buffer.data(), sz))
|
|
||||||
{
|
|
||||||
PanicAlertT("Failed to write the output file \"%s\".\n"
|
|
||||||
"Check that you have enough space available on the target drive.",
|
|
||||||
outfile_path.c_str());
|
|
||||||
success = false;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!success)
|
|
||||||
{
|
|
||||||
// Remove the incomplete output file.
|
|
||||||
outfile.Close();
|
|
||||||
File::Delete(outfile_path);
|
|
||||||
}
|
|
||||||
|
|
||||||
return success;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool IsGCZBlob(File::IOFile& file)
|
bool IsGCZBlob(File::IOFile& file)
|
||||||
{
|
{
|
||||||
const u64 position = file.Tell();
|
const u64 position = file.Tell();
|
||||||
|
|
|
@ -2,10 +2,15 @@
|
||||||
// Licensed under GPLv2+
|
// Licensed under GPLv2+
|
||||||
// Refer to the license.txt file included.
|
// Refer to the license.txt file included.
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "Common/Assert.h"
|
||||||
|
#include "Common/FileUtil.h"
|
||||||
|
#include "Common/MsgHandler.h"
|
||||||
#include "DiscIO/FileBlob.h"
|
#include "DiscIO/FileBlob.h"
|
||||||
|
|
||||||
namespace DiscIO
|
namespace DiscIO
|
||||||
|
@ -36,4 +41,83 @@ bool PlainFileReader::Read(u64 offset, u64 nbytes, u8* out_ptr)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool ConvertToPlain(const std::string& infile_path, const std::string& outfile_path,
|
||||||
|
CompressCB callback, void* arg)
|
||||||
|
{
|
||||||
|
std::unique_ptr<BlobReader> reader = CreateBlobReader(infile_path);
|
||||||
|
if (!reader)
|
||||||
|
{
|
||||||
|
PanicAlertT("Failed to open the input file \"%s\".", infile_path.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
ASSERT(reader->IsDataSizeAccurate());
|
||||||
|
|
||||||
|
File::IOFile outfile(outfile_path, "wb");
|
||||||
|
if (!outfile)
|
||||||
|
{
|
||||||
|
PanicAlertT("Failed to open the output file \"%s\".\n"
|
||||||
|
"Check that you have permissions to write the target folder and that the media can "
|
||||||
|
"be written.",
|
||||||
|
outfile_path.c_str());
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
constexpr size_t DESIRED_BUFFER_SIZE = 0x80000;
|
||||||
|
u64 buffer_size = reader->GetBlockSize();
|
||||||
|
if (buffer_size == 0)
|
||||||
|
{
|
||||||
|
buffer_size = DESIRED_BUFFER_SIZE;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
while (buffer_size < DESIRED_BUFFER_SIZE)
|
||||||
|
buffer_size *= 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<u8> buffer(buffer_size);
|
||||||
|
const u64 num_buffers = (reader->GetDataSize() + buffer_size - 1) / buffer_size;
|
||||||
|
int progress_monitor = std::max<int>(1, num_buffers / 100);
|
||||||
|
bool success = true;
|
||||||
|
|
||||||
|
for (u64 i = 0; i < num_buffers; i++)
|
||||||
|
{
|
||||||
|
if (i % progress_monitor == 0)
|
||||||
|
{
|
||||||
|
const bool was_cancelled =
|
||||||
|
!callback(Common::GetStringT("Unpacking"), (float)i / (float)num_buffers, arg);
|
||||||
|
if (was_cancelled)
|
||||||
|
{
|
||||||
|
success = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const u64 inpos = i * buffer_size;
|
||||||
|
const u64 sz = std::min(buffer_size, reader->GetDataSize() - inpos);
|
||||||
|
if (!reader->Read(inpos, sz, buffer.data()))
|
||||||
|
{
|
||||||
|
PanicAlertT("Failed to read from the input file \"%s\".", infile_path.c_str());
|
||||||
|
success = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (!outfile.WriteBytes(buffer.data(), sz))
|
||||||
|
{
|
||||||
|
PanicAlertT("Failed to write the output file \"%s\".\n"
|
||||||
|
"Check that you have enough space available on the target drive.",
|
||||||
|
outfile_path.c_str());
|
||||||
|
success = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!success)
|
||||||
|
{
|
||||||
|
// Remove the incomplete output file.
|
||||||
|
outfile.Close();
|
||||||
|
File::Delete(outfile_path);
|
||||||
|
}
|
||||||
|
|
||||||
|
return success;
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace DiscIO
|
} // namespace DiscIO
|
||||||
|
|
Loading…
Reference in New Issue