dlib/examples/file_to_code_ex.cpp

112 lines
3.7 KiB
C++

// The contents of this file are in the public domain. See LICENSE_FOR_EXAMPLE_PROGRAMS.txt
/*
This is an example illustrating the use of the compress_stream and
base64 components from the dlib C++ Library.
It reads in a file from the disk and compresses it in an in memory buffer and
then converts that buffer into base64 text. The final step is to output to
the screen some C++ code that contains this base64 encoded text and can decompress
it back into its original form.
*/
#include <iostream>
#include <fstream>
#include <sstream>
#include <string>
#include <cstdlib>
#include "dlib/compress_stream.h"
#include "dlib/base64.h"
using namespace std;
using namespace dlib;
int main(int argc, char** argv)
{
if (argc != 2)
{
cout << "You must give a file name as the argument to this program.\n" << endl;
cout << "This program reads in a file from the disk and compresses\n"
<< "it in an in memory buffer and then converts that buffer \n"
<< "into base64 text. The final step is to output to the screen\n"
<< "some C++ code that contains this base64 encoded text and can\n"
<< "decompress it back into its original form.\n" << endl;
return EXIT_FAILURE;
}
// open the file the user specified on the command line
ifstream fin(argv[1], ios::binary);
if (!fin) {
cout << "can't open file " << argv[1] << endl;
return EXIT_FAILURE;
}
ostringstream sout;
istringstream sin;
// this is the object we will use to do the base64 encoding
base64 base64_coder;
// this is the object we will use to do the data compression
compress_stream::kernel_1ea compressor;
// compress the contents of the file and store the results in the string stream sout
compressor.compress(fin,sout);
sin.str(sout.str());
sout.clear();
sout.str("");
// now base64 encode the compressed data
base64_coder.encode(sin,sout);
sin.clear();
sin.str(sout.str());
sout.str("");
// the following is a little funny looking but all it does is output some C++ code
// that contains the compressed/base64 data and the C++ code that can decode it back
// into its original form.
sout << "#include <sstream>\n";
sout << "#include <dlib/compress_stream.h>\n";
sout << "#include <dlib/base64.h>\n";
sout << "\n";
sout << "// This function returns the contents of the file '" << argv[1] << "'\n";
sout << "const std::string get_decoded_string()\n";
sout << "{\n";
sout << " dlib::base64 base64_coder;\n";
sout << " dlib::compress_stream::kernel_1ea compressor;\n";
sout << " std::ostringstream sout;\n";
sout << " std::istringstream sin;\n\n";
sout << " // The base64 encoded data from the file '" << argv[1] << "' we want to decode and return.\n";
string temp;
getline(sin,temp);
while (sin && temp.size() > 0)
{
sout << " sout << \"" << temp << "\";\n";
getline(sin,temp);
}
sout << "\n";
sout << " // Put the data into the istream sin\n";
sout << " sin.str(sout.str());\n";
sout << " sout.str(\"\");\n\n";
sout << " // Decode the base64 text into its compressed binary form\n";
sout << " base64_coder.decode(sin,sout);\n";
sout << " sin.clear();\n";
sout << " sin.str(sout.str());\n";
sout << " sout.str(\"\");\n\n";
sout << " // Decompress the data into its original form\n";
sout << " compressor.decompress(sin,sout);\n\n";
sout << " // Return the decoded and decompressed data\n";
sout << " return sout.str();\n";
sout << "}\n";
// finally output our encoded data and its C++ code to the screen
cout << sout.str() << endl;
}