1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120
|
#ifndef __BTANKS_MATRIX_H__
#define __BTANKS_MATRIX_H__
/* Battle Tanks Game
* Copyright (C) 2006-2009 Battle Tanks team
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
/*
* Additional rights can be granted beyond the GNU General Public License
* on the terms provided in the Exception. If you modify this file,
* you may extend this exception to your version of the file,
* but you are not obligated to do so. If you do not wish to provide this
* exception without modification, you must delete this exception statement
* from your version and license this file solely under the GPL without exception.
*/
#include "mrt/chunk.h"
#include "mrt/exception.h"
template <class T> class Matrix {
public:
Matrix() : _data(), _w(0), _h(0), _use_default(false), _default() {}
Matrix(const int h, const int w, const T v = 0): _use_default(false) {
set_size(h, w, v);
}
void useDefault(const T d) { _default = d; _use_default = true; }
void fill(const T v) {
T *ptr = (T*) _data.get_ptr();
for(int i = 0; i < _w * _h; ++i) {
*ptr++ = v;
}
}
void set_size(const int h, const int w, const T v = 0) {
_w = w;
_h = h;
_data.set_size(w * h * sizeof(T));
fill(v);
}
inline const T get(const int y, const int x) const {
if (x < 0 || x >= _w || y < 0 || y >= _h) {
if (_use_default)
return _default;
throw_ex(("get(%d, %d) is out of bounds", y, x));
}
register int idx = y * _w + x;
register const T *ptr = (const T*) _data.get_ptr();
return *(ptr + idx);
}
inline void set(const int y, const int x, const T v) {
if (x < 0 || x >= _w || y < 0 || y >= _h) {
if (_use_default)
return;
throw_ex(("set(%d, %d) is out of bounds", y, x));
}
register int idx = y * _w + x;
register T *ptr = (T*) _data.get_ptr();
*(ptr + idx) = v;
}
inline const int get_width() const { return _w; }
inline const int get_height() const { return _h; }
const std::string dump() const {
//fixme: add template functions for conversion int/float other types to string
std::string result;
result += " ";
for(int x = 0; x < _w; ++x)
result += mrt::format_string("%-2d ", x);
result += "\n";
for(int y = 0; y < _h; ++y) {
result += mrt::format_string("%-2d ", y);
result += "[ ";
for(int x = 0; x < _w; ++x) {
result += mrt::format_string("%-2d ", (int)get(y, x));
}
result += " ]";
result += mrt::format_string("%-2d\n", y);
}
result += " ";
for(int x = 0; x < _w; ++x)
result += mrt::format_string("%-2d ", x);
result += "\n";
return result;
}
protected:
mrt::Chunk _data;
int _w, _h;
bool _use_default;
T _default;
};
#endif
|