C++ 運算符重載（二）

關于輸入（cin）/輸出（cout）的重載。

在C++的頭文件中有 #include <iostream> ，實際上就包含著cin/cout，具體上：ostream中對應的是cout，而istream對應的是cin。

我先實現cout重載

#include <iostream>
using namespace std;
class Oc
{
public:
private:
    int cnt;
public:
    Oc( int cnt)
    {
        this->cnt = cnt;
    }
    friend ostream& operator << (ostream& ost , const Oc& oc);
};
ostream& operator << (ostream& ost , const Oc& oc)
{
    ost << oc.cnt;
    return ost;
}
int main()
{
    system("color 5A");
    Oc oc(2);
    cout << oc;
    return 0;
}

運行結果：

解釋：

①：  參數 ostream& ost , const Oc& oc 以及返回ostream& 都是使用引用 防止調用拷貝構造

②：  可以使用void返回 ， 但是這樣無法使用連續的輸出 << oc << XXX

③：  左操作數必須是ostream對象，所以此重載必須在類外實現

cin重載：

#include <iostream>
using namespace std;
class Oc
{
private:
    int cnt;
public:
    Oc( int cnt)
    {
        this->cnt = cnt;
    }
    friend istream& operator >>( istream& ist , Oc& oc);
    friend int main();
};
istream& operator >>( istream& ist , Oc& oc)
{
    ist >> oc.cnt;
    if( ist.fail() )
    {
        cout << "fail value to Oc.cnt!" << endl;
        oc.cnt = 0;
    }
    return ist;
}
int main()
{
    system("color 5A");
    Oc oc(2);
    cin >> oc;
    cout << "Oc.cnt value is : " << oc.cnt << endl;
    return 0;
}

①，我先輸入正確的值如 1 ， 其結果如下：

②，在輸入一個錯誤的值，字符串。結果如下：

ist.fail() 就是用來檢測是否是輸入失敗的。字符串無法給int賦值。所以---

cin和cout一樣只能在類外實現