File: PythonExport.cpp

package info (click to toggle)
bornagain 23.0-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 103,948 kB
  • sloc: cpp: 423,131; python: 40,997; javascript: 11,167; awk: 630; sh: 318; ruby: 173; xml: 130; makefile: 51; ansic: 24
file content (55 lines) | stat: -rw-r--r-- 2,089 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      GUI/View/IO/PythonExport.cpp
//! @brief     Implements Python export function in namespace IO.
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************

#include "GUI/View/IO/PythonExport.h"
#include "Base/Util/Assert.h"
#include "GUI/Model/ToCore/SimulationToCore.h"
#include "GUI/View/Widget/AppConfig.h"
#include "GUI/View/Widget/FileDialog.h"
#include "Sim/Export/ExportToPython.h"
#include "Sim/Simulation/ScatteringSimulation.h"
#include <QFile>
#include <QMessageBox>
#include <QStandardPaths>
#include <QTextStream>

void IO::Py::saveScript(const SampleItem* sampleItem, const InstrumentItem* instrumentItem,
                        const SimulationOptionsItem* optionItem)
{
    // Create output first (so that we don't go through the file dialog for nothing)
    const auto simulation = GUI::ToCore::itemsToSimulation(sampleItem, instrumentItem, optionItem);
    const QString script = QString::fromStdString(::Py::Export::simulationPlotCode(*simulation));

    QString& dirname = gApp->script_export_dir;
    if (dirname.isEmpty())
        dirname = QStandardPaths::writableLocation(QStandardPaths::HomeLocation);

    const QString filters("Python scripts (*.py)");

    QString fname = GUI::FileDialog::w1_1f("Save Python script", dirname, filters);
    if (fname.isEmpty())
        return;
    if (!fname.endsWith(".py"))
        fname += ".py";

    QFile file(fname);
    if (!file.open(QIODevice::WriteOnly)) {
        QMessageBox::warning(nullptr, "Save Python file", "File could not be opened for writing!");
        return;
    }

    QTextStream out(&file);
    out << script;
    file.close();
}