source: sasview/src/sas/sasgui/perspectives/file_converter/converter_panel.py @ be88076

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since be88076 was be88076, checked in by lewis, 8 years ago

Correctly read and write run_name

  • Property mode set to 100644
File size: 24.8 KB
RevLine 
[77d92cd]1"""
2This module provides a GUI for the file converter
3"""
4
5import wx
6import sys
[05595c4]7import os
[a58706d]8import numpy as np
[77d92cd]9from wx.lib.scrolledpanel import ScrolledPanel
10from sas.sasgui.guiframe.panel_base import PanelBase
11from sas.sasgui.perspectives.calculator import calculator_widgets as widget
[0e11ec7]12from sas.sasgui.perspectives.file_converter.converter_widgets import FileInput
[de0df2c]13from sas.sasgui.perspectives.file_converter.meta_panels import MetadataWindow
14from sas.sasgui.perspectives.file_converter.meta_panels import DetectorPanel
[2a7722b]15from sas.sasgui.perspectives.file_converter.meta_panels import SamplePanel
[55bc56bc]16from sas.sasgui.perspectives.file_converter.meta_panels import SourcePanel
[eb8da5f]17from sas.sasgui.perspectives.file_converter.frame_select_dialog import FrameSelectDialog
[a58706d]18from sas.sasgui.guiframe.events import StatusEvent
[11794f2]19from sas.sasgui.guiframe.documentation_window import DocumentationWindow
[a58706d]20from sas.sasgui.guiframe.dataFitting import Data1D
[e9e777c]21from sas.sascalc.dataloader.data_info import Data2D
[fdbea3c]22from sas.sasgui.guiframe.utils import check_float
[0b1a677]23from sas.sascalc.file_converter.cansas_writer import CansasWriter
24from sas.sascalc.file_converter.otoko_loader import OTOKOLoader
25from sas.sascalc.file_converter.bsl_loader import BSLLoader
[e9e777c]26from sas.sascalc.file_converter.nxcansas_writer import NXcanSASWriter
[f2b3f28]27from sas.sascalc.dataloader.data_info import Detector
[2a7722b]28from sas.sascalc.dataloader.data_info import Sample
[55bc56bc]29from sas.sascalc.dataloader.data_info import Source
[fdbea3c]30from sas.sascalc.dataloader.data_info import Vector
[77d92cd]31
32# Panel size
33if sys.platform.count("win32") > 0:
34    PANEL_TOP = 0
35    _STATICBOX_WIDTH = 410
36    _BOX_WIDTH = 200
[36f4debb]37    PANEL_SIZE = 480
[77d92cd]38    FONT_VARIANT = 0
39else:
40    PANEL_TOP = 60
41    _STATICBOX_WIDTH = 430
42    _BOX_WIDTH = 200
[36f4debb]43    PANEL_SIZE = 500
[77d92cd]44    FONT_VARIANT = 1
45
46class ConverterPanel(ScrolledPanel, PanelBase):
[ba65aff]47    """
48    This class provides the File Converter GUI
49    """
[77d92cd]50
51    def __init__(self, parent, base=None, *args, **kwargs):
52        ScrolledPanel.__init__(self, parent, *args, **kwargs)
53        PanelBase.__init__(self)
[a58706d]54        self.SetupScrolling()
[77d92cd]55        self.SetWindowVariant(variant=FONT_VARIANT)
56
57        self.base = base
58        self.parent = parent
[de0df2c]59        self.meta_frames = []
[77d92cd]60
[ba65aff]61        # GUI inputs
[a58706d]62        self.q_input = None
63        self.iq_input = None
64        self.output = None
[d6bf064]65        self.radiation_input = None
[35488b2]66        self.convert_btn = None
[d6bf064]67        self.metadata_section = None
[ba65aff]68
[4b862c4]69        self.data_type = "ascii"
[a58706d]70
[ba65aff]71        # Metadata values
[94667e27]72        self.title = ''
73        self.run = ''
74        self.run_name = ''
75        self.instrument = ''
[503cc34]76        self.detector = Detector()
77        self.sample = Sample()
78        self.source = Source()
79        self.properties = ['title', 'run', 'run_name', 'instrument']
80
81        self.detector.name = ''
82        self.source.radiation = 'neutron'
[4b862c4]83
[a58706d]84        self._do_layout()
[77d92cd]85        self.SetAutoLayout(True)
86        self.Layout()
87
[9e9f848]88    def convert_to_cansas(self, frame_data, filepath, single_file):
[ba65aff]89        """
90        Saves an array of Data1D objects to a single CanSAS file with multiple
91        <SasData> elements, or to multiple CanSAS files, each with one
92        <SasData> element.
93
[9e9f848]94        :param frame_data: If single_file is true, an array of Data1D objects.
95        If single_file is false, a dictionary of the form frame_number: Data1D.
96        :param filepath: Where to save the CanSAS file
[ba65aff]97        :param single_file: If true, array is saved as a single file, if false,
98        each item in the array is saved to it's own file
99        """
100        writer = CansasWriter()
[ea2a7348]101        entry_attrs = None
[94667e27]102        if self.run_name != '':
[ea2a7348]103            entry_attrs = { 'name': self.run_name }
[5451a78]104
[94f4518]105        if single_file:
[9e9f848]106            writer.write(filepath, frame_data,
[94f4518]107                sasentry_attrs=entry_attrs)
108        else:
[9e9f848]109            # Folder and base filename
110            [group_path, group_name] = os.path.split(filepath)
111            ext = "." + group_name.split('.')[-1] # File extension
112            for frame_number, frame_data in frame_data.iteritems():
113                # Append frame number to base filename
114                filename = group_name.replace(ext, str(frame_number)+ext)
115                destination = os.path.join(group_path, filename)
116                writer.write(destination, [frame_data],
[94f4518]117                    sasentry_attrs=entry_attrs)
[a58706d]118
[535e181]119    def extract_ascii_data(self, filename):
[ba65aff]120        """
121        Extracts data from a single-column ASCII file
122
123        :param filename: The file to load data from
124        :return: A numpy array containing the extracted data
125        """
[514a00e]126        try:
127            data = np.loadtxt(filename, dtype=str)
128        except:
129            is_bsl = False
130            # Check if file is a BSL or OTOKO header file
131            f = open(filename, 'r')
132            f.readline()
133            f.readline()
134            bsl_metadata = f.readline().strip().split()
135            f.close()
136            if len(bsl_metadata) == 10:
137                msg = ("Error parsing ASII data. {} looks like a BSL or OTOKO "
138                    "header file.")
139                raise Exception(msg.format(os.path.split(filename)[-1]))
[a58706d]140
[ff790b3]141        if len(data.shape) != 1:
142            msg = "Error reading {}: Only one column of data is allowed"
143            raise Exception(msg.format(filename.split('\\')[-1]))
144
[a58706d]145        is_float = True
146        try:
147            float(data[0])
148        except:
149            is_float = False
150
151        if not is_float:
152            end_char = data[0][-1]
153            # If lines end with comma or semi-colon, trim the last character
154            if end_char == ',' or end_char == ';':
155                data = map(lambda s: s[0:-1], data)
156            else:
157                msg = ("Error reading {}: Lines must end with a digit, comma "
158                    "or semi-colon").format(filename.split('\\')[-1])
159                raise Exception(msg)
160
161        return np.array(data, dtype=np.float32)
162
[535e181]163    def extract_otoko_data(self, filename):
[ba65aff]164        """
165        Extracts data from a 1D OTOKO file
166
167        :param filename: The OTOKO file to load the data from
168        :return: A numpy array containing the extracted data
169        """
[535e181]170        loader = OTOKOLoader(self.q_input.GetPath(),
171            self.iq_input.GetPath())
[9c500ab]172        otoko_data = loader.load_otoko_data()
173        qdata = otoko_data.q_axis.data
174        iqdata = otoko_data.data_axis.data
[535e181]175        if len(qdata) > 1:
176            msg = ("Q-Axis file has multiple frames. Only 1 frame is "
177                "allowed for the Q-Axis")
178            wx.PostEvent(self.parent.manager.parent,
179                StatusEvent(status=msg, info="error"))
180            return
181        else:
182            qdata = qdata[0]
183
184        return qdata, iqdata
185
[0b1a677]186    def extract_bsl_data(self, filename):
187        """
188        Extracts data from a 2D BSL file
189
190        :param filename: The header file to extract the data from
191        :return x_data: A 1D array containing all the x coordinates of the data
192        :return y_data: A 1D array containing all the y coordinates of the data
193        :return frame_data: A dictionary of the form frame_number: data, where
194        data is a 2D numpy array containing the intensity data
195        """
196        loader = BSLLoader(filename)
197        frames = [0]
198        should_continue = True
199
200        if loader.n_frames > 1:
201            params = self.ask_frame_range(loader.n_frames)
202            frames = params['frames']
203            if len(frames) == 0:
204                should_continue = False
205        elif loader.n_rasters == 1 and loader.n_frames == 1:
206            message = ("The selected file is an OTOKO file. Please select the "
207            "'OTOKO 1D' option if you wish to convert it.")
208            dlg = wx.MessageDialog(self,
209            message,
210            'Error!',
211            wx.OK | wx.ICON_WARNING)
212            dlg.ShowModal()
213            should_continue = False
214            dlg.Destroy()
215        else:
216            message = ("The selected data file only has 1 frame, it might be"
217                " a multi-frame OTOKO file.\nContinue conversion?")
218            dlg = wx.MessageDialog(self,
219            message,
220            'Warning!',
221            wx.YES_NO | wx.ICON_WARNING)
222            should_continue = (dlg.ShowModal() == wx.ID_YES)
223            dlg.Destroy()
224
225        if not should_continue:
[fe498b83]226            return None
[0b1a677]227
[fe498b83]228        frame_data = loader.load_frames(frames)
[0b1a677]229
[fe498b83]230        return frame_data
[0b1a677]231
[eb8da5f]232    def ask_frame_range(self, n_frames):
[ba65aff]233        """
234        Display a dialog asking the user to input the range of frames they
235        would like to export
236
237        :param n_frames: How many frames the loaded data file has
238        :return: A dictionary containing the parameters input by the user
239        """
[eb8da5f]240        valid_input = False
[bec6cb8]241        _, ext = os.path.splitext(self.output.GetPath())
242        show_single_btn = (ext == '.h5')
243        dlg = FrameSelectDialog(n_frames, show_single_btn)
[eb8da5f]244        frames = None
245        increment = None
[94f4518]246        single_file = True
[eb8da5f]247        while not valid_input:
248            if dlg.ShowModal() == wx.ID_OK:
249                msg = ""
250                try:
251                    first_frame = int(dlg.first_input.GetValue())
252                    last_frame = int(dlg.last_input.GetValue())
253                    increment = int(dlg.increment_input.GetValue())
[bec6cb8]254                    if not show_single_btn:
[05595c4]255                        single_file = dlg.single_btn.GetValue()
256
[eb8da5f]257                    if last_frame < 0 or first_frame < 0:
258                        msg = "Frame values must be positive"
259                    elif increment < 1:
260                        msg = "Increment must be greater than or equal to 1"
261                    elif first_frame > last_frame:
262                        msg = "First frame must be less than last frame"
[05595c4]263                    elif last_frame >= n_frames:
[eb8da5f]264                        msg = "Last frame must be less than {}".format(n_frames)
265                    else:
266                        valid_input = True
267                except:
268                    valid_input = False
269                    msg = "Please enter valid integer values"
270
271                if not valid_input:
272                    wx.PostEvent(self.parent.manager.parent,
273                        StatusEvent(status=msg))
274            else:
[94f4518]275                return { 'frames': [], 'inc': None, 'file': single_file }
[18544f8]276        frames = range(first_frame, last_frame + 1, increment)
[94f4518]277        return { 'frames': frames, 'inc': increment, 'file': single_file }
[eb8da5f]278
[edf0e06]279    def get_metadata(self):
280        # Prepare the metadata for writing to a file
281        if self.run is None:
282            self.run = []
283        elif not isinstance(self.run, list):
284            self.run = [self.run]
[0b1a677]285
[be88076]286        run_name = None
287        if len(self.run) > 0:
288            run_number = self.run[0]
289            run_name = { run_number: self.run_name }
290
[edf0e06]291        if self.title is None:
292            self.title = ''
[e9e777c]293
[edf0e06]294        metadata = {
295            'title': self.title,
296            'run': self.run,
[be88076]297            'run_name': run_name,
[edf0e06]298            'instrument': self.instrument,
299            'detector': [self.detector],
300            'sample': self.sample,
301            'source': self.source
302        }
[535e181]303
[edf0e06]304        return metadata
[a58706d]305
[edf0e06]306    def convert_1d_data(self, qdata, iqdata):
307        """
308        Formats a 1D array of q_axis data and a 2D array of I axis data (where
309        each row of iqdata is a separate row), into an array of Data1D objects
310        """
[535e181]311        frames = []
312        increment = 1
313        single_file = True
[05595c4]314        n_frames = iqdata.shape[0]
[535e181]315        # Standard file has 3 frames: SAS, calibration and WAS
[05595c4]316        if n_frames > 3:
[ba65aff]317            # File has multiple frames - ask the user which ones they want to
318            # export
[05595c4]319            params = self.ask_frame_range(n_frames)
[535e181]320            frames = params['frames']
321            increment = params['inc']
322            single_file = params['file']
323            if frames == []: return
324        else: # Only interested in SAS data
325            frames = [0]
326
[f2b3f28]327        output_path = self.output.GetPath()
[edf0e06]328        metadata = self.get_metadata()
[f2b3f28]329
[9e9f848]330        frame_data = {}
[eb8da5f]331        for i in frames:
332            data = Data1D(x=qdata, y=iqdata[i])
[9e9f848]333            frame_data[i] = data
[94f4518]334        if single_file:
335            # Only need to set metadata on first Data1D object
[9e9f848]336            frame_data = frame_data.values() # Don't need to know frame numbers
[94f4518]337            frame_data[0].filename = output_path.split('\\')[-1]
338            for key, value in metadata.iteritems():
339                setattr(frame_data[0], key, value)
340        else:
341            # Need to set metadata for all Data1D objects
[9e9f848]342            for datainfo in frame_data.values():
[94f4518]343                datainfo.filename = output_path.split('\\')[-1]
344                for key, value in metadata.iteritems():
345                    setattr(datainfo, key, value)
346
[bec6cb8]347        _, ext = os.path.splitext(output_path)
348        if ext == '.xml':
349            self.convert_to_cansas(frame_data, output_path, single_file)
350        else: # ext == '.h5'
351            w = NXcanSASWriter()
352            w.write(frame_data, output_path)
[edf0e06]353
354    def on_convert(self, event):
355        """Called when the Convert button is clicked"""
356        if not self.validate_inputs():
357            return
358
359        self.sample.ID = self.title
360
361        try:
362            if self.data_type == 'ascii':
363                qdata = self.extract_ascii_data(self.q_input.GetPath())
364                iqdata = np.array([self.extract_ascii_data(self.iq_input.GetPath())])
365                self.convert_1d_data(qdata, iqdata)
366            elif self.data_type == 'otoko':
367                qdata, iqdata = self.extract_otoko_data(self.q_input.GetPath())
368                self.convert_1d_data(qdata, iqdata)
369            else: # self.data_type == 'bsl'
370                dataset = self.extract_bsl_data(self.iq_input.GetPath())
371                if dataset is None:
372                    # Cancelled by user
373                    return
374
375                metadata = self.get_metadata()
376                for key, value in metadata.iteritems():
377                    setattr(dataset[0], key, value)
378
379                w = NXcanSASWriter()
380                w.write(dataset, self.output.GetPath())
381        except Exception as ex:
382            msg = str(ex)
383            wx.PostEvent(self.parent.manager.parent,
384                StatusEvent(status=msg, info='error'))
385            return
386
[a58706d]387        wx.PostEvent(self.parent.manager.parent,
388            StatusEvent(status="Conversion completed."))
389
[11794f2]390    def on_help(self, event):
[ba65aff]391        """
392        Show the File Converter documentation
393        """
[11794f2]394        tree_location = ("user/sasgui/perspectives/file_converter/"
395            "file_converter_help.html")
396        doc_viewer = DocumentationWindow(self, -1, tree_location,
397            "", "File Converter Help")
398
[fdbea3c]399    def validate_inputs(self):
400        msg = "You must select a"
[535e181]401        if self.q_input.GetPath() == '' and self.data_type != 'bsl':
[fdbea3c]402            msg += " Q Axis input file."
403        elif self.iq_input.GetPath() == '':
404            msg += "n Intensity input file."
405        elif self.output.GetPath() == '':
[35488b2]406            msg += " destination for the converted file."
[fdbea3c]407        if msg != "You must select a":
408            wx.PostEvent(self.parent.manager.parent,
409                StatusEvent(status=msg, info='error'))
410            return
411
412        return True
413
[af84162]414    def show_detector_window(self, event):
[ba65aff]415        """
[b7c21a7]416        Show the window for inputting Detector metadata
[ba65aff]417        """
[de0df2c]418        if self.meta_frames != []:
419            for frame in self.meta_frames:
420                frame.panel.on_close()
421        detector_frame = MetadataWindow(DetectorPanel,
422            parent=self.parent.manager.parent, manager=self,
[503cc34]423            metadata=self.detector, title='Detector Metadata')
[de0df2c]424        self.meta_frames.append(detector_frame)
425        self.parent.manager.put_icon(detector_frame)
426        detector_frame.Show(True)
[fdbea3c]427
[2a7722b]428    def show_sample_window(self, event):
[ba65aff]429        """
[b7c21a7]430        Show the window for inputting Sample metadata
[ba65aff]431        """
[2a7722b]432        if self.meta_frames != []:
433            for frame in self.meta_frames:
434                frame.panel.on_close()
435        sample_frame = MetadataWindow(SamplePanel,
436            parent=self.parent.manager.parent, manager=self,
[503cc34]437            metadata=self.sample, title='Sample Metadata')
[2a7722b]438        self.meta_frames.append(sample_frame)
439        self.parent.manager.put_icon(sample_frame)
440        sample_frame.Show(True)
441
[55bc56bc]442    def show_source_window(self, event):
[ba65aff]443        """
[b7c21a7]444        Show the window for inputting Source metadata
[ba65aff]445        """
[55bc56bc]446        if self.meta_frames != []:
447            for frame in self.meta_frames:
448                frame.panel.on_close()
449        source_frame = MetadataWindow(SourcePanel,
450            parent=self.parent.manager.parent, manager=self,
[503cc34]451            metadata=self.source, title="Source Metadata")
[55bc56bc]452        self.meta_frames.append(source_frame)
453        self.parent.manager.put_icon(source_frame)
454        source_frame.Show(True)
455
[a027549]456    def on_collapsible_pane(self, event):
[ba65aff]457        """
458        Resize the scrollable area to fit the metadata pane when it's
459        collapsed or expanded
460        """
[a027549]461        self.Freeze()
462        self.SetupScrolling()
463        self.parent.Layout()
464        self.Thaw()
465
[4b862c4]466    def datatype_changed(self, event):
[ba65aff]467        """
468        Update the UI and self.data_type when a data type radio button is
469        pressed
470        """
[4b862c4]471        event.Skip()
472        dtype = event.GetEventObject().GetName()
473        self.data_type = dtype
[535e181]474        if dtype == 'bsl':
[05595c4]475            self.q_input.SetPath("")
[535e181]476            self.q_input.Disable()
[e9e777c]477            self.output.SetWildcard("NXcanSAS HDF5 File (*.h5)|*.h5")
[535e181]478        else:
479            self.q_input.Enable()
[d6bf064]480            self.radiation_input.Enable()
481            self.metadata_section.Enable()
[bec6cb8]482            self.output.SetWildcard("CanSAS 1D (*.xml)|*.xml|NXcanSAS HDF5 File (*.h5)|*.h5")
[4b862c4]483
[55775e8]484    def radiationtype_changed(self, event):
485        event.Skip()
486        rtype = event.GetEventObject().GetValue().lower()
[503cc34]487        self.source.radiation = rtype
[55775e8]488
[f2b3f28]489    def metadata_changed(self, event):
490        event.Skip()
491        textbox = event.GetEventObject()
492        attr = textbox.GetName()
493        value = textbox.GetValue().strip()
[fdbea3c]494
[503cc34]495        setattr(self, attr, value)
[f2b3f28]496
497
[77d92cd]498    def _do_layout(self):
[f2b3f28]499        vbox = wx.BoxSizer(wx.VERTICAL)
[a58706d]500
[f5c52b0]501        instructions = ("Select either single column ASCII files or 1D OTOKO "
502        "files containing the Q-Axis and Intensity-Axis data, or a 2D BSL file"
503        ", then chose where to save the converted file, and click Convert.\n"
[d3e1870]504        "ASCII and OTOKO files can be converted to CanSAS XML, or NXcanSAS "
505        "HDF5 format. OTOKO files can only be converted to NXcanSAS.\nCanSAS "
506        "metadata can also be optionally input below.")
[f5c52b0]507
[5afe77f]508        instruction_label = wx.StaticText(self, -1, instructions,
509            size=(_STATICBOX_WIDTH+40, -1))
510        instruction_label.Wrap(_STATICBOX_WIDTH+40)
[f2b3f28]511        vbox.Add(instruction_label, flag=wx.TOP | wx.LEFT | wx.RIGHT, border=5)
[77d92cd]512
[a58706d]513        section = wx.StaticBox(self, -1)
514        section_sizer = wx.StaticBoxSizer(section, wx.VERTICAL)
515        section_sizer.SetMinSize((_STATICBOX_WIDTH, -1))
[77d92cd]516
517        input_grid = wx.GridBagSizer(5, 5)
518
[4b862c4]519        y = 0
520
[489bb46]521        data_type_label = wx.StaticText(self, -1, "Input Format: ")
522        input_grid.Add(data_type_label, (y,0), (1,1),
523            wx.ALIGN_CENTER_VERTICAL, 5)
524        radio_sizer = wx.BoxSizer(wx.HORIZONTAL)
525        ascii_btn = wx.RadioButton(self, -1, "ASCII", name="ascii",
526            style=wx.RB_GROUP)
527        ascii_btn.Bind(wx.EVT_RADIOBUTTON, self.datatype_changed)
528        radio_sizer.Add(ascii_btn)
529        otoko_btn = wx.RadioButton(self, -1, "OTOKO 1D", name="otoko")
530        otoko_btn.Bind(wx.EVT_RADIOBUTTON, self.datatype_changed)
531        radio_sizer.Add(otoko_btn)
532        input_grid.Add(radio_sizer, (y,1), (1,1), wx.ALL, 5)
533        bsl_btn = wx.RadioButton(self, -1, "BSL 2D", name="bsl")
534        bsl_btn.Bind(wx.EVT_RADIOBUTTON, self.datatype_changed)
535        radio_sizer.Add(bsl_btn)
536        y += 1
537
[13b9a63]538        q_label = wx.StaticText(self, -1, "Q-Axis Data: ")
[4b862c4]539        input_grid.Add(q_label, (y,0), (1,1), wx.ALIGN_CENTER_VERTICAL, 5)
[a58706d]540
541        self.q_input = wx.FilePickerCtrl(self, -1,
542            size=(_STATICBOX_WIDTH-80, -1),
[785fa53]543            message="Chose the Q-Axis data file.",
544            style=wx.FLP_USE_TEXTCTRL | wx.FLP_CHANGE_DIR)
[4b862c4]545        input_grid.Add(self.q_input, (y,1), (1,1), wx.ALL, 5)
546        y += 1
[a58706d]547
[13b9a63]548        iq_label = wx.StaticText(self, -1, "Intensity-Axis Data: ")
[4b862c4]549        input_grid.Add(iq_label, (y,0), (1,1), wx.ALIGN_CENTER_VERTICAL, 5)
[a58706d]550
551        self.iq_input = wx.FilePickerCtrl(self, -1,
552            size=(_STATICBOX_WIDTH-80, -1),
[785fa53]553            message="Chose the Intensity-Axis data file.",
554            style=wx.FLP_USE_TEXTCTRL | wx.FLP_CHANGE_DIR)
[4b862c4]555        input_grid.Add(self.iq_input, (y,1), (1,1), wx.ALL, 5)
556        y += 1
557
[55775e8]558        radiation_label = wx.StaticText(self, -1, "Radiation Type: ")
[489bb46]559        input_grid.Add(radiation_label, (y,0), (1,1), wx.ALIGN_CENTER_VERTICAL, 5)
[d6bf064]560        self.radiation_input = wx.ComboBox(self, -1,
[55775e8]561            choices=["Neutron", "X-Ray", "Muon", "Electron"],
562            name="radiation", style=wx.CB_READONLY, value="Neutron")
[d6bf064]563        self.radiation_input.Bind(wx.EVT_COMBOBOX, self.radiationtype_changed)
564        input_grid.Add(self.radiation_input, (y,1), (1,1), wx.ALL, 5)
[55775e8]565        y += 1
566
[a58706d]567        output_label = wx.StaticText(self, -1, "Output File: ")
[4b862c4]568        input_grid.Add(output_label, (y,0), (1,1), wx.ALIGN_CENTER_VERTICAL, 5)
[a58706d]569
[0e11ec7]570        self.output = FileInput(self,
[bec6cb8]571            wildcard="CanSAS 1D (*.xml)|*.xml|NXcanSAS HDF5 File (*.h5)|*.h5")
[0e11ec7]572        input_grid.Add(self.output.GetCtrl(), (y,1), (1,1), wx.EXPAND | wx.ALL, 5)
[4b862c4]573        y += 1
[a58706d]574
[35488b2]575        self.convert_btn = wx.Button(self, wx.ID_OK, "Stop Converstion")
576        self.convert_btn.SetLabel("Convert")
577        input_grid.Add(self.convert_btn, (y,0), (1,1), wx.ALL, 5)
578        self.convert_btn.Bind(wx.EVT_BUTTON, self.on_convert)
[77d92cd]579
[11794f2]580        help_btn = wx.Button(self, -1, "HELP")
581        input_grid.Add(help_btn, (y,1), (1,1), wx.ALL, 5)
582        help_btn.Bind(wx.EVT_BUTTON, self.on_help)
583
[a58706d]584        section_sizer.Add(input_grid)
[77d92cd]585
[f2b3f28]586        vbox.Add(section_sizer, flag=wx.ALL, border=5)
587
[d6bf064]588        self.metadata_section = wx.CollapsiblePane(self, -1, "Metadata",
[fdbea3c]589            size=(_STATICBOX_WIDTH+40, -1), style=wx.WS_EX_VALIDATE_RECURSIVELY)
[d6bf064]590        metadata_pane = self.metadata_section.GetPane()
[f2b3f28]591        metadata_grid = wx.GridBagSizer(5, 5)
592
[d6bf064]593        self.metadata_section.Bind(wx.EVT_COLLAPSIBLEPANE_CHANGED,
[a027549]594            self.on_collapsible_pane)
595
[f2b3f28]596        y = 0
[503cc34]597        for item in self.properties:
[7c8ddb83]598            # Capitalise each word
599            label_txt = " ".join(
600                [s.capitalize() for s in item.replace('_', ' ').split(' ')])
601            if item == 'run':
602                label_txt = "Run Number"
[fdbea3c]603            label = wx.StaticText(metadata_pane, -1, label_txt,
[f2b3f28]604                style=wx.ALIGN_CENTER_VERTICAL)
[fdbea3c]605            input_box = wx.TextCtrl(metadata_pane, name=item,
[f2b3f28]606                size=(_STATICBOX_WIDTH-80, -1))
607            input_box.Bind(wx.EVT_TEXT, self.metadata_changed)
608            metadata_grid.Add(label, (y,0), (1,1),
609                wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5)
[fdbea3c]610            metadata_grid.Add(input_box, (y,1), (1,2), wx.EXPAND)
[f2b3f28]611            y += 1
612
[fdbea3c]613        detector_label = wx.StaticText(metadata_pane, -1,
614            "Detector:")
615        metadata_grid.Add(detector_label, (y, 0), (1,1), wx.ALL | wx.EXPAND, 5)
[af84162]616        detector_btn = wx.Button(metadata_pane, -1, "Enter Detector Metadata")
617        metadata_grid.Add(detector_btn, (y, 1), (1,1), wx.ALL | wx.EXPAND, 5)
618        detector_btn.Bind(wx.EVT_BUTTON, self.show_detector_window)
[fdbea3c]619        y += 1
620
[2a7722b]621        sample_label = wx.StaticText(metadata_pane, -1, "Sample: ")
622        metadata_grid.Add(sample_label, (y,0), (1,1), wx.ALL | wx.EXPAND, 5)
623        sample_btn = wx.Button(metadata_pane, -1, "Enter Sample Metadata")
624        metadata_grid.Add(sample_btn, (y,1), (1,1), wx.ALL | wx.EXPAND, 5)
625        sample_btn.Bind(wx.EVT_BUTTON, self.show_sample_window)
626        y += 1
627
[55bc56bc]628        source_label = wx.StaticText(metadata_pane, -1, "Source: ")
629        metadata_grid.Add(source_label, (y,0), (1,1), wx.ALL | wx.EXPAND, 5)
630        source_btn = wx.Button(metadata_pane, -1, "Enter Source Metadata")
631        source_btn.Bind(wx.EVT_BUTTON, self.show_source_window)
632        metadata_grid.Add(source_btn, (y,1), (1,1), wx.ALL | wx.EXPAND, 5)
633        y += 1
634
[fdbea3c]635        metadata_pane.SetSizer(metadata_grid)
[f2b3f28]636
[d6bf064]637        vbox.Add(self.metadata_section, proportion=0, flag=wx.ALL, border=5)
[77d92cd]638
639        vbox.Fit(self)
640        self.SetSizer(vbox)
641
642class ConverterWindow(widget.CHILD_FRAME):
[ba65aff]643    """Displays ConverterPanel"""
[77d92cd]644
645    def __init__(self, parent=None, title='File Converter', base=None,
[f5c52b0]646        manager=None, size=(PANEL_SIZE * 1.05, PANEL_SIZE / 1.1),
[77d92cd]647        *args, **kwargs):
648        kwargs['title'] = title
649        kwargs['size'] = size
650        widget.CHILD_FRAME.__init__(self, parent, *args, **kwargs)
651
652        self.manager = manager
653        self.panel = ConverterPanel(self, base=None)
654        self.Bind(wx.EVT_CLOSE, self.on_close)
655        self.SetPosition((wx.LEFT, PANEL_TOP))
656        self.Show(True)
657
658    def on_close(self, event):
659        if self.manager is not None:
660            self.manager.converter_frame = None
661        self.Destroy()
Note: See TracBrowser for help on using the repository browser.