def __init__(self, parent=None):
QtGui.QMainWindow.__init__(self, parent)
self.state = {"test", "stuff", "other"}
self.data = GuiData()
self.vtkActorList = []
self.frame = QtGui.QFrame()
self.resize(1400, 1200)
self.setWindowTitle('ProteusCFD')
self.move(100, 100)
self.vl = QtGui.QHBoxLayout()
self.vl.setSpacing(10)
self.frame.setLayout(self.vl)
self.setCentralWidget(self.frame)
#create tabs dialog and add it to the layout
self.tabs = QtGui.QTabWidget()
self.vl.addWidget(self.tabs, 1.0)
#Setup our tree view
self.treeWidget = QtGui.QTreeWidget()
self.treeWidget.setHeaderHidden(True)
self.addItems(self.treeWidget.invisibleRootItem())
self.treeWidget.itemChanged.connect(self.handleChanged)
self.tabs.addTab(self.treeWidget, "Physics Options")
#setup solution control tab
SolutionControlTab = QtGui.QWidget()
self.tabs.addTab(SolutionControlTab, "Solution Control")
#setup boundary condition tab
self.BCTab1 = BCTab(self.data, self.visibilityChanged)
self.tabs.addTab(self.BCTab1, "Boundary Conditions")
#draw the tabs
self.tabs.show()
#Setup our menu and actions
exitAction = QtGui.QAction('Exit', self)
exitAction.setShortcut('Ctrl+Q')
exitAction.setStatusTip('Exit application')
exitAction.triggered.connect(self.close)
saveAction = QtGui.QAction('Save Case', self)
saveAction.setShortcut('Ctril+S')
saveAction.setStatusTip('Save config')
saveAction.triggered.connect(self.save)
loadAction = QtGui.QAction('Load Case', self)
loadAction.setStatusTip('Load config')
loadAction.triggered.connect(self.load)
openFileAction = QtGui.QAction('Import Mesh', self)
openFileAction.setStatusTip('Import Grid')
openFileAction.triggered.connect(self.selectFile)
self.menubar = self.menuBar()
fileMenu = self.menubar.addMenu('&File')
fileMenu.addAction(exitAction)
fileMenu.addAction(saveAction)
fileMenu.addAction(loadAction)
fileMenu.addAction(openFileAction)
self.statusBar().showMessage('Waiting...')
#Setup our vtk widget
self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
self.vl.addWidget(self.vtkWidget, 2.0)
#renderer inherits from vtkViewport so we have access to these members
self.ren = vtk.vtkRenderer()
self.ren.SetBackground(0.5, 0.5, 0.5)
self.ren.SetBackground2(.5, .8, .5)
self.ren.GradientBackgroundOn()
self.ren.ResetCamera()
self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
self.iren.AddObserver("KeyPressEvent", self.keyPressVTK)
#create axes
axes = vtk.vtkAxesActor()
self.ren.AddActor(axes)
scale = vtk.vtkTransform()
scale.Scale(0.5, 0.5, 0.5)
axes.SetUserTransform(scale)
translate = vtk.vtkTransform()
translate.Translate(-0.5, -0.5, -0.5)
axes.SetUserTransform(translate)
self.show()
self.iren.Initialize()
def __init__(self, parent = None):
QtGui.QMainWindow.__init__(self, parent)
self.state = {"test", "stuff", "other"}
self.data = GuiData()
self.vtkActorList = []
self.frame = QtGui.QFrame()
self.resize(1000,600)
self.setWindowTitle('ProteusCFD')
self.move(100,100)
self.vl = QtGui.QHBoxLayout()
self.vl.setSpacing(10)
self.frame.setLayout(self.vl)
self.setCentralWidget(self.frame)
#create tabs dialog and add it to the layout
self.tabs = QtGui.QTabWidget()
self.vl.addWidget(self.tabs,1.0)
#Setup our tree view
self.treeWidget = QtGui.QTreeWidget()
self.treeWidget.setHeaderHidden(True)
self.addItems(self.treeWidget.invisibleRootItem())
self.treeWidget.itemChanged.connect (self.handleChanged)
self.tabs.addTab(self.treeWidget, "Physics Options")
#setup solution control tab
SolutionControlTab = QtGui.QWidget()
self.tabs.addTab(SolutionControlTab, "Solution Control")
#setup boundary condition tab
self.BCTab1 = BCTab(self.data, self.visibilityChanged)
self.tabs.addTab(self.BCTab1, "Boundary Conditions")
#draw the tabs
self.tabs.show()
#Setup our menu and actions
exitAction = QtGui.QAction('Exit', self)
exitAction.setShortcut('Ctrl+Q')
exitAction.setStatusTip('Exit application')
exitAction.triggered.connect(self.close)
saveAction = QtGui.QAction('Save', self)
saveAction.setShortcut('Ctril+S')
saveAction.setStatusTip('Save config')
saveAction.triggered.connect(self.save)
loadAction = QtGui.QAction('Load', self)
loadAction.setStatusTip('Load config')
loadAction.triggered.connect(self.load)
openFileAction = QtGui.QAction('Import', self)
openFileAction.setStatusTip('Import Grid')
openFileAction.triggered.connect(self.selectFile)
self.menubar = self.menuBar()
fileMenu = self.menubar.addMenu('&File')
fileMenu.addAction(exitAction)
fileMenu.addAction(saveAction)
fileMenu.addAction(loadAction)
fileMenu.addAction(openFileAction)
self.statusBar().showMessage('Waiting...')
#Setup our vtk widget
self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
self.vl.addWidget(self.vtkWidget, 2.0)
#renderer inherits from vtkViewport so we have access to these members
self.ren = vtk.vtkRenderer()
self.ren.SetBackground(0.5,0.5,0.5)
self.ren.SetBackground2(.5,.8,.5)
self.ren.GradientBackgroundOn()
self.ren.ResetCamera()
self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
self.iren.AddObserver("KeyPressEvent", self.keyPressVTK)
#create axes
axes = vtk.vtkAxesActor()
self.ren.AddActor(axes)
scale = vtk.vtkTransform()
scale.Scale(0.5, 0.5, 0.5)
axes.SetUserTransform(scale)
translate = vtk.vtkTransform()
translate.Translate(-0.5,-0.5,-0.5)
axes.SetUserTransform(translate)
self.show()
self.iren.Initialize()
class MainWindow(QtGui.QMainWindow):
def __init__(self, parent=None):
QtGui.QMainWindow.__init__(self, parent)
self.state = {"test", "stuff", "other"}
self.data = GuiData()
self.vtkActorList = []
self.frame = QtGui.QFrame()
self.resize(1400, 1200)
self.setWindowTitle('ProteusCFD')
self.move(100, 100)
self.vl = QtGui.QHBoxLayout()
self.vl.setSpacing(10)
self.frame.setLayout(self.vl)
self.setCentralWidget(self.frame)
#create tabs dialog and add it to the layout
self.tabs = QtGui.QTabWidget()
self.vl.addWidget(self.tabs, 1.0)
#Setup our tree view
self.treeWidget = QtGui.QTreeWidget()
self.treeWidget.setHeaderHidden(True)
self.addItems(self.treeWidget.invisibleRootItem())
self.treeWidget.itemChanged.connect(self.handleChanged)
self.tabs.addTab(self.treeWidget, "Physics Options")
#setup solution control tab
SolutionControlTab = QtGui.QWidget()
self.tabs.addTab(SolutionControlTab, "Solution Control")
#setup boundary condition tab
self.BCTab1 = BCTab(self.data, self.visibilityChanged)
self.tabs.addTab(self.BCTab1, "Boundary Conditions")
#draw the tabs
self.tabs.show()
#Setup our menu and actions
exitAction = QtGui.QAction('Exit', self)
exitAction.setShortcut('Ctrl+Q')
exitAction.setStatusTip('Exit application')
exitAction.triggered.connect(self.close)
saveAction = QtGui.QAction('Save Case', self)
saveAction.setShortcut('Ctril+S')
saveAction.setStatusTip('Save config')
saveAction.triggered.connect(self.save)
loadAction = QtGui.QAction('Load Case', self)
loadAction.setStatusTip('Load config')
loadAction.triggered.connect(self.load)
openFileAction = QtGui.QAction('Import Mesh', self)
openFileAction.setStatusTip('Import Grid')
openFileAction.triggered.connect(self.selectFile)
self.menubar = self.menuBar()
fileMenu = self.menubar.addMenu('&File')
fileMenu.addAction(exitAction)
fileMenu.addAction(saveAction)
fileMenu.addAction(loadAction)
fileMenu.addAction(openFileAction)
self.statusBar().showMessage('Waiting...')
#Setup our vtk widget
self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
self.vl.addWidget(self.vtkWidget, 2.0)
#renderer inherits from vtkViewport so we have access to these members
self.ren = vtk.vtkRenderer()
self.ren.SetBackground(0.5, 0.5, 0.5)
self.ren.SetBackground2(.5, .8, .5)
self.ren.GradientBackgroundOn()
self.ren.ResetCamera()
self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
self.iren.AddObserver("KeyPressEvent", self.keyPressVTK)
#create axes
axes = vtk.vtkAxesActor()
self.ren.AddActor(axes)
scale = vtk.vtkTransform()
scale.Scale(0.5, 0.5, 0.5)
axes.SetUserTransform(scale)
translate = vtk.vtkTransform()
translate.Translate(-0.5, -0.5, -0.5)
axes.SetUserTransform(translate)
self.show()
self.iren.Initialize()
def addItems(self, parent):
column = 0
clients_item = self.addParent(parent, column, 'Clients',
'data Clients')
vendors_item = self.addParent(parent, column, 'Vendors',
'data Vendors')
time_period_item = self.addParent(parent, column, 'Time Period',
'data Time Period')
self.addChild(clients_item, column, 'Type A', 'data Type A')
self.addChild(clients_item, column, 'Type B', 'data Type B')
self.addChild(vendors_item, column, 'Mary', 'data Mary')
self.addChild(vendors_item, column, 'Arnold', 'data Arnold')
self.addChild(time_period_item, column, 'Init', 'data Init')
self.addChild(time_period_item, column, 'End', 'data End')
def addParent(self, parent, column, title, data):
item = QtGui.QTreeWidgetItem(parent, [title])
item.setData(column, QtCore.Qt.UserRole, data)
item.setChildIndicatorPolicy(QtGui.QTreeWidgetItem.ShowIndicator)
item.setExpanded(True)
return item
def addChild(self, parent, column, title, data):
item = QtGui.QTreeWidgetItem(parent, [title])
item.setData(column, QtCore.Qt.UserRole, data)
item.setCheckState(column, QtCore.Qt.Unchecked)
return item
def handleChanged(self, item, column):
if item.checkState(column) == QtCore.Qt.Checked:
print "checked", item, item.text(column)
if item.checkState(column) == QtCore.Qt.Unchecked:
print "unchecked", item, item.text(column)
def visibilityChanged(self, vtkGridPartId, factagId, state):
if state == QtCore.Qt.Checked:
self.vtkActorList[vtkGridPartId].VisibilityOn()
self.iren.Render()
print "visibility On for factag: ", factagId
else:
self.vtkActorList[vtkGridPartId].VisibilityOff()
self.iren.Render()
print "visibility Off for factag: ", factagId
def save(self):
filename = "proteusCFD.state"
pickle.dump(self.state, open(filename, "wb"))
def load(self):
filename = "proteusCFD.state"
self.state = pickle.load(open(filename, "rb"))
print self.state
def selectFile(self):
meshFile = str(QtGui.QFileDialog.getOpenFileName())
input = meshFile.split('.')
# The last decimal is before the h5 trailing name
# The second to last is before the processor id
# concatenate everything up to there
string = '/'
for part in input[0:-2]:
string = string + '.' + part
parts = string.split('/')
self.casename = parts[-1]
path = ""
for item in parts[0:-1]:
path = path + item + '/'
print 'Selected casename ' + self.casename
print 'Case path ' + path
self.clearMesh()
self.readMesh(path, self.casename)
# resets view for the camera to centered facing y
def resetCamera(self, direction):
self.ren.ResetCamera()
fp = self.ren.GetActiveCamera().GetFocalPoint()
p = self.ren.GetActiveCamera().GetPosition()
dist = math.sqrt((p[0] - fp[0])**2 + (p[1] - fp[1])**2 +
(p[2] - fp[2])**2)
if direction == "z":
self.ren.GetActiveCamera().SetPosition(fp[0], fp[1], fp[2] + dist)
self.ren.GetActiveCamera().SetViewUp(0.0, 1.0, 0.0)
elif direction == "y":
self.ren.GetActiveCamera().SetPosition(fp[0], fp[1] + dist, fp[2])
self.ren.GetActiveCamera().SetViewUp(0.0, 0.0, 1.0)
elif direction == "x":
self.ren.GetActiveCamera().SetPosition(fp[0] + dist, fp[1], fp[2])
self.ren.GetActiveCamera().SetViewUp(0.0, 1.0, 0.0)
self.iren.Render()
def keyPressVTK(self, obj, event):
print "Key pressed " + str(obj.GetKeyCode()) + " " + str(
obj.GetKeySym())
self.keyPressEvent(str(obj.GetKeyCode()))
def keyPressEvent(self, keyName):
if keyName == "x":
self.resetCamera("x")
elif keyName == "y":
self.resetCamera("y")
elif keyName == "z":
self.resetCamera("z")
elif keyName == "c":
self.clearMesh()
def clearMesh(self):
for part in self.vtkActorList:
self.ren.RemoveActor(part)
self.vtkActorList = []
self.data.clear()
self.iren.Render()
#update mesh visibility options
self.BCTab1.drawBCVizBoxes()
def readMesh(self, path, casename):
#read the mesh
self.data.loadMesh(path, casename)
# Create actors based on grid parts
self.vtkActorList = []
vtkgrids = self.data.buildVTKGrids()
itag = 0
c = colors()
for vtkgrid in vtkgrids:
mapper = vtk.vtkDataSetMapper()
mapper.SetInput(vtkgrid)
self.vtkActorList.append(vtk.vtkActor())
self.vtkActorList[-1].SetMapper(mapper)
self.vtkActorList[-1].GetProperty().SetColor(c.getNext())
# set visibility of volumes to false on startup
if int(self.data.elemGroups[itag]) < 0:
self.vtkActorList[-1].VisibilityOff()
itag = itag + 1
#visualize the grid
for actor in self.vtkActorList:
self.ren.AddActor(actor)
self.iren.Render()
#update mesh visibility options
self.BCTab1.drawBCVizBoxes()
class MainWindow(QtGui.QMainWindow):
def __init__(self, parent = None):
QtGui.QMainWindow.__init__(self, parent)
self.state = {"test", "stuff", "other"}
self.data = GuiData()
self.vtkActorList = []
self.frame = QtGui.QFrame()
self.resize(1000,600)
self.setWindowTitle('ProteusCFD')
self.move(100,100)
self.vl = QtGui.QHBoxLayout()
self.vl.setSpacing(10)
self.frame.setLayout(self.vl)
self.setCentralWidget(self.frame)
#create tabs dialog and add it to the layout
self.tabs = QtGui.QTabWidget()
self.vl.addWidget(self.tabs,1.0)
#Setup our tree view
self.treeWidget = QtGui.QTreeWidget()
self.treeWidget.setHeaderHidden(True)
self.addItems(self.treeWidget.invisibleRootItem())
self.treeWidget.itemChanged.connect (self.handleChanged)
self.tabs.addTab(self.treeWidget, "Physics Options")
#setup solution control tab
SolutionControlTab = QtGui.QWidget()
self.tabs.addTab(SolutionControlTab, "Solution Control")
#setup boundary condition tab
self.BCTab1 = BCTab(self.data, self.visibilityChanged)
self.tabs.addTab(self.BCTab1, "Boundary Conditions")
#draw the tabs
self.tabs.show()
#Setup our menu and actions
exitAction = QtGui.QAction('Exit', self)
exitAction.setShortcut('Ctrl+Q')
exitAction.setStatusTip('Exit application')
exitAction.triggered.connect(self.close)
saveAction = QtGui.QAction('Save', self)
saveAction.setShortcut('Ctril+S')
saveAction.setStatusTip('Save config')
saveAction.triggered.connect(self.save)
loadAction = QtGui.QAction('Load', self)
loadAction.setStatusTip('Load config')
loadAction.triggered.connect(self.load)
openFileAction = QtGui.QAction('Import', self)
openFileAction.setStatusTip('Import Grid')
openFileAction.triggered.connect(self.selectFile)
self.menubar = self.menuBar()
fileMenu = self.menubar.addMenu('&File')
fileMenu.addAction(exitAction)
fileMenu.addAction(saveAction)
fileMenu.addAction(loadAction)
fileMenu.addAction(openFileAction)
self.statusBar().showMessage('Waiting...')
#Setup our vtk widget
self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
self.vl.addWidget(self.vtkWidget, 2.0)
#renderer inherits from vtkViewport so we have access to these members
self.ren = vtk.vtkRenderer()
self.ren.SetBackground(0.5,0.5,0.5)
self.ren.SetBackground2(.5,.8,.5)
self.ren.GradientBackgroundOn()
self.ren.ResetCamera()
self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
self.iren.AddObserver("KeyPressEvent", self.keyPressVTK)
#create axes
axes = vtk.vtkAxesActor()
self.ren.AddActor(axes)
scale = vtk.vtkTransform()
scale.Scale(0.5, 0.5, 0.5)
axes.SetUserTransform(scale)
translate = vtk.vtkTransform()
translate.Translate(-0.5,-0.5,-0.5)
axes.SetUserTransform(translate)
self.show()
self.iren.Initialize()
def addItems(self, parent):
column = 0
clients_item = self.addParent(parent, column, 'Clients', 'data Clients')
vendors_item = self.addParent(parent, column, 'Vendors', 'data Vendors')
time_period_item = self.addParent(parent, column, 'Time Period', 'data Time Period')
self.addChild(clients_item, column, 'Type A', 'data Type A')
self.addChild(clients_item, column, 'Type B', 'data Type B')
self.addChild(vendors_item, column, 'Mary', 'data Mary')
self.addChild(vendors_item, column, 'Arnold', 'data Arnold')
self.addChild(time_period_item, column, 'Init', 'data Init')
self.addChild(time_period_item, column, 'End', 'data End')
def addParent(self, parent, column, title, data):
item = QtGui.QTreeWidgetItem(parent, [title])
item.setData(column, QtCore.Qt.UserRole, data)
item.setChildIndicatorPolicy(QtGui.QTreeWidgetItem.ShowIndicator)
item.setExpanded (True)
return item
def addChild(self, parent, column, title, data):
item = QtGui.QTreeWidgetItem(parent, [title])
item.setData(column, QtCore.Qt.UserRole, data)
item.setCheckState (column, QtCore.Qt.Unchecked)
return item
def handleChanged(self, item, column):
if item.checkState(column) == QtCore.Qt.Checked:
print "checked", item, item.text(column)
if item.checkState(column) == QtCore.Qt.Unchecked:
print "unchecked", item, item.text(column)
def visibilityChanged(self, vtkGridPartId, factagId, state):
if state == QtCore.Qt.Checked:
self.vtkActorList[vtkGridPartId].VisibilityOn()
self.iren.Render()
print "visibility On for factag: ", factagId
else:
self.vtkActorList[vtkGridPartId].VisibilityOff()
self.iren.Render()
print "visibility Off for factag: ", factagId
def save(self):
filename = "proteusCFD.state"
pickle.dump(self.state, open(filename, "wb"))
def load(self):
filename = "proteusCFD.state"
self.state = pickle.load(open(filename, "rb"))
print self.state
def selectFile(self):
meshFile = str(QtGui.QFileDialog.getOpenFileName())
input = meshFile.split('.')
parts = input[0].split('/')
self.casename = parts[-1]
parts = parts[0:-1]
path = "/"
for item in parts:
path = path + item + '/'
print 'Selected casename ' + self.casename
print 'Case path ' + path
self.clearMesh()
self.readMesh(path, self.casename)
# resets view for the camera to centered facing y
def resetCamera(self, direction):
self.ren.ResetCamera()
fp = self.ren.GetActiveCamera().GetFocalPoint()
p = self.ren.GetActiveCamera().GetPosition()
dist = math.sqrt( (p[0]-fp[0])**2 + (p[1]-fp[1])**2 + (p[2]-fp[2])**2 )
if direction == "z":
self.ren.GetActiveCamera().SetPosition(fp[0], fp[1], fp[2]+dist)
self.ren.GetActiveCamera().SetViewUp(0.0, 1.0, 0.0)
elif direction == "y":
self.ren.GetActiveCamera().SetPosition(fp[0], fp[1]+dist, fp[2])
self.ren.GetActiveCamera().SetViewUp(0.0, 0.0, 1.0)
elif direction == "x":
self.ren.GetActiveCamera().SetPosition(fp[0]+dist, fp[1], fp[2])
self.ren.GetActiveCamera().SetViewUp(0.0, 1.0, 0.0)
self.iren.Render()
def keyPressVTK(self, obj, event):
print "Key pressed " + str(obj.GetKeyCode()) + " " + str(obj.GetKeySym())
self.keyPressEvent(str(obj.GetKeyCode()))
def keyPressEvent(self, keyName):
if keyName == "x":
self.resetCamera("x")
elif keyName == "y":
self.resetCamera("y")
elif keyName == "z":
self.resetCamera("z")
elif keyName == "c":
self.clearMesh()
def clearMesh(self):
for part in self.vtkActorList:
self.ren.RemoveActor(part)
self.vtkActorList = []
self.data.clear()
self.iren.Render()
#update mesh visibility options
self.BCTab1.drawBCVizBoxes()
def readMesh(self, path, casename):
#read the mesh
self.data.loadMesh(path, casename)
# Create actors based on grid parts
self.vtkActorList = []
vtkgrids = self.data.buildVTKGrids()
itag = 0
c = colors()
for vtkgrid in vtkgrids:
mapper = vtk.vtkDataSetMapper()
mapper.SetInput(vtkgrid)
self.vtkActorList.append(vtk.vtkActor())
self.vtkActorList[-1].SetMapper(mapper)
self.vtkActorList[-1].GetProperty().SetColor(c.getNext())
# set visibility of volumes to false on startup
if int(self.data.elemGroups[itag]) < 0:
self.vtkActorList[-1].VisibilityOff()
itag = itag + 1
#visualize the grid
for actor in self.vtkActorList:
self.ren.AddActor(actor)
self.iren.Render()
#update mesh visibility options
self.BCTab1.drawBCVizBoxes()
