Files
dorayme/source/include/matrix.h
2020-02-14 17:49:51 +00:00

101 lines
2.0 KiB
C++

/*
* DoRayMe - a quick and dirty Raytracer
* Matrix header
*
* Created by Manoël Trapier
* Copyright (c) 2020 986-Studio.
*
*/
#ifndef DORAYME_MATRIX_H
#define DORAYME_MATRIX_H
#include <tuples.h>
class Matrix
{
private:
/* 4x4 is the default */
double data[4*4];
int width;
public:
Matrix(int width) : width(width)
{
int i;
for(i = 0; i < width*width; i++)
{
this->data[i] = 0;
}
};
Matrix(double values[], int width)
{
int x, y;
this->width = width;
for(y = 0; y < this->width; y++)
{
for (x = 0 ; x < this->width ; x++)
{
this->data[this->width * x + y] = values[this->width * x + y];
}
}
};
double get(int x, int y) const { return this->data[this->width * x + y]; };
void set(int x, int y, double v) { this->data[this->width * x + y] = v; };
Matrix identity()
{
int i;
for(i = 0; i < this->width; i++)
{
this->set(i, i, 1);
}
return *this;
}
Matrix transpose()
{
int x, y;
Matrix ret = Matrix(this->width);
for(y = 0; y < this->width; y++)
{
for (x = 0 ; x < this->width ; x++)
{
ret.set(y, x, this->get(x, y));
}
}
return ret;
}
bool operator==(const Matrix &b) const;
bool operator!=(const Matrix &b) const;
Matrix operator*(const Matrix &b) const;
Tuple operator*(const Tuple &b) const;
};
class Matrix4: public Matrix
{
public:
Matrix4() : Matrix(4) { };
Matrix4(double values[]) : Matrix(values, 4) { };
};
class Matrix2 : public Matrix
{
public:
Matrix2() : Matrix(2) { };
Matrix2(double values[]) : Matrix(values, 2) { };
};
class Matrix3 : public Matrix
{
public:
Matrix3() : Matrix(3) { };
Matrix3(double values[]) : Matrix(values, 3) { };
};
#endif /* DORAYME_MATRIX_H */