Wed 28 Aug 21:38:52 CEST 2024

src/SimNDT/gui/managerDialogsController.py

@@ -0,0 +1,516 @@
+__author__ = 'Miguel Molero'
+
+
+
+import os
+import copy
+import subprocess
+
+from PySide.QtCore import *
+from PySide.QtGui import *
+
+from SimNDT.gui.newScenarioController import NewScenario
+from SimNDT.gui.addEllipseController import AddEllipse
+from SimNDT.gui.addRectangleController import AddRectangle
+from SimNDT.gui.loadScenarioFromImageController import LoadScenarioFromImage
+from SimNDT.gui.previewScenarioController import PreviewScenario
+
+from SimNDT.gui.materialSetupController import MaterialSetup
+from SimNDT.gui.boundarySetupController import BoundarySetup
+from SimNDT.gui.singleLaunchSetupController import SingleLaunchSetup
+from SimNDT.gui.linearScanController import LinearScanSetup
+from SimNDT.gui.tomographySetupController import TomographySetup
+
+from SimNDT.gui.simulationSetupController import SimulationSetup
+from SimNDT.gui.checkSimulationController import CheckSimulation
+from SimNDT.gui.runSimulationController import RunSimulation
+from SimNDT.gui.engineController import EngineController
+from SimNDT.gui.generateVideoController import GenerateVideo
+
+from SimNDT.gui.twoPhaseModelDryCaseController import TwoPhaseModelDryCaseDialog
+from SimNDT.gui.threePhaseModelDryCaseController import ThreePhaseModelDryCaseDialog
+from SimNDT.gui.twoPhaseModelImmersionCaseController import TwoPhaseModelImmersionCaseDialog
+from SimNDT.gui.threePhaseModelImmersionCaseController import ThreePhaseModelImmersionCaseDialog
+
+from SimNDT.core.scenario import Scenario
+from SimNDT.core.geometryObjects import Ellipse, Circle, Square, Rectangle
+from SimNDT.core.simPack import SimPack
+from SimNDT.core.receivers import Receivers
+
+from SimNDT.gui.snapshots import SnapShots
+from SimNDT.gui.Warnings import WarningParms, DoneParms
+from SimNDT.gui import HelperMethods
+
+
+from SimNDT.core.concreteModel import TwoPhaseModel
+
+
+import numpy as np
+
+def toIntList(stringList):
+  splitList = stringList.split(',')
+  if len(splitList)==1:
+    if splitList[0]=='':
+      splitList=[] # empty list
+  return [int(x) for x in splitList]    
+
+
+class ManagerDialogs(object):
+    lastSimulationSetup = dict(
+              isView = False,
+              color = 0,
+              dB = 60,
+              step = 100,
+              field = 3,
+              isReceiverPlot = False,
+              isEnableFields = False,
+              isEnableSignals = False,
+              isEnableImages = False,
+              isEnableNumPy = False,
+              isEnableVolume = False,
+              sensorShape = [],
+              sensorPlacement = [],
+              sensorSize = 0,
+              lastPath = ''            
+            )
+
+    def newGeometryModel(self):
+
+        dialog = NewScenario()
+
+        if self.SimNDT_Scenario is not None:
+            dialog.widthLineEdit.setText(str(self.SimNDT_Scenario.Width))
+            dialog.heightLineEdit.setText(str(self.SimNDT_Scenario.Height))
+            dialog.pixelLineEdit.setText(str(self.SimNDT_Scenario.Pixel_mm))
+            dialog.labelSpinBox.setValue(int(self.SimNDT_Scenario.Label))
+
+        if dialog.exec_():
+            width = dialog.width
+            height = dialog.height
+            pixel = dialog.pixel
+            label = dialog.label
+
+            self.SimNDT_Scenario = Scenario(width,height,pixel,label)
+            self.SimNDT_ObjectList = None
+            self.SimNDT_Check = False
+
+            self.dirty = True
+
+
+            self.updateUI()
+
+
+    def addEllipse(self):
+        dlg = AddEllipse()
+        if dlg.exec_():
+            centerX, centerY, major, minor, theta, label = dlg.getParms()
+            if self.SimNDT_Scenario:
+                try:
+                    self.SimNDT_Scenario.addEllipse(centerX, centerY, major, minor, theta, label)
+                    if major!=minor:
+                        obj = Ellipse(centerX, centerY, major, minor, theta, label)
+                    else:
+                        obj = Circle(centerX,centerY, major, label)
+
+                except Exception as e:
+                    msg = WarningParms("Error in the object generation!!!. %s"%e)
+                    msg.exec_()
+                    return
+
+                if self.SimNDT_ObjectList is None:
+                    self.SimNDT_ObjectList = list()
+                self.SimNDT_ObjectList.append(obj)
+
+
+            self.dirty = True
+            self.updateUI()
+
+
+    def addRectangle(self):
+
+        dlg = AddRectangle()
+        if dlg.exec_():
+            centerX, centerY, width, height, theta, label = dlg.getParms()
+            if self.SimNDT_Scenario:
+
+                try:
+                    self.SimNDT_Scenario.addRectangle(centerX, centerY, width, height, theta, label)
+                    if width!=height:
+                        obj = Rectangle(centerX, centerY, width, height, theta, label)
+                    else:
+                        obj = Square(centerX,centerY, width, theta, label)
+
+                except Exception as e:
+                    msg = WarningParms("Error in the object generation!!!")
+                    msg.exec_()
+                    return
+
+                if self.SimNDT_ObjectList is None:
+                    self.SimNDT_ObjectList = list()
+                self.SimNDT_ObjectList.append(obj)
+
+            self.dirty = True
+            self.updateUI()
+
+
+
+    def loadImage(self):
+
+        dlg = LoadScenarioFromImage(self.filename)
+        if dlg.exec_():
+
+
+            M,N    = np.shape(dlg.image_labeled)
+            Pixel = float(dlg.Pixel)
+
+            Width  = N/Pixel
+            Height = M/Pixel
+
+            if self.SimNDT_Scenario is None:
+                self.SimNDT_Scenario = Scenario()
+                self.SimNDT_Scenario.setImage(dlg.image_labeled, Width, Height, Pixel, Label=0)
+            else:
+                self.SimNDT_Scenario.setImage(dlg.image_labeled, Width, Height, Pixel, Label=0)
+
+            self.GraphicView.setupZoom(100)
+            self.statusBarWidget.zoomSpinBox.setValue(100)
+
+            self.SimNDT_ObjectList	= None
+            self.SimNDT_Boundaries  = None
+            self.SimNDT_Check = False
+
+            self.dirty = True
+            self.updateUI()
+
+
+    def previewScenario(self):
+
+        dlg = PreviewScenario(self.SimNDT_Scenario, self)
+        dlg.show()
+
+
+    def rotateScenarioCounterClockwise(self):
+
+        self.SimNDT_Scenario.rotate(direction="clockwise")
+        self.dirty = True
+        self.SimNDT_Check = False
+        self.updateUI()
+
+
+    def rotateScenarioClockwise(self):
+        self.SimNDT_Scenario.rotate(direction="counterclockwise")
+        self.dirty = True
+        self.SimNDT_Check = False
+        self.updateUI()
+
+
+    def materialSetup(self):
+
+        dlg = MaterialSetup(self.SimNDT_Materials, filename = self.filename)
+        if dlg.exec_():
+            self.SimNDT_Materials = copy.deepcopy(dlg.SimNDT_Materials)
+            self.dirty = True
+
+            self.SimNDT_Check = False
+            self.updateUI()
+
+
+    def boundarySetup(self):
+
+        dlg = BoundarySetup(self.SimNDT_Boundaries)
+        if dlg.exec_():
+            self.SimNDT_Boundaries = copy.deepcopy(dlg.Boundaries)
+            self.SimNDT_Scenario.createBoundaries(self.SimNDT_Boundaries)
+
+            self.dirty = True
+            self.SimNDT_Check = False
+            self.updateUI()
+
+
+    def singleLaunchSetup(self):
+
+        dlg = SingleLaunchSetup(self.SimNDT_Scenario, self.SimNDT_Source, self.SimNDT_Inspection, self.SimNDT_Transducers, self.SimNDT_Signal)
+        if dlg.exec_():
+
+            self.SimNDT_Source = copy.deepcopy(dlg.Source)
+            self.SimNDT_Transducers = copy.deepcopy(dlg.Transducers)
+            self.SimNDT_Inspection = copy.deepcopy(dlg.Inspection)
+            self.SimNDT_Signal = copy.deepcopy(dlg.Signal)
+
+            self.SimNDT_Check = False
+            self.dirty = True
+            self.updateUI()
+
+
+    def linearScanSetup(self):
+        dlg = LinearScanSetup(self.SimNDT_Scenario, self.SimNDT_Source, self.SimNDT_Inspection, self.SimNDT_Transducers, self.SimNDT_Signal)
+        if dlg.exec_():
+
+            self.SimNDT_Source = copy.deepcopy(dlg.Source)
+            self.SimNDT_Transducers = copy.deepcopy(dlg.Transducers)
+            self.SimNDT_Inspection = copy.deepcopy(dlg.Inspection)
+            self.SimNDT_Signal = copy.deepcopy(dlg.Signal)
+
+            self.SimNDT_Check = False
+            self.dirty = True
+            self.updateUI()
+
+
+    def tomographySetup(self):
+        dlg = TomographySetup(self.SimNDT_Scenario, self.SimNDT_Source, self.SimNDT_Inspection, self.SimNDT_Transducers, self.SimNDT_Signal)
+        if dlg.exec_():
+
+
+            self.SimNDT_Source = copy.deepcopy(dlg.Source)
+            self.SimNDT_Transducers = copy.deepcopy(dlg.Transducers)
+            self.SimNDT_Inspection = copy.deepcopy(dlg.Inspection)
+            self.SimNDT_Signal = copy.deepcopy(dlg.Signal)
+
+            self.SimNDT_Check = False
+            self.dirty = True
+            self.updateUI()
+
+
+
+    def signalSetup(self):
+        pass
+
+    def simulationSetup(self):
+
+        dlg = SimulationSetup(self.SimNDT_Scenario, self.SimNDT_Materials,
+                              self.SimNDT_Transducers, self.SimNDT_Signal,
+                              self.SimNDT_Simulation)
+        if dlg.exec_():
+
+            self.SimNDT_Simulation = copy.deepcopy(dlg.Simulation)
+            self.dirty = True
+            self.OpenSimFile = False
+            self.SimNDT_Check = False
+            self.updateUI()
+
+
+    def checkSimulation(self):
+
+        dlg = CheckSimulation(self.SimNDT_Scenario, self.SimNDT_Materials, self.SimNDT_Boundaries,
+                              self.SimNDT_Transducers, self.SimNDT_Inspection, self.SimNDT_Signal, self.SimNDT_Simulation)
+
+        self.SimNDT_Check = False
+
+        if dlg.exec_():
+            self.SimNDT_Check = True
+
+        self.SimNDT_Scenario = copy.deepcopy(dlg.Scenario)
+        self.SimNDT_Boundaries = copy.deepcopy(dlg.Boundaries)
+        self.SimNDT_Inspection = copy.deepcopy(dlg.Inspection)
+        self.dirty = True
+        self.OpenSimFile = False
+        self.updateUI()
+
+
+    def runSimulation(self):
+        self.SimNDT_Simulation.lastSimulationSetup = self.lastSimulationSetup
+        dlg = RunSimulation(self.filename, self.SimNDT_Simulation)
+        if dlg.exec_():
+
+            DB = dlg.visualizacionRangeSpinBox.value()
+            color = dlg.colormapComboBox.currentIndex()
+            field = dlg.fieldSelectorComboBox.currentIndex()
+            step = dlg.snapshotStepSpinBox.value()
+            isView = dlg.viewCheckBox.isChecked()
+            isReceiverPlot = dlg.receiverCheckBox.isChecked()
+            isEnableFields = dlg.enableSavingFieldsCheckBox.isChecked()
+            isEnableSignals = dlg.enableSavingSignalsCheckBox.isChecked()
+            isEnableImages = dlg.enableSavingImagesCheckBox.isChecked()
+            isEnableNumPy = dlg.enableSavingNumpyCheckBox.isChecked()
+            isEnableVolume = dlg.enableSavingVolumeCheckBox.isChecked()
+            sensorShape = toIntList(dlg.signalShapeLineEdit.text())
+            sensorPlacement = toIntList(dlg.signalPlacementLineEdit.text())
+            sensorSize = float(dlg.signalSizeLineEdit.text())
+            lastPath = self.lastSimulationSetup["lastPath"]
+            self.lastSimulationSetup = dict(
+              isView          = dlg.viewCheckBox.isChecked(),
+              color           = color,
+              dB              = DB,
+              step            = step,
+              field           = field,
+              isReceiverPlot  = isReceiverPlot,
+              isEnableFields  = isEnableFields,
+              isEnableSignals = isEnableSignals,
+              isEnableImages  = isEnableImages,
+              isEnableNumPy   = isEnableNumPy,
+              isEnableVolume  = isEnableVolume,
+              sensorShape     = sensorShape,
+              sensorPlacement = sensorPlacement,
+              sensorSize      = sensorSize,
+              lastPath        = lastPath           
+            )
+            if isView:
+                self.GraphicView.setColormap(color)
+
+            if dlg.basename is not None:
+                self.SimNDT_SnapShots = SnapShots(True, Step=step, Filename = dlg.basename, dB = DB, 
+                                                  Color = color,
+                                                  Field = field, 
+                                                  enableFields=isEnableFields,
+                                                  enableSignals=isEnableSignals,
+                                                  enableImages=isEnableImages,
+                                                  enableNumPy=isEnableNumPy,
+                                                  enableVolume=isEnableVolume,
+                                                  enableView=isView,
+                                                  sensorShape = sensorShape,
+                                                  sensorPlacement = sensorPlacement,
+                                                  sensorSize = sensorSize )
+            else:
+                self.SimNDT_SnapShots = SnapShots(dB = DB,
+                                                  enableFields=isEnableFields,
+                                                  enableSignals=isEnableSignals,
+                                                  enableImages=isEnableImages,
+                                                  enableNumPy=isEnableNumPy,
+                                                  enableVolume=isEnableVolume,
+                                                  enableView=isView)
+
+
+            self.runEngine(isView, isReceiverPlot)
+
+
+
+    def runEngine(self, isView=False, isReceiverPlot=False):
+
+
+        self.dirty = True
+
+        self.menuFile.menuAction().setEnabled(False)
+        self.menuConfiguration.menuAction().setEnabled(False)
+        self.menuPlotting_Tools.menuAction().setEnabled(False)
+        self.menuTools.menuAction().setEnabled(False)
+        HelperMethods.setEnabled(self.fileMenuActions, False)
+        HelperMethods.setEnabled(self.geometryMenuActions, False)
+        HelperMethods.setEnabled(self.inspectionMenuActions, False)
+        HelperMethods.setEnabled(self.simulationMenuActions, False)
+
+
+        self.statusBarWidget.startSimulation(self.SimNDT_Simulation.TimeSteps)
+
+        if isView:
+            self.statusBarWidget.viewOn()
+        else:
+            self.statusBarWidget.updateGeometryFrame()
+
+
+        simPack = SimPack(self.SimNDT_Scenario, self.SimNDT_Materials, self.SimNDT_Boundaries,
+                          self.SimNDT_Inspection, self.SimNDT_Source, self.SimNDT_Transducers, self.SimNDT_Signal,
+                          self.SimNDT_Simulation)
+
+        engine = EngineController(simPack, isView, isReceiverPlot, self.statusBarWidget, self)
+
+
+        state = engine.run()
+
+
+        self.statusBarWidget.endSimulation()
+
+        self.PauseSimulation = False
+
+        if state == "Stop":
+            self.StopSimulation = False
+            self.status.showMessage("Stop by User!!!!!", 15000)
+            message = WarningParms("Stop by User!!!")
+            message.exec_()
+        else:
+            self.status.showMessage("Done", 15000)
+            message = DoneParms()
+            message.exec_()
+
+
+
+        self.menuFile.menuAction().setEnabled(True)
+        self.menuConfiguration.menuAction().setEnabled(True)
+        self.menuPlotting_Tools.menuAction().setEnabled(True)
+        self.menuTools.menuAction().setEnabled(True)
+        HelperMethods.setEnabled(self.fileMenuActions, True)
+        HelperMethods.setEnabled(self.geometryMenuActions, True)
+        HelperMethods.setEnabled(self.inspectionMenuActions, True)
+        HelperMethods.setEnabled(self.simulationMenuActions, True)
+
+
+        self.setPlotInspectionsMenu()
+        self.SimNDT_Receivers = Receivers(self.SimNDT_Inspection.Name)
+        self.SimNDT_Receivers.setReceivers(engine)
+
+
+    def generateVideo(self):
+
+        dlg = GenerateVideo()
+        dlg.exec_()
+
+
+    def twoPhaseModelDryCase(self):
+
+        dlg = TwoPhaseModelDryCaseDialog(None, self.SimNDT_Scenario, self.SimNDT_ConcreteMicrostructure)
+        if dlg.exec_():
+
+            self.SimNDT_Scenario.I = dlg.image
+            self.SimNDT_Scenario.resetBoundary()
+            self.SimNDT_ObjectList = None
+            self.SimNDT_Check = False
+            self.dirty = True
+
+
+            self.SimNDT_ConcreteMicrostructure = dlg.concrete2PhaseObject
+
+
+
+
+            self.updateUI()
+
+
+
+
+    def threePhaseModelDryCase(self):
+
+        dlg = ThreePhaseModelDryCaseDialog(None, self.SimNDT_Scenario, self.SimNDT_ConcreteMicrostructure)
+        if dlg.exec_():
+
+            self.SimNDT_Scenario.I = dlg.image
+            self.SimNDT_Scenario.resetBoundary()
+            self.SimNDT_ObjectList = None
+            self.SimNDT_Check = False
+            self.dirty = True
+
+            self.SimNDT_ConcreteMicrostructure = dlg.concrete3PhaseObject
+
+            self.updateUI()
+
+
+    def twoPhaseModelImmersionCase(self):
+
+        dlg = TwoPhaseModelImmersionCaseDialog(None, self.SimNDT_Scenario, self.SimNDT_ConcreteMicrostructure)
+        if dlg.exec_():
+
+            self.SimNDT_Scenario.I = dlg.image
+            self.SimNDT_Scenario.resetBoundary()
+            self.SimNDT_ObjectList = None
+            self.SimNDT_Check = False
+            self.dirty = True
+
+            self.SimNDT_ConcreteMicrostructure = dlg.concrete2PhaseObjectImmersion
+
+
+            self.updateUI()
+
+
+    def threePhaseModelImmersionCase(self):
+
+        dlg = ThreePhaseModelImmersionCaseDialog(None, self.SimNDT_Scenario, self.SimNDT_ConcreteMicrostructure)
+        if dlg.exec_():
+
+            self.SimNDT_Scenario.I = dlg.image
+            self.SimNDT_Scenario.resetBoundary()
+            self.SimNDT_ObjectList = None
+            self.SimNDT_Check = False
+            self.dirty = True
+
+            self.SimNDT_ConcreteMicrostructure = dlg.concrete3PhaseObjectImmersion
+
+            self.updateUI()
+