stringprintf string반환 sprintf 2

여기서는 %로 받는 형식문자열을 추론해서 알아본다.

template <typename _Any>
struct is_c_style_string : public false_type{};
template <>
struct is_c_style_string<const char*> : public true_type{};
template <>
struct is_c_style_string<char*> : public true_type{};
template <>
struct is_c_style_string<const char[]> : public true_type{};
template <>
struct is_c_style_string<char[]> : public true_type{};

이런 구조체를 추가하여 c_style의 문자열인지 알아낼 수 있도록 한다.

 
template<typename _Enum, typename = typename std::enable_if<std::is_enum<_Enum>::value>::type>
inline std::ostringstream& operator<<(std::ostringstream& o, _Enum e){
    o << ((long long)e);
    return o;
}
 
std::string stringprintf_internal(std::stringstream&& os, const char* str){
    for(; *str; str++){
        if(*str == '%' && *(++str) != '%'){
            //stringprintf("%d")같은 상황을 체크했을때
            //throw std::runtime_error("%d");
        }
        else{
            os << *str;
        }
    }
    return os.str();
}
 
template<typename Arg, typename... Args>
std::string stringprintf_internal(std::stringstream&& os, const char* str, Arg&& arg, Args&&... args){
    for(; *str; str++){
        //%%는 %
        //그외에는 Arg가 들어감
        if(*str == '%'){
            const char nextChar = *(++str);
            switch(nextChar){
            case 'u':
            case 'c':
            case 'd':
                if(std::is_integral<Arg>() == false){
                    throw std::runtime_error("bad format specifier");
                }
                break;
            case 'f':
                if(std::is_floating_point<Arg>() == false){
                    throw std::runtime_error("bad format specifier");
                }
                break;
            case 's':
                if(is_c_style_string<Arg>() == false){
                    throw std::runtime_error("bad format specifier");
                }
                break;
            }
            
            os << arg;
            
            return stringprintf_internal(std::move(os), ++str, std::forward<Args>(args)...);
        }
        else{
            os << *str;
        }
    }
    
    //stringprintf("abcde", 1,2,3,4,5); 같이 들어왔을때 던질려면
    //throw std::invalid_argument("argument");
    return os.str();
}
 
template<typename... Args>
std::string stringprintf(const char* str, Args&&... args) {
    if(str == nullptr){
        // null str이 들어왔을때 던질려면
        //throw std::invalid_argument("null");
        return std::string("");
    }
    std::stringstream os;
    return stringprintf_internal(std::move(os), str, std::forward<Args>(args)...);
}
 

테스트

#include <iostream>
using namespace std;
int main(int argc, const char * argv[]) {
    auto s = stringprintf("%d 1%d23 %d %% 4%d 5%d6 test %d 1%%", 1,2,3,4,5,6);
    cout << s<<endl;
}