X-Git-Url: https://code.delx.au/refind/blobdiff_plain/4c9f41e161bd197922912efbcf4cc676077d5c00..75afe82dbf9803baa447499a1e7942a4ca342535:/refind/icns.c diff --git a/refind/icns.c b/refind/icns.c index f9be58b..7b3af02 100644 --- a/refind/icns.c +++ b/refind/icns.c @@ -51,29 +51,29 @@ typedef struct { } BUILTIN_ICON; BUILTIN_ICON BuiltinIconTable[BUILTIN_ICON_COUNT] = { - { NULL, L"func_about.icns", 48 }, - { NULL, L"func_reset.icns", 48 }, - { NULL, L"func_shutdown.icns", 48 }, - { NULL, L"func_exit.icns", 48 }, - { NULL, L"tool_shell.icns", 48 }, - { NULL, L"tool_part.icns", 48 }, - { NULL, L"tool_rescue.icns", 48 }, - { NULL, L"vol_internal.icns", 32 }, - { NULL, L"vol_external.icns", 32 }, - { NULL, L"vol_optical.icns", 32 }, + { NULL, L"func_about", 48 }, + { NULL, L"func_reset", 48 }, + { NULL, L"func_shutdown", 48 }, + { NULL, L"func_exit", 48 }, + { NULL, L"tool_shell", 48 }, + { NULL, L"tool_part", 48 }, + { NULL, L"tool_rescue", 48 }, + { NULL, L"tool_apple_rescue", 48 }, + { NULL, L"tool_mok_tool", 48 }, + { NULL, L"vol_internal", 32 }, + { NULL, L"vol_external", 32 }, + { NULL, L"vol_optical", 32 }, }; EG_IMAGE * BuiltinIcon(IN UINTN Id) { - CHAR16 FileName[256]; - if (Id >= BUILTIN_ICON_COUNT) return NULL; if (BuiltinIconTable[Id].Image == NULL) { - SPrint(FileName, 255, L"%s\\%s", GlobalConfig.IconsDir ? GlobalConfig.IconsDir : DEFAULT_ICONS_DIR, - BuiltinIconTable[Id].FileName); - BuiltinIconTable[Id].Image = LoadIcnsFallback(SelfDir, FileName, BuiltinIconTable[Id].PixelSize); + BuiltinIconTable[Id].Image = egFindIcon(BuiltinIconTable[Id].FileName, BuiltinIconTable[Id].PixelSize); + if (BuiltinIconTable[Id].Image == NULL) + BuiltinIconTable[Id].Image = DummyImage(BuiltinIconTable[Id].PixelSize); } // if return BuiltinIconTable[Id].Image; @@ -84,54 +84,60 @@ EG_IMAGE * BuiltinIcon(IN UINTN Id) // // Load an OS icon from among the comma-delimited list provided in OSIconName. +// Searches for icons with extensions in the ICON_EXTENSIONS list (via +// egFindIcon()). +// Returns image data. On failure, returns an ugly "dummy" icon. EG_IMAGE * LoadOSIcon(IN CHAR16 *OSIconName OPTIONAL, IN CHAR16 *FallbackIconName, BOOLEAN BootLogo) { - EG_IMAGE *Image; - CHAR16 *CutoutName; - CHAR16 FileName[256]; + EG_IMAGE *Image = NULL; + CHAR16 *CutoutName, BaseName[256]; UINTN Index = 0; if (GlobalConfig.TextOnly) // skip loading if it's not used anyway return NULL; - // try the names from OSIconName - while ((CutoutName = FindCommaDelimited(OSIconName, Index++)) != NULL) { - SPrint(FileName, 255, L"%s\\%s_%s.icns", GlobalConfig.IconsDir ? GlobalConfig.IconsDir : DEFAULT_ICONS_DIR, - BootLogo ? L"boot" : L"os", CutoutName); + // First, try to find an icon from the OSIconName list.... + while (((CutoutName = FindCommaDelimited(OSIconName, Index++)) != NULL) && (Image == NULL)) { + SPrint(BaseName, 255, L"%s_%s", BootLogo ? L"boot" : L"os", CutoutName); + Image = egFindIcon(BaseName, 128); + } - // try to load it - Image = egLoadIcon(SelfDir, FileName, 128); - if (Image != NULL) { - FreePool(CutoutName); - return Image; - } - FreePool(CutoutName); - } // while + // If that fails, try again using the FallbackIconName.... + if (Image == NULL) { + SPrint(BaseName, 255, L"%s_%s", BootLogo ? L"boot" : L"os", FallbackIconName); + Image = egFindIcon(BaseName, 128); + } - // try the fallback name - SPrint(FileName, 255, L"%s\\%s_%s.icns", GlobalConfig.IconsDir ? GlobalConfig.IconsDir : DEFAULT_ICONS_DIR, - BootLogo ? L"boot" : L"os", FallbackIconName); - Image = egLoadIcon(SelfDir, FileName, 128); - if (Image != NULL) - return Image; + // If that fails and if BootLogo was set, try again using the "os_" start of the name.... + if (BootLogo && (Image == NULL)) { + SPrint(BaseName, 255, L"os_%s", FallbackIconName); + Image = egFindIcon(BaseName, 128); + } - // try the fallback name with os_ instead of boot_ - if (BootLogo) - return LoadOSIcon(NULL, FallbackIconName, FALSE); + // If all of these fail, return the dummy image.... + if (Image == NULL) + Image = DummyImage(128); - return DummyImage(128); + return Image; } /* EG_IMAGE * LoadOSIcon() */ -// -// Load an image from a .icns file -// -EG_IMAGE * LoadIcns(IN EFI_FILE_HANDLE BaseDir, IN CHAR16 *FileName, IN UINTN PixelSize) -{ - if (GlobalConfig.TextOnly) // skip loading if it's not used anyway - return NULL; - return egLoadIcon(BaseDir, FileName, PixelSize); -} +// Load an image from a .icns or .png file. FileNames contains a comma-delimited +// list of potential filanems; the function tries each in turn until it finds +// a loadable icon. If no file has a valid icon, it returns NULL. +EG_IMAGE * LoadIcns(IN EFI_FILE_HANDLE BaseDir, IN CHAR16 *FileNames, IN UINTN PixelSize) { + CHAR16 *FileName; + EG_IMAGE *Image = NULL; + UINTN Index = 0; + + if (GlobalConfig.TextOnly) // skip loading if it's not used anyway + return NULL; + + while (((FileName = FindCommaDelimited(FileNames, Index++)) != NULL) && (Image == NULL)) { + Image = egLoadIcon(BaseDir, FileName, PixelSize); + } + return Image; +} // EG_IMAGE * LoadIcns() static EG_PIXEL BlackPixel = { 0x00, 0x00, 0x00, 0 }; //static EG_PIXEL YellowPixel = { 0x00, 0xff, 0xff, 0 }; @@ -141,11 +147,11 @@ EG_IMAGE * DummyImage(IN UINTN PixelSize) EG_IMAGE *Image; UINTN x, y, LineOffset; CHAR8 *Ptr, *YPtr; - + Image = egCreateFilledImage(PixelSize, PixelSize, TRUE, &BlackPixel); - + LineOffset = PixelSize * 4; - + YPtr = (CHAR8 *)Image->PixelData + ((PixelSize - 32) >> 1) * (LineOffset + 4); for (y = 0; y < 32; y++) { Ptr = YPtr; @@ -166,16 +172,3 @@ EG_IMAGE * DummyImage(IN UINTN PixelSize) return Image; } - -EG_IMAGE * LoadIcnsFallback(IN EFI_FILE_HANDLE BaseDir, IN CHAR16 *FileName, IN UINTN PixelSize) -{ - EG_IMAGE *Image; - if (GlobalConfig.TextOnly) // skip loading if it's not used anyway - return NULL; - - Image = LoadIcns(BaseDir, FileName, PixelSize); - if (Image == NULL) { - Image = DummyImage(PixelSize); - } - return Image; -}