Ejemplo n.º 1
0
 def open_menu(self):
     dlg = FileDialog()
     dlg.open()
     if dlg.return_code == OK:
         self.img.load_image(dlg.path)
         self.update_affine()
         self.update_slice_index()
Ejemplo n.º 2
0
 def open_menu(self):
     dlg = FileDialog()
     dlg.open()
     if dlg.return_code == OK:
         self.img.load_image(dlg.path)
         self.update_affine()
         self.update_slice_index()
Ejemplo n.º 3
0
 def _save_as_button_fired(self):
     dialog = FileDialog(action="save as", wildcard=self.file_wildcard)
     dialog.open()
     if dialog.return_code == OK:
         self.filedir = dialog.directory
         self.filename = dialog.filename
         self._save_to_file()
Ejemplo n.º 4
0
    def _dialog(self, message="Select Folder", new_directory=True, mode='r'):
        """Creates a file dialog box for working with

        Args:
            message (string): Message to display in dialog
            new_file (bool): True if allowed to create new directory

        Returns:
            A directory to be used for the file operation.
        """
        try:
            from enthought.pyface.api import FileDialog, OK
        except ImportError:
            from pyface.api import FileDialog, OK
        # Wildcard pattern to be used in file dialogs.
        file_wildcard = "zip file (*.zip)|All files|*"

        if mode == "r":
            mode2 = "open"
        elif mode == "w":
            mode2 = "save as"

        dlg = FileDialog(action=mode2, wildcard=file_wildcard)
        dlg.open()
        if dlg.return_code == OK:
            self.directory = dlg.path
            self.File = zf.ZipFile(self.directory, mode)
            self.File.close()
            return self.directory
        else:
            return None
Ejemplo n.º 5
0
    def _dialog(self, message="Select Folder",  new_directory=True,mode='r'):
        """Creates a file dialog box for working with

        @param message Message to display in dialog
        @param new_file True if allowed to create new directory
        @return A directory to be used for the file operation."""
        try:
            from enthought.pyface.api import FileDialog, OK
        except ImportError:
            from pyface.api import FileDialog, OK
        # Wildcard pattern to be used in file dialogs.
        file_wildcard = "hdf file (*.hdf5)|*.hdf5|Data file (*.dat)|\
        *.dat|All files|*"

        if mode == "r":
            mode2 = "open"
        elif mode == "w":
            mode2 = "save as"

        if self.directory is not None:
            filename = path.basename(path.realpath(self.directory))
            dirname = path.dirname(path.realpath(self.directory))
        else:
            filename = ""
            dirname = ""
        dlg = FileDialog(action=mode2, wildcard=file_wildcard)
        dlg.open()
        if dlg.return_code == OK:
            self.directory = dlg.path
            self.File=h5py.File(self.directory,mode)
            self.File.close()
            return self.directory
        else:
            return None
Ejemplo n.º 6
0
 def _load_button_fired(self):
     dialog = FileDialog(action = "open", wildcard=self.file_wildcard)
     dialog.open()
     if dialog.return_code == OK:
         self.init(smear.RawData(dialog.path))
         #self.raw = smear.RawData(dialog.path, 'r')
         self.filename = dialog.filename
Ejemplo n.º 7
0
 def get_save_filename(self):
     """Get a filename from the user via a FileDialog. Returns the filename."""
     dialog = FileDialog(action="save as",
                         default_filename="spline_00",
                         wildcard="*.png")
     dialog.open()
     if dialog.return_code == OK:
         return dialog.path
    def _on_open(self, info):
        """ Menu action to load a script from file.  """
        file_dialog = FileDialog(action='open',
                                 default_directory=info.object.file_directory,
                                 wildcard='All files (*.*)|*.*')
        file_dialog.open()

        if file_dialog.path != '':
            info.object.load_code_from_file(file_dialog.path)
            
        return
    def _on_save_as(self, info):
        """ Menu action to save script to file of different name """
        file_dialog = FileDialog(action='save as',
                                 default_path=info.object.file_directory,
                                 wildcard='All files (*.*)|*.*')
        file_dialog.open()

        if file_dialog.path != '':
            info.object.save_code_to_file(file_dialog.path)
            msg = 'Saving script at ', file_dialog.path
            logger.debug(msg)
        return
Ejemplo n.º 10
0
 def _load_button_fired(self):
     dialog = FileDialog(action="open", wildcard=self.file_wildcard)
     dialog.open()
     if dialog.return_code == OK:
         f = open(dialog.path, 'r')
         data = f.readline()
         if data.endswith('\n'):
             data = data[:-1]
         self.value = data
         self.filedir = dialog.directory
         self.filename = dialog.filename
         self.saved = True
    def _on_import_data_file(self, info):
        """ Loading data files to import data for data contexts.

            File formats supported are ASCII, LAS, CSV, pickled files.
            ASCII files can be tab- or whitespace- delimited.
            If geo cannot be imported, only the pickled files can be
            used.
        """

        try:
            import geo
            wildcard = 'All files (*.*)|*.*|' + \
                       'ASCII files (*.asc)|*.asc|' + \
                       'Text files (*.txt)|*.txt|' + \
                       'CSV files (*.csv)|*.csv|' + \
                       'LAS files (*.las)|*.las|' + \
                       'Pickled files (*.pickle)|*.pickle|'+ \
                       'Segy files (*.segy)|*.segy'
        except ImportError:
            wildcard = 'Pickled files (*.pickle)|*.pickle'
        
        app = info.object
        file_dialog = FileDialog(action = 'open',
                                 default_directory = app.data_directory,
                                 wildcard = wildcard,
                                 )
        file_dialog.open()

        data_context, filename = None, file_dialog.path
        if file_dialog.path != '':
            filesplit = os.path.splitext(filename)
            if filesplit[1] != '': 
                configurable_import = ConfigurableImportUI(filename = filename,
                                                           wildcard = wildcard)
                ui = configurable_import.edit_traits(kind='livemodal')
                if ui.result and isinstance(configurable_import.context,DataContext):
                    data_context = configurable_import.context
                    app.load_context(
                        data_context, mode=configurable_import.save_choice_)

                    filename = configurable_import.filename

        if isinstance(data_context,DataContext):
            logger.debug('Loading data from %s' % filename)
        else:
            logger.error('Unidentified file format for data import: %s' %
                         filename)
        return
 def _load_button_fired(self):
     dialog = FileDialog(action="open", wildcard='EQ files (*.eq)|*.eq')
     result = dialog.open()
     if result == OK:
         f = file(dialog.path, "rb")
         self.equalizers = pickle.load(f)
         f.close()
Ejemplo n.º 13
0
    def perform(self, event, cfile=None):
        """ Performs the action. """

        logger.info("Performing save connectome file action")

        # helper variable to use this function not only in the menubar
        exec_as_funct = True

        cfile = self.window.application.get_service("cviewer.plugins.cff2.cfile.CFile")

        wildcard = "Connectome File Format v2.0 (*.cff)|*.cff|" "All files (*.*)|*.*"

        dlg = FileDialog(
            wildcard=wildcard,
            title="Save as Connectome File",
            resizeable=False,
            action="save as",
            default_directory=preference_manager.cviewerui.cffpath,
        )

        if dlg.open() == OK:

            if (dlg.paths[0]).endswith(".cff"):

                cfflib.save_to_cff(cfile.obj, dlg.paths[0])
                logger.info("Saved connectome file to %s" % dlg.paths[0])
Ejemplo n.º 14
0
    def perform(self, event):
        """ Performs the action. """
        mv = get_imayavi(self.window)
        s = get_scene(mv)
        if s is None:
            return

        wildcard = 'All files (*.*)|*.*'
        for src in registry.sources:
            if len(src.extensions) > 0:
                if wildcard.endswith('|') or \
                   src.wildcard.startswith('|'):
                    wildcard += src.wildcard
                else:
                    wildcard += '|' + src.wildcard

        parent = self.window.control
        dialog = FileDialog(parent=parent,
                            title='Open supported data file',
                            action='open',
                            wildcard=wildcard)
        if dialog.open() == OK:
            if not isfile(dialog.path):
                error("File '%s' does not exist!" % dialog.path, parent)
                return
            # FIXME: Ask for user input if a filetype is unknown and
            # choose appropriate reader.
            src = mv.open(dialog.path)
            if src is not None:
                mv.engine.current_selection = src
 def _save_button_fired(self):
     dialog = FileDialog(action="save as", wildcard='EQ files (*.eq)|*.eq')
     result = dialog.open()
     if result == OK:
         f = file(dialog.path, "wb")
         pickle.dump( self.equalizers , f)
         f.close()
Ejemplo n.º 16
0
def open_file(window, path=None):
    if path==None:
        wildcard="|".join([
            "All files (*)", "*",
            "Hermes2D mesh files (*.mesh)", "*.mesh",
            "Agros2D problem files (*.a2d)", "*.a2d",
            ])
        dialog = FileDialog(parent=None, title='Open supported data file',
                            action='open', wildcard=wildcard,
                            # use this to have Hermes2D by default:
                            #wildcard_index=1,
                            wildcard_index=0,
                            default_directory=get_data_dir(),
                            #default_filename="lshape.mesh",
                            )
        if dialog.open() == OK:
            path = dialog.path
        else:
            return
    ext = os.path.splitext(path)[1]
    if ext == ".a2d":
        scene = window.get_view_by_id("Problem")
        p, g = read_a2d(path)
        scene.problem = p
        scene.geometry = g
    else:
        scene = window.get_view_by_id("Scene")
        mesh = read_mesh(path)
        scene.mesh = mesh
        scene.mode = "mesh"
    def on_savefig ( self ):
        """ Handles the user requesting that the image of the function is to be saved.
        """
        import os
        dlg = FileDialog(parent = self.control, 
                         title = 'Save as image',
                         default_directory=os.getcwd(),
                         default_filename="", wildcard=WILDCARD,
                         action='save as')
        if dlg.open() == OK:
            path = dlg.path

            print "Saving plot to", path, "..."
            try:
                # Now we create a canvas of the appropriate size and ask it to render
                # our component.  (If we wanted to display this plot in a window, we
                # would not need to create the graphics context ourselves; it would be
                # created for us by the window.)
                size=(650,400)
                gc = GraphicsContext(size)
                self.plot_container.draw(gc)
                gc.save(path)
            except:
                print "Error saving!"
                raise
            print "Plot saved."
        return
Ejemplo n.º 18
0
    def perform(self, event):
        """ Performs the action. """
        mv = get_imayavi(self.window)
        s = get_scene(mv)
        if s is None:
            return

        wildcard = 'All files (*.*)|*.*'
        for src in registry.sources:
            if len(src.extensions) > 0:
                if wildcard.endswith('|') or \
                   src.wildcard.startswith('|'):
                       wildcard += src.wildcard
                else:
                    wildcard += '|' + src.wildcard
                   
        parent = self.window.control
        dialog = FileDialog(parent=parent,
                            title='Open supported data file',
                            action='open', wildcard=wildcard
                            )
        if dialog.open() == OK:
            if not isfile(dialog.path):
                error("File '%s' does not exist!"%dialog.path, parent)
                return
            # FIXME: Ask for user input if a filetype is unknown and
            # choose appropriate reader.
            src = mv.open(dialog.path)
            if src is not None:
                mv.engine.current_selection = src
Ejemplo n.º 19
0
    def perform(self, event):

        plot_component = self.container.component

        filter = 'PNG file (*.png)|*.png|\nTIFF file (*.tiff)|*.tiff|'
        dialog = FileDialog(action='save as', wildcard=filter)

        if dialog.open() != OK:
            return

        # Remove the toolbar before saving the plot, so the output doesn't
        # include the toolbar.
        plot_component.remove_toolbar()

        filename = dialog.path

        width, height = plot_component.outer_bounds

        gc = PlotGraphicsContext((width, height), dpi=72)
        gc.render_component(plot_component)
        try:
            gc.save(filename)
        except KeyError, e:
            errmsg = "The filename must have an extension that matches a graphics"
            errmsg = errmsg + " format, such as '.png' or '.tiff'."
            if str(e.message) != '':
                errmsg = ("Unknown filename extension: '%s'\n" % str(e.message)) + errmsg

            error(None, errmsg, title="Invalid Filename Extension")
Ejemplo n.º 20
0
    def save_as(self, info):
        """ Handles saving the current model to file.
        """
        if not info.initialized:
            return

#        retval = self.edit_traits(parent=info.ui.control, view="file_view")

        dlg = FileDialog( action = "save as",
            wildcard = "Graphviz Files (*.dot, *.xdot, *.txt)|" \
                "*.dot;*.xdot;*.txt|Dot Files (*.dot)|*.dot|" \
                "All Files (*.*)|*.*|")

        if dlg.open() == OK:
            fd = None
            try:
                fd = open(dlg.path, "wb")
                dot_code = str(self.model)
                fd.write(dot_code)

                self.save_file = dlg.path

            except:
                error(parent=info.ui.control, title="Save Error",
                      message="An error was encountered when saving\nto %s"
                      % self.file)

            finally:
                if fd is not None:
                    fd.close()

        del dlg
Ejemplo n.º 21
0
 def perform(self, event):
     """ Performs the action. """
     wildcard = 'Python files (*.py)|*.py'
     parent = self.window.control
     dialog = FileDialog(parent=parent,
                         title='Open Python file',
                         action='open', wildcard=wildcard
                         )
     if dialog.open() == OK:
         if not isfile(dialog.path):
             error("File '%s' does not exist"%dialog.path, parent)
             return
         
         # Get the globals.
         # The following code is taken from scripts/mayavi2.py.
         g = sys.modules['__main__'].__dict__
         if 'mayavi' not in g:
             mv = get_imayavi(self.window)
             g['mayavi'] = mv
             g['engine'] = mv.engine
         # Do execfile
         try:
             # If we don't pass globals twice we get NameErrors and nope,
             # using exec open(script_name).read() does not fix it.
             execfile(dialog.path, g, g)
         except Exception, msg:
             exception(str(msg))
Ejemplo n.º 22
0
 def _load_button_fired(self):
     dialog = FileDialog(action="open", wildcard='EQ files (*.eq)|*.eq')
     result = dialog.open()
     if result == OK:
         f = file(dialog.path, "rb")
         self.equalizers = pickle.load(f)
         f.close()
Ejemplo n.º 23
0
 def _save_button_fired(self):
     dialog = FileDialog(action="save as", wildcard='EQ files (*.eq)|*.eq')
     result = dialog.open()
     if result == OK:
         f = file(dialog.path, "wb")
         pickle.dump(self.equalizers, f)
         f.close()
Ejemplo n.º 24
0
    def on_savefig ( self ):
        """ Handles the user requesting that the image of the function is to be saved.
        """
        import os
        dlg = FileDialog(parent = self.control, 
                         title = 'Save as image',
                         default_directory=os.getcwd(),
                         default_filename="", wildcard=WILDCARD,
                         action='save as')
        if dlg.open() == OK:
            path = dlg.path

            print "Saving plot to", path, "..."
            try:

                # Now we create a canvas of the appropriate size and ask it to render
                # our component.  (If we wanted to display this plot in a window, we
                # would not need to create the graphics context ourselves; it would be
                # created for us by the window.)
                self._plot.bounds = [500,300]
                self._plot.padding = 50
                plot_gc = PlotGraphicsContext(self._plot.outer_bounds)
                print self._plot.outer_bounds
                plot_gc.render_component(self._plot)

                # Finally, we tell the graphics context to save itself to disk as an image.
                plot_gc.save(path)
                
            except:
                print "Error saving!"
                raise
            print "Plot saved."
        return
Ejemplo n.º 25
0
 def _save_btn_fired(self):
     f = FileDialog(action='save as',
                    default_path=self.bad_name,
                    wildcard='*.txt')
     if f.open() == OK:
         self.bad_name = f.path
         self.save_bad_data(self.bad_name)
    def on_savedata ( self ):
        """ Handles the user requesting that the data of the function is to be saved.
        """
        import os
        dlg = FileDialog(parent = self.control, 
                         title = 'Export function data',
                         default_directory=os.getcwd(),
                         default_filename="", wildcard='*.csv',
                         action='save as')
        if dlg.open() == OK:
            path = dlg.path

            print "Saving data to", path, "..."
            try:

#                factory  = self.factory
#                plotitem = factory.plotitem
#                x_values = getattr(self.object, plotitem.index)
#                y_values = getattr(self.object, plotitem.value)
                x_values = self.value.xdata
                y_values = self.value.ydata
                savetxt( path, vstack( (x_values,y_values) ).transpose() )
            except:
                print "Error saving!"
                raise
            print "Plot saved."
        return
    def on_savefig ( self ):
        """ Handles the user requesting that the image of the function is to be saved.
        """
        import os
        dlg = FileDialog(parent = self.control, 
                         title = 'Save as image',
                         default_directory=os.getcwd(),
                         default_filename="", wildcard=WILDCARD,
                         action='save as')
        if dlg.open() == OK:
            path = dlg.path

            print "Saving plot to", path, "..."
            try:

                # Now we create a canvas of the appropriate size and ask it to render
                # our component.  (If we wanted to display this plot in a window, we
                # would not need to create the graphics context ourselves; it would be
                # created for us by the window.)

               # plot_gc = PlotGraphicsContext(self._plot.outer_bounds)
               # plot_gc.render_component(self._plot)
                              
                #self._plot_container.outer_bounds = list((800,600))
#                plot_gc = PlotGraphicsContext((400,300), dpi=72.0)
#                plot_gc.render_component(self._plot_container)

#                self.line_plot.bounds = [500,300]
#                self.line_plot.padding = 50
#
#                win_size = self.line_plot.outer_bounds
#             #   win_size = self.component.outer_bounds
#                plot_gc = PlotGraphicsContext(win_size)
#        
#        
#                # Have the plot component into it
#                plot_gc.render_component(self.line_plot)
#
#                # Finally, we tell the graphics context to save itself to disk as an image.
#                plot_gc.save(path)

                DPI = 70.0
                size=(550,450)
             #   self.plot_container = create_plot()
                self.plot_container.bounds = list(size)
                self.plot_container.do_layout(force=True)
                
                gc = PlotGraphicsContext(size, dpi=DPI)
               

                #gc = GraphicsContext((size[0]+1, size[1]+1))
                gc.render_component(self.plot_container)
                gc.save(path)

                
            except:
                print "Error saving!"
                raise
            print "Plot saved."
        return
    def _on_export_as_script(self, info):
        """ Menu action to save script to file of different name 
        
        In this case the execution model and the local context are both
        merged in a script in the form of assignments.
        """
        file_dialog = FileDialog(action='save as',
                                 default_path=info.object.file_directory,
                                 wildcard='All files (*.*)|*.*')
        file_dialog.open()

        if file_dialog.path != '':
            info.object.export_as_script(file_dialog.path)
            msg = 'Saving script at ', file_dialog.path
            logger.debug(msg)
        return 
    def on_savedata ( self ):
        """ Handles the user requesting that the data of the function is to be saved.
        """
        import os
        dlg = FileDialog(parent = self.control, 
                         title = 'Export function data',
                         default_directory=os.getcwd(),
                         default_filename="", wildcard='*.csv',
                         action='save as')
        if dlg.open() == OK:
            path = dlg.path

            print "Saving data to", path, "..."
            try:
                vectors = []
                x_values = self.value.xdata
                y_values = self.value.ydata
                #savetxt( path, vstack( (x_values, y_values[:,0], y_values[:,1], y_values[:,2]) ).transpose() )
                
                print 'y_values', y_values
                y_values_tr = y_values.transpose()
                for vector in y_values_tr[:]:
                    vectors.append(vector)

                savetxt( path, vstack( (x_values, vectors) ).transpose() )
                    
            except:
                print "Error saving!"
                raise
            print "Plot saved."
        return
Ejemplo n.º 30
0
 def _load_fired(self):
     f = FileDialog(action = 'open', 
                    title = 'Load points',
                    default_path = self.filename, 
                    wildcard = '*.points')
     if f.open() == OK:
         self.filename = f.path
         self.from_file(f.path)
Ejemplo n.º 31
0
 def _save_fired(self):
     f = FileDialog(action='save as',
                    title='Save points',
                    default_path=self.filename,
                    wildcard='*.points')
     if f.open() == OK:
         self.filename = f.path
         self.to_file()
Ejemplo n.º 32
0
    def _open_file(self):
        """ Open a new file. """

        if self.control:
            dlg = FileDialog(parent=self.control, wildcard="*.py")

            if dlg.open() == OK:
                self._editor.path = dlg.path
Ejemplo n.º 33
0
 def perform(self):
     extns = ['*.pickle']
     dlg = FileDialog(parent=self._window.control, action='save as',
             wildcard='|'.join(extns), title="Save As")
     if dlg.open() == OK:
         file = open( dlg.path, 'w' )
         pickle.dump(self._window.scene, file)
         file.close()  
    def _on_export_as_function(self, info):
        """ Menu action to save function to file of different name 
        
        In this case the execution model and the local context are both
        included in the definition of a new function where free_variables
        take their default_value from the context.
        """
        file_dialog = FileDialog(action='save as',
                                 default_path=info.object.file_directory,
                                 wildcard='All files (*.*)|*.*')
        file_dialog.open()

        if file_dialog.path != '':
            info.object.export_as_function(file_dialog.path)
            msg = 'Saving script at ', file_dialog.path
            logger.debug(msg)
        return        
Ejemplo n.º 35
0
    def _open_file(self):
        """ Open a new file. """

        if self.control:
            dlg = FileDialog(parent=self.control, wildcard=WILDCARD)

            if dlg.open() == OK:
                self._editor.path = dlg.path
 def _set_filename_button_fired(self):
     dialog = FileDialog(
             default_directory=self.path,
             default_filename=self.filename,
             action="save as", 
             wildcard=self.file_wildcard
             )
     dialog.open()
     if dialog.return_code == OK:
         if not os.path.isdir(dialog.directory):
             raise IOError, 'directory does not exist'
         self.path = dialog.directory
         root, ext = os.path.splitext(dialog.filename)
         if not ext:
             ext = '.avi'
         filename = '{0}{1}'.format(root,ext)
         self.filename = os.path.join(dialog.directory, filename)
Ejemplo n.º 37
0
 def perform(self):
     extns = ['*.pickle']
     dlg = FileDialog(parent=self._window.control, action='open',
             wildcard='|'.join(extns), title="Open")
     if dlg.open() == OK:
         file = open( dlg.path, 'r' )
         self._window.scene = pickle.load(file)
         self. refresh()
         file.close()
Ejemplo n.º 38
0
 def _load(self):
     print "Load from EventTable"
     extension = "evt"
     wildcard = "EventTable|*.evt|VA MarkerFile|*.vmrk"
     fileDialog = FileDialog(action='open', title='Load EventTable',
                                  wildcard=wildcard)
     fileDialog.open()
     if fileDialog.path == '' or fileDialog.return_code == CANCEL:
         return False
     else:
         print "Opening", fileDialog.path
         #et = eegpy.load(str(fileDialog.path))
         et = eegpy.EventTable(str(fileDialog.path))
         for k in et.keys():
             for t in et[k]:
                 #print k,t
                 self.append(t,k,False,False)
         self.markers.sort(cmp=self.cmp_markers)
Ejemplo n.º 39
0
 def perform(self):
     """Pops up a dialog used to save the scene to an image."""
     extns = ['*.png', '*.jpg', '*.jpeg', '*.tiff', '*.bmp', '*.ps',
              '*.eps', '*.tex', '*.rib', '*.wrl', '*.oogl', '*.pdf',
              '*.vrml', '*.obj', '*.iv']
     dlg = FileDialog(parent=self._window.control, action='save as',
             wildcard='|'.join(extns), title="Save scene to image")
     if dlg.open() == OK:
         self._window.scene.save(dlg.path)
Ejemplo n.º 40
0
    def perform(self, event=None):
        logger.info('OpenFileAction.perform()')
        pref_script_path = preference_manager.cviewerui.scriptpath
        dialog = FileDialog(parent=self.window.control,
            title='Open File',
            default_directory=pref_script_path)
        if dialog.open() == OK:
            self.window.workbench.edit(File(dialog.path), kind=TextEditor)

            
Ejemplo n.º 41
0
 def perform(self, event):
     """ Performs the action. """
     wildcard = 'MayaVi2 files (*.mv2)|*.mv2|' + FileDialog.WILDCARD_ALL
     dialog = FileDialog(parent=self.window.control,
                         title='Save MayaVi2 file',
                         action='save as', wildcard=wildcard
                         )
     if dialog.open() == OK:
         mv = get_imayavi(self.window)
         mv.save_visualization(dialog.path)
Ejemplo n.º 42
0
 def _save_fired(self):
     f = FileDialog(action='save as',
                    default_path=self.filename,
                    wildcard='*.jpg;*.tiff;*.bmp;*.png')
     if f.open() == OK:
         try:
             self.save_image(f.path)
             self.error = ''
         except Exception as e:
             self.error = error_to_str(e)
Ejemplo n.º 43
0
 def _load(self):
     print "Load from EventTable"
     extension = "evt"
     wildcard = "EventTable|*.evt|VA MarkerFile|*.vmrk"
     fileDialog = FileDialog(action='open',
                             title='Load EventTable',
                             wildcard=wildcard)
     fileDialog.open()
     if fileDialog.path == '' or fileDialog.return_code == CANCEL:
         return False
     else:
         print "Opening", fileDialog.path
         #et = eegpy.load(str(fileDialog.path))
         et = eegpy.EventTable(str(fileDialog.path))
         for k in et.keys():
             for t in et[k]:
                 #print k,t
                 self.append(t, k, False, False)
         self.markers.sort(cmp=self.cmp_markers)
Ejemplo n.º 44
0
    def perform(self, event):
        """ Performs the action. """
        wildcard = 'Python files (*.py)|*.py'
        parent = self.window.control

        # path from preference manager
        pref_script_path = preference_manager.cviewerui.scriptpath

        if pref_script_path == '':
            # store executed script path in preferences
            dialog = FileDialog(
                parent=parent,
                title='Open Python file',
                action='open',
                wildcard=wildcard,
            )
        else:
            dialog = FileDialog(
                parent=parent,
                title='Open Python file',
                action='open',
                wildcard=wildcard,
                default_directory=pref_script_path,
            )

        if dialog.open() == OK:
            if not isfile(dialog.path):
                error("File '%s' does not exist" % dialog.path, parent)
                return

            # Get the globals.
            # The following code is taken from scripts/mayavi2.py.
            g = sys.modules['__main__'].__dict__
            if 'mayavi' not in g:
                mv = get_imayavi(self.window)
                g['mayavi'] = mv
                g['engine'] = mv.engine

            if 'cfile' not in g:
                # load cfile reference into gloabl name space
                cfile = self.window.application.get_service(
                    'cviewer.plugins.cff2.cfile.CFile')
                g['cfile'] = cfile

            # always store last executed path in preferences
            # but this only gets definitely stored when one open the preference manager
            preference_manager.cviewerui.scriptpath = dirname(dialog.path)

            # Do execfile
            try:
                # If we don't pass globals twice we get NameErrors and nope,
                # using exec open(script_name).read() does not fix it.
                execfile(dialog.path, g, g)
            except Exception, msg:
                exception(str(msg))
Ejemplo n.º 45
0
 def perform(self):
     """Pops up a dialog used to save the scene to an image."""
     extns = [
         '*.png', '*.jpg', '*.jpeg', '*.tiff', '*.bmp', '*.ps', '*.eps',
         '*.tex', '*.rib', '*.wrl', '*.oogl', '*.pdf', '*.vrml', '*.obj',
         '*.iv'
     ]
     dlg = FileDialog(parent=self._window.control,
                      action='save as',
                      wildcard='|'.join(extns),
                      title="Save scene to image")
     if dlg.open() == OK:
         self._window.scene.save(dlg.path)
Ejemplo n.º 46
0
 def _save_to(self):
     print "Save to EventTable"
     extension = "evt"
     wildcard = "*.evt"
     fileDialog = FileDialog(action='save as',
                             title='Save As',
                             wildcard=wildcard)
     fileDialog.open()
     if fileDialog.path == '' or fileDialog.return_code == CANCEL:
         return False
     else:
         extLen = len(extension)
         if extLen and fileDialog.path[-extLen - 1:] != '.' + extension:
             fileDialog.path += '.' + extension
     #print "fc.fn:", fileDialog.path
     #TODO: Check if file exists and join EventTables
     et = EventTable()
     for m in self.markers:
         try:
             et.add_trigger(m.name, m.t)
         except ValueError, e:
             et.add_trigger_type(m.name, [m.t])
Ejemplo n.º 47
0
    def perform(self, event):
        """ Perform the action. """

        dialog = FileDialog(parent=self.window.control,
                            title='Save scene to %s' % self.name,
                            action='save as',
                            wildcard=self.wildcard)
        if dialog.open() == OK:
            scene = self.scene_manager.current_scene
            if scene is not None:
                method = getattr(scene, self.save_method)
                method(dialog.path)

        return
Ejemplo n.º 48
0
    def _save_file(self):
        """ Save the file. """

        if self.control:
            try:
                self._editor.save()
            except IOError, e:
                # If you are trying to save to a file that doesn't exist,
                # open up a FileDialog with a 'save as' action.
                dlg = FileDialog(parent=self.control,
                                 action='save as',
                                 wildcard="*.py")
                if dlg.open() == OK:
                    self._editor.save(dlg.path)
Ejemplo n.º 49
0
 def ui_save(self):
     """Save recording to file, pop up a UI dialog to find out where
     and close the file when done.
     """
     from enthought.pyface.api import FileDialog, OK
     wildcard = 'Python files (*.py)|*.py|' + FileDialog.WILDCARD_ALL
     dialog = FileDialog(title='Save Script',
                         action='save as',
                         wildcard=wildcard)
     if dialog.open() == OK:
         fname = dialog.path
         f = open(fname, 'w')
         self.save(f)
         f.close()
Ejemplo n.º 50
0
    def _load_fired(self):
        import enthought.sweet_pickle as sp
        from enthought.pyface.api import FileDialog, OK

        wildcard = "CMP Configuration State (*.pkl)|*.pkl|" \
                        "All files (*.*)|*.*"
        dlg = FileDialog(wildcard=wildcard,title="Select a configuration state to load",\
                         resizeable=False, \
                         default_directory=self.project_dir,)

        if dlg.open() == OK:
            if not os.path.isfile(dlg.path):
                return
            else:
                self.load_state(dlg.path)
Ejemplo n.º 51
0
 def perform(self, event):
     """ Performs the action. """
     wildcard = 'MayaVi2 files (*.mv2)|*.mv2|' + FileDialog.WILDCARD_ALL
     parent = self.window.control
     dialog = FileDialog(parent=parent,
                         title='Open MayaVi2 file',
                         action='open', wildcard=wildcard
                         )
     if dialog.open() == OK:
         if not isfile(dialog.path):
             error("File '%s' does not exist"%dialog.path, parent)
             return
         
         mv = get_imayavi(self.window)
         mv.load_visualization(dialog.path)
Ejemplo n.º 52
0
 def _save_snapshot(self):
     """Invoked by the toolbar menu to save a snapshot of the scene
     to an image.  Note that the extension of the filename
     determines what image type is saved.  The default is PNG.
     """
     if self._panel is not None:
         wildcard = "PNG images (*.png)|*.png|Determine by extension (*.*)|*.*"
         dialog = FileDialog(parent=self._panel,
                             title='Save scene to image',
                             action='save as',
                             default_filename="snapshot.png",
                             wildcard=wildcard)
         if dialog.open() == OK:
             # The extension of the path will determine the actual
             # image type saved.
             self.save(dialog.path)
Ejemplo n.º 53
0
    def _save_fired(self):
        import pickle
        import enthought.sweet_pickle as sp
        import os.path
        from enthought.pyface.api import FileDialog, OK

        wildcard = "CMP Configuration State (*.pkl)|*.pkl|" \
                        "All files (*.*)|*.*"
        dlg = FileDialog(wildcard=wildcard,title="Filename to store configuration state",\
                         resizeable=False, action = 'save as', \
                         default_directory=self.subject_workingdir,)

        if dlg.open() == OK:
            if not dlg.path.endswith('.pkl'):
                dlg.path = dlg.path + '.pkl'
            self.save_state(dlg.path)
Ejemplo n.º 54
0
    def perform(self, event, cfile=None):
        """ Performs the action. """

        logger.info('Performing open connectome file action')

        # helper variable to use this function not only in the menubar
        exec_as_funct = True

        if cfile is None:
            # get the instance of the current CFile
            # with the help of the Service Registry
            cfile = self.window.application.get_service(
                'cviewer.plugins.cff2.cfile.CFile')
            exec_as_funct = False

        wildcard = "Connectome Markup File v2.0 (meta.cml)|meta.cml|" \
                    "Connectome File Format v2.0 (*.cff)|*.cff|" \
                    "All files (*.*)|*.*"
        dlg = FileDialog(wildcard=wildcard,title="Choose a Connectome File",\
                         resizeable=False, \
                         default_directory=preference_manager.cviewerui.cffpath,)

        if dlg.open() == OK:

            if not os.path.isfile(dlg.path):
                logger.error("File '%s' does not exist!" % dlg.path)
                return

            # if file exists and has .cff ending
            if os.path.exists(
                    dlg.paths[0]) and (dlg.paths[0]).endswith('.cff'):

                # close the cfile if one is currently loaded
                cfile.close_cfile()

                # load cfile data
                cfile.load_cfile(dlg.paths[0])

                self.window.status_bar_manager.message = ''
            elif os.path.exists(
                    dlg.paths[0]) and (dlg.paths[0]).endswith('meta.cml'):
                cfile.close_cfile()
                cfile.load_cfile(dlg.paths[0], ismetacml=True)
            else:
                logger.info('Could not load file: ' + dlg.paths)
Ejemplo n.º 55
0
    def save_as(self):
        """ Saves the text to disk after prompting for the file name. """

        dialog = FileDialog(parent=self.window.control,
                            action='save as',
                            default_filename=self.name,
                            wildcard=FileDialog.WILDCARD_PY)
        if dialog.open() != CANCEL:
            # Update the editor.
            self.id = dialog.path
            self.name = basename(dialog.path)

            # Update the resource.
            self.obj.path = dialog.path

            # Save it!
            self.save()

        return
Ejemplo n.º 56
0
 def _save_output_action(self):
     """Pops up a dialog box for the action to ask for a file."""
     # FIXME: in a refactor this should all go in a separate view
     # related object.
     from enthought.pyface.api import FileDialog, OK
     wildcard = 'All files (*.*)|*.*|'\
                'VTK XML files (*.xml)|*.xml|'\
                'Image Data (*.vti)|*.vti|'\
                'Poly Data (*.vtp)|*.vtp|'\
                'Rectilinear Grid (*.vtr)|*.vtr|'\
                'Structured Grid (*.vts)|*.vts|'\
                'Unstructured Grid (*.vtu)|*.vtu|'\
                'Old-style VTK files (*.vtk)|*.vtk'
                
     dialog = FileDialog(title='Save output to file',
                         action='save as', wildcard=wildcard
                         )
     if dialog.open() == OK:
         self.save_output(dialog.path)
Ejemplo n.º 57
0
def popup_save(parent=None):
    """Popup a dialog asking for an image name to save the scene to.
    This is used mainly to save a scene in full screen mode. Returns a
    filename, returns empty string if action was cancelled. `parent` is
    the parent widget over which the dialog will be popped up.
    """
    extns = ['*.png', '*.jpg', '*.jpeg', '*.tiff', '*.bmp', '*.ps', '*.eps',
             '*.tex', '*.rib', '*.wrl', '*.oogl', '*.pdf', '*.vrml', '*.obj',
             '*.iv']
    wildcard='|'.join(extns)

    dialog = FileDialog(
        parent = parent, title='Save scene to image',
        action='save as', wildcard=wildcard
    )
    if dialog.open() == OK:
        return dialog.path
    else:
        return ''