模板类中的友元函数

模板类中的友元函数

模板类中友元函数不要滥用，最好只用来重载左移和右移运算符("<<"和">>")，如果用于普通友元函数重载, 会非常麻烦，即使类模板定义和声明分开，也建议都写在.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
请按任意键继续. . .