I am facing a little problem that I can't solve on my own. I am writing a program which implements very simple operations on matrixes. The problem is that when I am trying to do something like this:
Matrix first(5);
Matrix e;
first.identityMatrix();
e = first;
cout << first;
cout << e;
Short explanation: I want to assign square matrix to a matrix without dimensions.
Second cout doesn't show anything. But when I change Matrix e() to Matrix e(5), everything works perfectly. I know that the bug resides in this piece of code:
Matrix& Matrix::operator=(const Matrix& tmp)
{
if (this->a==0 && this->b == 0)
{
this->matrix = new double*[tmp.a];
for (int i=0;i<tmp.a;i++)
this->matrix[i] = new double[tmp.b];
} else {
try {
if (this->a!=tmp.a || this->b!=tmp.b)
throw wrongSize();
} catch (wrongSize& e) {
e.message();
throw;
}
}
for (int i=0;i<tmp.a;i++)
{
for (int j=0;j<tmp.b;j++)
{
this->matrix[i][j] = tmp.matrix[i][j];
}
}
return *this;
}
After some attempts I guess that something is wrong with memory allocation, but I am not sure. To me it ought to work correctly due to the fact that I return the reference to the current object. I think that constructors might be useful as well:
Matrix::Matrix()
{
a = 0;
b = 0;
matrix = NULL;
}
Matrix::Matrix(int a)
{
try {
if (a==0)
throw wrongRowOrColNumber();
} catch (wrongRowOrColNumber& e) {
e.message();
throw;
}
this->a = a;
this->b = a;
this->matrix = new double*[a];
for (int i=0;i<a;i++)
matrix[i] = new double[a];
for (int i=0;i<a;i++)
for (int j=0;j<a;j++)
matrix[i][j] = 0;
}
Matrix::Matrix(int a, int b)
{
try {
if (a==0 || b==0)
throw wrongRowOrColNumber();
} catch (wrongRowOrColNumber& e) {
e.message();
throw;
}
if (a==b)
{
try {
if (a==0)
throw wrongRowOrColNumber();
} catch (wrongRowOrColNumber& e) {
e.message();
throw;
}
this->a = a;
this->b = a;
this->matrix = new double*[a];
for (int i=0;i<a;i++)
matrix[i] = new double[a];
for (int i=0;i<a;i++)
for (int j=0;j<a;j++)
matrix[i][j] = 0;
} else {
this->a = a;
this->b = b;
this->matrix = new double*[a];
for (int i=0;i<a;i++)
matrix[i] = new double[b];
for (int i=0;i<a;i++)
for (int j=0;j<b;j++)
matrix[i][j] = 0;
}
}
Operator <<:
friend ostream& operator<<(ostream& buffer, const Matrix& tmp)
{
for (int i=0;i<tmp.a;i++)
{
for (int j=0;j<tmp.b;j++)
{
buffer << tmp.matrix[i][j] << " ";
}
buffer << endl;
}
return buffer;
};
IdentityMatrix:
Matrix& Matrix::identityMatrix()
{
try {
if (this->a!=this->b)
{
throw wrongSize();
}
} catch (wrongSize& e) {
e.message();
throw wrongSize();
}
int row = this->a;
for (int i=0;i<row;i++)
{
for (int j=0;j<row;j++)
{
if (i==j)
this->matrix[i][j] = 1;
else
this->matrix[i][j] = 0;
}
}
return *this;
}