Marlin_Firmware/buildroot/share/dwin/bin/DWIN_ICO.py
2022-08-01 01:59:37 -05:00

343 lines
11 KiB
Python

# DWIN_ICO
# - Dissect and create DWIN .ico files for their LCD displays.
#
# Copyright (c) 2020 Brent Burton
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
#----------------------------------------------------------------
#
# This is not a normal Microsoft .ICO file, but it has a similar
# structure for containing a number of icon images. Each icon is
# a small JPG file.
#
# The file has a directory header containing fixed-length
# records, and each record points to its data at an offset later
# in the file.
#
# The directory entries are 16 bytes each, and the entire
# directory is 4KB (0 - 0x1000). This supports 256 entries.
#
# Multibyte values are in Big Endian format.
#
# Header: (offset 0x0)
# W H offset ?? len ?? ??
# Entry 0: xxxx xxxx 00001000 xx 10a2 00 00000000
# Entry 1: xxxx xxxx 000020a2 xx 0eac 00 00000000
# Entry 2: xxxx xxxx 00002f4e xx 0eaa 00 00000000
# ...
# 0x00001000: ffd8 ffe1 0018 ... jpeg exif and data follow .. ffd9
# 0x000020a2: ffd8 ffe1 ...
# ...rest of ICO entries' data...
#
# Header structure:
# Offset Len What
# 0 2 width
# 2 2 height
# 4 4 file byte position from SEEK_BEG
# 8 3 length of data
# 11 5 ??? all zeroes
#
# Other notes:
# * The index of each icon corresponds to the Icon number in dwin.h
# * One exception is number 39: that header entry is blank, and dwin.h
# does not define a name for 39. This is specially handled to
# prevent reordering stock icons.
import os
import struct
from PIL import Image
def getJpegResolution(jpegFile):
"""Returns a 2-tuple containing the jpegFile's (width, height).
"""
img = Image.open(jpegFile)
return img.size
class DWIN_ICO_File():
def __init__(self):
self.entries = [] # list of header entries
def splitFile(self, filename, outDir):
if not filename[-4:].lower() == '.ico':
raise RuntimeError('Input file must end in .ico')
with open(filename, 'rb') as infile:
self._parseHeader(infile)
self._splitEntryData(infile, outDir)
return
def _parseHeader(self, infile):
maxEntries = 256
count = 0
validEntries = 0
while count < maxEntries:
rawBytes = infile.read(16)
entry = Entry()
entry.parseRawData(rawBytes)
# check that it is valid: is offset nonzero?
# Special case: treat 39 as valid
if (entry.offset > 0) or (count == 39):
validEntries += 1
self.entries.append(entry)
count += 1
return
def _splitEntryData(self, infile, outDir):
print('Splitting Entry Data...')
if 0 == len(self.entries):
raise RuntimeError('.ico file is not loaded yet')
# check for output dir:
if not os.path.exists(outDir):
os.mkdir(outDir)
# keep a count
count = 0
for entry in self.entries:
# Skip any empty entries. (Special handling of 39.)
if entry.length == 0:
count += 1
continue
# Seek file position, read length bytes, and write to new output file.
print('%02d: offset: 0x%06x len: 0x%04x width: %d height: %d' %
(count, entry.offset, entry.length, entry.width, entry.height))
outfilename = os.path.join(outDir,
'%03d-%s.jpg' % (count, _iconNames[count]))
with open(outfilename, 'wb') as outfile:
infile.seek(entry.offset)
blob = infile.read(entry.length)
outfile.write(blob)
print('Wrote %d bytes to %s' % (entry.length, outfilename))
count += 1
return
def createFile(self, iconDir, filename):
'''Create a new .ico file from the contents of iconDir.
The contents of iconDir are processed to get image
resolution, and a new entry is created for each.
Each filename must have a leading number followed by a
dash, which is the icon index. E.g., "071-ICON_StepX.jpg".
'''
self.entries = [Entry() for i in range(0,256)]
# 1. Scan icon directory and record all valid files
print('Scanning icon directory', iconDir)
count = 0
for dirEntry in os.scandir(iconDir):
if not dirEntry.is_file():
print('...Ignoring', dirEntry.path)
continue
# process each file:
try:
index = int(dirEntry.name[0:3])
if not (0 <= index <= 255):
print('...Ignoring invalid index on', dirEntry.path)
continue
#dirEntry.path is iconDir/name
w,h = getJpegResolution(dirEntry.path)
length = dirEntry.stat().st_size
e = self.entries[index]
e.width = w
e.height = h
e.length = length
e.filename = dirEntry.path
count += 1
except Exception as e:
print('Whoops: ', e)
pass
print('...Scanned %d icon files' % (count))
# 2. Scan over valid header entries and update offsets
self._updateHeaderOffsets()
# 3. Write out header to .ico file, the append each icon file
self._combineAndWriteIcoFile(filename)
print('Scanning done. %d icons included.' % (count))
def _updateHeaderOffsets(self):
"""Iterate over all header entries and update their offsets.
"""
offset = 256 * 16
for i in range(0,256):
e = self.entries[i]
if e.length == 0:
continue
e.offset = offset
offset += e.length
#print('%03d: (%d x %d) len=%d off=%d' %
# (i, e.width, e.height, e.length, e.offset))
return
def _combineAndWriteIcoFile(self, filename):
"""Write out final .ico file.
All header entries are updated, so write out
the final header contents, and concat each icon
file to the .ico.
"""
with open(filename, 'wb') as outfile:
# 1. Write header directory
for e in self.entries:
outfile.write( e.serialize() )
if outfile.tell() != 4096:
raise RuntimeError('Header directory write failed. Not 4096 bytes')
# 2. For each entry, concat the icon file data
for e in self.entries:
if 0 == e.length: continue
guts = self._getFileContents(e.filename, e.length)
outfile.write(guts)
return
def _getFileContents(self, filename, length):
"""Read contents of filename, and return bytes"""
with open(filename, 'rb') as infile:
contents = infile.read(length)
if len(contents) != length:
raise RuntimeError('Failed to read contents of', filename)
return contents
class Entry():
'''Entry objects record resolution and size information
about each icon stored in an ICO file.
'''
__slots__ = ('width', 'height', 'offset', 'length', 'filename')
def __init__(self, w=0, h=0, length=0, offset=0):
self.width = w
self.height = h
self.offset = offset
self.length = length
self.filename = None
def parseRawData(self, rawEntryBytes):
if len(rawEntryBytes) != 16:
raise RuntimeError('Entry: data must be 16 bytes long')
# Split data into bigendian fields
(w, h, off, len3, len21, b1,b2,b3,b4,b5) = \
struct.unpack('>HHLBHBBBBB', rawEntryBytes)
self.width = w
self.height = h
self.offset = off
self.length = len3 * 65536 + len21
return
def serialize(self):
"""Convert this Entry's information into a 16-byte
.ico directory entry record. Return bytes object.
"""
len21 = self.length % 65536
len3 = self.length // 65536
rawdata = struct.pack('>HHLBHBBBBB', self.width, self.height,
self.offset, len3, len21,
0, 0, 0, 0, 0)
return rawdata
_iconNames = {
0 : 'ICON_LOGO',
1 : 'ICON_Print_0',
2 : 'ICON_Print_1',
3 : 'ICON_Prepare_0',
4 : 'ICON_Prepare_1',
5 : 'ICON_Control_0',
6 : 'ICON_Control_1',
7 : 'ICON_Leveling_0',
8 : 'ICON_Leveling_1',
9 : 'ICON_HotendTemp',
10 : 'ICON_BedTemp',
11 : 'ICON_Speed',
12 : 'ICON_Zoffset',
13 : 'ICON_Back',
14 : 'ICON_File',
15 : 'ICON_PrintTime',
16 : 'ICON_RemainTime',
17 : 'ICON_Setup_0',
18 : 'ICON_Setup_1',
19 : 'ICON_Pause_0',
20 : 'ICON_Pause_1',
21 : 'ICON_Continue_0',
22 : 'ICON_Continue_1',
23 : 'ICON_Stop_0',
24 : 'ICON_Stop_1',
25 : 'ICON_Bar',
26 : 'ICON_More',
27 : 'ICON_Axis',
28 : 'ICON_CloseMotor',
29 : 'ICON_Homing',
30 : 'ICON_SetHome',
31 : 'ICON_PLAPreheat',
32 : 'ICON_ABSPreheat',
33 : 'ICON_Cool',
34 : 'ICON_Language',
35 : 'ICON_MoveX',
36 : 'ICON_MoveY',
37 : 'ICON_MoveZ',
38 : 'ICON_Extruder',
# no 39
40 : 'ICON_Temperature',
41 : 'ICON_Motion',
42 : 'ICON_WriteEEPROM',
43 : 'ICON_ReadEEPROM',
44 : 'ICON_ResumeEEPROM',
45 : 'ICON_Info',
46 : 'ICON_SetEndTemp',
47 : 'ICON_SetBedTemp',
48 : 'ICON_FanSpeed',
49 : 'ICON_SetPLAPreheat',
50 : 'ICON_SetABSPreheat',
51 : 'ICON_MaxSpeed',
52 : 'ICON_MaxAccelerated',
53 : 'ICON_MaxJerk',
54 : 'ICON_Step',
55 : 'ICON_PrintSize',
56 : 'ICON_Version',
57 : 'ICON_Contact',
58 : 'ICON_StockConfiguraton',
59 : 'ICON_MaxSpeedX',
60 : 'ICON_MaxSpeedY',
61 : 'ICON_MaxSpeedZ',
62 : 'ICON_MaxSpeedE',
63 : 'ICON_MaxAccX',
64 : 'ICON_MaxAccY',
65 : 'ICON_MaxAccZ',
66 : 'ICON_MaxAccE',
67 : 'ICON_MaxSpeedJerkX',
68 : 'ICON_MaxSpeedJerkY',
69 : 'ICON_MaxSpeedJerkZ',
70 : 'ICON_MaxSpeedJerkE',
71 : 'ICON_StepX',
72 : 'ICON_StepY',
73 : 'ICON_StepZ',
74 : 'ICON_StepE',
75 : 'ICON_Setspeed',
76 : 'ICON_SetZOffset',
77 : 'ICON_Rectangle',
78 : 'ICON_BLTouch',
79 : 'ICON_TempTooLow',
80 : 'ICON_AutoLeveling',
81 : 'ICON_TempTooHigh',
82 : 'ICON_NoTips_C',
83 : 'ICON_NoTips_E',
84 : 'ICON_Continue_C',
85 : 'ICON_Continue_E',
86 : 'ICON_Cancel_C',
87 : 'ICON_Cancel_E',
88 : 'ICON_Confirm_C',
89 : 'ICON_Confirm_E',
90 : 'ICON_Info_0',
91 : 'ICON_Info_1'
}