Reformat code.

This commit is contained in:
2017-10-07 20:16:48 +02:00
parent ff94a1d3da
commit dbcb4ead51
5 changed files with 97 additions and 108 deletions

View File

@@ -1,15 +1,13 @@
#include <algorithm>
#include <cstdio>
#include <iostream>
#include <stdlib.h>
#include <unistd.h>
#include "Options.hpp"
using namespace fmri;
using namespace std;
static void show_help(const char* progname, int exitcode)
{
static void show_help(const char *progname, int exitcode) {
cerr << "Usage: " << progname << " -m MODEL -w WEIGHTS INPUTS..." << endl
<< endl
<< R"END(
@@ -24,16 +22,14 @@ Options:
exit(exitcode);
}
static void check_file(const char * filename)
{
static void check_file(const char *filename) {
if (access(filename, R_OK) != 0) {
perror(filename);
exit(1);
}
}
Options Options::parse(const int argc, char * const argv[])
{
Options Options::parse(const int argc, char *const argv[]) {
string model;
string weights;
@@ -85,24 +81,20 @@ Options Options::parse(const int argc, char * const argv[])
return Options(move(model), move(weights), move(inputs));
}
Options::Options(string&& model, string&& weights, vector<string>&& inputs) noexcept:
Options::Options(string &&model, string &&weights, vector<string> &&inputs) noexcept:
modelPath(move(model)),
weightsPath(move(weights)),
inputPaths(move(inputs))
{
inputPaths(move(inputs)) {
}
const string& Options::model() const
{
const string &Options::model() const {
return modelPath;
}
const string& Options::weights() const
{
const string &Options::weights() const {
return weightsPath;
}
const vector<string>& Options::inputs() const
{
const vector<string> &Options::inputs() const {
return inputPaths;
}

View File

@@ -8,20 +8,21 @@ namespace fmri {
using std::vector;
using std::string;
class Options
{
class Options {
public:
static Options parse(const int argc, char * const argv[]);
static Options parse(const int argc, char *const argv[]);
const string& model() const;
const string& weights() const;
const vector<string>& inputs() const;
const string &model() const;
const string &weights() const;
const vector<string> &inputs() const;
private:
const string modelPath;
const string weightsPath;
const vector<string> inputPaths;
Options(string&&, string&&, vector<string>&&) noexcept;
Options(string &&, string &&, vector<string> &&) noexcept;
};
}

View File

@@ -1,18 +1,15 @@
#include <iostream>
#include <string>
#include "Simulator.hpp"
using namespace caffe;
using namespace std;
using namespace fmri;
Simulator::Simulator(const string& model_file, const string& weights_file) :
net(model_file, TEST)
{
Simulator::Simulator(const string &model_file, const string &weights_file) :
net(model_file, TEST) {
net.CopyTrainedLayersFrom(weights_file);
}
void Simulator::simulate(const string& image_file)
{
void Simulator::simulate(const string &image_file) {
cerr << "This is not implemented yet." << endl;
}

View File

@@ -12,9 +12,9 @@ namespace fmri {
public:
typedef float DType;
Simulator(const string& model_file, const string& weights_file);
Simulator(const string &model_file, const string &weights_file);
void simulate(const string& input_file);
void simulate(const string &input_file);
private:
caffe::Net<DType> net;

View File

@@ -5,15 +5,14 @@
using namespace std;
using namespace fmri;
int main(int argc, char * const argv[])
{
int main(int argc, char *const argv[]) {
::google::InitGoogleLogging(argv[0]);
Options options = Options::parse(argc, argv);
Simulator simulator(options.model(), options.weights());
for (const auto& image : options.inputs()) {
for (const auto &image : options.inputs()) {
simulator.simulate(image);
}