Add CArray to the expression language

ianhinder · Mar 18, 2014 · a4b5f38 · a4b5f38
1 parent 4713bc4
commit a4b5f38
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/Tools/CodeGen/CodeGenKranc.m b/Tools/CodeGen/CodeGenKranc.m
@@ -291,6 +291,9 @@
     code = LineBreak[FlattenBlock[code], 70] <> "\n";
     {code}];
 
+Format[CArray[id_, {args__}], CForm] :=
+  SequenceForm[id, "[", Sequence @@ Riffle[{args}, ","], "]"];
+
 GenerateCodeFromExpression[expr_, vectorise_, noSimplify:Boolean : False] :=
   Module[{cleanExpr, code},
     cleanExpr = ProcessExpression[expr, vectorise, noSimplify];

diff --git a/Tools/CodeGen/Kranc.m b/Tools/CodeGen/Kranc.m
@@ -154,4 +154,7 @@
    "cse" -> CSE,
    "cakernel" -> UseCaKernel};
 
+(* CodeGenKranc.m *)
+CArray;
+
 EndPackage[];