]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Library/GraphicsLib/Graphics.c
Remove the unnecessary include, such as PiDxe.h, PiPei.h, Base.h and Uefi.h in header...
[mirror_edk2.git] / MdeModulePkg / Library / GraphicsLib / Graphics.c
index e944217028dda72f8f6307c27450f8c3fd8256ce..75aa91dc6c2f1a5e29a55f68428da166816ea409 100644 (file)
@@ -1,11 +1,10 @@
-/**@file\r
-  Support for Basic Graphics operations.\r
+/** @file\r
+  Library supports diplaying graphical splash screen,\r
+  locking of keyboard input and printing character on\r
+  screen. These basic graphics operations are based on UEFI HII, \r
+  Graphics Output protocol or UGA Draw protocol.\r
 \r
-  BugBug: Currently *.BMP files are supported. This will be replaced\r
-          when Tiano graphics format is supported.\r
-\r
-\r
-Copyright (c) 2006, Intel Corporation\r
+Copyright (c) 2006 - 2008, Intel Corporation\r
 All rights reserved. This program and the accompanying materials\r
 are licensed and made available under the terms and conditions of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
@@ -23,7 +22,6 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 #include <Protocol/OEMBadging.h>\r
 #include <Protocol/ConsoleControl.h>\r
 #include <Protocol/GraphicsOutput.h>\r
-#include <Protocol/FirmwareVolume2.h>\r
 #include <Protocol/UgaDraw.h>\r
 #include <Protocol/HiiFont.h>\r
 #include <Protocol/HiiImage.h>\r
@@ -60,77 +58,58 @@ STATIC EFI_GRAPHICS_OUTPUT_BLT_PIXEL mEfiColors[16] = {
 };\r
 \r
 \r
+/**\r
+  Return the graphics image file named FileNameGuid into Image and return it's\r
+  size in ImageSize. All Firmware Volumes (FV) in the system are searched for the\r
+  file name.\r
+\r
+  @param  FileNameGuid          File Name of graphics file in the FV(s).\r
+  @param  Image                 Pointer to pointer to return graphics image.  If NULL, a \r
+                                buffer will be allocated.\r
+  @param  ImageSize             Size of the graphics Image in bytes. Zero if no image found.\r
+\r
+  @retval EFI_SUCCESS           Image and ImageSize are valid. \r
+  @retval EFI_BUFFER_TOO_SMALL  Image not big enough. ImageSize has required size\r
+  @retval EFI_NOT_FOUND         FileNameGuid not found\r
+\r
+**/\r
 EFI_STATUS\r
+EFIAPI\r
 GetGraphicsBitMapFromFV (\r
   IN  EFI_GUID      *FileNameGuid,\r
   OUT VOID          **Image,\r
   OUT UINTN         *ImageSize\r
   )\r
-/*++\r
-\r
-Routine Description:\r
+{\r
+  return GetGraphicsBitMapFromFVEx (NULL, FileNameGuid, Image, ImageSize);\r
+}\r
 \r
+/**\r
   Return the graphics image file named FileNameGuid into Image and return it's\r
   size in ImageSize. All Firmware Volumes (FV) in the system are searched for the\r
   file name.\r
 \r
-Arguments:\r
-\r
-  FileNameGuid  - File Name of graphics file in the FV(s).\r
-\r
-  Image         - Pointer to pointer to return graphics image.  If NULL, a\r
-                  buffer will be allocated.\r
-\r
-  ImageSize     - Size of the graphics Image in bytes. Zero if no image found.\r
-\r
-\r
-Returns:\r
+  @param  ImageHandle           The driver image handle of the caller. The parameter is used to\r
+                                optimize the loading of the image file so that the FV from which\r
+                                the driver image is loaded will be tried first. \r
+  @param  FileNameGuid          File Name of graphics file in the FV(s).\r
+  @param  Image                 Pointer to pointer to return graphics image.  If NULL, a \r
+                                buffer will be allocated.\r
+  @param  ImageSize             Size of the graphics Image in bytes. Zero if no image found.\r
 \r
-  EFI_SUCCESS          - Image and ImageSize are valid.\r
-  EFI_BUFFER_TOO_SMALL - Image not big enough. ImageSize has required size\r
-  EFI_NOT_FOUND        - FileNameGuid not found\r
-\r
---*/\r
-{\r
-  return GetGraphicsBitMapFromFVEx (NULL, FileNameGuid, Image, ImageSize);\r
-}\r
+  @retval EFI_SUCCESS           Image and ImageSize are valid. \r
+  @retval EFI_BUFFER_TOO_SMALL  Image not big enough. ImageSize has required size\r
+  @retval EFI_NOT_FOUND         FileNameGuid not found\r
 \r
+**/\r
 EFI_STATUS\r
+EFIAPI\r
 GetGraphicsBitMapFromFVEx (\r
   IN  EFI_HANDLE    ImageHandle,\r
   IN  EFI_GUID      *FileNameGuid,\r
   OUT VOID          **Image,\r
   OUT UINTN         *ImageSize\r
   )\r
-/*++\r
-\r
-Routine Description:\r
-\r
-  Return the graphics image file named FileNameGuid into Image and return it's\r
-  size in ImageSize. All Firmware Volumes (FV) in the system are searched for the\r
-  file name.\r
-\r
-Arguments:\r
-\r
-  ImageHandle   - The driver image handle of the caller. The parameter is used to\r
-                  optimize the loading of the image file so that the FV from which\r
-                  the driver image is loaded will be tried first.\r
-\r
-  FileNameGuid  - File Name of graphics file in the FV(s).\r
-\r
-  Image         - Pointer to pointer to return graphics image.  If NULL, a\r
-                  buffer will be allocated.\r
-\r
-  ImageSize     - Size of the graphics Image in bytes. Zero if no image found.\r
-\r
-\r
-Returns:\r
-\r
-  EFI_SUCCESS          - Image and ImageSize are valid.\r
-  EFI_BUFFER_TOO_SMALL - Image not big enough. ImageSize has required size\r
-  EFI_NOT_FOUND        - FileNameGuid not found\r
-\r
---*/\r
 {\r
   return PiLibGetSectionFromAnyFv (\r
            FileNameGuid,\r
@@ -141,67 +120,58 @@ Returns:
            );\r
 }\r
 \r
-\r
-EFI_STATUS\r
-ConvertBmpToGopBlt (\r
-  IN  VOID      *BmpImage,\r
-  IN  UINTN     BmpImageSize,\r
-  IN OUT VOID   **GopBlt,\r
-  IN OUT UINTN  *GopBltSize,\r
-  OUT UINTN     *PixelHeight,\r
-  OUT UINTN     *PixelWidth\r
-  )\r
-/*++\r
-\r
-Routine Description:\r
-\r
-  Convert a *.BMP graphics image to a GOP/UGA blt buffer. If a NULL Blt buffer\r
+/**\r
+  Convert a *.BMP graphics image to a GOP blt buffer. If a NULL Blt buffer\r
   is passed in a GopBlt buffer will be allocated by this routine. If a GopBlt\r
   buffer is passed in it will be used if it is big enough.\r
 \r
-Arguments:\r
-\r
-  BmpImage      - Pointer to BMP file\r
-\r
-  BmpImageSize  - Number of bytes in BmpImage\r
-\r
-  GopBlt        - Buffer containing GOP version of BmpImage.\r
-\r
-  GopBltSize    - Size of GopBlt in bytes.\r
-\r
-  PixelHeight   - Height of GopBlt/BmpImage in pixels\r
+  @param  BmpImage      Pointer to BMP file\r
+  @param  BmpImageSize  Number of bytes in BmpImage\r
+  @param  GopBlt        Buffer containing GOP version of BmpImage.\r
+  @param  GopBltSize    Size of GopBlt in bytes.\r
+  @param  PixelHeight   Height of GopBlt/BmpImage in pixels\r
+  @param  PixelWidth    Width of GopBlt/BmpImage in pixels\r
 \r
-  PixelWidth    - Width of GopBlt/BmpImage in pixels\r
+  @retval EFI_SUCCESS           GopBlt and GopBltSize are returned. \r
+  @retval EFI_UNSUPPORTED       BmpImage is not a valid *.BMP image\r
+  @retval EFI_BUFFER_TOO_SMALL  The passed in GopBlt buffer is not big enough.\r
+                                GopBltSize will contain the required size.\r
+  @retval EFI_OUT_OF_RESOURCES  No enough buffer to allocate.\r
 \r
-\r
-Returns:\r
-\r
-  EFI_SUCCESS           - GopBlt and GopBltSize are returned.\r
-  EFI_UNSUPPORTED       - BmpImage is not a valid *.BMP image\r
-  EFI_BUFFER_TOO_SMALL  - The passed in GopBlt buffer is not big enough.\r
-                          GopBltSize will contain the required size.\r
-  EFI_OUT_OF_RESOURCES  - No enough buffer to allocate\r
-\r
---*/\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+ConvertBmpToGopBlt (\r
+  IN     VOID      *BmpImage,\r
+  IN     UINTN     BmpImageSize,\r
+  IN OUT VOID      **GopBlt,\r
+  IN OUT UINTN     *GopBltSize,\r
+     OUT UINTN     *PixelHeight,\r
+     OUT UINTN     *PixelWidth\r
+  )\r
 {\r
-  UINT8             *Image;\r
-  UINT8             *ImageHeader;\r
-  BMP_IMAGE_HEADER  *BmpHeader;\r
-  BMP_COLOR_MAP     *BmpColorMap;\r
+  UINT8                         *Image;\r
+  UINT8                         *ImageHeader;\r
+  BMP_IMAGE_HEADER              *BmpHeader;\r
+  BMP_COLOR_MAP                 *BmpColorMap;\r
   EFI_GRAPHICS_OUTPUT_BLT_PIXEL *BltBuffer;\r
   EFI_GRAPHICS_OUTPUT_BLT_PIXEL *Blt;\r
-  UINTN             BltBufferSize;\r
-  UINTN             Index;\r
-  UINTN             Height;\r
-  UINTN             Width;\r
-  UINTN             ImageIndex;\r
-  BOOLEAN           IsAllocated;\r
+  UINTN                         BltBufferSize;\r
+  UINTN                         Index;\r
+  UINTN                         Height;\r
+  UINTN                         Width;\r
+  UINTN                         ImageIndex;\r
+  BOOLEAN                       IsAllocated;\r
 \r
   BmpHeader = (BMP_IMAGE_HEADER *) BmpImage;\r
+\r
   if (BmpHeader->CharB != 'B' || BmpHeader->CharM != 'M') {\r
     return EFI_UNSUPPORTED;\r
   }\r
 \r
+  //\r
+  // Doesn't support compress.\r
+  //\r
   if (BmpHeader->CompressionType != 0) {\r
     return EFI_UNSUPPORTED;\r
   }\r
@@ -218,9 +188,15 @@ Returns:
   Image         = ((UINT8 *) BmpImage) + BmpHeader->ImageOffset;\r
   ImageHeader   = Image;\r
 \r
+  //\r
+  // Calculate the BltBuffer needed size.\r
+  //\r
   BltBufferSize = BmpHeader->PixelWidth * BmpHeader->PixelHeight * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL);\r
   IsAllocated   = FALSE;\r
   if (*GopBlt == NULL) {\r
+    //\r
+    // GopBlt is not allocated by caller.\r
+    //\r
     *GopBltSize = BltBufferSize;\r
     *GopBlt     = AllocatePool (*GopBltSize);\r
     IsAllocated = TRUE;\r
@@ -228,6 +204,9 @@ Returns:
       return EFI_OUT_OF_RESOURCES;\r
     }\r
   } else {\r
+    //\r
+    // GopBlt has been allocated by caller.\r
+    //\r
     if (*GopBltSize < BltBufferSize) {\r
       *GopBltSize = BltBufferSize;\r
       return EFI_BUFFER_TOO_SMALL;\r
@@ -247,7 +226,7 @@ Returns:
       switch (BmpHeader->BitPerPixel) {\r
       case 1:\r
         //\r
-        // Convert 1bit BMP to 24-bit color\r
+        // Convert 1-bit (2 colors) BMP to 24-bit color\r
         //\r
         for (Index = 0; Index < 8 && Width < BmpHeader->PixelWidth; Index++) {\r
           Blt->Red    = BmpColorMap[((*Image) >> (7 - Index)) & 0x1].Red;\r
@@ -263,7 +242,7 @@ Returns:
 \r
       case 4:\r
         //\r
-        // Convert BMP Palette to 24-bit color\r
+        // Convert 4-bit (16 colors) BMP Palette to 24-bit color\r
         //\r
         Index       = (*Image) >> 4;\r
         Blt->Red    = BmpColorMap[Index].Red;\r
@@ -281,7 +260,7 @@ Returns:
 \r
       case 8:\r
         //\r
-        // Convert BMP Palette to 24-bit color\r
+        // Convert 8-bit (256 colors) BMP Palette to 24-bit color\r
         //\r
         Blt->Red    = BmpColorMap[*Image].Red;\r
         Blt->Green  = BmpColorMap[*Image].Green;\r
@@ -289,14 +268,20 @@ Returns:
         break;\r
 \r
       case 24:\r
+        //\r
+        // It is 24-bit BMP.\r
+        //\r
         Blt->Blue   = *Image++;\r
         Blt->Green  = *Image++;\r
         Blt->Red    = *Image;\r
         break;\r
 \r
       default:\r
+        //\r
+        // Other bit format BMP is not supported.\r
+        //\r
         if (IsAllocated) {\r
-          gBS->FreePool (*GopBlt);\r
+          FreePool (*GopBlt);\r
           *GopBlt = NULL;\r
         }\r
         return EFI_UNSUPPORTED;\r
@@ -318,29 +303,23 @@ Returns:
 }\r
 \r
 \r
-EFI_STATUS\r
-LockKeyboards (\r
-  IN  CHAR16    *Password\r
-  )\r
-/*++\r
-\r
-Routine Description:\r
-  Use Console Control Protocol to lock the Console In Spliter virtual handle.\r
+/**\r
+  Use Console Control Protocol to lock the Console In Spliter virtual handle. \r
   This is the ConInHandle and ConIn handle in the EFI system table. All key\r
   presses will be ignored until the Password is typed in. The only way to\r
   disable the password is to type it in to a ConIn device.\r
 \r
-Arguments:\r
-  Password - Password used to lock ConIn device\r
+  @param  Password        Password used to lock ConIn device.\r
 \r
+  @retval EFI_SUCCESS     lock the Console In Spliter virtual handle successfully.\r
+  @retval EFI_UNSUPPORTED Password not found.\r
 \r
-Returns:\r
-\r
-  EFI_SUCCESS     - ConsoleControl has been flipped to graphics and logo\r
-                          displayed.\r
-  EFI_UNSUPPORTED - Logo not found\r
-\r
---*/\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+LockKeyboards (\r
+  IN  CHAR16    *Password\r
+  )\r
 {\r
   EFI_STATUS                    Status;\r
   EFI_CONSOLE_CONTROL_PROTOCOL  *ConsoleControl;\r
@@ -355,60 +334,44 @@ Returns:
 }\r
 \r
 \r
+/**\r
+  Use Console Control to turn off UGA based Simple Text Out consoles from going\r
+  to the UGA device. Put up LogoFile on every UGA device that is a console.\r
+\r
+  @param  LogoFile        File name of logo to display on the center of the screen.\r
+\r
+  @retval EFI_SUCCESS     ConsoleControl has been flipped to graphics and logo displayed.\r
+  @retval EFI_UNSUPPORTED Logo not found.\r
+\r
+**/\r
 EFI_STATUS\r
+EFIAPI\r
 EnableQuietBoot (\r
   IN  EFI_GUID  *LogoFile\r
   )\r
-/*++\r
-\r
-Routine Description:\r
+{\r
+  return EnableQuietBootEx (LogoFile, NULL);\r
+}\r
 \r
+/**\r
   Use Console Control to turn off UGA based Simple Text Out consoles from going\r
   to the UGA device. Put up LogoFile on every UGA device that is a console\r
 \r
-Arguments:\r
-\r
-  LogoFile - File name of logo to display on the center of the screen.\r
-\r
+  @param  LogoFile    File name of logo to display on the center of the screen.\r
+  @param  ImageHandle The driver image handle of the caller. The parameter is used to\r
+                      optimize the loading of the logo file so that the FV from which\r
+                      the driver image is loaded will be tried first.\r
 \r
-Returns:\r
-\r
-  EFI_SUCCESS           - ConsoleControl has been flipped to graphics and logo\r
-                          displayed.\r
-  EFI_UNSUPPORTED       - Logo not found\r
-\r
---*/\r
-{\r
-  return EnableQuietBootEx (LogoFile, NULL);\r
-}\r
+  @retval EFI_SUCCESS     ConsoleControl has been flipped to graphics and logo displayed.\r
+  @retval EFI_UNSUPPORTED Logo not found.\r
 \r
+**/\r
 EFI_STATUS\r
+EFIAPI\r
 EnableQuietBootEx (\r
   IN  EFI_GUID    *LogoFile,\r
   IN  EFI_HANDLE  ImageHandle\r
   )\r
-/*++\r
-\r
-Routine Description:\r
-\r
-  Use Console Control to turn off GOP/UGA based Simple Text Out consoles from going\r
-  to the GOP/UGA device. Put up LogoFile on every GOP/UGA device that is a console\r
-\r
-Arguments:\r
-\r
-  LogoFile    - File name of logo to display on the center of the screen.\r
-  ImageHandle - The driver image handle of the caller. The parameter is used to\r
-                optimize the loading of the logo file so that the FV from which\r
-                the driver image is loaded will be tried first.\r
-\r
-\r
-Returns:\r
-\r
-  EFI_SUCCESS           - ConsoleControl has been flipped to graphics and logo\r
-                          displayed.\r
-  EFI_UNSUPPORTED       - Logo not found\r
-\r
---*/\r
 {\r
   EFI_STATUS                    Status;\r
   EFI_CONSOLE_CONTROL_PROTOCOL  *ConsoleControl;\r
@@ -433,7 +396,7 @@ Returns:
   UINT32                        RefreshRate;\r
   EFI_GRAPHICS_OUTPUT_PROTOCOL  *GraphicsOutput;\r
 \r
-  Status = gBS->LocateProtocol (&gEfiConsoleControlProtocolGuid, NULL, (VOID**)&ConsoleControl);\r
+  Status = gBS->LocateProtocol (&gEfiConsoleControlProtocolGuid, NULL, (VOID **) &ConsoleControl);\r
   if (EFI_ERROR (Status)) {\r
     return EFI_UNSUPPORTED;\r
   }\r
@@ -442,21 +405,24 @@ Returns:
   //\r
   // Try to open GOP first\r
   //\r
-  Status = gBS->HandleProtocol (gST->ConsoleOutHandle, &gEfiGraphicsOutputProtocolGuid, (VOID**)&GraphicsOutput);\r
+  Status = gBS->HandleProtocol (gST->ConsoleOutHandle, &gEfiGraphicsOutputProtocolGuid, (VOID **) &GraphicsOutput);\r
   if (EFI_ERROR (Status) && FeaturePcdGet (PcdUgaConsumeSupport)) {\r
     GraphicsOutput = NULL;\r
     //\r
-    // Open GOP failed, try to open UGwhA\r
+    // Open GOP failed, try to open UGA\r
     //\r
-    Status = gBS->HandleProtocol (gST->ConsoleOutHandle, &gEfiUgaDrawProtocolGuid, (VOID**)&UgaDraw);\r
+    Status = gBS->HandleProtocol (gST->ConsoleOutHandle, &gEfiUgaDrawProtocolGuid, (VOID **) &UgaDraw);\r
   }\r
   if (EFI_ERROR (Status)) {\r
     return EFI_UNSUPPORTED;\r
   }\r
 \r
   Badging = NULL;\r
-  Status  = gBS->LocateProtocol (&gEfiOEMBadgingProtocolGuid, NULL, (VOID**)&Badging);\r
+  Status  = gBS->LocateProtocol (&gEfiOEMBadgingProtocolGuid, NULL, (VOID **) &Badging);\r
 \r
+  //\r
+  // Set console control to graphics mode.\r
+  //\r
   Status = ConsoleControl->SetMode (ConsoleControl, EfiConsoleControlScreenGraphics);\r
   if (EFI_ERROR (Status)) {\r
     return EFI_UNSUPPORTED;\r
@@ -470,6 +436,8 @@ Returns:
     if (EFI_ERROR (Status)) {\r
       return EFI_UNSUPPORTED;\r
     }\r
+  } else {\r
+    return EFI_UNSUPPORTED;\r
   }\r
 \r
   Instance = 0;\r
@@ -478,6 +446,9 @@ Returns:
     ImageSize = 0;\r
 \r
     if (Badging != NULL) {\r
+      //\r
+      // Get image from OEMBadging protocol.\r
+      //\r
       Status = Badging->GetImage (\r
                           Badging,\r
                           &Instance,\r
@@ -493,13 +464,16 @@ Returns:
       }\r
 \r
       //\r
-      // Currently only support BMP format\r
+      // Currently only support BMP format.\r
       //\r
       if (Format != EfiBadgingFormatBMP) {\r
-        gBS->FreePool (ImageData);\r
+        SafeFreePool (ImageData);\r
         continue;\r
       }\r
     } else {\r
+      //\r
+      // Get the specified image from FV.\r
+      //\r
       Status = GetGraphicsBitMapFromFVEx (ImageHandle, LogoFile, (VOID **) &ImageData, &ImageSize);\r
       if (EFI_ERROR (Status)) {\r
         return EFI_UNSUPPORTED;\r
@@ -514,13 +488,13 @@ Returns:
     Status = ConvertBmpToGopBlt (\r
               ImageData,\r
               ImageSize,\r
-              (VOID**)&Blt,\r
+              (VOID **) &Blt,\r
               &BltSize,\r
               &Height,\r
               &Width\r
               );\r
     if (EFI_ERROR (Status)) {\r
-      gBS->FreePool (ImageData);\r
+      SafeFreePool (ImageData);\r
       if (Badging == NULL) {\r
         return Status;\r
       } else {\r
@@ -528,6 +502,9 @@ Returns:
       }\r
     }\r
 \r
+    //\r
+    // Caculate the display position according to Attribute.\r
+    //\r
     switch (Attribute) {\r
     case EfiBadgingDisplayAttributeLeftTop:\r
       DestX = CoordinateX;\r
@@ -607,11 +584,13 @@ Returns:
                             Height,\r
                             Width * sizeof (EFI_UGA_PIXEL)\r
                             );\r
+      } else {\r
+      Status = EFI_UNSUPPORTED;\r
       }\r
     }\r
 \r
-    gBS->FreePool (ImageData);\r
-    gBS->FreePool (Blt);\r
+    SafeFreePool (ImageData);\r
+    SafeFreePool (Blt);\r
 \r
     if (Badging == NULL) {\r
       break;\r
@@ -621,28 +600,19 @@ Returns:
   return Status;\r
 }\r
 \r
+/**\r
+  Use Console Control to turn on UGA based Simple Text Out consoles. The UGA \r
+  Simple Text Out screens will now be synced up with all non UGA output devices\r
+\r
+  @retval EFI_SUCCESS          UGA devices are back in text mode and synced up.\r
+  @retval EFI_UNSUPPORTED      Logo not found\r
 \r
+**/\r
 EFI_STATUS\r
+EFIAPI\r
 DisableQuietBoot (\r
   VOID\r
   )\r
-/*++\r
-\r
-Routine Description:\r
-\r
-  Use Console Control to turn on GOP/UGA based Simple Text Out consoles. The GOP/UGA\r
-  Simple Text Out screens will now be synced up with all non GOP/UGA output devices\r
-\r
-Arguments:\r
-\r
-  NONE\r
-\r
-Returns:\r
-\r
-  EFI_SUCCESS           - GOP/UGA devices are back in text mode and synced up.\r
-  EFI_UNSUPPORTED       - Logo not found\r
-\r
---*/\r
 {\r
   EFI_STATUS                    Status;\r
   EFI_CONSOLE_CONTROL_PROTOCOL  *ConsoleControl;\r
@@ -652,11 +622,31 @@ Returns:
     return EFI_UNSUPPORTED;\r
   }\r
 \r
+  //\r
+  // Set console control to text mode.\r
+  //\r
   return ConsoleControl->SetMode (ConsoleControl, EfiConsoleControlScreenText);\r
 }\r
 \r
+/**\r
+  Internal display string worker function.\r
+\r
+  @param GraphicsOutput   Graphics output protocol interface.\r
+  @param UgaDraw          UGA draw protocol interface.\r
+  @param Sto              Simple text out protocol interface.\r
+  @param X                X coordinate to start printing.\r
+  @param Y                Y coordinate to start printing.\r
+  @param Foreground       Foreground color.\r
+  @param Background       Background color.\r
+  @param fmt              Format string.\r
+  @param args             Print arguments.\r
+\r
+  @return  Number of Characters printed. Zero means no any character \r
+           displayed successfully.\r
+\r
+**/\r
 UINTN\r
-_IPrint (\r
+Print (\r
   IN EFI_GRAPHICS_OUTPUT_PROTOCOL     *GraphicsOutput,\r
   IN EFI_UGA_DRAW_PROTOCOL            *UgaDraw,\r
   IN EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL  *Sto,\r
@@ -667,38 +657,6 @@ _IPrint (
   IN CHAR16                           *fmt,\r
   IN VA_LIST                          args\r
   )\r
-/*++\r
-\r
-Routine Description:\r
-\r
-  Display string worker for: Print, PrintAt, IPrint, IPrintAt\r
-\r
-Arguments:\r
-\r
-  GraphicsOutput  - Graphics output protocol interface\r
-\r
-  UgaDraw         - UGA draw protocol interface\r
-\r
-  Sto             - Simple text out protocol interface\r
-\r
-  X               - X coordinate to start printing\r
-\r
-  Y               - Y coordinate to start printing\r
-\r
-  Foreground      - Foreground color\r
-\r
-  Background      - Background color\r
-\r
-  fmt             - Format string\r
-\r
-  args            - Print arguments\r
-\r
-Returns:\r
-\r
-  EFI_SUCCESS             -  success\r
-  EFI_OUT_OF_RESOURCES    -  out of resources\r
-\r
---*/\r
 {\r
   VOID                           *Buffer;\r
   EFI_STATUS                     Status;\r
@@ -713,19 +671,23 @@ Returns:
   EFI_HII_FONT_PROTOCOL          *HiiFont;\r
   EFI_IMAGE_OUTPUT               *Blt;\r
   EFI_FONT_DISPLAY_INFO          *FontInfo;\r
+  EFI_HII_ROW_INFO               *RowInfoArray;\r
+  UINTN                          RowInfoArraySize;\r
+  UINTN                          PrintNum;     \r
 \r
   //\r
   // For now, allocate an arbitrarily long buffer\r
   //\r
   Buffer = AllocateZeroPool (0x10000);\r
   if (Buffer == NULL) {\r
-    return EFI_OUT_OF_RESOURCES;\r
+    return 0;\r
   }\r
 \r
   HorizontalResolution  = 0;\r
   VerticalResolution    = 0;\r
   Blt                   = NULL;\r
   FontInfo              = NULL;\r
+  PrintNum              = 0;\r
 \r
   if (GraphicsOutput != NULL) {\r
     HorizontalResolution = GraphicsOutput->Mode->Info->HorizontalResolution;\r
@@ -739,13 +701,12 @@ Returns:
 \r
   ASSERT ((HorizontalResolution != 0) && (VerticalResolution !=0));\r
 \r
-  ASSERT (GraphicsOutput != NULL);\r
   Status = gBS->LocateProtocol (&gEfiHiiFontProtocolGuid, NULL, (VOID **) &HiiFont);\r
   if (EFI_ERROR (Status)) {\r
     goto Error;\r
   }\r
 \r
-  UnicodeVSPrint (Buffer, 0x10000, fmt, args);\r
+  PrintNum = UnicodeVSPrint (Buffer, 0x10000, fmt, args);\r
 \r
   UnicodeWeight = (CHAR16 *) Buffer;\r
 \r
@@ -759,7 +720,6 @@ Returns:
 \r
   BufferLen = StrLen (Buffer);\r
 \r
-\r
   LineBufferLen = sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL) * HorizontalResolution * EFI_GLYPH_HEIGHT;\r
   if (EFI_GLYPH_WIDTH * EFI_GLYPH_HEIGHT * sizeof(EFI_GRAPHICS_OUTPUT_BLT_PIXEL) * BufferLen > LineBufferLen) {\r
      Status = EFI_INVALID_PARAMETER;\r
@@ -774,7 +734,6 @@ Returns:
 \r
   Blt->Width        = (UINT16) (HorizontalResolution);\r
   Blt->Height       = (UINT16) (VerticalResolution);\r
-  Blt->Image.Screen = GraphicsOutput;\r
 \r
   FontInfo = (EFI_FONT_DISPLAY_INFO *) AllocateZeroPool (sizeof (EFI_FONT_DISPLAY_INFO));\r
   if (FontInfo == NULL) {\r
@@ -800,28 +759,106 @@ Returns:
       );\r
   }\r
 \r
-  Status = HiiFont->StringToImage (\r
-                       HiiFont,\r
-                       EFI_HII_IGNORE_IF_NO_GLYPH | EFI_HII_DIRECT_TO_SCREEN,\r
-                       Buffer,\r
-                       FontInfo,\r
-                       &Blt,\r
-                       X,\r
-                       Y,\r
-                       NULL,\r
-                       NULL,\r
-                       NULL\r
-                       );\r
+  if (GraphicsOutput != NULL) {\r
+    Blt->Image.Screen = GraphicsOutput;\r
+    \r
+    Status = HiiFont->StringToImage (\r
+                         HiiFont,\r
+                         EFI_HII_IGNORE_IF_NO_GLYPH | EFI_HII_DIRECT_TO_SCREEN,\r
+                         Buffer,\r
+                         FontInfo,\r
+                         &Blt,\r
+                         X,\r
+                         Y,\r
+                         NULL,\r
+                         NULL,\r
+                         NULL\r
+                         );\r
+\r
+  } else if (FeaturePcdGet (PcdUgaConsumeSupport)) {\r
+    ASSERT (UgaDraw!= NULL);\r
+\r
+    Blt->Image.Bitmap = AllocateZeroPool (Blt->Width * Blt->Height * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL));\r
+    if (Blt->Image.Bitmap == NULL) {\r
+      SafeFreePool (Blt);\r
+      SafeFreePool (Buffer);\r
+      return EFI_OUT_OF_RESOURCES;\r
+    }\r
+\r
+    RowInfoArray = NULL;\r
+    //\r
+    //  StringToImage only support blt'ing image to device using GOP protocol. If GOP is not supported in this platform,\r
+    //  we ask StringToImage to print the string to blt buffer, then blt to device using UgaDraw.\r
+    //\r
+    Status = HiiFont->StringToImage (\r
+                         HiiFont,\r
+                         EFI_HII_IGNORE_IF_NO_GLYPH,\r
+                         Buffer,\r
+                         FontInfo,\r
+                         &Blt,\r
+                         X,\r
+                         Y,\r
+                         &RowInfoArray,\r
+                         &RowInfoArraySize,\r
+                         NULL\r
+                         );\r
+\r
+    if (!EFI_ERROR (Status)) {\r
+      //\r
+      // Line breaks are handled by caller of DrawUnicodeWeightAtCursorN, so the updated parameter RowInfoArraySize by StringToImage will\r
+      // always be 1 or 0 (if there is no valid Unicode Char can be printed). ASSERT here to make sure.\r
+      //\r
+      ASSERT (RowInfoArraySize <= 1);\r
+\r
+      Status = UgaDraw->Blt (\r
+                          UgaDraw,\r
+                          (EFI_UGA_PIXEL *) Blt->Image.Bitmap,\r
+                          EfiUgaBltBufferToVideo,\r
+                          X,\r
+                          Y,\r
+                          X,\r
+                          Y,\r
+                          RowInfoArray[0].LineWidth,\r
+                          RowInfoArray[0].LineHeight,\r
+                          Blt->Width * sizeof (EFI_GRAPHICS_OUTPUT_BLT_PIXEL)\r
+                          );\r
+    }\r
 \r
+    SafeFreePool (RowInfoArray);\r
+    SafeFreePool (Blt->Image.Bitmap);\r
+  } else {\r
+    Status = EFI_UNSUPPORTED;\r
+  }\r
 \r
 Error:\r
   SafeFreePool (Blt);\r
   SafeFreePool (FontInfo);\r
-  gBS->FreePool (Buffer);\r
-  return Status;\r
+  FreePool (Buffer);\r
+\r
+  if (EFI_ERROR (Status)) {\r
+    return PrintNum;\r
+  } else {\r
+    return 0;\r
+  }\r
 }\r
 \r
+/**\r
+  Print Unicode string to graphics screen at the given X,Y coordinates of the graphics screen.\r
+  see definition of Print to find rules for constructing Fmt.\r
+\r
+  @param  X            Row to start printing at.\r
+  @param  Y            Column to start printing at.\r
+  @param  ForeGround   Foreground color.\r
+  @param  BackGround   background color.\r
+  @param  Fmt          Print format sting. See definition of Print.\r
+  @param  ...          Argumnet stream defined by Fmt string.\r
+\r
+  @return  Number of Characters printed. Zero means no any character \r
+           displayed successfully.\r
+\r
+**/\r
 UINTN\r
+EFIAPI\r
 PrintXY (\r
   IN UINTN                            X,\r
   IN UINTN                            Y,\r
@@ -830,38 +867,13 @@ PrintXY (
   IN CHAR16                           *Fmt,\r
   ...\r
   )\r
-/*++\r
-\r
-Routine Description:\r
-\r
-    Prints a formatted unicode string to the default console\r
-\r
-Arguments:\r
-\r
-    X           - X coordinate to start printing\r
-\r
-    Y           - Y coordinate to start printing\r
-\r
-    ForeGround  - Foreground color\r
-\r
-    BackGround  - Background color\r
-\r
-    Fmt         - Format string\r
-\r
-    ...         - Print arguments\r
-\r
-Returns:\r
-\r
-    Length of string printed to the console\r
-\r
---*/\r
 {\r
-  EFI_HANDLE                    Handle;\r
-  EFI_GRAPHICS_OUTPUT_PROTOCOL  *GraphicsOutput;\r
-  EFI_UGA_DRAW_PROTOCOL         *UgaDraw;\r
+  EFI_HANDLE                       Handle;\r
+  EFI_GRAPHICS_OUTPUT_PROTOCOL     *GraphicsOutput;\r
+  EFI_UGA_DRAW_PROTOCOL            *UgaDraw;\r
   EFI_SIMPLE_TEXT_OUTPUT_PROTOCOL  *Sto;\r
-  EFI_STATUS                    Status;\r
-  VA_LIST                       Args;\r
+  EFI_STATUS                       Status;\r
+  VA_LIST                          Args;\r
 \r
   VA_START (Args, Fmt);\r
 \r
@@ -870,33 +882,36 @@ Returns:
   Status = gBS->HandleProtocol (\r
                   Handle,\r
                   &gEfiGraphicsOutputProtocolGuid,\r
-                  (VOID**)&GraphicsOutput\r
+                  (VOID **) &GraphicsOutput\r
                   );\r
 \r
   UgaDraw = NULL;\r
   if (EFI_ERROR (Status) && FeaturePcdGet (PcdUgaConsumeSupport)) {\r
+    //\r
+    // If no GOP available, try to open UGA Draw protocol if supported.\r
+    //\r
     GraphicsOutput = NULL;\r
 \r
     Status = gBS->HandleProtocol (\r
                     Handle,\r
                     &gEfiUgaDrawProtocolGuid,\r
-                    (VOID**)&UgaDraw\r
+                    (VOID **) &UgaDraw\r
                     );\r
   }\r
   if (EFI_ERROR (Status)) {\r
-    return Status;\r
+    return 0;\r
   }\r
 \r
   Status = gBS->HandleProtocol (\r
                   Handle,\r
                   &gEfiSimpleTextOutProtocolGuid,\r
-                  (VOID**)&Sto\r
+                  (VOID **) &Sto\r
                   );\r
 \r
   if (EFI_ERROR (Status)) {\r
-    return Status;\r
+    return 0;\r
   }\r
 \r
-  return _IPrint (GraphicsOutput, UgaDraw, Sto, X, Y, ForeGround, BackGround, Fmt, Args);\r
+  return Print (GraphicsOutput, UgaDraw, Sto, X, Y, ForeGround, BackGround, Fmt, Args);\r
 }\r
 \r