You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytqt/examples/tut14.py

379 lines
12 KiB

#!/usr/bin/env python
# TQt tutorial 14.
import sys
import math
import random
from PyTQt import tqt
class LCDRange(tqt.TQWidget):
def __init__(self, s=None, parent=None, name=None):
tqt.TQWidget.__init__(self, parent, name)
lcd = tqt.TQLCDNumber(2, self, "lcd")
self.slider = tqt.TQSlider(tqt.TQt.Horizontal, self, "slider")
self.slider.setRange(0, 99)
self.slider.setValue(0)
self.label = tqt.TQLabel(" ", self, "label")
self.label.setAlignment(tqt.TQt.AlignCenter)
self.connect(self.slider, tqt.TQ_SIGNAL("valueChanged(int)"), lcd, tqt.TQ_SLOT("display(int)"))
self.connect(self.slider, tqt.TQ_SIGNAL("valueChanged(int)"), self, tqt.PYSIGNAL("valueChanged(int)"))
self.setFocusProxy(self.slider)
l = tqt.TQVBoxLayout(self)
l.addWidget(lcd, 1)
l.addWidget(self.slider)
l.addWidget(self.label)
if s is not None:
self.setText(s)
def value(self):
return self.slider.value()
def setValue(self, value):
self.slider.setValue(value)
def setRange(self, minVal, maxVal):
if minVal < 0 or maxVal > 99 or minVal > maxVal:
raise ValueError("LCDRange.setRange(): invalid range")
self.slider.setRange(minVal, maxVal)
def text(self):
return self.label.text()
def setText(self, s):
self.label.setText(s)
class CannonField(tqt.TQWidget):
def __init__(self, parent=None, name=None):
tqt.TQWidget.__init__(self, parent, name)
self.ang = 45
self.f = 0
self.timerCount = 0
self.autoShootTimer = tqt.TQTimer(self, "movement handler")
self.connect(self.autoShootTimer, tqt.TQ_SIGNAL("timeout()"), self.moveShot)
self.shoot_ang = 0
self.shoot_f = 0
self.target = tqt.TQPoint(0, 0)
self.gameEnded = 0
self.barrelPressed = 0
self.setPalette(tqt.TQPalette(tqt.TQColor(250, 250, 200)))
self.barrelRect = tqt.TQRect(33, -4, 15, 8)
self.newTarget()
def angle(self):
return self.ang
def setAngle(self, degrees):
if degrees < 5:
degrees = 5
if degrees > 70:
degrees = 70
if self.ang == degrees:
return
self.ang = degrees
self.repaint(self.cannonRect(), 0)
self.emit(tqt.PYSIGNAL("angleChanged(int)"), (self.ang, ))
def force(self):
return self.f
def setForce(self, newton):
if newton < 0:
newton = 0
if self.f == newton:
return
self.f = newton
self.emit(tqt.PYSIGNAL("forceChanged(int)"), (self.f, ))
def shoot(self):
if self.isShooting():
return
self.timerCount = 0
self.shoot_ang = self.ang
self.shoot_f = self.f
self.autoShootTimer.start(50)
self.emit(tqt.PYSIGNAL("canShoot(bool)"), (0, ))
def newTarget(self):
r = tqt.TQRegion(self.targetRect())
self.target = tqt.TQPoint(random.randint(200, 390), random.randint(10, 265))
self.repaint(r.unite(tqt.TQRegion(self.targetRect())))
def gameOver(self):
return self.gameEnded
def setGameOver(self):
if self.gameEnded:
return
if self.isShooting():
self.autoShootTime.stop()
self.gameEnded = 1
self.repaint()
def restartGame(self):
if self.isShooting():
self.autoShootTime.stop()
self.gameEnded = 0
self.repaint()
self.emit(tqt.PYSIGNAL("canShoot(bool)"), (1, ))
def moveShot(self):
r = tqt.TQRegion(self.shotRect())
self.timerCount = self.timerCount + 1
shotR = self.shotRect()
if shotR.intersects(self.targetRect()):
self.autoShootTimer.stop()
self.emit(tqt.PYSIGNAL("hit()"), ())
self.emit(tqt.PYSIGNAL("canShoot(bool)"), (1, ))
elif shotR.x() > self.width() or shotR.y() > self.height() or shotR.intersects(self.barrierRect()):
self.autoShootTimer.stop()
self.emit(tqt.PYSIGNAL("missed()"), ())
self.emit(tqt.PYSIGNAL("canShoot(bool)"), (1, ))
else:
r = r.unite(tqt.TQRegion(shotR))
self.repaint(r)
def mousePressEvent(self, ev):
if ev.button() != tqt.TQt.LeftButton:
return
if self.barrelHit(ev.pos()):
self.barrelPressed = 1
def mouseMoveEvent(self, ev):
if not self.barrelPressed:
return
pnt = ev.pos()
if pnt.x() <= 0:
pnt.setX(1)
if pnt.y() >= self.height():
pnt.setY(self.height() - 1)
rad = math.atan(float(self.rect().bottom() - pnt.y()) / pnt.x())
self.setAngle(int(round(rad * 180 / math.pi)))
def mouseReleaseEvent(self, ev):
if ev.button() == tqt.TQt.LeftButton:
self.barrelPressed = 0
def paintEvent(self, ev):
updateR = ev.rect()
p = tqt.TQPainter(self)
if self.gameEnded:
p.setPen(tqt.TQt.black)
p.setFont(tqt.TQFont("Courier", 48, tqt.TQFont.Bold))
p.drawText(self.rect(), tqt.TQt.AlignCenter, "Game Over")
if updateR.intersects(self.cannonRect()):
self.paintCannon(p)
if updateR.intersects(self.barrierRect()):
self.paintBarrier(p)
if self.isShooting() and updateR.intersects(self.shotRect()):
self.paintShot(p)
if not self.gameEnded and updateR.intersects(self.targetRect()):
self.paintTarget(p)
def paintShot(self, p):
p.setBrush(tqt.TQt.black)
p.setPen(tqt.TQt.NoPen)
p.drawRect(self.shotRect())
def paintTarget(self, p):
p.setBrush(tqt.TQt.red)
p.setPen(tqt.TQt.black)
p.drawRect(self.targetRect())
def paintBarrier(self, p):
p.setBrush(tqt.TQt.yellow)
p.setPen(tqt.TQt.black)
p.drawRect(self.barrierRect())
def paintCannon(self, p):
cr = self.cannonRect()
pix = tqt.TQPixmap(cr.size())
pix.fill(self, cr.topLeft())
tmp = tqt.TQPainter(pix)
tmp.setBrush(tqt.TQt.blue)
tmp.setPen(tqt.TQt.NoPen)
tmp.translate(0, pix.height() - 1)
tmp.drawPie(tqt.TQRect(-35, -35, 70, 70), 0, 90 * 16)
tmp.rotate(-self.ang)
tmp.drawRect(self.barrelRect)
tmp.end()
p.drawPixmap(cr.topLeft(), pix)
def cannonRect(self):
r = tqt.TQRect(0, 0, 50, 50)
r.moveBottomLeft(self.rect().bottomLeft())
return r
def shotRect(self):
gravity = 4.0
time = self.timerCount / 4.0
velocity = self.shoot_f
radians = self.shoot_ang * math.pi / 180
velx = velocity * math.cos(radians)
vely = velocity * math.sin(radians)
x0 = (self.barrelRect.right() + 5) * math.cos(radians)
y0 = (self.barrelRect.right() + 5) * math.sin(radians)
x = x0 + velx * time
y = y0 + vely * time - 0.5 * gravity * time * time
r = tqt.TQRect(0, 0, 6, 6)
r.moveCenter(tqt.TQPoint(int(x), int(self.height() - 1 - y)))
return r
def targetRect(self):
r = tqt.TQRect(0, 0, 20, 10)
r.moveCenter(tqt.TQPoint(self.target.x(), self.height() - 1 - self.target.y()))
return r
def barrierRect(self):
return tqt.TQRect(145, self.height() - 100, 15, 100)
def barrelHit(self, p):
mtx = tqt.TQWMatrix()
mtx.translate(0, self.height() - 1)
mtx.rotate(-self.ang)
(mtx, invertable) = mtx.invert()
return self.barrelRect.contains(mtx.map(p))
def isShooting(self):
return self.autoShootTimer.isActive()
def sizePolicy(self):
return tqt.TQSizePolicy(tqt.TQSizePolicy.Expanding, tqt.TQSizePolicy.Expanding)
class GameBoard(tqt.TQWidget):
def __init__(self, parent=None, name=None):
tqt.TQWidget.__init__(self, parent, name)
quit = tqt.TQPushButton("&Quit", self, "quit")
quit.setFont(tqt.TQFont("Times", 18, tqt.TQFont.Bold))
self.connect(quit, tqt.TQ_SIGNAL("clicked()"), tqt.tqApp, tqt.TQ_SLOT("quit()"))
self.angle = LCDRange("ANGLE", self, "angle")
self.angle.setRange(5, 70)
self.force = LCDRange("FORCE", self, "force")
self.force.setRange(10, 50)
box = tqt.TQVBox(self, "cannonFrame")
box.setFrameStyle(tqt.TQFrame.WinPanel | tqt.TQFrame.Sunken)
self.cannonField = CannonField(box, "cannonField")
self.connect(self.angle, tqt.PYSIGNAL("valueChanged(int)"), self.cannonField.setAngle)
self.connect(self.cannonField, tqt.PYSIGNAL("angleChanged(int)"), self.angle.setValue)
self.connect(self.force, tqt.PYSIGNAL("valueChanged(int)"), self.cannonField.setForce)
self.connect(self.cannonField, tqt.PYSIGNAL("forceChanged(int)"), self.force.setValue)
self.connect(self.cannonField, tqt.PYSIGNAL("hit()"), self.hit)
self.connect(self.cannonField, tqt.PYSIGNAL("missed()"), self.missed)
self.shoot = tqt.TQPushButton("&Shoot", self, "shoot")
self.shoot.setFont(tqt.TQFont("Times", 18, tqt.TQFont.Bold))
self.connect(self.shoot, tqt.TQ_SIGNAL("clicked()"), self.fire)
self.connect(self.cannonField, tqt.PYSIGNAL("canShoot(bool)"), self.shoot, tqt.TQ_SLOT("setEnabled(bool)"))
restart = tqt.TQPushButton("&New Game", self, "newgame")
restart.setFont(tqt.TQFont("Times", 18, tqt.TQFont.Bold))
self.connect(restart, tqt.TQ_SIGNAL("clicked()"), self.newGame)
self.hits = tqt.TQLCDNumber(2, self, "hits")
self.shotsLeft = tqt.TQLCDNumber(2, self, "shotsleft")
hitsL = tqt.TQLabel("HITS", self, "hitsLabel")
shotsLeftL = tqt.TQLabel("SHOTS LEFT", self, "shotsleftLabel")
accel = tqt.TQAccel(self)
accel.connectItem(accel.insertItem(tqt.TQt.Key_Enter), self.fire)
accel.connectItem(accel.insertItem(tqt.TQt.Key_Return), self.fire)
accel.connectItem(accel.insertItem(tqt.TQt.CTRL + tqt.TQt.Key_Q), tqt.tqApp, tqt.TQ_SLOT("quit()"))
grid = tqt.TQGridLayout(self, 2, 2, 10)
grid.addWidget(quit, 0, 0)
grid.addWidget(box, 1, 1)
grid.setColStretch(1, 10)
leftBox = tqt.TQVBoxLayout()
grid.addLayout(leftBox, 1, 0)
leftBox.addWidget(self.angle)
leftBox.addWidget(self.force)
topBox = tqt.TQHBoxLayout()
grid.addLayout(topBox, 0, 1)
topBox.addWidget(self.shoot)
topBox.addWidget(self.hits)
topBox.addWidget(hitsL)
topBox.addWidget(self.shotsLeft)
topBox.addWidget(shotsLeftL)
topBox.addStretch(1)
topBox.addWidget(restart)
self.angle.setValue(60)
self.force.setValue(25)
self.angle.setFocus()
self.newGame()
def fire(self):
if self.cannonField.gameOver() or self.cannonField.isShooting():
return
self.shotsLeft.display(self.shotsLeft.intValue() - 1)
self.cannonField.shoot()
def hit(self):
self.hits.display(self.hits.intValue() + 1)
if self.shotsLeft.intValue() == 0:
self.cannonField.setGameOver()
else:
self.cannonField.newTarget()
def missed(self):
if self.shotsLeft.intValue() == 0:
self.cannonField.setGameOver()
def newGame(self):
self.shotsLeft.display(15)
self.hits.display(0)
self.cannonField.restartGame()
self.cannonField.newTarget()
tqt.TQApplication.setColorSpec(tqt.TQApplication.CustomColor)
a = tqt.TQApplication(sys.argv)
gb = GameBoard()
gb.setGeometry(100, 100, 500, 355)
a.setMainWidget(gb)
gb.show()
sys.exit(a.exec_loop())