1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
#!/usr/bin/python
# Amiga AGI game palette extractor.
# Extracts palette from an Amiga AGI game's executable file.
# Initial version written during summer of 2007 by Buddha^.
# Unoptimized.
import struct, sys, os.path
# Constants
maxComponentValue = 0x0F
colorsPerPalette = 16
componentsPerColor = 3
bytesPerComponent = 2
bytesPerColor = componentsPerColor * bytesPerComponent
paletteComponentCount = colorsPerPalette * componentsPerColor
paletteByteCount = paletteComponentCount * bytesPerComponent
black = tuple([0 for x in range(componentsPerColor)])
white = tuple([maxComponentValue for x in range(componentsPerColor)])
componentPrintFormat = "0x%1X"
def is12BitColor(color):
"""Is color 12-bit (i.e. 4 bits per color component)?"""
return color == filter(lambda component: 0 <= component <= maxComponentValue, color)
def printColor(color, tabulate = True, printLastComma = True, newLine = True):
"""Prints color with optional start tabulation, comma in the end and a newline"""
result = ""
if tabulate:
result = result + "\t"
for component in color[:-1]:
result = result + ((componentPrintFormat + ", ") % component)
result = result + (componentPrintFormat % color[-1])
if printLastComma:
result = result + ","
if newLine:
print result
else:
print result,
def printPalette(palette, arrayName):
"""Prints out palette as a C-style array"""
print "/** A 16-color, 12-bit RGB palette from an Amiga AGI game. */"
print "static const unsigned char " + arrayName + "[] = {"
for color in palette[:-1]:
printColor(color)
printColor(palette[-1], printLastComma = False)
print("};")
def isAmigaPalette(palette):
"""Test if the given palette is an Amiga-style palette"""
# Palette must be of correct size
if len(palette) != colorsPerPalette:
return False
# First palette color must be black, last color white
if palette[0] != black or palette[-1] != white:
return False
# All colors must be 12-bit (i.e. 4 bits per color component)
for color in palette:
if not is12BitColor(color):
return False
# All colors must be unique
if len(set(palette)) != colorsPerPalette:
return False
return True
# The main program starts here
if len(sys.argv) <= 2 or sys.argv[1] == "-h" or sys.argv[1] == "--help":
quit("Usage: " + os.path.basename(sys.argv[0]) + " FILE ARRAYNAME\n" \
" Searches FILE for an Amiga AGI game palette\n" \
" and prints out any found candidates as C-array style arrays with the name ARRAYNAME.");
# Open file and read it into memory
file = open(sys.argv[1], 'rb')
data = file.read()
foundPalettes = []
palette = [black for x in range(colorsPerPalette)]
# Search through the whole file
for searchPosition in range(len(data) - paletteByteCount + 1):
# Parse possible palette from byte data
for colorNum in range(colorsPerPalette):
paletteStart = searchPosition + colorNum * bytesPerColor
paletteEnd = paletteStart + bytesPerColor
# Parse color components as unsigned 16-bit big endian integers
color = struct.unpack('>' + 'H' * componentsPerColor, data[paletteStart:paletteEnd]);
palette[colorNum] = color;
# Save good candidates to a list
if isAmigaPalette(palette):
foundPalettes.append(tuple(palette));
# Close file
file.close()
# Remove duplicates from found palettes list
uniqueFoundPalettes = set(foundPalettes)
# Print out the found unique palettes
if len(uniqueFoundPalettes) > 0:
for palette in uniqueFoundPalettes:
printPalette(palette, sys.argv[2])
else:
print "Didn't find any Amiga AGI game palettes in file " + sys.argv[1]
print "Heuristic isn't perfect though, so caveat emptor"
|
