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));
}
