Syping/libragephoto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

RagePhoto-Extract: use RagePhoto::loadFile

Browse source
This commit is contained in:
Syping 2021-11-02 10:24:22 +01:00
parent a396d014f7
commit 9d9e0c8b05
1 changed files with 23 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
src/RagePhoto-Extract.cpp
View file

@ -28,47 +28,36 @@ int main(int argc, char *argv[])
return 0; return 0;
} }
// Make it crash when RagePhoto have a bug which make the deinitialisation to fail // Initialise RagePhoto
{ RagePhoto ragePhoto;
// Initialise RagePhoto
RagePhoto ragePhoto;
// Read file // Load Photo
std::ifstream ifs(argv[1], std::ios::in | std::ios::binary); const bool loaded = ragePhoto.loadFile(argv[1]);
if (!ifs.is_open()) {
if (!loaded) {
if (ragePhoto.error() == RagePhoto::Error::Uninitialised) {
std::cout << "Failed to open file: " << argv[1] << std::endl; std::cout << "Failed to open file: " << argv[1] << std::endl;
return 1; return 1;
} }
std::string data(std::istreambuf_iterator<char>{ifs}, {}); else if (ragePhoto.error() <= RagePhoto::Error::PhotoReadError) {
ifs.close(); std::cout << "Failed to load photo" << std::endl;
// Load Photo
const bool loaded = ragePhoto.load(data);
if (!loaded) {
if (ragePhoto.error() <= RagePhoto::Error::PhotoReadError) {
std::cout << "Failed to load photo" << std::endl;
return 1;
}
}
// Write jpeg
std::ofstream ofs(argv[2], std::ios::out | std::ios::binary | std::ios::trunc);
if (!ofs.is_open()) {
std::cout << "Failed to write file: " << argv[2] << std::endl;
return 1; return 1;
} }
ofs << ragePhoto.photo();
ofs.close();
if (ragePhoto.format() == RagePhoto::GTA5)
std::cout << "GTA V Photo successfully exported" << std::endl;
else
std::cout << "RDR 2 Photo successfully exported" << std::endl;
// Clear RagePhoto (provocate crash when pointer leak)
ragePhoto.clear();
} }
// Write jpeg
std::ofstream ofs(argv[2], std::ios::out | std::ios::binary | std::ios::trunc);
if (!ofs.is_open()) {
std::cout << "Failed to write file: " << argv[2] << std::endl;
return 1;
}
ofs << ragePhoto.photo();
ofs.close();
if (ragePhoto.format() == RagePhoto::GTA5)
std::cout << "GTA V Photo successfully exported" << std::endl;
else
std::cout << "RDR 2 Photo successfully exported" << std::endl;
return 0; return 0;
} }