From 1d343a079a166c1f9b9c7c188b087314ab35a23e Mon Sep 17 00:00:00 2001
From: Michael Mandl <mandlm@molez.net>
Date: Sun, 1 Nov 2015 15:00:32 +0100
Subject: [PATCH] Moved MNIST sample to QImage conversion to MNIST loader

---
 gui/NeuroUI/mnistloader.h | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/gui/NeuroUI/mnistloader.h b/gui/NeuroUI/mnistloader.h
index 6b1b57d..3e0f46f 100644
--- a/gui/NeuroUI/mnistloader.h
+++ b/gui/NeuroUI/mnistloader.h
@@ -6,6 +6,8 @@
 #include <memory>
 #include <inttypes.h>
 
+#include <QImage>
+
 class MnistLoader
 {
 private:
@@ -21,6 +23,11 @@ public:
     public:
         uint8_t label;
         uint8_t data[SAMPLE_WIDTH * SAMPLE_HEIGHT];
+
+        QImage toQImage() const
+        {
+            return QImage(data, SAMPLE_WIDTH, SAMPLE_HEIGHT, QImage::Format_Grayscale8);
+        }
     };
 
     using MnistSample = Sample<SampleWidth, SampleHeight>;