From c4aabb54fe450a4f5480cacd9ebe83a2f9af8bfb Mon Sep 17 00:00:00 2001
From: Peizhao Ou <peizhaoo@uci.edu>
Date: Wed, 24 Feb 2016 00:26:29 -0800
Subject: [PATCH 1/1] edits

---
 src/edu/uci/eecs/codeGenerator/CodeGenerator.java | 8 ++++++++
 1 file changed, 8 insertions(+)

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));
-- 
2.34.1