mirror of
https://github.com/open-goal/jak-project.git
synced 2024-10-20 11:26:18 -04:00
6446389263
* extractor: refactor and cleanup for multi-game support * deps: switch to `ghc::filesystem` as it is utf-8 everywhere by default * extractor: finally working with unicode * unicode: fix unicode cli args on windows in all `main` functions
79 lines
1.8 KiB
C++
79 lines
1.8 KiB
C++
#pragma once
|
|
|
|
/*!
|
|
* @file BinaryWriter.h
|
|
* Write raw data like a stream.
|
|
*/
|
|
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
#include <stdexcept>
|
|
#include <vector>
|
|
|
|
#include "common/util/Assert.h"
|
|
#include "common/util/FileUtil.h"
|
|
|
|
struct BinaryWriterRef {
|
|
size_t offset;
|
|
size_t write_size;
|
|
};
|
|
|
|
class BinaryWriter {
|
|
public:
|
|
BinaryWriter() = default;
|
|
|
|
template <typename T>
|
|
BinaryWriterRef add(const T& obj) {
|
|
auto orig_size = data.size();
|
|
data.resize(orig_size + sizeof(T));
|
|
memcpy(data.data() + orig_size, &obj, sizeof(T));
|
|
return {orig_size, sizeof(T)};
|
|
}
|
|
|
|
template <typename T>
|
|
void add_at_ref(const T& obj, const BinaryWriterRef& ref) {
|
|
ASSERT(ref.write_size == sizeof(T));
|
|
ASSERT(ref.offset + ref.write_size < get_size());
|
|
memcpy(data.data() + ref.offset, &obj, sizeof(T));
|
|
}
|
|
|
|
void add_str_len(const std::string& str, size_t len) { add_cstr_len(str.c_str(), len); }
|
|
|
|
void add_cstr_len(const char* str, size_t len) {
|
|
size_t i = 0;
|
|
while (*str) {
|
|
data.push_back(*str);
|
|
str++;
|
|
i++;
|
|
}
|
|
|
|
while (i < len) {
|
|
data.push_back(0);
|
|
i++;
|
|
}
|
|
}
|
|
|
|
BinaryWriterRef add_data(void* d, size_t len) {
|
|
auto orig_size = data.size();
|
|
data.resize(orig_size + len);
|
|
memcpy(data.data() + orig_size, d, len);
|
|
return {orig_size, len};
|
|
}
|
|
|
|
size_t get_size() { return data.size(); }
|
|
|
|
void* get_data() { return data.data(); }
|
|
|
|
void write_to_file(const fs::path& filename) {
|
|
auto fp = file_util::open_file(filename.string().c_str(), "wb");
|
|
if (!fp)
|
|
throw std::runtime_error("failed to open " + filename.string());
|
|
if (fwrite(get_data(), get_size(), 1, fp) != 1)
|
|
throw std::runtime_error("failed to write " + filename.string());
|
|
fclose(fp);
|
|
}
|
|
|
|
private:
|
|
std::vector<uint8_t> data;
|
|
};
|