Improved Uncontract to work with tensors (using DataType FCTensor) th…

…at are not defined in the Global` context.
FeynCalc · Jan 6, 2016 · 7deccf5 · 7deccf5
1 parent dc95bab
commit 7deccf5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/FeynCalc/Lorentz/TensorFunction.m b/FeynCalc/Lorentz/TensorFunction.m
@@ -40,6 +40,8 @@
 			at = Last[ef],
 			f = ef
 		];
+		DataType[ef,FCTensor]=True;
+
 		With[ {tf = f},
 			If [at=!="",
 				Which[

diff --git a/FeynCalc/Lorentz/Uncontract.m b/FeynCalc/Lorentz/Uncontract.m
@@ -122,8 +122,8 @@
 						Pair[Momentum[c,dimSelect[d]], li] Pair[Momentum[pe,dimSelect[d]],lidr[li]])/;MemberQ[par,pe]} /. times -> Times;
 
 				(* Uncontract tensor functions *)
-				If[ !FreeQ[nex, (tf_/;Context[tf]==="Global`")[___,Momentum[(q | Polarization[q,__]),___],___]],
-					nex = nex //. { (tf_/;Context[tf]==="Global`")[a___,Momentum[(c: q | Polarization[q,__]),d_:4],b___] :>
+				If[ !FreeQ[nex, (tf_/;DataType[tf,FCTensor])[___,Momentum[(q | Polarization[q,__]),___],___]],
+					nex = nex //. { (tf_/;DataType[tf,FCTensor])[a___,Momentum[(c: q | Polarization[q,__]),d_:4],b___] :>
 						(li = LorentzIndex[a$AL[inc = inc+1],dimSelect[d]];
 						tf[a, li, b] Pair[Momentum[c,dimSelect[d]],lidr[li]])}
 				];

diff --git a/FeynCalc/Shared/DataType.m b/FeynCalc/Shared/DataType.m
@@ -25,6 +25,9 @@
 PositiveNumber::usage =  "PositiveNumber is a data type. E.g. \
 DataType[Epsilon, PositiveNumber] = True (by default). ";
 
+FCTensor::usage =  "FCTensor is a data type. E.g. \
+DataType[R, FCTensor] = True. ";
+
 DataType::usage =
 "DataType[exp, type] = True   defines the object exp to have datatype type. \
 DataType[exp1, exp2, ..., type] defines the objects exp1, exp2, ... to \