60 lines
2.0 KiB
C++
60 lines
2.0 KiB
C++
#include "genericModel.h"
|
|
genericModel::genericModel(const std::string& storepath, const std::string &id, int index) : storepath(storepath), id(id), index(index) {
|
|
fs::current_path("/opfs");
|
|
fs::create_directories(storepath);
|
|
fs::current_path(storepath);
|
|
}
|
|
bool genericModel::checkModel() {
|
|
if(!checkModelFiles()) return false;
|
|
static std::error_code c{};
|
|
if(!fs::exists("id", c)) return false;
|
|
std::ifstream file {"id", std::ifstream::binary};
|
|
if(!file.is_open()) return false;
|
|
long long size {file.seekg(0, std::ios::end).tellg()};
|
|
std::string oldid(size, ' ');
|
|
file.seekg(0);
|
|
file.read(&oldid[0], size);
|
|
return id.compare(oldid) == 0 ? true : false;
|
|
}
|
|
void genericModel::afterFetch(int memAddr, size_t size) {
|
|
// FIXME: Recognizer reuse this thread if possible
|
|
std::thread t{[this, memAddr, size](){
|
|
char* modelData = reinterpret_cast<char*>(memAddr);
|
|
if(!extractModel(modelData, size)) {
|
|
free(modelData);
|
|
fireEv("_continue", "Unable to extract model", index);
|
|
return;
|
|
}
|
|
free(modelData);
|
|
std::ofstream idFile("id");
|
|
if(!idFile.is_open()) {
|
|
fs::current_path("/opfs");
|
|
fs::remove_all(storepath);
|
|
fireEv("_continue", "Unable to write model ID", index);
|
|
return;
|
|
}
|
|
idFile << id;
|
|
idFile.close();
|
|
load(false);
|
|
}};
|
|
t.detach();
|
|
}
|
|
bool genericModel::extractModel(const char* fileBuf, size_t size) {
|
|
std::string path{};
|
|
archive* src {archive_read_new()};
|
|
archive_entry* entry {};
|
|
archive_read_support_filter_all(src);
|
|
archive_read_support_format_all(src);
|
|
archive_read_open_memory(src, fileBuf, size);
|
|
if(archive_errno(src) != 0) return false;
|
|
while (archive_read_next_header(src, &entry) == ARCHIVE_OK) {
|
|
path = archive_entry_pathname(entry);
|
|
// Strip first component, keep relative path
|
|
path = "." + path.substr(path.find("/"));
|
|
archive_entry_set_pathname(entry, path.c_str());
|
|
if(archive_errno(src) != 0) return false;
|
|
archive_read_extract(src, entry, ARCHIVE_EXTRACT_UNLINK);
|
|
}
|
|
archive_read_free(src);
|
|
return true;
|
|
} |