mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
acd498bf25
Restrict the type of pipeline options to integers and strings. Rework the CLOpt mechanism to avoid parsing the string twice.
91 lines
2.1 KiB
C++
91 lines
2.1 KiB
C++
#pragma once
|
|
|
|
//
|
|
// This file is distributed under the MIT License. See LICENSE.md for details.
|
|
//
|
|
|
|
#include <cstddef>
|
|
#include <tuple>
|
|
#include <utility>
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
|
|
#include "revng/ADT/Concepts.h"
|
|
|
|
namespace pipeline {
|
|
template<typename T>
|
|
requires(anyOf<T, int, uint64_t, const char *>())
|
|
class Option {
|
|
public:
|
|
constexpr Option(const char *Name, T Default) :
|
|
Default(std::forward<T>(Default)), Name(Name) {}
|
|
|
|
using Type = std::decay_t<T>;
|
|
T Default;
|
|
const char *Name;
|
|
};
|
|
|
|
template<typename T>
|
|
concept HasOptions = requires(T) {
|
|
{ T::Options };
|
|
};
|
|
|
|
namespace detail {
|
|
|
|
template<typename Invokable, size_t Index>
|
|
constexpr auto &getOptionInfo() {
|
|
return std::get<Index>(Invokable::Options);
|
|
}
|
|
|
|
template<typename Invokable, size_t Index>
|
|
llvm::StringRef getOptionName() {
|
|
return getOptionInfo<Invokable, Index>().Name;
|
|
}
|
|
|
|
template<typename T>
|
|
llvm::StringRef typeNameImpl();
|
|
|
|
template<>
|
|
inline llvm::StringRef typeNameImpl<std::string>() {
|
|
return "string";
|
|
}
|
|
|
|
template<>
|
|
inline llvm::StringRef typeNameImpl<int>() {
|
|
return "int";
|
|
}
|
|
|
|
template<>
|
|
inline llvm::StringRef typeNameImpl<uint64_t>() {
|
|
return "uint64_t";
|
|
}
|
|
|
|
template<typename T, size_t Index>
|
|
using ArgTypeImpl = std::decay_t<decltype(getOptionInfo<T, Index>())>;
|
|
|
|
template<typename InvokableType, size_t Index>
|
|
using OptionTypeImpl = typename ArgTypeImpl<InvokableType, Index>::Type;
|
|
|
|
template<typename InvokableType, size_t Index>
|
|
constexpr bool
|
|
IsConstCharPtr = std::is_same_v<OptionTypeImpl<InvokableType, Index>,
|
|
const char *>;
|
|
|
|
template<typename InvokableType, size_t Index>
|
|
using OptionType = std::conditional_t<IsConstCharPtr<InvokableType, Index>,
|
|
std::string,
|
|
OptionTypeImpl<InvokableType, Index>>;
|
|
|
|
template<typename Invokable, size_t Index>
|
|
llvm::StringRef getTypeName() {
|
|
return typeNameImpl<OptionType<Invokable, Index>>();
|
|
}
|
|
|
|
template<typename InvokableType, size_t Index>
|
|
OptionType<InvokableType, Index> getOptionDefault() {
|
|
return getOptionInfo<InvokableType, Index>().Default;
|
|
}
|
|
|
|
} // namespace detail
|
|
} // namespace pipeline
|