* This file is part of the MindStudio project.
* Copyright (c) 2025 Huawei Technologies Co.,Ltd.
*
* MindStudio is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
*
* http://license.coscl.org.cn/MulanPSL2
*
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
* ------------------------------------------------------------------------- */
#ifndef CATLASS_TUNER_COMMAND_LINE_PARSER_H
#define CATLASS_TUNER_COMMAND_LINE_PARSER_H
#include <string>
#include <vector>
#include <map>
#include <unordered_set>
#include <algorithm>
#include <cstdint>
namespace Catlass {
#define GET_CHECK(stat, key) \
do { \
CommandLineParser::ERROR_CODE err = stat; \
if (err != CommandLineParser::ERROR_CODE::NONE) { \
LOGE("%s:%d get key --" key " failed, err: %s", __FILE__, __LINE__, \
CommandLineParser::GetErrorStr(err).data()); \
} \
} while (false)
class CommandLineParser {
public:
enum class ERROR_CODE : uint32_t {
NONE = 0,
CAST_FAILED,
KEY_NOT_EXIST,
INTEGER_OVERFLOW,
EXPECT_UNSIGNED_INTEGER,
NOT_DIGITAL_FORMAT,
END,
};
static std::string_view GetErrorStr(ERROR_CODE er)
{
static const std::string_view strs[] = {
"none",
"cast value from string failed",
"key not exist",
"integer overflow",
"expect unsigned integer",
"input is not digital format",
"unknown error"
};
auto idx = std::min(static_cast<int>(er), static_cast<int>(ERROR_CODE::END));
return strs[idx];
}
[[nodiscard]] inline bool HasKey(const std::string& key) const
{
return dataMap_.find(key) != dataMap_.end();
}
[[nodiscard]] inline bool Help() const { return help_; }
void Parse(int argc, const char* argv[]);
void PrintHelp() const;
void PrintUnusedKeys() const;
[[nodiscard]] std::vector<std::string> Keys() const;
template <typename T>
ERROR_CODE Get(const std::string& key, T &target);
private:
static std::string Trim(const std::string& str)
{
auto start = std::find_if_not(str.begin(), str.end(), [](unsigned char c) {
return std::isspace(c);
});
auto end = std::find_if_not(str.rbegin(), str.rend(), [](unsigned char c) {
return std::isspace(c);
}).base();
return (start < end) ? std::string(start, end) : "";
}
static bool IsDigitFormat(const std::string& str);
auto FindKey(const std::string& key)
{
auto it = dataMap_.find(key);
if (it != dataMap_.end()) {
usedSet_.insert(key);
}
return it;
}
std::map<std::string, std::string> dataMap_;
std::unordered_set<std::string> usedSet_;
bool help_{false};
};
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<std::string>(const std::string& key, std::string &target);
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<std::string_view>(const std::string& key,
std::string_view &target);
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<int64_t>(const std::string& key, int64_t &target);
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<uint64_t>(const std::string& key, uint64_t &target);
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<int32_t>(const std::string& key, int32_t &target);
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<uint32_t>(const std::string& key, uint32_t &target);
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<double>(const std::string& key, double &target);
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<float>(const std::string& key, float &target);
template<>
CommandLineParser::ERROR_CODE CommandLineParser::Get<bool>(const std::string& key, bool &target);
}
#endif