Homec4science
Diffusion glups (master) Browse

glups/NbodyMapWidget.py
61668ba8ebf3master

Last Change
Raw File

NbodyMapWidget.py
View Options

import sys
import math
from PyQt4 import QtCore, QtGui, QtOpenGL
from pNbody import *
from pNbody import libqt
import ImageQt as IQt
class NbodyMapWidget(QtGui.QWidget):
def __init__(self, parent=None):
QtGui.QWidget.__init__(self, parent)
self.setFocusPolicy(QtCore.Qt.StrongFocus) # Widget accepts focus (tab and mouse)
self.parent = parent
self.setMouseTracking(True) # see mouseMoveEvent
self.setCursor(QtCore.Qt.CrossCursor)
self.pixmapOffset = QtCore.QPoint()
self.pixmap = None
self.palette_pixmap = None
self.dyLabel = 20
self.x0Label = 5
self.y0Label = 5
self.colorLabel = QtCore.Qt.red
self.Mat = None
self.Min = 0
self.Max = 0
self.Cd = 0
self.CreateInfo()
########################################
# events
########################################
def keyPressEvent(self, event):
key = event.key()
if (key==QtCore.Qt.Key_D):
self.setCursor(QtCore.Qt.WaitCursor)
self.parent.Display(updateFromGL=False)
self.setCursor(QtCore.Qt.CrossCursor)
if (key==QtCore.Qt.Key_R):
self.setCursor(QtCore.Qt.WaitCursor)
self.parent.ReDisplay()
self.setCursor(QtCore.Qt.CrossCursor)
elif key==QtCore.Qt.Key_Escape:
sys.exit()
elif key==QtCore.Qt.Key_Q:
sys.exit()
def mousePressEvent(self, event):
if event.buttons() & QtCore.Qt.LeftButton:
self.lastPos = QtCore.QPoint(event.pos())
self.lastPosX = self.lastPos.x()
self.lastPosY = self.lastPos.y()
elif event.buttons() & QtCore.Qt.RightButton:
pass
elif event.buttons() & QtCore.Qt.MidButton:
pass
def mouseDoubleClickEvent(self,event):
if event.buttons() & QtCore.Qt.LeftButton:
pass
elif event.buttons() & QtCore.Qt.RightButton:
self.setCursor(QtCore.Qt.WaitCursor)
self.parent.ReDisplay()
self.setCursor(QtCore.Qt.CrossCursor)
elif event.buttons() & QtCore.Qt.MidButton:
pass
def mouseMoveEvent(self, event):
x = event.x()
y = event.y()
if self.Mat!=None:
shape = self.Mat.shape
x = min(shape[0]-1,x)
x = max(0,x)
y = min(shape[1]-1,y)
y = max(0,y)
val = self.Mat[x,y]
txt = "value = %g"%(val)
self.label.setText("""<font color="red">%s</font>"""%txt)
txt = "min=%10.3e max=%10.3e cd=%10.3e"%(self.Min,self.Max,self.Cd)
self.info.setText("""<font color="red">%s</font>"""%txt)
########################################
# palette
########################################
def DisplayPalette(self,p = "rainbow4"):
n = 512
matint = ( arange(n) * ones((64,n)) )/2
matint = matint.astype(uint8)
#imageQt = libqt.QNumarrayImage(matint, p)
imagePIL = get_image(matint,palette_name="rainbow4")
imageQt = IQt.ImageQt(imagePIL)
self.palette_pixmap = QtGui.QPixmap.fromImage(imageQt)
########################################
# display
########################################
def DisplayMap(self,mat,matint,mn,mx,cd, p = "rainbow4"):
self.Mat = mat
self.Min = mn
self.Max = mx
self.Cd = cd
self.info.setText("min=%10.3e max=%10.3e cd=%10.3e"%(self.Min,self.Max,self.Cd))
#imageQt = libqt.QNumarrayImage(matint, p)
imagePIL = get_image(matint,palette_name="rainbow4")
imageQt = IQt.ImageQt(imagePIL)
self.pixmap = QtGui.QPixmap.fromImage(imageQt)
self.DisplayPalette(p=p)
self.update()
def SaveMap(self,filename, mat,matint, p = "rainbow4"):
imageQt = libqt.QNumarrayImage(matint, p)
imageQt.save(filename)
########################################
# painter
########################################
def paintEvent(self, event):
if self.pixmap!=None:
painter = QtGui.QPainter(self)
painter.fillRect(self.rect(), QtCore.Qt.white)
painter.drawPixmap(self.pixmapOffset, self.pixmap)
if self.palette_pixmap!=None:
painter = QtGui.QPainter(self)
painter.fillRect(QtCore.QRect(), QtCore.Qt.white)
painter.drawPixmap(QtCore.QPoint(512,0), self.palette_pixmap,QtCore.QRect(0,0,64,512))
########################################
# info
########################################
def CreateInfo(self):
font = QtGui.QFont()
#font.setWeight(100)
#font.setBold(True)
self.label = QtGui.QLabel(self)
self.label.setGeometry(QtCore.QRect(self.x0Label,self.y0Label,512,18))
self.label.setFont(font)
self.label.setObjectName("label")
self.label.setText(QtGui.QApplication.translate("Form", "Observer position :", None, QtGui.QApplication.UnicodeUTF8))
txt = 'value = %g'%(0)
self.label.setText("""<font color="red">%s</font>"""%txt)
self.info = QtGui.QLabel(self)
self.info.setGeometry(QtCore.QRect(self.x0Label,self.y0Label+self.dyLabel,512,18))
self.info.setFont(font)
self.info.setObjectName("info")
self.info.setText(QtGui.QApplication.translate("Form", "Observer position :", None, QtGui.QApplication.UnicodeUTF8))
txt = "min=%10.3e max=%10.3e cd=%10.3e"%(self.Min,self.Max,self.Cd)
self.info.setText("""<font color="red">%s</font>"""%txt)
c4science · Help