X-Git-Url: https://www.ginac.de/ginac.git//ginac.git?p=ginac.git;a=blobdiff_plain;f=ginac%2Fmatrix.h;h=fc6a80bf4e6b3127eb701809792d948fdb694525;hp=8b0555c8eb102bb937e538e922f5c755f83efca7;hb=f78b1f296310b5f1c01b74c9fb10dd33af2a8f4a;hpb=5b090bb7e4951b48a28c20ba21bf9810c86eb0ca diff --git a/ginac/matrix.h b/ginac/matrix.h index 8b0555c8..fc6a80bf 100644 --- a/ginac/matrix.h +++ b/ginac/matrix.h @@ -3,7 +3,7 @@ * Interface to symbolic matrices */ /* - * GiNaC Copyright (C) 1999-2000 Johannes Gutenberg University Mainz, Germany + * GiNaC Copyright (C) 1999-2003 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 @@ -24,118 +24,161 @@ #define __GINAC_MATRIX_H__ #include +#include #include "basic.h" #include "ex.h" -#ifndef NO_NAMESPACE_GINAC namespace GiNaC { -#endif // ndef NO_NAMESPACE_GINAC -/** Symbolic matrices. */ -class matrix : public basic -{ - GINAC_DECLARE_REGISTERED_CLASS(matrix, basic) -// friends -// (none) -// member functions - - // default constructor, destructor, copy constructor, assignment operator - // and helpers: +/** Helper template to allow initialization of matrices via an overloaded + * comma operator (idea stolen from Blitz++). */ +template +class matrix_init { public: - matrix(); - ~matrix(); - matrix(const matrix & other); - const matrix & operator=(const matrix & other); -protected: - void copy(const matrix & other); - void destroy(bool call_parent); + matrix_init(It i) : iter(i) {} + + matrix_init operator,(const T & x) + { + *iter = x; + return matrix_init(++iter); + } + + // The following specializations produce much tighter code than the + // general case above + + matrix_init operator,(int x) + { + *iter = T(x); + return matrix_init(++iter); + } + + matrix_init operator,(unsigned int x) + { + *iter = T(x); + return matrix_init(++iter); + } + + matrix_init operator,(long x) + { + *iter = T(x); + return matrix_init(++iter); + } + + matrix_init operator,(unsigned long x) + { + *iter = T(x); + return matrix_init(++iter); + } + + matrix_init operator,(double x) + { + *iter = T(x); + return matrix_init(++iter); + } + + matrix_init operator,(const symbol & x) + { + *iter = T(x); + return matrix_init(++iter); + } + +private: + matrix_init(); + It iter; +}; + - // other constructors +/** Symbolic matrices. */ +class matrix : public basic +{ + GINAC_DECLARE_REGISTERED_CLASS(matrix, basic) + + // other constructors public: - matrix(unsigned r, unsigned c); - matrix(unsigned r, unsigned c, const exvector & m2); - - // functions overriding virtual functions from bases classes + matrix(unsigned r, unsigned c); + matrix(unsigned r, unsigned c, const exvector & m2); + matrix(unsigned r, unsigned c, const lst & l); + + // First step of initialization of matrix with a comma-separated seqeuence + // of expressions. Subsequent steps are handled by matrix_init<>::operator,(). + matrix_init operator=(const ex & x) + { + m[0] = x; + return matrix_init(++m.begin()); + } + + // functions overriding virtual functions from base classes public: - basic * duplicate() const; - void print(ostream & os, unsigned upper_precedence=0) const; - void printraw(ostream & os) const; - unsigned nops() const; - ex op(int i) const; - ex & let_op(int i); - ex expand(unsigned options=0) const; - bool has(const ex & other) const; - ex eval(int level=0) const; - ex evalf(int level=0) const; - // ex subs(const lst & ls, const lst & lr) const; + size_t nops() const; + ex op(size_t i) const; + ex & let_op(size_t i); + ex eval(int level=0) const; + ex evalm() const {return *this;} + ex subs(const exmap & m, unsigned options = 0) const; + ex eval_indexed(const basic & i) const; + ex add_indexed(const ex & self, const ex & other) const; + ex scalar_mul_indexed(const ex & self, const numeric & other) const; + bool contract_with(exvector::iterator self, exvector::iterator other, exvector & v) const; + protected: - int compare_same_type(const basic & other) const; - unsigned return_type(void) const { return return_types::noncommutative; }; - // new virtual functions which can be overridden by derived classes - // (none) - - // non-virtual functions in this class + bool match_same_type(const basic & other) const; + unsigned return_type() const { return return_types::noncommutative; }; + + // non-virtual functions in this class public: - unsigned rows(void) const //! get number of rows. - { return row; } - unsigned cols(void) const //! get number of columns. - { return col; } - matrix add(const matrix & other) const; - matrix sub(const matrix & other) const; - matrix mul(const matrix & other) const; - const ex & operator() (unsigned ro, unsigned co) const; - matrix & set(unsigned ro, unsigned co, ex value); - matrix transpose(void) const; - ex determinant(void) const; - ex trace(void) const; - ex charpoly(const ex & lambda) const; - matrix inverse(void) const; - matrix fraction_free_elim(const matrix & vars, const matrix & v) const; - matrix solve(const matrix & vars, const matrix & rhs) const; - matrix old_solve(const matrix & v) const; // FIXME: may be removed + unsigned rows() const /// Get number of rows. + { return row; } + unsigned cols() const /// Get number of columns. + { return col; } + matrix add(const matrix & other) const; + matrix sub(const matrix & other) const; + matrix mul(const matrix & other) const; + matrix mul(const numeric & other) const; + matrix mul_scalar(const ex & other) const; + matrix pow(const ex & expn) const; + const ex & operator() (unsigned ro, unsigned co) const; + ex & operator() (unsigned ro, unsigned co); + matrix & set(unsigned ro, unsigned co, const ex & value) { (*this)(ro, co) = value; return *this; } + matrix transpose() const; + ex determinant(unsigned algo = determinant_algo::automatic) const; + ex trace() const; + ex charpoly(const ex & lambda) const; + matrix inverse() const; + matrix solve(const matrix & vars, const matrix & rhs, + unsigned algo = solve_algo::automatic) const; protected: - ex determinant_numeric(void) const; - ex determinant_minor_sparse(void) const; - ex determinant_minor_dense(void) const; - ex determinant_bareiss(void) const; - ex determinant_perm(void) const; - int gauss_elimination(void); - int fraction_free_elimination(void); - int division_free_elimination(void); - int pivot(unsigned ro, bool symbolic=true); -private: // FIXME: these should be obsoleted - void ffe_swap(unsigned r1, unsigned c1, unsigned r2 ,unsigned c2); - void ffe_set(unsigned r, unsigned c, ex e); - ex ffe_get(unsigned r, unsigned c) const; - + ex determinant_minor() const; + int gauss_elimination(const bool det = false); + int division_free_elimination(const bool det = false); + int fraction_free_elimination(const bool det = false); + int pivot(unsigned ro, unsigned co, bool symbolic = true); + + void print_elements(const print_context & c, const char *row_start, const char *row_end, const char *row_sep, const char *col_sep) const; + void do_print(const print_context & c, unsigned level) const; + void do_print_latex(const print_latex & c, unsigned level) const; + void do_print_python_repr(const print_python_repr & c, unsigned level) const; + // member variables protected: - unsigned row; /**< number of rows */ - unsigned col; /**< number of columns */ - exvector m; /**< representation (cols indexed first) */ - static unsigned precedence; + unsigned row; ///< number of rows + unsigned col; ///< number of columns + exvector m; ///< representation (cols indexed first) }; -// global constants -extern const matrix some_matrix; -extern const type_info & typeid_matrix; // wrapper functions around member functions -inline unsigned nops(const matrix & m) +inline size_t nops(const matrix & m) { return m.nops(); } -inline ex expand(const matrix & m, unsigned options=0) +inline ex expand(const matrix & m, unsigned options = 0) { return m.expand(options); } -inline bool has(const matrix & m, const ex & other) -{ return m.has(other); } - -inline ex eval(const matrix & m, int level=0) +inline ex eval(const matrix & m, int level = 0) { return m.eval(level); } -inline ex evalf(const matrix & m, int level=0) +inline ex evalf(const matrix & m, int level = 0) { return m.evalf(level); } inline unsigned rows(const matrix & m) @@ -147,8 +190,8 @@ inline unsigned cols(const matrix & m) inline matrix transpose(const matrix & m) { return m.transpose(); } -inline ex determinant(const matrix & m) -{ return m.determinant(); } +inline ex determinant(const matrix & m, unsigned options = determinant_algo::automatic) +{ return m.determinant(options); } inline ex trace(const matrix & m) { return m.trace(); } @@ -160,13 +203,36 @@ inline matrix inverse(const matrix & m) { return m.inverse(); } // utility functions -inline const matrix &ex_to_matrix(const ex &e) + +/** Specialization of is_exactly_a(obj) for matrix objects. */ +template<> inline bool is_exactly_a(const basic & obj) { - return static_cast(*e.bp); + return obj.tinfo()==TINFO_matrix; } -#ifndef NO_NAMESPACE_GINAC +/** Convert list of lists to matrix. */ +extern ex lst_to_matrix(const lst & l); + +/** Convert list of diagonal elements to matrix. */ +extern ex diag_matrix(const lst & l); + +/** Create an r times c unit matrix. */ +extern ex unit_matrix(unsigned r, unsigned c); + +/** Create a x times x unit matrix. */ +inline ex unit_matrix(unsigned x) +{ return unit_matrix(x, x); } + +/** Create an r times c matrix of newly generated symbols consisting of the + * given base name plus the numeric row/column position of each element. + * The base name for LaTeX output is specified separately. */ +extern ex symbolic_matrix(unsigned r, unsigned c, const std::string & base_name, const std::string & tex_base_name); + +/** Create an r times c matrix of newly generated symbols consisting of the + * given base name plus the numeric row/column position of each element. */ +inline ex symbolic_matrix(unsigned r, unsigned c, const std::string & base_name) +{ return symbolic_matrix(r, c, base_name, base_name); } + } // namespace GiNaC -#endif // ndef NO_NAMESPACE_GINAC #endif // ndef __GINAC_MATRIX_H__