include Math
require 'Qt'

class CannonField < Qt::Widget
    
    signals 'hit()', 'missed()', 'angleChanged(int)', 'forceChanged(int)'
    slots 'setAngle(int)', 'setForce(int)', 'shoot()', 'moveShot()', 'newTarget()'
    
    def initialize(parent = nil)
        super
        @currentAngle = 45
        @currentForce = 0
        @timerCount = 0
        @autoShootTimer = Qt::Timer.new( self )
        connect( @autoShootTimer, SIGNAL('timeout()'),
                 self, SLOT('moveShot()') );
        @shootAngle = 0
        @shootForce = 0
        @target = Qt::Point.new(0, 0)
        setPalette( Qt::Palette.new( Qt::Color.new( 250, 250, 200) ) )
	setAutoFillBackground(true)
        newTarget()
        @barrelRect = Qt::Rect.new(33, -4, 15, 8)
    end

    def setAngle( degrees )
        if degrees < 5
            degrees = 5
        elsif degrees > 70
            degrees = 70
        end
        if @currentAngle == degrees
            return
        end
        @currentAngle = degrees
        update( cannonRect() )
        emit angleChanged( @currentAngle )
    end
    
    def setForce( newton )
        if newton < 0
            newton = 0
        end
        if @currentForce == newton
            return
        end
        @currentForce = newton
        emit forceChanged( @currentForce )
    end
    
    def shoot()
        if @autoShootTimer.isActive()
            return
        end;
        @timerCount = 0
        @shootAngle = @currentAngle
        @shootForce = @currentForce
        @autoShootTimer.start( 50 )
    end

    @@currentForceirst_time = true
    
    def newTarget()
        if @@currentForceirst_time
            @@currentForceirst_time = false
            midnight = Qt::Time.new( 0, 0, 0 )
            srand( midnight.secsTo(Qt::Time.currentTime()) )
        end
        r = Qt::Region.new( targetRect() )
        @target = Qt::Point.new( 200 + rand(190),
                     10  + rand(255) )
        repaint( r.unite( Qt::Region.new(targetRect()) ) )
    end
    
    def moveShot()
        r = Qt::Region.new( shotRect() )
        @timerCount += 1

        shotR = shotRect()

        if shotR.intersects( targetRect() ) 
            @autoShootTimer.stop()
            emit hit()
        elsif shotR.x() > width() || shotR.y() > height()
            @autoShootTimer.stop()
            emit missed()
        else
            r = r.unite( Qt::Region.new( shotR ) )
        end
        
        update( r )
    end

    def paintEvent( e )
        painter = Qt::Painter.new( self )
        paintCannon( painter )
        if @autoShootTimer.isActive()
            paintShot( painter )
        end
        paintTarget(painter)
        painter.end()
    end

    def paintShot( painter )
        painter.setPen( Qt::NoPen )
        painter.setBrush( Qt::Brush.new(Qt::black) )
        painter.drawRect( shotRect() )
    end

    def paintTarget( painter )
        painter.setBrush( Qt::Brush.new(Qt::red) )
        painter.setPen( Qt::Pen.new(Qt::Color.new(Qt::black)) )
        painter.drawRect( targetRect() )
    end
    
    def paintCannon(painter)                
        painter.setPen(Qt::NoPen)
        painter.setBrush(Qt::Brush.new(Qt::blue))

        painter.save()
        painter.translate(0, height())
        painter.drawPie( Qt::Rect.new(-35, -35, 70, 70), 0, 90*16 )
        painter.rotate( - @currentAngle )
        painter.drawRect( @barrelRect )
        painter.restore()
    end

    def cannonRect()
        r = Qt::Rect.new( 0, 0, 50, 50)
        r.moveBottomLeft( rect().bottomLeft() )
        return r
    end
    
    def shotRect()
        gravity = 4.0

        time      = @timerCount / 4.0
        velocity  = @shootForce
        radians   = @shootAngle*3.14159265/180.0

        velx      = velocity*cos( radians )
        vely      = velocity*sin( radians )
        x0        = ( @barrelRect.right()  + 5.0 )*cos(radians)
        y0        = ( @barrelRect.right()  + 5.0 )*sin(radians)
        x         = x0 + velx*time
        y         = y0 + vely*time - 0.5*gravity*time*time

        r = Qt::Rect.new( 0, 0, 6, 6 );
        r.moveCenter( Qt::Point.new( x.round, height() - 1 - y.round ) )
        return r
    end

    def targetRect()
        r = Qt::Rect.new( 0, 0, 20, 10 )
        r.moveCenter( Qt::Point.new(@target.x(),height() - 1 - @target.y()) );
        return r
    end
end