3dpcp/3rdparty/newmat/newmatrm.cpp

190 lines
4.6 KiB
C++
Raw Normal View History

2012-09-16 12:33:11 +00:00
//$$newmatrm.cpp rectangular matrix operations
// Copyright (C) 1991,2,3,4: R B Davies
#include "newmat.h"
#include "newmatrm.h"
#ifdef use_namespace
namespace NEWMAT {
#endif
#ifdef DO_REPORT
#define REPORT { static ExeCounter ExeCount(__LINE__,12); ++ExeCount; }
#else
#define REPORT {}
#endif
// operations on rectangular matrices
void RectMatrixRow::Reset (const Matrix& M, int row, int skip, int length)
{
REPORT
RectMatrixRowCol::Reset
( M.Store()+row*M.Ncols()+skip, length, 1, M.Ncols() );
}
void RectMatrixRow::Reset (const Matrix& M, int row)
{
REPORT
RectMatrixRowCol::Reset( M.Store()+row*M.Ncols(), M.Ncols(), 1, M.Ncols() );
}
void RectMatrixCol::Reset (const Matrix& M, int skip, int col, int length)
{
REPORT
RectMatrixRowCol::Reset
( M.Store()+col+skip*M.Ncols(), length, M.Ncols(), 1 );
}
void RectMatrixCol::Reset (const Matrix& M, int col)
{
REPORT
RectMatrixRowCol::Reset( M.Store()+col, M.Nrows(), M.Ncols(), 1 );
}
Real RectMatrixRowCol::SumSquare() const
{
REPORT
long_Real sum = 0.0; int i = n; Real* s = store; int d = spacing;
// while (i--) { sum += (long_Real)*s * *s; s += d; }
if (i) for(;;)
{ sum += (long_Real)*s * *s; if (!(--i)) break; s += d; }
return (Real)sum;
}
Real RectMatrixRowCol::operator*(const RectMatrixRowCol& rmrc) const
{
REPORT
long_Real sum = 0.0; int i = n;
Real* s = store; int d = spacing;
Real* s1 = rmrc.store; int d1 = rmrc.spacing;
if (i!=rmrc.n)
{
Tracer tr("newmatrm");
Throw(InternalException("Dimensions differ in *"));
}
// while (i--) { sum += (long_Real)*s * *s1; s += d; s1 += d1; }
if (i) for(;;)
{ sum += (long_Real)*s * *s1; if (!(--i)) break; s += d; s1 += d1; }
return (Real)sum;
}
void RectMatrixRowCol::AddScaled(const RectMatrixRowCol& rmrc, Real r)
{
REPORT
int i = n; Real* s = store; int d = spacing;
Real* s1 = rmrc.store; int d1 = rmrc.spacing;
if (i!=rmrc.n)
{
Tracer tr("newmatrm");
Throw(InternalException("Dimensions differ in AddScaled"));
}
// while (i--) { *s += *s1 * r; s += d; s1 += d1; }
if (i) for (;;)
{ *s += *s1 * r; if (!(--i)) break; s += d; s1 += d1; }
}
void RectMatrixRowCol::Divide(const RectMatrixRowCol& rmrc, Real r)
{
REPORT
int i = n; Real* s = store; int d = spacing;
Real* s1 = rmrc.store; int d1 = rmrc.spacing;
if (i!=rmrc.n)
{
Tracer tr("newmatrm");
Throw(InternalException("Dimensions differ in Divide"));
}
// while (i--) { *s = *s1 / r; s += d; s1 += d1; }
if (i) for (;;) { *s = *s1 / r; if (!(--i)) break; s += d; s1 += d1; }
}
void RectMatrixRowCol::Divide(Real r)
{
REPORT
int i = n; Real* s = store; int d = spacing;
// while (i--) { *s /= r; s += d; }
if (i) for (;;) { *s /= r; if (!(--i)) break; s += d; }
}
void RectMatrixRowCol::Negate()
{
REPORT
int i = n; Real* s = store; int d = spacing;
// while (i--) { *s = - *s; s += d; }
if (i) for (;;) { *s = - *s; if (!(--i)) break; s += d; }
}
void RectMatrixRowCol::Zero()
{
REPORT
int i = n; Real* s = store; int d = spacing;
// while (i--) { *s = 0.0; s += d; }
if (i) for (;;) { *s = 0.0; if (!(--i)) break; s += d; }
}
void ComplexScale(RectMatrixCol& U, RectMatrixCol& V, Real x, Real y)
{
REPORT
int n = U.n;
if (n != V.n)
{
Tracer tr("newmatrm");
Throw(InternalException("Dimensions differ in ComplexScale"));
}
Real* u = U.store; Real* v = V.store;
int su = U.spacing; int sv = V.spacing;
//while (n--)
//{
// Real z = *u * x - *v * y; *v = *u * y + *v * x; *u = z;
// u += su; v += sv;
//}
if (n) for (;;)
{
Real z = *u * x - *v * y; *v = *u * y + *v * x; *u = z;
if (!(--n)) break;
u += su; v += sv;
}
}
void Rotate(RectMatrixCol& U, RectMatrixCol& V, Real tau, Real s)
{
REPORT
// (U, V) = (U, V) * (c, s) where tau = s/(1+c), c^2 + s^2 = 1
int n = U.n;
if (n != V.n)
{
Tracer tr("newmatrm");
Throw(InternalException("Dimensions differ in Rotate"));
}
Real* u = U.store; Real* v = V.store;
int su = U.spacing; int sv = V.spacing;
//while (n--)
//{
// Real zu = *u; Real zv = *v;
// *u -= s * (zv + zu * tau); *v += s * (zu - zv * tau);
// u += su; v += sv;
//}
if (n) for(;;)
{
Real zu = *u; Real zv = *v;
*u -= s * (zv + zu * tau); *v += s * (zu - zv * tau);
if (!(--n)) break;
u += su; v += sv;
}
}
#ifdef use_namespace
}
#endif