2011-04-12 58 views
8

如何从流输入枚举类型?输入从流到枚举类型

我能做到这样

unsigned int sex = 0; 
stream >> sex; 
student.m_bio.sex = static_cast<Sex>(sex); 

否则?

回答

13
inline std::istream & operator>>(std::istream & str, Sex & v) { 
    unsigned int sex = 0; 
    if (str >> sex) 
    v = static_cast<Sex>(sex); 
    return str; 
} 

如果你想确保该值是有效的,你可以做这样的事情:

enum Sex { 
    Male, 
    Female, 
    Sex_COUNT 
}; 

inline std::istream & operator>>(std::istream & str, Sex & v) { 
    unsigned int sex = 0; 
    if (!(str >> sex)) 
    return str; 
    if (sex >= Sex_COUNT) { 
    str.setstate(str.rdstate() | std::ios::failbit); 
    return str; 
    } 
    v = static_cast<Sex>(sex); 
    return str; 
} 
+0

我认为你的意思'V =的static_cast (性);'......也就是说,参数的演员是“性”,而不是“v”。 :D – Nawaz 2011-04-12 10:34:02

+0

@Nawaz:是的,更正了,谢谢。 – Erik 2011-04-12 10:35:00

+0

这个答案缺少'return str;'。 – 2011-04-12 10:35:04

3

这个问题是问一个更一般的形式在这里:How to read enums from a std::istream in a generic fashion。 OP几乎有一个工作的解决方案,因为它是;他在const和一些不必要的尖括号中遇到了一些麻烦。这里的工作解决方案跃然纸上:

#include <iostream> 
#include <type_traits> 

enum enSide { eLeft, eRight }; 
enum enType { eConUndefined, eConRoom }; 

template<typename Enum> 
class EnumReader 
{ 
    Enum& e_; 

    friend std::istream& operator>>(std::istream& in, const EnumReader& val) { 
     typename std::underlying_type<Enum>::type asInt; 
     if (in >> asInt) val.e_ = static_cast<Enum>(asInt); 
     return in; 
    } 
    public: 
    EnumReader(Enum& e) : e_(e) {} 
}; 

template<typename Enum> 
EnumReader<Enum> read_enum(Enum& e) 
{ 
    return EnumReader<Enum>(e); 
} 

class MyClass { 
    enSide mSide; 
    enType mType; 
    int mTargetId; 
    public: 
    friend std::istream& operator>>(std::istream& in, MyClass& val) { 
     in >> read_enum(val.mSide) >> read_enum(val.mType) >> val.mTargetId; 
     return in; 
    } 
}; 

read_enum函数模板这里有异曲同工之妙的std::make_pair或标准库std::make_shared:它让我们用尖括号免除。你同样可以写in >> EnumReader<enSide>(val.mSide) >> EnumReader<enType>(val.mType),但这是更多的打字(双关意图)。

据称一些供应商的标准库仍然从其<type_traits>标头中缺少std::underlying_type。如果您有其中一个不完整的库,则可以使用How to know underlying type of class enum?中列出的其中一种解决方法。

0

这是不漂亮,但要做到这一点

stream >> reinterpret_cast<std::underlying_type<Sex>::type &>(student.m_bio.sex); 

干杯, CC