Poodletooth-iLand/panda/direct/ffi/FFIRename.py

import FFIConstants
from string import *


pythonKeywords = ['and','del','for','is','raise','assert','elif','from','lambda','return','break','else','global','not','try','class','except','if','or','while','continue','exec','import','pass','def','finally','in','print']


methodRenameDictionary = {
    'operator==':  '__eq__',
    'operator!=':  '__ne__',
    'operator<<':  '__lshift__',
    'operator>>':  '__rshift__',
    'operator<':   '__lt__',
    'operator>':   '__gt__',
    'operator<=':  '__le__',
    'operator>=':  '__ge__',
    'operator=':   'assign',
    'operator()':  '__call__',
    'operator[]':  '__getitem__',
    'operator++':  'increment',
    'operator--':  'decrement',
    'operator^':   '__xor__',
    'operator%':   '__mod__',
    'operator!':   'logicalNot',
    'operator~':   'bitwiseNot',
    'operator&':   '__and__',
    'operator&&':  'logicalAnd',
    'operator|':   '__or__',
    'operator||':  'logicalOr',
    'operator+':   '__add__',
    'operator-':   '__sub__',
    'operator*':   '__mul__',
    'operator/':   '__div__',
    'operator+=':  '__iadd__',
    'operator-=':  '__isub__',
    'operator*=':  '__imul__',
    'operator/=':  '__idiv__',
    'operator,':   'concatenate',
    'operator|=':  '__ior__',
    'operator&=':  '__iand__',
    'operator^=':  '__ixor__',
    'operator~=':  'bitwiseNotEqual',
    'operator->':  'dereference',
    'operator<<=': '__ilshift__',
    'operator>>=': '__irshift__',
    'print':       'Cprint',
    'CInterval.setT': '_priv__cSetT',
    }
    
classRenameDictionary = {
    'Loader':                    'PandaLoader',
    'String':                    'CString',
    'LMatrix4f':                 'Mat4',
    'LMatrix3f':                 'Mat3',
    'LVecBase4f':                'VBase4',
    'LVector4f':                 'Vec4',
    'LPoint4f':                  'Point4',
    'LVecBase3f':                'VBase3',
    'LVector3f':                 'Vec3',
    'LPoint3f':                  'Point3',
    'LVecBase2f':                'VBase2',
    'LVector2f':                 'Vec2',
    'LPoint2f':                  'Point2',
    'LQuaternionf':              'Quat',
    'LMatrix4d':                 'Mat4D',
    'LMatrix3d':                 'Mat3D',
    'LVecBase4d':                'VBase4D',
    'LVector4d':                 'Vec4D',
    'LPoint4d':                  'Point4D',
    'LVecBase3d':                'VBase3D',
    'LVector3d':                 'Vec3D',
    'LPoint3d':                  'Point3D',
    'LVecBase2d':                'VBase2D',
    'LVector2d':                 'Vec2D',
    'LPoint2d':                  'Point2D',
    'LQuaterniond':              'QuatD',
    'Plane':                     'PlaneBase',
    'Planef':                    'Plane',
    'Planed':                    'PlaneD',
    'Frustum':                   'FrustumBase',
    'Frustumf':                  'Frustum',
    'Frustumd':                  'FrustumD'
    }


def checkKeyword(cppName):
    if cppName in pythonKeywords:
        cppName = '_' + cppName
    return cppName

# TODO: Make faster - this thing is horribly slow    
def classNameFromCppName(cppName):
    # initialize to empty string
    className = ''
    # These are the characters we want to strip out of the name
    badChars = '!@#$%^&*()<>,.-=+~{}? '
    nextCap = 0
    firstChar = 1
    for char in cppName:
        if (char in badChars):
            continue
        elif (char == '_'):
            nextCap = 1
            continue
        elif (nextCap or firstChar):
            className = className + capitalize(char)
            nextCap = 0
            firstChar = 0
        else:
            className = className + char
    if className in classRenameDictionary:
        className = classRenameDictionary[className]

    if (className == ''):
        FFIConstants.notify.warning('Renaming class: ' + cppName + ' to empty string')
    # FFIConstants.notify.debug('Renaming class: ' + cppName + ' to: ' + className)
    # Note we do not have to check for keywords because class name are capitalized
    return className
    
def nonClassNameFromCppName(cppName):
    className = classNameFromCppName(cppName)
    # Make the first character lowercase
    newName = lower(className[0])+className[1:]
    # Mangle names that happen to be python keywords so they are not anymore
    newName = checkKeyword(newName)
    return newName

def methodNameFromCppName(cppName, className = None):
    methodName = ''
    badChars = ' '
    nextCap = 0
    for char in cppName:
        if (char in badChars):
            continue
        elif (char == '_'):
            nextCap = 1
            continue
        elif nextCap:
            methodName = methodName + capitalize(char)
            nextCap = 0
        else:
            methodName = methodName + char

    if className != None:
        methodName = methodRenameDictionary.get(className + '.' + methodName, methodName)
    methodName = methodRenameDictionary.get(methodName, methodName)
    
    # Mangle names that happen to be python keywords so they are not anymore
    methodName = checkKeyword(methodName)
    return methodName