/** @file matrix.cpp * * Implementation of symbolic matrices */ /* * GiNaC Copyright (C) 1999-2000 Johannes Gutenberg University Mainz, Germany * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ #include #include #include #include "matrix.h" #include "archive.h" #include "numeric.h" #include "lst.h" #include "utils.h" #include "debugmsg.h" #include "power.h" #include "symbol.h" #include "normal.h" #ifndef NO_NAMESPACE_GINAC namespace GiNaC { #endif // ndef NO_NAMESPACE_GINAC GINAC_IMPLEMENT_REGISTERED_CLASS(matrix, basic) ////////// // default constructor, destructor, copy constructor, assignment operator // and helpers: ////////// // public /** Default ctor. Initializes to 1 x 1-dimensional zero-matrix. */ matrix::matrix() : inherited(TINFO_matrix), row(1), col(1) { debugmsg("matrix default constructor",LOGLEVEL_CONSTRUCT); m.push_back(_ex0()); } matrix::~matrix() { debugmsg("matrix destructor",LOGLEVEL_DESTRUCT); destroy(false); } matrix::matrix(const matrix & other) { debugmsg("matrix copy constructor",LOGLEVEL_CONSTRUCT); copy(other); } const matrix & matrix::operator=(const matrix & other) { debugmsg("matrix operator=",LOGLEVEL_ASSIGNMENT); if (this != &other) { destroy(true); copy(other); } return *this; } // protected void matrix::copy(const matrix & other) { inherited::copy(other); row = other.row; col = other.col; m = other.m; // STL's vector copying invoked here } void matrix::destroy(bool call_parent) { if (call_parent) inherited::destroy(call_parent); } ////////// // other constructors ////////// // public /** Very common ctor. Initializes to r x c-dimensional zero-matrix. * * @param r number of rows * @param c number of cols */ matrix::matrix(unsigned r, unsigned c) : inherited(TINFO_matrix), row(r), col(c) { debugmsg("matrix constructor from unsigned,unsigned",LOGLEVEL_CONSTRUCT); m.resize(r*c, _ex0()); } // protected /** Ctor from representation, for internal use only. */ matrix::matrix(unsigned r, unsigned c, const exvector & m2) : inherited(TINFO_matrix), row(r), col(c), m(m2) { debugmsg("matrix constructor from unsigned,unsigned,exvector",LOGLEVEL_CONSTRUCT); } ////////// // archiving ////////// /** Construct object from archive_node. */ matrix::matrix(const archive_node &n, const lst &sym_lst) : inherited(n, sym_lst) { debugmsg("matrix constructor from archive_node", LOGLEVEL_CONSTRUCT); if (!(n.find_unsigned("row", row)) || !(n.find_unsigned("col", col))) throw (std::runtime_error("unknown matrix dimensions in archive")); m.reserve(row * col); for (unsigned int i=0; true; i++) { ex e; if (n.find_ex("m", e, sym_lst, i)) m.push_back(e); else break; } } /** Unarchive the object. */ ex matrix::unarchive(const archive_node &n, const lst &sym_lst) { return (new matrix(n, sym_lst))->setflag(status_flags::dynallocated); } /** Archive the object. */ void matrix::archive(archive_node &n) const { inherited::archive(n); n.add_unsigned("row", row); n.add_unsigned("col", col); exvector::const_iterator i = m.begin(), iend = m.end(); while (i != iend) { n.add_ex("m", *i); ++i; } } ////////// // functions overriding virtual functions from bases classes ////////// // public basic * matrix::duplicate() const { debugmsg("matrix duplicate",LOGLEVEL_DUPLICATE); return new matrix(*this); } void matrix::print(std::ostream & os, unsigned upper_precedence) const { debugmsg("matrix print",LOGLEVEL_PRINT); os << "[[ "; for (unsigned r=0; r=0); GINAC_ASSERT(isetflag(status_flags::dynallocated | status_flags::evaluated ); } /** evaluate matrix numerically entry by entry. */ ex matrix::evalf(int level) const { debugmsg("matrix evalf",LOGLEVEL_MEMBER_FUNCTION); // check if we have to do anything at all if (level==1) return *this; // emergency break if (level == -max_recursion_level) { throw (std::runtime_error("matrix::evalf(): recursion limit exceeded")); } // evalf() entry by entry exvector m2(row*col); --level; for (unsigned r=0; r(const_cast(other)); // compare number of rows if (row != o.rows()) return row < o.rows() ? -1 : 1; // compare number of columns if (col != o.cols()) return col < o.cols() ? -1 : 1; // equal number of rows and columns, compare individual elements int cmpval; for (unsigned r=0; r matrices are equal; return 0; } ////////// // non-virtual functions in this class ////////// // public /** Sum of matrices. * * @exception logic_error (incompatible matrices) */ matrix matrix::add(const matrix & other) const { if (col != other.col || row != other.row) throw (std::logic_error("matrix::add(): incompatible matrices")); exvector sum(this->m); exvector::iterator i; exvector::const_iterator ci; for (i=sum.begin(), ci=other.m.begin(); i!=sum.end(); ++i, ++ci) (*i) += (*ci); return matrix(row,col,sum); } /** Difference of matrices. * * @exception logic_error (incompatible matrices) */ matrix matrix::sub(const matrix & other) const { if (col != other.col || row != other.row) throw (std::logic_error("matrix::sub(): incompatible matrices")); exvector dif(this->m); exvector::iterator i; exvector::const_iterator ci; for (i=dif.begin(), ci=other.m.begin(); i!=dif.end(); ++i, ++ci) (*i) -= (*ci); return matrix(row,col,dif); } /** Product of matrices. * * @exception logic_error (incompatible matrices) */ matrix matrix::mul(const matrix & other) const { if (this->cols() != other.rows()) throw (std::logic_error("matrix::mul(): incompatible matrices")); exvector prod(this->rows()*other.cols()); for (unsigned r1=0; r1rows(); ++r1) { for (unsigned c=0; ccols(); ++c) { if (m[r1*col+c].is_zero()) continue; for (unsigned r2=0; r2=row || co>=col) throw (std::range_error("matrix::operator(): index out of range")); return m[ro*col+co]; } /** Set individual elements manually. * * @exception range_error (index out of range) */ matrix & matrix::set(unsigned ro, unsigned co, ex value) { if (ro>=row || co>=col) throw (std::range_error("matrix::set(): index out of range")); ensure_if_modifiable(); m[ro*col+co] = value; return *this; } /** Transposed of an m x n matrix, producing a new n x m matrix object that * represents the transposed. */ matrix matrix::transpose(void) const { exvector trans(this->cols()*this->rows()); for (unsigned r=0; rcols(); ++r) for (unsigned c=0; crows(); ++c) trans[r*this->rows()+c] = m[c*this->cols()+r]; return matrix(this->cols(),this->rows(),trans); } /** Determinant of square matrix. This routine doesn't actually calculate the * determinant, it only implements some heuristics about which algorithm to * run. If all the elements of the matrix are elements of an integral domain * the determinant is also in that integral domain and the result is expanded * only. If one or more elements are from a quotient field the determinant is * usually also in that quotient field and the result is normalized before it * is returned. This implies that the determinant of the symbolic 2x2 matrix * [[a/(a-b),1],[b/(a-b),1]] is returned as unity. (In this respect, it * behaves like MapleV and unlike Mathematica.) * * @param algo allows to chose an algorithm * @return the determinant as a new expression * @exception logic_error (matrix not square) * @see determinant_algo */ ex matrix::determinant(unsigned algo) const { if (row!=col) throw (std::logic_error("matrix::determinant(): matrix not square")); GINAC_ASSERT(row*col==m.capacity()); // Gather some statistical information about this matrix: bool numeric_flag = true; bool normal_flag = false; unsigned sparse_count = 0; // counts non-zero elements for (exvector::const_iterator r=m.begin(); r!=m.end(); ++r) { lst srl; // symbol replacement list ex rtest = (*r).to_rational(srl); if (!rtest.is_zero()) ++sparse_count; if (!rtest.info(info_flags::numeric)) numeric_flag = false; if (!rtest.info(info_flags::crational_polynomial) && rtest.info(info_flags::rational_function)) normal_flag = true; } // Here is the heuristics in case this routine has to decide: if (algo == determinant_algo::automatic) { // Minor expansion is generally a good guess: algo = determinant_algo::laplace; // Does anybody know when a matrix is really sparse? // Maybe <~row/2.236 nonzero elements average in a row? if (row>3 && 5*sparse_count<=row*col) algo = determinant_algo::bareiss; // Purely numeric matrix can be handled by Gauss elimination. // This overrides any prior decisions. if (numeric_flag) algo = determinant_algo::gauss; } // Trap the trivial case here, since some algorithms don't like it if (this->row==1) { // for consistency with non-trivial determinants... if (normal_flag) return m[0].normal(); else return m[0].expand(); } // Compute the determinant switch(algo) { case determinant_algo::gauss: { ex det = 1; matrix tmp(*this); int sign = tmp.gauss_elimination(true); for (unsigned d=0; d uintpair; std::vector c_zeros; // number of zeros in column for (unsigned c=0; c pre_sort; for (std::vector::iterator i=c_zeros.begin(); i!=c_zeros.end(); ++i) pre_sort.push_back(i->second); int sign = permutation_sign(pre_sort); exvector result(row*col); // represents sorted matrix unsigned c = 0; for (std::vector::iterator i=pre_sort.begin(); i!=pre_sort.end(); ++i,++c) { for (unsigned r=0; rmul(B); c = B.trace()/ex(i+1); poly -= c*power(lambda,row-i-1); } if (row%2) return -poly; else return poly; } matrix M(*this); for (unsigned r=0; rrows(); const unsigned n = this->cols(); const unsigned p = rhs.cols(); // syntax checks if ((rhs.rows() != m) || (vars.rows() != n) || (vars.col != p)) throw (std::logic_error("matrix::solve(): incompatible matrices")); for (unsigned ro=0; rom[r*n+c]; for (unsigned c=0; c=0; --r) { unsigned fnz = 1; // first non-zero in row while ((fnz<=n) && (aug.m[r*(n+p)+(fnz-1)].is_zero())) ++fnz; if (fnz>n) { // row consists only of zeros, corresponding rhs must be 0, too if (!aug.m[r*(n+p)+n+co].is_zero()) { throw (std::runtime_error("matrix::solve(): inconsistent linear system")); } } else { // assign solutions for vars between fnz+1 and // last_assigned_sol-1: free parameters for (unsigned c=fnz; ccols(); if (n==1) return m[0].expand(); if (n==2) return (m[0]*m[3]-m[2]*m[1]).expand(); if (n==3) return (m[0]*m[4]*m[8]-m[0]*m[5]*m[7]- m[1]*m[3]*m[8]+m[2]*m[3]*m[7]+ m[1]*m[5]*m[6]-m[2]*m[4]*m[6]).expand(); // This algorithm can best be understood by looking at a naive // implementation of Laplace-expansion, like this one: // ex det; // matrix minorM(this->rows()-1,this->cols()-1); // for (unsigned r1=0; r1rows(); ++r1) { // // shortcut if element(r1,0) vanishes // if (m[r1*col].is_zero()) // continue; // // assemble the minor matrix // for (unsigned r=0; r Pkey; Pkey.reserve(n); // key for minor determinant (a subpartition of Pkey) std::vector Mkey; Mkey.reserve(n-1); // we store our subminors in maps, keys being the rows they arise from typedef std::map,class ex> Rmap; typedef std::map,class ex>::value_type Rmap_value; Rmap A; Rmap B; ex det; // initialize A with last column: for (unsigned r=0; r=0; --c) { Pkey.erase(Pkey.begin(),Pkey.end()); // don't change capacity Mkey.erase(Mkey.begin(),Mkey.end()); for (unsigned i=0; i0; --fc) { ++Pkey[fc-1]; if (Pkey[fc-1]0) for (unsigned j=fc; jrows(); const unsigned n = this->cols(); GINAC_ASSERT(!det || n==m); int sign = 1; unsigned r0 = 0; for (unsigned r1=0; (r1=0) { if (indx > 0) sign = -sign; for (unsigned r2=r0+1; r2m[r2*n+r1].is_zero()) { // yes, there is something to do in this row ex piv = this->m[r2*n+r1] / this->m[r0*n+r1]; for (unsigned c=r1+1; cm[r2*n+c] -= piv * this->m[r0*n+c]; if (!this->m[r2*n+c].info(info_flags::numeric)) this->m[r2*n+c] = this->m[r2*n+c].normal(); } } // fill up left hand side with zeros for (unsigned c=0; c<=r1; ++c) this->m[r2*n+c] = _ex0(); } if (det) { // save space by deleting no longer needed elements for (unsigned c=r0+1; cm[r0*n+c] = _ex0(); } ++r0; } } return sign; } /** Perform the steps of division free elimination to bring the m x n matrix * into an upper echelon form. * * @param det may be set to true to save a lot of space if one is only * interested in the diagonal elements (i.e. for calculating determinants). * The others are set to zero in this case. * @return sign is 1 if an even number of rows was swapped, -1 if an odd * number of rows was swapped and 0 if the matrix is singular. */ int matrix::division_free_elimination(const bool det) { ensure_if_modifiable(); const unsigned m = this->rows(); const unsigned n = this->cols(); GINAC_ASSERT(!det || n==m); int sign = 1; unsigned r0 = 0; for (unsigned r1=0; (r1=0) { if (indx>0) sign = -sign; for (unsigned r2=r0+1; r2m[r2*n+c] = (this->m[r0*n+r1]*this->m[r2*n+c] - this->m[r2*n+r1]*this->m[r0*n+c]).expand(); // fill up left hand side with zeros for (unsigned c=0; c<=r1; ++c) this->m[r2*n+c] = _ex0(); } if (det) { // save space by deleting no longer needed elements for (unsigned c=r0+1; cm[r0*n+c] = _ex0(); } ++r0; } } return sign; } /** Perform the steps of Bareiss' one-step fraction free elimination to bring * the matrix into an upper echelon form. Fraction free elimination means * that divide is used straightforwardly, without computing GCDs first. This * is possible, since we know the divisor at each step. * * @param det may be set to true to save a lot of space if one is only * interested in the last element (i.e. for calculating determinants). The * others are set to zero in this case. * @return sign is 1 if an even number of rows was swapped, -1 if an odd * number of rows was swapped and 0 if the matrix is singular. */ int matrix::fraction_free_elimination(const bool det) { // Method: // (single-step fraction free elimination scheme, already known to Jordan) // // Usual division-free elimination sets m[0](r,c) = m(r,c) and then sets // m[k+1](r,c) = m[k](k,k) * m[k](r,c) - m[k](r,k) * m[k](k,c). // // Bareiss (fraction-free) elimination in addition divides that element // by m[k-1](k-1,k-1) for k>1, where it can be shown by means of the // Sylvester determinant that this really divides m[k+1](r,c). // // We also allow rational functions where the original prove still holds. // However, we must care for numerator and denominator separately and // "manually" work in the integral domains because of subtle cancellations // (see below). This blows up the bookkeeping a bit and the formula has // to be modified to expand like this (N{x} stands for numerator of x, // D{x} for denominator of x): // N{m[k+1](r,c)} = N{m[k](k,k)}*N{m[k](r,c)}*D{m[k](r,k)}*D{m[k](k,c)} // -N{m[k](r,k)}*N{m[k](k,c)}*D{m[k](k,k)}*D{m[k](r,c)} // D{m[k+1](r,c)} = D{m[k](k,k)}*D{m[k](r,c)}*D{m[k](r,k)}*D{m[k](k,c)} // where for k>1 we now divide N{m[k+1](r,c)} by // N{m[k-1](k-1,k-1)} // and D{m[k+1](r,c)} by // D{m[k-1](k-1,k-1)}. ensure_if_modifiable(); const unsigned m = this->rows(); const unsigned n = this->cols(); GINAC_ASSERT(!det || n==m); int sign = 1; if (m==1) return 1; ex divisor_n = 1; ex divisor_d = 1; ex dividend_n; ex dividend_d; // We populate temporary matrices to subsequently operate on. There is // one holding numerators and another holding denominators of entries. // This is a must since the evaluator (or even earlier mul's constructor) // might cancel some trivial element which causes divide() to fail. The // elements are normalized first (yes, even though this algorithm doesn't // need GCDs) since the elements of *this might be unnormalized, which // makes things more complicated than they need to be. matrix tmp_n(*this); matrix tmp_d(m,n); // for denominators, if needed lst srl; // symbol replacement list exvector::iterator it = this->m.begin(); exvector::iterator tmp_n_it = tmp_n.m.begin(); exvector::iterator tmp_d_it = tmp_d.m.begin(); for (; it!= this->m.end(); ++it, ++tmp_n_it, ++tmp_d_it) { (*tmp_n_it) = (*it).normal().to_rational(srl); (*tmp_d_it) = (*tmp_n_it).denom(); (*tmp_n_it) = (*tmp_n_it).numer(); } unsigned r0 = 0; for (unsigned r1=0; (r1=0) { if (indx>0) { sign = -sign; // tmp_n's rows r0 and indx were swapped, do the same in tmp_d: for (unsigned c=r1; cm.begin(); tmp_n_it = tmp_n.m.begin(); tmp_d_it = tmp_d.m.begin(); for (; it!= this->m.end(); ++it, ++tmp_n_it, ++tmp_d_it) (*it) = ((*tmp_n_it)/(*tmp_d_it)).subs(srl); return sign; } /** Partial pivoting method for matrix elimination schemes. * Usual pivoting (symbolic==false) returns the index to the element with the * largest absolute value in column ro and swaps the current row with the one * where the element was found. With (symbolic==true) it does the same thing * with the first non-zero element. * * @param ro is the row from where to begin * @param co is the column to be inspected * @param symbolic signal if we want the first non-zero element to be pivoted * (true) or the one with the largest absolute value (false). * @return 0 if no interchange occured, -1 if all are zero (usually signaling * a degeneracy) and positive integer k means that rows ro and k were swapped. */ int matrix::pivot(unsigned ro, unsigned co, bool symbolic) { unsigned k = ro; if (symbolic) { // search first non-zero element in column co beginning at row ro while ((km[k*col+co].expand().is_zero())) ++k; } else { // search largest element in column co beginning at row ro GINAC_ASSERT(is_ex_of_type(this->m[k*col+co],numeric)); unsigned kmax = k+1; numeric mmax = abs(ex_to_numeric(m[kmax*col+co])); while (kmaxm[kmax*col+co],numeric)); numeric tmp = ex_to_numeric(this->m[kmax*col+co]); if (abs(tmp) > mmax) { mmax = tmp; k = kmax; } ++kmax; } if (!mmax.is_zero()) k = kmax; } if (k==row) // all elements in column co below row ro vanish return -1; if (k==ro) // matrix needs no pivoting return 0; // matrix needs pivoting, so swap rows k and ro ensure_if_modifiable(); for (unsigned c=0; cm[k*col+c].swap(this->m[ro*col+c]); return k; } /** Convert list of lists to matrix. */ ex lst_to_matrix(const ex &l) { if (!is_ex_of_type(l, lst)) throw(std::invalid_argument("argument to lst_to_matrix() must be a lst")); // Find number of rows and columns unsigned rows = l.nops(), cols = 0, i, j; for (i=0; i cols) cols = l.op(i).nops(); // Allocate and fill matrix matrix &m = *new matrix(rows, cols); for (i=0; i j) m.set(i, j, l.op(i).op(j)); else m.set(i, j, ex(0)); return m; } ////////// // global constants ////////// const matrix some_matrix; const std::type_info & typeid_matrix = typeid(some_matrix); #ifndef NO_NAMESPACE_GINAC } // namespace GiNaC #endif // ndef NO_NAMESPACE_GINAC