- dirac_trace() handles arbitrary expressions (even unexpanded), including

[ginac.git] / ginac / color.h

diff --git a/ginac/color.h b/ginac/color.h
index 04d7c5d9e88c106869d37c2e025c3b1fe5e7faa9..700493253adad38b154c6040586753b7f6465861 100644 (file)
--- a/ginac/color.h
+++ b/ginac/color.h
@@ -57,6 +57,10 @@ protected:
        unsigned return_type(void) const { return return_types::noncommutative; }
        unsigned return_type_tinfo(void) const { return TINFO_color + representation_label; }
 
+       // non-virtual functions in this class
+public:
+       unsigned char get_representation_label(void) const {return representation_label;}
+
        // member variables
 private:
        unsigned char representation_label; /**< Representation label to distinguish independent color matrices coming from separated fermion lines */
@@ -149,6 +153,12 @@ 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 the trace of an expression containing color objects with a
+ *  specified representation label.
+ *
+ *  @param rl Representation label */
+ex color_trace(const ex & e, unsigned char rl = 0);
+
 
 } // namespace GiNaC