from direct.directnotify import DirectNotifyGlobal
from direct.showbase import PythonUtil
from toontown.toonbase import TTLocalizer
from panda3d.core import *
from KartShopGlobals import *
import types, copy

KartDNA = PythonUtil.Enum('bodyType, bodyColor, accColor,                             ebType, spType, fwwType,                             bwwType, rimsType, decalType')
InvalidEntry = -1
KartInfo = PythonUtil.Enum('name, model, cost, viewDist, decalId, LODmodel1, LODmodel2')
AccInfo = PythonUtil.Enum('name, model, cost, texCard, attach')
kNames = TTLocalizer.KartDNA_KartNames
KartDict = {0: (kNames[0],
     'phase_6/models/karting/Kart1_Final',
     100,
     7.0,
     'kart1',
     'phase_6/models/karting/Kart1_LOD_Final',
     'phase_6/models/karting/Kart1_LOD_Final',
     (Point3(1.5, 8.0, -0.5), Point3(1.5, 0.0, 2.0))),
 1: (kNames[1],
     'phase_6/models/karting/Kart2_Final',
     7500,
     7.0,
     'kart2',
     'phase_6/models/karting/Kart2_LOD2_Final',
     'phase_6/models/karting/Kart2_LOD3_Final',
     (Point3(0.25, 7, -2), Point3(1.25, -3, 0))),
 2: (kNames[2],
     'phase_6/models/karting/Kart3_Final',
     2500,
     8.5,
     'kart3',
     'phase_6/models/karting/Kart3_Final_LOD2',
     'phase_6/models/karting/Kart3_Final_LOD3',
     (Point3(1.25, 4.0, 1.0), Point3(1.25, -3.0, 2.5)))}
aNames = TTLocalizer.KartDNA_AccNames
AccessoryDict = {0: (aNames[1000],
     'phase_6/models/karting/accessory_frontMiddle_0',
     200,
     'accessory_frontMiddle_0_gui',
     'ebNode_0'),
 1: (aNames[1001],
     'phase_6/models/karting/accessory_frontMiddle_1',
     1250,
     'accessory_frontMiddle_1_gui',
     'ebNode_0'),
 2: (aNames[1002],
     'phase_6/models/karting/accessory_frontMiddle_2',
     450,
     'accessory_frontMiddle_2_gui',
     'ebNode_1'),
 3: (aNames[1003],
     'phase_6/models/karting/accessory_frontMiddle_3',
     5000,
     'accessory_frontMiddle_3_gui',
     'ebNode_1'),
 4: (aNames[1004],
     'phase_6/models/karting/accessory_frontMiddle_4',
     800,
     'accessory_frontMiddle_4_gui',
     'ebNode_0'),
 5: (aNames[1005],
     'phase_6/models/karting/accessory_frontMiddle_5',
     200,
     'accessory_frontMiddle_5_gui',
     'ebNode_0'),
 6: (aNames[1006],
     'phase_6/models/karting/accessory_frontMiddle_6',
     450,
     'accessory_frontMiddle_6_gui',
     'ebNode_0'),
 7: (aNames[1007],
     'phase_6/models/karting/accessory_frontMiddle_7',
     800,
     'accessory_frontMiddle_7_gui',
     'ebNode_0'),
 8: (aNames[1008],
     'phase_6/models/karting/accessory_frontMiddle_8',
     200,
     'accessory_frontMiddle_8_gui',
     'ebNode_1'),
 9: (aNames[1009],
     'phase_6/models/karting/accessory_frontMiddle_9',
     800,
     'accessory_frontMiddle_9_gui',
     'ebNode_1'),
 10: (aNames[1010],
      'phase_6/models/karting/accessory_frontMiddle_10',
      1250,
      'accessory_frontMiddle_10_gui',
      'ebNode_0'),
 11: (aNames[2000],
      'phase_6/models/karting/accessory_backMiddle_0',
      450,
      'accessory_backMiddle_0_gui',
      'spNode_1'),
 12: (aNames[2001],
      'phase_6/models/karting/accessory_backMiddle_1',
      200,
      'accessory_backMiddle_1_gui',
      'spNode_0'),
 13: (aNames[2002],
      'phase_6/models/karting/accessory_backMiddle_2',
      800,
      'accessory_backMiddle_2_gui',
      'spNode_1'),
 14: (aNames[2003],
      'phase_6/models/karting/accessory_backMiddle_3',
      1250,
      'accessory_backMiddle_3_gui',
      'spNode_1'),
 15: (aNames[2004],
      'phase_6/models/karting/accessory_backMiddle_4',
      5000,
      'accessory_backMiddle_4_gui',
      'spNode_1'),
 16: (aNames[2005],
      'phase_6/models/karting/accessory_backMiddle_5',
      800,
      'accessory_backMiddle_5_gui',
      'spNode_1'),
 17: (aNames[2006],
      'phase_6/models/karting/accessory_backMiddle_6',
      450,
      'accessory_backMiddle_6_gui',
      'spNode_0'),
 18: (aNames[2007],
      'phase_6/models/karting/accessory_backMiddle_7',
      1250,
      'accessory_backMiddle_7_gui',
      'spNode_1'),
 19: (aNames[3000],
      'phase_6/models/karting/accessory_front_ww_0',
      200,
      'accessory_front_ww_0_gui',
      '%sFWWNode_0'),
 20: (aNames[3001],
      'phase_6/models/karting/accessory_front_ww_1',
      200,
      'accessory_front_ww_1_gui',
      '%sFWWNode_1'),
 21: (aNames[3002],
      'phase_6/models/karting/accessory_front_ww_2',
      800,
      'accessory_front_ww_2_gui',
      '%sFWWNode_2'),
 22: (aNames[3003],
      'phase_6/models/karting/accessory_front_ww_3',
      5000,
      'accessory_front_ww_3_gui',
      '%sFWWNode_2'),
 23: (aNames[3004],
      'phase_6/models/karting/accessory_front_ww_4',
      1250,
      'accessory_front_ww_4_gui',
      '%sFWWNode_2'),
 24: (aNames[3005],
      'phase_6/models/karting/accessory_front_ww_5',
      800,
      'accessory_front_ww_5_gui',
      '%sFWWNode_1'),
 25: (aNames[3006],
      'phase_6/models/karting/accessory_front_ww_6',
      1250,
      'accessory_front_ww_6_gui',
      '%sFWWNode_2'),
 26: (aNames[3007],
      'phase_6/models/karting/accessory_front_ww_7',
      450,
      'accessory_front_ww_7_gui',
      '%sFWWNode_2'),
 27: (aNames[4000],
      'phase_6/models/karting/accessory_rear_ww_0',
      800,
      'accessory_rear_ww_0_gui',
      '%sBWWNode_0'),
 28: (aNames[4001],
      'phase_6/models/karting/accessory_rear_ww_1',
      200,
      'accessory_rear_ww_1_gui',
      '%sBWWNode_1'),
 29: (aNames[4002],
      'phase_6/models/karting/accessory_rear_ww_2',
      200,
      'accessory_rear_ww_2_gui',
      '%sBWWNode_2'),
 30: (aNames[4003],
      'phase_6/models/karting/accessory_rear_ww_3',
      1250,
      'accessory_rear_ww_3_gui',
      '%sBWWNode_0'),
 31: (aNames[4004],
      'phase_6/models/karting/accessory_rear_ww_4',
      200,
      'accessory_rear_ww_4_gui',
      '%sBWWNode_2'),
 32: (aNames[4005],
      'phase_6/models/karting/accessory_rear_ww_5',
      800,
      'accessory_rear_ww_5_gui',
      '%sBWWNode_2'),
 33: (aNames[4006],
      'phase_6/models/karting/accessory_rear_ww_6',
      450,
      'accessory_rear_ww_6_gui',
      '%sBWWNode_2'),
 34: (aNames[4007],
      'phase_6/models/karting/accessory_rear_ww_7',
      5000,
      'accessory_rear_ww_7_gui',
      '%sBWWNode_2'),
 35: (aNames[4008],
      'phase_6/models/karting/accessory_rear_ww_8',
      1250,
      'accessory_rear_ww_8_gui',
      '%sBWWNode_0'),
 36: (aNames[4009],
      'phase_6/models/karting/accessory_rear_ww_9',
      1250,
      'accessory_rear_ww_9_gui',
      '%sBWWNode_0'),
 37: (aNames[4010],
      'phase_6/models/karting/accessory_rear_ww_10',
      450,
      'accessory_rear_ww_10_gui',
      '%sBWWNode_2'),
 38: (aNames[4011],
      'phase_6/models/karting/accessory_rear_ww_11',
      800,
      'accessory_rear_ww_11_gui',
      '%sBWWNode_2'),
 39: (aNames[4012],
      'phase_6/models/karting/accessory_rear_ww_12',
      1250,
      'accessory_rear_ww_12_gui',
      '%sBWWNode_2'),
 40: (aNames[5000],
      'phase_6/maps/kart_Rim_1',
      InvalidEntry,
      'kart_Rim_1'),
 41: (aNames[5001],
      'phase_6/maps/kart_Rim_2',
      450,
      'kart_Rim_2'),
 42: (aNames[5002],
      'phase_6/maps/kart_Rim_3',
      100,
      'kart_Rim_3'),
 43: (aNames[5003],
      'phase_6/maps/kart_Rim_4',
      800,
      'kart_Rim_4'),
 44: (aNames[5004],
      'phase_6/maps/kart_Rim_5',
      100,
      'kart_Rim_5'),
 45: (aNames[5005],
      'phase_6/maps/kart_Rim_6',
      200,
      'kart_Rim_6'),
 46: (aNames[5006],
      'phase_6/maps/kart_Rim_7',
      200,
      'kart_Rim_7'),
 47: (aNames[5007],
      'phase_6/maps/kart_Rim_8',
      200,
      'kart_Rim_8'),
 48: (aNames[5008],
      'phase_6/maps/kart_Rim_9',
      200,
      'kart_Rim_9'),
 49: (aNames[5009],
      'phase_6/maps/kart_Rim_10',
      200,
      'kart_Rim_10'),
 50: (aNames[5010],
      'phase_6/maps/kart_Rim_11',
      200,
      'kart_Rim_11'),
 51: (aNames[5011],
      'phase_6/maps/kart_Rim_12',
      800,
      'kart_Rim_12'),
 52: (aNames[5012],
      'phase_6/maps/kart_Rim_13',
      450,
      'kart_Rim_13'),
 53: (aNames[5013],
      'phase_6/maps/kart_Rim_14',
      1250,
      'kart_Rim_14'),
 54: (aNames[5014],
      'phase_6/maps/kart_Rim_15',
      5000,
      'kart_Rim_15'),
 55: (aNames[6000],
      1,
      200,
      '%s_SideDecal_1'),
 56: (aNames[6001],
      2,
      450,
      '%s_SideDecal_2'),
 57: (aNames[6002],
      3,
      100,
      '%s_SideDecal_3'),
 58: (aNames[6003],
      4,
      5000,
      '%s_SideDecal_4'),
 59: (aNames[6004],
      5,
      200,
      '%s_SideDecal_5'),
 60: (aNames[6005],
      6,
      200,
      '%s_SideDecal_6'),
 61: (aNames[6006],
      7,
      450,
      '%s_SideDecal_7'),
 62: (aNames[6007],
      8,
      100,
      '%s_SideDecal_8'),
 63: (aNames[6008],
      9,
      1250,
      '%s_SideDecal_9'),
 64: (aNames[6009],
      10,
      800,
      '%s_SideDecal_10'),
 65: (aNames[7000], VBase4(0.8, 1.0, 0.5, 1.0), 100),
 66: (aNames[7001], VBase4(0.96875, 0.691406, 0.699219, 1.0), 100),
 67: (aNames[7002], VBase4(0.933594, 0.265625, 0.28125, 1.0), 2500),
 68: (aNames[7003], VBase4(0.863281, 0.40625, 0.417969, 1.0), 900),
 69: (aNames[7004], VBase4(0.710938, 0.234375, 0.4375, 1.0), 1600),
 70: (aNames[7005], VBase4(0.570312, 0.449219, 0.164062, 1.0), 1600),
 71: (aNames[7006], VBase4(0.640625, 0.355469, 0.269531, 1.0), 1600),
 72: (aNames[7007], VBase4(0.996094, 0.695312, 0.511719, 1.0), 100),
 73: (aNames[7008], VBase4(0.832031, 0.5, 0.296875, 1.0), 1600),
 74: (aNames[7009], VBase4(0.992188, 0.480469, 0.167969, 1.0), 1600),
 75: (aNames[7010], VBase4(0.996094, 0.898438, 0.320312, 1.0), 2500),
 76: (aNames[7011], VBase4(0.996094, 0.957031, 0.597656, 1.0), 900),
 77: (aNames[7012], VBase4(0.855469, 0.933594, 0.492188, 1.0), 900),
 78: (aNames[7013], VBase4(0.550781, 0.824219, 0.324219, 1.0), 900),
 79: (aNames[7014], VBase4(0.242188, 0.742188, 0.515625, 1.0), 1600),
 80: (aNames[7015], VBase4(0.304688, 0.96875, 0.402344, 1.0), 2500),
 81: (aNames[7016], VBase4(0.433594, 0.90625, 0.835938, 1.0), 900),
 82: (aNames[7017], VBase4(0.347656, 0.820312, 0.953125, 1.0), 900),
 83: (aNames[7018], VBase4(0.191406, 0.5625, 0.773438, 1.0), 1600),
 84: (aNames[7019], VBase4(0.558594, 0.589844, 0.875, 1.0), 900),
 85: (aNames[7020], VBase4(0.285156, 0.328125, 0.726562, 1.0), 2500),
 86: (aNames[7021], VBase4(0.460938, 0.378906, 0.824219, 1.0), 900),
 87: (aNames[7022], VBase4(0.546875, 0.28125, 0.75, 1.0), 1600),
 88: (aNames[7023], VBase4(0.726562, 0.472656, 0.859375, 1.0), 1600),
 89: (aNames[7024], VBase4(0.898438, 0.617188, 0.90625, 1.0), 1600),
 90: (aNames[7025], VBase4(0.7, 0.7, 0.8, 1.0), 2500),
 91: (aNames[7026], VBase4(0.3, 0.3, 0.35, 1.0), 10000)}
AccessoryTypeDict = {KartDNA.ebType: [0,
                  1,
                  2,
                  3,
                  4,
                  5,
                  6,
                  7,
                  8,
                  9,
                  10],
 KartDNA.spType: [11,
                  12,
                  13,
                  14,
                  15,
                  16,
                  17,
                  18],
 KartDNA.fwwType: [19,
                   20,
                   21,
                   22,
                   23,
                   24,
                   25,
                   26],
 KartDNA.bwwType: [27,
                   28,
                   29,
                   30,
                   31,
                   32,
                   33,
                   34,
                   35,
                   36,
                   37,
                   38,
                   39],
 KartDNA.rimsType: [40,
                    41,
                    42,
                    43,
                    44,
                    45,
                    46,
                    47,
                    48,
                    49,
                    50,
                    51,
                    52,
                    53,
                    54],
 KartDNA.decalType: [55,
                     56,
                     57,
                     58,
                     59,
                     60,
                     61,
                     62,
                     63,
                     64],
 KartDNA.bodyColor: [65,
                     66,
                     67,
                     68,
                     69,
                     70,
                     71,
                     72,
                     73,
                     74,
                     75,
                     76,
                     77,
                     78,
                     79,
                     80,
                     81,
                     82,
                     83,
                     84,
                     85,
                     86,
                     87,
                     88,
                     89,
                     90,
                     91]}
AccessoryTypeNameDict = [None,
 'KartShtikerBodyColor',
 'KartShtikerAccColor',
 'KartShtikerEngineBlock',
 'KartShtikerSpoiler',
 'KartShtikerFrontWheelWell',
 'KartShtikerBackWheelWell',
 'KartShtikerRim',
 'KartShtikerDecal']

def checkNumFieldsValidity(numFields):
    return KartDNA.decalType == numFields - 1


def checkKartFieldValidity(field):
    if field < KartDNA.bodyType or field > KartDNA.decalType:
        return 0
    return 1


def getNumFields():
    return KartDNA.decalType + 1


def getKartModelPath(kartType, lodLevel = 0):
    if lodLevel == 1:
        return KartDict.get(kartType)[KartInfo.LODmodel1]
    if lodLevel == 2:
        return KartDict.get(kartType)[KartInfo.LODmodel2]
    return KartDict.get(kartType)[KartInfo.model]


def getKartViewDist(kartType):
    return KartDict.get(kartType)[KartInfo.viewDist]


def getDecalId(kartType):
    return KartDict.get(kartType)[KartInfo.decalId]


def getAccessory(accId):
    return AccessoryDict.get(accId)[KartInfo.model]


def getAccessoryAttachNode(accId):
    accInfo = AccessoryDict.get(accId)
    if len(accInfo) == 5:
        return accInfo[4]
    return None


def getTexCardNode(accId):
    accInfo = AccessoryDict.get(accId)
    if len(accInfo) <= 5:
        return accInfo[3]
    return None


def checkKartDNAValidity(dna):
    if not checkNumFieldsValidity(len(dna)):
        return 0
    for field in xrange(len(dna)):
        if field == KartDNA.bodyType:
            if dna[field] not in KartDict.keys():
                return 0
        elif field == KartDNA.bodyColor or field == KartDNA.accColor:
            accList = [InvalidEntry] + AccessoryTypeDict.get(KartDNA.bodyColor)
            if dna[field] not in accList:
                return 0
        else:
            accList = [InvalidEntry] + AccessoryTypeDict.get(field)
            if dna[field] not in accList:
                return 0

    return 1


def getDefaultColor():
    return VBase4(1, 1, 1, 1)


def getDefaultRim():
    return AccessoryTypeDict[KartDNA.rimsType][0]


def getDefaultAccessory(category):
    if category in [KartDNA.bodyColor, KartDNA.accColor]:
        return getDefaultColor()
    elif category == KartDNA.rimsType:
        return getDefaultRim()
    else:
        return InvalidEntry


def getAccessoryItemList(accessoryType):
    return [ AccessoryDict[itemId] for itemId in AccessoryTypeDict[accessoryType] ]


def getKartTypeInfo(type):
    if type in KartDict.keys():
        return KartDict[type]
    return InvalidEntry


def getAccessoryInfo(index):
    if index in AccessoryDict.keys():
        return AccessoryDict[index]
    return InvalidEntry


def getAccessoryType(accessoryId):
    for key in AccessoryTypeDict.keys():
        if accessoryId in AccessoryTypeDict[key]:
            return key

    return InvalidEntry


def getAccessoryDictFromOwned(accessoryOwnedList, pType = -1):
    accessDict = copy.deepcopy(AccessoryTypeDict)
    accessDict[KartDNA.rimsType].remove(getDefaultRim())
    for accOwnedId in accessoryOwnedList:
        type = getAccessoryType(accOwnedId)
        if type != InvalidEntry and accOwnedId in accessDict[type]:
            accessDict[type].remove(accOwnedId)

    if pType != -1:
        return accessDict[pType]
    else:
        return accessDict


def getAccessDictByType(accessoryOwnedList):
    accessDict = {}
    if type(accessoryOwnedList) == types.ListType:
        for accOwnedId in accessoryOwnedList:
            accType = getAccessoryType(accOwnedId)
            if accType != InvalidEntry:
                if accType not in accessDict:
                    accessDict[accType] = []
                accessDict[accType].append(accOwnedId)

    else:
        print 'KartDNA: getAccessDictByType: bad accessory list: ', accessoryOwnedList
    return accessDict


def getKartCost(kartID):
    if kartID in KartDict:
        return KartDict[kartID][KartInfo.cost]
    else:
        return 'key error'


def getAccCost(accID):
    return AccessoryDict[accID][AccInfo.cost]


def getAccName(accID):
    try:
        return AccessoryDict[accID][AccInfo.name]
    except:
        return TTLocalizer.KartShtikerDefault