mibuild/generic_platform: add recursive parameter to add_source_dir

m-labs · Aug 2, 2014 · a0d0742 · a0d0742
1 parent 8baa957
commit a0d0742
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/mibuild/generic_platform.py b/mibuild/generic_platform.py
@@ -222,12 +222,20 @@ def add_sources(self, path, *filenames, language=None):
 		for f in filenames:
 			self.add_source(os.path.join(path, f), language)
 
-	def add_source_dir(self, path):
-		for root, dirs, files in os.walk(path):
-			for filename in files:
-				language = tools.language_by_filename(filename)
-				if language is not None:
-					self.add_source(os.path.join(root, filename), language)
+	def add_source_dir(self, path, recursive=True):
+		dir_files = []
+		if recursive:
+			for root, dirs, files in os.walk(path):
+				for filename in files:
+					dir_files.append(os.path.join(root, filename))
+		else:
+			for item in os.listdir(path):
+				if os.path.isfile(os.path.join(path, item)):
+					dir_files.append(os.path.join(path, item))
+		for filename in dir_files:
+			language = tools.language_by_filename(filename)
+			if language is not None:
+				self.add_source(filename, language)
 
 	def add_verilog_include_path(self, path):
 		self.verilog_include_paths.append(os.path.abspath(path))