Mac/Modules/cg/cgscan.py

# Scan an Apple header file, generating a Python file of generator calls.

import sys
import os
from bgenlocations import TOOLBOXDIR, BGENDIR
sys.path.append(BGENDIR)
from scantools import Scanner_OSX

LONG = "CoreGraphics"
SHORT = "cg"
OBJECTS = ("CGContextRef",
                )
# ADD object typenames here

def main():
    input = [
            "CGContext.h",
    ]
    output = SHORT + "gen.py"
    defsoutput = TOOLBOXDIR + LONG + ".py"
    scanner = MyScanner(input, output, defsoutput)
    scanner.scan()
    scanner.gentypetest(SHORT+"typetest.py")
    scanner.close()
    print "=== Testing definitions output code ==="
    execfile(defsoutput, {}, {})
    print "=== Done scanning and generating, now importing the generated code... ==="
    exec "import " + SHORT + "support"
    print "=== Done.  It's up to you to compile it now! ==="

class MyScanner(Scanner_OSX):

    def destination(self, type, name, arglist):
        classname = "Function"
        listname = "functions"
        if arglist:
            t, n, m = arglist[0]
            if t in OBJECTS and m == "InMode":
                classname = "Method"
                listname = t + "_methods"
            # Special case for the silly first AllocatorRef argument
            if t == 'CFAllocatorRef' and m == 'InMode' and len(arglist) > 1:
                t, n, m = arglist[1]
                if t in OBJECTS and m == "InMode":
                    classname = "MethodSkipArg1"
                    listname = t + "_methods"
        return classname, listname

    def writeinitialdefs(self):
        self.defsfile.write("def FOUR_CHAR_CODE(x): return x\n")

    def makeblacklistnames(self):
        return [
                "CGContextRetain",
                "CGContextRelease",
                ]

    def makegreylist(self):
        return []

    def makeblacklisttypes(self):
        return [
                "float_ptr",
                "CGRect_ptr",
                "CGPoint_ptr",
                "CGColorSpaceRef",
                "CGColorRenderingIntent",
                "CGFontRef",
#                       "char_ptr",
                "CGGlyph_ptr",
                "CGImageRef",
                "CGPDFDocumentRef",
                ]

    def makerepairinstructions(self):
        return [
                ([("char_ptr", "cstring", "InMode"), ("size_t", "length", "InMode")],
                 [("InBuffer", "*", "*")]),
#                       ([("char_ptr", "name", "InMode"),],
#                        [("CCCCC", "*", "*")]),
                ]

if __name__ == "__main__":
    main()