From 5087563069b609ba936d40cfec381cbd78b3eeae Mon Sep 17 00:00:00 2001
From: Ovidiu Mircescu <ovidiu.mircescu@edf.fr>
Date: Fri, 1 Mar 2019 16:18:49 +0100
Subject: [PATCH] Add setScriptText to Py2YacsDialog.

---
 src/py2yacsgui/Py2YacsDialog.cxx | 17 +++++++++++++++--
 src/py2yacsgui/Py2YacsDialog.hxx |  4 +++-
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/src/py2yacsgui/Py2YacsDialog.cxx b/src/py2yacsgui/Py2YacsDialog.cxx
index ccf3777f1..c6ba333b8 100644
--- a/src/py2yacsgui/Py2YacsDialog.cxx
+++ b/src/py2yacsgui/Py2YacsDialog.cxx
@@ -22,13 +22,13 @@
 #include <PyEditor_Widget.h>
 #include <py2yacs.hxx>
 
-Py2YacsDialog::Py2YacsDialog( QWidget* parent, bool exportXml)
+Py2YacsDialog::Py2YacsDialog( QWidget* parent)
 : QDialog(parent),
   _yacsFile(),
   _pyEditorWindow(0),
   _errorMessages(0),
   _okButton(0),
-  _exportXml(exportXml),
+  _exportXml(true),
   _pyScript()
 {
   QVBoxLayout *mainLayout = new QVBoxLayout;
@@ -128,3 +128,16 @@ std::string Py2YacsDialog::getScriptText()
 {
   return _pyScript;
 }
+
+void Py2YacsDialog::setScriptText(const std::string& pyScript)
+{
+  PyEditor_Widget* pyEdit = dynamic_cast<PyEditor_Widget*>
+                                             (_pyEditorWindow->centralWidget());
+  if(pyEdit != nullptr)
+    pyEdit->setText(QString(pyScript.c_str()));
+}
+
+void Py2YacsDialog::setExportXml(bool yes)
+{
+  _exportXml = yes;
+}
diff --git a/src/py2yacsgui/Py2YacsDialog.hxx b/src/py2yacsgui/Py2YacsDialog.hxx
index 782533738..e74ed4ac0 100644
--- a/src/py2yacsgui/Py2YacsDialog.hxx
+++ b/src/py2yacsgui/Py2YacsDialog.hxx
@@ -27,9 +27,11 @@ class PY2YACSGUILIB_EXPORT Py2YacsDialog : public QDialog
 {
   Q_OBJECT
   public:
-    Py2YacsDialog( QWidget* parent=0, bool exportXml = true);
+    Py2YacsDialog( QWidget* parent=0);
     QString getYacsFile();
     std::string getScriptText();
+    void setScriptText(const std::string& pyScript);
+    void setExportXml(bool yes);
 
   public slots:
     virtual void onExport();
-- 
2.30.2