X-Git-Url: https://www.ginac.de/ginac.git//ginac.git?p=ginac.git;a=blobdiff_plain;f=ginac%2Fcolor.h;h=826b4c7ffe68ae7489febb13aa82db7e705ab0f2;hp=2086909e1304dd1c7a35ac9e53c658263150348c;hb=ba0b56c5ac76274c5d748eaa29fbd83cb1d4fb9e;hpb=09f0415b7442714ea9adae30872d151452185962;ds=sidebyside

diff --git a/ginac/color.h b/ginac/color.h
index 2086909e..826b4c7f 100644
--- a/ginac/color.h
+++ b/ginac/color.h
@@ -26,6 +26,8 @@
 #include "indexed.h"
 #include "tensor.h"
 
+#include <set>
+
 namespace GiNaC {
 
 
@@ -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.
  *