Changeset a58706d in sasview for src/sas/sasgui/perspectives/file_converter/converter_panel.py

Timestamp:

Jul 13, 2016 5:56:54 AM (8 years ago)

Author:

lewis

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

5afe77f

Parents:

77d92cd

Message:

Implement basic file conversion

File:

• src/sas/sasgui/perspectives/file_converter/converter_panel.py (modified) (5 diffs)

src/sas/sasgui/perspectives/file_converter/converter_panel.py

@@ -5 +5 @@
 import wx
 import sys
+import numpy as np
 from wx.lib.scrolledpanel import ScrolledPanel
 from sas.sasgui.guiframe.panel_base import PanelBase
 from sas.sasgui.perspectives.calculator import calculator_widgets as widget
+from sas.sasgui.guiframe.events import StatusEvent
+from sas.sasgui.guiframe.dataFitting import Data1D
+from sas.sascalc.dataloader.readers.cansas_reader import Reader as CansasReader
 
 # Panel size
@@ -14 +18 @@
     _STATICBOX_WIDTH = 410
     _BOX_WIDTH = 200
-    PANEL_SIZE = 440
+    PANEL_SIZE = 460
     FONT_VARIANT = 0
 else:
@@ -20 +24 @@
     _STATICBOX_WIDTH = 430
     _BOX_WIDTH = 200
-    PANEL_SIZE = 460
+    PANEL_SIZE = 480
     FONT_VARIANT = 1
 
@@ -28 +32 @@
         ScrolledPanel.__init__(self, parent, *args, **kwargs)
         PanelBase.__init__(self)
-
+        self.SetupScrolling()
         self.SetWindowVariant(variant=FONT_VARIANT)
 
@@ -34 +38 @@
         self.parent = parent
 
-        self._do_layout
+        self.q_input = None
+        self.iq_input = None
+        self.output = None
+
+        self._do_layout()
         self.SetAutoLayout(True)
         self.Layout()
 
+    def convert_to_cansas(self, data, filename):
+        reader = CansasReader()
+        reader.write(filename, data)
+
+    def extract_data(self, filename):
+        data = np.loadtxt(filename, dtype=str)
+
+        is_float = True
+        try:
+            float(data[0])
+        except:
+            is_float = False
+
+        if not is_float:
+            end_char = data[0][-1]
+            # If lines end with comma or semi-colon, trim the last character
+            if end_char == ',' or end_char == ';':
+                data = map(lambda s: s[0:-1], data)
+            else:
+                msg = ("Error reading {}: Lines must end with a digit, comma "
+                    "or semi-colon").format(filename.split('\\')[-1])
+                raise Exception(msg)
+
+        return np.array(data, dtype=np.float32)
+
+    def on_convert(self, event):
+        try:
+            qdata = self.extract_data(self.q_input.GetPath())
+            iqdata = self.extract_data(self.iq_input.GetPath())
+        except Exception as ex:
+            msg = str(ex)
+            wx.PostEvent(self.parent.manager.parent,
+                StatusEvent(status=msg, info='error'))
+            return
+
+        data = Data1D(x=qdata, y=iqdata)
+        self.convert_to_cansas(data, self.output.GetPath())
+        wx.PostEvent(self.parent.manager.parent,
+            StatusEvent(status="Conversion completed."))
+
     def _do_layout(self):
-        # TODO: Get this working
-        vbox = wx.BoxSizer(wx.VERTICAL)
+        vbox = wx.GridBagSizer(wx.VERTICAL)
 
-        inputsection = wx.StaticBox(self, -1, "Input File")
-        inputsection_sizer = wx.StaticBoxSizer(inputsection, wx.VERTICAL)
+        instructions = ("Select the 1 column ASCII files containing the Q Axis"
+            "and Intensity data, chose where to save the converted file, then "
+            "click Convert to convert them to CanSAS XML format")
+        instruction_label = wx.StaticText(self, -1, "",
+            size=(_STATICBOX_WIDTH+40, 28))
+        instruction_label.SetLabel(instructions)
+        vbox.Add(instruction_label, (0,0), (1,1), wx.ALL, 5)
+
+        section = wx.StaticBox(self, -1)
+        section_sizer = wx.StaticBoxSizer(section, wx.VERTICAL)
+        section_sizer.SetMinSize((_STATICBOX_WIDTH, -1))
 
         input_grid = wx.GridBagSizer(5, 5)
 
-        input_box = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, -1))
-        input_grid.Add(input_box, (0,0), (1,1), wx.LEFT, 5)
+        q_label = wx.StaticText(self, -1, "Q Axis: ")
+        input_grid.Add(q_label, (0,0), (1,1), wx.ALIGN_CENTER_VERTICAL, 5)
 
-        inputsection_sizer.Add(input_grid)
+        self.q_input = wx.FilePickerCtrl(self, -1,
+            size=(_STATICBOX_WIDTH-80, -1),
+            message="Chose the Q Axis data file.")
+        input_grid.Add(self.q_input, (0,1), (1,1), wx.ALL, 5)
 
-        vbox.Add(input_section, (0,0), (0,1), wx.TOP, 15)
+        iq_label = wx.StaticText(self, -1, "Intensity Data: ")
+        input_grid.Add(iq_label, (1,0), (1,1), wx.ALIGN_CENTER_VERTICAL, 5)
+
+        self.iq_input = wx.FilePickerCtrl(self, -1,
+            size=(_STATICBOX_WIDTH-80, -1),
+            message="Chose the Intensity data file.")
+        input_grid.Add(self.iq_input, (1,1), (1,1), wx.ALL, 5)
+
+        output_label = wx.StaticText(self, -1, "Output File: ")
+        input_grid.Add(output_label, (2,0), (1,1), wx.ALIGN_CENTER_VERTICAL, 5)
+
+        self.output = wx.FilePickerCtrl(self, -1,
+            size=(_STATICBOX_WIDTH-80, -1),
+            message="Chose the Intensity data file.",
+            style=wx.FLP_SAVE | wx.FLP_OVERWRITE_PROMPT | wx.FLP_USE_TEXTCTRL,
+            wildcard="*.xml")
+        input_grid.Add(self.output, (2,1), (1,1), wx.ALL, 5)
+
+        convert_btn = wx.Button(self, -1, "Convert")
+        input_grid.Add(convert_btn, (3,0), (1,1), wx.ALL, 5)
+        convert_btn.Bind(wx.EVT_BUTTON, self.on_convert)
+
+        section_sizer.Add(input_grid)
+
+        vbox.Add(section_sizer, (1,0), (1,1), wx.ALL, 5)
 
         vbox.Fit(self)