TMSubVec diag(const TMat &m, Int diagNum) { CheckRange(diagNum, 1 - m.Rows(), m.Cols(), "(row(Mat)) illegal row index"); if (diagNum == 0) return(TMSubVec(Min(m.Rows(), m.Cols()), m.Cols() + 1, m.Ref())); else if (diagNum < 0) return(TMSubVec(Min(m.Rows() + diagNum, m.Cols()), m.Cols() + 1, m.Ref() - diagNum * m.Cols())); else return(TMSubVec(Min(m.Cols() - diagNum, m.Rows()), m.Cols() + 1, m.Ref() + diagNum)); }
TSubMat sub(const TMat &m, Int nrows, Int ncols) { Assert(ncols > 0 && nrows > 0 && nrows <= m.Rows() && ncols <= m.Cols(), "(sub(Mat)) illegal subset of matrix"); TSubMat result(nrows, ncols, m.Cols(), m.Ref()); return(result); }
TSubMat sub(const TMat &m, Int top, Int left, Int height, Int width) { Assert(left >= 0 && width > 0 && left + width <= m.Cols(), "(sub(Mat)) illegal subset of matrix"); Assert(top >= 0 && height > 0 && top + height <= m.Rows(), "(sub(Mat)) illegal subset of matrix"); TSubMat result(height, width, m.Cols(), m.Ref() + top * m.Cols() + left); return(result); }
TMSubVec col(const TMat &m, Int i) { CheckRange(i, 0, m.Cols(), "(col(Mat)) illegal column index"); return(TMSubVec(m.Rows(), m.Cols(), m.Ref() + i)); }