nouveau fichier : include/Appli/Option.hpp
nouveau fichier : src/Appli/Option.cpp
This commit is contained in:
Git Merciol
parent
4b4bbe889d
commit
eb0e4b26cb
37
include/Appli/Option.hpp
Normal file
37
include/Appli/Option.hpp
Normal file
@ -0,0 +1,37 @@
|
||||
#ifndef _OTB_TRISKELE_OPTION_HPP
|
||||
#define _OTB_TRISKELE_OPTION_HPP
|
||||
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <boost/thread.hpp>
|
||||
|
||||
#include "triskeleBase.hpp"
|
||||
#include "IImage.hpp"
|
||||
|
||||
namespace otb {
|
||||
namespace triskele {
|
||||
using namespace ::triskele;
|
||||
|
||||
// ========================================
|
||||
|
||||
class Option {
|
||||
public:
|
||||
bool debugFlag;
|
||||
IImage inputImage;
|
||||
IImage outputImage;
|
||||
Point topLeft = NullPoint;
|
||||
Size size = NullSize;
|
||||
DimChanel chanel = 0;
|
||||
unsigned int treeCoreCount = boost::thread::hardware_concurrency ();
|
||||
|
||||
Option ();
|
||||
Option (int argc, char** argv);
|
||||
|
||||
void usage (string msg = "", bool hidden = false);
|
||||
void parse (int argc, char** argv);
|
||||
};
|
||||
|
||||
}//namespace triskele
|
||||
}//namespace otb
|
||||
|
||||
#endif //_OTB_TRISKELE_OPTION_HPP
|
||||
183
src/Appli/Option.cpp
Normal file
183
src/Appli/Option.cpp
Normal file
@ -0,0 +1,183 @@
|
||||
#define LAST_VERSION "2018-02-17 (Debian Stretch)"
|
||||
|
||||
|
||||
#include <iostream>
|
||||
#include <string.h>
|
||||
|
||||
#include <boost/program_options/options_description.hpp>
|
||||
#include <boost/program_options/variables_map.hpp>
|
||||
#include <boost/program_options/parsers.hpp>
|
||||
|
||||
#include "triskeleDebug.hpp"
|
||||
#include "triskeleBase.hpp"
|
||||
#include "IImage.hpp"
|
||||
#include "Appli/Option.hpp"
|
||||
|
||||
#ifndef TRISKELE_ERROR
|
||||
#define TRISKELE_ERROR(expr) std::cerr << expr << std::endl << std::flush, std::exit (1)
|
||||
#endif
|
||||
|
||||
using namespace std;
|
||||
using namespace triskele;
|
||||
using namespace otb::triskele;
|
||||
|
||||
// ========================================
|
||||
Option::Option () {
|
||||
}
|
||||
|
||||
Option::Option (int argc, char** argv) {
|
||||
parse (argc, argv);
|
||||
}
|
||||
|
||||
// ========================================
|
||||
namespace po = boost::program_options;
|
||||
po::options_description desc ("Allowed options");
|
||||
po::options_description hide ("Hidded options");
|
||||
char *prog = NULL;
|
||||
|
||||
// ========================================
|
||||
void
|
||||
Option::usage (string msg, bool hidden) {
|
||||
if (!msg.empty ())
|
||||
cout << msg << endl;
|
||||
cout << endl
|
||||
<<"Usage: " << endl
|
||||
<<" " << prog << " [options] [-i] inputFileName [-o] outputFileName" << endl << endl
|
||||
<< endl << desc << endl;
|
||||
if (hidden)
|
||||
cout << hide << endl;
|
||||
exit (1);
|
||||
}
|
||||
|
||||
void
|
||||
version () {
|
||||
cout << endl << prog << " version " << LAST_VERSION << endl << endl
|
||||
<< " GDAL : read and write image (http://www.gdal.org/)" << endl
|
||||
<< " cct : inspired by jirihavel library to build trees (https://github.com/jirihavel/libcct)" << endl
|
||||
<< " Boost : C++ libraries (http://www.boost.org/)" << endl
|
||||
<< endl << " This software is a Obelix team production (http://www-obelix.irisa.fr/)" << endl << endl;
|
||||
exit (0);
|
||||
}
|
||||
|
||||
// ========================================
|
||||
static const string inputFile = "input-file";
|
||||
static const char *const inputFileC = inputFile.c_str ();
|
||||
|
||||
void
|
||||
Option::parse (int argc, char** argv) {
|
||||
prog = argv [0];
|
||||
bool helpFlag = false, versionFlag = false, useTheForceLuke = false;
|
||||
string inputFileName, outputFileName;
|
||||
long left = -1, top = -1, width = -1, height = -1;
|
||||
try {
|
||||
desc.add_options ()
|
||||
("help", po::bool_switch (&helpFlag), "produce this help message")
|
||||
("version", po::bool_switch (&versionFlag), "display version information")
|
||||
("debug", po::bool_switch (&debugFlag), "debug mode")
|
||||
("band,b", po::value<DimChanel> (&chanel), (string ("select input band (first band is 0) (default ")+
|
||||
boost::lexical_cast<std::string> (chanel)+")").c_str ())
|
||||
("left,x", po::value<long> (&left), "left crop (default center)")
|
||||
("top,y", po::value<long> (&top), "top crop (default middle)")
|
||||
("width,w", po::value<long> (&width), "width crop (default input width)")
|
||||
("height,h", po::value<long> (&height), "height crop (default input height)")
|
||||
("input,i", po::value<string> (&inputFileName), "input file name image")
|
||||
("output,o", po::value<string> (&outputFileName), "output file name hyperbands image (contains attributs profiles)")
|
||||
//("min-tree", po::bool_switch (&options.noTreeFlag[minE]), "build min-tree")
|
||||
//("area,A", po::value<string> (&areaThresholdsName), "produce area attributs")
|
||||
;
|
||||
hide.add_options ()
|
||||
("use-the-force-luke", po::bool_switch (&useTheForceLuke), "display hidded options")
|
||||
("tree-core-count", po::value<unsigned int> (&treeCoreCount), "thread used to build tree (default hardware value)")
|
||||
// ("max-tree", po::bool_switch (&options.noTreeFlag[maxE]), "build max-tree")
|
||||
// ("tos-tree", po::bool_switch (&options.noTreeFlag[tosE]), "build tree-of-shape")
|
||||
// ("standard-deviation,S", po::value<string> (&sdThresholdsName), "produce standard deviation attributs")
|
||||
// ("moment-of-inertia,I", po::value<string> (&moiThresholdsName), "produce moment of inertia attributs")
|
||||
// ("no-border", po::bool_switch (&options.noBorder), "build tree with all pixels (included no-data)")
|
||||
// ("dap", po::bool_switch (&options.dapFlag), "produce DAP rather than AP")
|
||||
// ("use-all-orig", po::bool_switch (&options.useAllOrigFlag), "force use all original band")
|
||||
// ("dap-no-orig", po::bool_switch (&options.dapNoOrigFlag), "force don't use original AP band")
|
||||
// ("dap-orig-pos", po::value<unsigned int> (&apOrigPos), (string ("position of origin on DAP (default ")+
|
||||
// boost::lexical_cast<std::string> (apOrigPos)+
|
||||
// " = "+apOrigPosName [apOrigPos]+")").c_str ())
|
||||
// ("dap-orig-weight", po::bool_switch (&options.dapOrigWeightFlag), "use origin weight for dap")
|
||||
;
|
||||
|
||||
po::options_description cmd ("All options");
|
||||
cmd.add (desc).add (hide).add_options ()
|
||||
(inputFileC, po::value<vector<string> > (), "input thresholds output")
|
||||
;
|
||||
|
||||
po::positional_options_description p;
|
||||
p.add (inputFileC, -1);
|
||||
po::variables_map vm;
|
||||
po::store (po::command_line_parser (argc, argv).options (cmd).positional (p).run (), vm);
|
||||
po::store (po::command_line_parser (argc, argv).options (desc).positional (p).run (), vm);
|
||||
po::notify (vm);
|
||||
|
||||
if (useTheForceLuke)
|
||||
usage ("", true);
|
||||
if (versionFlag)
|
||||
version ();
|
||||
if (helpFlag)
|
||||
usage ();
|
||||
|
||||
int required = 2;
|
||||
if (vm.count ("input"))
|
||||
required--;
|
||||
if (vm.count ("output"))
|
||||
required--;
|
||||
|
||||
int nbArgs = 0;
|
||||
if (vm.count (inputFileC))
|
||||
nbArgs = vm[inputFileC].as<vector<string> > ().size ();
|
||||
if (required-nbArgs != 0)
|
||||
usage ("Need one input and one output");
|
||||
if (required) {
|
||||
vector<string> var = vm[inputFileC].as<vector<string> > ();
|
||||
if (outputFileName.empty ())
|
||||
outputFileName = var [--required];
|
||||
if (inputFileName.empty ())
|
||||
inputFileName = var [--required];
|
||||
}
|
||||
|
||||
if (debugFlag) {
|
||||
#ifndef ENABLE_LOG
|
||||
cout << "No debug option available. You must compile without: -DENABLE_LOG" << endl;
|
||||
#endif
|
||||
debug = true;
|
||||
}
|
||||
|
||||
} catch (...) {
|
||||
usage ("Bad options");
|
||||
}
|
||||
inputImage.setFileName (inputFileName);
|
||||
outputImage.setFileName (outputFileName);
|
||||
inputImage.readImage ();
|
||||
|
||||
GDALDataType inputType = inputImage.getDataType ();
|
||||
|
||||
int bandInputCount = inputImage.getBandCount ();
|
||||
if (!(chanel < bandInputCount))
|
||||
TRISKELE_ERROR ("chanel " << chanel << " out of " << bandInputCount << " bands!");
|
||||
|
||||
Size orgSize = inputImage.getSize ();
|
||||
if (width < 0 || width > orgSize.width)
|
||||
width = left <= 0 ? orgSize.width : min (1L, orgSize.width-left);
|
||||
if (height < 0 || height > orgSize.height)
|
||||
height = top <= 0 ? orgSize.height : min (1L, orgSize.height-top);
|
||||
if (left < 0)
|
||||
left = (orgSize.width - width)/2;
|
||||
if (top < 0)
|
||||
top = (orgSize.height - height)/2;
|
||||
if (left > orgSize.width - width)
|
||||
left = orgSize.width - width;
|
||||
if (top > orgSize.height - height)
|
||||
top = orgSize.height - height;
|
||||
topLeft = Point ((DimSideImg) left, (DimSideImg) top);
|
||||
size = Size ((DimSideImg) width, (DimSideImg) height);
|
||||
|
||||
cout
|
||||
<< "Input :" << inputFileName << " " << orgSize << " (" << bandInputCount << " chanels of " << GDALGetDataTypeName (inputType) << ")" << endl
|
||||
<< "Crop topLeft:" << topLeft << " size:" << size << " band:" << chanel << endl
|
||||
<< "Output :" << outputFileName << endl;
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user