103 lines
3.5 KiB
Python
103 lines
3.5 KiB
Python
from panda3d.core import *
|
|
from panda3d.toontown import *
|
|
from direct.directnotify.DirectNotifyGlobal import *
|
|
from direct.showbase import Loader
|
|
from toontown.toontowngui import ToontownLoadingScreen
|
|
|
|
class ToontownLoader(Loader.Loader):
|
|
TickPeriod = 0.2
|
|
|
|
def __init__(self, base):
|
|
Loader.Loader.__init__(self, base)
|
|
self.inBulkBlock = None
|
|
self.blockName = None
|
|
self.loadingScreen = ToontownLoadingScreen.ToontownLoadingScreen()
|
|
return
|
|
|
|
def destroy(self):
|
|
self.loadingScreen.destroy()
|
|
del self.loadingScreen
|
|
Loader.Loader.destroy(self)
|
|
|
|
def beginBulkLoad(self, name, label, range, gui, tipCategory):
|
|
self._loadStartT = globalClock.getRealTime()
|
|
Loader.Loader.notify.info("starting bulk load of block '%s'" % name)
|
|
if self.inBulkBlock:
|
|
Loader.Loader.notify.warning("Tried to start a block ('%s'), but am already in a block ('%s')" % (name, self.blockName))
|
|
return None
|
|
self.inBulkBlock = 1
|
|
self._lastTickT = globalClock.getRealTime()
|
|
self.blockName = name
|
|
self.loadingScreen.begin(range, label, gui, tipCategory)
|
|
return None
|
|
|
|
def endBulkLoad(self, name):
|
|
if not self.inBulkBlock:
|
|
Loader.Loader.notify.warning("Tried to end a block ('%s'), but not in one" % name)
|
|
return None
|
|
if name != self.blockName:
|
|
Loader.Loader.notify.warning("Tried to end a block ('%s'), other then the current one ('%s')" % (name, self.blockName))
|
|
return None
|
|
self.inBulkBlock = None
|
|
expectedCount, loadedCount = self.loadingScreen.end()
|
|
now = globalClock.getRealTime()
|
|
Loader.Loader.notify.info("At end of block '%s', expected %s, loaded %s, duration=%s" % (self.blockName,
|
|
expectedCount,
|
|
loadedCount,
|
|
now - self._loadStartT))
|
|
return
|
|
|
|
def abortBulkLoad(self):
|
|
if self.inBulkBlock:
|
|
Loader.Loader.notify.info("Aborting block ('%s')" % self.blockName)
|
|
self.inBulkBlock = None
|
|
self.loadingScreen.abort()
|
|
return
|
|
|
|
def tick(self):
|
|
if self.inBulkBlock:
|
|
now = globalClock.getRealTime()
|
|
if now - self._lastTickT > self.TickPeriod:
|
|
self._lastTickT += self.TickPeriod
|
|
self.loadingScreen.tick()
|
|
try:
|
|
base.cr.considerHeartbeat()
|
|
except:
|
|
pass
|
|
|
|
def loadModel(self, *args, **kw):
|
|
ret = Loader.Loader.loadModel(self, *args, **kw)
|
|
self.tick()
|
|
return ret
|
|
|
|
def loadFont(self, *args, **kw):
|
|
ret = Loader.Loader.loadFont(self, *args, **kw)
|
|
self.tick()
|
|
return ret
|
|
|
|
def loadTexture(self, texturePath, alphaPath = None, okMissing = False):
|
|
ret = Loader.Loader.loadTexture(self, texturePath, alphaPath, okMissing=okMissing)
|
|
self.tick()
|
|
if alphaPath:
|
|
self.tick()
|
|
return ret
|
|
|
|
def loadSfx(self, soundPath):
|
|
ret = Loader.Loader.loadSfx(self, soundPath)
|
|
self.tick()
|
|
return ret
|
|
|
|
def loadMusic(self, soundPath):
|
|
ret = Loader.Loader.loadMusic(self, soundPath)
|
|
self.tick()
|
|
return ret
|
|
|
|
def loadDNAFileAI(self, dnaStore, dnaFile):
|
|
ret = loadDNAFileAI(dnaStore, dnaFile, CSDefault)
|
|
self.tick()
|
|
return ret
|
|
|
|
def loadDNAFile(self, dnaStore, dnaFile):
|
|
ret = loadDNAFile(dnaStore, dnaFile, CSDefault, 0)
|
|
self.tick()
|
|
return ret
|