From 1e5d9a07b5e902991ea1978fa9fe9efbc658e8ea Mon Sep 17 00:00:00 2001
From: D G Turner
Date: Tue, 29 Nov 2011 19:08:51 +0000
Subject: DEVTOOLS: Add flag to tasm-recover to omit blacklisted function
 headers.

This flag when set omits the blacklisted function header from the
dreamgen.h output, rather than including it commented out with //.
---
 devtools/tasmrecover/tasm/cpp.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'devtools/tasmrecover/tasm/cpp.py')

diff --git a/devtools/tasmrecover/tasm/cpp.py b/devtools/tasmrecover/tasm/cpp.py
index 6ffce010ce..f93f994681 100644
--- a/devtools/tasmrecover/tasm/cpp.py
+++ b/devtools/tasmrecover/tasm/cpp.py
@@ -33,7 +33,7 @@ def parse_bin(s):
 	return v
 
 class cpp:
-	def __init__(self, context, namespace, skip_first = 0, blacklist = [], skip_output = [], skip_dispatch_call = False):
+	def __init__(self, context, namespace, skip_first = 0, blacklist = [], skip_output = [], skip_dispatch_call = False, header_omit_blacklisted = False):
 		self.namespace = namespace
 		fname = namespace.lower() + ".cpp"
 		header = namespace.lower() + ".h"
@@ -81,6 +81,7 @@ class cpp:
 		self.failed = list(blacklist)
 		self.skip_output = skip_output
 		self.skip_dispatch_call = skip_dispatch_call
+		self.header_omit_blacklisted = header_omit_blacklisted
 		self.translated = []
 		self.proc_addr = []
 		self.used_data_offsets = set()
@@ -637,7 +638,8 @@ public:
 		self.hd.write("\n")
 		for p in set(self.methods):
 			if p in self.blacklist:
-				self.hd.write("\t//void %s();\n" %p)
+				if self.header_omit_blacklisted == False:
+					self.hd.write("\t//void %s();\n" %p)
 			else:
 				self.hd.write("\tvoid %s();\n" %p)
 
-- 
cgit v1.2.3