def make_map_img(imgfile, mapfile): twmap = Teemap("dummy") img2map(Image.open(imgfile).convert("RGB"), twmap.gamelayer, twmap.layers[2]) twmap.save(mapfile)
def make_map_txt(txtfile, mapfile): twmap = Teemap("dummy") txt2map(open(txtfile, "r"), twmap.gamelayer, twmap.layers[2]) twmap.save(mapfile)
class TestTeemap(unittest.TestCase): def setUp(self): os.mkdir("test_tmp") self.teemap = Teemap("test_maps/vanilla") def tearDown(self): if os.path.isdir("test_tmp"): shutil.rmtree("test_tmp") def test_load(self): pass # with warnings.catch_warnings(record=True) as w: # warnings.simplefilter('always') # Teemap('tml/test_maps/vanilla') # self.assertEqual(len(w), 1) # self.assertTrue(issubclass(w[0].category, UserWarning)) # self.assertIn(str(w[0].message), 'External image „test2“ does not exist') # assert Teemap('tml/maps/dm1.map') # assert Teemap('tml/maps/dm1') def test_groups(self): self.assertEqual(len(self.teemap.groups), 7) names = [None, None, "Game", "NamedGroup", None, None, "OtherGroup"] for i, group in enumerate(self.teemap.groups): self.assertEqual(group.name, names[i]) def test_layers(self): self.assertEqual(len(self.teemap.layers), 6) for i, num in enumerate([1, 2, 1, 0, 2, 0, 0]): self.assertEqual(len(self.teemap.groups[i].layers), num) self.assertIs(self.teemap.groups[0].layers[0], self.teemap.layers[0]) self.assertIs(self.teemap.groups[1].layers[0], self.teemap.layers[1]) self.assertIs(self.teemap.groups[1].layers[1], self.teemap.layers[2]) self.assertIs(self.teemap.groups[2].layers[0], self.teemap.layers[3]) self.assertIs(self.teemap.groups[4].layers[0], self.teemap.layers[4]) self.assertIs(self.teemap.groups[4].layers[1], self.teemap.layers[5]) names = ["TestQuads", "Quads", "TestTiles", "Game", None, "LastTiles"] types = [ "quadlayer", "quadlayer", "tilelayer", "tilelayer", "tilelayer", "tilelayer", ] for i, layer in enumerate(self.teemap.layers): self.assertEqual(layer.type, types[i]) self.assertEqual(layer.name, names[i]) for layer in self.teemap.layers: if layer.type == "tilelayer": if layer.name == "TestTiles": self.assertEqual(layer.width, 5) self.assertEqual(layer.height, 3) else: self.assertEqual(layer.width, 50) self.assertEqual(layer.height, 50) self.assertIs(self.teemap.layers[3], self.teemap.gamelayer) def test_envelopes(self): self.assertEqual(len(self.teemap.envelopes), 2) self.assertEqual(self.teemap.envelopes[0].name, "PosEnv") self.assertEqual(self.teemap.envelopes[1].name, "ColorEnv") def test_envpoints(self): self.assertEqual(len(self.teemap.envpoints), 9) self.assertEqual(len(self.teemap.envelopes[0].envpoints), 4) self.assertEqual(len(self.teemap.envelopes[1].envpoints), 5) for i, envpoint in enumerate(self.teemap.envelopes[0].envpoints): self.assertIs(envpoint, self.teemap.envpoints[i]) for i, envpoint in enumerate(self.teemap.envelopes[1].envpoints): self.assertIs(envpoint, self.teemap.envpoints[i + 4]) def test_images(self): images = [None, "grass_main", "grass_main", None, "grass_main", "test"] for i, layer in enumerate(self.teemap.layers): if images[i] is None: self.assertIs(layer.image_id, -1) else: self.assertEqual(self.teemap.images[layer.image_id].name, images[i]) self.assertIs(self.teemap.layers[2].image_id, 0) self.assertIs(self.teemap.layers[4].image_id, 0) self.assertIs(self.teemap.layers[5].image_id, 1) self.assertTrue(self.teemap.images[0].external) self.assertFalse(self.teemap.images[1].external) self.assertTrue(self.teemap.images[2].external) self.teemap.images[0].save("test_tmp/grass_main.png") self.teemap.images[1].save("test_tmp/test.png") self.assertRaises(ValueError, self.teemap.images[2].save, "tests/test_tmp/test2.png") self.assertTrue( filecmp.cmp("test_tmp/grass_main.png", "../tml/mapres/grass_main.png")) self.assertTrue( filecmp.cmp("test_tmp/test.png", "test_mapres/test.png")) def test_tiles(self): layer = self.teemap.layers[2] tiles = layer.tiles self.assertEqual(len(tiles), 15) # XXX: wouldn't be coordinates in layer make more sense? for i, tile in enumerate(tiles[:5]): self.assertEqual(tile.index, i) self.assertEqual(tile.coords, (i, 0)) for tile in tiles[5:10]: self.assertEqual(tile.index, 0) self.assertEqual(tile.coords, (0, 0)) for i, tile in enumerate(tiles[10:]): self.assertEqual(tile.index, i + 251) self.assertEqual(tile.coords, (11 + i, 15)) flags = [ { "rotation": False, "hflip": True, "vflip": False }, { "rotation": False, "hflip": False, "vflip": True }, { "rotation": False, "hflip": True, "vflip": True }, { "rotation": True, "hflip": True, "vflip": True }, { "rotation": True, "hflip": False, "vflip": False }, { "rotation": False, "hflip": False, "vflip": False }, ] for i, tile in enumerate(tiles[:6]): self.assertEqual(tile.flags, flags[i]) def test_quads(self): self.assertEqual(len(self.teemap.layers[0].quads), 1) self.assertEqual(len(self.teemap.layers[1].quads), 2) pos_envs = [-1, -1, 0] pos_env_offsets = [0, 0, 0] color_envs = [-1, -1, 1] color_env_offsets = [0, 0, 0] points = [ [ (-800000, -600000), (800000, -600000), (-800000, 600000), (800000, 600000), (32768, 32768), ], [ (329099, 208820), (608709, 162163), (304402, 515873), (773381, 589973), (361867, 241588), ], [ (-194633, 306165), (-13007, 301225), (-286028, 556954), (-13007, 492731), (-111637, 400134), ], ] texcoords = [ [(0, 0), (1024, 0), (0, 1024), (1024, 1024)], [(0, 0), (1024, 0), (0, 1024), (1024, 1024)], [(2215, -1076), (3204, -1076), (2215, 946), (3204, 946)], ] quads = [] for layer in self.teemap.layers: if layer.type == "quadlayer": quads.extend(layer.quads) for i, quad in enumerate(quads): self.assertEqual(quad.pos_env, pos_envs[i]) self.assertEqual(quad.pos_env_offset, pos_env_offsets[i]) self.assertEqual(quad.color_env, color_envs[i]) self.assertEqual(quad.color_env_offset, color_env_offsets[i]) self.assertEqual(quad.texcoords, texcoords[i]) self.assertEqual(quad.points, points[i]) def test_envpoints(self): times = [0, 722, 835, 2062, 0, 99, 643, 861, 1000] curvetypes = [4, 0, 4, 1, 1, 1, 1, 1, 1] values = [ [0, 0, 0, 0], [0, 0, 207678, 9600], [0, 0, 213958, 9600], [0, 0, -3105, 0], [1, 1, 1, 767], [1, 355, 514, 843], [1, 289, 995, 959], [824, 143, 86, 597], [1, 1, 1, 1], ] for i, envpoint in enumerate(self.teemap.envpoints): self.assertEqual(envpoint.time, times[i]) self.assertEqual(envpoint.curvetype, curvetypes[i]) self.assertEqual(envpoint.values, values[i]) def test_save(self): self.teemap.save("test_tmp/copy.map") self.teemap.save("test_tmp/copy2") self.assertTrue( filecmp.cmp("test_maps/vanilla.map", "test_tmp/copy.map")) self.assertTrue( filecmp.cmp("test_maps/vanilla.map", "test_tmp/copy2.map")) def test_validate(self): teemap = Teemap() self.assertRaises(MapError, teemap.validate) group = items.Group() group.layers.append(items.TileLayer(game=True)) teemap.groups.append(group) self.assertTrue(teemap.validate()) group.layers.append(items.TileLayer(game=True)) self.assertRaises(MapError, teemap.validate)
class TestTeemap(unittest.TestCase): def setUp(self): os.mkdir("test_tmp") self.teemap = Teemap("test_maps/vanilla") def tearDown(self): if os.path.isdir("test_tmp"): shutil.rmtree("test_tmp") def test_load(self): pass # with warnings.catch_warnings(record=True) as w: # warnings.simplefilter('always') # Teemap('tml/test_maps/vanilla') # self.assertEqual(len(w), 1) # self.assertTrue(issubclass(w[0].category, UserWarning)) # self.assertIn(str(w[0].message), 'External image „test2“ does not exist') # assert Teemap('tml/maps/dm1.map') # assert Teemap('tml/maps/dm1') def test_groups(self): self.assertEqual(len(self.teemap.groups), 7) names = [None, None, "Game", "NamedGroup", None, None, "OtherGroup"] for i, group in enumerate(self.teemap.groups): self.assertEqual(group.name, names[i]) def test_layers(self): self.assertEqual(len(self.teemap.layers), 6) for i, num in enumerate([1, 2, 1, 0, 2, 0, 0]): self.assertEqual(len(self.teemap.groups[i].layers), num) self.assertIs(self.teemap.groups[0].layers[0], self.teemap.layers[0]) self.assertIs(self.teemap.groups[1].layers[0], self.teemap.layers[1]) self.assertIs(self.teemap.groups[1].layers[1], self.teemap.layers[2]) self.assertIs(self.teemap.groups[2].layers[0], self.teemap.layers[3]) self.assertIs(self.teemap.groups[4].layers[0], self.teemap.layers[4]) self.assertIs(self.teemap.groups[4].layers[1], self.teemap.layers[5]) names = ["TestQuads", "Quads", "TestTiles", "Game", None, "LastTiles"] types = [ "quadlayer", "quadlayer", "tilelayer", "tilelayer", "tilelayer", "tilelayer", ] for i, layer in enumerate(self.teemap.layers): self.assertEqual(layer.type, types[i]) self.assertEqual(layer.name, names[i]) for layer in self.teemap.layers: if layer.type == "tilelayer": if layer.name == "TestTiles": self.assertEqual(layer.width, 5) self.assertEqual(layer.height, 3) else: self.assertEqual(layer.width, 50) self.assertEqual(layer.height, 50) self.assertIs(self.teemap.layers[3], self.teemap.gamelayer) def test_envelopes(self): self.assertEqual(len(self.teemap.envelopes), 2) self.assertEqual(self.teemap.envelopes[0].name, "PosEnv") self.assertEqual(self.teemap.envelopes[1].name, "ColorEnv") def test_envpoints(self): self.assertEqual(len(self.teemap.envpoints), 9) self.assertEqual(len(self.teemap.envelopes[0].envpoints), 4) self.assertEqual(len(self.teemap.envelopes[1].envpoints), 5) for i, envpoint in enumerate(self.teemap.envelopes[0].envpoints): self.assertIs(envpoint, self.teemap.envpoints[i]) for i, envpoint in enumerate(self.teemap.envelopes[1].envpoints): self.assertIs(envpoint, self.teemap.envpoints[i + 4]) def test_images(self): images = [None, "grass_main", "grass_main", None, "grass_main", "test"] for i, layer in enumerate(self.teemap.layers): if images[i] is None: self.assertIs(layer.image_id, -1) else: self.assertEqual(self.teemap.images[layer.image_id].name, images[i]) self.assertIs(self.teemap.layers[2].image_id, 0) self.assertIs(self.teemap.layers[4].image_id, 0) self.assertIs(self.teemap.layers[5].image_id, 1) self.assertTrue(self.teemap.images[0].external) self.assertFalse(self.teemap.images[1].external) self.assertTrue(self.teemap.images[2].external) self.teemap.images[0].save("test_tmp/grass_main.png") self.teemap.images[1].save("test_tmp/test.png") self.assertRaises(ValueError, self.teemap.images[2].save, "tests/test_tmp/test2.png") self.assertTrue( filecmp.cmp("test_tmp/grass_main.png", "../tml/mapres/grass_main.png") ) self.assertTrue(filecmp.cmp("test_tmp/test.png", "test_mapres/test.png")) def test_tiles(self): layer = self.teemap.layers[2] tiles = layer.tiles self.assertEqual(len(tiles), 15) # XXX: wouldn't be coordinates in layer make more sense? for i, tile in enumerate(tiles[:5]): self.assertEqual(tile.index, i) self.assertEqual(tile.coords, (i, 0)) for tile in tiles[5:10]: self.assertEqual(tile.index, 0) self.assertEqual(tile.coords, (0, 0)) for i, tile in enumerate(tiles[10:]): self.assertEqual(tile.index, i + 251) self.assertEqual(tile.coords, (11 + i, 15)) flags = [ {"rotation": False, "hflip": True, "vflip": False}, {"rotation": False, "hflip": False, "vflip": True}, {"rotation": False, "hflip": True, "vflip": True}, {"rotation": True, "hflip": True, "vflip": True}, {"rotation": True, "hflip": False, "vflip": False}, {"rotation": False, "hflip": False, "vflip": False}, ] for i, tile in enumerate(tiles[:6]): self.assertEqual(tile.flags, flags[i]) def test_quads(self): self.assertEqual(len(self.teemap.layers[0].quads), 1) self.assertEqual(len(self.teemap.layers[1].quads), 2) pos_envs = [-1, -1, 0] pos_env_offsets = [0, 0, 0] color_envs = [-1, -1, 1] color_env_offsets = [0, 0, 0] points = [ [ (-800000, -600000), (800000, -600000), (-800000, 600000), (800000, 600000), (32768, 32768), ], [ (329099, 208820), (608709, 162163), (304402, 515873), (773381, 589973), (361867, 241588), ], [ (-194633, 306165), (-13007, 301225), (-286028, 556954), (-13007, 492731), (-111637, 400134), ], ] texcoords = [ [(0, 0), (1024, 0), (0, 1024), (1024, 1024)], [(0, 0), (1024, 0), (0, 1024), (1024, 1024)], [(2215, -1076), (3204, -1076), (2215, 946), (3204, 946)], ] quads = [] for layer in self.teemap.layers: if layer.type == "quadlayer": quads.extend(layer.quads) for i, quad in enumerate(quads): self.assertEqual(quad.pos_env, pos_envs[i]) self.assertEqual(quad.pos_env_offset, pos_env_offsets[i]) self.assertEqual(quad.color_env, color_envs[i]) self.assertEqual(quad.color_env_offset, color_env_offsets[i]) self.assertEqual(quad.texcoords, texcoords[i]) self.assertEqual(quad.points, points[i]) def test_envpoints(self): times = [0, 722, 835, 2062, 0, 99, 643, 861, 1000] curvetypes = [4, 0, 4, 1, 1, 1, 1, 1, 1] values = [ [0, 0, 0, 0], [0, 0, 207678, 9600], [0, 0, 213958, 9600], [0, 0, -3105, 0], [1, 1, 1, 767], [1, 355, 514, 843], [1, 289, 995, 959], [824, 143, 86, 597], [1, 1, 1, 1], ] for i, envpoint in enumerate(self.teemap.envpoints): self.assertEqual(envpoint.time, times[i]) self.assertEqual(envpoint.curvetype, curvetypes[i]) self.assertEqual(envpoint.values, values[i]) def test_save(self): self.teemap.save("test_tmp/copy.map") self.teemap.save("test_tmp/copy2") self.assertTrue(filecmp.cmp("test_maps/vanilla.map", "test_tmp/copy.map")) self.assertTrue(filecmp.cmp("test_maps/vanilla.map", "test_tmp/copy2.map")) def test_validate(self): teemap = Teemap() self.assertRaises(MapError, teemap.validate) group = items.Group() group.layers.append(items.TileLayer(game=True)) teemap.groups.append(group) self.assertTrue(teemap.validate()) group.layers.append(items.TileLayer(game=True)) self.assertRaises(MapError, teemap.validate)
print(TML_DIR) map_path = os.sep.join([TML_DIR, '/maps/dm1']) t = Teemap(map_path) pickups = { 'shotgun': 0, 'grenade': 0, 'rifle': 0, 'ninja': 0, 'health': 0, 'armor': 0, 'solid': 0, 'air': 0, 'death': 0, 'nohook': 0, } for tile in t.gamelayer.tiles: for key, value in pickups.iteritems(): if tile.index == TILEINDEX[key]: pickups[key] += 1 for k, v in pickups.iteritems(): print '{value:3}x {key}'.format(value=v, key=k) t.gamelayer.tiles[0] = Tile(3) # set first tile x/y 0/0 to unhook t.gamelayer.set_tile(2, 0, Tile(2)) # zilli dreghun stinkt c: print("should be air(0): " + str(t.gamelayer.get_tile(5, 6))) print("should be shield(197): " + str(t.gamelayer.get_tile(3, 7))) print("should be spawn(192): " + str(t.gamelayer.get_tile(10, 10))) t.save('tml_map')