from otp.ai.AIBaseGlobal import * from direct.distributed.ClockDelta import * from otp.avatar import DistributedAvatarAI from otp.avatar.DistributedPlayerAI import DistributedPlayerAI from direct.directnotify import DirectNotifyGlobal from toontown.toonbase import ToontownGlobals import random class DistributedCCharBaseAI(DistributedAvatarAI.DistributedAvatarAI): notify = DirectNotifyGlobal.directNotify.newCategory('DistributedCCharBaseAI') def __init__(self, air, name): DistributedAvatarAI.DistributedAvatarAI.__init__(self, air) self.setName(name) self.exitOff() self.transitionToCostume = 0 self.diffPath = None return def generate(self): DistributedAvatarAI.DistributedAvatarAI.generate(self) if config.GetBool('classic-char-client-spam', 0): self._ccharSpamTask = taskMgr.add(self._simSpam, 'cchar-spam-%s' % serialNum()) def _simSpam(self, task): AvEnter = 'avatarEnter' AvExit = 'avatarExit' SetChat = 'setNearbyAvatarChat' SetSC = 'setNearbyAvatarSC' SetSCCustom = 'setNearbyAvatarSCCustom' SetSCToontask = 'setNearbyAvatarSCToontask' msgs = (AvEnter, AvExit, SetChat, SetSC, SetSCCustom, SetSCToontask) msg = random.choice(msgs) r = random.random() fixedAvId = 1000000006 if r < 0.3: avId = random.randrange(1L << 32) else: if r < 0.6: players = self.air.doFindAllOfType('DistributedToonAI')[0] if len(players): player = random.choice(players) avId = player.doId else: avId = fixedAvId else: avId = fixedAvId savedIdFunc = self.air.getAvatarIdFromSender self.air.getAvatarIdFromSender = lambda : avId rrange = random.randrange if msg is AvEnter: self.avatarEnter() elif msg is AvExit: self.avatarExit() else: if msg is SetChat: length = rrange(1024) s = '' for i in xrange(length): s += chr(rrange(1 << 8)) self.setNearbyAvatarChat(s) else: if msg is SetSC: self.setNearbyAvatarSC(rrange(1 << 16)) else: if msg is SetSCCustom: self.setNearbyAvatarSCCustom(rrange(1 << 16)) else: if msg is SetSCToontask: self.setNearbyAvatarSCToontask(rrange(1 << 32), rrange(1 << 32), rrange(1 << 32), rrange(1 << 8)) self.air.getAvatarIdFromSender = savedIdFunc return task.cont def delete(self): self.ignoreAll() if hasattr(self, '_ccharSpamTask'): taskMgr.remove(self._ccharSpamTask) self._ccharSpamTask = None DistributedAvatarAI.DistributedAvatarAI.delete(self) return def exitOff(self): self.__initAttentionSpan() self.__clearNearbyAvatars() def avatarEnter(self): avId = self.air.getAvatarIdFromSender() if avId not in self.air.doId2do: self.air.writeServerEvent('suspicious', avId, 'CCharBaseAI.avatarEnter from unknown avId') return av = self.air.getDo(avId) if not isinstance(av, DistributedPlayerAI): self.air.writeServerEvent('suspicious', avId, 'CCharBaseAI.avatarEnter from non-player object to cchar %s' % self.doId) return if av.zoneId != self.zoneId: self.air.writeServerEvent('suspicious', avId, 'CCharBaseAI.avatarEnter from av in zone %s, my zone is %s' % (av.zoneId, self.zoneId)) return self.notify.debug('adding avatar ' + str(avId) + ' to the nearby avatar list') if avId not in self.nearbyAvatars: self.nearbyAvatars.append(avId) else: self.air.writeServerEvent('suspicious', avId, 'CCharBase.avatarEnter') self.notify.warning('Avatar %s already in nearby avatars!' % avId) self.nearbyAvatarInfoDict[avId] = {} self.nearbyAvatarInfoDict[avId]['enterTime'] = globalClock.getRealTime() self.nearbyAvatarInfoDict[avId]['lastChatTime'] = 0 self.sortNearbyAvatars() self.__interestingAvatarEventOccured() avExitEvent = self.air.getAvatarExitEvent(avId) self.acceptOnce(avExitEvent, self.__handleExitedAvatar, [avId]) self.avatarEnterNextState() def avatarExit(self): avId = self.air.getAvatarIdFromSender() self.__doAvatarExit(avId) def __doAvatarExit(self, avId): avId = self.air.getAvatarIdFromSender() self.notify.debug('removing avatar ' + str(avId) + ' from the nearby avatar list') if avId not in self.nearbyAvatars: self.notify.debug('avatar ' + str(avId) + ' not in the nearby avatar list') else: avExitEvent = self.air.getAvatarExitEvent(avId) self.ignore(avExitEvent) del self.nearbyAvatarInfoDict[avId] self.nearbyAvatars.remove(avId) def avatarEnterNextState(): pass def avatarExitNextState(): pass def __clearNearbyAvatars(self): self.nearbyAvatars = [] self.nearbyAvatarInfoDict = {} def sortNearbyAvatars(self): def nAv_compare(a, b, nAvIDict=self.nearbyAvatarInfoDict): tsA = nAvIDict[a]['enterTime'] tsB = nAvIDict[b]['enterTime'] if tsA == tsB: return 0 else: if tsA < tsB: return -1 else: return 1 self.nearbyAvatars.sort(nAv_compare) def getNearbyAvatars(self): return self.nearbyAvatars def __avatarSpoke(self, avId): now = globalClock.getRealTime() if self.nearbyAvatarInfoDict.has_key(avId): self.nearbyAvatarInfoDict[avId]['lastChatTime'] = now self.__interestingAvatarEventOccured() def __initAttentionSpan(self): self.__avatarTimeoutBase = 0 def __interestingAvatarEventOccured(self, t=None): if t == None: t = globalClock.getRealTime() self.__avatarTimeoutBase = t return def lostInterest(self): now = globalClock.getRealTime() if now > self.__avatarTimeoutBase + 50.0: return 1 return 0 def __handleExitedAvatar(self, avId): self.__doAvatarExit(avId) def setNearbyAvatarChat(self, msg): avId = self.air.getAvatarIdFromSender() self.notify.debug('setNearbyAvatarChat: avatar ' + str(avId) + ' said ' + str(msg)) self.__avatarSpoke(avId) def setNearbyAvatarSC(self, msgIndex): avId = self.air.getAvatarIdFromSender() self.notify.debug('setNearbyAvatarSC: avatar %s said SpeedChat phrase %s' % (avId, msgIndex)) self.__avatarSpoke(avId) def setNearbyAvatarSCCustom(self, msgIndex): avId = self.air.getAvatarIdFromSender() self.notify.debug('setNearbyAvatarSCCustom: avatar %s said custom SpeedChat phrase %s' % (avId, msgIndex)) self.__avatarSpoke(avId) def setNearbyAvatarSCToontask(self, taskId, toNpcId, toonProgress, msgIndex): avId = self.air.getAvatarIdFromSender() self.notify.debug('setNearbyAvatarSCToontask: avatar %s said %s' % (avId, (taskId, toNpcId, toonProgress, msgIndex))) self.__avatarSpoke(avId) def getWalk(self): return ('', '', 0) def walkSpeed(self): return 0.1 def handleHolidays(self): self.CCChatter = 0 if hasattr(simbase.air, 'holidayManager'): if ToontownGlobals.CRASHED_LEADERBOARD in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.CRASHED_LEADERBOARD elif ToontownGlobals.CIRCUIT_RACING_EVENT in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.CIRCUIT_RACING_EVENT elif ToontownGlobals.WINTER_CAROLING in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.WINTER_CAROLING elif ToontownGlobals.WINTER_DECORATIONS in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.WINTER_DECORATIONS elif ToontownGlobals.VALENTINES_DAY in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.VALENTINES_DAY elif ToontownGlobals.APRIL_FOOLS_COSTUMES in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.APRIL_FOOLS_COSTUMES elif ToontownGlobals.SILLY_CHATTER_ONE in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.SILLY_CHATTER_ONE elif ToontownGlobals.SILLY_CHATTER_TWO in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.SILLY_CHATTER_TWO elif ToontownGlobals.SILLY_CHATTER_THREE in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.SILLY_CHATTER_THREE elif ToontownGlobals.SILLY_CHATTER_FOUR in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.SILLY_CHATTER_FOUR elif ToontownGlobals.SILLY_CHATTER_FIVE in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.SILLY_CHATTER_FOUR elif ToontownGlobals.HALLOWEEN_COSTUMES in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.HALLOWEEN_COSTUMES elif ToontownGlobals.SELLBOT_FIELD_OFFICE in simbase.air.holidayManager.currentHolidays: self.CCChatter = ToontownGlobals.SELLBOT_FIELD_OFFICE def getCCLocation(self): return 0 def getCCChatter(self): self.handleHolidays() return self.CCChatter def fadeAway(self): self.sendUpdate('fadeAway', []) def transitionCostume(self): self.transitionToCostume = 1