1 files changed, 114 insertions, 0 deletions
diff --git a/waflib/Tools/winres.py b/waflib/Tools/winres.py
new file mode 100644
index 00000000..ddddd949
--- /dev/null
+++ b/waflib/Tools/winres.py
@@ -0,0 +1,114 @@
+#!/usr/bin/env python
+# encoding: utf-8
+# Brant Young, 2007
+
+"Process *.rc* files for C/C++: X{.rc -> [.res|.rc.o]}"
+
+import re, traceback
+from waflib import Task, Logs, Utils
+from waflib.TaskGen import extension
+from waflib.Tools import c_preproc
+
+@extension('.rc')
+def rc_file(self, node):
+	"""
+	Bind the .rc extension to a winrc task
+	"""
+	obj_ext = '.rc.o'
+	if self.env['WINRC_TGT_F'] == '/fo':
+		obj_ext = '.res'
+	rctask = self.create_task('winrc', node, node.change_ext(obj_ext))
+	try:
+		self.compiled_tasks.append(rctask)
+	except AttributeError:
+		self.compiled_tasks = [rctask]
+
+re_lines = re.compile(
+	'(?:^[ \t]*(#|%:)[ \t]*(ifdef|ifndef|if|else|elif|endif|include|import|define|undef|pragma)[ \t]*(.*?)\s*$)|'\
+	'(?:^\w+[ \t]*(ICON|BITMAP|CURSOR|HTML|FONT|MESSAGETABLE|TYPELIB|REGISTRY|D3DFX)[ \t]*(.*?)\s*$)',
+	re.IGNORECASE | re.MULTILINE)
+
+class rc_parser(c_preproc.c_parser):
+	def filter_comments(self, filepath):
+		code = Utils.readf(filepath)
+		if c_preproc.use_trigraphs:
+			for (a, b) in c_preproc.trig_def: code = code.split(a).join(b)
+		code = c_preproc.re_nl.sub('', code)
+		code = c_preproc.re_cpp.sub(c_preproc.repl, code)
+		ret = []
+		for m in re.finditer(re_lines, code):
+			if m.group(2):
+				ret.append((m.group(2), m.group(3)))
+			else:
+				ret.append(('include', m.group(5)))
+		return ret
+
+	def addlines(self, node):
+		self.currentnode_stack.append(node.parent)
+		filepath = node.abspath()
+
+		self.count_files += 1
+		if self.count_files > c_preproc.recursion_limit:
+			raise c_preproc.PreprocError("recursion limit exceeded")
+		pc = self.parse_cache
+		Logs.debug('preproc: reading file %r', filepath)
+		try:
+			lns = pc[filepath]
+		except KeyError:
+			pass
+		else:
+			self.lines.extend(lns)
+			return
+
+		try:
+			lines = self.filter_comments(filepath)
+			lines.append((c_preproc.POPFILE, ''))
+			lines.reverse()
+			pc[filepath] = lines
+			self.lines.extend(lines)
+		except IOError:
+			raise c_preproc.PreprocError("could not read the file %s" % filepath)
+		except Exception:
+			if Logs.verbose > 0:
+				Logs.error("parsing %s failed" % filepath)
+				traceback.print_exc()
+
+class winrc(Task.Task):
+	"""
+	Task for compiling resource files
+	"""
+	run_str = '${WINRC} ${WINRCFLAGS} ${CPPPATH_ST:INCPATHS} ${DEFINES_ST:DEFINES} ${WINRC_TGT_F} ${TGT} ${WINRC_SRC_F} ${SRC}'
+	color   = 'BLUE'
+
+	def scan(self):
+		tmp = rc_parser(self.generator.includes_nodes)
+		tmp.start(self.inputs[0], self.env)
+		nodes = tmp.nodes
+		names = tmp.names
+
+		if Logs.verbose:
+			Logs.debug('deps: deps for %s: %r; unresolved %r' % (str(self), nodes, names))
+
+		return (nodes, names)
+
+def configure(conf):
+	"""
+	Detect the programs RC or windres, depending on the C/C++ compiler in use
+	"""
+	v = conf.env
+	v['WINRC_TGT_F'] = '-o'
+	v['WINRC_SRC_F'] = '-i'
+
+	# find rc.exe
+	if not conf.env.WINRC:
+		if v.CC_NAME == 'msvc':
+			conf.find_program('RC', var='WINRC', path_list = v['PATH'])
+			v['WINRC_TGT_F'] = '/fo'
+			v['WINRC_SRC_F'] = ''
+		else:
+			conf.find_program('windres', var='WINRC', path_list = v['PATH'])
+	if not conf.env.WINRC:
+		conf.fatal('winrc was not found!')
+
+	v['WINRCFLAGS'] = []
+