# -*- coding: utf-8 -*- #------------------------------------------------------------------------------- # This file is part of Code_Saturne, a general-purpose CFD tool. # # Copyright (C) 1998-2014 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-2013 EDF S.A., France @author: U{EDF R&D} @license: GNU GPL v2, see COPYING for details. """ #------------------------------------------------------------------------------- # Standard modules #------------------------------------------------------------------------------- import os, sys, string, shutil, signal, logging import subprocess, platform try: import ConfigParser # Python2 configparser = ConfigParser except Exception: import configparser # Python3 #------------------------------------------------------------------------------- # Third-party modules #------------------------------------------------------------------------------- from PyQt4.QtCore import * from PyQt4.QtGui import * #------------------------------------------------------------------------------- # Application modules #------------------------------------------------------------------------------- import cs_info from cs_exec_environment import \ separate_args, update_command_single_value, assemble_args, enquote_arg try: from Base.MainForm import Ui_MainForm from 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 Base.MainForm import Ui_MainForm from Base.NewCaseDialogForm import Ui_NewCaseDialogForm from Base.IdView import IdView from Base.BrowserView import BrowserView from Base import XMLengine from Base.XMLinitialize import * from Base.XMLmodel import * from Base.Toolbox import GuiParam, displaySelectedPage from Base.Common import XML_DOC_VERSION try: import Pages except: sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from Pages.WelcomeView import WelcomeView from Pages.IdentityAndPathesModel import IdentityAndPathesModel from Pages.XMLEditorView import XMLEditorView from Base.QtPage import setGreenColor, getexistingdirectory from Base.QtPage import from_qvariant, to_text_string, getopenfilename, getsavefilename #------------------------------------------------------------------------------- # 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.connect(self.listViewDirectory, SIGNAL("doubleClicked(QModelIndex)"), self.slotParentDirectory) self.connect(self.listViewFolder, SIGNAL("clicked(QModelIndex)"), self.slotSelectFolder) self.connect(self.lineEditCurrentPath, SIGNAL("textChanged(const QString &)"), self.slotChangeDirectory) self.connect(self.lineEditCaseName, SIGNAL("textChanged(const QString &)"), self.slotChangeName) self.connect(self.checkBoxPost, SIGNAL("clicked()"), self.slotPostStatus) self.connect(self.checkBoxMesh, SIGNAL("clicked()"), self.slotMeshStatus) self.connect(self.checkBoxCopyFrom, SIGNAL("clicked()"), self.slotCopyFromStatus) self.connect(self.pushButtonCopyFrom, SIGNAL("clicked()"), self.slotCopyFromCase) 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) def slotSelectFolder(self, index): if index.row() > 1: self.lineEditCaseName.setText(str(self.model.fileName(index))) @pyqtSignature("const QString &") 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)) @pyqtSignature("const QString &") def slotChangeName(self, text): """ change case name """ self.caseName = str(text) @pyqtSignature("") def slotPostStatus(self): if self.checkBoxPost.isChecked(): self.createMesh = True else: self.createMesh = False @pyqtSignature("") def slotMeshStatus(self): if self.checkBoxMesh.isChecked(): self.createPost = True else: self.createPost = False @pyqtSignature("") def slotCopyFromStatus(self): if self.checkBoxCopyFrom.isChecked(): self.copyFrom = True self.pushButtonCopyFrom.setEnabled(True) setGreenColor(self.pushButtonCopyFrom, True) else: self.copyFrom = False setGreenColor(self.pushButtonCopyFrom, False) self.pushButtonCopyFrom.setEnabled(False) @pyqtSignature("") 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) setGreenColor(self.pushButtonCopyFrom, False) @pyqtSignature("") def accept(self): """ Method called when user clicks 'OK' """ if self.caseName == None or self.caseName == "": msg = "name case 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) output = p.communicate() if p.returncode != 0: sys.stderr.write(output[1] + '\n') if self.createMesh == True: path = os.path.join(self.currentPath, "MESH") os.mkdir(path) if self.createPost == True: path = os.path.join(self.currentPath, "POST") 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 """ if cmd_package.name == 'code_saturne': return MainViewSaturne.__new__(MainViewSaturne, cmd_package, cmd_case, cmd_salome) elif cmd_package.name == 'neptune_cfd': from core.MainView import MainViewNeptune return MainViewNeptune.__new__(MainViewNeptune, 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.setWindowTitle(self.package.code_name + " GUI" + " - " + self.package.version) self.Id = IdView() self.dockWidgetIdentity.setWidget(self.Id) 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.connect(self.fileOpenAction, SIGNAL("triggered()"), self.fileOpen) self.connect(self.caseNewAction, SIGNAL("triggered()"), self.caseNew) self.connect(self.fileNewAction, SIGNAL("triggered()"), self.fileNew) self.connect(self.menuRecent, SIGNAL("aboutToShow()"), self.updateRecentFileMenu) self.connect(self.fileSaveAction, SIGNAL("triggered()"), self.fileSave) self.connect(self.fileSaveAsAction, SIGNAL("triggered()"), self.fileSaveAs) self.connect(self.fileCloseAction, SIGNAL("triggered()"), self.close) self.connect(self.fileQuitAction, SIGNAL("triggered()"), self.fileQuit) self.connect(self.openXtermAction, SIGNAL("triggered()"), self.openXterm) self.connect(self.displayCaseAction, SIGNAL("triggered()"), self.displayCase) self.connect(self.IdentityAction, SIGNAL("toggled(bool)"), self.dockWidgetIdentityDisplay) self.connect(self.BrowserAction, SIGNAL("toggled(bool)"), self.dockWidgetBrowserDisplay) self.connect(self.displayAboutAction, SIGNAL("triggered()"), self.displayAbout) self.connect(self.backgroundColorAction, SIGNAL("triggered()"), self.setColor) self.connect(self.actionFont, SIGNAL("triggered()"), self.setFontSize) self.connect(self.displayLicenceAction, SIGNAL("triggered()"), self.displayLicence) # connection for page layout self.connect(self.Browser.treeView, SIGNAL("pressed(const QModelIndex &)"), self.displayNewPage) self.connect(self, SIGNAL("destroyed(QObject*)"), 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()) color = settings.value("MainWindow/Color", self.palette().color(QPalette.Window).name()) color = QColor(color) if color.isValid(): self.setPalette(QPalette(color)) app = QCoreApplication.instance() app.setPalette(QPalette(color)) f = settings.value("MainWindow/Font", str(self.font())) if f: font = QFont() if (font.fromString(from_qvariant(f, to_text_string))): self.setFont(font) app = QCoreApplication.instance() 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.caseNew() 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) @pyqtSignature("bool") def dockWidgetIdentityDisplay(self, bool=True): """ Private slot. Show or hide the the identity dock window. @type bool: C{True} or C{False} @param bool: if C{True}, shows the identity dock window """ if bool: self.dockWidgetIdentity.show() else: self.dockWidgetIdentity.hide() @pyqtSignature("bool") 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() @pyqtSignature("") 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) self.connect(action, SIGNAL("triggered()"), 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: self.recentFiles.pop() 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()) settings.setValue("MainWindow/Color", self.palette().color(QPalette.Window).name()) settings.setValue("MainWindow/Font", self.font().toString()) 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 @pyqtSignature("") def fileQuit(self): """ Public slot. try to quit all window """ QApplication.closeAllWindows() @pyqtSignature("") def fileNew(self): """ Public slot. create new Code_Saturne case """ if not hasattr(self, 'case'): self.case = XMLengine.Case(package=self.package) self.case.root()['version'] = self.XML_DOC_VERSION self.initCase() title = self.tr("New parameters set") + \ " - " + self.tr(self.package.code_name) + self.tr(" GUI") \ + " - " + self.package.version self.setWindowTitle(title) self.Browser.configureTree(self.case) self.dockWidgetBrowserDisplay(True) self.case['salome'] = self.salome self.scrollArea.setWidget(self.displayFisrtPage()) self.case['saved'] = "yes" self.connect(self.case, SIGNAL("undo"), self.slotUndoRedoView) else: MainView(cmd_package=self.package, cmd_case="new case").show() @pyqtSignature("") def caseNew(self): """ Public slot. create new Code_Saturne case """ if not hasattr(self, 'case'): dialog = NewCaseDialogView(self, self.package) if dialog.exec_(): self.case = XMLengine.Case(package=self.package) self.case.root()['version'] = self.XML_DOC_VERSION self.initCase() title = self.tr("New parameters set") + \ " - " + self.tr(self.package.code_name) + self.tr(" GUI") \ + " - " + self.package.version self.setWindowTitle(title) self.Browser.configureTree(self.case) self.dockWidgetBrowserDisplay(True) self.case['salome'] = self.salome self.scrollArea.setWidget(self.displayFisrtPage()) self.case['saved'] = "yes" self.connect(self.case, SIGNAL("undo"), 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 = XMLengine.Case(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 # Check if the root node is the good one if self.case.xmlRootNode().tagName != self.package.code_name + "_GUI": msg = self.tr("This file seems not to be a %s file of parameters.\n\n" \ "XML root node found: %s" % \ (self.package.code_name, self.case.xmlRootNode().tagName)) self.loadingAborted(msg, fn) return if self.case.root()['version'] != self.XML_DOC_VERSION: msg = self.tr("The syntax version %s of this file of parameters does\n" \ "not match with the version required y the GUI: %s." % \ (self.case.root()['version'], self.XML_DOC_VERSION) ) self.loadingAborted(msg, fn) return # Cleaning the '\n' and '\t' from file_name (except in formula) self.case.xmlCleanAllBlank(self.case.xmlRootNode()) msg = self.initCase() if msg: self.loadingAborted(msg, fn) return # 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 title = fn + " - " + self.tr(self.package.code_name) + self.tr(" GUI") \ + " - " + self.package.version self.setWindowTitle(title) msg = self.tr("Loaded: %s" % fn) self.statusbar.showMessage(msg, 2000) self.scrollArea.setWidget(self.displayFisrtPage()) self.case['saved'] = "yes" # Update the Tree Navigator layout if self.batch == True: self.initializeBatchRunningWindow() self.currentEntry = 'Prepare batch calculation' self.connect(self.case, SIGNAL("undo"), self.slotUndoRedoView) @pyqtSignature("") 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() @pyqtSignature("") 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) @pyqtSignature("") 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 updateStudyId(self): """ private method update the Study Identity dock widget """ study = self.case.root().xmlGetAttribute('study') case = self.case.root().xmlGetAttribute('case') file_name = XMLengine._encode(self.case['xmlfile']) self.Id.setStudyName(study) self.Id.setCaseName(case) self.Id.setXMLFileName(file_name) @pyqtSignature("") 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: self.fileSaveAs() 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.updateStudyId() self.case.xmlSaveDocument() self.batchFileSave() # force to blank after save self.case['undo'] = [] self.case['redo'] = [] self.slotUndoRedoView() log.debug("fileSave(): ok") msg = self.tr("%s saved" % file_name) self.statusbar.showMessage(msg, 2000) @pyqtSignature("") 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.updateStudyId() self.case.xmlSaveDocument() self.batchFileSave() title = os.path.basename(self.case['xmlfile']) + " - " + self.tr(self.package.code_name) + self.tr(" GUI") \ + " - " + self.package.version self.setWindowTitle(title) # 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 displayManual(self, manual, reader = None): """ private method open a manual """ argv_info = ['--guide'] argv_info.append(manual) cs_info.main(argv_info, self.package) 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['batch'] = self.batch_file del IdentityAndPathesModel self.updateStudyId() def batchFileUpdate(self): """ Update the run command """ cmd_name = self.case['package'].name parameters = os.path.basename(self.case['xmlfile']) batch_lines = self.batch_lines for i in range(len(batch_lines)): if batch_lines[i][0:1] != '#': index = string.find(batch_lines[i], cmd_name) if index < 0: continue line = batch_lines[i] if line[index + len(cmd_name):].strip()[0:3] != 'run': continue args = update_command_single_value(separate_args(line.rstrip()), ('--param', '--param=', '-p'), enquote_arg(parameters)) batch_lines[i] = assemble_args(args) + '\n' break self.batch_lines = batch_lines @pyqtSignature("") def batchFileSave(self): """ public slot save the current case """ log.debug("batchFileSave()") if not hasattr(self, 'case'): return if self.case['batch'] and self.batch_lines: self.batchFileUpdate() batch = os.path.join(self.case['scripts_path'], self.case['batch']) f = open(batch, 'w') f.writelines(self.batch_lines) f.close() else: try: node_models = self.case.xmlGetNode('thermophysical_models') node_cht = node_models.xmlGetNode('conjugate_heat_transfer') node_syr = node_cht.xmlGetNode('external_coupling') if len(node_syr.xmlGetNodeList('syrthes')) > 0: msg = "The Calculation management/Prepare batch page should be " \ + "visited before saving so that the script file is updated " \ + "relative to the XML file." QMessageBox.warning(self, self.package.code_name + ' Interface', msg) except Exception: pass @pyqtSignature("const QModelIndex &") 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 if self.Browser.isFolder(): return # 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.Id, self.Browser) if self.page is not None: self.scrollArea.setWidget(self.page) else: log.debug("displayNewPage() self.page == None") raise @pyqtSignature("") 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 + ' Interface', msg) @pyqtSignature("") def displayLicence(self): """ public slot GNU GPL license dialog window """ QMessageBox.about(self, self.package.code_name + ' Interface', "see COPYING file") # TODO @pyqtSignature("") def displayConfig(self): """ public slot configuration information window """ QMessageBox.about(self, self.package.code_name + ' Interface', "see config.py") # TODO @pyqtSignature("") def setColor(self): """ public slot choose GUI color """ c = self.palette().color(QPalette.Window) color = QColorDialog.getColor(c, self) if color.isValid(): self.setPalette(QPalette(color)) app = QCoreApplication.instance() app.setPalette(QPalette(color)) @pyqtSignature("") def setFontSize(self): """ public slot choose GUI font """ font, ok = QFontDialog.getFont(self) log.debug("setFont -> %s" % ok) if ok: self.setFont(font) app = QCoreApplication.instance() app.setFont(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 betwenn 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) # Get batch type config = configparser.ConfigParser() config.read(cmd_package.get_configfiles()) cs_batch_type = None if config.has_option('install', 'batch'): cs_batch_type = config.get('install', 'batch') # create some instance variables self.cmd_case = cmd_case self.salome = cmd_salome self.batch_type = cs_batch_type self.batch = None self.batch_file = cmd_package.runcase self.batch_lines = [] self.package = cmd_package self.XML_DOC_VERSION = XML_DOC_VERSION self.Browser = BrowserView() self.ui_initialize() self.connect(self.displayCSManualAction, SIGNAL("triggered()"), self.displayCSManual) self.connect(self.displayCSTutorialAction, SIGNAL("triggered()"), self.displayCSTutorial) self.connect(self.displayCSKernelAction, SIGNAL("triggered()"), self.displayCSKernel) self.connect(self.displayCSRefcardAction, SIGNAL("triggered()"), self.displayCSRefcard) self.connect(self.actionUndo, SIGNAL("activated()"), self.slotUndo) self.connect(self.actionRedo, SIGNAL("activated()"), self.slotRedo) def initCase(self): """ Initializes the new case with default xml nodes. If previous case, just check if all mandatory nodes exist. """ return XMLinit(self.case).initialize() def displayWelcomePage(self): """ Display the Welcome (and the default) page """ self.page = WelcomeView() self.scrollArea.setWidget(self.page) def displayFisrtPage(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('Identity and paths', self, self.case, stbar=self.statusbar, study=self.Id, tree=self.Browser) @pyqtSignature("") def displayCSManual(self): """ public slot open the user manual """ self.displayManual('user') @pyqtSignature("") def displayCSTutorial(self): """ public slot open the tutorial for Code_Saturne """ self.displayManual('tutorial') @pyqtSignature("") def displayCSKernel(self): """ public slot open the theory and programmer's guide """ self.displayManual('theory') @pyqtSignature("") def displayCSRefcard(self): """ public slot open the quick reference card for Code_Saturne """ self.displayManual('refcard') @pyqtSignature("") def slotUndo(self): """ public slot """ if self.case['undo'] != []: 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, study=self.Id, tree=self.Browser) self.scrollArea.setWidget(p) @pyqtSignature("") def slotRedo(self): """ public slot """ if self.case['redo'] != []: 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, study=self.Id, tree=self.Browser) self.scrollArea.setWidget(p) @pyqtSignature("") 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 exist @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 #-------------------------------------------------------------------------------