ConfigParser.hpp

#ifndef CONFIGPARSER_HPP_
#define CONFIGPARSER_HPP_

#include "io/Configuration.hpp"
#include <istream>
#include <string>

namespace PV {

class ConfigParser {
  public:
   ConfigParser(std::istream &configStream, bool allowUnrecognizedArguments);

   bool getAllowUnrecognizedArguments() const { return mAllowUnrecognizedArguments; }

   bool const &getBooleanArgument(std::string const &name) const {
      return mConfig.getBooleanArgument(name);
   }

   int const &getIntegerArgument(std::string const &name) const {
      return mConfig.getIntegerArgument(name);
   }

   unsigned int const &getUnsignedIntArgument(std::string const &name) const {
      return mConfig.getUnsignedIntArgument(name);
   }

   std::string getStringArgument(std::string const &name) const {
      return mConfig.getStringArgument(name);
   }

   Configuration::IntOptional getIntOptionalArgument(std::string const &name) const {
      return mConfig.getIntOptionalArgument(name);
   }

   std::string printConfig() const { return mConfig.printConfig(); }

   static std::string createString(
         bool requireReturnFlag,
         std::string const &outputPath,
         std::string const &paramsFile,
         std::string const &logFile,
         std::string const &gpuDevices,
         unsigned int randomSeed,
         std::string const &workingDir,
         bool restartFlag,
         std::string const &checkpointReadDir,
         bool useDefaultNumThreads,
         int numThreads,
         int numRows,
         int numColumns,
         int batchWidth,
         bool dryRunFlag);

   Configuration const &getConfig() const { return mConfig; }

  private:
   void initialize(std::istream &inputStream, bool allowUnrecognizedArguments);

   void handleUnrecognized(std::string const &argument, std::string const &value, int linenumber);

   std::string stripLeadingTrailingWhitespace(std::string const &inString);

  private:
   bool mAllowUnrecognizedArguments = false;
   Configuration mConfig;
};

} // end namespace PV

#endif // CONFIGPARSER_HPP_