def __init__(self):
super(ICECacheExplorerWindow, self).__init__()
self.setStyleSheet(CONSTS.SS_BACKGROUND)
self.setWindowTitle("ICE Explorer")
self.setWindowIcon(QtGui.QIcon('./resources/icex.png'))
self._create_statusbar()
self.prefs = ICEPreferences(self)
# setup the OGL window
self.viewer = ICEViewer(self)
self.viewer.cacheLoaded.connect(self._on_cache_loaded)
self.viewer.beginCacheLoading.connect(self._on_begin_cache_loading)
self.viewer.endCacheLoading.connect(self._on_end_cache_loading)
self.setCentralWidget(self.viewer)
# create object responsible for exporting attributes in the browser
self.exporter = ICEExporter(self)
self.exporter.cacheExporting.connect(self._on_cache_exporting)
self.exporter.beginCacheExporting.connect(
self._on_begin_cache_exporting)
self.exporter.endCacheExporting.connect(self._on_end_cache_exporting)
# create object responsible for loading attributes data in the browser
self.data_loader = ICEDataLoader(self)
self.data_loader.cacheDataLoaded.connect(self._on_cache_data_loaded)
self.data_loader.beginCacheDataLoading.connect(
self._on_begin_cache_data_loading)
self.data_loader.endCacheDataLoading.connect(
self._on_end_cache_data_loading)
self._create_actions()
self._create_menus()
self._create_toolbars()
self._create_dock_windows()
self.current_job = None
self._cache_files = []
def __init__(self):
super(ICECacheExplorerWindow, self).__init__()
self.setStyleSheet(CONSTS.SS_BACKGROUND)
self.setWindowTitle("ICE Explorer")
self.setWindowIcon(QtGui.QIcon('./resources/icex.png'))
self._create_statusbar()
self.prefs = ICEPreferences(self)
# setup the OGL window
self.viewer = ICEViewer(self)
self.viewer.cacheLoaded.connect(self._on_cache_loaded)
self.viewer.beginCacheLoading.connect(self._on_begin_cache_loading)
self.viewer.endCacheLoading.connect(self._on_end_cache_loading)
self.setCentralWidget(self.viewer)
# create object responsible for exporting attributes in the browser
self.exporter = ICEExporter( self )
self.exporter.cacheExporting.connect(self._on_cache_exporting)
self.exporter.beginCacheExporting.connect(self._on_begin_cache_exporting)
self.exporter.endCacheExporting.connect(self._on_end_cache_exporting)
# create object responsible for loading attributes data in the browser
self.data_loader = ICEDataLoader(self)
self.data_loader.cacheDataLoaded.connect(self._on_cache_data_loaded)
self.data_loader.beginCacheDataLoading.connect(self._on_begin_cache_data_loading)
self.data_loader.endCacheDataLoading.connect(self._on_end_cache_data_loading)
self._create_actions()
self._create_menus()
self._create_toolbars()
self._create_dock_windows()
self.current_job = None
self._cache_files = []
class ICECacheExplorerWindow(QtGui.QMainWindow):
""" This is the main window for ICE Explorer """
# Supported I/O jobs
LOAD = 0
EXPORT = 1
def __init__(self):
super(ICECacheExplorerWindow, self).__init__()
self.setStyleSheet(CONSTS.SS_BACKGROUND)
self.setWindowTitle("ICE Explorer")
self.setWindowIcon(QtGui.QIcon('./resources/icex.png'))
self._create_statusbar()
self.prefs = ICEPreferences(self)
# setup the OGL window
self.viewer = ICEViewer(self)
self.viewer.cacheLoaded.connect(self._on_cache_loaded)
self.viewer.beginCacheLoading.connect(self._on_begin_cache_loading)
self.viewer.endCacheLoading.connect(self._on_end_cache_loading)
self.setCentralWidget(self.viewer)
# create object responsible for exporting attributes in the browser
self.exporter = ICEExporter( self )
self.exporter.cacheExporting.connect(self._on_cache_exporting)
self.exporter.beginCacheExporting.connect(self._on_begin_cache_exporting)
self.exporter.endCacheExporting.connect(self._on_end_cache_exporting)
# create object responsible for loading attributes data in the browser
self.data_loader = ICEDataLoader(self)
self.data_loader.cacheDataLoaded.connect(self._on_cache_data_loaded)
self.data_loader.beginCacheDataLoading.connect(self._on_begin_cache_data_loading)
self.data_loader.endCacheDataLoading.connect(self._on_end_cache_data_loading)
self._create_actions()
self._create_menus()
self._create_toolbars()
self._create_dock_windows()
self.current_job = None
self._cache_files = []
def _load_cache_folder(self):
""" Load cache folder from a directory dialog """
self.statusBar().clearMessage()
fileDialog = QtGui.QFileDialog(self, caption="Select ICECache Folder To Load", directory=".")
fileDialog.setFileMode(QtGui.QFileDialog.DirectoryOnly)
fileDialog.setAcceptMode(QtGui.QFileDialog.AcceptOpen)
fileDialog.setOptions(QtGui.QFileDialog.ShowDirsOnly)
if not fileDialog.exec_():
return
self.current_job = self.LOAD
self.viewer.load_files_from_folder(fileDialog.directory().absolutePath())
def _load_cache(self):
""" Load cache file from a file dialog """
self.statusBar().clearMessage()
fileDialog = QtGui.QFileDialog(self, caption="Select ICECache File(s) To Load", directory=".", filter="ICECACHE (*.icecache *.sih5)")
fileDialog.setFileMode(QtGui.QFileDialog.ExistingFiles)
fileDialog.setAcceptMode(QtGui.QFileDialog.AcceptOpen)
if not fileDialog.exec_():
return
self.viewer.load_files(fileDialog.selectedFiles())
def _export_to_text(self):
""" Select folder cache and export cache files to text """
self.exporter.folder_dialog.open( CONSTS.TEXT_FMT )
def _export_to_sih5(self):
""" Select folder cache and export cache files to SIH5"""
self.exporter.folder_dialog.open( CONSTS.SIH5_FMT )
def _export_all_to_text(self):
""" Export all cache files to text """
self.statusBar().clearMessage()
self.exporter.file_dialog.open( self._cache_files, CONSTS.TEXT_FMT )
def _export_all_to_sih5(self):
""" Export all cache files to SIH5 """
self.statusBar().clearMessage()
self.exporter.file_dialog.open( self._cache_files, CONSTS.SIH5_FMT )
def preferences(self):
self.prefs.exec_()
def close(self):
sys.exit()
# Internals
# slots for the export job
def _on_begin_cache_exporting(self, num ):
self.current_job = self.EXPORT
self.cancel_current_job_act.setDisabled (False)
self._start_progressbar(num,'Exporting...')
def _on_cache_exporting(self, export_file):
self._update_progressbar()
def _on_end_cache_exporting(self):
# and we are done
self.current_job = None
self.cancel_current_job_act.setDisabled(True)
self._stop_progressbar()
# slots for the cache data load job
def _on_begin_cache_data_loading(self, cache_attrib, num ):
self._start_progressbar( num, 'Loading data: %s' % (cache_attrib) )
def _on_cache_data_loaded(self ):
self._update_progressbar()
def _on_end_cache_data_loading(self):
self._stop_progressbar()
# slots for the cache load job
def _on_begin_cache_loading(self, count, start, end ):
self.current_job = self.LOAD
self.treeWidget.clear()
self._cache_files = []
self.cancel_current_job_act.setDisabled(False)
self._start_progressbar( count, 'Loading Cache Files ...' )
def _on_end_cache_loading(self, count, start, end ):
self.current_job = None
self.cancel_current_job_act.setDisabled(True)
self._stop_progressbar()
def _on_cache_loaded(self, cacheindex, filename):
"""update the browser when a cache has finished loading"""
if cacheindex == None:
self.statusBar().showMessage( 'Error - invalid file: %s' % filename )
return
self._update_progressbar()
self._cache_files.append( filename )
# Creates browser top level items only, rest will be filled when items get expanded
# cache item
itemcache = QtGui.QTreeWidgetItem(self.treeWidget)
itemcache.setText(0, 'Cache %d' % (cacheindex))
indexVar = QtCore.QVariant(cacheindex)
itemcache.setData(0, QtCore.Qt.UserRole, indexVar)
self.treeWidget.addTopLevelItem(itemcache)
# header item
headeritem = QtGui.QTreeWidgetItem(itemcache)
headeritem.setText(0, 'Header')
headeritem.setText(1, "ICECACHE")
def _cancel_current_job(self):
""" cancel the current executing I/O job """
if self.current_job == self.LOAD:
self.viewer.stop_loading()
elif self.current_job == self.EXPORT:
self.exporter.cancel()
def _create_actions(self):
""" All actions required by the GUI components """
self.load_cache_folder_act = QtGui.QAction(QtGui.QIcon(r'./resources/load-cache-folder.png'), "&Load Cache Folder...", self, statusTip="Load Cache Folder", triggered=self._load_cache_folder)
self.load_cache_act = QtGui.QAction(QtGui.QIcon(r'./resources/load-cache-file.png'), "Load Cache &File(s)...", self, statusTip="Load Cache File(s)", triggered=self._load_cache)
self.export_all_caches_to_text_act = QtGui.QAction(QtGui.QIcon(r'./resources/export-cache.png'), "Export Caches to &Text", self, statusTip="Export Caches To Text", triggered=self._export_all_to_text)
self.export_all_caches_to_sih5_act = QtGui.QAction(QtGui.QIcon(r'./resources/export-cache.png'), "Export Caches to &SIH5", self, statusTip="Export Caches To SIH5", triggered=self._export_all_to_sih5)
self.export_caches_to_text_act = QtGui.QAction(QtGui.QIcon(r'./resources/export-cache-folder.png'), "Export Folder to &Text", self, statusTip="Export Folder To Text", triggered=self._export_to_text)
self.export_caches_to_sih5_act = QtGui.QAction(QtGui.QIcon(r'./resources/export-cache-folder.png'), "Export Folder to &SIH5", self, statusTip="Export Folder To SIH5", triggered=self._export_to_sih5)
self.export_selected_cache_to_text_act = QtGui.QAction(QtGui.QIcon(r'./resources/export-cache.png'), "Export Selected Cache To &Text", self, statusTip="Export To Text")
self.export_selected_cache_to_sih5_act = QtGui.QAction(QtGui.QIcon(r'./resources/export-cache.png'), "Export Selected Cache To &SIH5", self, statusTip="Export To SIH5")
self.cancel_current_job_act = QtGui.QAction(QtGui.QIcon(r'./resources/cancel_loading.png'), "Cancel", self, statusTip="Cancel", triggered=self._cancel_current_job)
self.cancel_current_job_act.setDisabled(True)
self.prefs_act = QtGui.QAction(QtGui.QIcon(r'./resources/preferences.png'), "&Preferences...", self, statusTip="ICE Explorer Preferences", triggered=self.preferences)
self.quit_act = QtGui.QAction("&Quit", self, shortcut="Ctrl+Q", statusTip="Quit ICE Explorer", triggered=self.close)
self.about_act = QtGui.QAction("&About", self, statusTip="Show About Dialog", triggered=self.about)
self.show_pers_act = QtGui.QAction(QtGui.QIcon(r'./resources/perspective.png'), "&Perspective", self, statusTip="Show Perspective View", triggered=self.viewer.perspective_view)
self.show_top_act = QtGui.QAction(QtGui.QIcon(r'./resources/top.png'), "&Top", self, statusTip="Show Top View", triggered=self.viewer.top_view)
self.show_front_act = QtGui.QAction(QtGui.QIcon(r'./resources/front.png'), "&Front", self, statusTip="Show Front View", triggered=self.viewer.front_view)
self.show_right_act = QtGui.QAction(QtGui.QIcon(r'./resources/right.png'), "&Right", self, statusTip="Show Right View", triggered=self.viewer.right_view)
self.zoom_tool_act = QtGui.QAction(QtGui.QIcon(r'./resources/zoom.png'), "&Zoom", self, statusTip="Zoom Tool", triggered=self.viewer.zoom_tool)
self.orbit_tool_act = QtGui.QAction(QtGui.QIcon(r'./resources/orbit.png'), "&Orbit", self, statusTip="Orbit Tool", triggered=self.viewer.orbit_tool)
self.pan_tool_act = QtGui.QAction(QtGui.QIcon(r'./resources/pan.png'), "&Pan", self, statusTip="Pan Tool", triggered=self.viewer.pan_tool)
def _create_menus(self):
self.menuBar().setStyleSheet(CONSTS.SS_MENUBAR)
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenu.setStyleSheet(CONSTS.SS_MENU)
self.fileMenu.addAction(self.load_cache_folder_act)
self.fileMenu.addAction(self.load_cache_act)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.export_caches_to_text_act)
self.fileMenu.addAction(self.export_caches_to_sih5_act)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.export_all_caches_to_text_act)
self.fileMenu.addAction(self.export_all_caches_to_sih5_act)
self.fileMenu.addAction(self.cancel_current_job_act)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.cancel_current_job_act)
self.fileMenu.addAction(self.quit_act)
self.editMenu = self.menuBar().addMenu("&Edit")
self.editMenu.setStyleSheet(CONSTS.SS_MENU)
self.editMenu.addAction(self.prefs_act)
self.viewMenu = self.menuBar().addMenu("&View")
self.viewMenu.setStyleSheet(CONSTS.SS_MENU)
self.helpMenu = self.menuBar().addMenu("&Help")
self.helpMenu.addAction(self.about_act)
self.helpMenu.setStyleSheet(CONSTS.SS_MENU)
def _create_toolbars(self):
self.fileToolBar = QtGui.QToolBar( "File" )
self.fileToolBar.addAction(self.load_cache_folder_act)
self.fileToolBar.addAction(self.load_cache_act)
self.fileToolBar.addSeparator()
self.fileToolBar.addAction(self.cancel_current_job_act)
self.addToolBar( QtCore.Qt.LeftToolBarArea, self.fileToolBar )
self.toolsToolBar = QtGui.QToolBar( "Tools" )
self.toolsToolBar.addAction(self.orbit_tool_act)
self.toolsToolBar.addAction(self.pan_tool_act)
self.toolsToolBar.addAction(self.zoom_tool_act)
self.addToolBar( QtCore.Qt.LeftToolBarArea, self.toolsToolBar )
self.viewToolBar = QtGui.QToolBar( "Views" )
self.viewToolBar.addAction(self.show_pers_act)
self.viewToolBar.addAction(self.show_top_act)
self.viewToolBar.addAction(self.show_front_act)
self.viewToolBar.addAction(self.show_right_act)
self.addToolBar( QtCore.Qt.LeftToolBarArea, self.viewToolBar )
def _create_statusbar(self):
""" The status bar contains a temporary message (left area ), a progress bar (left area) and a permanent msg on the right. """
self.pbar = QtGui.QProgressBar(self.statusBar())
self.pbar.setTextVisible(False)
self.pbar.hide()
self.middle_msg = QtGui.QLabel(self.statusBar())
self.right_msg = QtGui.QLabel(self.statusBar())
self.statusBar().addWidget(self.pbar,2)
self.statusBar().addWidget(self.middle_msg,1)
self.statusBar().addPermanentWidget(self.right_msg)
self.statusBar().showMessage("Ready")
def _create_dock_windows(self):
""" Creates and organizes all dockable windows"""
# Cache file browser
dock = QtGui.QDockWidget("Browser", self)
dock.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea | QtCore.Qt.RightDockWidgetArea)
self.treeWidget = QtGui.QTreeWidget(dock)
self.treeWidget.itemExpanded.connect( self._handle_item_expanded )
self.treeWidget.setObjectName('ICECacheInspectorWidget')
self.treeWidget.setColumnCount(2)
self.treeWidget.headerItem().setText(0, 'Item')
self.treeWidget.headerItem().setText(1, 'Value')
self.treeWidget.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.treeWidget.customContextMenuRequested.connect(self._handle_browser_contextmenu)
self.treeWidget.setStyleSheet(CONSTS.SS_BACKGROUND)
# ... browser context menus
self._browser_contextmenu = QtGui.QMenu(self)
self._browser_contextmenu.setStyleSheet(CONSTS.SS_MENU)
self._browser_contextmenu.addAction(self.export_selected_cache_to_text_act)
self._browser_contextmenu.addAction(self.export_selected_cache_to_sih5_act)
self.attribute_contextmenu = QtGui.QMenu(self)
self.attribute_contextmenu.setStyleSheet(CONSTS.SS_MENU)
dock.setWidget(self.treeWidget)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, dock)
self.viewMenu.addAction(dock.toggleViewAction())
# Playback widget
dock = QtGui.QDockWidget("Play Back", self)
dock.setAllowedAreas(QtCore.Qt.BottomDockWidgetArea)
self.playback = PlaybackWidget(self.viewer, dock)
dock.setWidget(self.playback)
self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, dock)
self.viewMenu.addAction(dock.toggleViewAction())
def _handle_item_expanded( self, item ):
""" Fill items data when the node gets expanded """
var = item.data(0, QtCore.Qt.UserRole)
if var == None:
# nothing to do
return
#Fill this item with cache attributes
(cacheindex,flag) = var.toInt()
h5cache = self.viewer.cache[ cacheindex ]
if item.text(0) == 'Data':
self.data_loader.load(h5cache,item)
return
#done with the user data
item.setData(0, QtCore.Qt.UserRole, None)
reader = H5Reader( h5cache )
reader.load()
# Cache
# > Header
# <header items>
headeritem = item.child(0)
headeritem.setText(0, 'Header')
headeritem.setText(1, reader.header['name'])
version = QtGui.QTreeWidgetItem(headeritem)
type = QtGui.QTreeWidgetItem(headeritem)
edge_count = QtGui.QTreeWidgetItem(headeritem)
particle_count = QtGui.QTreeWidgetItem(headeritem)
polygon_count = QtGui.QTreeWidgetItem(headeritem)
sample_count = QtGui.QTreeWidgetItem(headeritem)
blob_count = QtGui.QTreeWidgetItem(headeritem)
attribute_count = QtGui.QTreeWidgetItem(headeritem)
version.setText(0, 'Version')
version.setText(1, str(reader.header['version']))
type.setText(0, 'Type')
type.setText(1, objtype_to_string(reader.header['type']))
particle_count.setText(0, 'Particle Count')
particle_count.setText(1, str(reader.header['particle_count']))
edge_count.setText(0, 'Edge Count')
edge_count.setText(1, str(reader.header['edge_count']))
polygon_count.setText(0, 'Polygon Count')
polygon_count.setText(1, str(reader.header['polygon_count']))
sample_count.setText(0, 'Sample Count')
sample_count.setText(1, str(reader.header['sample_count']))
blob_count.setText(0, 'Blob Count')
blob_count.setText(1, str(reader.header['blob_count']))
attribute_count.setText(0, 'Attribute Count')
attribute_count.setText(1, str(reader.header['attribute_count']))
# attribute items
# Cache
# > Header
# <header items>
# > Attribute <name>
# <attribute items>
for attrib in reader.attributes:
attribitem = QtGui.QTreeWidgetItem(item)
attribitem.setText(0, 'Attribute')
attribitem.setText(1, attrib['name'] )
# attribute description
datatype = QtGui.QTreeWidgetItem(attribitem)
structtype = QtGui.QTreeWidgetItem(attribitem)
contexttype = QtGui.QTreeWidgetItem(attribitem)
objid = QtGui.QTreeWidgetItem(attribitem)
category = QtGui.QTreeWidgetItem(attribitem)
ptlocator_size = QtGui.QTreeWidgetItem(attribitem)
blobtype_count = QtGui.QTreeWidgetItem(attribitem)
blobtype_names = QtGui.QTreeWidgetItem(attribitem)
isconstant = QtGui.QTreeWidgetItem(attribitem)
dataitem = QtGui.QTreeWidgetItem(attribitem)
datatype.setText(0, 'Data Type')
datatype.setText(1, datatype_to_string(attrib['datatype']))
structtype.setText(0, 'Structure Type')
structtype.setText(1, structtype_to_string(attrib['structtype']))
contexttype.setText(0, 'Context Type')
contexttype.setText(1, contexttype_to_string(attrib['contexttype']))
objid.setText(0, 'Object ID')
objid.setText(1, str(attrib['objid']))
category.setText(0, 'Category')
category.setText(1, categorytype_to_string(attrib['category']))
ptlocator_size.setText(0, 'PointLocator Size')
ptlocator_size.setText(1, str(attrib['ptlocator_size']))
blobtype_count.setText(0, 'Blob Type Count')
blobtype_count.setText(1, str(attrib['blobtype_count']))
blobtype_names.setText(0, 'Blob Type Name')
blobtype_names.setText(1, str(attrib['blobtype_names']))
# will be filled when the data gets loaded by the user
isconstant.setText(0, 'Constant Data')
isconstant.setText(1, '')
# attribute data items, empty for now
# Cache
# > Header
# <header items>
# > Attribute <name>
# <attribute items>
# > Data
# <values>
dataitem.setText(0, 'Data')
var = QtCore.QVariant(cacheindex)
dataitem.setData(0, QtCore.Qt.UserRole, var )
# add dummy value so data item can get filled with values when it get expanded
emptyvalue = QtGui.QTreeWidgetItem( dataitem )
emptyvalue.setText( 0, '' )
def _handle_browser_contextmenu(self, point):
""" Context menus for exporting individual cache files"""
item = self.treeWidget.currentItem()
# show browser context menu on the widget
if item.parent() != None:
# context menu supported on the top level item only
return
action = self._browser_contextmenu.exec_(self.treeWidget.mapToGlobal(point))
var = item.data(0, QtCore.Qt.UserRole)
(cache_index,flag) = var.toInt()
fmt=None
if action == self.export_selected_cache_to_text_act:
fmt = CONSTS.TEXT_FMT
elif action == self.export_selected_cache_to_sih5_act:
fmt = CONSTS.SIH5_FMT
else:
return
h5cache = self.viewer.cache[ cache_index ]
self.exporter.export_files( [h5cache.filename], self.prefs.export_folder, fmt)
# Progress bar helpers
def _start_progressbar(self,num,msg=''):
self.pbar.show()
self.pbar.setRange(0, num)
self.pbar.setValue(1)
self.middle_msg.setText( msg )
def _update_progressbar(self):
self.pbar.setValue(self.pbar.value() + 1)
def _stop_progressbar(self):
self.pbar.hide()
self.pbar.reset()
self.middle_msg.clear( )
def about(self):
msg = "<b>ICE Explorer</b> Copyright (C) 2010 M.A.Belzile <br>"\
"A tool for viewing and browsing ICE cache data.<br>"\
"Version %d.%d.%d %s %d<br><br>"\
"This program comes with ABSOLUTELY NO WARRANTY; for details see the GNU General Public License."\
"This is free software, and you are welcome to redistribute it under certain conditions; see the GNU General Public License for details."
QtGui.QMessageBox.about(self, "About ICE Explorer", msg % (__version__[0], __version__[1], __version__[2], __version__[3], __version__[4]))
class ICECacheExplorerWindow(QtGui.QMainWindow):
""" This is the main window for ICE Explorer """
# Supported I/O jobs
LOAD = 0
EXPORT = 1
def __init__(self):
super(ICECacheExplorerWindow, self).__init__()
self.setStyleSheet(CONSTS.SS_BACKGROUND)
self.setWindowTitle("ICE Explorer")
self.setWindowIcon(QtGui.QIcon('./resources/icex.png'))
self._create_statusbar()
self.prefs = ICEPreferences(self)
# setup the OGL window
self.viewer = ICEViewer(self)
self.viewer.cacheLoaded.connect(self._on_cache_loaded)
self.viewer.beginCacheLoading.connect(self._on_begin_cache_loading)
self.viewer.endCacheLoading.connect(self._on_end_cache_loading)
self.setCentralWidget(self.viewer)
# create object responsible for exporting attributes in the browser
self.exporter = ICEExporter(self)
self.exporter.cacheExporting.connect(self._on_cache_exporting)
self.exporter.beginCacheExporting.connect(
self._on_begin_cache_exporting)
self.exporter.endCacheExporting.connect(self._on_end_cache_exporting)
# create object responsible for loading attributes data in the browser
self.data_loader = ICEDataLoader(self)
self.data_loader.cacheDataLoaded.connect(self._on_cache_data_loaded)
self.data_loader.beginCacheDataLoading.connect(
self._on_begin_cache_data_loading)
self.data_loader.endCacheDataLoading.connect(
self._on_end_cache_data_loading)
self._create_actions()
self._create_menus()
self._create_toolbars()
self._create_dock_windows()
self.current_job = None
self._cache_files = []
def _load_cache_folder(self):
""" Load cache folder from a directory dialog """
self.statusBar().clearMessage()
fileDialog = QtGui.QFileDialog(
self, caption="Select ICECache Folder To Load", directory=".")
fileDialog.setFileMode(QtGui.QFileDialog.DirectoryOnly)
fileDialog.setAcceptMode(QtGui.QFileDialog.AcceptOpen)
fileDialog.setOptions(QtGui.QFileDialog.ShowDirsOnly)
if not fileDialog.exec_():
return
self.current_job = self.LOAD
self.viewer.load_files_from_folder(
fileDialog.directory().absolutePath())
def _load_cache(self):
""" Load cache file from a file dialog """
self.statusBar().clearMessage()
fileDialog = QtGui.QFileDialog(
self,
caption="Select ICECache File(s) To Load",
directory=".",
filter="ICECACHE (*.icecache *.sih5)")
fileDialog.setFileMode(QtGui.QFileDialog.ExistingFiles)
fileDialog.setAcceptMode(QtGui.QFileDialog.AcceptOpen)
if not fileDialog.exec_():
return
self.viewer.load_files(fileDialog.selectedFiles())
def _export_to_text(self):
""" Select folder cache and export cache files to text """
self.exporter.folder_dialog.open(CONSTS.TEXT_FMT)
def _export_to_sih5(self):
""" Select folder cache and export cache files to SIH5"""
self.exporter.folder_dialog.open(CONSTS.SIH5_FMT)
def _export_all_to_text(self):
""" Export all cache files to text """
self.statusBar().clearMessage()
self.exporter.file_dialog.open(self._cache_files, CONSTS.TEXT_FMT)
def _export_all_to_sih5(self):
""" Export all cache files to SIH5 """
self.statusBar().clearMessage()
self.exporter.file_dialog.open(self._cache_files, CONSTS.SIH5_FMT)
def preferences(self):
self.prefs.exec_()
def close(self):
sys.exit()
# Internals
# slots for the export job
def _on_begin_cache_exporting(self, num):
self.current_job = self.EXPORT
self.cancel_current_job_act.setDisabled(False)
self._start_progressbar(num, 'Exporting...')
def _on_cache_exporting(self, export_file):
self._update_progressbar()
def _on_end_cache_exporting(self):
# and we are done
self.current_job = None
self.cancel_current_job_act.setDisabled(True)
self._stop_progressbar()
# slots for the cache data load job
def _on_begin_cache_data_loading(self, cache_attrib, num):
self._start_progressbar(num, 'Loading data: %s' % (cache_attrib))
def _on_cache_data_loaded(self):
self._update_progressbar()
def _on_end_cache_data_loading(self):
self._stop_progressbar()
# slots for the cache load job
def _on_begin_cache_loading(self, count, start, end):
self.current_job = self.LOAD
self.treeWidget.clear()
self._cache_files = []
self.cancel_current_job_act.setDisabled(False)
self._start_progressbar(count, 'Loading Cache Files ...')
def _on_end_cache_loading(self, count, start, end):
self.current_job = None
self.cancel_current_job_act.setDisabled(True)
self._stop_progressbar()
def _on_cache_loaded(self, cacheindex, filename):
"""update the browser when a cache has finished loading"""
if cacheindex == None:
self.statusBar().showMessage('Error - invalid file: %s' % filename)
return
self._update_progressbar()
self._cache_files.append(filename)
# Creates browser top level items only, rest will be filled when items get expanded
# cache item
itemcache = QtGui.QTreeWidgetItem(self.treeWidget)
itemcache.setText(0, 'Cache %d' % (cacheindex))
indexVar = QtCore.QVariant(cacheindex)
itemcache.setData(0, QtCore.Qt.UserRole, indexVar)
self.treeWidget.addTopLevelItem(itemcache)
# header item
headeritem = QtGui.QTreeWidgetItem(itemcache)
headeritem.setText(0, 'Header')
headeritem.setText(1, "ICECACHE")
def _cancel_current_job(self):
""" cancel the current executing I/O job """
if self.current_job == self.LOAD:
self.viewer.stop_loading()
elif self.current_job == self.EXPORT:
self.exporter.cancel()
def _create_actions(self):
""" All actions required by the GUI components """
self.load_cache_folder_act = QtGui.QAction(
QtGui.QIcon(r'./resources/load-cache-folder.png'),
"&Load Cache Folder...",
self,
statusTip="Load Cache Folder",
triggered=self._load_cache_folder)
self.load_cache_act = QtGui.QAction(
QtGui.QIcon(r'./resources/load-cache-file.png'),
"Load Cache &File(s)...",
self,
statusTip="Load Cache File(s)",
triggered=self._load_cache)
self.export_all_caches_to_text_act = QtGui.QAction(
QtGui.QIcon(r'./resources/export-cache.png'),
"Export Caches to &Text",
self,
statusTip="Export Caches To Text",
triggered=self._export_all_to_text)
self.export_all_caches_to_sih5_act = QtGui.QAction(
QtGui.QIcon(r'./resources/export-cache.png'),
"Export Caches to &SIH5",
self,
statusTip="Export Caches To SIH5",
triggered=self._export_all_to_sih5)
self.export_caches_to_text_act = QtGui.QAction(
QtGui.QIcon(r'./resources/export-cache-folder.png'),
"Export Folder to &Text",
self,
statusTip="Export Folder To Text",
triggered=self._export_to_text)
self.export_caches_to_sih5_act = QtGui.QAction(
QtGui.QIcon(r'./resources/export-cache-folder.png'),
"Export Folder to &SIH5",
self,
statusTip="Export Folder To SIH5",
triggered=self._export_to_sih5)
self.export_selected_cache_to_text_act = QtGui.QAction(
QtGui.QIcon(r'./resources/export-cache.png'),
"Export Selected Cache To &Text",
self,
statusTip="Export To Text")
self.export_selected_cache_to_sih5_act = QtGui.QAction(
QtGui.QIcon(r'./resources/export-cache.png'),
"Export Selected Cache To &SIH5",
self,
statusTip="Export To SIH5")
self.cancel_current_job_act = QtGui.QAction(
QtGui.QIcon(r'./resources/cancel_loading.png'),
"Cancel",
self,
statusTip="Cancel",
triggered=self._cancel_current_job)
self.cancel_current_job_act.setDisabled(True)
self.prefs_act = QtGui.QAction(
QtGui.QIcon(r'./resources/preferences.png'),
"&Preferences...",
self,
statusTip="ICE Explorer Preferences",
triggered=self.preferences)
self.quit_act = QtGui.QAction("&Quit",
self,
shortcut="Ctrl+Q",
statusTip="Quit ICE Explorer",
triggered=self.close)
self.about_act = QtGui.QAction("&About",
self,
statusTip="Show About Dialog",
triggered=self.about)
self.show_pers_act = QtGui.QAction(
QtGui.QIcon(r'./resources/perspective.png'),
"&Perspective",
self,
statusTip="Show Perspective View",
triggered=self.viewer.perspective_view)
self.show_top_act = QtGui.QAction(QtGui.QIcon(r'./resources/top.png'),
"&Top",
self,
statusTip="Show Top View",
triggered=self.viewer.top_view)
self.show_front_act = QtGui.QAction(
QtGui.QIcon(r'./resources/front.png'),
"&Front",
self,
statusTip="Show Front View",
triggered=self.viewer.front_view)
self.show_right_act = QtGui.QAction(
QtGui.QIcon(r'./resources/right.png'),
"&Right",
self,
statusTip="Show Right View",
triggered=self.viewer.right_view)
self.zoom_tool_act = QtGui.QAction(
QtGui.QIcon(r'./resources/zoom.png'),
"&Zoom",
self,
statusTip="Zoom Tool",
triggered=self.viewer.zoom_tool)
self.orbit_tool_act = QtGui.QAction(
QtGui.QIcon(r'./resources/orbit.png'),
"&Orbit",
self,
statusTip="Orbit Tool",
triggered=self.viewer.orbit_tool)
self.pan_tool_act = QtGui.QAction(QtGui.QIcon(r'./resources/pan.png'),
"&Pan",
self,
statusTip="Pan Tool",
triggered=self.viewer.pan_tool)
def _create_menus(self):
self.menuBar().setStyleSheet(CONSTS.SS_MENUBAR)
self.fileMenu = self.menuBar().addMenu("&File")
self.fileMenu.setStyleSheet(CONSTS.SS_MENU)
self.fileMenu.addAction(self.load_cache_folder_act)
self.fileMenu.addAction(self.load_cache_act)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.export_caches_to_text_act)
self.fileMenu.addAction(self.export_caches_to_sih5_act)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.export_all_caches_to_text_act)
self.fileMenu.addAction(self.export_all_caches_to_sih5_act)
self.fileMenu.addAction(self.cancel_current_job_act)
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.cancel_current_job_act)
self.fileMenu.addAction(self.quit_act)
self.editMenu = self.menuBar().addMenu("&Edit")
self.editMenu.setStyleSheet(CONSTS.SS_MENU)
self.editMenu.addAction(self.prefs_act)
self.viewMenu = self.menuBar().addMenu("&View")
self.viewMenu.setStyleSheet(CONSTS.SS_MENU)
self.helpMenu = self.menuBar().addMenu("&Help")
self.helpMenu.addAction(self.about_act)
self.helpMenu.setStyleSheet(CONSTS.SS_MENU)
def _create_toolbars(self):
self.fileToolBar = QtGui.QToolBar("File")
self.fileToolBar.addAction(self.load_cache_folder_act)
self.fileToolBar.addAction(self.load_cache_act)
self.fileToolBar.addSeparator()
self.fileToolBar.addAction(self.cancel_current_job_act)
self.addToolBar(QtCore.Qt.LeftToolBarArea, self.fileToolBar)
self.toolsToolBar = QtGui.QToolBar("Tools")
self.toolsToolBar.addAction(self.orbit_tool_act)
self.toolsToolBar.addAction(self.pan_tool_act)
self.toolsToolBar.addAction(self.zoom_tool_act)
self.addToolBar(QtCore.Qt.LeftToolBarArea, self.toolsToolBar)
self.viewToolBar = QtGui.QToolBar("Views")
self.viewToolBar.addAction(self.show_pers_act)
self.viewToolBar.addAction(self.show_top_act)
self.viewToolBar.addAction(self.show_front_act)
self.viewToolBar.addAction(self.show_right_act)
self.addToolBar(QtCore.Qt.LeftToolBarArea, self.viewToolBar)
def _create_statusbar(self):
""" The status bar contains a temporary message (left area ), a progress bar (left area) and a permanent msg on the right. """
self.pbar = QtGui.QProgressBar(self.statusBar())
self.pbar.setTextVisible(False)
self.pbar.hide()
self.middle_msg = QtGui.QLabel(self.statusBar())
self.right_msg = QtGui.QLabel(self.statusBar())
self.statusBar().addWidget(self.pbar, 2)
self.statusBar().addWidget(self.middle_msg, 1)
self.statusBar().addPermanentWidget(self.right_msg)
self.statusBar().showMessage("Ready")
def _create_dock_windows(self):
""" Creates and organizes all dockable windows"""
# Cache file browser
dock = QtGui.QDockWidget("Browser", self)
dock.setAllowedAreas(QtCore.Qt.LeftDockWidgetArea
| QtCore.Qt.RightDockWidgetArea)
self.treeWidget = QtGui.QTreeWidget(dock)
self.treeWidget.itemExpanded.connect(self._handle_item_expanded)
self.treeWidget.setObjectName('ICECacheInspectorWidget')
self.treeWidget.setColumnCount(2)
self.treeWidget.headerItem().setText(0, 'Item')
self.treeWidget.headerItem().setText(1, 'Value')
self.treeWidget.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.treeWidget.customContextMenuRequested.connect(
self._handle_browser_contextmenu)
self.treeWidget.setStyleSheet(CONSTS.SS_BACKGROUND)
# ... browser context menus
self._browser_contextmenu = QtGui.QMenu(self)
self._browser_contextmenu.setStyleSheet(CONSTS.SS_MENU)
self._browser_contextmenu.addAction(
self.export_selected_cache_to_text_act)
self._browser_contextmenu.addAction(
self.export_selected_cache_to_sih5_act)
self.attribute_contextmenu = QtGui.QMenu(self)
self.attribute_contextmenu.setStyleSheet(CONSTS.SS_MENU)
dock.setWidget(self.treeWidget)
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, dock)
self.viewMenu.addAction(dock.toggleViewAction())
# Playback widget
dock = QtGui.QDockWidget("Play Back", self)
dock.setAllowedAreas(QtCore.Qt.BottomDockWidgetArea)
self.playback = PlaybackWidget(self.viewer, dock)
dock.setWidget(self.playback)
self.addDockWidget(QtCore.Qt.BottomDockWidgetArea, dock)
self.viewMenu.addAction(dock.toggleViewAction())
def _handle_item_expanded(self, item):
""" Fill items data when the node gets expanded """
var = item.data(0, QtCore.Qt.UserRole)
if var == None:
# nothing to do
return
#Fill this item with cache attributes
(cacheindex, flag) = var.toInt()
h5cache = self.viewer.cache[cacheindex]
if item.text(0) == 'Data':
self.data_loader.load(h5cache, item)
return
#done with the user data
item.setData(0, QtCore.Qt.UserRole, None)
reader = H5Reader(h5cache)
reader.load()
# Cache
# > Header
# <header items>
headeritem = item.child(0)
headeritem.setText(0, 'Header')
headeritem.setText(1, reader.header['name'])
version = QtGui.QTreeWidgetItem(headeritem)
type = QtGui.QTreeWidgetItem(headeritem)
edge_count = QtGui.QTreeWidgetItem(headeritem)
particle_count = QtGui.QTreeWidgetItem(headeritem)
polygon_count = QtGui.QTreeWidgetItem(headeritem)
sample_count = QtGui.QTreeWidgetItem(headeritem)
blob_count = QtGui.QTreeWidgetItem(headeritem)
attribute_count = QtGui.QTreeWidgetItem(headeritem)
version.setText(0, 'Version')
version.setText(1, str(reader.header['version']))
type.setText(0, 'Type')
type.setText(1, objtype_to_string(reader.header['type']))
particle_count.setText(0, 'Particle Count')
particle_count.setText(1, str(reader.header['particle_count']))
edge_count.setText(0, 'Edge Count')
edge_count.setText(1, str(reader.header['edge_count']))
polygon_count.setText(0, 'Polygon Count')
polygon_count.setText(1, str(reader.header['polygon_count']))
sample_count.setText(0, 'Sample Count')
sample_count.setText(1, str(reader.header['sample_count']))
blob_count.setText(0, 'Blob Count')
blob_count.setText(1, str(reader.header['blob_count']))
attribute_count.setText(0, 'Attribute Count')
attribute_count.setText(1, str(reader.header['attribute_count']))
# attribute items
# Cache
# > Header
# <header items>
# > Attribute <name>
# <attribute items>
for attrib in reader.attributes:
attribitem = QtGui.QTreeWidgetItem(item)
attribitem.setText(0, 'Attribute')
attribitem.setText(1, attrib['name'])
# attribute description
datatype = QtGui.QTreeWidgetItem(attribitem)
structtype = QtGui.QTreeWidgetItem(attribitem)
contexttype = QtGui.QTreeWidgetItem(attribitem)
objid = QtGui.QTreeWidgetItem(attribitem)
category = QtGui.QTreeWidgetItem(attribitem)
ptlocator_size = QtGui.QTreeWidgetItem(attribitem)
blobtype_count = QtGui.QTreeWidgetItem(attribitem)
blobtype_names = QtGui.QTreeWidgetItem(attribitem)
isconstant = QtGui.QTreeWidgetItem(attribitem)
dataitem = QtGui.QTreeWidgetItem(attribitem)
datatype.setText(0, 'Data Type')
datatype.setText(1, datatype_to_string(attrib['datatype']))
structtype.setText(0, 'Structure Type')
structtype.setText(1, structtype_to_string(attrib['structtype']))
contexttype.setText(0, 'Context Type')
contexttype.setText(1,
contexttype_to_string(attrib['contexttype']))
objid.setText(0, 'Object ID')
objid.setText(1, str(attrib['objid']))
category.setText(0, 'Category')
category.setText(1, categorytype_to_string(attrib['category']))
ptlocator_size.setText(0, 'PointLocator Size')
ptlocator_size.setText(1, str(attrib['ptlocator_size']))
blobtype_count.setText(0, 'Blob Type Count')
blobtype_count.setText(1, str(attrib['blobtype_count']))
blobtype_names.setText(0, 'Blob Type Name')
blobtype_names.setText(1, str(attrib['blobtype_names']))
# will be filled when the data gets loaded by the user
isconstant.setText(0, 'Constant Data')
isconstant.setText(1, '')
# attribute data items, empty for now
# Cache
# > Header
# <header items>
# > Attribute <name>
# <attribute items>
# > Data
# <values>
dataitem.setText(0, 'Data')
var = QtCore.QVariant(cacheindex)
dataitem.setData(0, QtCore.Qt.UserRole, var)
# add dummy value so data item can get filled with values when it get expanded
emptyvalue = QtGui.QTreeWidgetItem(dataitem)
emptyvalue.setText(0, '')
def _handle_browser_contextmenu(self, point):
""" Context menus for exporting individual cache files"""
item = self.treeWidget.currentItem()
# show browser context menu on the widget
if item.parent() != None:
# context menu supported on the top level item only
return
action = self._browser_contextmenu.exec_(
self.treeWidget.mapToGlobal(point))
var = item.data(0, QtCore.Qt.UserRole)
(cache_index, flag) = var.toInt()
fmt = None
if action == self.export_selected_cache_to_text_act:
fmt = CONSTS.TEXT_FMT
elif action == self.export_selected_cache_to_sih5_act:
fmt = CONSTS.SIH5_FMT
else:
return
h5cache = self.viewer.cache[cache_index]
self.exporter.export_files([h5cache.filename],
self.prefs.export_folder, fmt)
# Progress bar helpers
def _start_progressbar(self, num, msg=''):
self.pbar.show()
self.pbar.setRange(0, num)
self.pbar.setValue(1)
self.middle_msg.setText(msg)
def _update_progressbar(self):
self.pbar.setValue(self.pbar.value() + 1)
def _stop_progressbar(self):
self.pbar.hide()
self.pbar.reset()
self.middle_msg.clear()
def about(self):
msg = "<b>ICE Explorer</b> Copyright (C) 2010 M.A.Belzile <br>"\
"A tool for viewing and browsing ICE cache data.<br>"\
"Version %d.%d.%d %s %d<br><br>"\
"This program comes with ABSOLUTELY NO WARRANTY; for details see the GNU General Public License."\
"This is free software, and you are welcome to redistribute it under certain conditions; see the GNU General Public License for details."
QtGui.QMessageBox.about(
self, "About ICE Explorer",
msg % (__version__[0], __version__[1], __version__[2],
__version__[3], __version__[4]))