Poodletooth-iLand/toontown/coghq/FactoryManagerAI.py
2015-03-03 17:10:12 -05:00

27 lines
1.1 KiB
Python

from direct.directnotify import DirectNotifyGlobal
import DistributedBrutalFactoryAI
import DistributedFactoryAI
from toontown.toonbase import ToontownGlobals
from direct.showbase import DirectObject
class FactoryManagerAI(DirectObject.DirectObject):
notify = DirectNotifyGlobal.directNotify.newCategory('FactoryManagerAI')
factoryId = None
def __init__(self, air):
DirectObject.DirectObject.__init__(self)
self.air = air
def getDoId(self):
return 0
def createFactory(self, factoryId, entranceId, players):
factoryZone = self.air.allocateZone()
if FactoryManagerAI.factoryId is not None:
factoryId = FactoryManagerAI.factoryId
if entranceId == 2:
factory = DistributedBrutalFactoryAI.DistributedBrutalFactoryAI(self.air, factoryId, factoryZone, entranceId, players)
else:
factory = DistributedFactoryAI.DistributedFactoryAI(self.air, factoryId, factoryZone, entranceId, players)
factory.generateWithRequired(factoryZone)
return factoryZone