joshua/mozaic
Archived
1
Fork 0
Code Activity

Adds program options and also adds generate_database binary

Browse source
This commit is contained in:
Joshua Moerman 2015-06-28 13:05:57 +02:00
parent 89629a0b30
commit 6ee3eb6cfb
2 changed files with 50 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
src/generate_database.cpp Normal file
View file

@ -0,0 +1,39 @@
#include <fingerprints.hpp>
#include <read_database.hpp>
#include <utilities.hpp>
extern "C" {
#include <libavformat/avformat.h>
}
#include <iostream>
using namespace std;
template <typename Metric>
void generate(const string & database_directory) {
auto const db = read_database<Metric>(database_directory);
cout << colors::green("Read database: ") << db.size() << endl;
}
int main(int argc, char * argv[]) {
if (argc != 3) {
cerr << "usage: main <dir> <type>\n";
return 1;
}
av_register_all();
av_log_set_level(AV_LOG_QUIET);
string const database_directory = argv[1];
string const type = argv[2];
if (type == "rgb") {
generate<fingerprints::rgb>(database_directory);
} else if (type == "wvlt_rgb") {
generate<fingerprints::wvlt_rgb>(database_directory);
} else {
cerr << "No known type given\n";
return 1;
}
}

18
src/main.cpp
View file

@ -10,7 +10,6 @@
#include <boost/filesystem.hpp> #include <boost/filesystem.hpp>
extern "C" { extern "C" {
#include <libavformat/avformat.h>
#include <libavcodec/avcodec.h> #include <libavcodec/avcodec.h>
#include <libavformat/avformat.h> #include <libavformat/avformat.h>
} }
@ -74,16 +73,21 @@ static void save_mozaic(Mozaiq const & mozaic, string filename){
encode_as_jpg(frame, filename); encode_as_jpg(frame, filename);
} }
int main(){ int main(int argc, char *argv[]){
if(argc != 6){
cerr << "usage: main <dir> <photo> <output> <h tiles> <v tiles>\n";
return 1;
}
av_register_all(); av_register_all();
av_log_set_level(AV_LOG_QUIET); av_log_set_level(AV_LOG_QUIET);
// TODO: use boost::program_options // TODO: use boost::program_options
string const database_directory = "basbram"; string const database_directory = argv[1];
string const filename = "basbram.jpg"; string const filename = argv[2];
string const output = "output/basbram.jpg"; string const output = argv[3];
int const h_tiles = 4*4; int const h_tiles = stoi(argv[4]);
int const v_tiles = 3*4; int const v_tiles = stoi(argv[5]);
auto const db = read_database<Metric>(database_directory); auto const db = read_database<Metric>(database_directory);
cout << colors::green("Read database: ") << db.size() << endl; cout << colors::green("Read database: ") << db.size() << endl;