Created
June 14, 2012 09:53
-
-
Save cstein/2929371 to your computer and use it in GitHub Desktop.
basic gui for simulation program
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from time import sleep | |
import matplotlib | |
import matplotlib.pyplot as plt | |
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas | |
from matplotlib.figure import Figure | |
matplotlib.rcParams.update({'font.size': 8}) | |
class Canvas(FigureCanvas): | |
def __init__(self,parent,dpi=100.0): | |
size = parent.size() | |
self.dpi = dpi | |
self.width = size.width() / dpi | |
self.height = size.height() / dpi | |
self.figure = Figure(figsize=(self.width, self.height), dpi=self.dpi, facecolor='w', edgecolor='k', frameon=False) | |
self.figure.subplots_adjust(left=0.05, bottom=0.05, right=0.95, top=0.95, wspace=None, hspace=None) | |
self.axes = self.figure.add_subplot(111) | |
self.axes.axis((-1,1,-1,1)) | |
FigureCanvas.__init__(self, self.figure) | |
self.updateGeometry() | |
self.draw() | |
self.cc = self.copy_from_bbox(self.axes.bbox) | |
self.particle_plot = None | |
self.setParent(parent) | |
self.blit(self.axes.bbox) | |
def on_pre_draw(self): | |
self.restore_region(self.cc) | |
def on_draw(self): | |
raise NotImplementedError | |
def on_post_draw(self): | |
self.blit(self.axes.bbox) | |
sleep(0.005) | |
def redraw(self): | |
self.on_pre_draw() | |
self.on_draw() | |
self.on_post_draw() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
import sys | |
from PyQt4.QtGui import QApplication | |
from simulator import MainForm | |
def main(): | |
app = QApplication(sys.argv) | |
form = MainForm() | |
form.show() | |
app.exec_() | |
if __name__ == "__main__": | |
main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
simulator_ui.py: simulator.ui | |
pyuic4 simulator.ui -o simulator_ui.py | |
clean: | |
rm -f *.pyc |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from numpy import array, where, abs | |
from numpy.random import random | |
from canvas import Canvas | |
class ParticleCanvas(Canvas): | |
def __init__(self,parent,dpi=100.0): | |
Canvas.__init__(self,parent,dpi) | |
self.pos = None | |
self.vel = None | |
def reinitialize(self, npart): | |
self.pos = 2.0*(random((2,npart))-0.5) | |
self.vel = 2.0*(random((2,npart))-0.5) | |
self.particle_plot = None | |
self.redraw() | |
def move(self): | |
dt = 0.01 | |
# change velocity if they go beyond box | |
self.vel[where(abs(self.pos + dt*self.vel)>1.0)] *= -1.0 | |
self.pos += dt*self.vel | |
def on_draw(self): | |
if self.pos is None or self.vel is None: return | |
if self.particle_plot is None: | |
self.particle_plot, = self.axes.plot(self.pos[0],self.pos[1], 'ro', animated=True) | |
self.particle_plot.set_xdata(self.pos[0]) | |
self.particle_plot.set_ydata(self.pos[1]) | |
self.axes.draw_artist(self.particle_plot) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from PyQt4 import QtGui, QtCore | |
from simulator_ui import Ui_main | |
from particlecanvas import ParticleCanvas | |
class MainForm(QtGui.QMainWindow): | |
def __init__(self, parent=None): | |
QtGui.QWidget.__init__(self,parent) | |
self.ui = Ui_main() | |
self.ui.setupUi(self) | |
QtCore.QObject.connect(self.ui.btnSetup, QtCore.SIGNAL("clicked()"), self.on_setup_clicked) | |
QtCore.QObject.connect(self.ui.btnRun, QtCore.SIGNAL("clicked()"), self.on_run_clicked) | |
self.c = ParticleCanvas(self.ui.frame) | |
def on_setup_clicked(self): | |
self.c.reinitialize(int(self.ui.txtNPart.text())) | |
def on_run_clicked(self): | |
for i in range(int(self.ui.txtNSteps.text())): | |
self.c.move() | |
self.c.redraw() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
<?xml version="1.0" encoding="UTF-8"?> | |
<ui version="4.0"> | |
<class>main</class> | |
<widget class="QWidget" name="main"> | |
<property name="geometry"> | |
<rect> | |
<x>0</x> | |
<y>0</y> | |
<width>412</width> | |
<height>270</height> | |
</rect> | |
</property> | |
<property name="windowTitle"> | |
<string>Simulator</string> | |
</property> | |
<widget class="QLineEdit" name="txtNPart"> | |
<property name="geometry"> | |
<rect> | |
<x>270</x> | |
<y>30</y> | |
<width>131</width> | |
<height>27</height> | |
</rect> | |
</property> | |
<property name="inputMask"> | |
<string notr="true"/> | |
</property> | |
<property name="text"> | |
<string notr="true">100</string> | |
</property> | |
<property name="alignment"> | |
<set>Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter</set> | |
</property> | |
</widget> | |
<widget class="QLabel" name="label"> | |
<property name="geometry"> | |
<rect> | |
<x>270</x> | |
<y>10</y> | |
<width>141</width> | |
<height>17</height> | |
</rect> | |
</property> | |
<property name="text"> | |
<string notr="true">Number of Particles</string> | |
</property> | |
</widget> | |
<widget class="QFrame" name="frame"> | |
<property name="geometry"> | |
<rect> | |
<x>10</x> | |
<y>10</y> | |
<width>250</width> | |
<height>250</height> | |
</rect> | |
</property> | |
<property name="frameShape"> | |
<enum>QFrame::StyledPanel</enum> | |
</property> | |
<property name="frameShadow"> | |
<enum>QFrame::Raised</enum> | |
</property> | |
</widget> | |
<widget class="QPushButton" name="btnSetup"> | |
<property name="geometry"> | |
<rect> | |
<x>340</x> | |
<y>60</y> | |
<width>61</width> | |
<height>27</height> | |
</rect> | |
</property> | |
<property name="text"> | |
<string notr="true">Setup</string> | |
</property> | |
</widget> | |
<widget class="QPushButton" name="btnRun"> | |
<property name="geometry"> | |
<rect> | |
<x>340</x> | |
<y>230</y> | |
<width>61</width> | |
<height>27</height> | |
</rect> | |
</property> | |
<property name="text"> | |
<string notr="true">Run</string> | |
</property> | |
</widget> | |
<widget class="QLineEdit" name="txtNSteps"> | |
<property name="geometry"> | |
<rect> | |
<x>270</x> | |
<y>200</y> | |
<width>131</width> | |
<height>27</height> | |
</rect> | |
</property> | |
<property name="inputMask"> | |
<string notr="true"/> | |
</property> | |
<property name="text"> | |
<string notr="true">1000</string> | |
</property> | |
<property name="alignment"> | |
<set>Qt::AlignRight|Qt::AlignTrailing|Qt::AlignVCenter</set> | |
</property> | |
</widget> | |
<widget class="QLabel" name="label_2"> | |
<property name="geometry"> | |
<rect> | |
<x>270</x> | |
<y>180</y> | |
<width>131</width> | |
<height>17</height> | |
</rect> | |
</property> | |
<property name="text"> | |
<string notr="true">Number of Steps</string> | |
</property> | |
</widget> | |
</widget> | |
<resources/> | |
<connections/> | |
</ui> |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment