140 lines
3.3 KiB
C++
140 lines
3.3 KiB
C++
#include "common.h"
|
|
#include "quam/pak.h"
|
|
|
|
struct PakHeader {
|
|
quint32 dirOffset;
|
|
quint32 dirNum;
|
|
};
|
|
|
|
struct PakEntry {
|
|
std::string name;
|
|
PakFile file;
|
|
};
|
|
|
|
static constexpr quint32 sizeOfPakEntry = 64;
|
|
|
|
static PakHeader readPakHeader(std::istream &st) {
|
|
auto magic = readBytes<4>(st);
|
|
|
|
if(magic != std::array{'P', 'A', 'C', 'K'}) {
|
|
throw std::runtime_error("not a pak file (invalid magic number)");
|
|
}
|
|
|
|
auto dirOffset = readLE<quint32>(st);
|
|
auto dirSize = readLE<quint32>(st);
|
|
|
|
if(dirSize % sizeOfPakEntry != 0) {
|
|
throw std::runtime_error("invalid directory size");
|
|
}
|
|
|
|
PakHeader hdr;
|
|
hdr.dirOffset = dirOffset;
|
|
hdr.dirNum = dirSize / sizeOfPakEntry;
|
|
return hdr;
|
|
}
|
|
|
|
static PakEntry readPakEntry(std::istream &st, quint32 id) {
|
|
auto entName = readBytes<56>(st);
|
|
auto entOffset = readLE<quint32>(st);
|
|
auto entSize = readLE<quint32>(st);
|
|
|
|
auto pos = st.tellg();
|
|
|
|
st.seekg(entOffset);
|
|
|
|
PakFile file;
|
|
file.id = id;
|
|
file.resize(entSize);
|
|
st.read(file.data(), entSize);
|
|
|
|
st.seekg(pos);
|
|
|
|
auto zero = std::find(entName.cbegin(), entName.cend(), '\0');
|
|
|
|
std::string name;
|
|
std::copy(entName.cbegin(), zero, std::back_inserter(name));
|
|
|
|
if(name.front() == '/') {
|
|
throw std::runtime_error("empty root directory name");
|
|
}
|
|
|
|
PakEntry ent;
|
|
ent.name = std::move(name);
|
|
ent.file = std::move(file);
|
|
return ent;
|
|
}
|
|
|
|
void insertFile(PakDir &dir, std::string name, PakFile file) {
|
|
if(auto slash = name.find('/'); slash != std::string::npos) {
|
|
auto folder = name.substr(0, slash);
|
|
auto next = name.substr(slash + 1);
|
|
dir[folder] = PakNode{PakDir{}};
|
|
insertFile(std::get<PakDir>(dir[folder]),
|
|
std::move(next),
|
|
std::move(file));
|
|
} else {
|
|
dir[name] = PakNode{std::move(file)};
|
|
}
|
|
}
|
|
|
|
PakDir readPak(std::istream &st) {
|
|
auto hdr = readPakHeader(st);
|
|
st.seekg(hdr.dirOffset);
|
|
|
|
PakDir root;
|
|
|
|
for(quint32 id = 0; id < hdr.dirNum; id++) {
|
|
auto ent = readPakEntry(st, id);
|
|
insertFile(root, std::move(ent.name), std::move(ent.file));
|
|
}
|
|
|
|
return root;
|
|
}
|
|
|
|
void setTableToPakDir(QTableWidget &table, PakDir const &dir) {
|
|
constexpr auto Flags = Qt::ItemIsSelectable |
|
|
Qt::ItemIsDragEnabled |
|
|
Qt::ItemIsEnabled |
|
|
Qt::ItemNeverHasChildren;
|
|
|
|
auto sorted = table.isSortingEnabled();
|
|
table.clearContents();
|
|
table.setSortingEnabled(false);
|
|
quint32 row{0};
|
|
for(auto const &kv : dir) {
|
|
auto const &name = kv.first;
|
|
auto const &node = kv.second;
|
|
table.setRowCount(row + 1);
|
|
{
|
|
auto item = new QTableWidgetItem;
|
|
item->setFlags(Flags);
|
|
if(auto file = std::get_if<PakFile>(&node)) {
|
|
item->setText(QString::number(file->id));
|
|
}
|
|
table.setItem(row, Pak::ColumnId, item);
|
|
}
|
|
{
|
|
auto item = new QTableWidgetItem;
|
|
item->setFlags(Flags);
|
|
if(auto file = std::get_if<PakFile>(&node)) {
|
|
item->setText(QString::number(file->size()));
|
|
}
|
|
table.setItem(row, Pak::ColumnSize, item);
|
|
}
|
|
{
|
|
auto item = new QTableWidgetItem;
|
|
auto icon = std::holds_alternative<PakDir>(node) ? "folder" :
|
|
"text-x-generic";
|
|
item->setFlags(Flags);
|
|
item->setText(QString::fromStdString(name));
|
|
item->setIcon(QIcon::fromTheme(icon));
|
|
table.setItem(row, Pak::ColumnName, item);
|
|
}
|
|
++row;
|
|
}
|
|
table.setSortingEnabled(sorted);
|
|
table.resizeColumnsToContents();
|
|
}
|
|
|
|
// EOF
|