Move document printing to DiagramDocument::print

1 files changed, 17 insertions, 0 deletions

diff --git a/src/diagramdocument.cpp b/src/diagramdocument.cpp
index af2f4ec..a697bc7 100644
--- a/src/diagramdocument.cpp
+++ b/src/diagramdocument.cpp
@@ -30,6 +30,8 @@
 #include <QTextStream>
 #include <QStyle>
 #include <QSet>
+#include <QPainter>
+#include <QPrinter>
 #include "diagram/linelayouter.h"
 
 using namespace std;
@@ -468,3 +470,18 @@ DiagramDocument::itemById(const QUuid &id)
 			return item;
 	return 0;
 }
+
+void
+DiagramDocument::print(QPrinter *printer, const QRectF &target, const QRectF &source)
+{
+    QPainter painter(printer);
+    print(&painter, target, source);
+}
+
+void
+DiagramDocument::print(QPainter *painter, const QRectF &target, const QRectF &source)
+{
+    setPrinting(true);
+    render(painter, target, source);
+    setPrinting(false);
+}