Move the engine->string function into ImageItem.

2 files changed, 16 insertions, 0 deletions

diff --git a/tests/arthur/common/baselineprotocol.cpp b/tests/arthur/common/baselineprotocol.cpp
index a6306b9..881c9cc 100644
--- a/tests/arthur/common/baselineprotocol.cpp
+++ b/tests/arthur/common/baselineprotocol.cpp
@@ -44,6 +44,21 @@ quint64 ImageItem::computeChecksum(const QImage &image)
     return qChecksum((const char *)image.constScanLine(50), image.bytesPerLine());
 }
 
+QString ImageItem::engineAsString()
+{
+    switch (engine) {
+    case Raster:
+        return QLatin1String("Raster");
+        break;
+    case OpenGL:
+        return QLatin1String("OpenGL");
+        break;
+    default:
+        break;
+    }
+    return QLatin1String("Unknown");
+}
+
 QDataStream & operator<< (QDataStream &stream, const ImageItem &ii)
 {
     stream << ii.scriptName << ii.scriptChecksum << quint8(ii.status) << quint8(ii.renderFormat)
diff --git a/tests/arthur/common/baselineprotocol.h b/tests/arthur/common/baselineprotocol.h
index 749b623..70cb227 100644
--- a/tests/arthur/common/baselineprotocol.h
+++ b/tests/arthur/common/baselineprotocol.h
@@ -31,6 +31,7 @@ public:
     {}
     ImageItem &operator=(const ImageItem &other);
     static quint64 computeChecksum(const QImage& image);
+    QString engineAsString();
 
     enum ItemStatus {
         Ok = 0,