From: Peizhao Ou <peizhaoo@uci.edu>
Date: Wed, 24 Feb 2016 08:26:29 +0000 (-0800)
Subject: edits
X-Git-Url: http://plrg.eecs.uci.edu/git/?p=cdsspec-compiler.git;a=commitdiff_plain;h=c4aabb54fe450a4f5480cacd9ebe83a2f9af8bfb

edits
---

diff --git a/src/edu/uci/eecs/codeGenerator/CodeGenerator.java b/src/edu/uci/eecs/codeGenerator/CodeGenerator.java
index 93809c6..41a5bdd 100644
--- a/src/edu/uci/eecs/codeGenerator/CodeGenerator.java
+++ b/src/edu/uci/eecs/codeGenerator/CodeGenerator.java
@@ -178,6 +178,14 @@ public class CodeGenerator {
 			additionIdx = 0;
 			curAddition = additions.codeAdditions.get(0);
 		}
+		
+		// Include the header for C/C++ files (.c/.cc/.cpp)
+		String name = file.getName();
+		if (name.endsWith(".c") || name.endsWith(".cc") || name.endsWith(".cpp")) {
+			newCode.addLine(CodeGeneratorUtils.Comment("Add the" + SpecNaming.CDSSpecGeneratedHeader + " header file"));
+			newCode.addLine(CodeGeneratorUtils.IncludeHeader(SpecNaming.CDSSpecGeneratedHeader));
+			newCode.addLine("");
+		}
 
 		try {
 			br = new BufferedReader(new FileReader(file));