mystr.cpp

来自「矩阵类 简单的矩阵类的编写」· C++ 代码 · 共 67 行

// MyStr.cpp: implementation of the MyStr class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "MyStr.h"
#include "string.h"
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

MyStr::MyStr()
{
	this->m_pStr = NULL;
}

MyStr MyStr::operator =(MyStr& str)
{
	delete []this->m_pStr;
	char *p = str.m_pStr;
	if(p)
	{
		int n = strlen(p);
		m_pStr = new char[n+1];
		strcpy(m_pStr, str.m_pStr);
	}
	else
		m_pStr = NULL;

	return *this;
}

MyStr::MyStr(MyStr &str)
{
	char *p = str.m_pStr;
	if(p)
	{
		int n = strlen(p);
		m_pStr = new char[n+1];
		strcpy(m_pStr, str.m_pStr);
	}
	else
		m_pStr = NULL;
}

MyStr::MyStr(char *p)
{
	int n = strlen(p);
	m_pStr = new char[n+1];
	strcpy(m_pStr, p);
}

MyStr::~MyStr()
{
	delete []m_pStr;
}

#define MaxSize 256
istream & operator >>( istream & input, MyStr& str)
{
	char s[MaxSize];
	input >> s;
	int n = strlen(s);
	str.m_pStr = new char[n+1];
	strcpy(str.m_pStr, s);
	return input;
}