drewcassidy/quicktex
1
0
Fork 0
mirror of https://github.com/drewcassidy/quicktex.git synced 2024-09-13 06:37:34 +00:00
Code Issues Projects Releases Activity

ColorSet extension/alias for Matrix

Browse Source
This commit is contained in:
Andrew Cassidy 2022-06-02 23:45:53 -07:00
parent d3515c1db8
commit f88212af85
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
quicktex/Matrix.h
View File

@ -20,6 +20,7 @@
#pragma once
#include <array>
#include <type_traits>
#include "Vec.h"
@ -43,8 +44,27 @@ template <typename T, size_t M, size_t N> class Matrix : Vec<Vec<T, N>, M> {
std::array<T *, M> get_column_ptrs(size_t index) const {
std::array<T *, M> ptrs;
for (unsigned i = 0; i < M; i++) { ptrs[i] = &(row(i)[index]); }
for (unsigned m = 0; m < M; m++) { ptrs[m] = &(row(m)[index]); }
return ptrs;
}
Matrix<T, N, M> transpose() {
Matrix<T, N, M> res;
for (unsigned m = 0; m < M; m++) { res.set_column(m, row(m)); }
return res;
}
};
/**
* Extension of Matrix with some helper aliases for use as a vector of channels
*/
template <typename T, size_t M, size_t N> class ChannelSet : Matrix<T, M, N> {
Vec<T, N> &channel(size_t m) { return this->row(m); }
const Vec<T, N> &channel(size_t m) const { return this->row(m); }
Vec<T, M> get_pixel(size_t n) { return this->get_column(n); }
void set_pixel(size_t n, const Vec<T, M> &pixel) { this->set_column(n, pixel); }
};
} // namespace quicktex