# -*- coding: utf-8 -*- #------------------------------------------------------------------------------- # This file is part of Code_Saturne, a general-purpose CFD tool. # # Copyright (C) 1998-2019 EDF S.A. # # 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. #------------------------------------------------------------------------------- """ This module defines the main application classes for the Qt GUI. This GUI provides a simple way to display independante pages, in order to put informations in the XML document, which reflets the treated case. This module defines the following classes: - NewCaseDialogView - MainView @copyright: 1998-2019 EDF S.A., France @author: U{EDF} @license: GNU GPL v2 or later, see COPYING for details. """ #------------------------------------------------------------------------------- # Standard modules #------------------------------------------------------------------------------- import os, sys, shutil, signal, logging import subprocess, platform try: import ConfigParser # Python2 configparser = ConfigParser except Exception: import configparser # Python3 #------------------------------------------------------------------------------- # Third-party modules #------------------------------------------------------------------------------- from code_saturne.Base.QtCore import * from code_saturne.Base.QtGui import * from code_saturne.Base.QtWidgets import * #------------------------------------------------------------------------------- # Application modules #------------------------------------------------------------------------------- import cs_info from cs_exec_environment import \ separate_args, update_command_single_value, assemble_args, enquote_arg import cs_runcase try: from code_saturne.Base.MainForm import Ui_MainForm from code_saturne.Base.NewCaseDialogForm import Ui_NewCaseDialogForm except: sys.path.insert(1, os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "Base")) from code_saturne.Base.MainForm import Ui_MainForm from code_saturne.Base.NewCaseDialogForm import Ui_NewCaseDialogForm from code_saturne.Base.BrowserView import BrowserView from code_saturne.model import XMLengine from code_saturne.Base import QtCase from code_saturne.model.XMLinitialize import * from code_saturne.model.XMLinitializeNeptune import * from code_saturne.model.XMLmodel import * from code_saturne.model.Common import GuiParam, XML_DOC_VERSION from code_saturne.Base.Toolbox import displaySelectedPage try: import code_saturne.Pages except: sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from code_saturne.Pages.WelcomeView import WelcomeView from code_saturne.model.IdentityAndPathesModel import IdentityAndPathesModel from code_saturne.Pages.XMLEditorView import XMLEditorView from code_saturne.Pages.BatchRunningDialogView import BatchRunningDialogView from code_saturne.Pages.OpenTurnsDialogView import OpenTurnsDialogView from code_saturne.model.ScriptRunningModel import ScriptRunningModel from code_saturne.model.SolutionDomainModel import getRunType from code_saturne.Base.QtPage import getexistingdirectory from code_saturne.Base.QtPage import from_qvariant, to_text_string, getopenfilename, getsavefilename from code_saturne.cs_mei_to_c import mei_to_c_interpreter #------------------------------------------------------------------------------- # log config #------------------------------------------------------------------------------- logging.basicConfig() log = logging.getLogger("MainView") log.setLevel(GuiParam.DEBUG) #------------------------------------------------------------------------------- # New case dialog #------------------------------------------------------------------------------- class NewCaseDialogView(QDialog, Ui_NewCaseDialogForm): """ Advanced dialog """ def __init__(self, parent, pkg): """ Constructor """ QDialog.__init__(self, parent) Ui_NewCaseDialogForm.__init__(self) self.setupUi(self) self.package = pkg self.createMesh = False self.createPost = False self.copyFrom = False self.copyFromName = None self.caseName = None self.currentPath = str(QDir.currentPath()) self.model = QFileSystemModel(self) self.model.setRootPath(self.currentPath) self.model.setFilter(QDir.Dirs) self.listViewDirectory.setModel(self.model) self.modelFolder = QStringListModel() self.listViewFolder.setModel(self.modelFolder) self.listViewDirectory.setRootIndex(self.model.index(self.currentPath)) self.lineEditCurrentPath.setText(str(self.currentPath)) self.pushButtonCopyFrom.setEnabled(False) self.listViewDirectory.doubleClicked[QModelIndex].connect(self.slotParentDirectory) self.listViewFolder.clicked[QModelIndex].connect(self.slotSelectFolder) self.lineEditCurrentPath.textChanged[str].connect(self.slotChangeDirectory) self.lineEditCaseName.textChanged[str].connect(self.slotChangeName) self.checkBoxPost.clicked.connect(self.slotPostStatus) self.checkBoxMesh.clicked.connect(self.slotMeshStatus) self.checkBoxCopyFrom.clicked.connect(self.slotCopyFromStatus) self.pushButtonCopyFrom.clicked.connect(self.slotCopyFromCase) @pyqtSlot("QModelIndex") def slotParentDirectory(self, index): if index.row() == 1 and index.column() == 0: # ".." change directory self.currentPath = os.path.abspath(os.path.join(self.currentPath, "..")) self.model.setRootPath(self.currentPath) self.listViewDirectory.setRootIndex(self.model.index(self.currentPath)) elif index.row() > 1: self.currentPath = self.model.filePath(index) self.model.setRootPath(self.currentPath) self.listViewDirectory.setRootIndex(self.model.index(self.currentPath)) self.lineEditCurrentPath.setText(str(self.currentPath)) # construct list of study in current directory cpath = QDir(str(self.currentPath)) cpath.setFilter(QDir.Dirs | QDir.NoSymLinks) lst = [] self.modelFolder.setStringList([]) for name in cpath.entryList(): stud = True base = os.path.abspath(os.path.join(self.currentPath, str(name))) for rep in ['DATA', 'RESU', 'SRC', 'SCRIPTS']: directory = os.path.abspath(os.path.join(base, rep)) if not(os.path.isdir(directory)): stud = False if stud: lst.append(name) self.modelFolder.setStringList(lst) @pyqtSlot("QModelIndex") def slotSelectFolder(self, index): if index.row() > 1: self.lineEditCaseName.setText(str(self.model.fileName(index))) @pyqtSlot(str) def slotChangeDirectory(self, text): """ change directory """ if os.path.isdir(text): self.currentPath = str(text) self.model.setRootPath(self.currentPath) self.listViewDirectory.setRootIndex(self.model.index(self.currentPath)) @pyqtSlot(str) def slotChangeName(self, text): """ change case name """ self.caseName = str(text) @pyqtSlot() def slotPostStatus(self): if self.checkBoxPost.isChecked(): self.createPost = True else: self.createPost = False @pyqtSlot() def slotMeshStatus(self): if self.checkBoxMesh.isChecked(): self.createMesh = True else: self.createMesh = False @pyqtSlot() def slotCopyFromStatus(self): if self.checkBoxCopyFrom.isChecked(): self.copyFrom = True self.pushButtonCopyFrom.setEnabled(True) self.pushButtonCopyFrom.setStyleSheet("background-color: red") else: self.copyFrom = False self.pushButtonCopyFrom.setStyleSheet("background-color: green") self.pushButtonCopyFrom.setEnabled(False) @pyqtSlot() def slotCopyFromCase(self): title = self.tr("Choose an existing case") path = os.getcwd() dataPath = os.path.join(path, "..", "DATA") if os.path.isdir(dataPath): path = dataPath dirName = getexistingdirectory(self, title, path, QFileDialog.ShowDirsOnly | QFileDialog.DontResolveSymlinks) if not dirName: self.copyFromName = None else: self.copyFromName = str(dirName) self.pushButtonCopyFrom.setStyleSheet("background-color: green") @pyqtSlot() def accept(self): """ Method called when user clicks 'OK' """ if self.caseName == None or self.caseName == "": msg = "case name not defined" sys.stderr.write(msg + '\n') elif self.copyFrom == True and self.copyFromName == None: msg = "copy from case not defined" sys.stderr.write(msg + '\n') else: cmd = [os.path.join(self.package.get_dir("bindir"), self.package.name), 'create', '-c', os.path.join(self.currentPath, self.caseName)] if self.copyFrom == True: cmd += ['--copy-from', self.copyFromName] p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True) output = p.communicate() if p.returncode != 0: sys.stderr.write(output[1] + '\n') if self.createMesh == True: path = os.path.join(self.currentPath, "MESH") if not os.path.isdir(path): os.mkdir(path) if self.createPost == True: path = os.path.join(self.currentPath, "POST") if not os.path.isdir(path): os.mkdir(path) self.currentPath = os.path.join(self.currentPath, self.caseName, "DATA") self.model.setRootPath(self.currentPath) self.listViewDirectory.setRootIndex(self.model.index(self.currentPath)) self.lineEditCurrentPath.setText(str(self.currentPath)) os.chdir(self.currentPath) QDialog.accept(self) def reject(self): """ Method called when user clicks 'Cancel' """ QDialog.reject(self) #------------------------------------------------------------------------------- # Base Main Window #------------------------------------------------------------------------------- class MainView(object): """ Abstract class """ NextId = 1 Instances = set() def __new__(cls, cmd_package = None, cmd_case = "", cmd_salome = None): """ Factory """ return MainViewSaturne.__new__(MainViewSaturne, cmd_package, cmd_case, cmd_salome) @staticmethod def updateInstances(qobj): """ Overwrites the Instances set with a set that contains only those window instances that are still alive. """ MainView.Instances = set([window for window \ in MainView.Instances if isAlive(window)]) def ui_initialize(self): self.setAttribute(Qt.WA_DeleteOnClose) MainView.Instances.add(self) self.dockWidgetBrowser.setWidget(self.Browser) self.scrollArea = QScrollArea(self.frame) self.gridlayout1.addWidget(self.scrollArea,0,0,1,1) #self.gridlayout1.setMargin(0) self.gridlayout1.setSpacing(0) self.gridlayout.addWidget(self.frame,0,0,1,1) self.scrollArea.setWidgetResizable(True) self.scrollArea.setFrameShape(QFrame.StyledPanel) self.scrollArea.setFrameShadow(QFrame.Raised) self.scrollArea.setFrameStyle(QFrame.NoFrame) # connections self.fileOpenAction.triggered.connect(self.fileOpen) self.caseNewAction.triggered.connect(self.caseNew) self.fileNewAction.triggered.connect(self.fileNew) self.menuRecent.aboutToShow.connect(self.updateRecentFileMenu) self.fileSaveAction.triggered.connect(self.fileSave) self.filePythonAction.triggered.connect(self.filePythonSave) self.fileSaveAsAction.triggered.connect(self.fileSaveAs) self.fileCloseAction.triggered.connect(self.close) self.fileQuitAction.triggered.connect(self.fileQuit) self.openTextEditorAction.triggered.connect(self.fileEditorOpen) self.openResultsFileAction.triggered.connect(self.fileViewerOpen) self.testUserCompilationAction.triggered.connect(self.testUserFilesCompilation) self.runOrSubmitAction.triggered.connect(self.runOrSubmit) self.openXtermAction.triggered.connect(self.openXterm) self.displayCaseAction.triggered.connect(self.displayCase) self.BrowserAction.toggled.connect(self.dockWidgetBrowserDisplay) self.displayAboutAction.triggered.connect(self.displayAbout) self.backgroundColorAction.triggered.connect(self.setColor) self.actionFont.triggered.connect(self.setFontSize) self.RestoreStyleDefaults.triggered.connect(self.restoreStyleDefaults) self.displayLicenceAction.triggered.connect(self.displayLicence) # connection for page layout self.Browser.treeView.pressed.connect(self.displayNewPage) self.destroyed.connect(MainView.updateInstances) # Ctrl+C signal handler (allow to shutdown the GUI with Ctrl+C) signal.signal(signal.SIGINT, signal.SIG_DFL) self.resize(800, 700) # restore system settings settings = QSettings() try: # API 2 if settings.value("RecentFiles", []) is not None: try: recentFiles = settings.value("RecentFiles").toStringList() self.recentFiles = [] for f in recentFiles: self.recentFiles.append(str(f)) except: self.recentFiles = list(settings.value("RecentFiles", [])) else: self.recentFiles = [] self.restoreGeometry(settings.value("MainWindow/Geometry", QByteArray())) self.restoreState(settings.value("MainWindow/State", QByteArray())) except: # API 1 self.recentFiles = settings.value("RecentFiles").toStringList() self.restoreGeometry(settings.value("MainWindow/Geometry").toByteArray()) self.restoreState(settings.value("MainWindow/State").toByteArray()) app = QCoreApplication.instance() self.palette_default = None self.font_default = None if settings.contains("MainWindow/Color"): color = settings.value("MainWindow/Color", self.palette().color(QPalette.Window).name()) color = QColor(color) if color.isValid(): if not self.palette_default: self.palette_default = QPalette().resolve(self.palette()) self.setPalette(QPalette(color)) app.setPalette(QPalette(color)) if settings.contains("MainWindow/Font"): f = settings.value("MainWindow/Font", str(self.font())) if f: if not self.font_default: self.font_default = self.font() font = QFont() if (font.fromString(from_qvariant(f, to_text_string))): self.setFont(font) app.setFont(font) self.updateRecentFileMenu() QTimer.singleShot(0, self.loadInitialFile) self.statusbar.setSizeGripEnabled(False) self.statusbar.showMessage(self.tr("Ready"), 5000) self.actionRedo.setEnabled(False) self.actionUndo.setEnabled(False) def loadInitialFile(self): """ Private method. Checks the opening mode (from command line). """ log.debug("loadInitialFile -> %s" % self.cmd_case) # 1) new case if self.cmd_case == "new case": MainView.NextId += 1 self.fileNew() self.dockWidgetBrowserDisplay(True) # 2) existing case elif self.cmd_case: try: self.loadFile(self.cmd_case) self.dockWidgetBrowserDisplay(True) except: raise # 3) neutral point (default page layout) else: self.displayWelcomePage() self.dockWidgetBrowserDisplay(False) def dockWidgetBrowserDisplay(self, bool=True): """ Private slot. Show or hide the browser dock window. @type bool: C{True} or C{False} @param bool: if C{True}, shows the browser dock window """ if bool: self.dockWidgetBrowser.show() else: self.dockWidgetBrowser.hide() def updateRecentFileMenu(self): """ private method update the File menu with the recent files list """ self.menuRecent.clear() if hasattr(self, 'case'): current = str(self.case['xmlfile']) else: current = "" recentFiles = [] for f in self.recentFiles: if f != current and QFile.exists(f): recentFiles.append(f) if recentFiles: for i, f in enumerate(recentFiles): action = QAction(QIcon(":/icons/22x22/document-open.png"), "&%d %s" % ( i + 1, QFileInfo(f).fileName()), self) action.setData(f) action.triggered.connect(self.loadRecentFile) self.menuRecent.addAction(action) def addRecentFile(self, fname): """ private method creates and update the list of the recent files @type fname: C{str} @param fname: filename to add in the recent files list """ if fname is None: return if not fname in self.recentFiles: self.recentFiles.insert(0, str(fname)) # l'idée est de garder les 8 premiers élements de la # liste. On pourrait donc remplacer le code ci-dessus par # : # self.recentFiles = self.recentFiles[:8] while len(self.recentFiles) > 9: try: self.recentFiles.removeLast() except: try: self.recentFiles.pop() except: self.recentFiles.takeLast() def closeEvent(self, event): """ public slot try to quit all the current MainWindow """ if self.okToContinue(): settings = QSettings() if self.recentFiles: recentFiles = self.recentFiles else: recentFiles = [] settings.setValue("RecentFiles", recentFiles) settings.setValue("MainWindow/Geometry", self.saveGeometry()) settings.setValue("MainWindow/State", self.saveState()) event.accept() log.debug("closeEvent -> accept") else: event.ignore() log.debug("closeEvent -> ignore") def okToContinue(self): """ private method ask for unsaved changes before quit @return: C{True} or C{False} """ title = self.tr("Quit") msg = self.tr("Save unsaved changes?") if hasattr(self, 'case'): log.debug("okToContinue -> %s" % self.case.isModified()) if hasattr(self, 'case') and self.case.isModified(): reply = QMessageBox.question(self, title, msg, QMessageBox.Yes| QMessageBox.No| QMessageBox.Cancel) if reply == QMessageBox.Cancel: return False elif reply == QMessageBox.Yes: self.fileSave() return True def fileQuit(self): """ Public slot. try to quit all window """ QApplication.closeAllWindows() def fileNew(self): """ Public slot. create new Code_Saturne case """ if not hasattr(self, 'case'): self.case = QtCase.QtCase(package=self.package) self.case.root()['version'] = self.XML_DOC_VERSION self.initCase() self.Browser.configureTree(self.case) self.dockWidgetBrowserDisplay(True) self.case['salome'] = self.salome self.scrollArea.setWidget(self.displayFirstPage()) self.case['saved'] = "yes" self.case.undo_signal.connect(self.slotUndoRedoView) else: MainView(cmd_package=self.package, cmd_case="new case").show() self.updateTitleBar() def caseNew(self): """ Public slot. create new Code_Saturne case """ if not hasattr(self, 'case') or not self.case['xmlfile']: dialog = NewCaseDialogView(self, self.package) if dialog.exec_(): self.case = QtCase.QtCase(package=self.package) self.case.root()['version'] = self.XML_DOC_VERSION self.initCase() self.updateTitleBar() self.Browser.configureTree(self.case) self.dockWidgetBrowserDisplay(True) self.case['salome'] = self.salome self.scrollArea.setWidget(self.displayFirstPage()) self.case['saved'] = "yes" self.case.undo_signal.connect(self.slotUndoRedoView) else: MainView(cmd_package=self.package, cmd_case="new case").show() def fileAlreadyLoaded(self, f): """ private method check if the file to load is not already loaded @type fname: C{str} @param fname: file name to load @return: C{True} or C{False} """ for win in MainView.Instances: if isAlive(win) and hasattr(win, 'case') \ and win.case['xmlfile'] == f: win.activateWindow() win.raise_() return True return False def loadRecentFile(self, file_name=None): """ private slot reload an existing recent file @type fname: C{str} @param fname: file name to load """ # reload from File menu if file_name is None: action = self.sender() if isinstance(action, QAction): file_name = unicode(action.data().toString()) if not self.okToContinue(): return else: return # check if the file to load is not already loaded if hasattr(self, 'case'): if not self.fileAlreadyLoaded(file_name): MainView(cmd_package=self.package, cmd_case = file_name).show() else: self.loadFile(file_name) def loadingAborted(self, msg, fn): """Show a message window dialog. Delete the case if it is already loaded, but non conformal. @param msg text to display in the popup window @param fn name of the file pf parameters """ msg += self.tr("\n\nThe loading of %s is aborted." % fn) title = self.tr("File of parameters reading error") QMessageBox.critical(self, title, msg) if hasattr(self, 'case'): delattr(self, 'case') def loadFile(self, file_name=None): """ Private method Load an existing file. @type fname: C{str} @param fname: file name to load """ file_name = os.path.abspath(str(file_name)) fn = os.path.basename(file_name) log.debug("loadFile -> %s" % file_name) # XML syntax checker msg = XMLengine.xmlChecker(file_name) if msg: self.loadingAborted(msg, fn) return # Instantiate a new case try: self.case = QtCase.QtCase(package=self.package, file_name=file_name) except: msg = self.tr("This file is not in accordance with XML specifications.") self.loadingAborted(msg, fn) return # Cleaning the '\n' and '\t' from file_name (except in formula) self.case.xmlCleanAllBlank(self.case.xmlRootNode()) # we determine if we are in calculation mode when we open an xml file mdl = ScriptRunningModel(self.case) self.case['run_type'] = getRunType(self.case) msg = self.initCase() # All checks are fine, wan can continue... self.addRecentFile(fn) self.Browser.configureTree(self.case) self.dockWidgetBrowserDisplay(True) self.case['salome'] = self.salome # Update the case and the StudyIdBar self.case['xmlfile'] = file_name msg = self.tr("Loaded: %s" % fn) self.statusbar.showMessage(msg, 2000) self.scrollArea.setWidget(self.displayFirstPage()) self.case['saved'] = "yes" # Update the Tree Navigator layout self.case.undo_signal.connect(self.slotUndoRedoView) self.updateTitleBar() def updateTitleBar(self): """ Update Icon, Window Title Name and package name. """ icondir = os.path.dirname(os.path.abspath(__file__)) + '/' title = "" file_name = '' datadir = '' if hasattr(self, 'case'): file_name = self.case['xmlfile'] if file_name: file_name = os.path.basename(file_name) datadir = os.path.split(file_name)[0] else: file_name = '' if not datadir: datadir = os.getcwd() (casedir, data) = os.path.split(datadir) if data != 'DATA': # inconsistent paramaters location. casedir = '' case_name = os.path.basename(casedir) if case_name: title += case_name + ' : ' if file_name: title += file_name + ' - ' if hasattr(self, 'case'): package = self.case['package'] else: package = self.package title += self.tr(package.code_name) if package.code_name == "NEPTUNE_CFD": icon = QIcon(QPixmap(icondir+"logoneptune.png")) else: icon = QIcon(QPixmap(icondir+"MONO-bulle-HD.png")) self.setWindowIcon(icon) self.setWindowTitle(title) def fileOpen(self): """ public slot open an existing file """ msg = self.tr("Opening an existing case.") self.statusbar.showMessage(msg, 2000) title = self.tr("Open existing file.") if hasattr(self, 'case') and os.path.isdir(self.case['data_path']): path = self.case['data_path'] else: path = os.getcwd() dataPath = os.path.join(path, "..", "DATA") if os.path.isdir(dataPath): path = dataPath filetypes = self.tr(self.package.code_name) + self.tr(" GUI files (*.xml);;""All Files (*)") file_name, _selfilter = getopenfilename(self, title, path, filetypes) if not file_name: msg = self.tr("Loading aborted") self.statusbar.showMessage(msg, 2000) file_name = None return else: file_name = str(file_name) log.debug("fileOpen -> %s" % file_name) if hasattr(self, 'case'): if not self.fileAlreadyLoaded(file_name): MainView(cmd_package=self.package, cmd_case = file_name).show() else: self.loadFile(file_name) self.statusbar.clearMessage() def fileEditorOpen(self): """ public open a text file """ from code_saturne.Base.QFileEditor import QFileEditor # We do several checks: # - Has a case structure been initialized ? # - Does the xml file exists ? # - Is the xml file within a DATA folder ? # # Only if all of these tests are passed do we open the file editor. open_editor = True if not hasattr(self, 'case'): open_editor = False else: if not hasattr(self, 'IdPthMdl'): self.IdPthMdl = IdentityAndPathesModel(self.case) fic = self.IdPthMdl.getXmlFileName() if not fic: open_editor = False else: file_dir = os.path.split(fic)[0] if file_dir: if not os.path.basename(file_dir) == "DATA": open_editor = False if not open_editor: title = self.tr("Warning") msg = self.tr("Warning: you can only manage user files for a "\ "Code_Saturne CASE with an xml file.") QMessageBox.warning(self, title, msg) return fileEditor = QFileEditor(parent=self, case_dir=self.case['case_path']+'/SRC', noOpen=True) fileEditor.show() def fileViewerOpen(self): """ public open a text file """ from code_saturne.Base.QFileEditor import QFileEditor # We do several checks: # - Has a case structure been initialized ? # - Does the xml file exists ? # - Is the xml file within a DATA folder ? # # Only if all of these tests are passed do we open the file editor. open_viewer = True if not hasattr(self, 'case'): open_viewer = False else: if not hasattr(self, 'IdPthMdl'): self.IdPthMdl = IdentityAndPathesModel(self.case) fic = self.IdPthMdl.getXmlFileName() if not fic: open_viewer = False else: file_dir = os.path.split(fic)[0] if file_dir: if not os.path.basename(file_dir) == "DATA": open_viewer = False if not open_viewer: title = self.tr("Warning") msg = self.tr("Warning: you can only view log files for a "\ "Code_Saturne CASE with an xml file.") QMessageBox.warning(self, title, msg) return fileViewer = QFileEditor(parent=self, case_dir=self.case['case_path']+'/RESU', readOnly=True, noOpen=True, useHighlight=False) fileViewer.show() def testUserFilesCompilation(self): """ public slot test the compilation for the files within the SRC sub-folder """ if not hasattr(self, 'case'): return test_title = "Compilation test" ori_dir = os.getcwd() src_dir = self.case['case_path'] + '/SRC' if not os.path.exists(src_dir): msg = "You are not in a CASE structure.\n" msg += "Compilation aborted.\n" QMessageBox.warning(self, self.tr(test_title), msg) return os.chdir(src_dir) # Check if there are any user files within the SRC folder from glob import glob usr_files = glob("*.c") + glob("*.cxx") + glob("*.f90") if len(usr_files) > 0: from code_saturne import cs_compile from code_saturne.Base.QFileEditor import QExpandingMessageBox box = QExpandingMessageBox() out = open('comp.out', 'w') err = open('comp.err', 'w') state = cs_compile.compile_and_link(self.case['package'], self.case['package'].solver, src_dir, destdir=None, stdout=out, stderr=err) out.close() err.close() errors = open('comp.err', 'r').readlines() n_errors = 0 if state == 0: box.setIcon(QMessageBox.Information) msg = "User functions compilation succeeded." if len(errors) > 0: msg += '\n' msg += 'Warnings were found:' msg += '\n' else: box.setIcon(QMessageBox.Critical) msg = 'User functions compilation failed with the following errors:\n' box.setText(msg) box.setWindowTitle(self.tr(test_title)) if len(errors) > 0: warnings = '' for line in errors: warnings += line if sys.version_info[0] < 3: warnings = warnings.decode('utf-8').replace(u"\u2018", "'") warnings = warnings.replace(u"\u2019", "'") box.setDetailedText(warnings) box.exec_() else: msg = "There are no user files inside the SRC subfolder." QMessageBox.information(self, self.tr(test_title), msg) # Cleanup if os.path.exists('comp.out'): os.remove('comp.out') if os.path.exists('comp.err'): os.remove('comp.err') os.chdir(ori_dir) return def openXterm(self): """ public slot open an xterm window """ if sys.platform.startswith("win"): cmd = "start cmd" else: if hasattr(self, 'case'): cmd = "cd " + self.case['case_path'] + " && xterm -sb &" else: cmd = "xterm -sb &" p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True) def displayCase(self): """ public slot print the case (xml file) on the current terminal """ if hasattr(self, 'case'): dialog = XMLEditorView(self, self.case) dialog.show() def fileSave(self): """ public slot save the current case """ log.debug("fileSave()") if not hasattr(self, 'case'): return file_name = self.case['xmlfile'] log.debug("fileSave(): %s" % file_name) if not file_name: default_name = os.path.join(os.getcwd(), "setup.xml") if os.path.exists(default_name) or os.path.islink(default_name): self.fileSaveAs() return else: self.case['xmlfile'] = default_name file_name = default_name log.debug("fileSave(): %s" % os.path.dirname(file_name)) log.debug("fileSave(): %s" % os.access(os.path.dirname(file_name), os.W_OK)) if not os.access(os.path.dirname(file_name), os.W_OK): title = self.tr("Save error") msg = self.tr("Failed to write %s " % file_name) QMessageBox.critical(self, title, msg) msg = self.tr("Saving aborted") self.statusbar.showMessage(msg, 2000) return self.updateTitleBar() self.case.xmlSaveDocument() self.batchFileSave() meg_state = self.saveUserFormulaInC() if meg_state == -1: return # force to blank after save self.case['saved'] = 'yes' self.case['undo'] = [] self.case['redo'] = [] self.slotUndoRedoView() log.debug("fileSave(): ok") msg = self.tr("%s saved" % file_name) self.statusbar.showMessage(msg, 2000) def fileSaveAs(self): """ public slot save the current case with a new name """ log.debug("fileSaveAs()") if hasattr(self,'case'): filetypes = self.tr(self.package.code_name) + self.tr(" GUI files (*.xml);;""All Files (*)") fname, _selfilter = getsavefilename(self, self.tr("Save File As"), self.case['data_path'], filetypes) if fname: f = str(fname) self.case['xmlfile'] = f self.addRecentFile(f) self.fileSave() self.updateTitleBar() self.case.xmlSaveDocument() self.batchFileSave() self.updateTitleBar() # force to blank after save self.case['undo'] = [] self.case['redo'] = [] self.slotUndoRedoView() else: msg = self.tr("Saving aborted") self.statusbar.showMessage(msg, 2000) def filePythonSave(self): """ public slot dump python for the current case """ log.debug("filePythonSave()") if not hasattr(self, 'case'): return file_name = self.case['pythonfile'] log.debug("filePythonSave(): %s" % file_name) if not file_name: self.filePythonSaveAs() return log.debug("fileSave(): %s" % os.path.dirname(file_name)) log.debug("fileSave(): %s" % os.access(os.path.dirname(file_name), os.W_OK)) if not os.access(os.path.dirname(file_name), os.W_OK): title = self.tr("Save error") msg = self.tr("Failed to write %s " % file_name) QMessageBox.critical(self, title, msg) msg = self.tr("Saving aborted") self.statusbar.showMessage(msg, 2000) return self.case.pythonSaveDocument() log.debug("filePythonSave(): ok") msg = self.tr("%s saved" % file_name) self.statusbar.showMessage(msg, 2000) def filePythonSaveAs(self): """ public slot save the current case with a new name """ log.debug("filePythonSaveAs()") if hasattr(self,'case'): filetypes = self.tr(self.package.code_name) + self.tr(" python files (*.py);;""All Files (*)") fname, _selfilter = getsavefilename(self, self.tr("Save Python File As"), self.case['data_path'], filetypes) if fname: f = str(fname) self.case['pythonfile'] = f self.case.pythonSaveDocument() else: msg = self.tr("Saving aborted") self.statusbar.showMessage(msg, 2000) def displayManual(self, pkg, manual, reader = None): """ private method open a manual """ argv_info = ['--guide'] argv_info.append(manual) cs_info.main(argv_info, pkg) def initializeBatchRunningWindow(self): """ initializes variables concerning the display of batchrunning """ self.IdPthMdl = IdentityAndPathesModel(self.case) fic = self.IdPthMdl.getXmlFileName() if not fic: file_name = os.getcwd() if os.path.basename(f) == 'DATA': file_name = os.path.dirname(file_name) self.IdPthMdl.setCasePath(file_name) else: file_dir = os.path.split(fic)[0] if file_dir: self.IdPthMdl.setCasePath(os.path.split(file_dir)[0]) if not os.path.basename(file_dir) == 'DATA': self.IdPthMdl.setCasePath(file_dir) else: file_dir = os.path.split(os.getcwd())[0] self.IdPthMdl.setCasePath(file_dir) for p, rep in [('data_path', 'DATA'), ('resu_path', 'RESU'), ('user_src_path', 'SRC'), ('scripts_path', 'SCRIPTS')]: self.IdPthMdl.setPathI(p, os.path.join(file_dir, rep)) self.IdPthMdl.setRelevantSubdir("yes", "") self.IdPthMdl.setPathI('mesh_path', os.path.join(os.path.abspath(os.path.split(file_dir)[0], 'MESH'))) self.case['runcase'] = cs_runcase.runcase(os.path.join(self.case['scripts_path'], self.batch_file), package=self.package) del IdentityAndPathesModel self.updateTitleBar() @pyqtSlot() def batchFileSave(self): """ public slot save the current case """ log.debug("batchFileSave()") if not hasattr(self, 'case'): return if not self.case['runcase']: self.case['runcase'] \ = cs_runcase.runcase(os.path.join(self.case['scripts_path'], self.batch_file), package=self.package) parameters = os.path.basename(self.case['xmlfile']) self.case['runcase'].set_parameters(parameters) self.case['runcase'].save() def runOrSubmit(self): """ public slot print the case (xml file) on the current terminal """ if hasattr(self, 'case'): dialog = BatchRunningDialogView(self, self.case) dialog.show() def runOTMode(self): """ public slot CFD to OpenTURNS mode """ # Test if pydefx is available for the OpenTURNS coupling. # Needed for Salome 9.3 try: from pydefx import Parameters except: ydefx_root=os.environ['YDEFX_ROOT_DIR'] sys.path.insert(0, os.path.join(ydefx_root, 'lib/salome')) sys.path.insert(0, os.path.join(ydefx_root, 'lib/python3.6/site-packages/salome')) if hasattr(self, 'case'): dialog = OpenTurnsDialogView(self, self.case) dialog.show() def displayNewPage(self, index): """ private slot display a new page when the Browser send the order @type index: C{QModelIndex} @param index: index of the item in the C{QTreeView} clicked in the browser """ # stop if the entry is a folder or a file # warning and stop if is no case if not hasattr(self, 'case'): log.debug("displayNewPage(): no attr. 'case', return ") msg = self.tr("You have to create a new case or load\n"\ "an existing case before selecting an item") w = QMessageBox(self) w.information(self, self.tr("Warning"), msg, self.tr('OK')) return self.page = self.Browser.display(self, self.case, self.statusbar, self.Browser) if self.page is not None: self.scrollArea.setWidget(self.page) else: log.debug("displayNewPage() self.page == None") raise # Auto expand nodes in the Browser view when clicked self.Browser.treeView.setExpanded(index, True) def saveUserFormulaInC(self): """ Save user defined laws with MEI to C functions """ state = 0 if self.case['run_type'] == 'standard': mci = mei_to_c_interpreter(self.case) mci_state = mci.save_all_functions() state = mci_state['state'] if state == -1: msg = '' title = self.tr("Save error") msg = "Expressions are missing !\n" for i, d in enumerate(mci_state['exps']): msg += "(%d/%d) %s %s is not provided or zone %s\n" % (i+1, mci_state['nexps'], d['var'], d['func'], d['zone']) QMessageBox.critical(self, title, msg) msg = self.tr("Saving aborted") self.statusbar.showMessage(msg, 2000) if state == 1: title = self.tr("Warning!") msg = "You are within a RESU folder!\n" msg += "The xml file was saved, as the cs_meg C functions " msg += "for mathematical expressions.\n" msg += "The C functions are hence different from those in " msg += "your DATA folder!\n" err_msg = self.tr(msg) QMessageBox.warning(self, title, err_msg) msg = self.tr("Saving was done within a RESU folder and not in CASE.") self.statusbar.showMessage(msg, 2000) self.updateTitleBar() return state def displayAbout(self): """ public slot the About dialog window shows: - title - version - contact """ msg = self.package.code_name + "\n" +\ "version " + self.package.version + "\n\n" +\ "For information about this application " +\ "please contact:\n\n" +\ self.package.bugreport + "\n\n" +\ "Please visit our site:\n" +\ self.package.url QMessageBox.about(self, self.package.name, msg) def displayLicence(self): """ public slot GNU GPL license dialog window """ QMessageBox.about(self, self.package.code_name, cs_info.licence_text) def displayConfig(self): """ public slot configuration information window """ QMessageBox.about(self, self.package.code_name, "see config.py") # TODO def setColor(self): """ public slot choose GUI color """ c = self.palette().color(QPalette.Window) color = QColorDialog.getColor(c, self) if color.isValid(): app = QCoreApplication.instance() if not self.palette_default: self.palette_default = QPalette().resolve(self.palette()) app.setPalette(QPalette(color)) settings = QSettings() settings.setValue("MainWindow/Color", self.palette().color(QPalette.Window).name()) def setFontSize(self): """ public slot choose GUI font """ font, ok = QFontDialog.getFont(self) log.debug("setFont -> %s" % ok) if ok: if not self.font_default: self.font_default = self.font() self.setFont(font) app = QCoreApplication.instance() app.setFont(font) settings = QSettings() settings.setValue("MainWindow/Font", self.font().toString()) def restoreStyleDefaults(self): """ public slot Restore default style. """ reply = QMessageBox.question(self, "Restore defaults", "Restore default color and font ?", QMessageBox.Yes | QMessageBox.No) if reply == QMessageBox.Yes: app = QCoreApplication.instance() if self.palette_default: app.setPalette(self.palette_default) if self.font_default: print(self.font_default) print(self.font()) self.setFont(self.font_default) app.setFont(self.font_default) settings = QSettings() settings.remove("MainWindow/Color") settings.remove("MainWindow/Font") def tr(self, text): """ private method translation @param text: text to translate @return: translated text """ return text #------------------------------------------------------------------------------- # Main Window for Code_Saturne #------------------------------------------------------------------------------- class MainViewSaturne(QMainWindow, Ui_MainForm, MainView): def __new__(cls, cmd_package = None, cmd_case = "", cmd_salome = None): return super(MainViewSaturne, cls). __new__(cls, cmd_package, cmd_case, cmd_salome) def __init__(self, cmd_package = None, cmd_case = "", cmd_salome = None): """ Initializes a Main Window for a new document: 1. finish the Main Window layout 2. connection between signal and slot 3. Ctrl+C signal handler 4. create some instance variables 5. restore system settings @type cmd_case: @param cmd_case: """ QMainWindow.__init__(self) Ui_MainForm.__init__(self) self.setupUi(self) # create some instance variables self.cmd_case = cmd_case self.salome = cmd_salome self.batch_file = cmd_package.runcase self.package = cmd_package self.XML_DOC_VERSION = XML_DOC_VERSION self.Browser = BrowserView() self.ui_initialize() # Code_Saturne doc self.displayCSManualAction.triggered.connect(self.displayCSManual) self.displayCSTutorialAction.triggered.connect(self.displayCSTutorial) self.displayCSTheoryAction.triggered.connect(self.displayCSTheory) self.displayCSSmgrAction.triggered.connect(self.displayCSSmgr) self.displayCSRefcardAction.triggered.connect(self.displayCSRefcard) self.displayCSDoxygenAction.triggered.connect(self.displayCSDoxygen) # NCFD doc self.displayNCManualAction.triggered.connect(self.displayNCManual) self.displayNCTutorialAction.triggered.connect(self.displayNCTutorial) self.displayNCTheoryAction.triggered.connect(self.displayNCTheory) self.displayNCDoxygenAction.triggered.connect(self.displayNCDoxygen) self.actionUndo.triggered.connect(self.slotUndo) self.actionRedo.triggered.connect(self.slotRedo) docdir = self.package.get_dir('docdir') # Code_Saturne doc ddcs = os.path.join(docdir, '../code_saturne') if os.path.isdir(ddcs): liste = os.listdir(ddcs) else: liste = [] if 'user.pdf' not in liste: self.displayCSManualAction.setEnabled(False) if 'theory.pdf' not in liste: self.displayCSTheoryAction.setEnabled(False) if 'studymanager.pdf' not in liste: self.displayCSSmgrAction.setEnabled(False) if 'refcard.pdf' not in liste: self.displayCSRefcardAction.setEnabled(False) if 'doxygen' not in liste: self.displayCSDoxygenAction.setEnabled(False) # NCFD doc ddnc = os.path.join(docdir, '../neptune_cfd') if os.path.isdir(ddnc): liste = os.listdir(ddnc) else: liste = [] if 'user.pdf' not in liste: self.displayNCManualAction.setEnabled(False) if 'tutorial.pdf' not in liste: self.displayNCTutorialAction.setEnabled(False) if 'theory.pdf' not in liste: self.displayNCTheoryAction.setEnabled(False) if 'doxygen' not in liste: self.displayNCDoxygenAction.setEnabled(False) self.updateTitleBar() def initCase(self): """ Initializes the new case with default xml nodes. If previous case, just check if all mandatory nodes exist. """ prepro_only = self.case['run_type'] != 'standard' if self.case.xmlRootNode().tagName == "NEPTUNE_CFD_GUI" : from neptune_cfd.nc_package import package as nc_package self.package = nc_package() return XMLinitNeptune(self.case).initialize(prepro_only) elif self.case.xmlRootNode().tagName == "Code_Saturne_GUI" : from code_saturne.cs_package import package as cs_package self.package = cs_package() return XMLinit(self.case).initialize(prepro_only) def displayWelcomePage(self): """ Display the Welcome (and the default) page """ self.page = WelcomeView() self.scrollArea.setWidget(self.page) def displayFirstPage(self): """ Display the first page if a file of parameters (new or previous) is loaded """ self.case['current_tab'] = 0 self.case['current_index'] = None return displaySelectedPage('Calculation environment', self, self.case, stbar=self.statusbar, tree=self.Browser) def displayCSManual(self): """ public slot open the user manual """ if self.package.name == 'code_saturne': self.displayManual(self.package, 'user') else: from code_saturne.cs_package import package as cs_package pkg = cs_package() self.displayManual(pkg, 'user') def displayCSTutorial(self): """ public slot open the tutorial for Code_Saturne """ msg = "See " + self.package.url + " web site for tutorials." QMessageBox.about(self, self.package.name, msg) def displayCSTheory(self): """ public slot open the theory and programmer's guide """ if self.package.name == 'code_saturne': self.displayManual(self.package, 'theory') else: from code_saturne.cs_package import package as cs_package pkg = cs_package() self.displayManual(pkg, 'theory') def displayCSSmgr(self): """ public slot open the studymanager guide """ if self.package.name == 'code_saturne': self.displayManual(self.package, 'studymanager') else: from code_saturne.cs_package import package as cs_package pkg = cs_package() self.displayManual(pkg, 'studymanager') def displayCSRefcard(self): """ public slot open the quick reference card for Code_Saturne """ if self.package.name == 'code_saturne': self.displayManual(self.package, 'refcard') else: from code_saturne.cs_package import package as cs_package pkg = cs_package() self.displayManual(pkg, 'refcard') def displayCSDoxygen(self): """ public slot open the quick doxygen for Code_Saturne """ if self.package.name == 'code_saturne': self.displayManual(self.package, 'Doxygen') else: from code_saturne.cs_package import package as cs_package pkg = cs_package() self.displayManual(pkg, 'Doxygen') def displayNCManual(self): """ public slot open the user manual """ if self.package.name == 'neptune_cfd': self.displayManual(self.package, 'user') else: from neptune_cfd.nc_package import package as nc_package pkg = nc_package() self.displayManual(pkg, 'user') def displayNCTutorial(self): """ public slot open the user manual """ if self.package.name == 'neptune_cfd': self.displayManual(self.package, 'tutorial') else: from neptune_cfd.nc_package import package as nc_package pkg = nc_package() self.displayManual(pkg, 'tutorial') def displayNCTheory(self): """ public slot open the user manual """ if self.package.name == 'neptune_cfd': self.displayManual(self.package, 'theory') else: from neptune_cfd.nc_package import package as nc_package pkg = nc_package() self.displayManual(pkg, 'theory') def displayNCDoxygen(self): """ public slot open the user manual """ if self.package.name == 'neptune_cfd': self.displayManual(self.package, 'Doxygen') else: from neptune_cfd.nc_package import package as nc_package pkg = nc_package() self.displayManual(pkg, 'Doxygen') def slotUndo(self): """ public slot """ if self.case['undo'] != []: python_record = self.case['dump_python'].pop() self.case['python_redo'].append([python_record[0], python_record[1], python_record[2]]) last_record = self.case['undo'].pop() self.case.record_func_prev = None self.case.xml_prev = "" self.case['redo'].append([last_record[0], self.case.toString(), last_record[2], last_record[3]]) self.case.parseString(last_record[1]) self.Browser.activeSelectedPage(last_record[2]) self.Browser.configureTree(self.case) self.case['current_index'] = last_record[2] self.case['current_tab'] = last_record[3] self.slotUndoRedoView() p = displaySelectedPage(last_record[0], self, self.case, stbar=self.statusbar, tree=self.Browser) self.scrollArea.setWidget(p) def slotRedo(self): """ public slot """ if self.case['redo'] != []: python_record = self.case['python_redo'].pop() self.case['dump_python'].append([python_record[0], python_record[1], python_record[2]]) last_record = self.case['redo'].pop() self.case.record_func_prev = None self.case.xml_prev = "" self.case['undo'].append([last_record[0], self.case.toString(), last_record[2], last_record[3]]) self.case.parseString(last_record[1]) self.Browser.activeSelectedPage(last_record[2]) self.Browser.configureTree(self.case) self.case['current_index'] = last_record[2] self.case['current_tab'] = last_record[3] self.slotUndoRedoView() p = displaySelectedPage(last_record[0], self, self.case, stbar=self.statusbar, tree=self.Browser) self.scrollArea.setWidget(p) def slotUndoRedoView(self): if self.case['undo']: self.actionUndo.setEnabled(True) else: self.actionUndo.setEnabled(False) if self.case['redo']: self.actionRedo.setEnabled(True) else: self.actionRedo.setEnabled(False) #------------------------------------------------------------------------------- def isAlive(qobj): """ return True if the object qobj exists @param qobj: the name of the attribute @return: C{True} or C{False} """ import sip try: sip.unwrapinstance(qobj) except RuntimeError: return False return True #------------------------------------------------------------------------------- # End #-------------------------------------------------------------------------------