CodeGen.m: Rewrite FlattenBlock and make it more strict

ianhinder · Oct 6, 2011 · a50298e · a50298e
1 parent 25b122f
commit a50298e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Tools/CodeGen/CodeGen.m b/Tools/CodeGen/CodeGen.m
@@ -176,7 +176,10 @@
 
 (* Code generation utilities; not specific to any language *)
 
-FlattenBlock[b_] := Apply[StringJoin,Map[ToString,If[! AtomQ[b], Flatten[b, Infinity], b]]];
+(* FlattenBlock *)
+FlattenBlock[x_String] := x;
+FlattenBlock[l_List] := StringJoin@@Map[FlattenBlock, l];
+FlattenBlock[a_?AtomQ] := ToString[a];
 
 indentBlock[block_] :=
   StringDrop["  " <> StringReplace[FlattenBlock[block], {"\n" -> "\n  "}],-2];