mirror of
https://github.com/veekun/pokedex.git
synced 2024-08-20 18:16:34 +00:00
22 lines
708 B
Python
22 lines
708 B
Python
"""Allegedly stands for 'Pokémon Container'. Completely generic, dead-simple
|
|
container format.
|
|
"""
|
|
from .base import _ContainerFile, Substream
|
|
|
|
|
|
class PokemonContainerFile(_ContainerFile):
|
|
magic = b'PC'
|
|
|
|
def __init__(self, stream):
|
|
self.stream = stream = Substream(stream)
|
|
|
|
magic, entry_ct = stream.unpack('<2sH')
|
|
assert magic == b'PC'
|
|
|
|
# Offsets are "A B C ...", where entry 0 ranges from A to B, entry 1
|
|
# from B to C, etc.
|
|
offsets = stream.unpack('<{}L'.format(entry_ct + 1))
|
|
self.slices = []
|
|
for i in range(entry_ct):
|
|
start, end = offsets[i:i + 2]
|
|
self.slices.append(self.stream.slice(start, end - start))
|