X-Git-Url: https://www.ginac.de/ginac.git//ginac.git?p=ginac.git;a=blobdiff_plain;f=ginac%2Fidx.cpp;h=cc5f52c3fef153a30596efb2980bc8f012ab81f4;hp=e2b76bec65b43e42f899166258b03475990ddb69;hb=006b3274b157c4772c16d30f49079913bd236652;hpb=276f1bc79bd0f41a12fb45ece6c7857e13c6668e diff --git a/ginac/idx.cpp b/ginac/idx.cpp index e2b76bec..cc5f52c3 100644 --- a/ginac/idx.cpp +++ b/ginac/idx.cpp @@ -3,7 +3,7 @@ * Implementation of GiNaC's indices. */ /* - * GiNaC Copyright (C) 1999-2001 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 @@ -20,16 +20,18 @@ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ +#include +#include #include -#include #include "idx.h" #include "symbol.h" #include "lst.h" +#include "relational.h" +#include "operators.h" #include "print.h" #include "archive.h" #include "utils.h" -#include "debugmsg.h" namespace GiNaC { @@ -38,56 +40,27 @@ GINAC_IMPLEMENT_REGISTERED_CLASS(varidx, idx) GINAC_IMPLEMENT_REGISTERED_CLASS(spinidx, varidx) ////////// -// default constructor, destructor, copy constructor assignment operator and helpers +// default constructor ////////// -idx::idx() : inherited(TINFO_idx) -{ - debugmsg("idx default constructor", LOGLEVEL_CONSTRUCT); -} +idx::idx() : inherited(TINFO_idx) {} varidx::varidx() : covariant(false) { - debugmsg("varidx default constructor", LOGLEVEL_CONSTRUCT); tinfo_key = TINFO_varidx; } spinidx::spinidx() : dotted(false) { - debugmsg("spinidx default constructor", LOGLEVEL_CONSTRUCT); tinfo_key = TINFO_spinidx; } -void idx::copy(const idx & other) -{ - inherited::copy(other); - value = other.value; - dim = other.dim; -} - -void varidx::copy(const varidx & other) -{ - inherited::copy(other); - covariant = other.covariant; -} - -void spinidx::copy(const spinidx & other) -{ - inherited::copy(other); - dotted = other.dotted; -} - -DEFAULT_DESTROY(idx) -DEFAULT_DESTROY(varidx) -DEFAULT_DESTROY(spinidx) - ////////// // other constructors ////////// idx::idx(const ex & v, const ex & d) : inherited(TINFO_idx), value(v), dim(d) { - debugmsg("idx constructor from ex,ex", LOGLEVEL_CONSTRUCT); if (is_dim_numeric()) if (!dim.info(info_flags::posint)) throw(std::invalid_argument("dimension of space must be a positive integer")); @@ -95,13 +68,11 @@ idx::idx(const ex & v, const ex & d) : inherited(TINFO_idx), value(v), dim(d) varidx::varidx(const ex & v, const ex & d, bool cov) : inherited(v, d), covariant(cov) { - debugmsg("varidx constructor from ex,ex,bool", LOGLEVEL_CONSTRUCT); tinfo_key = TINFO_varidx; } spinidx::spinidx(const ex & v, const ex & d, bool cov, bool dot) : inherited(v, d, cov), dotted(dot) { - debugmsg("spinidx constructor from ex,ex,bool,bool", LOGLEVEL_CONSTRUCT); tinfo_key = TINFO_spinidx; } @@ -109,22 +80,19 @@ spinidx::spinidx(const ex & v, const ex & d, bool cov, bool dot) : inherited(v, // archiving ////////// -idx::idx(const archive_node &n, const lst &sym_lst) : inherited(n, sym_lst) +idx::idx(const archive_node &n, lst &sym_lst) : inherited(n, sym_lst) { - debugmsg("idx constructor from archive_node", LOGLEVEL_CONSTRUCT); n.find_ex("value", value, sym_lst); n.find_ex("dim", dim, sym_lst); } -varidx::varidx(const archive_node &n, const lst &sym_lst) : inherited(n, sym_lst) +varidx::varidx(const archive_node &n, lst &sym_lst) : inherited(n, sym_lst) { - debugmsg("varidx constructor from archive_node", LOGLEVEL_CONSTRUCT); n.find_bool("covariant", covariant); } -spinidx::spinidx(const archive_node &n, const lst &sym_lst) : inherited(n, sym_lst) +spinidx::spinidx(const archive_node &n, lst &sym_lst) : inherited(n, sym_lst) { - debugmsg("spinidx constructor from archive_node", LOGLEVEL_CONSTRUCT); n.find_bool("dotted", dotted); } @@ -152,14 +120,12 @@ DEFAULT_UNARCHIVE(varidx) DEFAULT_UNARCHIVE(spinidx) ////////// -// functions overriding virtual functions from bases classes +// functions overriding virtual functions from base classes ////////// void idx::print(const print_context & c, unsigned level) const { - debugmsg("idx print", LOGLEVEL_PRINT); - - if (is_of_type(c, print_tree)) { + if (is_a(c)) { c.s << std::string(level, ' ') << class_name() << std::hex << ", hash=0x" << hashvalue << ", flags=0x" << flags << std::dec @@ -170,22 +136,29 @@ void idx::print(const print_context & c, unsigned level) const } else { - if (!is_of_type(c, print_latex)) + if (is_a(c)) + c.s << "{"; + else c.s << "."; - bool need_parens = !(is_ex_exactly_of_type(value, numeric) || is_ex_of_type(value, symbol)); + bool need_parens = !(is_exactly_a(value) || is_a(value)); if (need_parens) c.s << "("; value.print(c); if (need_parens) c.s << ")"; + if (c.options & print_options::print_index_dimensions) { + c.s << "["; + dim.print(c); + c.s << "]"; + } + if (is_a(c)) + c.s << "}"; } } void varidx::print(const print_context & c, unsigned level) const { - debugmsg("varidx print", LOGLEVEL_PRINT); - - if (is_of_type(c, print_tree)) { + if (is_a(c)) { c.s << std::string(level, ' ') << class_name() << std::hex << ", hash=0x" << hashvalue << ", flags=0x" << flags << std::dec @@ -196,27 +169,33 @@ void varidx::print(const print_context & c, unsigned level) const dim.print(c, level + delta_indent); } else { - - if (!is_of_type(c, print_latex)) { + if (is_a(c)) + c.s << "{"; + else { if (covariant) c.s << "."; else c.s << "~"; } - bool need_parens = !(is_ex_exactly_of_type(value, numeric) || is_ex_of_type(value, symbol)); + bool need_parens = !(is_exactly_a(value) || is_a(value)); if (need_parens) c.s << "("; value.print(c); if (need_parens) c.s << ")"; + if (c.options & print_options::print_index_dimensions) { + c.s << "["; + dim.print(c); + c.s << "]"; + } + if (is_a(c)) + c.s << "}"; } } void spinidx::print(const print_context & c, unsigned level) const { - debugmsg("spinidx print", LOGLEVEL_PRINT); - - if (is_of_type(c, print_tree)) { + if (is_a(c)) { c.s << std::string(level, ' ') << class_name() << std::hex << ", hash=0x" << hashvalue << ", flags=0x" << flags << std::dec @@ -229,8 +208,13 @@ void spinidx::print(const print_context & c, unsigned level) const } else { - bool is_tex = is_of_type(c, print_latex); - if (!is_tex) { + bool is_tex = is_a(c); + if (is_tex) { + if (covariant) + c.s << "_{"; + else + c.s << "^{"; + } else { if (covariant) c.s << "."; else @@ -242,7 +226,7 @@ void spinidx::print(const print_context & c, unsigned level) const else c.s << "*"; } - bool need_parens = !(is_ex_exactly_of_type(value, numeric) || is_ex_of_type(value, symbol)); + bool need_parens = !(is_exactly_a(value) || is_a(value)); if (need_parens) c.s << "("; value.print(c); @@ -250,6 +234,8 @@ void spinidx::print(const print_context & c, unsigned level) const c.s << ")"; if (is_tex && dotted) c.s << "}"; + if (is_tex) + c.s << "}"; } } @@ -260,23 +246,32 @@ bool idx::info(unsigned inf) const return inherited::info(inf); } -unsigned idx::nops() const +size_t idx::nops() const { // don't count the dimension as that is not really a sub-expression return 1; } -ex & idx::let_op(int i) +ex idx::op(size_t i) const { GINAC_ASSERT(i == 0); return value; } +ex idx::map(map_function & f) const +{ + idx *copy = duplicate(); + copy->setflag(status_flags::dynallocated); + copy->clearflag(status_flags::hash_calculated); + copy->value = f(value); + return *copy; +} + /** Returns order relation between two indices of the same type. The order * must be such that dummy indices lie next to each other. */ int idx::compare_same_type(const basic & other) const { - GINAC_ASSERT(is_of_type(other, idx)); + GINAC_ASSERT(is_a(other)); const idx &o = static_cast(other); int cmpval = value.compare(o.value); @@ -285,9 +280,17 @@ int idx::compare_same_type(const basic & other) const return dim.compare(o.dim); } +bool idx::match_same_type(const basic & other) const +{ + GINAC_ASSERT(is_a(other)); + const idx &o = static_cast(other); + + return dim.is_equal(o.dim); +} + int varidx::compare_same_type(const basic & other) const { - GINAC_ASSERT(is_of_type(other, varidx)); + GINAC_ASSERT(is_a(other)); const varidx &o = static_cast(other); int cmpval = inherited::compare_same_type(other); @@ -297,12 +300,24 @@ int varidx::compare_same_type(const basic & other) const // Check variance last so dummy indices will end up next to each other if (covariant != o.covariant) return covariant ? -1 : 1; + return 0; } +bool varidx::match_same_type(const basic & other) const +{ + GINAC_ASSERT(is_a(other)); + const varidx &o = static_cast(other); + + if (covariant != o.covariant) + return false; + + return inherited::match_same_type(other); +} + int spinidx::compare_same_type(const basic & other) const { - GINAC_ASSERT(is_of_type(other, spinidx)); + GINAC_ASSERT(is_a(other)); const spinidx &o = static_cast(other); // Check dottedness first so dummy indices will end up next to each other @@ -316,67 +331,59 @@ int spinidx::compare_same_type(const basic & other) const return 0; } -bool idx::match(const ex & pattern, lst & repl_lst) const +bool spinidx::match_same_type(const basic & other) const { - if (!is_ex_of_type(pattern, idx)) - return false; - const idx &o = ex_to_idx(pattern); - if (!dim.is_equal(o.dim)) - return false; - return value.match(o.value, repl_lst); -} + GINAC_ASSERT(is_a(other)); + const spinidx &o = static_cast(other); -bool varidx::match(const ex & pattern, lst & repl_lst) const -{ - if (!is_ex_of_type(pattern, varidx)) - return false; - const varidx &o = ex_to_varidx(pattern); - if (covariant != o.covariant) + if (dotted != o.dotted) return false; - return inherited::match(pattern, repl_lst); + return inherited::match_same_type(other); } -bool spinidx::match(const ex & pattern, lst & repl_lst) const +/** By default, basic::evalf would evaluate the index value but we don't want + * a.1 to become a.(1.0). */ +ex idx::evalf(int level) const { - if (!is_ex_of_type(pattern, spinidx)) - return false; - const spinidx &o = ex_to_spinidx(pattern); - if (dotted != o.dotted) - return false; - return inherited::match(pattern, repl_lst); + return *this; } -ex idx::subs(const lst & ls, const lst & lr, bool no_pattern) const +ex idx::subs(const exmap & m, unsigned options) const { - GINAC_ASSERT(ls.nops() == lr.nops()); - // First look for index substitutions - for (unsigned i=0; iindex - if (is_ex_of_type(lr.op(i), idx)) - return lr.op(i); - - // Otherwise substitute value - idx *i_copy = static_cast(duplicate()); - i_copy->value = lr.op(i); - i_copy->clearflag(status_flags::hash_calculated); - return i_copy->setflag(status_flags::dynallocated); - } + exmap::const_iterator it = m.find(*this); + if (it != m.end()) { + + // Substitution index->index + if (is_a(it->second)) + return it->second; + + // Otherwise substitute value + idx *i_copy = duplicate(); + i_copy->value = it->second; + i_copy->clearflag(status_flags::hash_calculated); + return i_copy->setflag(status_flags::dynallocated); } // None, substitute objects in value (not in dimension) - const ex &subsed_value = value.subs(ls, lr, no_pattern); + const ex &subsed_value = value.subs(m, options); if (are_ex_trivially_equal(value, subsed_value)) return *this; - idx *i_copy = static_cast(duplicate()); + idx *i_copy = duplicate(); i_copy->value = subsed_value; i_copy->clearflag(status_flags::hash_calculated); return i_copy->setflag(status_flags::dynallocated); } +/** Implementation of ex::diff() for an index always returns 0. + * + * @see ex::diff */ +ex idx::derivative(const symbol & s) const +{ + return _ex0; +} + ////////// // new virtual functions ////////// @@ -386,15 +393,19 @@ bool idx::is_dummy_pair_same_type(const basic & other) const const idx &o = static_cast(other); // Only pure symbols form dummy pairs, "2n+1" doesn't - if (!is_ex_of_type(value, symbol)) + if (!is_a(value)) return false; // Value must be equal, of course if (!value.is_equal(o.value)) return false; - // Also the dimension - return dim.is_equal(o.dim); + // Dimensions need not be equal but must be comparable (so we can + // determine the minimum dimension of contractions) + if (dim.is_equal(o.dim)) + return true; + + return (dim < o.dim || dim > o.dim || (is_exactly_a(dim) && is_a(o.dim)) || (is_a(dim) && is_exactly_a(o.dim))); } bool varidx::is_dummy_pair_same_type(const basic & other) const @@ -424,25 +435,38 @@ bool spinidx::is_dummy_pair_same_type(const basic & other) const // non-virtual functions ////////// -ex varidx::toggle_variance(void) const +ex idx::replace_dim(const ex & new_dim) const +{ + idx *i_copy = duplicate(); + i_copy->dim = new_dim; + i_copy->clearflag(status_flags::hash_calculated); + return i_copy->setflag(status_flags::dynallocated); +} + +ex idx::minimal_dim(const idx & other) const +{ + return GiNaC::minimal_dim(dim, other.dim); +} + +ex varidx::toggle_variance() const { - varidx *i_copy = static_cast(duplicate()); + varidx *i_copy = duplicate(); i_copy->covariant = !i_copy->covariant; i_copy->clearflag(status_flags::hash_calculated); return i_copy->setflag(status_flags::dynallocated); } -ex spinidx::toggle_dot(void) const +ex spinidx::toggle_dot() const { - spinidx *i_copy = static_cast(duplicate()); + spinidx *i_copy = duplicate(); i_copy->dotted = !i_copy->dotted; i_copy->clearflag(status_flags::hash_calculated); return i_copy->setflag(status_flags::dynallocated); } -ex spinidx::toggle_variance_dot(void) const +ex spinidx::toggle_variance_dot() const { - spinidx *i_copy = static_cast(duplicate()); + spinidx *i_copy = duplicate(); i_copy->covariant = !i_copy->covariant; i_copy->dotted = !i_copy->dotted; i_copy->clearflag(status_flags::hash_calculated); @@ -466,10 +490,10 @@ bool is_dummy_pair(const idx & i1, const idx & i2) bool is_dummy_pair(const ex & e1, const ex & e2) { // The expressions must be indices - if (!is_ex_of_type(e1, idx) || !is_ex_of_type(e2, idx)) + if (!is_a(e1) || !is_a(e2)) return false; - return is_dummy_pair(ex_to_idx(e1), ex_to_idx(e2)); + return is_dummy_pair(ex_to(e1), ex_to(e2)); } void find_free_and_dummy(exvector::const_iterator it, exvector::const_iterator itend, exvector & out_free, exvector & out_dummy) @@ -483,7 +507,7 @@ void find_free_and_dummy(exvector::const_iterator it, exvector::const_iterator i // Only one index? Then it is a free one if it's not numeric if (itend - it == 1) { - if (ex_to_idx(*it).is_symbolic()) + if (ex_to(*it).is_symbolic()) out_free.push_back(*it); return; } @@ -503,13 +527,26 @@ void find_free_and_dummy(exvector::const_iterator it, exvector::const_iterator i if (it == itend) return; } else { - if (!it->is_equal(*last) && ex_to_idx(*last).is_symbolic()) + if (!it->is_equal(*last) && ex_to(*last).is_symbolic()) out_free.push_back(*last); } last = it++; } - if (ex_to_idx(*last).is_symbolic()) + if (ex_to(*last).is_symbolic()) out_free.push_back(*last); } +ex minimal_dim(const ex & dim1, const ex & dim2) +{ + if (dim1.is_equal(dim2) || dim1 < dim2 || (is_exactly_a(dim1) && is_a(dim2))) + return dim1; + else if (dim1 > dim2 || (is_a(dim1) && is_exactly_a(dim2))) + return dim2; + else { + std::ostringstream s; + s << "minimal_dim(): index dimensions " << dim1 << " and " << dim2 << " cannot be ordered"; + throw (std::runtime_error(s.str())); + } +} + } // namespace GiNaC