#!/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.SIGNAL("valueChanged(int)"), lcd, tqt.SLOT("display(int)"))
        self.connect(self.slider, tqt.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.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.SIGNAL("clicked()"), tqt.tqApp, tqt.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.SIGNAL("clicked()"), self.fire)
        self.connect(self.cannonField, tqt.PYSIGNAL("canShoot(bool)"), self.shoot, tqt.SLOT("setEnabled(bool)"))

        restart = tqt.TQPushButton("&New Game", self, "newgame")
        restart.setFont(tqt.TQFont("Times", 18, tqt.TQFont.Bold))
        self.connect(restart, tqt.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.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())