diff options
| -rw-r--r-- | src/tileset.py | 161 |
1 files changed, 159 insertions, 2 deletions
diff --git a/src/tileset.py b/src/tileset.py index 7fa8733..a193a7f 100644 --- a/src/tileset.py +++ b/src/tileset.py @@ -82,8 +82,165 @@ class KPObjectDefinition(object): class KPTileset(object): - def __init__(self): - self.texture = None + def __init__(self, imageBuffer, objectBuffer, groupBuffer): + '''A Koopatlas Tileset class. To initialize, pass it an image buffer, an + object buffer, and a group buffer from a Koopatlas tileset file. + + tiles -> has a list of all 512 tiles. + objects -> has a list of objects in the tileset + groupModel -> has a model containing groups with items for all objects + and groups in the tileset. + + + Methods of Note: + + overrideTile(Tile Index, QPixmap) # Takes a 24x24 QPixmap and a tile index + + + + + ''' + + + self.tiles = [] self.objects = [] + self.groupModel = QtGui.QTreeModel + + self.processImage(imageBuffer) + self.processObjects(objectBuffer) + self.processGroup(groupBuffer) + + + + + def processImage(self, imageBuffer): + '''Takes an imageBuffer from a Koopatlas Tileset and converts it into 24x24 + tiles, which get put into KPTileset.tiles.''' + + dest = self.RGB4A3Decode(Image) + + self.tileImage = QtGui.QPixmap.fromImage(dest) + + # Makes us some nice Tiles! + Xoffset = 4 + Yoffset = 4 + for i in range(512): + self.tiles.append(self.tileImage.copy(Xoffset,Yoffset,24,24)) + Xoffset += 32 + if Xoffset >= 1024: + Xoffset = 4 + Yoffset += 32 + + + + def RGB4A3Decode(tex): + dest = QtGui.QImage(1024,512,QtGui.QImage.Format_ARGB32) + dest.fill(QtCore.Qt.transparent) + + i = 0 + for ytile in xrange(0, 512, 4): + for xtile in xrange(0, 1024, 4): + for ypixel in xrange(ytile, ytile + 4): + for xpixel in xrange(xtile, xtile + 4): + + if(xpixel >= 1024 or ypixel >= 512): + continue + + newpixel = struct.unpack_from('>H', tex, i)[0] + # newpixel = (int(tex[i]) << 8) | int(tex[i+1]) + + + if(newpixel >= 0x8000): # Check if it's RGB555 + red = ((newpixel >> 10) & 0x1F) * 255 / 0x1F + green = ((newpixel >> 5) & 0x1F) * 255 / 0x1F + blue = (newpixel & 0x1F) * 255 / 0x1F + alpha = 0xFF + + else: # If not, it's RGB4A3 + alpha = ((newpixel & 0x7000) >> 12) * 255 / 0x7 + blue = ((newpixel & 0xF00) >> 8) * 255 / 0xF + green = ((newpixel & 0xF0) >> 4) * 255 / 0xF + red = (newpixel & 0xF) * 255 / 0xF + + argb = (blue) | (green << 8) | (red << 16) | (alpha << 24) + dest.setPixel(xpixel, ypixel, argb) + i += 2 + return dest + + + def processObjects(self, objectBuffer): + + + + + def processGroup(self, groupBuffer): + + grovyle = cPickle.loads(group) + makeTree(grovyle, treecko) + + + def makeTree(self, grovyle, treecko): + + for razorLeaf in grovyle: + + if (type(razorLeaf) is str) and (razorLeaf[:6] == "Object"): + + pix = QtGui.QPixmap(24, 24) + pix.fill(QtCore.Qt.transparent) + painter = QtGui.QPainter(pix) + painter.drawPixmap(0, 0, pix) + painter.end() + + a = QtGui.QTreeWidgetItem(treecko) + a.setText(0, razorLeaf) + a.setFlags(QtCore.Qt.ItemFlags(0x25)) + a.setIcon(1, QtGui.QIcon(pix)) + + else: + + a = QtGui.QTreeWidgetItem(treecko) + a.setText(0, razorLeaf[0]) + a.setFlags(QtCore.Qt.ItemFlags(0x2F)) + a.setChildIndicatorPolicy(QtGui.QTreeWidgetItem.ShowIndicator) + a.setExpanded(True) + + + makeTree(razorLeaf[1], a) + + + def overrideTile(index, pixmap): + '''Takes a 24x24 QPixmap and a tile index, and returns true if it succeeds.''' + + if index > 511: + return false + + if not isInstance(pixmap, QtGui.QPixmap): + return false + + if (QtGui.QPixmap.height() != 24) or (QtGui.QPixmap.width() != 24): + return false + + + self.tiles[index] = pixmap + + return true + + + + + + + + + + + + + + + + + + |