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

import sys

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()