You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1053 lines
32 KiB
C++

//$$ newmat7.cpp Invert, solve, binary operations
// Copyright (C) 1991,2,3,4: R B Davies
#include "include.h"
#include "newmat.h"
#include "newmatrc.h"
#ifdef use_namespace
namespace NEWMAT {
#endif
#ifdef DO_REPORT
#define REPORT { static ExeCounter ExeCount(__LINE__,7); ++ExeCount; }
#else
#define REPORT {}
#endif
//***************************** solve routines ******************************/
GeneralMatrix* GeneralMatrix::MakeSolver()
{
REPORT
GeneralMatrix* gm = new CroutMatrix(*this);
MatrixErrorNoSpace(gm); gm->ReleaseAndDelete(); return gm;
}
GeneralMatrix* Matrix::MakeSolver()
{
REPORT
GeneralMatrix* gm = new CroutMatrix(*this);
MatrixErrorNoSpace(gm); gm->ReleaseAndDelete(); return gm;
}
void CroutMatrix::Solver(MatrixColX& mcout, const MatrixColX& mcin)
{
REPORT
int i = mcin.skip; Real* el = mcin.data-i; Real* el1 = el;
while (i--) *el++ = 0.0;
el += mcin.storage; i = nrows - mcin.skip - mcin.storage;
while (i--) *el++ = 0.0;
lubksb(el1, mcout.skip);
}
// Do we need check for entirely zero output?
void UpperTriangularMatrix::Solver(MatrixColX& mcout,
const MatrixColX& mcin)
{
REPORT
int i = mcin.skip-mcout.skip; Real* elx = mcin.data-i;
while (i-- > 0) *elx++ = 0.0;
int nr = mcin.skip+mcin.storage;
elx = mcin.data+mcin.storage; Real* el = elx;
int j = mcout.skip+mcout.storage-nr; int nc = ncols-nr; i = nr-mcout.skip;
while (j-- > 0) *elx++ = 0.0;
Real* Ael = store + (nr*(2*ncols-nr+1))/2; j = 0;
while (i-- > 0)
{
elx = el; Real sum = 0.0; int jx = j++; Ael -= nc;
while (jx--) sum += *(--Ael) * *(--elx);
elx--; *elx = (*elx - sum) / *(--Ael);
}
}
void LowerTriangularMatrix::Solver(MatrixColX& mcout,
const MatrixColX& mcin)
{
REPORT
int i = mcin.skip-mcout.skip; Real* elx = mcin.data-i;
while (i-- > 0) *elx++ = 0.0;
int nc = mcin.skip; i = nc+mcin.storage; elx = mcin.data+mcin.storage;
int nr = mcout.skip+mcout.storage; int j = nr-i; i = nr-nc;
while (j-- > 0) *elx++ = 0.0;
Real* el = mcin.data; Real* Ael = store + (nc*(nc+1))/2; j = 0;
while (i-- > 0)
{
elx = el; Real sum = 0.0; int jx = j++; Ael += nc;
while (jx--) sum += *Ael++ * *elx++;
*elx = (*elx - sum) / *Ael++;
}
}
//******************* carry out binary operations *************************/
static GeneralMatrix*
GeneralMult(GeneralMatrix*,GeneralMatrix*,MultipliedMatrix*,MatrixType);
static GeneralMatrix*
GeneralSolv(GeneralMatrix*,GeneralMatrix*,BaseMatrix*,MatrixType);
static GeneralMatrix*
GeneralSolvI(GeneralMatrix*,BaseMatrix*,MatrixType);
static GeneralMatrix*
GeneralKP(GeneralMatrix*,GeneralMatrix*,KPMatrix*,MatrixType);
GeneralMatrix* MultipliedMatrix::Evaluate(MatrixType mt)
{
REPORT
gm2 = ((BaseMatrix*&)bm2)->Evaluate();
gm2 = gm2->Evaluate(gm2->Type().MultRHS()); // no symmetric on RHS
gm1=((BaseMatrix*&)bm1)->Evaluate();
#ifdef TEMPS_DESTROYED_QUICKLY
GeneralMatrix* gmx;
Try { gmx = GeneralMult(gm1, gm2, this, mt); }
CatchAll { delete this; ReThrow; }
delete this; return gmx;
#else
return GeneralMult(gm1, gm2, this, mt);
#endif
}
GeneralMatrix* SolvedMatrix::Evaluate(MatrixType mt)
{
REPORT
gm1=((BaseMatrix*&)bm1)->Evaluate();
gm2=((BaseMatrix*&)bm2)->Evaluate();
#ifdef TEMPS_DESTROYED_QUICKLY
GeneralMatrix* gmx;
Try { gmx = GeneralSolv(gm1,gm2,this,mt); }
CatchAll { delete this; ReThrow; }
delete this; return gmx;
#else
return GeneralSolv(gm1,gm2,this,mt);
#endif
}
GeneralMatrix* KPMatrix::Evaluate(MatrixType mt)
{
REPORT
gm1=((BaseMatrix*&)bm1)->Evaluate();
gm2=((BaseMatrix*&)bm2)->Evaluate();
#ifdef TEMPS_DESTROYED_QUICKLY
GeneralMatrix* gmx;
Try { gmx = GeneralKP(gm1,gm2,this,mt); }
CatchAll { delete this; ReThrow; }
delete this; return gmx;
#else
return GeneralKP(gm1,gm2,this,mt);
#endif
}
// routines for adding or subtracting matrices of identical storage structure
static void Add(GeneralMatrix* gm, GeneralMatrix* gm1, GeneralMatrix* gm2)
{
REPORT
Real* s1=gm1->Store(); Real* s2=gm2->Store();
Real* s=gm->Store(); int i=gm->Storage() >> 2;
while (i--)
{
*s++ = *s1++ + *s2++; *s++ = *s1++ + *s2++;
*s++ = *s1++ + *s2++; *s++ = *s1++ + *s2++;
}
i=gm->Storage() & 3; while (i--) *s++ = *s1++ + *s2++;
}
static void Add(GeneralMatrix* gm, GeneralMatrix* gm2)
{
REPORT
Real* s2=gm2->Store(); Real* s=gm->Store(); int i=gm->Storage() >> 2;
while (i--)
{ *s++ += *s2++; *s++ += *s2++; *s++ += *s2++; *s++ += *s2++; }
i=gm->Storage() & 3; while (i--) *s++ += *s2++;
}
static void Subtract(GeneralMatrix* gm, GeneralMatrix* gm1, GeneralMatrix* gm2)
{
REPORT
Real* s1=gm1->Store(); Real* s2=gm2->Store();
Real* s=gm->Store(); int i=gm->Storage() >> 2;
while (i--)
{
*s++ = *s1++ - *s2++; *s++ = *s1++ - *s2++;
*s++ = *s1++ - *s2++; *s++ = *s1++ - *s2++;
}
i=gm->Storage() & 3; while (i--) *s++ = *s1++ - *s2++;
}
static void Subtract(GeneralMatrix* gm, GeneralMatrix* gm2)
{
REPORT
Real* s2=gm2->Store(); Real* s=gm->Store(); int i=gm->Storage() >> 2;
while (i--)
{ *s++ -= *s2++; *s++ -= *s2++; *s++ -= *s2++; *s++ -= *s2++; }
i=gm->Storage() & 3; while (i--) *s++ -= *s2++;
}
static void ReverseSubtract(GeneralMatrix* gm, GeneralMatrix* gm2)
{
REPORT
Real* s2=gm2->Store(); Real* s=gm->Store(); int i=gm->Storage() >> 2;
while (i--)
{
*s = *s2++ - *s; s++; *s = *s2++ - *s; s++;
*s = *s2++ - *s; s++; *s = *s2++ - *s; s++;
}
i=gm->Storage() & 3; while (i--) { *s = *s2++ - *s; s++; }
}
static void SP(GeneralMatrix* gm, GeneralMatrix* gm1, GeneralMatrix* gm2)
{
REPORT
Real* s1=gm1->Store(); Real* s2=gm2->Store();
Real* s=gm->Store(); int i=gm->Storage() >> 2;
while (i--)
{
*s++ = *s1++ * *s2++; *s++ = *s1++ * *s2++;
*s++ = *s1++ * *s2++; *s++ = *s1++ * *s2++;
}
i=gm->Storage() & 3; while (i--) *s++ = *s1++ * *s2++;
}
static void SP(GeneralMatrix* gm, GeneralMatrix* gm2)
{
REPORT
Real* s2=gm2->Store(); Real* s=gm->Store(); int i=gm->Storage() >> 2;
while (i--)
{ *s++ *= *s2++; *s++ *= *s2++; *s++ *= *s2++; *s++ *= *s2++; }
i=gm->Storage() & 3; while (i--) *s++ *= *s2++;
}
// routines for adding or subtracting matrices of different storage structure
static void AddDS(GeneralMatrix* gm, GeneralMatrix* gm1, GeneralMatrix* gm2)
{
REPORT
int nr = gm->Nrows();
MatrixRow mr1(gm1, LoadOnEntry); MatrixRow mr2(gm2, LoadOnEntry);
MatrixRow mr(gm, StoreOnExit+DirectPart);
while (nr--) { mr.Add(mr1,mr2); mr1.Next(); mr2.Next(); mr.Next(); }
}
static void AddDS(GeneralMatrix* gm, GeneralMatrix* gm2)
// Add into first argument
{
REPORT
int nr = gm->Nrows();
MatrixRow mr(gm, StoreOnExit+LoadOnEntry+DirectPart);
MatrixRow mr2(gm2, LoadOnEntry);
while (nr--) { mr.Add(mr2); mr.Next(); mr2.Next(); }
}
static void SubtractDS
(GeneralMatrix* gm, GeneralMatrix* gm1, GeneralMatrix* gm2)
{
REPORT
int nr = gm->Nrows();
MatrixRow mr1(gm1, LoadOnEntry); MatrixRow mr2(gm2, LoadOnEntry);
MatrixRow mr(gm, StoreOnExit+DirectPart);
while (nr--) { mr.Sub(mr1,mr2); mr1.Next(); mr2.Next(); mr.Next(); }
}
static void SubtractDS(GeneralMatrix* gm, GeneralMatrix* gm2)
{
REPORT
int nr = gm->Nrows();
MatrixRow mr(gm, LoadOnEntry+StoreOnExit+DirectPart);
MatrixRow mr2(gm2, LoadOnEntry);
while (nr--) { mr.Sub(mr2); mr.Next(); mr2.Next(); }
}
static void ReverseSubtractDS(GeneralMatrix* gm, GeneralMatrix* gm2)
{
REPORT
int nr = gm->Nrows();
MatrixRow mr(gm, LoadOnEntry+StoreOnExit+DirectPart);
MatrixRow mr2(gm2, LoadOnEntry);
while (nr--) { mr.RevSub(mr2); mr2.Next(); mr.Next(); }
}
static void SPDS(GeneralMatrix* gm, GeneralMatrix* gm1, GeneralMatrix* gm2)
{
REPORT
int nr = gm->Nrows();
MatrixRow mr1(gm1, LoadOnEntry); MatrixRow mr2(gm2, LoadOnEntry);
MatrixRow mr(gm, StoreOnExit+DirectPart);
while (nr--) { mr.Multiply(mr1,mr2); mr1.Next(); mr2.Next(); mr.Next(); }
}
static void SPDS(GeneralMatrix* gm, GeneralMatrix* gm2)
// SP into first argument
{
REPORT
int nr = gm->Nrows();
MatrixRow mr(gm, StoreOnExit+LoadOnEntry+DirectPart);
MatrixRow mr2(gm2, LoadOnEntry);
while (nr--) { mr.Multiply(mr2); mr.Next(); mr2.Next(); }
}
static GeneralMatrix* GeneralMult1(GeneralMatrix* gm1, GeneralMatrix* gm2,
MultipliedMatrix* mm, MatrixType mtx)
{
REPORT
Tracer tr("GeneralMult1");
int nr=gm1->Nrows(); int nc=gm2->Ncols();
if (gm1->Ncols() !=gm2->Nrows())
Throw(IncompatibleDimensionsException(*gm1, *gm2));
GeneralMatrix* gmx = mtx.New(nr,nc,mm);
MatrixCol mcx(gmx, StoreOnExit+DirectPart);
MatrixCol mc2(gm2, LoadOnEntry);
while (nc--)
{
MatrixRow mr1(gm1, LoadOnEntry, mcx.Skip());
Real* el = mcx.Data(); // pointer to an element
int n = mcx.Storage();
while (n--) { *(el++) = DotProd(mr1,mc2); mr1.Next(); }
mc2.Next(); mcx.Next();
}
gmx->ReleaseAndDelete(); gm1->tDelete(); gm2->tDelete(); return gmx;
}
static GeneralMatrix* GeneralMult2(GeneralMatrix* gm1, GeneralMatrix* gm2,
MultipliedMatrix* mm, MatrixType mtx)
{
// version that accesses by row only - not good for thin matrices
// or column vectors in right hand term.
REPORT
Tracer tr("GeneralMult2");
int nr=gm1->Nrows(); int nc=gm2->Ncols();
if (gm1->Ncols() !=gm2->Nrows())
Throw(IncompatibleDimensionsException(*gm1, *gm2));
GeneralMatrix* gmx = mtx.New(nr,nc,mm);
MatrixRow mrx(gmx, LoadOnEntry+StoreOnExit+DirectPart);
MatrixRow mr1(gm1, LoadOnEntry);
while (nr--)
{
MatrixRow mr2(gm2, LoadOnEntry, mr1.Skip());
Real* el = mr1.Data(); // pointer to an element
int n = mr1.Storage();
mrx.Zero();
while (n--) { mrx.AddScaled(mr2, *el++); mr2.Next(); }
mr1.Next(); mrx.Next();
}
gmx->ReleaseAndDelete(); gm1->tDelete(); gm2->tDelete(); return gmx;
}
static GeneralMatrix* mmMult(GeneralMatrix* gm1, GeneralMatrix* gm2)
{
// matrix multiplication for type Matrix only
REPORT
Tracer tr("MatrixMult");
int nr=gm1->Nrows(); int ncr=gm1->Ncols(); int nc=gm2->Ncols();
if (ncr != gm2->Nrows()) Throw(IncompatibleDimensionsException(*gm1,*gm2));
Matrix* gm = new Matrix(nr,nc); MatrixErrorNoSpace(gm);
Real* s1=gm1->Store(); Real* s2=gm2->Store(); Real* s=gm->Store();
if (ncr)
{
while (nr--)
{
Real* s2x = s2; int j = ncr;
Real* sx = s; Real f = *s1++; int k = nc;
while (k--) *sx++ = f * *s2x++;
while (--j)
{ sx = s; f = *s1++; k = nc; while (k--) *sx++ += f * *s2x++; }
s = sx;
}
}
else *gm = 0.0;
gm->ReleaseAndDelete(); gm1->tDelete(); gm2->tDelete(); return gm;
}
static GeneralMatrix* GeneralMult(GeneralMatrix* gm1, GeneralMatrix* gm2,
MultipliedMatrix* mm, MatrixType mtx)
{
if ( Rectangular(gm1->Type(), gm2->Type(), mtx))
{
REPORT
return mmMult(gm1, gm2);
}
else
{
REPORT
Compare(gm1->Type() * gm2->Type(),mtx);
int nr = gm2->Nrows(); int nc = gm2->Ncols();
if (nc <= 5 && nr > nc) { REPORT return GeneralMult1(gm1, gm2, mm, mtx); }
else { REPORT return GeneralMult2(gm1, gm2, mm, mtx); }
}
}
static GeneralMatrix* GeneralKP(GeneralMatrix* gm1, GeneralMatrix* gm2,
KPMatrix* kp, MatrixType mtx)
{
REPORT
Tracer tr("GeneralKP");
int nr1 = gm1->Nrows(); int nc1 = gm1->Ncols();
int nr2 = gm2->Nrows(); int nc2 = gm2->Ncols();
Compare((gm1->Type()).KP(gm2->Type()),mtx);
GeneralMatrix* gmx = mtx.New(nr1*nr2, nc1*nc2, kp);
MatrixRow mrx(gmx, LoadOnEntry+StoreOnExit+DirectPart);
MatrixRow mr1(gm1, LoadOnEntry);
for (int i = 1; i <= nr1; ++i)
{
MatrixRow mr2(gm2, LoadOnEntry);
for (int j = 1; j <= nr2; ++j)
{ mrx.KP(mr1,mr2); mr2.Next(); mrx.Next(); }
mr1.Next();
}
gmx->ReleaseAndDelete(); gm1->tDelete(); gm2->tDelete(); return gmx;
}
static GeneralMatrix* GeneralSolv(GeneralMatrix* gm1, GeneralMatrix* gm2,
BaseMatrix* sm, MatrixType mtx)
{
REPORT
Tracer tr("GeneralSolv");
Compare(gm1->Type().i() * gm2->Type(),mtx);
int nr = gm1->Nrows();
if (nr != gm1->Ncols()) Throw(NotSquareException(*gm1));
int nc = gm2->Ncols();
if (gm1->Ncols() != gm2->Nrows())
Throw(IncompatibleDimensionsException(*gm1, *gm2));
GeneralMatrix* gmx = mtx.New(nr,nc,sm); MatrixErrorNoSpace(gmx);
Real* r = new Real [nr]; MatrixErrorNoSpace(r);
MONITOR_REAL_NEW("Make (GenSolv)",nr,r)
GeneralMatrix* gms = gm1->MakeSolver();
Try
{
MatrixColX mcx(gmx, r, StoreOnExit+DirectPart); // copy to and from r
// this must be inside Try so mcx is destroyed before gmx
MatrixColX mc2(gm2, r, LoadOnEntry);
while (nc--) { gms->Solver(mcx, mc2); mcx.Next(); mc2.Next(); }
}
CatchAll
{
if (gms) gms->tDelete();
delete gmx; // <--------------------
gm2->tDelete();
MONITOR_REAL_DELETE("Delete (GenSolv)",nr,r)
// ATandT version 2.1 gives an internal error
delete [] r;
ReThrow;
}
gms->tDelete(); gmx->ReleaseAndDelete(); gm2->tDelete();
MONITOR_REAL_DELETE("Delete (GenSolv)",nr,r)
// ATandT version 2.1 gives an internal error
delete [] r;
return gmx;
}
// version for inverses - gm2 is identity
static GeneralMatrix* GeneralSolvI(GeneralMatrix* gm1, BaseMatrix* sm,
MatrixType mtx)
{
REPORT
Tracer tr("GeneralSolvI");
Compare(gm1->Type().i(),mtx);
int nr = gm1->Nrows();
if (nr != gm1->Ncols()) Throw(NotSquareException(*gm1));
int nc = nr;
// DiagonalMatrix I(nr); I = 1;
IdentityMatrix I(nr);
GeneralMatrix* gmx = mtx.New(nr,nc,sm); MatrixErrorNoSpace(gmx);
Real* r = new Real [nr]; MatrixErrorNoSpace(r);
MONITOR_REAL_NEW("Make (GenSolvI)",nr,r)
GeneralMatrix* gms = gm1->MakeSolver();
Try
{
MatrixColX mcx(gmx, r, StoreOnExit+DirectPart); // copy to and from r
// this must be inside Try so mcx is destroyed before gmx
MatrixColX mc2(&I, r, LoadOnEntry);
while (nc--) { gms->Solver(mcx, mc2); mcx.Next(); mc2.Next(); }
}
CatchAll
{
if (gms) gms->tDelete();
delete gmx;
MONITOR_REAL_DELETE("Delete (GenSolvI)",nr,r)
// ATandT version 2.1 gives an internal error
delete [] r;
ReThrow;
}
gms->tDelete(); gmx->ReleaseAndDelete();
MONITOR_REAL_DELETE("Delete (GenSolvI)",nr,r)
// ATandT version 2.1 gives an internal error
delete [] r;
return gmx;
}
GeneralMatrix* InvertedMatrix::Evaluate(MatrixType mtx)
{
// Matrix Inversion - use solve routines
Tracer tr("InvertedMatrix::Evaluate");
REPORT
gm=((BaseMatrix*&)bm)->Evaluate();
#ifdef TEMPS_DESTROYED_QUICKLY
GeneralMatrix* gmx;
Try { gmx = GeneralSolvI(gm,this,mtx); }
CatchAll { delete this; ReThrow; }
delete this; return gmx;
#else
return GeneralSolvI(gm,this,mtx);
#endif
}
//*************************** New versions ************************
GeneralMatrix* AddedMatrix::Evaluate(MatrixType mtd)
{
REPORT
Tracer tr("AddedMatrix::Evaluate");
gm1=((BaseMatrix*&)bm1)->Evaluate(); gm2=((BaseMatrix*&)bm2)->Evaluate();
int nr=gm1->Nrows(); int nc=gm1->Ncols();
if (nr!=gm2->Nrows() || nc!=gm2->Ncols())
{
Try { Throw(IncompatibleDimensionsException(*gm1, *gm2)); }
CatchAll
{
gm1->tDelete(); gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
}
MatrixType mt1 = gm1->Type(), mt2 = gm2->Type(); MatrixType mts = mt1 + mt2;
if (!mtd) { REPORT mtd = mts; }
else if (!(mtd.DataLossOK || mtd >= mts))
{
REPORT
gm1->tDelete(); gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
Throw(ProgramException("Illegal Conversion", mts, mtd));
}
GeneralMatrix* gmx;
bool c1 = (mtd == mt1), c2 = (mtd == mt2);
if ( c1 && c2 && (gm1->SimpleAddOK(gm2) == 0) )
{
if (gm1->reuse()) { REPORT Add(gm1,gm2); gm2->tDelete(); gmx = gm1; }
else if (gm2->reuse()) { REPORT Add(gm2,gm1); gmx = gm2; }
else
{
REPORT
// what if new throws an exception
Try { gmx = mt1.New(nr,nc,this); }
CatchAll
{
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
gmx->ReleaseAndDelete(); Add(gmx,gm1,gm2);
}
}
else
{
if (c1 && c2)
{
short SAO = gm1->SimpleAddOK(gm2);
if (SAO & 1) { REPORT c1 = false; }
if (SAO & 2) { REPORT c2 = false; }
}
if (c1 && gm1->reuse() ) // must have type test first
{ REPORT AddDS(gm1,gm2); gm2->tDelete(); gmx = gm1; }
else if (c2 && gm2->reuse() )
{ REPORT AddDS(gm2,gm1); if (!c1) gm1->tDelete(); gmx = gm2; }
else
{
REPORT
Try { gmx = mtd.New(nr,nc,this); }
CatchAll
{
if (!c1) gm1->tDelete(); if (!c2) gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
AddDS(gmx,gm1,gm2);
if (!c1) gm1->tDelete(); if (!c2) gm2->tDelete();
gmx->ReleaseAndDelete();
}
}
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
return gmx;
}
GeneralMatrix* SubtractedMatrix::Evaluate(MatrixType mtd)
{
REPORT
Tracer tr("SubtractedMatrix::Evaluate");
gm1=((BaseMatrix*&)bm1)->Evaluate(); gm2=((BaseMatrix*&)bm2)->Evaluate();
int nr=gm1->Nrows(); int nc=gm1->Ncols();
if (nr!=gm2->Nrows() || nc!=gm2->Ncols())
{
Try { Throw(IncompatibleDimensionsException(*gm1, *gm2)); }
CatchAll
{
gm1->tDelete(); gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
}
MatrixType mt1 = gm1->Type(), mt2 = gm2->Type(); MatrixType mts = mt1 + mt2;
if (!mtd) { REPORT mtd = mts; }
else if (!(mtd.DataLossOK || mtd >= mts))
{
gm1->tDelete(); gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
Throw(ProgramException("Illegal Conversion", mts, mtd));
}
GeneralMatrix* gmx;
bool c1 = (mtd == mt1), c2 = (mtd == mt2);
if ( c1 && c2 && (gm1->SimpleAddOK(gm2) == 0) )
{
if (gm1->reuse()) { REPORT Subtract(gm1,gm2); gm2->tDelete(); gmx = gm1; }
else if (gm2->reuse()) { REPORT ReverseSubtract(gm2,gm1); gmx = gm2; }
else
{
REPORT
Try { gmx = mt1.New(nr,nc,this); }
CatchAll
{
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
gmx->ReleaseAndDelete(); Subtract(gmx,gm1,gm2);
}
}
else
{
if (c1 && c2)
{
short SAO = gm1->SimpleAddOK(gm2);
if (SAO & 1) { REPORT c1 = false; }
if (SAO & 2) { REPORT c2 = false; }
}
if (c1 && gm1->reuse() ) // must have type test first
{ REPORT SubtractDS(gm1,gm2); gm2->tDelete(); gmx = gm1; }
else if (c2 && gm2->reuse() )
{
REPORT ReverseSubtractDS(gm2,gm1);
if (!c1) gm1->tDelete(); gmx = gm2;
}
else
{
REPORT
// what if New throws and exception
Try { gmx = mtd.New(nr,nc,this); }
CatchAll
{
if (!c1) gm1->tDelete(); if (!c2) gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
SubtractDS(gmx,gm1,gm2);
if (!c1) gm1->tDelete(); if (!c2) gm2->tDelete();
gmx->ReleaseAndDelete();
}
}
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
return gmx;
}
GeneralMatrix* SPMatrix::Evaluate(MatrixType mtd)
{
REPORT
Tracer tr("SPMatrix::Evaluate");
gm1=((BaseMatrix*&)bm1)->Evaluate(); gm2=((BaseMatrix*&)bm2)->Evaluate();
int nr=gm1->Nrows(); int nc=gm1->Ncols();
if (nr!=gm2->Nrows() || nc!=gm2->Ncols())
{
Try { Throw(IncompatibleDimensionsException(*gm1, *gm2)); }
CatchAll
{
gm1->tDelete(); gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
}
MatrixType mt1 = gm1->Type(), mt2 = gm2->Type();
MatrixType mts = mt1.SP(mt2);
if (!mtd) { REPORT mtd = mts; }
else if (!(mtd.DataLossOK || mtd >= mts))
{
gm1->tDelete(); gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
Throw(ProgramException("Illegal Conversion", mts, mtd));
}
GeneralMatrix* gmx;
bool c1 = (mtd == mt1), c2 = (mtd == mt2);
if ( c1 && c2 && (gm1->SimpleAddOK(gm2) == 0) )
{
if (gm1->reuse()) { REPORT SP(gm1,gm2); gm2->tDelete(); gmx = gm1; }
else if (gm2->reuse()) { REPORT SP(gm2,gm1); gmx = gm2; }
else
{
REPORT
Try { gmx = mt1.New(nr,nc,this); }
CatchAll
{
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
gmx->ReleaseAndDelete(); SP(gmx,gm1,gm2);
}
}
else
{
if (c1 && c2)
{
short SAO = gm1->SimpleAddOK(gm2);
if (SAO & 1) { REPORT c2 = false; } // c1 and c2 swapped
if (SAO & 2) { REPORT c1 = false; }
}
if (c1 && gm1->reuse() ) // must have type test first
{ REPORT SPDS(gm1,gm2); gm2->tDelete(); gmx = gm1; }
else if (c2 && gm2->reuse() )
{ REPORT SPDS(gm2,gm1); if (!c1) gm1->tDelete(); gmx = gm2; }
else
{
REPORT
// what if New throws and exception
Try { gmx = mtd.New(nr,nc,this); }
CatchAll
{
if (!c1) gm1->tDelete(); if (!c2) gm2->tDelete();
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
ReThrow;
}
SPDS(gmx,gm1,gm2);
if (!c1) gm1->tDelete(); if (!c2) gm2->tDelete();
gmx->ReleaseAndDelete();
}
}
#ifdef TEMPS_DESTROYED_QUICKLY
delete this;
#endif
return gmx;
}
//*************************** norm functions ****************************/
Real BaseMatrix::Norm1() const
{
// maximum of sum of absolute values of a column
REPORT
GeneralMatrix* gm = ((BaseMatrix&)*this).Evaluate();
int nc = gm->Ncols(); Real value = 0.0;
MatrixCol mc(gm, LoadOnEntry);
while (nc--)
{ Real v = mc.SumAbsoluteValue(); if (value < v) value = v; mc.Next(); }
gm->tDelete(); return value;
}
Real BaseMatrix::NormInfinity() const
{
// maximum of sum of absolute values of a row
REPORT
GeneralMatrix* gm = ((BaseMatrix&)*this).Evaluate();
int nr = gm->Nrows(); Real value = 0.0;
MatrixRow mr(gm, LoadOnEntry);
while (nr--)
{ Real v = mr.SumAbsoluteValue(); if (value < v) value = v; mr.Next(); }
gm->tDelete(); return value;
}
//********************** Concatenation and stacking *************************/
GeneralMatrix* ConcatenatedMatrix::Evaluate(MatrixType mtx)
{
REPORT
Tracer tr("Concatenate");
#ifdef TEMPS_DESTROYED_QUICKLY
Try
{
gm2 = ((BaseMatrix*&)bm2)->Evaluate();
gm1 = ((BaseMatrix*&)bm1)->Evaluate();
Compare(gm1->Type() | gm2->Type(),mtx);
int nr=gm1->Nrows(); int nc = gm1->Ncols() + gm2->Ncols();
if (nr != gm2->Nrows())
Throw(IncompatibleDimensionsException(*gm1, *gm2));
GeneralMatrix* gmx = mtx.New(nr,nc,this);
MatrixRow mr1(gm1, LoadOnEntry); MatrixRow mr2(gm2, LoadOnEntry);
MatrixRow mr(gmx, StoreOnExit+DirectPart);
while (nr--) { mr.ConCat(mr1,mr2); mr1.Next(); mr2.Next(); mr.Next(); }
gmx->ReleaseAndDelete(); gm1->tDelete(); gm2->tDelete(); delete this;
return gmx;
}
CatchAll { delete this; ReThrow; }
#ifndef UseExceptions
return 0;
#endif
#else
gm2 = ((BaseMatrix*&)bm2)->Evaluate();
gm1 = ((BaseMatrix*&)bm1)->Evaluate();
Compare(gm1->Type() | gm2->Type(),mtx);
int nr=gm1->Nrows(); int nc = gm1->Ncols() + gm2->Ncols();
if (nr != gm2->Nrows())
Throw(IncompatibleDimensionsException(*gm1, *gm2));
GeneralMatrix* gmx = mtx.New(nr,nc,this);
MatrixRow mr1(gm1, LoadOnEntry); MatrixRow mr2(gm2, LoadOnEntry);
MatrixRow mr(gmx, StoreOnExit+DirectPart);
while (nr--) { mr.ConCat(mr1,mr2); mr1.Next(); mr2.Next(); mr.Next(); }
gmx->ReleaseAndDelete(); gm1->tDelete(); gm2->tDelete(); return gmx;
#endif
}
GeneralMatrix* StackedMatrix::Evaluate(MatrixType mtx)
{
REPORT
Tracer tr("Stack");
#ifdef TEMPS_DESTROYED_QUICKLY
Try
{
gm2 = ((BaseMatrix*&)bm2)->Evaluate();
gm1 = ((BaseMatrix*&)bm1)->Evaluate();
Compare(gm1->Type() & gm2->Type(),mtx);
int nc=gm1->Ncols();
int nr1 = gm1->Nrows(); int nr2 = gm2->Nrows();
if (nc != gm2->Ncols())
Throw(IncompatibleDimensionsException(*gm1, *gm2));
GeneralMatrix* gmx = mtx.New(nr1+nr2,nc,this);
MatrixRow mr1(gm1, LoadOnEntry); MatrixRow mr2(gm2, LoadOnEntry);
MatrixRow mr(gmx, StoreOnExit+DirectPart);
while (nr1--) { mr.Copy(mr1); mr1.Next(); mr.Next(); }
while (nr2--) { mr.Copy(mr2); mr2.Next(); mr.Next(); }
gmx->ReleaseAndDelete(); gm1->tDelete(); gm2->tDelete(); delete this;
return gmx;
}
CatchAll { delete this; ReThrow; }
#ifndef UseExceptions
return 0;
#endif
#else
gm2 = ((BaseMatrix*&)bm2)->Evaluate();
gm1 = ((BaseMatrix*&)bm1)->Evaluate();
Compare(gm1->Type() & gm2->Type(),mtx);
int nc=gm1->Ncols();
int nr1 = gm1->Nrows(); int nr2 = gm2->Nrows();
if (nc != gm2->Ncols())
Throw(IncompatibleDimensionsException(*gm1, *gm2));
GeneralMatrix* gmx = mtx.New(nr1+nr2,nc,this);
MatrixRow mr1(gm1, LoadOnEntry); MatrixRow mr2(gm2, LoadOnEntry);
MatrixRow mr(gmx, StoreOnExit+DirectPart);
while (nr1--) { mr.Copy(mr1); mr1.Next(); mr.Next(); }
while (nr2--) { mr.Copy(mr2); mr2.Next(); mr.Next(); }
gmx->ReleaseAndDelete(); gm1->tDelete(); gm2->tDelete(); return gmx;
#endif
}
// ************************* equality of matrices ******************** //
static bool RealEqual(Real* s1, Real* s2, int n)
{
int i = n >> 2;
while (i--)
{
if (*s1++ != *s2++) return false; if (*s1++ != *s2++) return false;
if (*s1++ != *s2++) return false; if (*s1++ != *s2++) return false;
}
i = n & 3; while (i--) if (*s1++ != *s2++) return false;
return true;
}
static bool intEqual(int* s1, int* s2, int n)
{
int i = n >> 2;
while (i--)
{
if (*s1++ != *s2++) return false; if (*s1++ != *s2++) return false;
if (*s1++ != *s2++) return false; if (*s1++ != *s2++) return false;
}
i = n & 3; while (i--) if (*s1++ != *s2++) return false;
return true;
}
bool operator==(const BaseMatrix& A, const BaseMatrix& B)
{
Tracer tr("BaseMatrix ==");
REPORT
GeneralMatrix* gmA = ((BaseMatrix&)A).Evaluate();
GeneralMatrix* gmB = ((BaseMatrix&)B).Evaluate();
if (gmA == gmB) // same matrix
{ REPORT gmA->tDelete(); return true; }
if ( gmA->Nrows() != gmB->Nrows() || gmA->Ncols() != gmB->Ncols() )
// different dimensions
{ REPORT gmA->tDelete(); gmB->tDelete(); return false; }
// check for CroutMatrix or BandLUMatrix
MatrixType AType = gmA->Type(); MatrixType BType = gmB->Type();
if (AType.CannotConvert() || BType.CannotConvert() )
{
REPORT
bool bx = gmA->IsEqual(*gmB);
gmA->tDelete(); gmB->tDelete();
return bx;
}
// is matrix storage the same
// will need to modify if further matrix structures are introduced
if (AType == BType && gmA->BandWidth() == gmB->BandWidth())
{ // compare store
REPORT
bool bx = RealEqual(gmA->Store(),gmB->Store(),gmA->Storage());
gmA->tDelete(); gmB->tDelete();
return bx;
}
// matrix storage different - just subtract
REPORT return IsZero(*gmA-*gmB);
}
bool operator==(const GeneralMatrix& A, const GeneralMatrix& B)
{
Tracer tr("GeneralMatrix ==");
// May or may not call tDeletes
REPORT
if (&A == &B) // same matrix
{ REPORT return true; }
if ( A.Nrows() != B.Nrows() || A.Ncols() != B.Ncols() )
{ REPORT return false; } // different dimensions
// check for CroutMatrix or BandLUMatrix
MatrixType AType = A.Type(); MatrixType BType = B.Type();
if (AType.CannotConvert() || BType.CannotConvert() )
{ REPORT return A.IsEqual(B); }
// is matrix storage the same
// will need to modify if further matrix structures are introduced
if (AType == BType && A.BandWidth() == B.BandWidth())
{ REPORT return RealEqual(A.Store(),B.Store(),A.Storage()); }
// matrix storage different - just subtract
REPORT return IsZero(A-B);
}
bool GeneralMatrix::IsZero() const
{
REPORT
Real* s=store; int i = storage >> 2;
while (i--)
{
if (*s++) return false; if (*s++) return false;
if (*s++) return false; if (*s++) return false;
}
i = storage & 3; while (i--) if (*s++) return false;
return true;
}
bool IsZero(const BaseMatrix& A)
{
Tracer tr("BaseMatrix::IsZero");
REPORT
GeneralMatrix* gm1 = 0; bool bx;
Try { gm1=((BaseMatrix&)A).Evaluate(); bx = gm1->IsZero(); }
CatchAll { if (gm1) gm1->tDelete(); ReThrow; }
gm1->tDelete();
return bx;
}
// IsEqual functions - insist matrices are of same type
// as well as equal values to be equal
bool GeneralMatrix::IsEqual(const GeneralMatrix& A) const
{
Tracer tr("GeneralMatrix IsEqual");
if (A.Type() != Type()) // not same types
{ REPORT return false; }
if (&A == this) // same matrix
{ REPORT return true; }
if (A.nrows != nrows || A.ncols != ncols)
// different dimensions
{ REPORT return false; }
// is matrix storage the same - compare store
REPORT
return RealEqual(A.store,store,storage);
}
bool CroutMatrix::IsEqual(const GeneralMatrix& A) const
{
Tracer tr("CroutMatrix IsEqual");
if (A.Type() != Type()) // not same types
{ REPORT return false; }
if (&A == this) // same matrix
{ REPORT return true; }
if (A.nrows != nrows || A.ncols != ncols)
// different dimensions
{ REPORT return false; }
// is matrix storage the same - compare store
REPORT
return RealEqual(A.store,store,storage)
&& intEqual(((CroutMatrix&)A).indx, indx, nrows);
}
bool BandLUMatrix::IsEqual(const GeneralMatrix& A) const
{
Tracer tr("BandLUMatrix IsEqual");
if (A.Type() != Type()) // not same types
{ REPORT return false; }
if (&A == this) // same matrix
{ REPORT return true; }
if ( A.Nrows() != nrows || A.Ncols() != ncols
|| ((BandLUMatrix&)A).m1 != m1 || ((BandLUMatrix&)A).m2 != m2 )
// different dimensions
{ REPORT return false; }
// matrix storage the same - compare store
REPORT
return RealEqual(A.Store(),store,storage)
&& RealEqual(((BandLUMatrix&)A).store2,store2,storage2)
&& intEqual(((BandLUMatrix&)A).indx, indx, nrows);
}
#ifdef use_namespace
}
#endif