From f99bf3e9e1d16505a3f7a74d32e1b03d58dd7f11 Mon Sep 17 00:00:00 2001
From: Robin Dunn <robin@alldunn.com>
Date: Fri, 9 Mar 2018 21:26:56 -0800
Subject: [PATCH] Add the ability to inject wxClassInfo macros into the sip
 generated wrapper classes. Only needed for wxTreeCtrl so far.

---
 build.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/build.py b/build.py
index 8fc83da4..d452bc1f 100755
--- a/build.py
+++ b/build.py
@@ -1109,6 +1109,7 @@ def cmd_sip(options, args):
             (sip, cfg.SIPOPTS, tmpdir, sbf, pycode, src_name)
         runcmd(cmd)
 
+        classesNeedingClassInfo = { 'sip_corewxTreeCtrl.cpp' : 'wxTreeCtrl', }
 
         def processSrc(src, keepHashLines=False):
             with textfile_open(src, 'rt') as f:
@@ -1120,8 +1121,29 @@ def cmd_sip(options, args):
                     # ...or totally remove them by replacing those lines with ''
                     import re
                     srcTxt = re.sub(r'^#line.*\n', '', srcTxt, flags=re.MULTILINE)
+                className = classesNeedingClassInfo.get(os.path.basename(src))
+                if className:
+                    srcTxt = injectClassInfo(className, srcTxt)
             return srcTxt
 
+        def injectClassInfo(className, srcTxt):
+            # inject wxClassInfo macros into the sip generated wrapper class
+            lines = srcTxt.splitlines()
+            # find the beginning of the class declaration
+            for idx, line in enumerate(lines):
+                if line.startswith('class sip{}'.format(className)):
+                    # next line is '{', insert after that
+                    lines[idx+2:idx+2] = ['wxDECLARE_ABSTRACT_CLASS(sip{});'.format(className)]
+                    break
+            # find the '};' that terminates the class
+            for idx, line in enumerate(lines[idx+2:], idx+2):
+                if line.startswith('};'):
+                    lines[idx+1:idx+1] = ['\nwxIMPLEMENT_ABSTRACT_CLASS(sip{0}, {0});'.format(className)]
+                    break
+            # join it back up and return
+            return '\n'.join(lines)
+
+
         # Check each file in tmpdir to see if it is different than the same file
         # in cfg.SIPOUT. If so then copy the new one to cfg.SIPOUT, otherwise
         # ignore it.