# 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 . #---------------------------------------------------------------- # # 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' }