294 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
		
		
			
		
	
	
			294 lines
		
	
	
		
			8.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
|   | from .rom_addresses import rom_addresses | ||
|  | 
 | ||
|  | disallowed1F = [[2, 2], [3, 2], [1, 8], [2, 8], [7, 7], [8, 7], [10, 4], [11, 4], [11, 12], | ||
|  |               [11, 13], [16, 10], [17, 10], [18, 10], [16, 12], [17, 12], [18, 12]] | ||
|  | disallowed2F = [[16, 2], [17, 2], [18, 2], [15, 5], [15, 6], [10, 10], [11, 10], [12, 10], [7, 14], [8, 14], [1, 15], | ||
|  |                 [13, 15], [13, 16], [1, 12], [1, 10], [3, 5], [3, 6], [5, 6], [5, 7], [5, 8], [1, 2], [1, 3], [1, 4], | ||
|  |                 [11, 1]] | ||
|  | 
 | ||
|  | 
 | ||
|  | def randomize_rock_tunnel(data, random): | ||
|  | 
 | ||
|  |     seed = random.randint(0, 999999999999999999) | ||
|  |     random.seed(seed) | ||
|  | 
 | ||
|  |     map1f = [] | ||
|  |     map2f = [] | ||
|  | 
 | ||
|  |     address = rom_addresses["Map_Rock_Tunnel1F"] | ||
|  |     for y in range(0, 18): | ||
|  |         row = [] | ||
|  |         for x in range(0, 20): | ||
|  |             row.append(data[address]) | ||
|  |             address += 1 | ||
|  |         map1f.append(row) | ||
|  | 
 | ||
|  |     address = rom_addresses["Map_Rock_TunnelB1F"] | ||
|  |     for y in range(0, 18): | ||
|  |         row = [] | ||
|  |         for x in range(0, 20): | ||
|  |             row.append(data[address]) | ||
|  |             address += 1 | ||
|  |         map2f.append(row) | ||
|  | 
 | ||
|  |     current_map = map1f | ||
|  | 
 | ||
|  |     def floor(x, y): | ||
|  |         current_map[y][x] = 1 | ||
|  | 
 | ||
|  |     def wide(x, y): | ||
|  |         current_map[y][x] = 32 | ||
|  |         current_map[y][x + 1] = 34 | ||
|  | 
 | ||
|  |     def tall(x, y): | ||
|  |         current_map[y][x] = 23 | ||
|  |         current_map[y + 1][x] = 31 | ||
|  | 
 | ||
|  |     def single(x, y): | ||
|  |         current_map[y][x] = 2 | ||
|  | 
 | ||
|  |     # 0 = top left, 1 = middle, 2 = top right, 3 = bottom right | ||
|  |     entrance_c = random.choice([0, 1, 2]) | ||
|  |     exit_c = [0, 1, 3] | ||
|  |     if entrance_c == 2: | ||
|  |         exit_c.remove(1) | ||
|  |     else: | ||
|  |         exit_c.remove(entrance_c) | ||
|  |     exit_c = random.choice(exit_c) | ||
|  |     remaining = [i for i in [0, 1, 2, 3] if i not in [entrance_c, exit_c]] | ||
|  | 
 | ||
|  |     if entrance_c == 0: | ||
|  |         floor(6, 3) | ||
|  |         floor(6, 4) | ||
|  |         tall(random.randint(8, 10), 2) | ||
|  |         wide(4, random.randint(5, 7)) | ||
|  |         wide(1, random.choice([5, 6, 7, 9])) | ||
|  |     elif entrance_c == 1: | ||
|  |         if remaining == [0, 2] or random.randint(0, 1): | ||
|  |             tall(random.randint(8, 10), 2) | ||
|  |             floor(7, 4) | ||
|  |             floor(8, 4) | ||
|  |         else: | ||
|  |             tall(random.randint(11, 12), 5) | ||
|  |             floor(9, 5) | ||
|  |             floor(9, 6) | ||
|  |     elif entrance_c == 2: | ||
|  |         floor(16, 2) | ||
|  |         floor(16, 3) | ||
|  |         if remaining == [1, 3]: | ||
|  |             wide(17, 4) | ||
|  |         else: | ||
|  |             tall(random.randint(11, 17), random.choice([2, 5])) | ||
|  | 
 | ||
|  |     if exit_c == 0: | ||
|  |         r = random.sample([0, 1, 2], 2) | ||
|  |         if 0 in r: | ||
|  |             floor(1, 11) | ||
|  |             floor(2, 11) | ||
|  |         if 1 in r: | ||
|  |             floor(3, 11) | ||
|  |             floor(4, 11) | ||
|  |         if 2 in r: | ||
|  |             floor(5, 11) | ||
|  |             floor(6, 11) | ||
|  |     elif exit_c == 1 or (exit_c == 3 and entrance_c == 0): | ||
|  |         r = random.sample([1, 3, 5, 7], random.randint(1, 2)) | ||
|  |         for i in r: | ||
|  |             floor(i, 11) | ||
|  |             floor(i + 1, 11) | ||
|  |     if exit_c != 3: | ||
|  |         tall(random.choice([9, 10, 12]), 12) | ||
|  | 
 | ||
|  |     # 0 = top left, 1 = middle, 2 = top right, 3 = bottom right | ||
|  |     # [0, 1] [0, 2] [1, 2] [1, 3], [2, 3] | ||
|  |     if remaining[0] == 1: | ||
|  |         floor(9, 5) | ||
|  |         floor(9, 6) | ||
|  | 
 | ||
|  |     if remaining == [0, 2]: | ||
|  |         if random.randint(0, 1): | ||
|  |             tall(9, 4) | ||
|  |             floor(9, 6) | ||
|  |             floor(9, 7) | ||
|  |         else: | ||
|  |             floor(10, 7) | ||
|  |             floor(11, 7) | ||
|  | 
 | ||
|  |     if remaining == [1, 2]: | ||
|  |         floor(16, 2) | ||
|  |         floor(16, 3) | ||
|  |         tall(random.randint(11, 17), random.choice([2, 5])) | ||
|  |     if remaining in [[1, 3], [2, 3]]: | ||
|  |         r = round(random.triangular(0, 3, 0)) | ||
|  |         floor(12 + (r * 2), 7) | ||
|  |         if r < 3: | ||
|  |             floor(13 + (r * 2), 7) | ||
|  |     if remaining == [1, 3]: | ||
|  |         wide(10, random.choice([3, 5])) | ||
|  | 
 | ||
|  |     if remaining != [0, 1] and exit_c != 1: | ||
|  |         wide(7, 6) | ||
|  | 
 | ||
|  |     if entrance_c != 0: | ||
|  |         if random.randint(0, 1): | ||
|  |             wide(4, random.randint(4, 7)) | ||
|  |         else: | ||
|  |             wide(1, random.choice([5, 6, 7, 9])) | ||
|  | 
 | ||
|  |     current_map = map2f | ||
|  | 
 | ||
|  |     if 3 in remaining: | ||
|  |         c = random.choice([entrance_c, exit_c]) | ||
|  |     else: | ||
|  |         c = random.choice(remaining) | ||
|  | 
 | ||
|  |     # 0 = top right, 1 = middle, 2 = bottom right, 3 = top left | ||
|  |     if c in [0, 1]: | ||
|  |         if random.randint(0, 2): | ||
|  |             tall(random.choice([2, 4]), 5) | ||
|  |             r = random.choice([1, 3, 7, 9, 11]) | ||
|  |             floor(3 if r < 11 else random.randint(1, 2), r) | ||
|  |             floor(3 if r < 11 else random.randint(1, 2), r + 1) | ||
|  |         if random.randint(0, 2): | ||
|  |             tall(random.randint(6, 7), 7) | ||
|  |             r = random.choice([1, 3, 5, 9]) | ||
|  |             floor(6, r) | ||
|  |             floor(6, r + 1) | ||
|  |         if random.randint(0, 2): | ||
|  |             wide(7, 15) | ||
|  |             r = random.randint(0, 4) | ||
|  |             if r == 0: | ||
|  |                 floor(9, 14) | ||
|  |                 floor(10, 14) | ||
|  |             elif r == 1: | ||
|  |                 floor(11, 14) | ||
|  |                 floor(12, 14) | ||
|  |             elif r == 2: | ||
|  |                 floor(13, 13) | ||
|  |                 floor(13, 14) | ||
|  |             elif r == 3: | ||
|  |                 floor(13, 11) | ||
|  |                 floor(13, 12) | ||
|  |             elif r == 4: | ||
|  |                 floor(13, 10) | ||
|  |                 floor(14, 10) | ||
|  |     if c == 0: | ||
|  |         tall(random.randint(9, 10), 5) | ||
|  |         if random.randint(0, 1): | ||
|  |             floor(10, 7) | ||
|  |             floor(11, 7) | ||
|  |             tall(random.randint(12, 17), 8) | ||
|  |         else: | ||
|  |             floor(12, 5) | ||
|  |             floor(12, 6) | ||
|  |             wide(13, random.randint(4, 5)) | ||
|  |             wide(17, random.randint(3, 5)) | ||
|  |         r = random.choice([1, 3]) | ||
|  |         floor(12, r) | ||
|  |         floor(12,  + 1) | ||
|  | 
 | ||
|  |     elif c == 2: | ||
|  |         r = random.randint(0, 6) | ||
|  |         if r == 0: | ||
|  |             floor(12, 1) | ||
|  |             floor(12, 2) | ||
|  |         elif r == 1: | ||
|  |             floor(12, 3) | ||
|  |             floor(12, 4) | ||
|  |         elif r == 2: | ||
|  |             floor(12, 5) | ||
|  |             floor(12, 6) | ||
|  |         elif r == 3: | ||
|  |             floor(10, 7) | ||
|  |             floor(11, 7) | ||
|  |         elif r == 4: | ||
|  |             floor(9, 7) | ||
|  |             floor(9, 8) | ||
|  |         elif r == 5: | ||
|  |             floor(9, 9) | ||
|  |             floor(9, 10) | ||
|  |         elif r == 6: | ||
|  |             floor(8, 11) | ||
|  |             floor(9, 11) | ||
|  |         if r < 2 or (r in [2, 3] and random.randint(0, 1)): | ||
|  |             wide(7, random.randint(6, 7)) | ||
|  |         elif r in [2, 3]: | ||
|  |             tall(random.randint(9, 10), 5) | ||
|  |         else: | ||
|  |             tall(random.randint(6, 7), 7) | ||
|  |         r = random.randint(r, 6) | ||
|  |         if r == 0: | ||
|  |             #early block | ||
|  |             wide(13, random.randint(2, 5)) | ||
|  |             tall(random.randint(14, 15), 1) | ||
|  |         elif r == 1: | ||
|  |             if random.randint(0, 1): | ||
|  |                 tall(16, 5) | ||
|  |                 tall(random.choice([14, 15, 17]), 1) | ||
|  |             else: | ||
|  |                 wide(16, random.randint(6,8)) | ||
|  |                 single(18, 7) | ||
|  |         elif r == 2: | ||
|  |             tall(random.randint(12, 16), 8) | ||
|  |         elif r == 3: | ||
|  |             wide(10, 9) | ||
|  |             single(12, 9) | ||
|  |         elif r == 4: | ||
|  |             wide(10, random.randint(11, 12)) | ||
|  |             single(12, random.randint(11, 12)) | ||
|  |         elif r == 5: | ||
|  |             tall(random.randint(8, 10), 12) | ||
|  |         elif r == 6: | ||
|  |             wide(7, 15) | ||
|  |         r = random.randint(r, 6) | ||
|  |         if r == 6: | ||
|  |             #late open | ||
|  |             r2 = random.randint(0, 2) | ||
|  |             floor(1 + (r2 * 2), 14) | ||
|  |             floor(2 + (r2 * 2), 14) | ||
|  |         elif r == 5: | ||
|  |             floor(6, 12) | ||
|  |             floor(6, 13) | ||
|  |         elif r == 4: | ||
|  |             if random.randint(0, 1): | ||
|  |                 floor(6, 11) | ||
|  |                 floor(7, 11) | ||
|  |             else: | ||
|  |                 floor(8, 11) | ||
|  |                 floor(9, 11) | ||
|  |         elif r == 3: | ||
|  |             floor(9, 9) | ||
|  |             floor(9, 10) | ||
|  |         elif r < 3: | ||
|  |             single(9, 7) | ||
|  |             floor(9, 8) | ||
|  | 
 | ||
|  |     def check_addable_block(check_map, disallowed): | ||
|  |         if check_map[y][x] == 1 and [x, y] not in disallowed: | ||
|  |             i = 0 | ||
|  |             for xx in range(x-1, x+2): | ||
|  |                 for yy in range(y-1, y+2): | ||
|  |                     if check_map[yy][xx] == 1: | ||
|  |                         i += 1 | ||
|  |             if i >= 8: | ||
|  |                 single(x, y) | ||
|  | 
 | ||
|  |     for _ in range(100): | ||
|  |         y = random.randint(1, 16) | ||
|  |         x = random.randint(1, 18) | ||
|  |         current_map = map1f | ||
|  |         check_addable_block(map1f, disallowed1F) | ||
|  |         current_map = map2f | ||
|  |         check_addable_block(map2f, disallowed2F) | ||
|  | 
 | ||
|  |     address = rom_addresses["Map_Rock_Tunnel1F"] | ||
|  |     for y in map1f: | ||
|  |         for x in y: | ||
|  |             data[address] = x | ||
|  |             address += 1 | ||
|  |     address = rom_addresses["Map_Rock_TunnelB1F"] | ||
|  |     for y in map2f: | ||
|  |         for x in y: | ||
|  |             data[address] = x | ||
|  |             address += 1 | ||
|  |     return seed |