OASYS2-SRW 1.0.0__tar.gz

oasys2_srw-1.0.0/LICENSE

@@ -0,0 +1,28 @@
+BSD 3-Clause License
+
+Copyright (c) 2025, Organization managing the OASYS project
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

oasys2_srw-1.0.0/MANIFEST.in

@@ -0,0 +1,6 @@
+recursive-include orangecontrib *.pyx
+recursive-include orangecontrib *.py
+
+recursive-include orangecontrib/srw/widgets *.png *.svg *.ico *.jpg *.bmp
+
+include README.txt

oasys2_srw-1.0.0/OASYS2_SRW.egg-info/PKG-INFO

@@ -0,0 +1,37 @@
+Metadata-Version: 2.4
+Name: OASYS2-SRW
+Version: 1.0.0
+Summary: SRW in OASYS2
+Home-page: https://github.com/lucarebuffi/OASYS-SRW
+Download-URL: https://github.com/lucarebuffi/OASYS-SRW
+Author: Luca Rebuffi
+Author-email: lrebuffi@anl.gov
+License: GPLv3
+Keywords: waveoptics,simulator,oasys2
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Environment :: X11 Applications :: Qt
+Classifier: Environment :: Console
+Classifier: Environment :: Plugins
+Classifier: Programming Language :: Python :: 3
+Classifier: Intended Audience :: Science/Research
+License-File: LICENSE
+Requires-Dist: oasys2>=0.0.1
+Requires-Dist: syned-gui-2>=1.0.0
+Requires-Dist: wofrysrw>=1.1.31
+Requires-Dist: xoppylib>=1.0.38
+Requires-Dist: srwpy==4.1.1
+Requires-Dist: scikit-image
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: download-url
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: summary
+
+# OASYS-SRW
+SRW widgets for OASYS

oasys2_srw-1.0.0/OASYS2_SRW.egg-info/SOURCES.txt

@@ -0,0 +1,137 @@
+LICENSE
+MANIFEST.in
+README.md
+pyproject.toml
+setup.py
+OASYS2_SRW.egg-info/PKG-INFO
+OASYS2_SRW.egg-info/SOURCES.txt
+OASYS2_SRW.egg-info/dependency_links.txt
+OASYS2_SRW.egg-info/entry_points.txt
+OASYS2_SRW.egg-info/not-zip-safe
+OASYS2_SRW.egg-info/requires.txt
+OASYS2_SRW.egg-info/top_level.txt
+orangecontrib/__init__.py
+orangecontrib/srw/__init__.py
+orangecontrib/srw/menu/__init__.py
+orangecontrib/srw/menu/srw_tools_menu.py
+orangecontrib/srw/util/__init__.py
+orangecontrib/srw/util/python_script.py
+orangecontrib/srw/util/scanning.py
+orangecontrib/srw/util/srw_objects.py
+orangecontrib/srw/util/srw_util.py
+orangecontrib/srw/widgets/__init__.py
+orangecontrib/srw/widgets/gui/__init__.py
+orangecontrib/srw/widgets/gui/ow_srw_absorber.py
+orangecontrib/srw/widgets/gui/ow_srw_crystal.py
+orangecontrib/srw/widgets/gui/ow_srw_grating.py
+orangecontrib/srw/widgets/gui/ow_srw_mirror.py
+orangecontrib/srw/widgets/gui/ow_srw_optical_element.py
+orangecontrib/srw/widgets/gui/ow_srw_power_density_viewer.py
+orangecontrib/srw/widgets/gui/ow_srw_source.py
+orangecontrib/srw/widgets/gui/ow_srw_wavefront_viewer.py
+orangecontrib/srw/widgets/gui/ow_srw_widget.py
+orangecontrib/srw/widgets/gui/misc/crystal_usage.png
+orangecontrib/srw/widgets/gui/misc/dabam_height_profile_usage.png
+orangecontrib/srw/widgets/gui/misc/height_error_profile_usage.png
+orangecontrib/srw/widgets/gui/misc/propagator_info.png
+orangecontrib/srw/widgets/gui/misc/reflectivity_generator_usage.png
+orangecontrib/srw/widgets/light_sources/__init__.py
+orangecontrib/srw/widgets/light_sources/ow_srw_3d_light_source.py
+orangecontrib/srw/widgets/light_sources/ow_srw_bending_magnet.py
+orangecontrib/srw/widgets/light_sources/ow_srw_gaussian_source.py
+orangecontrib/srw/widgets/light_sources/ow_srw_power_density.py
+orangecontrib/srw/widgets/light_sources/ow_srw_radiation.py
+orangecontrib/srw/widgets/light_sources/ow_srw_spectrum.py
+orangecontrib/srw/widgets/light_sources/ow_srw_undulator.py
+orangecontrib/srw/widgets/light_sources/icons/3d.png
+orangecontrib/srw/widgets/light_sources/icons/bending_magnet.png
+orangecontrib/srw/widgets/light_sources/icons/gaussian_source.png
+orangecontrib/srw/widgets/light_sources/icons/power_density.png
+orangecontrib/srw/widgets/light_sources/icons/radiation.png
+orangecontrib/srw/widgets/light_sources/icons/source.png
+orangecontrib/srw/widgets/light_sources/icons/spectrum.png
+orangecontrib/srw/widgets/light_sources/icons/undulator.png
+orangecontrib/srw/widgets/light_sources/misc/symmetry.png
+orangecontrib/srw/widgets/loops/__init__.py
+orangecontrib/srw/widgets/loops/ow_histogram_scan.py
+orangecontrib/srw/widgets/loops/ow_multiple_height_profile_simulator_S.py
+orangecontrib/srw/widgets/loops/ow_multiple_height_profile_simulator_T.py
+orangecontrib/srw/widgets/loops/ow_multiple_oasys_data_converter.py
+orangecontrib/srw/widgets/loops/ow_scan_file_node_point.py
+orangecontrib/srw/widgets/loops/ow_scan_variable_node_point.py
+orangecontrib/srw/widgets/loops/ow_srw_accumulation_point.py
+orangecontrib/srw/widgets/loops/icons/accumulation.png
+orangecontrib/srw/widgets/loops/icons/cycle_file.png
+orangecontrib/srw/widgets/loops/icons/cycle_variable.png
+orangecontrib/srw/widgets/loops/icons/histogram.png
+orangecontrib/srw/widgets/loops/icons/loops.png
+orangecontrib/srw/widgets/loops/icons/multiple_oasys_data_converter.png
+orangecontrib/srw/widgets/loops/icons/simulator_S.png
+orangecontrib/srw/widgets/loops/icons/simulator_T.png
+orangecontrib/srw/widgets/native/__init__.py
+orangecontrib/srw/widgets/native/ow_srw_intensity_plotter.py
+orangecontrib/srw/widgets/native/ow_srw_me_degcoh_plotter.py
+orangecontrib/srw/widgets/native/ow_srw_python_script_me.py
+orangecontrib/srw/widgets/native/ow_srw_python_script_se.py
+orangecontrib/srw/widgets/native/icons/degcoh.png
+orangecontrib/srw/widgets/native/icons/intensity.png
+orangecontrib/srw/widgets/native/icons/native.png
+orangecontrib/srw/widgets/native/icons/python_script_me.png
+orangecontrib/srw/widgets/native/icons/python_script_se.png
+orangecontrib/srw/widgets/native/util/__init__.py
+orangecontrib/srw/widgets/native/util/native_util.py
+orangecontrib/srw/widgets/optical_elements/__init__.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_aperture.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_back_propagation.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_crl.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_elliptical_mirror.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_filter.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_ideal_lens.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_obstacle.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_plane_crystal.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_plane_grating.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_plane_mirror.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_screen.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_spherical_mirror.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_toroidal_mirror.py
+orangecontrib/srw/widgets/optical_elements/ow_srw_zone_plate.py
+orangecontrib/srw/widgets/optical_elements/icons/backpropagation.png
+orangecontrib/srw/widgets/optical_elements/icons/beam_stopper.png
+orangecontrib/srw/widgets/optical_elements/icons/crl.png
+orangecontrib/srw/widgets/optical_elements/icons/ellipsoid_grating.png
+orangecontrib/srw/widgets/optical_elements/icons/ellipsoid_mirror.png
+orangecontrib/srw/widgets/optical_elements/icons/filter.png
+orangecontrib/srw/widgets/optical_elements/icons/ideal_lens.png
+orangecontrib/srw/widgets/optical_elements/icons/opticalElements.png
+orangecontrib/srw/widgets/optical_elements/icons/plane_crystal.png
+orangecontrib/srw/widgets/optical_elements/icons/plane_grating.png
+orangecontrib/srw/widgets/optical_elements/icons/plane_mirror.png
+orangecontrib/srw/widgets/optical_elements/icons/screen.png
+orangecontrib/srw/widgets/optical_elements/icons/slit.png
+orangecontrib/srw/widgets/optical_elements/icons/spherical_mirror.png
+orangecontrib/srw/widgets/optical_elements/icons/toroidal_mirror.png
+orangecontrib/srw/widgets/optical_elements/icons/zone_plate.png
+orangecontrib/srw/widgets/tools/__init__.py
+orangecontrib/srw/widgets/tools/ow_beamline_renderer.py
+orangecontrib/srw/widgets/tools/ow_dabam_height_profile.py
+orangecontrib/srw/widgets/tools/ow_from_wofry_wavefront.py
+orangecontrib/srw/widgets/tools/ow_height_profile_simulator.py
+orangecontrib/srw/widgets/tools/ow_oasys_data_converter.py
+orangecontrib/srw/widgets/tools/ow_srw_info.py
+orangecontrib/srw/widgets/tools/ow_srw_reflectivity_generator.py
+orangecontrib/srw/widgets/tools/ow_srw_wavefront_file_reader.py
+orangecontrib/srw/widgets/tools/ow_srw_wavefront_file_writer.py
+orangecontrib/srw/widgets/tools/ow_srw_wavefront_to_wofry.py
+orangecontrib/srw/widgets/tools/icons/cycle.png
+orangecontrib/srw/widgets/tools/icons/dabam.png
+orangecontrib/srw/widgets/tools/icons/file_reader.png
+orangecontrib/srw/widgets/tools/icons/file_writer.png
+orangecontrib/srw/widgets/tools/icons/from_wofry_wavefront_2d.png
+orangecontrib/srw/widgets/tools/icons/info.png
+orangecontrib/srw/widgets/tools/icons/oasys_data_converter.png
+orangecontrib/srw/widgets/tools/icons/pin.png
+orangecontrib/srw/widgets/tools/icons/reflectivity.png
+orangecontrib/srw/widgets/tools/icons/renderer.png
+orangecontrib/srw/widgets/tools/icons/simulator.png
+orangecontrib/srw/widgets/tools/icons/to_wofry_wavefront_2d.png
+orangecontrib/srw/widgets/tools/icons/tools.png

oasys2_srw-1.0.0/OASYS2_SRW.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

oasys2_srw-1.0.0/OASYS2_SRW.egg-info/entry_points.txt

@@ -0,0 +1,12 @@
+[oasys2.addons]
+SRW = orangecontrib.srw
+
+[oasys2.menus]
+srwmenu = orangecontrib.srw.menu
+
+[oasys2.widgets]
+SRW Light Sources = orangecontrib.srw.widgets.light_sources
+SRW Loops = orangecontrib.srw.widgets.loops
+SRW Native = orangecontrib.srw.widgets.native
+SRW Optical Elements = orangecontrib.srw.widgets.optical_elements
+SRW Tools = orangecontrib.srw.widgets.tools

oasys2_srw-1.0.0/OASYS2_SRW.egg-info/not-zip-safe

@@ -0,0 +1 @@
+

oasys2_srw-1.0.0/OASYS2_SRW.egg-info/requires.txt

@@ -0,0 +1,6 @@
+oasys2>=0.0.1
+syned-gui-2>=1.0.0
+wofrysrw>=1.1.31
+xoppylib>=1.0.38
+srwpy==4.1.1
+scikit-image

oasys2_srw-1.0.0/OASYS2_SRW.egg-info/top_level.txt

@@ -0,0 +1 @@
+orangecontrib

oasys2_srw-1.0.0/PKG-INFO

@@ -0,0 +1,37 @@
+Metadata-Version: 2.4
+Name: OASYS2-SRW
+Version: 1.0.0
+Summary: SRW in OASYS2
+Home-page: https://github.com/lucarebuffi/OASYS-SRW
+Download-URL: https://github.com/lucarebuffi/OASYS-SRW
+Author: Luca Rebuffi
+Author-email: lrebuffi@anl.gov
+License: GPLv3
+Keywords: waveoptics,simulator,oasys2
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Environment :: X11 Applications :: Qt
+Classifier: Environment :: Console
+Classifier: Environment :: Plugins
+Classifier: Programming Language :: Python :: 3
+Classifier: Intended Audience :: Science/Research
+License-File: LICENSE
+Requires-Dist: oasys2>=0.0.1
+Requires-Dist: syned-gui-2>=1.0.0
+Requires-Dist: wofrysrw>=1.1.31
+Requires-Dist: xoppylib>=1.0.38
+Requires-Dist: srwpy==4.1.1
+Requires-Dist: scikit-image
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: download-url
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: summary
+
+# OASYS-SRW
+SRW widgets for OASYS

oasys2_srw-1.0.0/README.md

@@ -0,0 +1,2 @@
+# OASYS-SRW
+SRW widgets for OASYS

oasys2_srw-1.0.0/orangecontrib/srw/menu/__init__.py

@@ -0,0 +1 @@
+MENU = "SRW MENU"

oasys2_srw-1.0.0/orangecontrib/srw/menu/srw_tools_menu.py

@@ -0,0 +1,225 @@
+
+from orangecanvas.scheme.link import SchemeLink
+from oasys2.canvas.menus.menu import OMenu
+
+from wofry.propagator.propagator import PropagationManager
+from wofrysrw.propagator.propagators2D.srw_fresnel_native import SRW_APPLICATION
+from wofrysrw.propagator.propagators2D.srw_propagation_mode import SRWPropagationMode
+
+from orangecontrib.srw.util.srw_util import showWarningMessage, showCriticalMessage
+from orangecontrib.srw.widgets.optical_elements.ow_srw_screen import OWSRWScreen
+from orangecontrib.srw.widgets.native.ow_srw_intensity_plotter import OWSRWIntensityPlotter
+from orangecontrib.srw.widgets.native.ow_srw_me_degcoh_plotter import OWSRWDegCohPlotter
+from orangecontrib.srw.widgets.loops.ow_srw_accumulation_point import OWSRWAccumulationPoint
+from orangecontrib.srw.widgets.gui.ow_srw_widget import SRWWidget
+
+class SRWToolsMenu(OMenu):
+
+    def __init__(self):
+        super().__init__(name="SRW")
+
+        self.openContainer()
+        self.addContainer("Propagation Mode")
+        self.addSubMenu("Element by Element (Wofry)")
+        self.addSubMenu("Element by Element (SRW Native)")
+        self.addSubMenu("Whole beamline at Final Screen (SRW Native)")
+        self.addSeparator()
+        self.addSubMenu("Disable Wavefront Propagation on all the Final Screens")
+        self.closeContainer()
+        self.openContainer()
+        self.addContainer("Plotting")
+        self.addSubMenu("Select Plots \'No\' on all Source and O.E. widgets")
+        self.addSubMenu("Select Plots \'Yes\' on all Source and O.E. widgets")
+        self.closeContainer()
+        self.openContainer()
+        self.addContainer("Cumulative Loops")
+        self.addSubMenu("Reload Saved Data on the all the Accumulation Points")
+        self.closeContainer()
+
+    def executeAction_1(self, action):
+        try:
+            propagation_mode =  PropagationManager.Instance().get_propagation_mode(SRW_APPLICATION)
+
+            try:
+                PropagationManager.Instance().set_propagation_mode(SRW_APPLICATION, SRWPropagationMode.STEP_BY_STEP_WOFRY)
+
+                self.set_srw_live_propagation_mode()
+
+                showWarningMessage("Propagation Mode: Element by Element (Wofry)")
+            except Exception as exception:
+                showCriticalMessage(exception.args[0])
+
+                try:
+                    PropagationManager.Instance().set_propagation_mode(SRW_APPLICATION, propagation_mode)
+                    self.set_srw_live_propagation_mode()
+                except:
+                    pass
+        except:
+            pass
+
+    def executeAction_2(self, action):
+        try:
+            propagation_mode =  PropagationManager.Instance().get_propagation_mode(SRW_APPLICATION)
+
+            try:
+                PropagationManager.Instance().set_propagation_mode(SRW_APPLICATION, SRWPropagationMode.STEP_BY_STEP)
+
+                self.set_srw_live_propagation_mode()
+
+                showWarningMessage("Propagation Mode: Element by Element (SRW Native)")
+            except Exception as exception:
+                showCriticalMessage(exception.args[0])
+
+                try:
+                    PropagationManager.Instance().set_propagation_mode(SRW_APPLICATION, propagation_mode)
+                    self.set_srw_live_propagation_mode()
+                except:
+                    pass
+        except:
+            pass
+
+    def executeAction_3(self, action):
+        try:
+            propagation_mode =  PropagationManager.Instance().get_propagation_mode(SRW_APPLICATION)
+
+            try:
+                PropagationManager.Instance().set_propagation_mode(SRW_APPLICATION, SRWPropagationMode.WHOLE_BEAMLINE)
+
+                self.set_srw_live_propagation_mode()
+
+                showWarningMessage("Propagation Mode: Whole beamline at Final Screen (SRW Native)")
+            except Exception as exception:
+                showCriticalMessage(exception.args[0])
+
+                try:
+                    PropagationManager.Instance().set_propagation_mode(SRW_APPLICATION, propagation_mode)
+                    self.set_srw_live_propagation_mode()
+                except:
+                    pass
+        except:
+            pass
+
+
+
+    def executeAction_4(self, action):
+        try:
+            for node in self.canvas_main_window.current_document().scheme().nodes:
+                widget = self.canvas_main_window.current_document().scheme().widget_for_node(node)
+
+                if isinstance(widget, OWSRWScreen):
+                    if hasattr(widget, "is_final_screen") and hasattr(widget, "set_is_final_screen"):
+                        if (PropagationManager.Instance().get_propagation_mode(SRW_APPLICATION) != SRWPropagationMode.WHOLE_BEAMLINE):
+                            raise Exception("Action possibile only while Propagation Mode: Whole beamline at Final Screen (SRW Native)")
+
+                        if hasattr(widget, "show_view_box") and getattr(widget, "show_view_box"):
+                            widget.is_final_screen = 0
+                            widget.set_is_final_screen()
+        except Exception as exception:
+            showCriticalMessage(exception.args[0])
+
+    def executeAction_5(self, action):
+        try:
+            for node in self.canvas_main_window.current_document().scheme().nodes:
+                widget = self.canvas_main_window.current_document().scheme().widget_for_node(node)
+
+                if isinstance(widget, SRWWidget) and not (isinstance(widget, OWSRWIntensityPlotter) or
+                                                          isinstance(widget, OWSRWDegCohPlotter) or
+                                                          isinstance(widget, OWSRWAccumulationPoint)):
+                    if hasattr(widget, "view_type") and hasattr(widget, "set_PlotQuality"):
+                        if (PropagationManager.Instance().get_propagation_mode(SRW_APPLICATION) == SRWPropagationMode.WHOLE_BEAMLINE):
+                            raise Exception("Action not possibile while Propagation Mode: Whole beamline at Final Screen (SRW Native)")
+
+                        if hasattr(widget, "show_view_box") and getattr(widget, "show_view_box"):
+                            widget.view_type = 0
+                            widget.set_PlotQuality()
+
+        except Exception as exception:
+            showCriticalMessage(exception.args[0])
+
+    def executeAction_6(self, action):
+        try:
+            for node in self.canvas_main_window.current_document().scheme().nodes:
+                widget = self.canvas_main_window.current_document().scheme().widget_for_node(node)
+
+                if isinstance(widget, SRWWidget) and not (isinstance(widget, OWSRWIntensityPlotter) or
+                                                          isinstance(widget, OWSRWDegCohPlotter) or
+                                                          isinstance(widget, OWSRWAccumulationPoint)):
+                    if hasattr(widget, "view_type") and hasattr(widget, "set_PlotQuality"):
+                        if (PropagationManager.Instance().get_propagation_mode(SRW_APPLICATION) == SRWPropagationMode.WHOLE_BEAMLINE):
+                            raise Exception("Action not possibile while Propagation Mode: Whole beamline at Final Screen (SRW Native)")
+
+                    if hasattr(widget, "show_view_box") and getattr(widget, "show_view_box"):
+                        widget.view_type = 1
+                        widget.set_PlotQuality()
+
+        except Exception as exception:
+            showCriticalMessage(exception.args[0])
+
+    def executeAction_7(self, action):
+        try:
+            for node in self.canvas_main_window.current_document().scheme().nodes:
+                widget = self.canvas_main_window.current_document().scheme().widget_for_node(node)
+
+                if isinstance(widget, OWSRWAccumulationPoint) and widget.autosave == 1: widget.reload_autosave_file()
+        except Exception as exception:
+            showCriticalMessage(exception.args[0])
+
+   #################################################################
+
+    def set_srw_live_propagation_mode(self):
+        for node in self.canvas_main_window.current_document().scheme().nodes:
+            widget = self.canvas_main_window.current_document().scheme().widget_for_node(node)
+
+            if hasattr(widget, "set_srw_live_propagation_mode"):
+                widget.set_srw_live_propagation_mode()
+
+                if (PropagationManager.Instance().get_propagation_mode(SRW_APPLICATION) == SRWPropagationMode.WHOLE_BEAMLINE):
+                    if not (isinstance(widget, OWSRWScreen) or
+                            isinstance(widget, OWSRWIntensityPlotter) or
+                            isinstance(widget, OWSRWDegCohPlotter) or
+                            isinstance(widget, OWSRWAccumulationPoint)) \
+                            or getattr(widget, "is_final_screen") == False:
+                        if hasattr(widget, "view_type") and hasattr(widget, "set_PlotQuality"):
+                            if hasattr(widget, "show_view_box") and getattr(widget, "show_view_box"):
+                                widget.view_type = 0
+                                widget.set_PlotQuality()
+
+                if isinstance(widget, OWSRWScreen): widget.set_is_final_screen()
+
+
+    #################################################################
+    #
+    # SCHEME MANAGEMENT
+    #
+    #################################################################
+
+    def getWidgetFromNode(self, node):
+        return self.canvas_main_window.current_document().scheme().widget_for_node(node)
+
+    def createLinks(self, nodes):
+        previous_node = None
+        for node in nodes:
+            if not (isinstance(node, str)) and not previous_node is None and not (isinstance(previous_node, str)):
+                link = SchemeLink(source_node=previous_node, source_channel="Beam", sink_node=node, sink_channel="Input Beam")
+                self.canvas_main_window.current_document().addLink(link=link)
+            previous_node = node
+
+    def getWidgetDesc(self, widget_name):
+        return self.canvas_main_window.widget_registry.widget(widget_name)
+
+    def createNewNode(self, widget_desc):
+        return self.canvas_main_window.current_document().createNewNode(widget_desc)
+
+    def createNewNodeAndWidget(self, widget_desc):
+        messages = []
+
+        try:
+            node = self.createNewNode(widget_desc)
+            widget = self.getWidgetFromNode(node)
+
+            # here you can put values on the attrubutes
+
+        except Exception as exception:
+            messages.append(exception.args[0])
+
+        return widget, node, messages