2 files changed, 8 insertions, 0 deletions
diff --git a/src/data/tikzdocument.cpp b/src/data/tikzdocument.cpp
index 4a813ad..bf39f67 100644
--- a/src/data/tikzdocument.cpp
+++ b/src/data/tikzdocument.cpp
@@ -96,6 +96,12 @@ void TikzDocument::save() {
     }
 }
 
+void TikzDocument::setGraph(Graph *graph)
+{
+    _graph = graph;
+    refreshTikz();
+}
+
 void TikzDocument::saveAs() {
     QSettings settings("tikzit", "tikzit");
     QString fileName = QFileDialog::getSaveFileName(tikzit->activeWindow(),
diff --git a/src/data/tikzdocument.h b/src/data/tikzdocument.h
index edb1beb..9b6893a 100644
--- a/src/data/tikzdocument.h
+++ b/src/data/tikzdocument.h
@@ -19,6 +19,7 @@ public:
     ~TikzDocument();
 
     Graph *graph() const;
+    void setGraph(Graph *graph);
     QString tikz() const;
     QUndoStack *undoStack() const;
     bool parseSuccess() const;
@@ -30,6 +31,7 @@ public:
 
     void saveAs();
     void save();
+
 private:
     Graph *_graph;
     QString _tikz;