add gradients test

2 files changed, 204 insertions, 0 deletions

diff --git a/UefiMonitorTest/tests/Gradients.c b/UefiMonitorTest/tests/Gradients.c
new file mode 100644
index 0000000..03af391
--- /dev/null
+++ b/UefiMonitorTest/tests/Gradients.c
@@ -0,0 +1,171 @@
+#include <Library/DebugLib.h>
+#include <Library/HiiLib.h>
+#include <Library/MemoryAllocationLib.h>
+
+#include "Gradients.h"
+
+enum UMT_GRADIENTS_PARAM {
+  UMT_GRADIENTS_PARAM_COLOR = 0,
+  UMT_GRADIENTS_PARAM_STEP
+};
+
+enum UMT_GRADIENTS_STEPS {
+  UMT_GRADIENTS_STEP_8 = 0,
+  UMT_GRADIENTS_STEP_16,
+  UMT_GRADIENTS_STEP_32,
+  UMT_GRADIENTS_STEP_64,
+  UMT_GRADIENTS_STEP_128,
+  UMT_GRADIENTS_STEP_256,
+  UMT_GRADIENTS_STEP_END
+};
+
+STATIC enum UMT_COLORS CurrentColor           = 1;
+STATIC enum UMT_GRADIENTS_STEPS CurrentStep   = UMT_GRADIENTS_STEP_8;
+STATIC enum UMT_GRADIENTS_PARAM CurrentParam  = UMT_GRADIENTS_PARAM_COLOR;
+
+CONST UINT32 mGradientsSteps[UMT_GRADIENTS_STEP_END] = {
+  8, 16, 32, 64, 128, 256
+};
+
+VOID
+GradientsTestInit (
+  IN UMT_CONTEXT *Ctx
+  )
+{
+  UINT32 Index;
+  GRAPHICS_PIXEL_COLOR ColorOutput = { 0x00 };
+  GRAPHICS_PIXEL_COLOR ColorInc    = { 0x00 };
+  GRAPHICS_CONTEXT *Graphics = Ctx->Graphics;
+
+  ColorInc.Red   = ((gUmtColors[CurrentColor].Color.Red) / (mGradientsSteps[CurrentStep]-1));
+  ColorInc.Green = ((gUmtColors[CurrentColor].Color.Green) / (mGradientsSteps[CurrentStep]-1));
+  ColorInc.Blue  = ((gUmtColors[CurrentColor].Color.Blue) / (mGradientsSteps[CurrentStep]-1));
+
+  for (Index = 0; Index < mGradientsSteps[CurrentStep]; Index++)
+  {
+    PutRect (Graphics, (Graphics->Width * Index) / mGradientsSteps[CurrentStep],
+                        0,
+                        (Graphics->Width * (Index + 1)) / mGradientsSteps[CurrentStep],
+                        Graphics->Height,
+                        &ColorOutput);
+
+    ColorOutput.Red   += ColorInc.Red;
+    ColorOutput.Green += ColorInc.Green;
+    ColorOutput.Blue  += ColorInc.Blue;
+  }
+
+  if (Ctx->ShowTip) {
+    GradientsTestTip (Ctx);
+  }
+}
+
+VOID
+GradientsTestDoit (
+  IN UMT_CONTEXT *Ctx
+  )
+{
+
+}
+
+VOID
+GradientsTestTip (
+  IN UMT_CONTEXT *Ctx
+  )
+{
+  GRAPHICS_CONTEXT *Graphics;
+  CHAR16           *ColorMsg;
+
+  Graphics = Ctx->Graphics;
+
+  if (Ctx->ShowTip == FALSE) {
+    // Restore
+    GradientsTestInit (Ctx);
+    return;
+  }
+
+  ColorMsg = HiiGetString (gUmtHiiHandle, gUmtColors[CurrentColor].StringId, NULL);
+
+  DrawRectWithBorder (Graphics,
+                      15,
+                      Graphics->Height - 15 - 174,
+                      510, Graphics->Height - 15,
+                      3,
+                      &gUmtColors[UMT_COLOR_WHITE].Color,
+                      &gUmtColors[UMT_COLOR_NAVY].Color);
+
+  DrawHiiStringF (Graphics,
+                  25,
+                  Graphics->Height - 15 - 164,
+                  &gUmtColors[UMT_COLOR_NAVY].Color,
+                  STRING_TOKEN (STR_GRADIENTS_TITLE), gUmtHiiHandle);
+
+  DrawHiiStringF (Graphics,
+                  25,
+                  Graphics->Height - 15 - 144,
+                  &gUmtColors[UMT_COLOR_BLACK].Color,
+                  STRING_TOKEN (STR_GRADIENTS_MSG), gUmtHiiHandle,
+                  (CurrentParam == UMT_GRADIENTS_PARAM_COLOR) ? L'*' : L' ',
+                  ColorMsg,
+                  (CurrentParam == UMT_GRADIENTS_PARAM_STEP) ? L'*' : L' ',
+                  mGradientsSteps[CurrentStep]);
+
+  FreePool (ColorMsg);
+}
+
+VOID
+GradientsTestChangeValue (
+  IN  UMT_CONTEXT *Ctx,
+  IN  INT8        ValueStep
+  )
+{
+  switch (CurrentParam) {
+  case UMT_GRADIENTS_PARAM_COLOR:
+    if (CurrentColor == UMT_COLOR_WHITE && ValueStep < 0) {
+      CurrentColor = UMT_COLOR_MAGENTA;
+    } else {
+      CurrentColor += ValueStep;
+      if (CurrentColor > UMT_COLOR_MAGENTA) {
+        CurrentColor = UMT_COLOR_WHITE;
+      }
+    }
+    break;
+
+  case UMT_GRADIENTS_PARAM_STEP:
+    if (CurrentStep == 0 && ValueStep < 0) {
+      return;
+    } else {
+      CurrentStep += ValueStep;
+      if (CurrentStep > UMT_GRADIENTS_STEP_256)
+      {
+        CurrentStep = UMT_GRADIENTS_STEP_256;
+        return;
+      }
+    }
+    break;
+
+  default:
+    ASSERT(FALSE);
+    return;
+  }
+
+  GradientsTestInit (Ctx);
+}
+
+VOID
+GradientsTestChangeParam (
+  IN  UMT_CONTEXT *Ctx,
+  IN  INT8        ParamStep
+  )
+{
+  if (CurrentParam == 0 && ParamStep < 0) {
+    CurrentParam = UMT_GRADIENTS_PARAM_STEP;
+  } else {
+    CurrentParam += ParamStep;
+    if (CurrentParam > UMT_GRADIENTS_PARAM_STEP)
+    {
+      CurrentParam = UMT_GRADIENTS_PARAM_COLOR;
+    }
+  }
+
+  GradientsTestInit (Ctx);
+}
diff --git a/UefiMonitorTest/tests/Gradients.h b/UefiMonitorTest/tests/Gradients.h
new file mode 100644
index 0000000..7c6141b
--- /dev/null
+++ b/UefiMonitorTest/tests/Gradients.h
@@ -0,0 +1,33 @@
+#ifndef UMT_GRADIENTS_TEST_H
+#define UMT_GRADIENTS_TEST_H
+
+#include "UefiMonitorTest.h"
+
+VOID
+GradientsTestInit (
+  IN UMT_CONTEXT *Ctx
+  );
+
+VOID
+GradientsTestDoit (
+  IN UMT_CONTEXT *Ctx
+  );
+
+VOID
+GradientsTestTip (
+  IN UMT_CONTEXT *Ctx
+  );
+
+VOID
+GradientsTestChangeParam (
+  IN  UMT_CONTEXT *Ctx,
+  IN  INT8        ParamStep
+  );
+
+VOID
+GradientsTestChangeValue (
+  IN  UMT_CONTEXT *Ctx,
+  IN  INT8        ValueStep
+  );
+
+#endif /* UMT_GRADIENTS_TEST_H */