X-Git-Url: https://ginac.de/ginac.git//ginac.git?a=blobdiff_plain;f=ginac%2Fcolor.h;h=700493253adad38b154c6040586753b7f6465861;hb=fdade42bf337fa1650bb9d69ee06e3bd1562c951;hp=c5c57c5598942080a6f467b4ead100d17c166558;hpb=61b8ecb02c8212efbc15f59face8a9c620dc823c;p=ginac.git

diff --git a/ginac/color.h b/ginac/color.h
index c5c57c55..70049325 100644
--- a/ginac/color.h
+++ b/ginac/color.h
@@ -31,24 +31,23 @@ namespace GiNaC {
 
 /** This class holds a generator T_a or the unity element of the Lie algebra
  *  of SU(3), as used for calculations in quantum chromodynamics. A
- *  representation label (an unsigned integer) is used to distinguish
- *  elements from different Lie algebra representations (only objects with
- *  the same label "interact" with each other). 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. */
+ *  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
+ *  group, not a specific matrix representation. The indices used for color
+ *  objects should not have a variance. */
 class color : public indexed
 {
 	GINAC_DECLARE_REGISTERED_CLASS(color, indexed)
 
 	// other constructors
 public:
-	color(const ex & b, unsigned rl = 0);
-	color(const ex & b, const ex & i1, unsigned rl = 0);
+	color(const ex & b, unsigned char rl = 0);
+	color(const ex & b, const ex & i1, unsigned char rl = 0);
 
 	// internal constructors
-	color(unsigned rl, const exvector & v, bool discardable = false);
-	color(unsigned rl, exvector * vp); // vp will be deleted
+	color(unsigned char rl, const exvector & v, bool discardable = false);
+	color(unsigned char rl, exvector * vp); // vp will be deleted
 
 	// functions overriding virtual functions from base classes
 protected:
@@ -56,11 +55,15 @@ protected:
 	ex thisexprseq(const exvector & v) const;
 	ex thisexprseq(exvector * vp) const;
 	unsigned return_type(void) const { return return_types::noncommutative; }
-	unsigned return_type_tinfo(void) const { return TINFO_color; }
+	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 representation_label; /**< Representation label to distinguish independent color matrices coming from separated fermion lines */
+	unsigned char representation_label; /**< Representation label to distinguish independent color matrices coming from separated fermion lines */
 };
 
 
@@ -71,7 +74,7 @@ class su3one : public tensor
 
 	// functions overriding virtual functions from bases classes
 public:
-	void print(std::ostream & os, unsigned upper_precedence=0) const;
+	void print(const print_context & c, unsigned level = 0) const;
 };
 
 /** This class represents an su(3) generator. */
@@ -81,7 +84,7 @@ class su3t : public tensor
 
 	// functions overriding virtual functions from bases classes
 public:
-	void print(std::ostream & os, unsigned upper_precedence=0) const;
+	void print(const print_context & c, unsigned level = 0) const;
 };
 
 /** This class represents the tensor of antisymmetric su(3) structure
@@ -92,7 +95,7 @@ class su3f : public tensor
 
 	// functions overriding virtual functions from bases classes
 public:
-	void print(std::ostream & os, unsigned upper_precedence=0) const;
+	void print(const print_context & c, unsigned level = 0) const;
 	ex eval_indexed(const basic & i) const;
 	bool contract_with(exvector::iterator self, exvector::iterator other, exvector & v) const;
 };
@@ -104,7 +107,7 @@ class su3d : public tensor
 
 	// functions overriding virtual functions from bases classes
 public:
-	void print(std::ostream & os, unsigned upper_precedence=0) const;
+	void print(const print_context & c, unsigned level = 0) const;
 	ex eval_indexed(const basic & i) const;
 	bool contract_with(exvector::iterator self, exvector::iterator other, exvector & v) const;
 };
@@ -122,14 +125,14 @@ inline const color &ex_to_color(const ex &e)
  *
  *  @param rl Representation label
  *  @return newly constructed unity element */
-ex color_ONE(unsigned rl = 0);
+ex color_ONE(unsigned char rl = 0);
 
 /** Create an su(3) generator.
  *
  *  @param a Index
  *  @param rl Representation label
  *  @return newly constructed unity generator */
-ex color_T(const ex & a, unsigned rl = 0);
+ex color_T(const ex & a, unsigned char rl = 0);
 
 /** Create an su(3) antisymmetric structure constant.
  *
@@ -150,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