模板类中的友元函数
模板类中友元函数不要滥用,最好只用来重载左移和右移运算符("<<"和">>"),如果用于普通友元函数重载, 会非常麻烦,即使类模板定义和声明分开,也建议都写在.h或.hpp文件中,即要保证模板类的定义和声明在一个文件中,否则,将不能找到函数的定义,这和模板的二次编译有关。参见官方的STL库风格可知,它也遵循这一原则,即类模板的定义和声明都写在一个文件中,由于包含类模板定义,文件后缀理应为.cpp,但使用模板又要#include此cpp文件,所以业界一般将后缀改为.hpp。
complex.hpp文件
#pragma once
#include <iostream>
#include <cstdlib>
using namespace std;
//非运算符重载的友元函数如果将定义和声明分开写的话,必须要进行以下四句声明,
//否则出错(运算符重载的友元函数除外),所以不建议滥用友元函数,特别是在类模板中
//一般我们的使用原则是除了重载左移和右移运算符,不要用其他类型的友元函数
template <typename T>
class Complex;
template<typename T>
Complex<T> SubCom(const Complex<T> &obj1, const Complex<T> &obj2);
template <typename T>
class Complex
{
//注意语法,operator后面要加<T> 所有模板类中的友元函数声明都要这么写,否则出错
//除非你把函数的定义写在类的内部
friend ostream & operator<< <T>(ostream &out, const Complex &obj);
friend Complex SubCom <T>(const Complex &obj1, const Complex &obj2);
public:
Complex(T i, T j);
void printComplex();
Complex operator+(Complex & obj);
private:
T i;
T j;
};
template<typename T>
Complex<T>::Complex(T i, T j) :i(i), j(j)
{ }
template<typename T>
void Complex<T>::printComplex()
{
cout << i << " + " << j << "i" << endl;
}
template<typename T>
Complex<T> Complex<T>::operator+(Complex<T> & obj)
{
Complex temp(this->i + obj.i, this->j + obj.j);
return temp;
}
template<typename T>
ostream & operator<<(ostream &out, const Complex<T> & obj)
{
out << obj.i << " + " << obj.j << "i" << endl;
return out;
}
//滥用 友元函数
template <typename T>
Complex<T> SubCom(const Complex<T> &obj1, const Complex<T> &obj2)
{
Complex<T> tmp(obj1.i - obj2.i, obj1.j - obj2.j);
return tmp;
}
/*
请注意最重要的两句:
1、除了左移和右移运算符重载不要使用其他友元函数
2、即使类模板的定义和声明分开来写,也要放在同一个文件中,建议都放在.h或者.hpp文件中(官方的STL模板都是这么做的)
*/
注意第23,24行友元函数声明的写法,<T>所放的位置,否则模板类的二次编译会导致编译出错
main.cpp文件
#include <iostream>
#include "complex.hpp"
using namespace std;
int main()
{
Complex<int> c1(1, 2);
Complex<int> c2(2, 3);
Complex<int> c3 = c2 + c1;
c1.printComplex();
c2.printComplex();
c3.printComplex();
cout << c1 << c2 << c3;
c3 = SubCom<int>(c1, c2);
cout << c3;
system("pause");
return 0;
}
运行结果
1 + 2i
2 + 3i
3 + 5i
1 + 2i
2 + 3i
3 + 5i
-1 + -1i
请按任意键继续. . .