コード例 #1
0
 def __init__(self):
     super(CompressedGraphicsModule, self).__init__()
     self.town_maps = [EbTownMap() for x in TOWN_MAP_POINTER_OFFSETS]
     self.town_map_icons = EbTownMapIcons()
     self.company_logos = [EbCompanyLogo() for x in COMPANY_LOGO_INFOS]
     self.attract_mode_logos = [EbAttractModeLogo() for x in ATTRACT_MODE_INFOS]
     self.gas_station_logo = EbGasStationLogo()
コード例 #2
0
 def __init__(self):
     super(CompressedGraphicsModule, self).__init__()
     self.town_maps = [EbTownMap() for x in TOWN_MAP_POINTER_OFFSETS]
     self.town_map_icons = EbTownMapIcons()
     self.company_logos = [EbCompanyLogo() for x in COMPANY_LOGO_INFOS]
     self.attract_mode_logos = [EbAttractModeLogo() for x in ATTRACT_MODE_INFOS]
     self.gas_station_logo = EbGasStationLogo()
コード例 #3
0
class CompressedGraphicsModule(EbModule):
    NAME = "Compressed Graphics"
    FREE_RANGES = [(0x2021a8, 0x20ed02),  # Town Map data
                   (0x214ec1, 0x21ae7b),  # Company Logos, "Produced by" and "Presented by", and Gas Station
                   (0x21ea50, 0x21f203)]  # Town map icon graphics and palette

    def __init__(self):
        super(CompressedGraphicsModule, self).__init__()
        self.town_maps = [EbTownMap() for x in TOWN_MAP_POINTER_OFFSETS]
        self.town_map_icons = EbTownMapIcons()
        self.company_logos = [EbCompanyLogo() for x in COMPANY_LOGO_INFOS]
        self.attract_mode_logos = [EbAttractModeLogo() for x in ATTRACT_MODE_INFOS]
        self.gas_station_logo = EbGasStationLogo()

    def __exit__(self, type, value, traceback):
        del self.town_maps
        del self.town_map_icons
        del self.company_logos
        del self.attract_mode_logos
        del self.gas_station_logo

    def read_from_rom(self, rom):
        self.read_town_maps_from_rom(rom)
        self.read_town_map_icons_from_rom(rom)
        self.read_company_logos_from_rom(rom)
        self.read_attract_mode_logos_from_rom(rom)
        self.read_gas_station_from_rom(rom)

    def write_to_rom(self, rom):
        self.write_town_maps_to_rom(rom)
        self.write_town_map_icons_to_rom(rom)
        self.write_company_logos_to_rom(rom)
        self.write_attract_mode_logos_to_rom(rom)
        self.write_gas_station_to_rom(rom)

    def read_town_maps_from_rom(self, rom):
        log.debug("Reading town maps")
        for pointer_offset, town_map in zip(TOWN_MAP_POINTER_OFFSETS, self.town_maps):
            offset = from_snes_address(rom.read_multi(pointer_offset, size=4))
            town_map.from_block(block=rom,
                                offset=offset)

    def write_town_maps_to_rom(self, rom):
        log.debug("Writing town maps")
        for pointer_offset, town_map in zip(TOWN_MAP_POINTER_OFFSETS, self.town_maps):
            offset = town_map.to_block(rom)
            rom.write_multi(pointer_offset, to_snes_address(offset), size=4)

    def read_town_map_icons_from_rom(self, rom):
        log.debug("Reading town map icons")
        graphics_offset = from_snes_address(read_asm_pointer(block=rom,
                                                             offset=TOWN_MAP_ICON_GRAPHICS_ASM_POINTER_OFFSET))
        palette_offset = from_snes_address(read_asm_pointer(block=rom,
                                                            offset=TOWN_MAP_ICON_PALETTE_ASM_POINTER_OFFSET))
        self.town_map_icons.from_block(block=rom,
                                       graphics_offset=graphics_offset,
                                       arrangement_offset=0,
                                       palette_offsets=[palette_offset])

    def write_town_map_icons_to_rom(self, rom):
        log.debug("Writing town map icons")
        graphics_offset, arrangement_offset, palette_offsets = self.town_map_icons.to_block(rom)
        write_asm_pointer(block=rom,
                          offset=TOWN_MAP_ICON_GRAPHICS_ASM_POINTER_OFFSET,
                          pointer=to_snes_address(graphics_offset))
        write_asm_pointer(block=rom,
                          offset=TOWN_MAP_ICON_PALETTE_ASM_POINTER_OFFSET,
                          pointer=to_snes_address(palette_offsets[0]))

    def read_company_logos_from_rom(self, rom):
        log.debug("Reading company logos")
        self.read_logos_from_rom(rom, self.company_logos, COMPANY_LOGO_INFOS)

    def write_company_logos_to_rom(self, rom):
        log.debug("Writing company logos")
        self.write_logos_to_rom(rom, self.company_logos, COMPANY_LOGO_INFOS)

    def read_attract_mode_logos_from_rom(self, rom):
        log.debug("Reading attract mode logos")
        self.read_logos_from_rom(rom, self.attract_mode_logos, ATTRACT_MODE_INFOS)

    def write_attract_mode_logos_to_rom(self, rom):
        log.debug("Writing attract mode logos")
        self.write_logos_to_rom(rom, self.attract_mode_logos, ATTRACT_MODE_INFOS)

    def read_gas_station_from_rom(self, rom):
        log.debug("Reading gas station logo")
        self.read_logos_from_rom(rom, [self.gas_station_logo], [GAS_STATION_INFO])

    def write_gas_station_to_rom(self, rom):
        log.debug("Writing gas station logo")
        self.write_logos_to_rom(rom, [self.gas_station_logo], [GAS_STATION_INFO])

    def read_logos_from_rom(self, rom, logos, infos):
        for info, logo in zip(infos, logos):
            graphics_offset = from_snes_address(read_asm_pointer(rom, info.graphics_asm_pointer_offsets[0]))
            arrangement_offset = from_snes_address(read_asm_pointer(rom, info.arrangement_asm_pointer_offsets[0]))
            palette_offsets = [from_snes_address(read_asm_pointer(rom, x)) for x in info.palette_asm_pointer_offsets]

            logo.from_block(block=rom,
                            graphics_offset=graphics_offset,
                            arrangement_offset=arrangement_offset,
                            palette_offsets=palette_offsets)

    def write_logos_to_rom(self, rom, logos, infos):
        for info, logo in zip(infos, logos):
            graphics_offset, arrangement_offset, palette_offsets = logo.to_block(rom)

            for asm_pointer_offset in info.graphics_asm_pointer_offsets:
                write_asm_pointer(block=rom, offset=asm_pointer_offset, pointer=to_snes_address(graphics_offset))
            for asm_pointer_offset in info.arrangement_asm_pointer_offsets:
                write_asm_pointer(block=rom, offset=asm_pointer_offset, pointer=to_snes_address(arrangement_offset))
            for offset, asm_pointer_offset in zip(palette_offsets, info.palette_asm_pointer_offsets):
                write_asm_pointer(block=rom, offset=asm_pointer_offset, pointer=to_snes_address(offset))

    def read_from_project(self, resource_open):
        self.read_town_maps_from_project(resource_open)
        self.read_town_map_icons_from_project(resource_open)
        self.read_company_logos_from_project(resource_open)
        self.read_attract_mode_logos_from_project(resource_open)
        self.read_gas_station_from_project(resource_open)

    def write_to_project(self, resource_open):
        self.write_town_maps_to_project(resource_open)
        self.write_town_map_icons_to_project(resource_open)
        self.write_company_logos_to_project(resource_open)
        self.write_attract_mode_logos_to_project(resource_open)
        self.write_gas_station_to_project(resource_open)

    def read_town_maps_from_project(self, resource_open):
        for resource_name, town_map in zip(TOWN_MAP_RESOURCE_NAMES, self.town_maps):
            log.info("- Reading {}".format(resource_name))
            with resource_open(resource_name, "png") as image_file:
                image = open_indexed_image(image_file)
                town_map.from_image(image)

    def write_town_maps_to_project(self, resource_open):
        log.debug("Writing town maps")
        for resource_name, town_map in zip(TOWN_MAP_RESOURCE_NAMES, self.town_maps):
            image = town_map.image()
            with resource_open(resource_name, "png") as image_file:
                image.save(image_file, "png")

    def read_town_map_icons_from_project(self, resource_open):
        log.info("- Reading town map icons")
        with resource_open("TownMaps/icons", "png") as image_file:
            image = open_indexed_image(image_file)
            self.town_map_icons.from_image(image=image, arrangement=TOWN_MAP_ICON_PREVIEW_ARRANGEMENT)

    def write_town_map_icons_to_project(self, resource_open):
        log.debug("Writing town map icons")
        image = self.town_map_icons.image(TOWN_MAP_ICON_PREVIEW_ARRANGEMENT)
        with resource_open("TownMaps/icons", "png") as image_file:
            image.save(image_file, "png")

    def read_company_logos_from_project(self, resource_open):
        self.read_logos_from_project(resource_open, self.company_logos, COMPANY_LOGO_INFOS)

    def write_company_logos_to_project(self, resource_open):
        log.debug("Writing company logos")
        self.write_logos_to_project(resource_open, self.company_logos, COMPANY_LOGO_INFOS)

    def read_attract_mode_logos_from_project(self, resource_open):
        self.read_logos_from_project(resource_open, self.attract_mode_logos, ATTRACT_MODE_INFOS)

    def write_attract_mode_logos_to_project(self, resource_open):
        log.debug("Writing attract mode logos")
        self.write_logos_to_project(resource_open, self.attract_mode_logos, ATTRACT_MODE_INFOS)

    def read_logos_from_project(self, resource_open, logos, infos):
        for info, logo in zip(infos, logos):
            log.info("- Reading " + info.name)
            with resource_open(info.name, "png") as image_file:
                image = open_indexed_image(image_file)
                logo.from_image(image)

    def write_logos_to_project(self, resource_open, logos, infos):
        for info, logo in zip(infos, logos):
            image = logo.image()
            with resource_open(info.name, "png") as image_file:
                image.save(image_file, "png")

    def read_gas_station_from_project(self, resource_open):
        log.info("- Reading gas station logo")
        with resource_open(GAS_STATION_INFO.name + "1", "png") as image1_file:
            image1 = open_image(image1_file)
            with resource_open(GAS_STATION_INFO.name + "2", "png") as image2_file:
                image2 = open_image(image2_file)
                with resource_open(GAS_STATION_INFO.name + "3", "png") as image3_file:
                    image3 = open_image(image3_file)
                    self.gas_station_logo.from_images([image1, image2, image3])

    def write_gas_station_to_project(self, resource_open):
        log.debug("Writing gas station logo")
        images = self.gas_station_logo.images()
        with resource_open(GAS_STATION_INFO.name + "1", "png") as image_file:
            images[0].save(image_file, "png")
        with resource_open(GAS_STATION_INFO.name + "2", "png") as image_file:
            images[1].save(image_file, "png")
        with resource_open(GAS_STATION_INFO.name + "3", "png") as image_file:
            images[2].save(image_file, "png")

    def upgrade_project(self, old_version, new_version, rom, resource_open_r, resource_open_w, resource_delete):
        if old_version == new_version:
            return
        elif old_version <= 2:
            self.read_town_map_icons_from_rom(rom)
            self.write_town_map_icons_to_project(resource_open_w)

            self.read_attract_mode_logos_from_rom(rom)
            self.write_attract_mode_logos_to_project(resource_open_w)

            self.read_gas_station_from_rom(rom)
            self.write_gas_station_to_project(resource_open_w)

            self.upgrade_project(3, new_version, rom, resource_open_r, resource_open_w, resource_delete)
        else:
            self.upgrade_project(old_version + 1, new_version, rom, resource_open_r, resource_open_w, resource_delete)
コード例 #4
0
class CompressedGraphicsModule(EbModule):
    NAME = "Compressed Graphics"
    FREE_RANGES = [
        (0x2021a8, 0x20ed02),  # Town Map data
        (0x214ec1, 0x21ae7b
         ),  # Company Logos, "Produced by" and "Presented by", and Gas Station
        (0x21ea50, 0x21f203)
    ]  # Town map icon graphics and palette

    def __init__(self):
        super(CompressedGraphicsModule, self).__init__()
        self.town_maps = [EbTownMap() for x in TOWN_MAP_POINTER_OFFSETS]
        self.town_map_icons = EbTownMapIcons()
        self.company_logos = [EbCompanyLogo() for x in COMPANY_LOGO_INFOS]
        self.attract_mode_logos = [
            EbAttractModeLogo() for x in ATTRACT_MODE_INFOS
        ]
        self.gas_station_logo = EbGasStationLogo()

    def __exit__(self, type, value, traceback):
        del self.town_maps
        del self.town_map_icons
        del self.company_logos
        del self.attract_mode_logos
        del self.gas_station_logo

    def read_from_rom(self, rom):
        self.read_town_maps_from_rom(rom)
        self.read_town_map_icons_from_rom(rom)
        self.read_company_logos_from_rom(rom)
        self.read_attract_mode_logos_from_rom(rom)
        self.read_gas_station_from_rom(rom)

    def write_to_rom(self, rom):
        self.write_town_maps_to_rom(rom)
        self.write_town_map_icons_to_rom(rom)
        self.write_company_logos_to_rom(rom)
        self.write_attract_mode_logos_to_rom(rom)
        self.write_gas_station_to_rom(rom)

    def read_town_maps_from_rom(self, rom):
        log.debug("Reading town maps")
        for pointer_offset, town_map in zip(TOWN_MAP_POINTER_OFFSETS,
                                            self.town_maps):
            offset = from_snes_address(rom.read_multi(pointer_offset, size=4))
            town_map.from_block(block=rom, offset=offset)

    def write_town_maps_to_rom(self, rom):
        log.debug("Writing town maps")
        for pointer_offset, town_map in zip(TOWN_MAP_POINTER_OFFSETS,
                                            self.town_maps):
            offset = town_map.to_block(rom)
            rom.write_multi(pointer_offset, to_snes_address(offset), size=4)

    def read_town_map_icons_from_rom(self, rom):
        log.debug("Reading town map icons")
        graphics_offset = from_snes_address(
            read_asm_pointer(block=rom,
                             offset=TOWN_MAP_ICON_GRAPHICS_ASM_POINTER_OFFSET))
        palette_offset = from_snes_address(
            read_asm_pointer(block=rom,
                             offset=TOWN_MAP_ICON_PALETTE_ASM_POINTER_OFFSET))
        self.town_map_icons.from_block(block=rom,
                                       graphics_offset=graphics_offset,
                                       arrangement_offset=0,
                                       palette_offsets=[palette_offset])

    def write_town_map_icons_to_rom(self, rom):
        log.debug("Writing town map icons")
        graphics_offset, arrangement_offset, palette_offsets = self.town_map_icons.to_block(
            rom)
        write_asm_pointer(block=rom,
                          offset=TOWN_MAP_ICON_GRAPHICS_ASM_POINTER_OFFSET,
                          pointer=to_snes_address(graphics_offset))
        write_asm_pointer(block=rom,
                          offset=TOWN_MAP_ICON_PALETTE_ASM_POINTER_OFFSET,
                          pointer=to_snes_address(palette_offsets[0]))

    def read_company_logos_from_rom(self, rom):
        log.debug("Reading company logos")
        self.read_logos_from_rom(rom, self.company_logos, COMPANY_LOGO_INFOS)

    def write_company_logos_to_rom(self, rom):
        log.debug("Writing company logos")
        self.write_logos_to_rom(rom, self.company_logos, COMPANY_LOGO_INFOS)

    def read_attract_mode_logos_from_rom(self, rom):
        log.debug("Reading attract mode logos")
        self.read_logos_from_rom(rom, self.attract_mode_logos,
                                 ATTRACT_MODE_INFOS)

    def write_attract_mode_logos_to_rom(self, rom):
        log.debug("Writing attract mode logos")
        self.write_logos_to_rom(rom, self.attract_mode_logos,
                                ATTRACT_MODE_INFOS)

    def read_gas_station_from_rom(self, rom):
        log.debug("Reading gas station logo")
        self.read_logos_from_rom(rom, [self.gas_station_logo],
                                 [GAS_STATION_INFO])

    def write_gas_station_to_rom(self, rom):
        log.debug("Writing gas station logo")
        self.write_logos_to_rom(rom, [self.gas_station_logo],
                                [GAS_STATION_INFO])

    def read_logos_from_rom(self, rom, logos, infos):
        for info, logo in zip(infos, logos):
            graphics_offset = from_snes_address(
                read_asm_pointer(rom, info.graphics_asm_pointer_offsets[0]))
            arrangement_offset = from_snes_address(
                read_asm_pointer(rom, info.arrangement_asm_pointer_offsets[0]))
            palette_offsets = [
                from_snes_address(read_asm_pointer(rom, x))
                for x in info.palette_asm_pointer_offsets
            ]

            logo.from_block(block=rom,
                            graphics_offset=graphics_offset,
                            arrangement_offset=arrangement_offset,
                            palette_offsets=palette_offsets)

    def write_logos_to_rom(self, rom, logos, infos):
        for info, logo in zip(infos, logos):
            graphics_offset, arrangement_offset, palette_offsets = logo.to_block(
                rom)

            for asm_pointer_offset in info.graphics_asm_pointer_offsets:
                write_asm_pointer(block=rom,
                                  offset=asm_pointer_offset,
                                  pointer=to_snes_address(graphics_offset))
            for asm_pointer_offset in info.arrangement_asm_pointer_offsets:
                write_asm_pointer(block=rom,
                                  offset=asm_pointer_offset,
                                  pointer=to_snes_address(arrangement_offset))
            for offset, asm_pointer_offset in zip(
                    palette_offsets, info.palette_asm_pointer_offsets):
                write_asm_pointer(block=rom,
                                  offset=asm_pointer_offset,
                                  pointer=to_snes_address(offset))

    def read_from_project(self, resource_open):
        self.read_town_maps_from_project(resource_open)
        self.read_town_map_icons_from_project(resource_open)
        self.read_company_logos_from_project(resource_open)
        self.read_attract_mode_logos_from_project(resource_open)
        self.read_gas_station_from_project(resource_open)

    def write_to_project(self, resource_open):
        self.write_town_maps_to_project(resource_open)
        self.write_town_map_icons_to_project(resource_open)
        self.write_company_logos_to_project(resource_open)
        self.write_attract_mode_logos_to_project(resource_open)
        self.write_gas_station_to_project(resource_open)

    def read_town_maps_from_project(self, resource_open):
        for resource_name, town_map in zip(TOWN_MAP_RESOURCE_NAMES,
                                           self.town_maps):
            log.info("- Reading {}".format(resource_name))
            with resource_open(resource_name, "png") as image_file:
                image = open_indexed_image(image_file)
                town_map.from_image(image)

    def write_town_maps_to_project(self, resource_open):
        log.debug("Writing town maps")
        for resource_name, town_map in zip(TOWN_MAP_RESOURCE_NAMES,
                                           self.town_maps):
            image = town_map.image()
            with resource_open(resource_name, "png") as image_file:
                image.save(image_file, "png")

    def read_town_map_icons_from_project(self, resource_open):
        log.info("- Reading town map icons")
        with resource_open("TownMaps/icons", "png") as image_file:
            image = open_indexed_image(image_file)
            self.town_map_icons.from_image(
                image=image, arrangement=TOWN_MAP_ICON_PREVIEW_ARRANGEMENT)

    def write_town_map_icons_to_project(self, resource_open):
        log.debug("Writing town map icons")
        image = self.town_map_icons.image(TOWN_MAP_ICON_PREVIEW_ARRANGEMENT)
        with resource_open("TownMaps/icons", "png") as image_file:
            image.save(image_file, "png")

    def read_company_logos_from_project(self, resource_open):
        self.read_logos_from_project(resource_open, self.company_logos,
                                     COMPANY_LOGO_INFOS)

    def write_company_logos_to_project(self, resource_open):
        log.debug("Writing company logos")
        self.write_logos_to_project(resource_open, self.company_logos,
                                    COMPANY_LOGO_INFOS)

    def read_attract_mode_logos_from_project(self, resource_open):
        self.read_logos_from_project(resource_open, self.attract_mode_logos,
                                     ATTRACT_MODE_INFOS)

    def write_attract_mode_logos_to_project(self, resource_open):
        log.debug("Writing attract mode logos")
        self.write_logos_to_project(resource_open, self.attract_mode_logos,
                                    ATTRACT_MODE_INFOS)

    def read_logos_from_project(self, resource_open, logos, infos):
        for info, logo in zip(infos, logos):
            log.info("- Reading " + info.name)
            with resource_open(info.name, "png") as image_file:
                image = open_indexed_image(image_file)
                logo.from_image(image)

    def write_logos_to_project(self, resource_open, logos, infos):
        for info, logo in zip(infos, logos):
            image = logo.image()
            with resource_open(info.name, "png") as image_file:
                image.save(image_file, "png")

    def read_gas_station_from_project(self, resource_open):
        log.info("- Reading gas station logo")
        with resource_open(GAS_STATION_INFO.name + "1", "png") as image1_file:
            image1 = open_image(image1_file)
            with resource_open(GAS_STATION_INFO.name + "2",
                               "png") as image2_file:
                image2 = open_image(image2_file)
                with resource_open(GAS_STATION_INFO.name + "3",
                                   "png") as image3_file:
                    image3 = open_image(image3_file)
                    self.gas_station_logo.from_images([image1, image2, image3])

    def write_gas_station_to_project(self, resource_open):
        log.debug("Writing gas station logo")
        images = self.gas_station_logo.images()
        with resource_open(GAS_STATION_INFO.name + "1", "png") as image_file:
            images[0].save(image_file, "png")
        with resource_open(GAS_STATION_INFO.name + "2", "png") as image_file:
            images[1].save(image_file, "png")
        with resource_open(GAS_STATION_INFO.name + "3", "png") as image_file:
            images[2].save(image_file, "png")

    def upgrade_project(self, old_version, new_version, rom, resource_open_r,
                        resource_open_w, resource_delete):
        if old_version == new_version:
            return
        elif old_version <= 2:
            self.read_town_map_icons_from_rom(rom)
            self.write_town_map_icons_to_project(resource_open_w)

            self.read_attract_mode_logos_from_rom(rom)
            self.write_attract_mode_logos_to_project(resource_open_w)

            self.read_gas_station_from_rom(rom)
            self.write_gas_station_to_project(resource_open_w)

            self.upgrade_project(3, new_version, rom, resource_open_r,
                                 resource_open_w, resource_delete)
        else:
            self.upgrade_project(old_version + 1, new_version, rom,
                                 resource_open_r, resource_open_w,
                                 resource_delete)