- added variants of dirac_trace() and color_trace() that take the trace over

[ginac.git] / ginac / color.h

diff --git a/ginac/color.h b/ginac/color.h
index 8ff75442fa608de3dfe2c58861450837973314fc..826b4c7ffe68ae7489febb13aa82db7e705ab0f2 100644 (file)
--- a/ginac/color.h
+++ b/ginac/color.h
@@ -3,7 +3,7 @@
  *  Interface to GiNaC's color (SU(3) Lie algebra) objects. */
 
 /*
- *  GiNaC Copyright (C) 1999-2003 Johannes Gutenberg University Mainz, Germany
+ *  GiNaC Copyright (C) 1999-2004 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
@@ -26,6 +26,8 @@
 #include "indexed.h"
 #include "tensor.h"
 
+#include <set>
+
 namespace GiNaC {
 
 
@@ -33,7 +35,7 @@ namespace GiNaC {
  *  of SU(3), as used for calculations in quantum chromodynamics. A
  *  representation label (an unsigned 8-bit integer) is used to distinguish
  *  elements from different Lie algebras (objects with different labels
- *  commute). These objects implement an abstract representation of the
+ *  commutate). These objects implement an abstract representation of the
  *  group, not a specific matrix representation. The indices used for color
  *  objects should not have a variance. */
 class color : public indexed
@@ -169,6 +171,18 @@ ex color_d(const ex & a, const ex & b, const ex & c);
 /** This returns the linear combination d.a.b.c+I*f.a.b.c. */
 ex color_h(const ex & a, const ex & b, const ex & c);
 
+/** Calculate color traces over the specified set of representation labels.
+ *
+ *  @param e Expression to take the trace of
+ *  @param rls Set of representation labels */
+ex color_trace(const ex & e, const std::set<unsigned char> & rls);
+
+/** Calculate color traces over the specified list of representation labels.
+ *
+ *  @param e Expression to take the trace of
+ *  @param rll List of representation labels */
+ex color_trace(const ex & e, const lst & rll);
+
 /** Calculate the trace of an expression containing color objects with a
  *  specified representation label.
  *