HAWD::Formatter to print dataset

1 files changed, 52 insertions, 0 deletions

diff --git a/tests/hawd/formatter.cpp b/tests/hawd/formatter.cpp
new file mode 100644
index 0000000..9328d44
--- /dev/null
+++ b/tests/hawd/formatter.cpp
@@ -0,0 +1,52 @@
+/*
+ * Copyright (C) 2015 Christian Mollekopf <mollekopf@kolabsys.com>
+ *
+ *   This program is free software; you can redistribute it and/or modify
+ *   it under the terms of the GNU General Public License as published by
+ *   the Free Software Foundation; either version 2 of the License, or
+ *   (at your option) any later version.
+ *
+ *   This program is distributed in the hope that it will be useful,
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *   GNU General Public License for more details.
+ *
+ *   You should have received a copy of the GNU General Public License
+ *   along with this program; if not, write to the
+ *   Free Software Foundation, Inc.,
+ *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
+ */
+
+#include "formatter.h"
+
+#include <QDir>
+#include <QObject>
+
+#include <iostream>
+
+namespace HAWD
+{
+
+void Formatter::print(const QString &datasetName, const QStringList &cols, State &state)
+{
+    QDir project(state.projectPath());
+    Dataset dataset(datasetName, state);
+
+    if (!dataset.isValid()) {
+        std::cout << QObject::tr("The dataset %1 could not be loaded; try checking it with the check command").arg(datasetName).toStdString() << std::endl;
+        return;
+    }
+    print(dataset);
+}
+
+void Formatter::print(Dataset &dataset, const QStringList &cols)
+{
+    QStringList cols;
+    std::cout << dataset.tableHeaders(cols).toStdString() << std::endl;
+    dataset.eachRow(
+        [cols](const Dataset::Row &row) {
+            std::cout << row.toString(cols).toStdString() << std::endl;
+        });
+}
+
+}