KVM WORKING! ! !
diff --git a/src/ast_jpeg_decoder_test.cpp b/src/ast_jpeg_decoder_test.cpp
new file mode 100644
index 0000000..bc6857e
--- /dev/null
+++ b/src/ast_jpeg_decoder_test.cpp
@@ -0,0 +1,156 @@
+#include "ast_jpeg_decoder.hpp"
+#include "gtest/gtest.h"
+#include "gmock/gmock.h"
+
+#ifdef BUILD_CIMG
+#define cimg_display 0
+#include <CImg.h>
+#endif
+
+using namespace testing;
+MATCHER_P2(IsBetween, a, b, std::string(negation ? "isn't" : "is") +
+ " between " + PrintToString(a) + " and " +
+ PrintToString(b)) {
+ return a <= arg && arg <= b;
+};
+
+TEST(AstJpegDecoder, AllBlue) {
+ AstVideo::RawVideoBuffer out;
+
+ // This binary blog was created on the aspeed hardware using a blue screen
+ // consisting of the color 0x8EFFFA in a web browser window
+ FILE *fp = fopen("test_resources/aspeedbluescreen.bin", "rb");
+ EXPECT_NE(fp, nullptr);
+ size_t bufferlen = fread(out.buffer.data(), sizeof(char),
+ out.buffer.size() * sizeof(long), fp);
+ fclose(fp);
+
+ ASSERT_GT(bufferlen, 0);
+
+ out.y_selector = 0;
+ out.uv_selector = 0;
+ out.mode = AstVideo::YuvMode::YUV444;
+ out.width = 800;
+ out.height = 600;
+
+ AstVideo::AstJpegDecoder d;
+ d.decode(out.buffer, out.width, out.height, out.mode, out.y_selector,
+ out.uv_selector);
+
+ int tolerance = 16;
+
+ // All pixels should be blue (0x8EFFFA) to within a tolerance (due to jpeg
+ // compression artifacts and quanitization)
+ for (int i = 0; i < out.width * out.height; i++) {
+ AstVideo::RGB &pixel = d.OutBuffer[i];
+ EXPECT_GT(pixel.R, 0x8E - tolerance);
+ EXPECT_LT(pixel.R, 0x8E + tolerance);
+ EXPECT_GT(pixel.G, 0xFF - tolerance);
+ EXPECT_LT(pixel.G, 0xFF + tolerance);
+ EXPECT_GT(pixel.B, 0xF1 - tolerance);
+ EXPECT_LT(pixel.B, 0xF1 + tolerance);
+ }
+}
+
+TEST(AstJpegDecoder, AllBlack) {
+ AstVideo::RawVideoBuffer out;
+
+ // This binary blog was created on the aspeed hardware using a black screen
+ FILE *fp = fopen("test_resources/aspeedblackscreen.bin", "rb");
+ EXPECT_NE(fp, nullptr);
+ size_t bufferlen = fread(out.buffer.data(), sizeof(char),
+ out.buffer.size() * sizeof(long), fp);
+ fclose(fp);
+
+ ASSERT_GT(bufferlen, 0);
+
+ out.y_selector = 0;
+ out.uv_selector = 0;
+ out.mode = AstVideo::YuvMode::YUV444;
+ out.width = 800;
+ out.height = 600;
+
+ AstVideo::AstJpegDecoder d;
+ d.decode(out.buffer, out.width, out.height, out.mode, out.y_selector,
+ out.uv_selector);
+
+ // All pixels should be blue (0x8EFFFA) to within a tolerance (due to jpeg
+ // compression artifacts and quanitization)
+ for (int x = 0; x < out.width; x++) {
+ for (int y = 0; y < out.height; y++) {
+ AstVideo::RGB pixel = d.OutBuffer[x + (y * out.width)];
+ ASSERT_EQ(pixel.R, 0x00) << "X:" << x << " Y: " << y;
+ ASSERT_EQ(pixel.G, 0x00) << "X:" << x << " Y: " << y;
+ ASSERT_EQ(pixel.B, 0x00) << "X:" << x << " Y: " << y;
+ }
+ }
+}
+
+TEST(AstJpegDecoder, TestColors) {
+ AstVideo::RawVideoBuffer out;
+
+ // This binary blog was created on the aspeed hardware using a blue screen
+ // consisting of the color 0x8EFFFA in a web browser window
+ FILE *fp = fopen("test_resources/ubuntu_444_800x600_0chrom_0lum.bin", "rb");
+ EXPECT_NE(fp, nullptr);
+ size_t bufferlen = fread(out.buffer.data(), sizeof(char),
+ out.buffer.size() * sizeof(long), fp);
+ fclose(fp);
+
+ ASSERT_GT(bufferlen, 0);
+
+ out.y_selector = 0;
+ out.uv_selector = 0;
+ out.mode = AstVideo::YuvMode::YUV444;
+ out.width = 800;
+ out.height = 600;
+
+ AstVideo::AstJpegDecoder d;
+ d.decode(out.buffer, out.width, out.height, out.mode, out.y_selector,
+ out.uv_selector);
+
+ int tolerance = 16;
+
+ for (int i = 0; i < out.width * out.height; i++) {
+ AstVideo::RGB &pixel = d.OutBuffer[i];
+ EXPECT_GT(pixel.R, 0x8E - tolerance);
+ EXPECT_LT(pixel.R, 0x8E + tolerance);
+ EXPECT_GT(pixel.G, 0xFF - tolerance);
+ EXPECT_LT(pixel.G, 0xFF + tolerance);
+ EXPECT_GT(pixel.B, 0xF1 - tolerance);
+ EXPECT_LT(pixel.B, 0xF1 + tolerance);
+ }
+}
+// Tests the buffers around the screen aren't written to
+TEST(AstJpegDecoder, BufferLimits) {
+ AstVideo::RawVideoBuffer out;
+
+ // This binary blog was created on the aspeed hardware using a black screen
+ FILE *fp = fopen("test_resources/aspeedblackscreen.bin", "rb");
+ EXPECT_NE(fp, nullptr);
+ size_t bufferlen = fread(out.buffer.data(), sizeof(char),
+ out.buffer.size() * sizeof(long), fp);
+ fclose(fp);
+
+ ASSERT_GT(bufferlen, 0);
+
+ out.y_selector = 0;
+ out.uv_selector = 0;
+ out.mode = AstVideo::YuvMode::YUV444;
+ out.width = 800;
+ out.height = 600;
+
+ AstVideo::AstJpegDecoder d;
+ d.decode(out.buffer, out.width, out.height, out.mode, out.y_selector,
+ out.uv_selector);
+ // Reserved pixel should be default value
+ for (auto &pixel : d.OutBuffer) {
+ EXPECT_EQ(pixel.Reserved, 0xAA);
+ }
+ // All pixels beyond the buffer should be zero
+ for (int i = out.width * out.height; i < d.OutBuffer.size(); i++) {
+ EXPECT_EQ(d.OutBuffer[i].R, 0x00) << "index:" << i;
+ EXPECT_EQ(d.OutBuffer[i].B, 0x00) << "index:" << i;
+ EXPECT_EQ(d.OutBuffer[i].G, 0x00) << "index:" << i;
+ }
+}
\ No newline at end of file