148 lines
5 KiB
C++
Executable file
148 lines
5 KiB
C++
Executable file
/**
|
|
* @file DenseMatrix.cpp
|
|
*/
|
|
|
|
// Copyright 2001 California Institute of Technology
|
|
|
|
#ifdef WIN32
|
|
#pragma warning(disable:4786)
|
|
#pragma warning(disable:4503)
|
|
#endif
|
|
|
|
#include "ct_defs.h"
|
|
#include "ctlapack.h"
|
|
#include "utilities.h"
|
|
#include "DenseMatrix.h"
|
|
|
|
namespace Cantera {
|
|
|
|
/// assignment.
|
|
DenseMatrix& DenseMatrix::operator=(const DenseMatrix& y) {
|
|
if (&y == this) return *this;
|
|
Array2D::operator=(y);
|
|
m_ipiv = y.ipiv();
|
|
return *this;
|
|
}
|
|
|
|
void DenseMatrix::resize(int n, int m, doublereal v) {
|
|
Array2D::resize(n,m,v);
|
|
m_ipiv.resize( max(n,m) );
|
|
}
|
|
|
|
void DenseMatrix::mult(const double* b, double* prod) const {
|
|
ct_dgemv(ctlapack::ColMajor, ctlapack::NoTranspose,
|
|
static_cast<int>(nRows()),
|
|
static_cast<int>(nRows()), 1.0, ptrColumn(0), //begin(),
|
|
static_cast<int>(nRows()), b, 1, 0.0, prod, 1);
|
|
}
|
|
|
|
void DenseMatrix::leftMult(const double* b, double* prod) const {
|
|
int nc = static_cast<int>(nColumns());
|
|
int nr = static_cast<int>(nRows());
|
|
int n, i;
|
|
double sum = 0.0;
|
|
for (n = 0; n < nc; n++) {
|
|
sum = 0.0;
|
|
for (i = 0; i < nr; i++) {
|
|
sum += value(i,n)*b[i];
|
|
}
|
|
prod[n] = sum;
|
|
}
|
|
}
|
|
|
|
int solve(DenseMatrix& A, double* b) {
|
|
int info=0;
|
|
ct_dgetrf(static_cast<int>(A.nRows()),
|
|
static_cast<int>(A.nColumns()), A.ptrColumn(0), //begin(),
|
|
static_cast<int>(A.nRows()), &A.ipiv()[0], info);
|
|
if (info != 0)
|
|
throw CanteraError("DenseMatrix::solve",
|
|
"DGETRF returned INFO = "+int2str(info));
|
|
ct_dgetrs(ctlapack::NoTranspose,
|
|
static_cast<int>(A.nRows()), 1, A.ptrColumn(0), //begin(),
|
|
static_cast<int>(A.nRows()),
|
|
&A.ipiv()[0], b,
|
|
static_cast<int>(A.nColumns()), info);
|
|
if (info != 0)
|
|
throw CanteraError("DenseMatrix::solve",
|
|
"DGETRS returned INFO = "+int2str(info));
|
|
return 0;
|
|
}
|
|
|
|
int solve(DenseMatrix& A, DenseMatrix& b) {
|
|
int info=0;
|
|
ct_dgetrf(static_cast<int>(A.nRows()),
|
|
static_cast<int>(A.nColumns()), A.ptrColumn(0),
|
|
static_cast<int>(A.nRows()), &A.ipiv()[0], info);
|
|
if (info != 0)
|
|
throw CanteraError("DenseMatrix::solve",
|
|
"DGETRF returned INFO = "+int2str(info));
|
|
ct_dgetrs(ctlapack::NoTranspose, static_cast<int>(A.nRows()),
|
|
static_cast<int>(b.nColumns()),
|
|
A.ptrColumn(0), static_cast<int>(A.nRows()),
|
|
&A.ipiv()[0], b.ptrColumn(0),
|
|
static_cast<int>(b.nRows()), info);
|
|
if (info != 0)
|
|
throw CanteraError("DenseMatrix::solve",
|
|
"DGETRS returned INFO = "+int2str(info));
|
|
return 0;
|
|
}
|
|
|
|
|
|
#ifdef INCL_LEAST_SQUARES
|
|
/** @todo fix lwork */
|
|
int leastSquares(DenseMatrix& A, double* b) {
|
|
int info = 0;
|
|
int rank = 0;
|
|
double rcond = -1.0;
|
|
// fix this!
|
|
int lwork = 6000; // 2*(3*min(m,n) + max(2*min(m,n), max(m,n)));
|
|
vector_fp work(lwork);
|
|
vector_fp s(min(static_cast<int>(A.nRows()),
|
|
static_cast<int>(A.nColumns())));
|
|
ct_dgelss(static_cast<int>(A.nRows()),
|
|
static_cast<int>(A.nColumns()), 1, A.ptrColumn(0),
|
|
static_cast<int>(A.nRows()), b,
|
|
static_cast<int>(A.nColumns()), &s[0], //.begin(),
|
|
rcond, rank, &work[0], work.size(), info);
|
|
if (info != 0)
|
|
throw CanteraError("DenseMatrix::leaseSquares",
|
|
"DGELSS returned INFO = "+int2str(info));
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
void multiply(const DenseMatrix& A, const double* b, double* prod) {
|
|
ct_dgemv(ctlapack::ColMajor, ctlapack::NoTranspose,
|
|
static_cast<int>(A.nRows()), static_cast<int>(A.nColumns()), 1.0,
|
|
A.ptrColumn(0), static_cast<int>(A.nRows()), b, 1, 0.0, prod, 1);
|
|
}
|
|
|
|
void increment(const DenseMatrix& A,
|
|
const double* b, double* prod) {
|
|
ct_dgemv(ctlapack::ColMajor, ctlapack::NoTranspose,
|
|
static_cast<int>(A.nRows()), static_cast<int>(A.nRows()), 1.0,
|
|
A.ptrColumn(0), static_cast<int>(A.nRows()), b, 1, 1.0, prod, 1);
|
|
}
|
|
|
|
int invert(DenseMatrix& A, int nn) {
|
|
integer n = (nn > 0 ? nn : static_cast<int>(A.nRows()));
|
|
int info=0;
|
|
ct_dgetrf(n, n, A.ptrColumn(0), static_cast<int>(A.nRows()),
|
|
&A.ipiv()[0], info);
|
|
if (info != 0)
|
|
throw CanteraError("invert",
|
|
"DGETRF returned INFO="+int2str(info));
|
|
|
|
vector_fp work(n);
|
|
integer lwork = static_cast<int>(work.size());
|
|
ct_dgetri(n, A.ptrColumn(0), static_cast<int>(A.nRows()),
|
|
&A.ipiv()[0],
|
|
&work[0], lwork, info);
|
|
if (info != 0)
|
|
throw CanteraError("invert",
|
|
"DGETRI returned INFO="+int2str(info));
|
|
return 0;
|
|
}
|
|
}
|
|
|