src/tools/lvr2_transform/Options.cpp

Go to the documentation of this file.

 
/*
 * Options.cpp
 *
 * @date 2012-08-23
 * @author Christian Wansart <cwansart@uos.de>
 * @author Thomas Wiemann
 */
 
#include "Options.hpp"
//#include <omp.h>
 
#define vc(x) ( m_variables.count(x) )
#define srtOR ( !vc("sx") && !vc("sy") && !vc("sz") && \
                !vc("rx") && !vc("ry") && !vc("rz") && \
                !vc("tx") && !vc("ty") && !vc("tz") )
 
namespace transform {
 
Options::Options(int argc, char** argv) : m_descr("Supported options")
{
  // Create option descriptions
 
  m_descr.add_options()
    ("help", "Produce help message")
    ("inputFile,i",     value< string >(),          "Input file name.")
    ("outputFile,o",    value< string >(),          "Output file name. Don't forget to append a file extension.")
    ("transformFile,t", value< string >(),          "*.pose or *.frame file for transformation")
 
    ("sx",              value< float >(),           "Scale in x axis.")
    ("sy",              value< float >(),           "Scale in y axis.")
    ("sz",              value< float >(),           "Scale in z axis.")
//    ("scale,s",         value< vector< float >>() "Scale in all axis. (Format: x,y,z; e.g. --scale 1.0,4.2,0.4)")
 
    ("rx",              value< float >(),           "Rotation in x axis.")
    ("ry",              value< float >(),           "Rotation in y axis.")
    ("rz",              value< float >(),           "Rotation in z axis.")
//    ("rotation",        value< vector< float >>() "Rotation in all axis. (Format: x,y,z; e.g. --rotation 4.0,5.0,0.0)")
 
    ("tx",              value< float >(),           "Transform in x axis.")
    ("ty",              value< float >(),           "Transform in y axis.")
    ("tz",              value< float >(),           "Transform in z axis.")
//    ("translate",       value< vector< float >>(),  "Transform in all axis. (Format: x,y,z; e.g.: --translate 6.0,0.0,1.2)")
    ;
 
  // Parse command line and generate variables map
  store(command_line_parser(argc, argv).options(m_descr).positional(m_pdescr).run(), m_variables);
  notify(m_variables);
 
  if(m_variables.count("help")) {
    cout<< m_descr << endl;
  }
 
}
 
string Options::getInputFile() const
{
  return (m_variables["inputFile"].as< string >());
}
 
string Options::getOutputFile() const
{
  return (m_variables["outputFile"].as< string >());
}
 
string Options::getTransformFile() const
{
  return (m_variables["transformFile"].as< string >());
}
 
bool Options::anyTransformFile() const
{
  return (m_variables.count("transformFile"));
}
 
float Options::getScaleX() const
{
  return (m_variables["sx"].as< float >());
}
 
float Options::getScaleY() const
{
  return (m_variables["sy"].as< float >());
}
 
float Options::getScaleZ() const
{
  return (m_variables["sz"].as< float >());
}
 
float Options::getRotationX() const
{
  return (m_variables["rx"].as< float >());
}
 
float Options::getRotationY() const
{
  return (m_variables["ry"].as< float >());
}
 
float Options::getRotationZ() const
{
  return (m_variables["rz"].as< float >());
}
 
float Options::getTranslationX() const
{
  return (m_variables["tx"].as< float >());
}
 
float Options::getTranslationY() const
{
  return (m_variables["ty"].as< float >());
}
 
float Options::getTranslationZ() const
{
  return (m_variables["tz"].as< float >());
}
 
bool Options::anyScale() const
{
  return (anyScaleX() || anyScaleY() || anyScaleZ());
}
 
bool Options::anyScaleX() const
{
  return (m_variables.count("sx"));
}
 
bool Options::anyScaleY() const
{
  return (m_variables.count("sy"));
}
 
bool Options::anyScaleZ() const
{
  return (m_variables.count("sz"));
}
 
bool Options::anyRotation() const
{
  return (anyRotationX() || anyRotationY() || anyRotationZ());
}
 
bool Options::anyRotationX() const
{
  return (m_variables.count("rx"));
}
 
bool Options::anyRotationY() const
{
  return (m_variables.count("ry"));
}
 
bool Options::anyRotationZ() const
{
  return (m_variables.count("rz"));
}
 
bool Options::anyTranslation() const
{
  return (anyTranslationX() || anyTranslationY() || anyTranslationZ());
}
 
bool Options::anyTranslationX() const
{
  return (m_variables.count("tx"));
}
 
bool Options::anyTranslationY() const
{
  return (m_variables.count("ty"));
}
 
bool Options::anyTranslationZ() const
{
  return (m_variables.count("tz"));
}
 
bool Options::printUsage() const
{
  if(!m_variables.count("inputFile"))
  {
    cout << "Error: You must specify an input file." << endl;
    cout << endl;
    cout << m_descr << endl;
    return true;
  }
  
  if(!m_variables.count("outputFile"))
  {
    cout << "Error: You must specify an output file." << endl;
    cout << endl;
    cout << m_descr << endl;
    return true;
  }
 
  // TODO:  Need to check for s, t or r
  //        add notice that s, t and r will be ignored if transformFile is given
  if(!m_variables.count("transformFile") && srtOR)
  {
    cout << "Error: You must specify a transform file or sx, sy, sz, rx, ry, rz, tx, ty, tz." << endl;
    cout << endl;
    cout << m_descr << endl;
    return true;
  }
 
  if(m_variables.count("help"))
  {
    cout << endl;
    cout << m_descr << endl;
    return true;
  }
  return false;
}
 
Options::~Options() {
}
 
} // namespace transform