def _save_images(self):
"""Saves the current frames in individual images."""
dialog = DirectoryDialog()
if dialog.open() != OK:
return
self.progress_dialog.title = 'Saving images...'
self.progress_dialog.open()
dirpath = dialog.path
self.interactive = False
try:
for frame_index in xrange(len(self.frames)):
self.current_frame_index = frame_index
mlab.savefig(os.path.join(
dirpath, "frame-{}.png".format(frame_index)
))
progress = 100*frame_index/len(self.frames)
self.progress_dialog.update(progress)
self.progress_dialog.update(100)
except Exception:
self.calculation_error_event = traceback.format_exc()
log.exception('Error while saving')
finally:
self.interactive = True
def _add_root_button_fired(self):
dlg = DirectoryDialog(default_path=paths.mdd_data_dir)
if dlg.open() == OK and dlg.path:
name = os.path.basename(dlg.path)
if os.path.isfile(os.path.join(dlg.path, '{}.in'.format(name))):
self.roots.append(dlg.path)
else:
warning(None, 'Invalid MDD directory. {}. Directory must contain file '
'named {}.in'.format(dlg.path, name))
def _add_root_button_fired(self):
dlg = DirectoryDialog(default_path=paths.mdd_data_dir)
if dlg.open() == OK and dlg.path:
name = os.path.basename(dlg.path)
if os.path.isfile(os.path.join(dlg.path, '{}.in'.format(name))):
self.roots.append(dlg.path)
else:
warning(
None,
'Invalid MDD directory. {}. Directory must contain file '
'named {}.in'.format(dlg.path, name))
def open_directory_dialog(self, **kw):
if 'default_directory' not in kw:
kw['default_directory'] = self.default_directory
if 'wildcard' not in kw:
if self.wildcard:
kw['wildcard'] = self.wildcard
dialog = DirectoryDialog(action='open', **kw)
if dialog.open() == OK:
r = dialog.path
return r
def open_directory_dialog(self, **kw):
if 'default_directory' not in kw:
kw['default_directory'] = self.default_directory
if 'wildcard' not in kw:
if self.wildcard:
kw['wildcard'] = self.wildcard
dialog = DirectoryDialog(
#parent=self.window.control,
action='open',
**kw)
if dialog.open() == OK:
r = dialog.path
# if action == 'open files':
# r = dialog.paths
return r
def open_directory_dialog(self, **kw):
if 'default_directory' not in kw:
kw['default_directory'] = self.default_directory
if 'wildcard' not in kw:
if self.wildcard:
kw['wildcard'] = self.wildcard
dialog = DirectoryDialog(
# parent=self.window.control,
action='open',
**kw)
if dialog.open() == OK:
r = dialog.path
# if action == 'open files':
# r = dialog.paths
return r
def initialize_version(appname, debug):
root = os.path.dirname(__file__)
if not debug:
add_eggs(root)
else:
build_sys_path()
# can now use pychron.
from pychron.envisage.user_login import get_user
user = get_user()
if not user:
logger.info('user login failed')
return
if appname.startswith('py'):
appname = appname[2:]
from pychron.paths import paths
pref_path = os.path.join(paths.base, '.enthought',
'pychron.{}.application.{}'.format(appname, user),
'preferences.ini')
from ConfigParser import ConfigParser
cp = ConfigParser()
cp.read(pref_path)
proot = None
try:
proot = cp.get('pychron.general', 'root_dir')
except BaseException, e:
print 'root_dir exception={}'.format(e)
from pyface.directory_dialog import DirectoryDialog
information(
None,
'Pychron root directory not set in Preferences/General. Please select a valid directory'
)
dlg = DirectoryDialog(action='open',
default_directory=os.path.expanduser('~'))
result = dlg.open()
if result == OK:
proot = str(dlg.path)
def open_directory_dialog(self, **kw):
if "default_directory" not in kw:
kw["default_directory"] = self.default_directory
if "wildcard" not in kw:
if self.wildcard:
kw["wildcard"] = self.wildcard
dialog = DirectoryDialog(
# parent=self.window.control,
action="open",
**kw
)
if dialog.open() == OK:
r = dialog.path
# if action == 'open files':
# r = dialog.paths
return r
def initialize_version(appname, debug):
root = os.path.dirname(__file__)
if not debug:
add_eggs(root)
else:
build_sys_path()
# can now use pychron.
from pychron.envisage.user_login import get_user
user = get_user()
if not user:
logger.info("user login failed")
return
if appname.startswith("py"):
appname = appname[2:]
from pychron.paths import paths
pref_path = os.path.join(
paths.base, ".enthought", "pychron.{}.application.{}".format(appname, user), "preferences.ini"
)
from ConfigParser import ConfigParser
cp = ConfigParser()
cp.read(pref_path)
proot = None
try:
proot = cp.get("pychron.general", "root_dir")
except BaseException, e:
print "root_dir exception={}".format(e)
from pyface.directory_dialog import DirectoryDialog
information(None, "Pychron root directory not set in Preferences/General. Please select a valid directory")
dlg = DirectoryDialog(action="open", default_directory=os.path.expanduser("~"))
result = dlg.open()
if result == OK:
proot = str(dlg.path)
def _get_rid_root(self):
#=======================================================================
# #debug
# d = DummyDirectoryDialog()
# =======================================================================
d = DirectoryDialog(action='open',
default_path=self.data_directory
# default_path=paths.modeling_data_dir
)
if d.open() == OK:
rid = os.path.basename(d.path)
root = os.path.dirname(d.path)
# set this root as the working directory
os.chdir(d.path)
self.info('setting working directory to {}'.format(d.path))
return rid, root
return None, None
def _get_rid_root(self):
# =======================================================================
# #debug
# d = DummyDirectoryDialog()
# =======================================================================
d = DirectoryDialog(
action='open',
default_path=self.data_directory
# default_path=paths.modeling_data_dir
)
if d.open() == OK:
rid = os.path.basename(d.path)
root = os.path.dirname(d.path)
# set this root as the working directory
os.chdir(d.path)
self.info('setting working directory to {}'.format(d.path))
return rid, root
return None, None
def _directory_select_button_fired(self):
dlg = DirectoryDialog(default_path=HOME)
if dlg.open() == OK and dlg.path:
self.environments.append(dlg.path)
self.environment = dlg.path
def _open_directory_dialog(self, **kw):
dialog = DirectoryDialog(action='open', **kw)
if dialog.open() == OK:
r = dialog.path
return r
def _changeDestDir_changed(self):
dlg = DirectoryDialog()
if dlg.open() == OK:
self.destDir = dlg.path
def _open_directory_dialog(self, **kw):
dialog = DirectoryDialog(action='open', **kw)
if dialog.open() == OK:
r = dialog.path
return r
