1 files changed, 92 insertions, 0 deletions

diff --git a/UefiMonitorTest/tests/ChessBoard.c b/UefiMonitorTest/tests/ChessBoard.c
index 6b90067..7b3b976 100644
--- a/UefiMonitorTest/tests/ChessBoard.c
+++ b/UefiMonitorTest/tests/ChessBoard.c
@@ -1,11 +1,48 @@
+#include <Library/DebugLib.h>
+#include <Library/HiiLib.h>
+#include <Library/MemoryAllocationLib.h>
+
 #include "tests/ChessBoard.h"
 
+STATIC UINT32 CurrentSideLength = 1;
+
 VOID
 ChessBoardTestInit (
   IN UMT_CONTEXT *Ctx
   )
 {
+  UINT32 I;
+  UINT32 J;
+  enum UMT_COLORS ColorSquare;
+  enum UMT_COLORS FirstColorSquare;
+  enum UMT_COLORS TempColor;
+  GRAPHICS_CONTEXT *Graphics = Ctx->Graphics;
+
+  FirstColorSquare = UMT_COLOR_WHITE;
+  ColorSquare      = UMT_COLOR_BLACK;
+
+  for (J = 0; J < Graphics->Height; J += CurrentSideLength)
+  {
+    TempColor = FirstColorSquare;
+    FirstColorSquare = ColorSquare;
+    ColorSquare = TempColor;
 
+    for (I = 0; I < Graphics->Width; I += CurrentSideLength)
+    {
+      PutRect (Graphics,
+               I,
+               J,
+               I + CurrentSideLength,
+               J + CurrentSideLength,
+               &gUmtColors[ColorSquare].Color);
+
+      ColorSquare = ((ColorSquare == UMT_COLOR_WHITE) ? UMT_COLOR_BLACK : UMT_COLOR_WHITE);
+    }
+  }
+
+  if (Ctx->ShowTip) {
+    ChessBoardTestTip (Ctx);
+  }
 }
 
 VOID
@@ -21,7 +58,36 @@ ChessBoardTestTip (
   IN UMT_CONTEXT *Ctx
   )
 {
+  GRAPHICS_CONTEXT *Graphics;
+
+  Graphics = Ctx->Graphics;
 
+  if (Ctx->ShowTip == FALSE) {
+    // Restore
+    ChessBoardTestInit (Ctx);
+    return;
+  }
+
+  DrawRectWithBorder (Graphics,
+                      15,
+                      Graphics->Height - 15 - 124,
+                      430, Graphics->Height - 15,
+                      3,
+                      &gUmtColors[UMT_COLOR_WHITE].Color,
+                      &gUmtColors[UMT_COLOR_NAVY].Color);
+
+  DrawHiiStringF (Graphics,
+                  25,
+                  Graphics->Height - 15 - 114,
+                  &gUmtColors[UMT_COLOR_NAVY].Color,
+                  STRING_TOKEN (STR_CHESSBOARD_TITLE), gUmtHiiHandle);
+
+  DrawHiiStringF (Graphics,
+                  25,
+                  Graphics->Height - 15 - 94,
+                  &gUmtColors[UMT_COLOR_BLACK].Color,
+                  STRING_TOKEN (STR_CHESSBOARD_MSG), gUmtHiiHandle,
+                  CurrentSideLength);
 }
 
 VOID
@@ -39,5 +105,31 @@ ChessBoardTestChangeValue (
   IN  INT8        ValueStep
   )
 {
+  UINT32 ValueOut;
+  UINT32 Height;
+  UINT32 Width;
+
+  Height   = Ctx->Graphics->Height;
+  Width    = Ctx->Graphics->Width;
+  ValueOut = CurrentSideLength + ValueStep;
+
+  if (ValueOut == 0) {
+    return;
+  }
+
+  while ((((Width / ValueOut) * ValueOut != Width) ||
+         ((Height / ValueOut) * ValueOut != Height)) &&
+         ((ValueOut < Height) && (ValueOut < Width)) &&
+         (ValueOut > 1))
+  {
+    ValueOut += ValueStep;
+  }
+
+  if ((ValueOut >= Height) || (ValueOut >= Width))
+  {
+    return;
+  }
 
+  CurrentSideLength = ValueOut;
+  ChessBoardTestInit (Ctx);
 }